ZDR: remove dependency on zdr.h from the examples and nfs-ls
[deb_libnfs.git] / examples / nfsclient-bcast.c
index 6f48618259a15ec8a21c04b49ce928e0827b3a01..ec19a69f958b23ed0f3c4b9949268f294f73250e 100644 (file)
 
 /* Example program using the lowlevel raw broadcast interface.
  */
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
 
-#include <stdio.h>
+#ifdef WIN32
+#include "win32_compat.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 <string.h>
-#include <poll.h>
 #include <errno.h>
-#include <sys/socket.h>
-#include <sys/ioctl.h>
+
+#ifdef HAVE_SYS_TIME_H
 #include <sys/time.h>
+#endif
+
+#ifdef HAVE_NET_IF_H
 #include <net/if.h>
+#endif
+
+#ifdef HAVE_NETDB_H
 #include <netdb.h>
+#endif
+
+#ifdef HAVE_SYS_IOCTL_H
+#include <sys/ioctl.h>
+#endif
+
+#ifdef HAVE_SYS_SOCKET_H
+#include <sys/socket.h>
+#endif
+
 #include "libnfs.h"
 #include "libnfs-raw.h"
+#include "libnfs-private.h"
 #include "libnfs-raw-mount.h"
 #include "libnfs-raw-portmap.h"
-#include "libnfs-private.h"
 
-void pm_cb(struct rpc_context *rpc _U_, int status, void *data, void *private_data _U_)
+struct nfs_list_data {
+       int status;
+       struct nfs_server_list *srvrs;
+};
+
+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;
+       }
+}           
+
+void pm_cb(struct rpc_context *rpc, int status, void *data _U_, void *private_data)
 {
-       pmap_call_result *res = (pmap_call_result *)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) {
-               printf("callback for CALLIT failed\n");
-               exit(10);
+               srv_data->status = -1;
+               return;
        }
 
        sin = rpc_get_recv_sockaddr(rpc);
        if (sin == NULL) {
-               printf("failed to get sockaddr for received pdu\n");
-               exit(10);
+               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) {
-               printf("getnameinfo failed\n");
-               exit(10);
+               rpc_set_error(rpc, "getnameinfo failed in CALLIT callback");
+               srv_data->status = -1;
+               return;
+       }
+
+       
+       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;
        }
 
-       printf("NFS server at %s\n", hostdd);
+       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;
 }
 
 int main(int argc _U_, char *argv[] _U_)
@@ -68,8 +133,11 @@ int main(int argc _U_, char *argv[] _U_)
        struct rpc_context *rpc;
        struct pollfd pfd;
        struct ifconf ifc;
-       int i, size;
+       int size;
        struct timeval tv_start, tv_current;
+       struct nfs_list_data data = {0, NULL};
+       struct nfs_server_list *srvr;
+       char *ptr;
        
        rpc = rpc_init_udp_context();
        if (rpc == NULL) {
@@ -88,51 +156,62 @@ int main(int argc _U_, char *argv[] _U_)
        ifc.ifc_buf = NULL;
        ifc.ifc_len = size;
 
-       while (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) {
                        printf("ioctl SIOCGIFCONF failed\n");
                        exit(10);
                }
        }       
 
-       for (i=0; i<ifc.ifc_len / sizeof(struct ifconf); i++) {
+       for (ptr =(char *)ifc.ifc_buf; ptr < ((char *)ifc.ifc_buf) + ifc.ifc_len; ) {
+               struct ifreq *ifr;
                char bcdd[16];
 
-               if (ifc.ifc_req[i].ifr_addr.sa_family != AF_INET) {
+               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, &ifc.ifc_req[i]) < 0) {
+               if (ioctl(rpc_get_fd(rpc), SIOCGIFFLAGS, ifr) < 0) {
                        printf("ioctl DRBADDR failed\n");
                        exit(10);
                }
-               if (!(ifc.ifc_req[i].ifr_flags & IFF_UP)) {
+               if (!(ifr->ifr_flags & IFF_UP)) {
                        continue;
                }
-               if (ifc.ifc_req[i].ifr_flags & IFF_LOOPBACK) {
+               if (ifr->ifr_flags & IFF_LOOPBACK) {
                        continue;
                }
-               if (!(ifc.ifc_req[i].ifr_flags & IFF_BROADCAST)) {
+               if (!(ifr->ifr_flags & IFF_BROADCAST)) {
                        continue;
                }
-               if (ioctl(rpc_get_fd(rpc), SIOCGIFBRDADDR, &ifc.ifc_req[i]) < 0) {
-                       printf("ioctl DRBADDR failed\n");
-                       exit(10);
+               if (ioctl(rpc_get_fd(rpc), SIOCGIFBRDADDR, ifr) < 0) {
+                       continue;
                }
-               if (getnameinfo(&ifc.ifc_req[i].ifr_broadaddr, sizeof(struct sockaddr_in), &bcdd[0], sizeof(bcdd), NULL, 0, NI_NUMERICHOST) < 0) {
-                       printf("getnameinfo failed\n");
-                       exit(10);
+               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) {
                        printf("failed to set udp destination %s\n", rpc_get_error(rpc));
                        exit(10);
                }
 
-               if (rpc_pmap_callit_async(rpc, MOUNT_PROGRAM, 2, 0, NULL, 0, pm_cb, NULL) < 0) {
+               if (rpc_pmap2_callit_async(rpc, MOUNT_PROGRAM, 2, 0, NULL, 0, pm_cb, &data) < 0) {
                        printf("Failed to set up callit function\n");
                        exit(10);
                }
@@ -165,6 +244,11 @@ int main(int argc _U_, char *argv[] _U_)
                }
        }
 
+       for (srvr=data.srvrs; srvr; srvr = srvr->next) {
+               printf("NFS SERVER @ %s\n", srvr->addr);
+       }
+       free_nfs_srvr_list(data.srvrs);
+
        rpc_destroy_context(rpc);
        rpc=NULL;
        return 0;