NetBSD/usr.sbin/rpc.statd/stat_proc.c

372 lines
10 KiB
C

/* $NetBSD: stat_proc.c,v 1.8 2009/04/18 13:04:50 lukem Exp $ */
/*
* Copyright (c) 1995
* A.R. Gordon (andrew.gordon@net-tel.co.uk). All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed for the FreeBSD project
* 4. Neither the name of the author nor the names of any co-contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY ANDREW GORDON AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*
*/
#include <sys/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: stat_proc.c,v 1.8 2009/04/18 13:04:50 lukem Exp $");
#endif
#include <errno.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <syslog.h>
#include <signal.h>
#include <unistd.h>
#include <rpc/rpc.h>
#include "statd.h"
/* sm_stat_1 --------------------------------------------------------------- */
/*
* Purpose: RPC call to enquire if a host can be monitored
* Returns: TRUE for any hostname that can be looked up to give
* an address.
*/
struct sm_stat_res *
sm_stat_1_svc(arg, req)
sm_name *arg;
struct svc_req *req;
{
static sm_stat_res smres;
struct addrinfo *ai;
NO_ALARM;
if (debug)
syslog(LOG_DEBUG, "stat called for host %s", arg->mon_name);
if (getaddrinfo(arg->mon_name, NULL, NULL, &ai) == 0) {
smres.res_stat = stat_succ;
freeaddrinfo(ai);
} else {
syslog(LOG_ERR, "invalid hostname to sm_stat: %s",
arg->mon_name);
smres.res_stat = stat_fail;
}
smres.state = status_info.ourState;
ALARM;
return (&smres);
}
/* sm_mon_1 ---------------------------------------------------------------- */
/*
* Purpose: RPC procedure to establish a monitor request
* Returns: Success, unless lack of resources prevents
* the necessary structures from being set up
* to record the request, or if the hostname is not
* valid (as judged by gethostbyname())
*/
struct sm_stat_res *
sm_mon_1_svc(arg, req)
mon *arg;
struct svc_req *req;
{
static sm_stat_res smres;
struct addrinfo *ai;
HostInfo *hp, h;
MonList *lp;
NO_ALARM;
if (debug) {
syslog(LOG_DEBUG, "monitor request for host %s",
arg->mon_id.mon_name);
syslog(LOG_DEBUG, "recall host: %s prog: %d ver: %d proc: %d",
arg->mon_id.my_id.my_name, arg->mon_id.my_id.my_prog,
arg->mon_id.my_id.my_vers, arg->mon_id.my_id.my_proc);
}
smres.res_stat = stat_fail; /* Assume fail until set otherwise */
smres.state = status_info.ourState;
/*
* Find existing host entry, or create one if not found. If
* find_host() fails, it will have logged the error already.
*/
if (getaddrinfo(arg->mon_id.mon_name, NULL, NULL, &ai) != 0) {
syslog(LOG_ERR, "Invalid hostname to sm_mon: %s",
arg->mon_id.mon_name);
return &smres;
}
freeaddrinfo(ai);
if ((hp = find_host(arg->mon_id.mon_name, &h)) == NULL)
memset(hp = &h, 0, sizeof(h));
lp = (MonList *)malloc(sizeof(MonList));
if (!lp)
syslog(LOG_ERR, "Out of memory");
else {
strncpy(lp->notifyHost, arg->mon_id.my_id.my_name,
SM_MAXSTRLEN);
lp->notifyProg = arg->mon_id.my_id.my_prog;
lp->notifyVers = arg->mon_id.my_id.my_vers;
lp->notifyProc = arg->mon_id.my_id.my_proc;
memcpy(lp->notifyData, arg->priv,
sizeof(lp->notifyData));
lp->next = hp->monList;
hp->monList = lp;
change_host(arg->mon_id.mon_name, hp);
sync_file();
smres.res_stat = stat_succ; /* Report success */
}
ALARM;
return (&smres);
}
/* do_unmon ---------------------------------------------------------------- */
/*
* Purpose: Remove a monitor request from a host
* Returns: TRUE if found, FALSE if not found.
* Notes: Common code from sm_unmon_1_svc and sm_unmon_all_1_svc
* In the unlikely event of more than one identical monitor
* request, all are removed.
*/
int
do_unmon(name, hp, ptr)
char *name;
HostInfo *hp;
void *ptr;
{
my_id *idp = ptr;
MonList *lp, *next;
MonList *last = NULL;
int result = FALSE;
lp = hp->monList;
while (lp) {
if (!strncasecmp(idp->my_name, lp->notifyHost, SM_MAXSTRLEN)
&& (idp->my_prog == lp->notifyProg)
&& (idp->my_proc == lp->notifyProc)
&& (idp->my_vers == lp->notifyVers)) {
/* found one. Unhook from chain and free. */
next = lp->next;
if (last)
last->next = next;
else
hp->monList = next;
free(lp);
lp = next;
result = TRUE;
} else {
last = lp;
lp = lp->next;
}
}
return (result);
}
/* sm_unmon_1 -------------------------------------------------------------- */
/*
* Purpose: RPC procedure to release a monitor request.
* Returns: Local machine's status number
* Notes: The supplied mon_id should match the value passed in an
* earlier call to sm_mon_1
*/
struct sm_stat *
sm_unmon_1_svc(arg, req)
mon_id *arg;
struct svc_req *req;
{
static sm_stat smres;
HostInfo *hp, h;
NO_ALARM;
if (debug) {
syslog(LOG_DEBUG, "un-monitor request for host %s",
arg->mon_name);
syslog(LOG_DEBUG, "recall host: %s prog: %d ver: %d proc: %d",
arg->my_id.my_name, arg->my_id.my_prog,
arg->my_id.my_vers, arg->my_id.my_proc);
}
if ((hp = find_host(arg->mon_name, &h)) != NULL) {
if (do_unmon(arg->mon_name, hp, &arg->my_id)) {
change_host(arg->mon_name, hp);
sync_file();
}
else
syslog(LOG_ERR,
"unmon request from %s, no matching monitor",
arg->my_id.my_name);
} else
syslog(LOG_ERR, "unmon request from %s for unknown host %s",
arg->my_id.my_name, arg->mon_name);
smres.state = status_info.ourState;
ALARM;
return (&smres);
}
/* sm_unmon_all_1 ---------------------------------------------------------- */
/*
* Purpose: RPC procedure to release monitor requests.
* Returns: Local machine's status number
* Notes: Releases all monitor requests (if any) from the specified
* host and program number.
*/
struct sm_stat *
sm_unmon_all_1_svc(arg, req)
my_id *arg;
struct svc_req *req;
{
static sm_stat smres;
NO_ALARM;
if (debug) {
syslog(LOG_DEBUG,
"unmon_all for host: %s prog: %d ver: %d proc: %d",
arg->my_name, arg->my_prog, arg->my_vers, arg->my_proc);
}
unmon_hosts();
sync_file();
smres.state = status_info.ourState;
ALARM;
return (&smres);
}
/* sm_simu_crash_1 --------------------------------------------------------- */
/*
* Purpose: RPC procedure to simulate a crash
* Returns: Nothing
* Notes: Standardised mechanism for debug purposes
* The specification says that we should drop all of our
* status information (apart from the list of monitored hosts
* on disc). However, this would confuse the rpc.lockd
* which would be unaware that all of its monitor requests
* had been silently junked. Hence we in fact retain all
* current requests and simply increment the status counter
* and inform all hosts on the monitor list.
*/
void *
sm_simu_crash_1_svc(v, req)
void *v;
struct svc_req *req;
{
static char dummy;
NO_ALARM;
if (debug)
syslog(LOG_DEBUG, "simu_crash called!!");
reset_database();
ALARM;
notify_handler(0);
return (&dummy);
}
/* sm_notify_1 ------------------------------------------------------------- */
/*
* Purpose: RPC procedure notifying local statd of the crash of another
* Returns: Nothing
* Notes: There is danger of deadlock, since it is quite likely that
* the client procedure that we call will in turn call us
* to remove or adjust the monitor request.
* We therefore fork() a process to do the notifications.
* Note that the main HostInfo structure is in a mmap()
* region and so will be shared with the child, but the
* monList pointed to by the HostInfo is in normal memory.
* Hence if we read the monList before forking, we are
* protected from the parent servicing other requests
* that modify the list.
*/
void *
sm_notify_1_svc(arg, req)
stat_chge *arg;
struct svc_req *req;
{
struct timeval timeout = {20, 0}; /* 20 secs timeout */
CLIENT *cli;
static char dummy;
status tx_arg; /* arg sent to callback procedure */
MonList *lp;
HostInfo *hp, h;
pid_t pid;
if (debug)
syslog(LOG_DEBUG, "notify from host %s, new state %d",
arg->mon_name, arg->state);
hp = find_host(arg->mon_name, &h);
if (!hp) {
/* Never heard of this host - why is it notifying us? */
syslog(LOG_DEBUG, "Unsolicited notification from host %s",
arg->mon_name);
return (&dummy);
}
lp = hp->monList;
if (!lp) /* We know this host, but have no outstanding requests. */
return (&dummy);
sync_file();
pid = fork();
if (pid == -1) {
syslog(LOG_ERR, "Unable to fork notify process - %s",
strerror(errno));
return (FALSE);
}
if (pid)
return (&dummy); /* Parent returns */
while (lp) {
tx_arg.mon_name = arg->mon_name;
tx_arg.state = arg->state;
memcpy(tx_arg.priv, lp->notifyData, sizeof(tx_arg.priv));
cli = clnt_create(lp->notifyHost, lp->notifyProg,
lp->notifyVers, "udp");
if (!cli)
syslog(LOG_ERR, "Failed to contact host %s%s",
lp->notifyHost, clnt_spcreateerror(""));
else {
if (clnt_call(cli, lp->notifyProc, xdr_status, &tx_arg,
xdr_void, &dummy, timeout) != RPC_SUCCESS)
syslog(LOG_ERR,
"Failed to call rpc.statd client at host %s",
lp->notifyHost);
clnt_destroy(cli);
}
lp = lp->next;
}
exit(0); /* Child quits */
}