1 /* 2 * Copyright (c) 1995 3 * A.R. Gordon (andrew.gordon@net-tel.co.uk). All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. All advertising materials mentioning features or use of this software 14 * must display the following acknowledgement: 15 * This product includes software developed for the FreeBSD project 16 * 4. Neither the name of the author nor the names of any co-contributors 17 * may be used to endorse or promote products derived from this software 18 * without specific prior written permission. 19 * 20 * THIS SOFTWARE IS PROVIDED BY ANDREW GORDON AND CONTRIBUTORS ``AS IS'' AND 21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 23 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 30 * SUCH DAMAGE. 31 * 32 * $NetBSD: lockd.c,v 1.7 2000/08/12 18:08:44 thorpej Exp $ 33 * $FreeBSD: src/usr.sbin/rpc.lockd/lockd.c,v 1.6 2001/03/19 12:50:09 alfred Exp $ 34 * $DragonFly: src/usr.sbin/rpc.lockd/lockd.c,v 1.5 2005/11/25 00:32:49 swildner Exp $ 35 */ 36 37 /* 38 * main() function for NFS lock daemon. Most of the code in this 39 * file was generated by running rpcgen /usr/include/rpcsvc/nlm_prot.x. 40 * 41 * The actual program logic is in the file lock_proc.c 42 */ 43 44 #include <sys/types.h> 45 #include <sys/socket.h> 46 47 #include <netinet/in.h> 48 #include <arpa/inet.h> 49 50 #include <err.h> 51 #include <stdio.h> 52 #include <stdlib.h> 53 #include <errno.h> 54 #include <syslog.h> 55 #include <signal.h> 56 #include <string.h> 57 #include <unistd.h> 58 #include <libutil.h> 59 #include <netconfig.h> 60 #include <netdb.h> 61 62 #include <rpc/rpc.h> 63 #include <rpc/rpc_com.h> 64 #include <rpcsvc/sm_inter.h> 65 66 #include "lockd.h" 67 #include <rpcsvc/nlm_prot.h> 68 69 int debug_level = 0; /* 0 = no debugging syslog() calls */ 70 int _rpcsvcdirty = 0; 71 72 int grace_expired; 73 char **hosts, *svcport_str = NULL; 74 int nhosts = 0; 75 int xcreated = 0; 76 77 #ifndef IPPORT_MAX 78 #define IPPORT_MAX 65535 79 #endif 80 81 void create_service(struct netconfig *); 82 83 void nlm_prog_0(struct svc_req *, SVCXPRT *); 84 void nlm_prog_1(struct svc_req *, SVCXPRT *); 85 void nlm_prog_3(struct svc_req *, SVCXPRT *); 86 void nlm_prog_4(struct svc_req *, SVCXPRT *); 87 void out_of_mem(void); 88 89 static void usage(void); 90 91 void sigalarm_handler(void); 92 93 int 94 main(int argc, char **argv) 95 { 96 int ch, i, s; 97 void *nc_handle; 98 char *endptr, **hosts_bak; 99 struct sigaction sigalarm; 100 int grace_period = 30; 101 struct netconfig *nconf; 102 int have_v6 = 1; 103 int maxrec = RPC_MAXDATASIZE; 104 in_port_t svcport = 0; 105 106 while ((ch = getopt(argc, argv, "d:g:h:p:")) != (-1)) { 107 switch (ch) { 108 case 'd': 109 debug_level = atoi(optarg); 110 if (!debug_level) { 111 usage(); 112 /* NOTREACHED */ 113 } 114 break; 115 case 'g': 116 grace_period = atoi(optarg); 117 if (!grace_period) { 118 usage(); 119 /* NOTREACHED */ 120 } 121 break; 122 case 'h': 123 ++nhosts; 124 hosts_bak = hosts; 125 hosts_bak = realloc(hosts, nhosts * sizeof(char *)); 126 if (hosts_bak == NULL) { 127 if (hosts != NULL) { 128 for (i = 0; i < nhosts; i++) 129 free(hosts[i]); 130 free(hosts); 131 out_of_mem(); 132 } 133 } 134 hosts = hosts_bak; 135 hosts[nhosts - 1] = strdup(optarg); 136 if (hosts[nhosts - 1] == NULL) { 137 for (i = 0; i < (nhosts - 1); i++) 138 free(hosts[i]); 139 free(hosts); 140 out_of_mem(); 141 } 142 break; 143 case 'p': 144 endptr = NULL; 145 svcport = (in_port_t)strtoul(optarg, &endptr, 10); 146 if (endptr == NULL || *endptr != '\0' || 147 svcport == 0 || svcport >= IPPORT_MAX) 148 usage(); 149 svcport_str = strdup(optarg); 150 break; 151 default: 152 case '?': 153 usage(); 154 /* NOTREACHED */ 155 } 156 } 157 if (geteuid()) { /* This command allowed only to root */ 158 fprintf(stderr, "Sorry. You are not superuser\n"); 159 exit(1); 160 } 161 162 rpcb_unset(NLM_PROG, NLM_SM, NULL); 163 rpcb_unset(NLM_PROG, NLM_VERS, NULL); 164 rpcb_unset(NLM_PROG, NLM_VERSX, NULL); 165 rpcb_unset(NLM_PROG, NLM_VERS4, NULL); 166 167 /* 168 * Check if IPv6 support is present. 169 */ 170 s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP); 171 if (s < 0) 172 have_v6 = 0; 173 else 174 close(s); 175 176 rpc_control(RPC_SVC_CONNMAXREC_SET, &maxrec); 177 178 /* 179 * If no hosts were specified, add a wildcard entry to bind to 180 * INADDR_ANY. Otherwise make sure 127.0.0.1 and ::1 are added to the 181 * list. 182 */ 183 if (nhosts == 0) { 184 hosts = malloc(sizeof(char**)); 185 if (hosts == NULL) 186 out_of_mem(); 187 188 hosts[0] = "*"; 189 nhosts = 1; 190 } else { 191 hosts_bak = hosts; 192 if (have_v6) { 193 hosts_bak = realloc(hosts, (nhosts + 2) * 194 sizeof(char *)); 195 if (hosts_bak == NULL) { 196 for (i = 0; i < nhosts; i++) 197 free(hosts[i]); 198 free(hosts); 199 out_of_mem(); 200 } else 201 hosts = hosts_bak; 202 203 nhosts += 2; 204 hosts[nhosts - 2] = "::1"; 205 } else { 206 hosts_bak = realloc(hosts, (nhosts + 1) * sizeof(char *)); 207 if (hosts_bak == NULL) { 208 for (i = 0; i < nhosts; i++) 209 free(hosts[i]); 210 211 free(hosts); 212 out_of_mem(); 213 } else { 214 nhosts += 1; 215 hosts = hosts_bak; 216 } 217 } 218 hosts[nhosts - 1] = "127.0.0.1"; 219 } 220 221 nc_handle = setnetconfig(); 222 while ((nconf = getnetconfig(nc_handle))) { 223 /* We want to listen only on udp6, tcp6, udp, tcp transports */ 224 if (nconf->nc_flag & NC_VISIBLE) { 225 /* Skip if there's no IPv6 support */ 226 if (have_v6 == 0 && strcmp(nconf->nc_protofmly, "inet6") == 0) { 227 /* DO NOTHING */ 228 } else { 229 create_service(nconf); 230 } 231 } 232 } 233 endnetconfig(nc_handle); 234 235 /* 236 * Note that it is NOT sensible to run this program from inetd - the 237 * protocol assumes that it will run immediately at boot time. 238 */ 239 if (daemon(0, debug_level > 0)) { 240 err(1, "cannot fork"); 241 /* NOTREACHED */ 242 } 243 244 openlog("rpc.lockd", 0, LOG_DAEMON); 245 if (debug_level) 246 syslog(LOG_INFO, "Starting, debug level %d", debug_level); 247 else 248 syslog(LOG_INFO, "Starting"); 249 250 sigalarm.sa_handler = (sig_t) sigalarm_handler; 251 sigemptyset(&sigalarm.sa_mask); 252 sigalarm.sa_flags = SA_RESETHAND; /* should only happen once */ 253 sigalarm.sa_flags |= SA_RESTART; 254 if (sigaction(SIGALRM, &sigalarm, NULL) != 0) { 255 syslog(LOG_WARNING, "sigaction(SIGALRM) failed: %s", 256 strerror(errno)); 257 exit(1); 258 } 259 grace_expired = 0; 260 alarm(grace_period); 261 262 svc_run(); /* Should never return */ 263 exit(1); 264 } 265 266 /* 267 * This routine creates and binds sockets on the appropriate 268 * addresses. It gets called one time for each transport and 269 * registrates the service with rpcbind on that trasport. 270 */ 271 void 272 create_service(struct netconfig *nconf) 273 { 274 struct addrinfo hints, *res = NULL; 275 struct sockaddr_in *sin; 276 struct sockaddr_in6 *sin6; 277 struct __rpc_sockinfo si; 278 struct netbuf servaddr; 279 SVCXPRT *transp = NULL; 280 int aicode; 281 int fd; 282 int nhostsbak; 283 int r; 284 int registered = 0; 285 u_int32_t host_addr[4]; /* IPv4 or IPv6 */ 286 287 if ((nconf->nc_semantics != NC_TPI_CLTS) && 288 (nconf->nc_semantics != NC_TPI_COTS) && 289 (nconf->nc_semantics != NC_TPI_COTS_ORD)) 290 return; /* not my type */ 291 292 /* 293 * XXX - using RPC library internal functions. 294 */ 295 if (!__rpc_nconf2sockinfo(nconf, &si)) { 296 syslog(LOG_ERR, "cannot get information for %s", 297 nconf->nc_netid); 298 return; 299 } 300 301 /* Get rpc.statd's address on this transport */ 302 memset(&hints, 0, sizeof hints); 303 hints.ai_flags = AI_PASSIVE; 304 hints.ai_family = si.si_af; 305 hints.ai_socktype = si.si_socktype; 306 hints.ai_protocol = si.si_proto; 307 308 /* 309 * Bind to specific IPs if asked to 310 */ 311 nhostsbak = nhosts; 312 while (nhostsbak > 0) { 313 --nhostsbak; 314 315 /* 316 * XXX - using RPC library internal functions. 317 */ 318 if ((fd = __rpc_nconf2fd(nconf)) < 0) { 319 syslog(LOG_ERR, "cannot create socket for %s", 320 nconf->nc_netid); 321 continue; 322 } 323 324 switch (hints.ai_family) { 325 case AF_INET: 326 if (inet_pton(AF_INET, hosts[nhostsbak], 327 host_addr) == 1) { 328 hints.ai_flags &= AI_NUMERICHOST; 329 } else { 330 /* 331 * Skip if we have an AF_INET6 address. 332 */ 333 if (inet_pton(AF_INET6, hosts[nhostsbak], 334 host_addr) == 1) { 335 close(fd); 336 continue; 337 } 338 } 339 break; 340 case AF_INET6: 341 if (inet_pton(AF_INET6, hosts[nhostsbak], 342 host_addr) == 1) { 343 hints.ai_flags &= AI_NUMERICHOST; 344 } else { 345 /* 346 * Skip if we have an AF_INET address. 347 */ 348 if (inet_pton(AF_INET, hosts[nhostsbak], 349 host_addr) == 1) { 350 close(fd); 351 continue; 352 } 353 } 354 break; 355 default: 356 break; 357 } 358 359 /* 360 * If no hosts were specified, just bind to INADDR_ANY 361 */ 362 if (strcmp("*", hosts[nhostsbak]) == 0) { 363 if (svcport_str == NULL) { 364 res = malloc(sizeof(struct addrinfo)); 365 if (res == NULL) 366 out_of_mem(); 367 res->ai_flags = hints.ai_flags; 368 res->ai_family = hints.ai_family; 369 res->ai_protocol = hints.ai_protocol; 370 switch (res->ai_family) { 371 case AF_INET: 372 sin = malloc(sizeof(struct sockaddr_in)); 373 if (sin == NULL) 374 out_of_mem(); 375 sin->sin_family = AF_INET; 376 sin->sin_port = htons(0); 377 sin->sin_addr.s_addr = htonl(INADDR_ANY); 378 res->ai_addr = (struct sockaddr*) sin; 379 res->ai_addrlen = (socklen_t) 380 sizeof(res->ai_addr); 381 break; 382 case AF_INET6: 383 sin6 = malloc(sizeof(struct sockaddr_in6)); 384 if (sin6 == NULL) 385 out_of_mem(); 386 sin6->sin6_family = AF_INET6; 387 sin6->sin6_port = htons(0); 388 sin6->sin6_addr = in6addr_any; 389 res->ai_addr = (struct sockaddr*) sin6; 390 res->ai_addrlen = (socklen_t) sizeof(res->ai_addr); 391 break; 392 default: 393 break; 394 } 395 } else { 396 if ((aicode = getaddrinfo(NULL, svcport_str, 397 &hints, &res)) != 0) { 398 syslog(LOG_ERR, 399 "cannot get local address for %s: %s", 400 nconf->nc_netid, 401 gai_strerror(aicode)); 402 continue; 403 } 404 } 405 } else { 406 if ((aicode = getaddrinfo(hosts[nhostsbak], svcport_str, 407 &hints, &res)) != 0) { 408 syslog(LOG_ERR, 409 "cannot get local address for %s: %s", 410 nconf->nc_netid, gai_strerror(aicode)); 411 continue; 412 } 413 } 414 415 r = bindresvport_sa(fd, res->ai_addr); 416 if (r != 0) { 417 syslog(LOG_ERR, "bindresvport_sa: %m"); 418 exit(1); 419 } 420 421 transp = svc_tli_create(fd, nconf, NULL, 422 RPC_MAXDATASIZE, RPC_MAXDATASIZE); 423 424 if (transp != NULL) { 425 if (!svc_reg(transp, NLM_PROG, NLM_SM, nlm_prog_0, 426 NULL)) 427 syslog(LOG_ERR, 428 "can't register %s NLM_PROG, NLM_SM service", 429 nconf->nc_netid); 430 431 if (!svc_reg(transp, NLM_PROG, NLM_VERS, nlm_prog_1, 432 NULL)) 433 syslog(LOG_ERR, 434 "can't register %s NLM_PROG, NLM_VERS service", 435 nconf->nc_netid); 436 437 if (!svc_reg(transp, NLM_PROG, NLM_VERSX, nlm_prog_3, 438 NULL)) 439 syslog(LOG_ERR, 440 "can't register %s NLM_PROG, NLM_VERSX service", 441 nconf->nc_netid); 442 443 if (!svc_reg(transp, NLM_PROG, NLM_VERS4, nlm_prog_4, 444 NULL)) 445 syslog(LOG_ERR, 446 "can't register %s NLM_PROG, NLM_VERS4 service", 447 nconf->nc_netid); 448 449 } else 450 syslog(LOG_WARNING, "can't create %s services", 451 nconf->nc_netid); 452 453 if (registered == 0) { 454 registered = 1; 455 memset(&hints, 0, sizeof hints); 456 hints.ai_flags = AI_PASSIVE; 457 hints.ai_family = si.si_af; 458 hints.ai_socktype = si.si_socktype; 459 hints.ai_protocol = si.si_proto; 460 461 if (svcport_str == NULL) { 462 svcport_str = malloc(NI_MAXSERV * sizeof(char)); 463 if (svcport_str == NULL) 464 out_of_mem(); 465 466 if (getnameinfo(res->ai_addr, 467 res->ai_addr->sa_len, NULL, NI_MAXHOST, 468 svcport_str, NI_MAXSERV * sizeof(char), 469 NI_NUMERICHOST | NI_NUMERICSERV)) 470 errx(1, "Cannot get port number"); 471 } 472 473 if((aicode = getaddrinfo(NULL, svcport_str, &hints, 474 &res)) != 0) { 475 syslog(LOG_ERR, "cannot get local address: %s", 476 gai_strerror(aicode)); 477 exit(1); 478 } 479 480 servaddr.buf = malloc(res->ai_addrlen); 481 memcpy(servaddr.buf, res->ai_addr, res->ai_addrlen); 482 servaddr.len = res->ai_addrlen; 483 484 rpcb_set(NLM_PROG, NLM_SM, nconf, &servaddr); 485 rpcb_set(NLM_PROG, NLM_VERS, nconf, &servaddr); 486 rpcb_set(NLM_PROG, NLM_VERSX, nconf, &servaddr); 487 rpcb_set(NLM_PROG, NLM_VERS4, nconf, &servaddr); 488 489 xcreated++; 490 freeaddrinfo(res); 491 } 492 } /* end while */ 493 } 494 495 void 496 sigalarm_handler(void) 497 { 498 499 grace_expired = 1; 500 } 501 502 static void 503 usage(void) 504 { 505 errx(1, "usage: rpc.lockd [-d <debuglevel>]" 506 " [-g <grace period>] [-h <bindip>] [-p <port>]"); 507 } 508 509 /* 510 * Out of memory, fatal 511 */ 512 void 513 out_of_mem(void) 514 { 515 syslog(LOG_ERR, "out of memory"); 516 exit(2); 517 } 518