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 #ifdef HAVE_CONFIG_H
  21 #include "config.h"
  22 #endif
  23
  24 #ifdef WIN32
  25 #include "win32_compat.h"
  26 #endif
  27
  28 #ifdef HAVE_POLL_H
  29 #include <poll.h>
  30 #endif
  31
  32 #ifdef HAVE_UNISTD_H
  33 #include <unistd.h>
  34 #endif
  35
  36 #include <stdio.h>
  37 #include <stdlib.h>
  38 #include <string.h>
  39 #include <errno.h>
  40
  41 #ifdef HAVE_SYS_TIME_H
  42 #include <sys/time.h>
  43 #endif
  44
  45 #ifdef HAVE_NET_IF_H
  46 #include <net/if.h>
  47 #endif
  48
  49 #ifdef HAVE_NETDB_H
  50 #include <netdb.h>
  51 #endif
  52
  53 #ifdef HAVE_SYS_IOCTL_H
  54 #include <sys/ioctl.h>
  55 #endif
  56
  57 #ifdef HAVE_SYS_SOCKET_H
  58 #include <sys/socket.h>
  59 #endif
  60
  61 #include "libnfs-zdr.h"
  62 #include "libnfs.h"
  63 #include "libnfs-raw.h"
  64 #include "libnfs-private.h"
  65 #include "libnfs-raw-mount.h"
  66 #include "libnfs-raw-portmap.h"
  67
  68 struct nfs_list_data {
  69        int status;
  70        struct nfs_server_list *srvrs;
  71 };
  72
  73 void free_nfs_srvr_list(struct nfs_server_list *srv)
  74 {
  75         while (srv != NULL) {
  76                 struct nfs_server_list *next = srv->next;
  77
  78                 free(srv->addr);
  79                 free(srv);
  80                 srv = next;
  81         }
  82 }
  83
  84 void pm_cb(struct rpc_context *rpc, int status, void *data _U_, void *private_data)
  85 {
  86         struct nfs_list_data *srv_data = private_data;
  87         struct sockaddr *sin;
  88         char hostdd[16];
  89         struct nfs_server_list *srvr;
  90
  91         if (status == RPC_STATUS_CANCEL) {
  92                 return;
  93         }
  94         if (status != 0) {
  95                 srv_data->status = -1;
  96                 return;
  97         }
  98
  99         sin = rpc_get_recv_sockaddr(rpc);
 100         if (sin == NULL) {
 101                 rpc_set_error(rpc, "failed to get sockaddr in CALLIT callback");
 102                 srv_data->status = -1;
 103                 return;
 104         }
 105
 106         if (getnameinfo(sin, sizeof(struct sockaddr_in), &hostdd[0], sizeof(hostdd), NULL, 0, NI_NUMERICHOST) < 0) {
 107                 rpc_set_error(rpc, "getnameinfo failed in CALLIT callback");
 108                 srv_data->status = -1;
 109                 return;
 110         }
 111
 112
 113         srvr = malloc(sizeof(struct nfs_server_list));
 114         if (srvr == NULL) {
 115                 rpc_set_error(rpc, "Malloc failed when allocating server structure");
 116                 srv_data->status = -1;
 117                 return;
 118         }
 119
 120         srvr->addr = strdup(hostdd);
 121         if (srvr->addr == NULL) {
 122                 rpc_set_error(rpc, "Strdup failed when allocating server structure");
 123                 free(srvr);
 124                 srv_data->status = -1;
 125                 return;
 126         }
 127
 128         srvr->next = srv_data->srvrs;
 129         srv_data->srvrs = srvr;
 130 }
 131
 132 int main(int argc _U_, char *argv[] _U_)
 133 {
 134         struct rpc_context *rpc;
 135         struct pollfd pfd;
 136         struct ifconf ifc;
 137         int size;
 138         struct timeval tv_start, tv_current;
 139         struct nfs_list_data data = {0, NULL};
 140         struct nfs_server_list *srvr;
 141         char *ptr;
 142
 143         rpc = rpc_init_udp_context();
 144         if (rpc == NULL) {
 145                 printf("failed to init context\n");
 146                 exit(10);
 147         }
 148
 149         if (rpc_bind_udp(rpc, "0.0.0.0", 0) < 0) {
 150                 printf("failed to bind to udp %s\n", rpc_get_error(rpc));
 151                 exit(10);
 152         }
 153
 154
 155         /* get list of all interfaces */
 156         size = sizeof(struct ifreq);
 157         ifc.ifc_buf = NULL;
 158         ifc.ifc_len = size;
 159
 160         while(ifc.ifc_len > (size - sizeof(struct ifreq))) {
 161                 size *= 2;
 162
 163                 free(ifc.ifc_buf);
 164                 ifc.ifc_len = size;
 165                 ifc.ifc_buf = malloc(size);
 166                 memset(ifc.ifc_buf, 0, size);
 167                 if (ioctl(rpc_get_fd(rpc), SIOCGIFCONF, (caddr_t)&ifc) < 0) {
 168                         printf("ioctl SIOCGIFCONF failed\n");
 169                         exit(10);
 170                 }
 171         }
 172
 173         for (ptr =(char *)ifc.ifc_buf; ptr < ((char *)ifc.ifc_buf) + ifc.ifc_len; ) {
 174                 struct ifreq *ifr;
 175                 char bcdd[16];
 176
 177                 ifr = (struct ifreq *)ptr;
 178 #ifdef HAVE_SOCKADDR_LEN
 179                 if (ifr->ifr_addr.sa_len > sizeof(struct sockaddr)) {
 180                         ptr += sizeof(ifr->ifr_name) + ifr->ifr_addr.sa_len;
 181                 } else {
 182                         ptr += sizeof(ifr->ifr_name) + sizeof(struct sockaddr);
 183                 }
 184 #else
 185                 ptr += sizeof(struct ifreq);
 186 #endif
 187
 188                 if (ifr->ifr_addr.sa_family != AF_INET) {
 189                         continue;
 190                 }
 191                 if (ioctl(rpc_get_fd(rpc), SIOCGIFFLAGS, ifr) < 0) {
 192                         printf("ioctl DRBADDR failed\n");
 193                         exit(10);
 194                 }
 195                 if (!(ifr->ifr_flags & IFF_UP)) {
 196                         continue;
 197                 }
 198                 if (ifr->ifr_flags & IFF_LOOPBACK) {
 199                         continue;
 200                 }
 201                 if (!(ifr->ifr_flags & IFF_BROADCAST)) {
 202                         continue;
 203                 }
 204                 if (ioctl(rpc_get_fd(rpc), SIOCGIFBRDADDR, ifr) < 0) {
 205                         continue;
 206                 }
 207                 if (getnameinfo(&ifr->ifr_broadaddr, sizeof(struct sockaddr_in), &bcdd[0], sizeof(bcdd), NULL, 0, NI_NUMERICHOST) < 0) {
 208                         continue;
 209                 }
 210                 if (rpc_set_udp_destination(rpc, bcdd, 111, 1) < 0) {
 211                         printf("failed to set udp destination %s\n", rpc_get_error(rpc));
 212                         exit(10);
 213                 }
 214
 215                 if (rpc_pmap2_callit_async(rpc, MOUNT_PROGRAM, 2, 0, NULL, 0, pm_cb, &data) < 0) {
 216                         printf("Failed to set up callit function\n");
 217                         exit(10);
 218                 }
 219         }
 220         free(ifc.ifc_buf);
 221
 222         gettimeofday(&tv_start, NULL);
 223         for(;;) {
 224                 int mpt;
 225
 226                 pfd.fd = rpc_get_fd(rpc);
 227                 pfd.events = rpc_which_events(rpc);
 228
 229                 gettimeofday(&tv_current, NULL);
 230                 mpt = 1000
 231                 -    (tv_current.tv_sec *1000 + tv_current.tv_usec / 1000)
 232                 +    (tv_start.tv_sec *1000 + tv_start.tv_usec / 1000);
 233
 234                 if (poll(&pfd, 1, mpt) < 0) {
 235                         printf("Poll failed");
 236                         exit(10);
 237                 }
 238                 if (pfd.revents == 0) {
 239                         break;
 240                 }
 241
 242                 if (rpc_service(rpc, pfd.revents) < 0) {
 243                         printf("rpc_service failed with %s\n", rpc_get_error(rpc));
 244                         break;
 245                 }
 246         }
 247
 248         for (srvr=data.srvrs; srvr; srvr = srvr->next) {
 249                 printf("NFS SERVER @ %s\n", srvr->addr);
 250         }
 251         free_nfs_srvr_list(data.srvrs);
 252
 253         rpc_destroy_context(rpc);
 254         rpc=NULL;
 255         return 0;
 256 }