Imported Upstream version 1.3.0
[deb_libnfs.git] / examples / nfsclient-bcast.c
1 /*
2 Copyright (C) by Ronnie Sahlberg <ronniesahlberg@gmail.com> 2011
3
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 3 of the License, or
7 (at your option) any later version.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, see <http://www.gnu.org/licenses/>.
16 */
17
18 /* Example program using the lowlevel raw broadcast interface.
19 */
20
21 #include "config.h"
22 #include <stdio.h>
23 #include <unistd.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <poll.h>
27 #include <errno.h>
28 #include <sys/socket.h>
29 #include <sys/ioctl.h>
30 #include <sys/time.h>
31 #include <net/if.h>
32 #include <netdb.h>
33 #include "libnfs.h"
34 #include "libnfs-raw.h"
35 #include "libnfs-raw-mount.h"
36 #include "libnfs-raw-portmap.h"
37
38 struct nfs_list_data {
39 int status;
40 struct nfs_server_list *srvrs;
41 };
42
43 void free_nfs_srvr_list(struct nfs_server_list *srv)
44 {
45 while (srv != NULL) {
46 struct nfs_server_list *next = srv->next;
47
48 free(srv->addr);
49 free(srv);
50 srv = next;
51 }
52 }
53
54 void pm_cb(struct rpc_context *rpc, int status, void *data _U_, void *private_data)
55 {
56 struct nfs_list_data *srv_data = private_data;
57 struct sockaddr *sin;
58 char hostdd[16];
59 struct nfs_server_list *srvr;
60
61 if (status == RPC_STATUS_CANCEL) {
62 return;
63 }
64 if (status != 0) {
65 srv_data->status = -1;
66 return;
67 }
68
69 sin = rpc_get_recv_sockaddr(rpc);
70 if (sin == NULL) {
71 rpc_set_error(rpc, "failed to get sockaddr in CALLIT callback");
72 srv_data->status = -1;
73 return;
74 }
75
76 if (getnameinfo(sin, sizeof(struct sockaddr_in), &hostdd[0], sizeof(hostdd), NULL, 0, NI_NUMERICHOST) < 0) {
77 rpc_set_error(rpc, "getnameinfo failed in CALLIT callback");
78 srv_data->status = -1;
79 return;
80 }
81
82
83 srvr = malloc(sizeof(struct nfs_server_list));
84 if (srvr == NULL) {
85 rpc_set_error(rpc, "Malloc failed when allocating server structure");
86 srv_data->status = -1;
87 return;
88 }
89
90 srvr->addr = strdup(hostdd);
91 if (srvr->addr == NULL) {
92 rpc_set_error(rpc, "Strdup failed when allocating server structure");
93 free(srvr);
94 srv_data->status = -1;
95 return;
96 }
97
98 srvr->next = srv_data->srvrs;
99 srv_data->srvrs = srvr;
100 }
101
102 int main(int argc _U_, char *argv[] _U_)
103 {
104 struct rpc_context *rpc;
105 struct pollfd pfd;
106 struct ifconf ifc;
107 int size;
108 struct timeval tv_start, tv_current;
109 struct nfs_list_data data = {0, NULL};
110 struct nfs_server_list *srvr;
111 char *ptr;
112
113 rpc = rpc_init_udp_context();
114 if (rpc == NULL) {
115 printf("failed to init context\n");
116 exit(10);
117 }
118
119 if (rpc_bind_udp(rpc, "0.0.0.0", 0) < 0) {
120 printf("failed to bind to udp %s\n", rpc_get_error(rpc));
121 exit(10);
122 }
123
124
125 /* get list of all interfaces */
126 size = sizeof(struct ifreq);
127 ifc.ifc_buf = NULL;
128 ifc.ifc_len = size;
129
130 while(ifc.ifc_len > (size - sizeof(struct ifreq))) {
131 size *= 2;
132
133 free(ifc.ifc_buf);
134 ifc.ifc_len = size;
135 ifc.ifc_buf = malloc(size);
136 memset(ifc.ifc_buf, 0, size);
137 if (ioctl(rpc_get_fd(rpc), SIOCGIFCONF, (caddr_t)&ifc) < 0) {
138 printf("ioctl SIOCGIFCONF failed\n");
139 exit(10);
140 }
141 }
142
143 for (ptr =(char *)ifc.ifc_buf; ptr < ((char *)ifc.ifc_buf) + ifc.ifc_len; ) {
144 struct ifreq *ifr;
145 char bcdd[16];
146
147 ifr = (struct ifreq *)ptr;
148 #ifdef HAVE_SOCKADDR_LEN
149 if (ifr->ifr_addr.sa_len > sizeof(struct sockaddr)) {
150 ptr += sizeof(ifr->ifr_name) + ifr->ifr_addr.sa_len;
151 } else {
152 ptr += sizeof(ifr->ifr_name) + sizeof(struct sockaddr);
153 }
154 #else
155 ptr += sizeof(struct ifreq);
156 #endif
157
158 if (ifr->ifr_addr.sa_family != AF_INET) {
159 continue;
160 }
161 if (ioctl(rpc_get_fd(rpc), SIOCGIFFLAGS, ifr) < 0) {
162 printf("ioctl DRBADDR failed\n");
163 exit(10);
164 }
165 if (!(ifr->ifr_flags & IFF_UP)) {
166 continue;
167 }
168 if (ifr->ifr_flags & IFF_LOOPBACK) {
169 continue;
170 }
171 if (!(ifr->ifr_flags & IFF_BROADCAST)) {
172 continue;
173 }
174 if (ioctl(rpc_get_fd(rpc), SIOCGIFBRDADDR, ifr) < 0) {
175 continue;
176 }
177 if (getnameinfo(&ifr->ifr_broadaddr, sizeof(struct sockaddr_in), &bcdd[0], sizeof(bcdd), NULL, 0, NI_NUMERICHOST) < 0) {
178 continue;
179 }
180 if (rpc_set_udp_destination(rpc, bcdd, 111, 1) < 0) {
181 printf("failed to set udp destination %s\n", rpc_get_error(rpc));
182 exit(10);
183 }
184
185 if (rpc_pmap_callit_async(rpc, MOUNT_PROGRAM, 2, 0, NULL, 0, pm_cb, &data) < 0) {
186 printf("Failed to set up callit function\n");
187 exit(10);
188 }
189 }
190 free(ifc.ifc_buf);
191
192 gettimeofday(&tv_start, NULL);
193 for(;;) {
194 int mpt;
195
196 pfd.fd = rpc_get_fd(rpc);
197 pfd.events = rpc_which_events(rpc);
198
199 gettimeofday(&tv_current, NULL);
200 mpt = 1000
201 - (tv_current.tv_sec *1000 + tv_current.tv_usec / 1000)
202 + (tv_start.tv_sec *1000 + tv_start.tv_usec / 1000);
203
204 if (poll(&pfd, 1, mpt) < 0) {
205 printf("Poll failed");
206 exit(10);
207 }
208 if (pfd.revents == 0) {
209 break;
210 }
211
212 if (rpc_service(rpc, pfd.revents) < 0) {
213 printf("rpc_service failed with %s\n", rpc_get_error(rpc));
214 break;
215 }
216 }
217
218 for (srvr=data.srvrs; srvr; srvr = srvr->next) {
219 printf("NFS SERVER @ %s\n", srvr->addr);
220 }
221 free_nfs_srvr_list(data.srvrs);
222
223 rpc_destroy_context(rpc);
224 rpc=NULL;
225 return 0;
226 }