since UDP is unreliable, send the CALLIT broadcasts 3 times, one second apart when...
[deb_libnfs.git] / lib / libnfs-sync.c
index 88b89ee6ee5870e90f15f7e56f69f41f40e25531..e964c144b17fc23a164a5913792bd43450d8f447 100644 (file)
@@ -18,6 +18,7 @@
  * High level api to nfs filesystems
  */
 
+#include "config.h"
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <sys/statvfs.h>
+#include <sys/ioctl.h>
 #include <unistd.h>
 #include <fcntl.h>
 #include <errno.h>
 #include <poll.h>
+#include <sys/socket.h>
+#include <net/if.h>
+#include <netdb.h>
 #include "libnfs.h"
 #include "libnfs-raw.h"
 #include "libnfs-raw-mount.h"
@@ -44,7 +49,7 @@ struct sync_cb_data {
 };
 
 
-static void wait_for_reply(struct nfs_context *nfs, struct sync_cb_data *cb_data)
+static void wait_for_reply(struct rpc_context *rpc, struct sync_cb_data *cb_data)
 {
        struct pollfd pfd;
 
@@ -52,16 +57,16 @@ static void wait_for_reply(struct nfs_context *nfs, struct sync_cb_data *cb_data
                if (cb_data->is_finished) {
                        break;
                }
-               pfd.fd = nfs_get_fd(nfs);
-               pfd.events = nfs_which_events(nfs);
+               pfd.fd = rpc_get_fd(rpc);
+               pfd.events = rpc_which_events(rpc);
 
                if (poll(&pfd, 1, -1) < 0) {
-                       nfs_set_error(nfs, "Poll failed");
+                       rpc_set_error(rpc, "Poll failed");
                        cb_data->status = -EIO;
                        break;
                }
-               if (nfs_service(nfs, pfd.revents) < 0) {
-                       nfs_set_error(nfs, "nfs_service failed");
+               if (rpc_service(rpc, pfd.revents) < 0) {
+                       rpc_set_error(rpc, "rpc_service failed");
                        cb_data->status = -EIO;
                        break;
                }
@@ -100,7 +105,7 @@ int nfs_mount(struct nfs_context *nfs, const char *server, const char *export)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -136,7 +141,7 @@ int nfs_stat(struct nfs_context *nfs, const char *path, struct stat *st)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -177,7 +182,7 @@ int nfs_open(struct nfs_context *nfs, const char *path, int mode, struct nfsfh *
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -216,7 +221,7 @@ int nfs_pread(struct nfs_context *nfs, struct nfsfh *nfsfh, off_t offset, size_t
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -255,7 +260,7 @@ int nfs_close(struct nfs_context *nfs, struct nfsfh *nfsfh)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -278,7 +283,7 @@ int nfs_fstat(struct nfs_context *nfs, struct nfsfh *nfsfh, struct stat *st)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -310,7 +315,7 @@ int nfs_pwrite(struct nfs_context *nfs, struct nfsfh *nfsfh, off_t offset, size_
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -350,7 +355,7 @@ int nfs_fsync(struct nfs_context *nfs, struct nfsfh *nfsfh)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -384,7 +389,7 @@ int nfs_ftruncate(struct nfs_context *nfs, struct nfsfh *nfsfh, off_t length)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -417,7 +422,7 @@ int nfs_truncate(struct nfs_context *nfs, const char *path, off_t length)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -452,7 +457,7 @@ int nfs_mkdir(struct nfs_context *nfs, const char *path)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -487,7 +492,7 @@ int nfs_rmdir(struct nfs_context *nfs, const char *path)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -527,7 +532,7 @@ int nfs_creat(struct nfs_context *nfs, const char *path, int mode, struct nfsfh
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -562,7 +567,7 @@ int nfs_unlink(struct nfs_context *nfs, const char *path)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -602,7 +607,7 @@ int nfs_opendir(struct nfs_context *nfs, const char *path, struct nfsdir **nfsdi
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -640,7 +645,7 @@ int nfs_lseek(struct nfs_context *nfs, struct nfsfh *nfsfh, off_t offset, int wh
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -677,7 +682,7 @@ int nfs_statvfs(struct nfs_context *nfs, const char *path, struct statvfs *svfs)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -723,7 +728,7 @@ int nfs_readlink(struct nfs_context *nfs, const char *path, char *buf, int bufsi
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -757,7 +762,7 @@ int nfs_chmod(struct nfs_context *nfs, const char *path, int mode)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -792,7 +797,7 @@ int nfs_fchmod(struct nfs_context *nfs, struct nfsfh *nfsfh, int mode)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -827,7 +832,7 @@ int nfs_chown(struct nfs_context *nfs, const char *path, int uid, int gid)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -859,7 +864,7 @@ int nfs_fchown(struct nfs_context *nfs, struct nfsfh *nfsfh, int uid, int gid)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -893,7 +898,7 @@ int nfs_utimes(struct nfs_context *nfs, const char *path, struct timeval *times)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -927,7 +932,7 @@ int nfs_utime(struct nfs_context *nfs, const char *path, struct utimbuf *times)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -962,7 +967,7 @@ int nfs_access(struct nfs_context *nfs, const char *path, int mode)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -996,7 +1001,7 @@ int nfs_symlink(struct nfs_context *nfs, const char *oldpath, const char *newpat
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -1030,7 +1035,7 @@ int nfs_rename(struct nfs_context *nfs, const char *oldpath, const char *newpath
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
@@ -1064,7 +1069,276 @@ int nfs_link(struct nfs_context *nfs, const char *oldpath, const char *newpath)
                return -1;
        }
 
-       wait_for_reply(nfs, &cb_data);
+       wait_for_reply(nfs_get_rpc_context(nfs), &cb_data);
 
        return cb_data.status;
 }
+
+void mount_getexports_cb(struct rpc_context *mount_context, int status, void *data, void *private_data)
+{
+       struct sync_cb_data *cb_data = private_data;
+       exports export = *(exports *)data;
+
+       cb_data->is_finished = 1;
+       cb_data->status = status;
+       cb_data->return_data = NULL;
+
+       if (status != 0) {
+               rpc_set_error(mount_context, "mount/export call failed with \"%s\"", (char *)data);
+               return;
+       }
+
+       while (export != NULL) {
+               exports new_export;
+
+               new_export = malloc(sizeof(*new_export));
+               memset(new_export, 0, sizeof(*new_export));
+               new_export->ex_dir  = strdup(export->ex_dir);
+               new_export->ex_next = cb_data->return_data;
+
+               cb_data->return_data = new_export;
+
+               export = export->ex_next;
+       }
+}
+
+struct exportnode *mount_getexports(const char *server)
+{
+       struct sync_cb_data cb_data;
+       struct rpc_context *rpc;
+
+
+       cb_data.is_finished = 0;
+       cb_data.return_data = NULL;
+
+       rpc = rpc_init_context();
+       if (mount_getexports_async(rpc, server, mount_getexports_cb, &cb_data) != 0) {
+               rpc_destroy_context(rpc);
+               return NULL;
+       }
+
+       wait_for_reply(rpc, &cb_data);
+       rpc_destroy_context(rpc);
+
+       return cb_data.return_data;
+}
+
+void mount_free_export_list(struct exportnode *exports)
+{
+       struct exportnode *tmp;
+
+       while ((tmp = exports)) {
+               exports = exports->ex_next;
+               free(tmp->ex_dir);
+               free(tmp);
+       }
+}
+
+
+
+
+void free_nfs_srvr_list(struct nfs_server_list *srv)
+{
+       while (srv != NULL) {
+               struct nfs_server_list *next = srv->next;
+
+               free(srv->addr);
+               free(srv);
+               srv = next;
+       }
+}           
+
+struct nfs_list_data {
+       int status;
+       struct nfs_server_list *srvrs;
+};
+
+void callit_cb(struct rpc_context *rpc, int status, void *data _U_, void *private_data)
+{
+       struct nfs_list_data *srv_data = private_data;
+       struct sockaddr *sin;
+       char hostdd[16];
+       struct nfs_server_list *srvr;
+
+       if (status == RPC_STATUS_CANCEL) {
+               return;
+       }
+       if (status != 0) {
+               srv_data->status = -1;
+               return;
+       }
+
+       sin = rpc_get_recv_sockaddr(rpc);
+       if (sin == NULL) {
+               rpc_set_error(rpc, "failed to get sockaddr in CALLIT callback");
+               srv_data->status = -1;
+               return;
+       }
+
+       if (getnameinfo(sin, sizeof(struct sockaddr_in), &hostdd[0], sizeof(hostdd), NULL, 0, NI_NUMERICHOST) < 0) {
+               rpc_set_error(rpc, "getnameinfo failed in CALLIT callback");
+               srv_data->status = -1;
+               return;
+       }
+       
+       /* check for dupes */
+       for (srvr = srv_data->srvrs; srvr; srvr = srvr->next) {
+               if (!strcmp(hostdd, srvr->addr)) {
+                       return 0;
+               }
+       }
+
+       srvr = malloc(sizeof(struct nfs_server_list));
+       if (srvr == NULL) {
+               rpc_set_error(rpc, "Malloc failed when allocating server structure");   
+               srv_data->status = -1;
+               return;
+       }
+
+       srvr->addr = strdup(hostdd);
+       if (srvr->addr == NULL) {
+               rpc_set_error(rpc, "Strdup failed when allocating server structure");
+               free(srvr);
+               srv_data->status = -1;
+               return;
+       }
+
+       srvr->next = srv_data->srvrs;
+       srv_data->srvrs = srvr;
+}
+
+static int send_nfsd_probes(struct rpc_context *rpc, struct ifconf *ifc, struct nfs_list_data *data)
+{
+       char *ptr;
+
+       for (ptr =(char *)(ifc->ifc_buf); ptr < (char *)(ifc->ifc_buf) + ifc->ifc_len; ) {
+               struct ifreq *ifr;
+               char bcdd[16];
+
+               ifr = (struct ifreq *)ptr;
+#ifdef HAVE_SOCKADDR_LEN
+               if (ifr->ifr_addr.sa_len > sizeof(struct sockaddr)) {
+                       ptr += sizeof(ifr->ifr_name) + ifr->ifr_addr.sa_len;
+               } else {
+                       ptr += sizeof(ifr->ifr_name) + sizeof(struct sockaddr);
+               }
+#else
+               ptr += sizeof(struct ifreq);
+#endif
+
+               if (ifr->ifr_addr.sa_family != AF_INET) {
+                       continue;
+               }
+               if (ioctl(rpc_get_fd(rpc), SIOCGIFFLAGS, ifr) < 0) {
+                       return -1;
+               }
+               if (!(ifr->ifr_flags & IFF_UP)) {
+                       continue;
+               }
+               if (ifr->ifr_flags & IFF_LOOPBACK) {
+                       continue;
+               }
+               if (!(ifr->ifr_flags & IFF_BROADCAST)) {
+                       continue;
+               }
+               if (ioctl(rpc_get_fd(rpc), SIOCGIFBRDADDR, ifr) < 0) {
+                       continue;
+               }
+               if (getnameinfo(&ifr->ifr_broadaddr, sizeof(struct sockaddr_in), &bcdd[0], sizeof(bcdd), NULL, 0, NI_NUMERICHOST) < 0) {
+                       continue;
+               }
+               if (rpc_set_udp_destination(rpc, bcdd, 111, 1) < 0) {
+                       return -1;
+               }
+
+               if (rpc_pmap_callit_async(rpc, MOUNT_PROGRAM, 2, 0, NULL, 0, callit_cb, data) < 0) {
+                       return -1;
+               }
+       }
+
+       return 0;
+}
+
+struct nfs_server_list *nfs_find_local_servers(void)
+{
+       struct rpc_context *rpc;
+       struct nfs_list_data data = {0, NULL};
+       struct timeval tv_start, tv_current;
+       struct ifconf ifc;
+       int size, loop;
+       struct pollfd pfd;
+
+       rpc = rpc_init_udp_context();
+       if (rpc == NULL) {
+               return NULL;
+       }
+
+       if (rpc_bind_udp(rpc, "0.0.0.0", 0) < 0) {
+               rpc_destroy_context(rpc);
+               return NULL;
+       }
+
+
+       /* get list of all interfaces */
+       size = sizeof(struct ifreq);
+       ifc.ifc_buf = NULL;
+       ifc.ifc_len = size;
+
+       while(ifc.ifc_len > (size - sizeof(struct ifreq))) {
+               size *= 2;
+
+               free(ifc.ifc_buf);      
+               ifc.ifc_len = size;
+               ifc.ifc_buf = malloc(size);
+               memset(ifc.ifc_buf, 0, size);
+               if (ioctl(rpc_get_fd(rpc), SIOCGIFCONF, (caddr_t)&ifc) < 0) {
+                       rpc_destroy_context(rpc);
+                       free(ifc.ifc_buf);      
+                       return NULL;
+               }
+       }       
+
+       for (loop=0; loop<3; loop++) {
+               if (send_nfsd_probes(rpc, &ifc, &data) != 0) {
+                       rpc_destroy_context(rpc);
+                       free(ifc.ifc_buf);      
+                       return NULL;
+               }
+
+               gettimeofday(&tv_start, NULL);
+               for(;;) {
+                       int mpt;
+
+                       pfd.fd = rpc_get_fd(rpc);
+                       pfd.events = rpc_which_events(rpc);
+
+                       gettimeofday(&tv_current, NULL);
+                       mpt = 1000
+                       -    (tv_current.tv_sec *1000 + tv_current.tv_usec / 1000)
+                       +    (tv_start.tv_sec *1000 + tv_start.tv_usec / 1000);
+
+                       if (poll(&pfd, 1, mpt) < 0) {
+                               free_nfs_srvr_list(data.srvrs);
+                               rpc_destroy_context(rpc);
+                               return NULL;
+                       }
+                       if (pfd.revents == 0) {
+                               break;
+                       }
+               
+                       if (rpc_service(rpc, pfd.revents) < 0) {
+                               break;
+                       }
+               }
+       }
+
+       free(ifc.ifc_buf);      
+       rpc_destroy_context(rpc);
+
+       if (data.status != 0) {
+               free_nfs_srvr_list(data.srvrs);
+               return NULL;
+       }
+
+       return data.srvrs;
+}