From 552c766578d37c4d1eb150436da7eeed104f1fe9 Mon Sep 17 00:00:00 2001 From: Ronnie Sahlberg Date: Sat, 2 Jul 2011 10:30:16 +1000 Subject: [PATCH] add example for sync interface to probe for all local servers --- Makefile.in | 6 +- examples/nfsclient-bcast.c | 10 +- examples/nfsclient-listservers.c | 50 +++++++++ include/libnfs.h | 10 ++ lib/libnfs-sync.c | 187 +++++++++++++++++++++++++++++++ 5 files changed, 254 insertions(+), 9 deletions(-) create mode 100644 examples/nfsclient-listservers.c diff --git a/Makefile.in b/Makefile.in index d2d0c83..e4952b8 100644 --- a/Makefile.in +++ b/Makefile.in @@ -35,7 +35,7 @@ nfs/libnfs-raw-nfs.o nfs/nfs.o \ rquota/libnfs-raw-rquota.o rquota/rquota.o ifeq ("$(ENABLE_EXAMPLES)","yes") - EXAMPLES=bin/nfsclient-raw bin/nfsclient-async bin/nfsclient-sync bin/nfsclient-bcast + EXAMPLES=bin/nfsclient-raw bin/nfsclient-async bin/nfsclient-sync bin/nfsclient-bcast bin/nfsclient-listservers endif all: $(LIBNFS_SO) $(EXAMPLES) @@ -56,6 +56,10 @@ bin/nfsclient-bcast: examples/nfsclient-bcast.c $(LIBNFS_A) mkdir -p bin $(CC) $(CFLAGS) -o $@ examples/nfsclient-bcast.c $(LIBNFS_A) $(LIBS) +bin/nfsclient-listservers: examples/nfsclient-listservers.c $(LIBNFS_A) + mkdir -p bin + $(CC) $(CFLAGS) -o $@ examples/nfsclient-listservers.c $(LIBNFS_A) $(LIBS) + $(LIBNFS_A): $(LIBNFS_OBJS) $(LIB_OBJS) @echo Creating library $@ ar r $(LIBNFS_A) $(LIBNFS_OBJS) $(LIB_OBJS) diff --git a/examples/nfsclient-bcast.c b/examples/nfsclient-bcast.c index 2075f09..9fb8073 100644 --- a/examples/nfsclient-bcast.c +++ b/examples/nfsclient-bcast.c @@ -35,11 +35,6 @@ #include "libnfs-raw-portmap.h" #include "libnfs-private.h" -struct nfs_server_list { - struct nfs_server_list *next; - char *addr; -}; - struct nfs_list_data { int status; struct nfs_server_list *srvrs; @@ -56,9 +51,8 @@ void free_nfs_srvr_list(struct nfs_server_list *srv) } } -void pm_cb(struct rpc_context *rpc, int status, void *data, void *private_data _U_) +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]; @@ -144,7 +138,7 @@ int main(int argc _U_, char *argv[] _U_) } } - for (i=0; i 2011 + + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program; if not, see . +*/ + +/* Example program showing sync interface to probe for all local servers + */ + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include "libnfs.h" +#include "libnfs-raw.h" +#include "libnfs-raw-mount.h" +#include "libnfs-raw-portmap.h" +#include "libnfs-private.h" + + +int main(int argc _U_, char *argv[] _U_) +{ + struct nfs_server_list *srvrs; + struct nfs_server_list *srv; + + srvrs = nfs_find_local_servers(); + for (srv=srvrs; srv; srv = srv->next) { + printf("NFS SERVER @ %s\n", srv->addr); + } + free_nfs_srvr_list(srvrs); + return 0; +} diff --git a/include/libnfs.h b/include/libnfs.h index e77e1d6..81febb2 100644 --- a/include/libnfs.h +++ b/include/libnfs.h @@ -952,3 +952,13 @@ void mount_free_export_list(struct exportnode *exports); //qqq replace later with lseek(cur, 0) off_t nfs_get_current_offset(struct nfsfh *nfsfh); + + + +struct nfs_server_list { + struct nfs_server_list *next; + char *addr; +}; + +struct nfs_server_list *nfs_find_local_servers(void); +void free_nfs_srvr_list(struct nfs_server_list *srv); diff --git a/lib/libnfs-sync.c b/lib/libnfs-sync.c index 4c8ba76..6e4eff5 100644 --- a/lib/libnfs-sync.c +++ b/lib/libnfs-sync.c @@ -25,10 +25,13 @@ #include #include #include +#include #include #include #include #include +#include +#include #include "libnfs.h" #include "libnfs-raw.h" #include "libnfs-raw-mount.h" @@ -1129,3 +1132,187 @@ void mount_free_export_list(struct exportnode *exports) } } + + + +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; + } +} + +struct nfs_list_data { + int status; + struct nfs_server_list *srvrs; +}; + +void callit_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) { + srv_data->status = -1; + return; + } + + sin = rpc_get_recv_sockaddr(rpc); + if (sin == NULL) { + 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) { + 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; + } + + 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; +} + +struct nfs_server_list *nfs_find_local_servers(void) +{ + struct rpc_context *rpc; + struct nfs_list_data data = {0, NULL}; + struct timeval tv_start, tv_current; + struct ifconf ifc; + int i, size; + struct pollfd pfd; + + rpc = rpc_init_udp_context(); + if (rpc == NULL) { + return NULL; + } + + if (rpc_bind_udp(rpc, "0.0.0.0", 0) < 0) { + rpc_destroy_context(rpc); + return NULL; + } + + + /* get list of all interfaces */ + size = sizeof(struct ifreq); + ifc.ifc_buf = NULL; + ifc.ifc_len = size; + + while (ifc.ifc_len == size) { + size *= 2; + + free(ifc.ifc_buf); + ifc.ifc_len = size; + ifc.ifc_buf = malloc(size); + if (ioctl(rpc_get_fd(rpc), SIOCGIFCONF, (caddr_t)&ifc) < 0) { + rpc_destroy_context(rpc); + free(ifc.ifc_buf); + return NULL; + } + } + + for (i = 0; (unsigned)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) { + rpc_destroy_context(rpc); + free(ifc.ifc_buf); + return NULL; + } + 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) { + rpc_destroy_context(rpc); + free(ifc.ifc_buf); + return NULL; + } + if (getnameinfo(&ifc.ifc_req[i].ifr_broadaddr, sizeof(struct sockaddr_in), &bcdd[0], sizeof(bcdd), NULL, 0, NI_NUMERICHOST) < 0) { + rpc_destroy_context(rpc); + free(ifc.ifc_buf); + return NULL; + } + if (rpc_set_udp_destination(rpc, bcdd, 111, 1) < 0) { + rpc_destroy_context(rpc); + free(ifc.ifc_buf); + return NULL; + } + + if (rpc_pmap_callit_async(rpc, MOUNT_PROGRAM, 2, 0, NULL, 0, callit_cb, &data) < 0) { + rpc_destroy_context(rpc); + free(ifc.ifc_buf); + return NULL; + } + } + free(ifc.ifc_buf); + + gettimeofday(&tv_start, NULL); + for(;;) { + int mpt; + + pfd.fd = rpc_get_fd(rpc); + pfd.events = rpc_which_events(rpc); + + 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) { + free_nfs_srvr_list(data.srvrs); + rpc_destroy_context(rpc); + return NULL; + } + if (pfd.revents == 0) { + break; + } + + if (rpc_service(rpc, pfd.revents) < 0) { + break; + } + } + + rpc_destroy_context(rpc); + + if (data.status != 0) { + free_nfs_srvr_list(data.srvrs); + return NULL; + } + + return data.srvrs; +} -- 2.34.1