libnfs/tests/prog_open_write.c

131 lines
3.2 KiB
C

/* -*- mode:c; tab-width:8; c-basic-offset:8; indent-tabs-mode:nil; -*- */
/*
Copyright (C) by Ronnie Sahlberg <ronniesahlberg@gmail.com> 2017
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, see <http://www.gnu.org/licenses/>.
*/
#define _FILE_OFFSET_BITS 64
#define _GNU_SOURCE
#include <fcntl.h>
#include <inttypes.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include "libnfs.h"
void usage(void)
{
fprintf(stderr, "Usage: prog_open_write <url> <cwd> <path> <flags>"
"\t<data>\n");
exit(1);
}
int main(int argc, char *argv[])
{
struct nfs_context *nfs = NULL;
struct nfs_url *url = NULL;
int ret = 0;
int flags = 0, count;
struct nfsfh *fh;
if (argc != 6) {
usage();
}
if (strstr(argv[4], "O_RDONLY")) {
flags |= O_RDONLY;
}
if (strstr(argv[4], "O_RDWR")) {
flags |= O_RDWR;
}
if (strstr(argv[4], "O_WRONLY")) {
flags |= O_WRONLY;
}
if (strstr(argv[4], "O_NOFOLLOW")) {
flags |= O_NOFOLLOW;
}
if (strstr(argv[4], "O_APPEND")) {
flags |= O_APPEND;
}
if (strstr(argv[4], "O_TRUNC")) {
flags |= O_TRUNC;
}
if (strstr(argv[4], "O_CREAT")) {
flags |= O_CREAT;
}
nfs = nfs_init_context();
if (nfs == NULL) {
printf("failed to init context\n");
exit(1);
}
url = nfs_parse_url_full(nfs, argv[1]);
if (url == NULL) {
fprintf(stderr, "%s\n", nfs_get_error(nfs));
exit(1);
}
if (nfs_mount(nfs, url->server, url->path) != 0) {
fprintf(stderr, "Failed to mount nfs share : %s\n",
nfs_get_error(nfs));
ret = 1;
goto finished;
}
if (nfs_chdir(nfs, argv[2]) != 0) {
fprintf(stderr, "Failed to chdir to \"%s\" : %s\n",
argv[2], nfs_get_error(nfs));
ret = 1;
goto finished;
}
if (nfs_open(nfs, argv[3],flags, &fh)) {
fprintf(stderr, "Failed to open(): %s\n",
nfs_get_error(nfs));
ret = 1;
goto finished;
}
if (strlen(argv[5]) > 0) {
count = nfs_write(nfs, fh, strlen(argv[5]), argv[5]);
if (count < 0) {
fprintf(stderr, "Failed to write(): %s\n",
nfs_get_error(nfs));
ret = 1;
goto finished;
}
}
if (nfs_close(nfs, fh)) {
fprintf(stderr, "Failed to close(): %s\n",
nfs_get_error(nfs));
ret = 1;
goto finished;
}
finished:
nfs_destroy_url(url);
nfs_destroy_context(nfs);
return ret;
}