X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=examples%2Fnfsclient-bcast.c;h=ec19a69f958b23ed0f3c4b9949268f294f73250e;hb=2db42ce2fc89bc88050564037688888bbb8876e9;hp=6f48618259a15ec8a21c04b49ce928e0827b3a01;hpb=cb46c8fb9dfde942b531545359ed92313112e668;p=deb_libnfs.git diff --git a/examples/nfsclient-bcast.c b/examples/nfsclient-bcast.c index 6f48618..ec19a69 100644 --- a/examples/nfsclient-bcast.c +++ b/examples/nfsclient-bcast.c @@ -17,50 +17,115 @@ /* Example program using the lowlevel raw broadcast interface. */ +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif -#include +#ifdef WIN32 +#include "win32_compat.h" +#endif + +#ifdef HAVE_POLL_H +#include +#endif + +#ifdef HAVE_UNISTD_H #include +#endif + +#include #include #include -#include #include -#include -#include + +#ifdef HAVE_SYS_TIME_H #include +#endif + +#ifdef HAVE_NET_IF_H #include +#endif + +#ifdef HAVE_NETDB_H #include +#endif + +#ifdef HAVE_SYS_IOCTL_H +#include +#endif + +#ifdef HAVE_SYS_SOCKET_H +#include +#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; iifr_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;