X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=examples%2Fnfsclient-bcast.c;h=62ea6549d29ba295003feeb82932565545c0aa44;hb=2384420387a7c6c2aaa983c48e98d4ab788ce65f;hp=c74c5b37764b8e12a9aa002ee714aadf24deb17f;hpb=28f7bd66967e39c8f2290f32c9b3dee1306cb11c;p=deb_libnfs.git diff --git a/examples/nfsclient-bcast.c b/examples/nfsclient-bcast.c index c74c5b3..62ea654 100644 --- a/examples/nfsclient-bcast.c +++ b/examples/nfsclient-bcast.c @@ -17,50 +17,128 @@ /* 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 + +#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-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) +{ + 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; } - printf("NFS server at %s\n", hostdd); + + 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; } int main(int argc _U_, char *argv[] _U_) { struct rpc_context *rpc; struct pollfd pfd; + struct ifconf ifc; + 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) { @@ -73,43 +151,106 @@ int main(int argc _U_, char *argv[] _U_) exit(10); } - if (rpc_set_udp_destination(rpc, "10.1.1.255", 111, 1) < 0) { - printf("failed to set udp destination %s\n", rpc_get_error(rpc)); - exit(10); - } - if (rpc_pmap_callit_async(rpc, 100005, 2, 0, NULL, 0, pm_cb, NULL) < 0) { - printf("Failed to set up callit function\n"); - exit(10); - } - if (rpc_set_udp_destination(rpc, "10.9.2.255", 111, 1) < 0) { - printf("failed to set udp destination %s\n", rpc_get_error(rpc)); - exit(10); - } + /* get list of all interfaces */ + size = sizeof(struct ifreq); + ifc.ifc_buf = NULL; + ifc.ifc_len = size; - if (rpc_pmap_callit_async(rpc, 100005, 2, 0, NULL, 0, pm_cb, NULL) < 0) { - printf("Failed to set up callit function\n"); - exit(10); - } + while(ifc.ifc_len > (size - sizeof(struct ifreq))) { + size *= 2; - alarm(3); + 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 (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) { + printf("ioctl DRBADDR failed\n"); + exit(10); + } + 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) { + 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, &data) < 0) { + printf("Failed to set up callit function\n"); + exit(10); + } + } + free(ifc.ifc_buf); + + 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; }