X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=examples%2Fnfsclient-bcast.c;h=6f48618259a15ec8a21c04b49ce928e0827b3a01;hb=cb46c8fb9dfde942b531545359ed92313112e668;hp=c74c5b37764b8e12a9aa002ee714aadf24deb17f;hpb=28f7bd66967e39c8f2290f32c9b3dee1306cb11c;p=deb_libnfs.git

diff --git a/examples/nfsclient-bcast.c b/examples/nfsclient-bcast.c
index c74c5b3..6f48618 100644
--- a/examples/nfsclient-bcast.c
+++ b/examples/nfsclient-bcast.c
@@ -25,6 +25,9 @@
 #include <poll.h>
 #include <errno.h>
 #include <sys/socket.h>
+#include <sys/ioctl.h>
+#include <sys/time.h>
+#include <net/if.h>
 #include <netdb.h>
 #include "libnfs.h"
 #include "libnfs-raw.h"
@@ -38,6 +41,9 @@ void pm_cb(struct rpc_context *rpc _U_, int status, void *data, void *private_da
 	struct sockaddr *sin;
 	char hostdd[16];
 
+	if (status == RPC_STATUS_CANCEL) {
+		return;
+	}
 	if (status != 0) {
 		printf("callback for CALLIT failed\n");
 		exit(10);
@@ -61,6 +67,9 @@ int main(int argc _U_, char *argv[] _U_)
 {
 	struct rpc_context *rpc;
 	struct pollfd pfd;
+	struct ifconf ifc;
+	int i, size;
+	struct timeval tv_start, tv_current;
 	
 	rpc = rpc_init_udp_context();
 	if (rpc == NULL) {
@@ -73,43 +82,90 @@ 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) {
+		size *= 2;
 
-	alarm(3);
+		free(ifc.ifc_buf);	
+		ifc.ifc_len = size;
+		ifc.ifc_buf = malloc(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++) {
+		char bcdd[16];
 
+		if (ifc.ifc_req[i].ifr_addr.sa_family != AF_INET) {
+			continue;
+		}
+		if (ioctl(rpc_get_fd(rpc), SIOCGIFFLAGS, &ifc.ifc_req[i]) < 0) {
+			printf("ioctl DRBADDR failed\n");
+			exit(10);
+		}
+		if (!(ifc.ifc_req[i].ifr_flags & IFF_UP)) {
+			continue;
+		}
+		if (ifc.ifc_req[i].ifr_flags & IFF_LOOPBACK) {
+			continue;
+		}
+		if (!(ifc.ifc_req[i].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 (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 (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) {
+			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;
 		}
 	}
-	
+
 	rpc_destroy_context(rpc);
 	rpc=NULL;
-	printf("nfsclient finished\n");
 	return 0;
 }