2 Copyright (C) by Ronnie Sahlberg <ronniesahlberg@gmail.com> 2011
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.
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.
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/>.
18 /* Example program using the lowlevel raw broadcast interface.
28 #include <sys/socket.h>
29 #include <sys/ioctl.h>
34 #include "libnfs-raw.h"
35 #include "libnfs-raw-mount.h"
36 #include "libnfs-raw-portmap.h"
38 struct nfs_list_data
{
40 struct nfs_server_list
*srvrs
;
43 void free_nfs_srvr_list(struct nfs_server_list
*srv
)
46 struct nfs_server_list
*next
= srv
->next
;
54 void pm_cb(struct rpc_context
*rpc
, int status
, void *data _U_
, void *private_data
)
56 struct nfs_list_data
*srv_data
= private_data
;
59 struct nfs_server_list
*srvr
;
61 if (status
== RPC_STATUS_CANCEL
) {
65 srv_data
->status
= -1;
69 sin
= rpc_get_recv_sockaddr(rpc
);
71 rpc_set_error(rpc
, "failed to get sockaddr in CALLIT callback");
72 srv_data
->status
= -1;
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;
83 srvr
= malloc(sizeof(struct nfs_server_list
));
85 rpc_set_error(rpc
, "Malloc failed when allocating server structure");
86 srv_data
->status
= -1;
90 srvr
->addr
= strdup(hostdd
);
91 if (srvr
->addr
== NULL
) {
92 rpc_set_error(rpc
, "Strdup failed when allocating server structure");
94 srv_data
->status
= -1;
98 srvr
->next
= srv_data
->srvrs
;
99 srv_data
->srvrs
= srvr
;
102 int main(int argc _U_
, char *argv
[] _U_
)
104 struct rpc_context
*rpc
;
108 struct timeval tv_start
, tv_current
;
109 struct nfs_list_data data
= {0, NULL
};
110 struct nfs_server_list
*srvr
;
113 rpc
= rpc_init_udp_context();
115 printf("failed to init context\n");
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
));
125 /* get list of all interfaces */
126 size
= sizeof(struct ifreq
);
130 while(ifc
.ifc_len
> (size
- sizeof(struct ifreq
))) {
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");
143 for (ptr
=(char *)ifc
.ifc_buf
; ptr
< ((char *)ifc
.ifc_buf
) + ifc
.ifc_len
; ) {
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
;
152 ptr
+= sizeof(ifr
->ifr_name
) + sizeof(struct sockaddr
);
155 ptr
+= sizeof(struct ifreq
);
158 if (ifr
->ifr_addr
.sa_family
!= AF_INET
) {
161 if (ioctl(rpc_get_fd(rpc
), SIOCGIFFLAGS
, ifr
) < 0) {
162 printf("ioctl DRBADDR failed\n");
165 if (!(ifr
->ifr_flags
& IFF_UP
)) {
168 if (ifr
->ifr_flags
& IFF_LOOPBACK
) {
171 if (!(ifr
->ifr_flags
& IFF_BROADCAST
)) {
174 if (ioctl(rpc_get_fd(rpc
), SIOCGIFBRDADDR
, ifr
) < 0) {
177 if (getnameinfo(&ifr
->ifr_broadaddr
, sizeof(struct sockaddr_in
), &bcdd
[0], sizeof(bcdd
), NULL
, 0, NI_NUMERICHOST
) < 0) {
180 if (rpc_set_udp_destination(rpc
, bcdd
, 111, 1) < 0) {
181 printf("failed to set udp destination %s\n", rpc_get_error(rpc
));
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");
192 gettimeofday(&tv_start
, NULL
);
196 pfd
.fd
= rpc_get_fd(rpc
);
197 pfd
.events
= rpc_which_events(rpc
);
199 gettimeofday(&tv_current
, NULL
);
201 - (tv_current
.tv_sec
*1000 + tv_current
.tv_usec
/ 1000)
202 + (tv_start
.tv_sec
*1000 + tv_start
.tv_usec
/ 1000);
204 if (poll(&pfd
, 1, mpt
) < 0) {
205 printf("Poll failed");
208 if (pfd
.revents
== 0) {
212 if (rpc_service(rpc
, pfd
.revents
) < 0) {
213 printf("rpc_service failed with %s\n", rpc_get_error(rpc
));
218 for (srvr
=data
.srvrs
; srvr
; srvr
= srvr
->next
) {
219 printf("NFS SERVER @ %s\n", srvr
->addr
);
221 free_nfs_srvr_list(data
.srvrs
);
223 rpc_destroy_context(rpc
);