X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=examples%2Fnfsclient-bcast.c;h=ae08f16e5fe3f02d681adcbed92b114167ddb585;hb=7057e733c1465661c410b65d90e4c5d0939f1617;hp=47fd28cfbd50578ef85af221453c5f4c372d53ff;hpb=92e787feacb4134aa0641b18ea95b014e7006684;p=deb_libnfs.git diff --git a/examples/nfsclient-bcast.c b/examples/nfsclient-bcast.c index 47fd28c..ae08f16 100644 --- a/examples/nfsclient-bcast.c +++ b/examples/nfsclient-bcast.c @@ -17,46 +17,107 @@ /* 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 +#include #include #include + +#ifdef HAVE_SYS_IOCTL_H +#include +#endif + +#ifdef HAVE_SYS_SOCKET_H +#include +#endif + +#include "libnfs-zdr.h" #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) { - pmap_call_result *res = (pmap_call_result *)data; + 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) +{ + 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_) @@ -64,7 +125,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) { @@ -83,76 +148,100 @@ 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_pmap_callit_async(rpc, MOUNT_PROGRAM, 2, 0, NULL, 0, pm_cb, &data) < 0) { printf("Failed to set up callit function\n"); exit(10); } } free(ifc.ifc_buf); - - alarm(3); - + gettimeofday(&tv_start, NULL); for(;;) { + int mpt; + pfd.fd = rpc_get_fd(rpc); pfd.events = rpc_which_events(rpc); - if (poll(&pfd, 1, -1) < 0) { + 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) { printf("Poll failed"); exit(10); } + if (pfd.revents == 0) { + break; + } + if (rpc_service(rpc, pfd.revents) < 0) { printf("rpc_service failed with %s\n", rpc_get_error(rpc)); break; } } + 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; - printf("nfsclient finished\n"); return 0; }