Add two more examples: nfs-ls and nfs-cp
authorRonnie Sahlberg <ronniesahlberg@gmail.com>
Tue, 16 Apr 2013 04:41:38 +0000 (21:41 -0700)
committerRonnie Sahlberg <ronniesahlberg@gmail.com>
Tue, 16 Apr 2013 04:41:38 +0000 (21:41 -0700)
examples/Makefile.am
examples/nfs-cp.c [new file with mode: 0644]
examples/nfs-ls.c [new file with mode: 0644]

index fec0ff3a5161789fce02ea3522a61eef870350e0..200ba115fb44b7898d904b9522628e0c513601ca 100644 (file)
@@ -1,4 +1,4 @@
-noinst_PROGRAMS = nfsclient-async nfsclient-raw nfsclient-sync nfsclient-bcast nfsclient-listservers
+noinst_PROGRAMS = nfsclient-async nfsclient-raw nfsclient-sync nfsclient-bcast nfsclient-listservers nfs-ls nfs-cp
 
 AM_CPPFLAGS = \
        -I$(abs_top_srcdir)/include \
@@ -11,6 +11,10 @@ AM_CPPFLAGS = \
 
 AM_LDFLAGS = ../lib/.libs/libnfs.la -lpopt
 
+nfs_ls_SOURCES = nfs-ls.c
+
+nfs_cp_SOURCES = nfs-cp.c
+
 nfsclient_async_SOURCES = nfsclient-async.c
 
 nfsclient_raw_SOURCES = nfsclient-raw.c
diff --git a/examples/nfs-cp.c b/examples/nfs-cp.c
new file mode 100644 (file)
index 0000000..53e22d6
--- /dev/null
@@ -0,0 +1,302 @@
+/* 
+   Copyright (C) by Ronnie Sahlberg <ronniesahlberg@gmail.com> 2013
+   
+   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
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#ifdef AROS
+#include "aros_compat.h"
+#endif
+
+
+#ifdef WIN32
+#include "win32_compat.h"
+#else
+#include <sys/stat.h>
+#include <string.h>
+#endif
+#ifdef HAVE_POLL_H
+#include <poll.h>
+#endif
+
+#ifdef HAVE_UNISTD_H
+#include <unistd.h>
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <sys/types.h>
+#include <fcntl.h>
+#include "libnfs-zdr.h"
+#include "libnfs.h"
+#include "libnfs-raw.h"
+#include "libnfs-raw-mount.h"
+
+struct file_context {
+       int is_nfs;
+       int fd;
+       struct nfs_context *nfs;
+       struct nfsfh *nfsfh;
+};
+
+void usage(void)
+{
+       fprintf(stderr, "Usage: nfs-cp <src> <dst>\n");
+       fprintf(stderr, "<src>,<dst> can either be a local file or "
+                       "an nfs URL.\n");
+       exit(0);
+}
+
+static void
+free_file_context(struct file_context *file_context)
+{
+       if (file_context->fd != -1) {
+               close(file_context->fd);
+       }
+       if (file_context->nfsfh != NULL) {
+               nfs_close(file_context->nfs, file_context->nfsfh);
+       }
+       if (file_context->nfs != NULL) {
+               nfs_destroy_context(file_context->nfs);
+       }
+       free(file_context);
+}
+
+static int
+fstat_file(struct file_context *fc, struct stat *st)
+{
+       if (fc->is_nfs == 0) {
+               return fstat(fc->fd, st);
+       } else {
+               return nfs_fstat(fc->nfs, fc->nfsfh, st);
+       }
+}
+
+static int64_t
+file_pread(struct file_context *fc, char *buf, int64_t count, uint64_t off)
+{
+       if (fc->is_nfs == 0) {
+               return pread(fc->fd, buf, count, off);
+       } else {
+               return nfs_pread(fc->nfs, fc->nfsfh, off, count, buf);
+       }
+}
+
+static int64_t
+file_pwrite(struct file_context *fc, char *buf, int64_t count, uint64_t off)
+{
+       if (fc->is_nfs == 0) {
+               return pwrite(fc->fd, buf, count, off);
+       } else {
+               return nfs_pwrite(fc->nfs, fc->nfsfh, off, count, buf);
+       }
+}
+
+static struct file_context *
+open_file(const char *url, int flags)
+{
+       struct file_context *file_context;
+       char *server = NULL, *path = NULL, *file = NULL, *strp;
+
+       file_context = malloc(sizeof(struct file_context));
+       if (file_context == NULL) {
+               fprintf(stderr, "Failed to malloc file_context\n");
+               return NULL;
+       }
+       file_context->is_nfs = 0;
+       file_context->fd     = -1;
+       file_context->nfs    = NULL;
+       file_context->nfsfh  = NULL;
+       
+
+       if (strncmp(url, "nfs://", 6)) {
+               file_context->is_nfs = 0;
+               file_context->fd = open(url, flags, 0660);
+               if (file_context->fd == -1) {           
+                       fprintf(stderr, "Failed to open %s\n", url);
+                       free_file_context(file_context);
+                       return NULL;
+               }
+               return file_context;
+       }
+
+       file_context->is_nfs = 1;
+
+       file_context->nfs = nfs_init_context();
+       if (file_context->nfs == NULL) {
+               fprintf(stderr, "failed to init context\n");
+               free_file_context(file_context);
+               return NULL;
+       }
+
+       server = strdup(url + 6);
+       if (server == NULL) {
+               fprintf(stderr, "Failed to strdup server string\n");
+               free_file_context(file_context);
+               return NULL;
+       }
+       if (server[0] == '/' || server[0] == '\0') {
+               fprintf(stderr, "Invalid server string.\n");
+               free(server);
+               free_file_context(file_context);
+               return NULL;
+       }
+       strp = strchr(server, '/');
+       path = strdup(strp);
+       *strp = 0;
+       if (path == NULL) {
+               fprintf(stderr, "Invalid URL specified.\n");
+               free(server);
+               free_file_context(file_context);
+               return NULL;
+       }
+
+       strp = strrchr(path, '/');
+       if (strp == NULL) {
+               fprintf(stderr, "Invalid URL specified.\n");
+               free(path);
+               free(server);
+               free_file_context(file_context);
+               return NULL;
+       }
+       file = strdup(strp);
+       *strp = 0;
+
+       if (nfs_mount(file_context->nfs, server, path) != 0) {
+               fprintf(stderr, "Failed to mount nfs share : %s\n",
+                              nfs_get_error(file_context->nfs));
+               free(file);
+               free(path);
+               free(server);
+               free_file_context(file_context);
+               return NULL;
+       }
+
+       if (flags == O_RDONLY) {
+               if (nfs_open(file_context->nfs, file, flags,
+                               &file_context->nfsfh) != 0) {
+                       fprintf(stderr, "Failed to open file : %s\n",
+                                      nfs_get_error(file_context->nfs));
+                       free(file);
+                       free(path);
+                       free(server);
+                       free_file_context(file_context);
+                       return NULL;
+               }
+       } else {
+               if (nfs_creat(file_context->nfs, file, 0660,
+                               &file_context->nfsfh) != 0) {
+                       fprintf(stderr, "Failed to creat file %s  %s\n", file,
+                                      nfs_get_error(file_context->nfs));
+                       free(file);
+                       free(path);
+                       free(server);
+                       free_file_context(file_context);
+                       return NULL;
+               }
+       }
+
+       free(file);
+       free(path);
+       free(server);
+
+       return file_context;
+}
+
+#define BUFSIZE 1024*1024
+static char buf[BUFSIZE];
+
+int main(int argc, char *argv[])
+{
+       int ret;
+       struct stat st;
+       struct file_context *src;
+       struct file_context *dst;
+       uint64_t off;
+       int64_t count;
+       
+#ifdef WIN32
+       if (WSAStartup(MAKEWORD(2,2), &wsaData) != 0) {
+               printf("Failed to start Winsock2\n");
+               return 10;
+       }
+#endif
+
+#ifdef AROS
+       aros_init_socket();
+#endif
+
+       if (argc != 3) {
+               usage();
+       }
+
+       src = open_file(argv[1], O_RDONLY);
+       if (src == NULL) {
+               fprintf(stderr, "Failed to open %s\n", argv[1]);
+               return 10;
+       }
+
+       dst = open_file(argv[2], O_WRONLY|O_CREAT|O_TRUNC);
+       if (dst == NULL) {
+               fprintf(stderr, "Failed to open %s\n", argv[2]);
+               free_file_context(src);
+               return 10;
+       }
+
+       if (fstat_file(src, &st) != 0) {
+               fprintf(stderr, "Failed to fstat source file\n");
+               free_file_context(src);
+               free_file_context(dst);
+               return 10;
+       }
+
+       off = 0;
+       while (off < st.st_size) {
+               count = st.st_size - off;
+               if (count > BUFSIZE) {
+                       count = BUFSIZE;
+               }
+               count = file_pread(src, buf, count, off);
+               if (count < 0) {
+                       fprintf(stderr, "Failed to read from source file\n");
+                       free_file_context(src);
+                       free_file_context(dst);
+                       return 10;
+               }
+               count = file_pwrite(dst, buf, count, off);
+               if (count < 0) {
+                       fprintf(stderr, "Failed to write to dest file\n");
+                       free_file_context(src);
+                       free_file_context(dst);
+                       return 10;
+               }
+
+               off += count;
+       }
+       printf("copied %d bytes\n", (int)count);
+
+       free_file_context(src);
+       free_file_context(dst);
+
+       return 0;
+}
diff --git a/examples/nfs-ls.c b/examples/nfs-ls.c
new file mode 100644 (file)
index 0000000..c4b9fa9
--- /dev/null
@@ -0,0 +1,225 @@
+/* 
+   Copyright (C) by Ronnie Sahlberg <ronniesahlberg@gmail.com> 2010
+   
+   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
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#ifdef AROS
+#include "aros_compat.h"
+#endif
+#ifdef WIN32
+#include "win32_compat.h"
+#pragma comment(lib, "ws2_32.lib")
+WSADATA wsaData;
+#else
+#include <string.h>
+#include <sys/stat.h>
+#ifndef AROS
+#include <sys/statvfs.h>
+#endif
+#endif
+
+#ifdef HAVE_UNISTD_H
+#include <unistd.h>
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <inttypes.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include "libnfs-zdr.h"
+#include "libnfs.h"
+#include "libnfs-raw.h"
+#include "libnfs-raw-mount.h"
+
+struct client {
+       char *server;
+       char *export;
+       uint32_t mount_port;
+       int is_finished;
+};
+
+void print_usage(void)
+{
+       fprintf(stderr, "Usage: nfs-ls [-?|--help] [--usage] <url>\n");
+}
+
+int main(int argc, char *argv[])
+{
+       struct nfs_context *nfs = NULL;
+       int i, ret, res;
+       uint64_t offset;
+       struct client client;
+       struct stat st;
+       struct nfsfh  *nfsfh;
+       struct nfsdir *nfsdir;
+       struct nfsdirent *nfsdirent;
+       struct statvfs svfs;
+       exports export, tmp;
+       const char *url = NULL;
+       char *server = NULL, *path = NULL, *strp;
+
+#ifdef WIN32
+       if (WSAStartup(MAKEWORD(2,2), &wsaData) != 0) {
+               printf("Failed to start Winsock2\n");
+               exit(10);
+       }
+#endif
+
+#ifdef AROS
+       aros_init_socket();
+#endif
+
+       url = argv[1];
+
+       if (url == NULL) {
+               fprintf(stderr, "No URL specified.\n");
+               print_usage();
+               exit(0);
+       }
+
+       if (strncmp(url, "nfs://", 6)) {
+               fprintf(stderr, "Invalid URL specified.\n");
+               print_usage();
+               exit(0);
+       }
+
+       server = strdup(url + 6);
+       if (server == NULL) {
+               fprintf(stderr, "Failed to strdup server string\n");
+               exit(10);
+       }
+       if (server[0] == '/' || server[0] == '\0') {
+               fprintf(stderr, "Invalid server string.\n");
+               free(server);
+               exit(10);
+       }
+       strp = strchr(server, '/');
+       if (strp == NULL) {
+               fprintf(stderr, "Invalid URL specified.\n");
+               print_usage();
+               free(server);
+               exit(0);
+       }
+       path = strdup(strp);
+       if (path == NULL) {
+               fprintf(stderr, "Failed to strdup server string\n");
+               free(server);
+               exit(10);
+       }
+       if (path[0] != '/') {
+               fprintf(stderr, "Invalid path.\n");
+               free(server);
+               free(path);
+               exit(10);
+       }
+       *strp = 0;
+       
+       client.server = server;
+       client.export = path;
+       client.is_finished = 0;
+
+
+       nfs = nfs_init_context();
+       if (nfs == NULL) {
+               printf("failed to init context\n");
+               goto finished;
+       }
+
+       ret = nfs_mount(nfs, client.server, client.export);
+       if (ret != 0) {
+               printf("Failed to mount nfs share : %s\n", nfs_get_error(nfs));
+               goto finished;
+       }
+
+
+       ret = nfs_opendir(nfs, "/", &nfsdir);
+       if (ret != 0) {
+               printf("Failed to opendir(\"/\")\n", nfs_get_error(nfs));
+               exit(10);
+       }
+       while((nfsdirent = nfs_readdir(nfs, nfsdir)) != NULL) {
+               char path[1024];
+
+               if (!strcmp(nfsdirent->name, ".") || !strcmp(nfsdirent->name, "..")) {
+                       continue;
+               }
+
+               snprintf(path, 1024, "%s/%s", "/", nfsdirent->name);
+               ret = nfs_stat(nfs, path, &st);
+               if (ret != 0) {
+                       fprintf(stderr, "Failed to stat(%s) %s\n", path, nfs_get_error(nfs));
+                       continue;
+               }
+
+               switch (st.st_mode & S_IFMT) {
+               case S_IFLNK:
+               case S_IFREG:
+                       printf("-");
+                       break;
+               case S_IFDIR:
+                       printf("d");
+                       break;
+               case S_IFCHR:
+                       printf("c");
+                       break;
+               case S_IFBLK:
+                       printf("b");
+                       break;
+               }
+               printf("%c%c%c",
+                       "-r"[!!(st.st_mode & S_IRUSR)],
+                       "-w"[!!(st.st_mode & S_IWUSR)],
+                       "-x"[!!(st.st_mode & S_IXUSR)]
+               );
+               printf("%c%c%c",
+                       "-r"[!!(st.st_mode & S_IRGRP)],
+                       "-w"[!!(st.st_mode & S_IWGRP)],
+                       "-x"[!!(st.st_mode & S_IXGRP)]
+               );
+               printf("%c%c%c",
+                       "-r"[!!(st.st_mode & S_IROTH)],
+                       "-w"[!!(st.st_mode & S_IWOTH)],
+                       "-x"[!!(st.st_mode & S_IXOTH)]
+               );
+               printf(" %2d", st.st_nlink);
+               printf(" %5d", st.st_uid);
+               printf(" %5d", st.st_gid);
+               printf(" %12" PRId64, st.st_size);
+
+               printf(" %s\n", nfsdirent->name);
+       }
+       nfs_closedir(nfs, nfsdir);
+
+
+finished:
+       free(server);
+       free(path);
+       if (nfs != NULL) {              
+               nfs_destroy_context(nfs);
+       }
+       return 0;
+}
+