lockd.c revision 75631
1/* $NetBSD: lockd.c,v 1.7 2000/08/12 18:08:44 thorpej Exp $ */ 2/* $FreeBSD: head/usr.sbin/rpc.lockd/lockd.c 75631 2001-04-17 20:45:23Z alfred $ */ 3 4/* 5 * Copyright (c) 1995 6 * A.R. Gordon (andrew.gordon@net-tel.co.uk). All rights reserved. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 3. All advertising materials mentioning features or use of this software 17 * must display the following acknowledgement: 18 * This product includes software developed for the FreeBSD project 19 * 4. Neither the name of the author nor the names of any co-contributors 20 * may be used to endorse or promote products derived from this software 21 * without specific prior written permission. 22 * 23 * THIS SOFTWARE IS PROVIDED BY ANDREW GORDON AND CONTRIBUTORS ``AS IS'' AND 24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 26 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 33 * SUCH DAMAGE. 34 * 35 */ 36 37#include <sys/cdefs.h> 38#ifndef lint 39__RCSID("$NetBSD: lockd.c,v 1.7 2000/08/12 18:08:44 thorpej Exp $"); 40#endif 41 42/* 43 * main() function for NFS lock daemon. Most of the code in this 44 * file was generated by running rpcgen /usr/include/rpcsvc/nlm_prot.x. 45 * 46 * The actual program logic is in the file lock_proc.c 47 */ 48 49#include <sys/types.h> 50#include <sys/socket.h> 51 52#include <err.h> 53#include <stdio.h> 54#include <stdlib.h> 55#include <errno.h> 56#include <syslog.h> 57#include <signal.h> 58#include <string.h> 59#include <unistd.h> 60#include <libutil.h> 61#include <netconfig.h> 62 63#include <rpc/rpc.h> 64#include <rpcsvc/sm_inter.h> 65 66#include "lockd.h" 67#include <rpcsvc/nlm_prot.h> 68 69int debug_level = 0; /* 0 = no debugging syslog() calls */ 70int _rpcsvcdirty = 0; 71 72int grace_expired; 73int nsm_state; 74pid_t client_pid; 75struct mon mon_host; 76 77void init_nsm(void); 78int main __P((int, char **)); 79void nlm_prog_0 __P((struct svc_req *, SVCXPRT *)); 80void nlm_prog_1 __P((struct svc_req *, SVCXPRT *)); 81void nlm_prog_3 __P((struct svc_req *, SVCXPRT *)); 82void nlm_prog_4 __P((struct svc_req *, SVCXPRT *)); 83void usage __P((void)); 84 85void sigalarm_handler __P((int)); 86 87char *transports[] = { "udp", "tcp", "udp6", "tcp6" }; 88 89int 90main(argc, argv) 91 int argc; 92 char **argv; 93{ 94 SVCXPRT *transp; 95 int ch, i, maxindex, s; 96 struct sigaction sigchild, sigalarm; 97 int grace_period = 30; 98 struct netconfig *nconf; 99 100 while ((ch = getopt(argc, argv, "d:g:")) != (-1)) { 101 switch (ch) { 102 case 'd': 103 debug_level = atoi(optarg); 104 if (!debug_level) { 105 usage(); 106 /* NOTREACHED */ 107 } 108 break; 109 case 'g': 110 grace_period = atoi(optarg); 111 if (!grace_period) { 112 usage(); 113 /* NOTREACHED */ 114 } 115 break; 116 default: 117 case '?': 118 usage(); 119 /* NOTREACHED */ 120 } 121 } 122 if (geteuid()) { /* This command allowed only to root */ 123 fprintf(stderr, "Sorry. You are not superuser\n"); 124 exit(1); 125 } 126 127 (void)rpcb_unset(NLM_PROG, NLM_SM, NULL); 128 (void)rpcb_unset(NLM_PROG, NLM_VERS, NULL); 129 (void)rpcb_unset(NLM_PROG, NLM_VERSX, NULL); 130 (void)rpcb_unset(NLM_PROG, NLM_VERS4, NULL); 131 132 /* 133 * Check if IPv6 support is present. 134 */ 135 s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP); 136 if (s < 0) 137 maxindex = 2; 138 else { 139 close(s); 140 maxindex = 4; 141 } 142 143 for (i = 0; i < maxindex; i++) { 144 nconf = getnetconfigent(transports[i]); 145 if (nconf == NULL) 146 errx(1, "cannot get udp netconf."); 147 148 transp = svc_tli_create(RPC_ANYFD, nconf, NULL, 0, 0); 149 if (transp == NULL) { 150 errx(1, "cannot create %s service.", transports[i]); 151 /* NOTREACHED */ 152 } 153 if (!svc_reg(transp, NLM_PROG, NLM_SM, nlm_prog_0, nconf)) { 154 errx(1, "unable to register (NLM_PROG, NLM_SM, %s)", 155 transports[i]); 156 /* NOTREACHED */ 157 } 158 if (!svc_reg(transp, NLM_PROG, NLM_VERS, nlm_prog_1, nconf)) { 159 errx(1, "unable to register (NLM_PROG, NLM_VERS, %s)", 160 transports[i]); 161 /* NOTREACHED */ 162 } 163 if (!svc_reg(transp, NLM_PROG, NLM_VERSX, nlm_prog_3, nconf)) { 164 errx(1, "unable to register (NLM_PROG, NLM_VERSX, %s)", 165 transports[i]); 166 /* NOTREACHED */ 167 } 168 if (!svc_reg(transp, NLM_PROG, NLM_VERS4, nlm_prog_4, nconf)) { 169 errx(1, "unable to register (NLM_PROG, NLM_VERS4, %s)", 170 transports[i]); 171 /* NOTREACHED */ 172 } 173 freenetconfigent(nconf); 174 } 175 176 /* 177 * Note that it is NOT sensible to run this program from inetd - the 178 * protocol assumes that it will run immediately at boot time. 179 */ 180 if (daemon(0, 0)) { 181 err(1, "cannot fork"); 182 /* NOTREACHED */ 183 } 184 185 openlog("rpc.lockd", 0, LOG_DAEMON); 186 if (debug_level) 187 syslog(LOG_INFO, "Starting, debug level %d", debug_level); 188 else 189 syslog(LOG_INFO, "Starting"); 190 191 sigchild.sa_handler = sigchild_handler; 192 sigemptyset(&sigchild.sa_mask); 193 sigchild.sa_flags = SA_RESTART; 194 if (sigaction(SIGCHLD, &sigchild, NULL) != 0) { 195 syslog(LOG_WARNING, "sigaction(SIGCHLD) failed: %s", 196 strerror(errno)); 197 exit(1); 198 } 199 sigalarm.sa_handler = sigalarm_handler; 200 sigemptyset(&sigalarm.sa_mask); 201 sigalarm.sa_flags = SA_RESETHAND; /* should only happen once */ 202 sigalarm.sa_flags |= SA_RESTART; 203 if (sigaction(SIGALRM, &sigalarm, NULL) != 0) { 204 syslog(LOG_WARNING, "sigaction(SIGALRM) failed: %s", 205 strerror(errno)); 206 exit(1); 207 } 208 grace_expired = 0; 209 if (alarm(10) < 0) { 210 syslog(LOG_WARNING, "alarm failed: %s", 211 strerror(errno)); 212 exit(1); 213 } 214 215 init_nsm(); 216 217 client_pid = client_request(); 218 219 svc_run(); /* Should never return */ 220 exit(1); 221} 222 223void 224sigalarm_handler(s) 225 int s; 226{ 227 grace_expired = 1; 228} 229 230void 231usage() 232{ 233 errx(1, "usage: rpc.lockd [-d <debuglevel>] [-g <grace period>]"); 234} 235 236/* 237 * init_nsm -- 238 * Reset the NSM state-of-the-world and acquire its state. 239 */ 240void 241init_nsm(void) 242{ 243 enum clnt_stat ret; 244 my_id id; 245 sm_stat stat; 246 247 /* 248 * !!! 249 * The my_id structure isn't used by the SM_UNMON_ALL call, as far 250 * as I know. Leave it empty for now. 251 */ 252 memset(&id, 0, sizeof(id)); 253 id.my_name = "NFS NLM"; 254 255 /* 256 * !!! 257 * The statd program must already be registered when lockd runs. 258 */ 259 do { 260 ret = callrpc("localhost", SM_PROG, SM_VERS, SM_UNMON_ALL, 261 xdr_my_id, &id, xdr_sm_stat, &stat); 262 if (ret == RPC_PROGUNAVAIL) { 263 warnx("%lu %s", SM_PROG, clnt_sperrno(ret)); 264 sleep(2); 265 continue; 266 } 267 break; 268 } while (0); 269 270 if (ret != 0) { 271 errx(1, "%lu %s", SM_PROG, clnt_sperrno(ret)); 272 } 273 274 nsm_state = stat.state; 275 276 /* setup constant data for SM_MON calls */ 277 mon_host.mon_id.my_id.my_name = "localhost"; 278 mon_host.mon_id.my_id.my_prog = NLM_PROG; 279 mon_host.mon_id.my_id.my_vers = NLM_SM; 280 mon_host.mon_id.my_id.my_proc = NLM_SM_NOTIFY; /* bsdi addition */ 281} 282