X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=utils%2Fstatd%2Fsm-notify.c;h=581234e15e02adac4f7b2216818c8f9b8355ae21;hb=99979a6cf2f862d2365d27fa90fab4416c374903;hp=d58e0be421859ed6c59880e917d53a53a874b877;hpb=f846abde5faa4742b4823fa981080b1f5dac66b1;p=nfs-utils.git diff --git a/utils/statd/sm-notify.c b/utils/statd/sm-notify.c index d58e0be..581234e 100644 --- a/utils/statd/sm-notify.c +++ b/utils/statd/sm-notify.c @@ -8,6 +8,7 @@ #include #endif +#include #include #include #include @@ -28,22 +29,9 @@ #include #include -#ifndef BASEDIR -# ifdef NFS_STATEDIR -# define BASEDIR NFS_STATEDIR -# else -# define BASEDIR "/var/lib/nfs" -# endif -#endif - -#define DEFAULT_SM_STATE_PATH BASEDIR "/state" -#define DEFAULT_SM_DIR_PATH BASEDIR "/sm" -#define DEFAULT_SM_BAK_PATH DEFAULT_SM_DIR_PATH ".bak" - -char *_SM_BASE_PATH = BASEDIR; -char *_SM_STATE_PATH = DEFAULT_SM_STATE_PATH; -char *_SM_DIR_PATH = DEFAULT_SM_DIR_PATH; -char *_SM_BAK_PATH = DEFAULT_SM_BAK_PATH; +#include "xlog.h" +#include "nsm.h" +#include "nfsrpc.h" #define NSM_PROG 100024 #define NSM_PROGRAM 100024 @@ -56,7 +44,6 @@ char *_SM_BAK_PATH = DEFAULT_SM_BAK_PATH; struct nsm_host { struct nsm_host * next; char * name; - char * path; struct sockaddr_storage addr; struct addrinfo *ai; time_t last_used; @@ -67,74 +54,75 @@ struct nsm_host { }; static char nsm_hostname[256]; -static uint32_t nsm_state; +static int nsm_state; static int opt_debug = 0; -static int opt_quiet = 0; -static int opt_update_state = 1; +static _Bool opt_update_state = true; static unsigned int opt_max_retry = 15 * 60; static char * opt_srcaddr = 0; static uint16_t opt_srcport = 0; -static int log_syslog = 0; -static unsigned int nsm_get_state(int); static void notify(void); static int notify_host(int, struct nsm_host *); static void recv_reply(int); -static void backup_hosts(const char *, const char *); -static void get_hosts(const char *); static void insert_host(struct nsm_host *); static struct nsm_host *find_host(uint32_t); -static void nsm_log(int fac, const char *fmt, ...); static int record_pid(void); -static void drop_privs(void); -static void set_kernel_nsm_state(int state); static struct nsm_host * hosts = NULL; -/* - * Address handling utilities - */ - -static unsigned short smn_get_port(const struct sockaddr *sap) +static struct addrinfo *smn_lookup(const char *name) { - switch (sap->sa_family) { - case AF_INET: - return ntohs(((struct sockaddr_in *)sap)->sin_port); - case AF_INET6: - return ntohs(((struct sockaddr_in6 *)sap)->sin6_port); + struct addrinfo *ai, hint = { +#if HAVE_DECL_AI_ADDRCONFIG + .ai_flags = AI_ADDRCONFIG, +#endif /* HAVE_DECL_AI_ADDRCONFIG */ + .ai_family = AF_INET, + .ai_protocol = IPPROTO_UDP, + }; + int error; + + error = getaddrinfo(name, NULL, &hint, &ai); + if (error) { + xlog(D_GENERAL, "getaddrinfo(3): %s", gai_strerror(error)); + return NULL; } - return 0; + + return ai; } -static void smn_set_port(struct sockaddr *sap, const unsigned short port) +__attribute_malloc__ +static struct nsm_host * +smn_alloc_host(const char *hostname, const time_t timestamp) { - switch (sap->sa_family) { - case AF_INET: - ((struct sockaddr_in *)sap)->sin_port = htons(port); - break; - case AF_INET6: - ((struct sockaddr_in6 *)sap)->sin6_port = htons(port); - break; + struct nsm_host *host; + + host = calloc(1, sizeof(*host)); + if (host == NULL) + goto out_nomem; + + host->name = strdup(hostname); + if (host->name == NULL) { + free(host); + goto out_nomem; } -} -static struct addrinfo *smn_lookup(const sa_family_t family, const char *name) -{ - struct addrinfo *ai, hint = { - .ai_family = family, - .ai_protocol = IPPROTO_UDP, - }; + host->last_used = timestamp; + host->timeout = NSM_TIMEOUT; + host->retries = 100; /* force address retry */ - if (getaddrinfo(name, NULL, &hint, &ai) != 0) - return NULL; + return host; - return ai; +out_nomem: + xlog_warn("Unable to allocate memory"); + return NULL; } static void smn_forget_host(struct nsm_host *host) { - unlink(host->path); - free(host->path); + xlog(D_CALL, "Removing %s from notify list", host->name); + + nsm_delete_notified_host(host->name); + free(host->name); if (host->ai) freeaddrinfo(host->ai); @@ -142,13 +130,37 @@ static void smn_forget_host(struct nsm_host *host) free(host); } +static unsigned int +smn_get_host(const char *hostname, + __attribute__ ((unused)) const struct sockaddr *sap, + __attribute__ ((unused)) const struct mon *m, + const time_t timestamp) +{ + struct nsm_host *host; + + host = smn_alloc_host(hostname, timestamp); + if (host == NULL) + return 0; + + insert_host(host); + xlog(D_GENERAL, "Added host %s to notify list", hostname); + return 1; +} + int main(int argc, char **argv) { int c; int force = 0; + char * progname; - while ((c = getopt(argc, argv, "dm:np:v:qP:f")) != -1) { + progname = strrchr(argv[0], '/'); + if (progname != NULL) + progname++; + else + progname = argv[0]; + + while ((c = getopt(argc, argv, "dm:np:v:P:f")) != -1) { switch (c) { case 'f': force = 1; @@ -160,7 +172,7 @@ main(int argc, char **argv) opt_max_retry = atoi(optarg) * 60; break; case 'n': - opt_update_state = 0; + opt_update_state = false; break; case 'p': opt_srcport = atoi(optarg); @@ -168,24 +180,9 @@ main(int argc, char **argv) case 'v': opt_srcaddr = optarg; break; - case 'q': - opt_quiet = 1; - break; case 'P': - _SM_BASE_PATH = strdup(optarg); - _SM_STATE_PATH = malloc(strlen(optarg)+1+sizeof("state")); - _SM_DIR_PATH = malloc(strlen(optarg)+1+sizeof("sm")); - _SM_BAK_PATH = malloc(strlen(optarg)+1+sizeof("sm.bak")); - if (_SM_BASE_PATH == NULL || - _SM_STATE_PATH == NULL || - _SM_DIR_PATH == NULL || - _SM_BAK_PATH == NULL) { - nsm_log(LOG_ERR, "unable to allocate memory"); + if (!nsm_setup_pathnames(argv[0], optarg)) exit(1); - } - strcat(strcpy(_SM_STATE_PATH, _SM_BASE_PATH), "/state"); - strcat(strcpy(_SM_DIR_PATH, _SM_BASE_PATH), "/sm"); - strcat(strcpy(_SM_BAK_PATH, _SM_BASE_PATH), "/sm.bak"); break; default: @@ -195,18 +192,26 @@ main(int argc, char **argv) if (optind < argc) { usage: fprintf(stderr, - "Usage: sm-notify [-dfq] [-m max-retry-minutes] [-p srcport]\n" - " [-P /path/to/state/directory] [-v my_host_name]\n"); + "Usage: %s -notify [-dfq] [-m max-retry-minutes] [-p srcport]\n" + " [-P /path/to/state/directory] [-v my_host_name]\n", + progname); exit(1); } - log_syslog = 1; - openlog("sm-notify", LOG_PID, LOG_DAEMON); + xlog_syslog(1); + if (opt_debug) { + xlog_stderr(1); + xlog_config(D_ALL, 1); + } else + xlog_stderr(0); + + xlog_open(progname); + xlog(L_NOTICE, "Version " VERSION " starting"); - if (strcmp(_SM_BASE_PATH, BASEDIR) == 0) { - if (record_pid() == 0 && force == 0 && opt_update_state == 1) { + if (nsm_is_default_parentdir()) { + if (record_pid() == 0 && force == 0 && opt_update_state) { /* already run, don't try again */ - nsm_log(LOG_NOTICE, "Already notifying clients; Exiting!"); + xlog(L_NOTICE, "Already notifying clients; Exiting!"); exit(0); } } @@ -215,31 +220,26 @@ usage: fprintf(stderr, strncpy(nsm_hostname, opt_srcaddr, sizeof(nsm_hostname)-1); } else if (gethostname(nsm_hostname, sizeof(nsm_hostname)) < 0) { - nsm_log(LOG_ERR, "Failed to obtain name of local host: %s", - strerror(errno)); + xlog(L_ERROR, "Failed to obtain name of local host: %m"); exit(1); } - backup_hosts(_SM_DIR_PATH, _SM_BAK_PATH); - get_hosts(_SM_BAK_PATH); - - /* If there are not hosts to notify, just exit */ - if (!hosts) { - nsm_log(LOG_DEBUG, "No hosts to notify; exiting"); + (void)nsm_retire_monitored_hosts(); + if (nsm_load_notify_list(smn_get_host) == 0) { + xlog(D_GENERAL, "No hosts to notify; exiting"); return 0; } - /* Get and update the NSM state. This will call sync() */ nsm_state = nsm_get_state(opt_update_state); - set_kernel_nsm_state(nsm_state); + if (nsm_state == 0) + exit(1); + nsm_update_kernel_state(nsm_state); if (!opt_debug) { - if (!opt_quiet) - printf("Backgrounding to notify hosts...\n"); + xlog(L_NOTICE, "Backgrounding to notify hosts...\n"); if (daemon(0, 0) < 0) { - nsm_log(LOG_ERR, "unable to background: %s", - strerror(errno)); + xlog(L_ERROR, "unable to background: %m"); exit(1); } @@ -255,8 +255,7 @@ usage: fprintf(stderr, while ((hp = hosts) != 0) { hosts = hp->next; - nsm_log(LOG_NOTICE, - "Unable to notify %s, giving up", + xlog(L_NOTICE, "Unable to notify %s, giving up", hp->name); } exit(1); @@ -280,8 +279,7 @@ notify(void) retry: sock = socket(AF_INET, SOCK_DGRAM, 0); if (sock < 0) { - nsm_log(LOG_ERR, "Failed to create RPC socket: %s", - strerror(errno)); + xlog(L_ERROR, "Failed to create RPC socket: %m"); exit(1); } fcntl(sock, F_SETFL, O_NONBLOCK); @@ -291,9 +289,9 @@ notify(void) /* Bind source IP if provided on command line */ if (opt_srcaddr) { - struct addrinfo *ai = smn_lookup(AF_INET, opt_srcaddr); + struct addrinfo *ai = smn_lookup(opt_srcaddr); if (!ai) { - nsm_log(LOG_ERR, + xlog(L_ERROR, "Not a valid hostname or address: \"%s\"", opt_srcaddr); exit(1); @@ -308,10 +306,9 @@ notify(void) /* Use source port if provided on the command line, * otherwise use bindresvport */ if (opt_srcport) { - smn_set_port(local_addr, opt_srcport); + nfs_set_port(local_addr, opt_srcport); if (bind(sock, local_addr, sizeof(struct sockaddr_in)) < 0) { - nsm_log(LOG_ERR, "Failed to bind RPC socket: %s", - strerror(errno)); + xlog(L_ERROR, "Failed to bind RPC socket: %m"); exit(1); } } else { @@ -330,7 +327,8 @@ notify(void) if (opt_max_retry) failtime = time(NULL) + opt_max_retry; - drop_privs(); + if (!nsm_drop_privileges(-1)) + exit(1); while (hosts) { struct pollfd pfd; @@ -367,7 +365,7 @@ notify(void) if (hosts == NULL) return; - nsm_log(LOG_DEBUG, "Host %s due in %ld seconds", + xlog(D_GENERAL, "Host %s due in %ld seconds", hosts->name, wait); pfd.fd = sock; @@ -402,13 +400,11 @@ notify_host(int sock, struct nsm_host *host) host->xid = xid++; if (host->ai == NULL) { - host->ai = smn_lookup(AF_UNSPEC, host->name); + host->ai = smn_lookup(host->name); if (host->ai == NULL) { - nsm_log(LOG_WARNING, - "%s doesn't seem to be a valid address," - " skipped", host->name); - smn_forget_host(host); - return 1; + xlog_warn("DNS resolution of %s failed; " + "retrying later", host->name); + return 0; } } @@ -424,29 +420,37 @@ notify_host(int sock, struct nsm_host *host) * point. */ if (host->retries >= 4) { - struct addrinfo *first = host->ai; - struct addrinfo **next = &host->ai; - - /* remove the first entry from the list */ - host->ai = first->ai_next; - first->ai_next = NULL; - /* find the end of the list */ - next = &first->ai_next; - while ( *next ) - next = & (*next)->ai_next; - /* put first entry at end */ - *next = first; - memcpy(&host->addr, first->ai_addr, first->ai_addrlen); - smn_set_port((struct sockaddr *)&host->addr, 0); + /* don't rotate if there is only one addrinfo */ + if (host->ai->ai_next == NULL) + memcpy(&host->addr, host->ai->ai_addr, + host->ai->ai_addrlen); + else { + struct addrinfo *first = host->ai; + struct addrinfo **next = &host->ai; + + /* remove the first entry from the list */ + host->ai = first->ai_next; + first->ai_next = NULL; + /* find the end of the list */ + next = &first->ai_next; + while ( *next ) + next = & (*next)->ai_next; + /* put first entry at end */ + *next = first; + memcpy(&host->addr, first->ai_addr, + first->ai_addrlen); + } + + nfs_set_port((struct sockaddr *)&host->addr, 0); host->retries = 0; } memcpy(dest, &host->addr, destlen); - if (smn_get_port(dest) == 0) { + if (nfs_get_port(dest) == 0) { /* Build a PMAP packet */ - nsm_log(LOG_DEBUG, "Sending portmap query to %s", host->name); + xlog(D_GENERAL, "Sending portmap query to %s", host->name); - smn_set_port(dest, 111); + nfs_set_port(dest, 111); *p++ = htonl(100000); *p++ = htonl(2); *p++ = htonl(3); @@ -461,7 +465,7 @@ notify_host(int sock, struct nsm_host *host) *p++ = 0; } else { /* Build an SM_NOTIFY packet */ - nsm_log(LOG_DEBUG, "Sending SM_NOTIFY to %s", host->name); + xlog(D_GENERAL, "Sending SM_NOTIFY to %s", host->name); *p++ = htonl(NSM_PROGRAM); *p++ = htonl(NSM_VERSION); @@ -481,8 +485,8 @@ notify_host(int sock, struct nsm_host *host) len = (p - msgbuf) << 2; if (sendto(sock, msgbuf, len, 0, dest, destlen) < 0) - nsm_log(LOG_WARNING, "Sending Reboot Notification to " - "'%s' failed: errno %d (%s)", host->name, errno, strerror(errno)); + xlog_warn("Sending Reboot Notification to " + "'%s' failed: errno %d (%m)", host->name, errno); return 0; } @@ -503,7 +507,7 @@ recv_reply(int sock) if (res < 0) return; - nsm_log(LOG_DEBUG, "Received packet..."); + xlog(D_GENERAL, "Received packet..."); p = msgbuf; end = p + (res >> 2); @@ -522,7 +526,7 @@ recv_reply(int sock) return; sap = (struct sockaddr *)&hp->addr; - if (smn_get_port(sap) == 0) { + if (nfs_get_port(sap) == 0) { /* This was a portmap request */ unsigned int port; @@ -534,11 +538,11 @@ recv_reply(int sock) if (port == 0) { /* No binding for statd. Delay the next * portmap query for max timeout */ - nsm_log(LOG_DEBUG, "No statd on %s", hp->name); + xlog(D_GENERAL, "No statd on %s", hp->name); hp->timeout = NSM_MAX_TIMEOUT; hp->send_next += NSM_MAX_TIMEOUT; } else { - smn_set_port(sap, port); + nfs_set_port(sap, port); if (hp->timeout >= NSM_MAX_TIMEOUT / 4) hp->timeout = NSM_MAX_TIMEOUT / 4; } @@ -550,7 +554,7 @@ recv_reply(int sock) * packet) */ if (p <= end) { - nsm_log(LOG_DEBUG, "Host %s notified successfully", + xlog(D_GENERAL, "Host %s notified successfully", hp->name); smn_forget_host(hp); return; @@ -561,87 +565,6 @@ fail: /* Re-insert the host */ insert_host(hp); } -/* - * Back up all hosts from the sm directory to sm.bak - */ -static void -backup_hosts(const char *dirname, const char *bakname) -{ - struct dirent *de; - DIR *dir; - - if (!(dir = opendir(dirname))) { - nsm_log(LOG_WARNING, - "Failed to open %s: %s", dirname, strerror(errno)); - return; - } - - while ((de = readdir(dir)) != NULL) { - char src[1024], dst[1024]; - - if (de->d_name[0] == '.') - continue; - - snprintf(src, sizeof(src), "%s/%s", dirname, de->d_name); - snprintf(dst, sizeof(dst), "%s/%s", bakname, de->d_name); - if (rename(src, dst) < 0) { - nsm_log(LOG_WARNING, - "Failed to rename %s -> %s: %m", - src, dst); - } - } - closedir(dir); -} - -/* - * Get all entries from sm.bak and convert them to host entries - */ -static void -get_hosts(const char *dirname) -{ - struct nsm_host *host; - struct dirent *de; - DIR *dir; - - if (!(dir = opendir(dirname))) { - nsm_log(LOG_WARNING, - "Failed to open %s: %s", dirname, strerror(errno)); - return; - } - - host = NULL; - while ((de = readdir(dir)) != NULL) { - struct stat stb; - char path[1024]; - - if (de->d_name[0] == '.') - continue; - if (host == NULL) - host = calloc(1, sizeof(*host)); - if (host == NULL) { - nsm_log(LOG_WARNING, "Unable to allocate memory"); - return; - } - - snprintf(path, sizeof(path), "%s/%s", dirname, de->d_name); - if (stat(path, &stb) < 0) - continue; - - host->last_used = stb.st_mtime; - host->timeout = NSM_TIMEOUT; - host->path = strdup(path); - host->name = strdup(de->d_name); - host->retries = 100; /* force address retry */ - - insert_host(host); - host = NULL; - } - closedir(dir); - - if (host) - free(host); -} - /* * Insert host into sorted list */ @@ -689,84 +612,6 @@ find_host(uint32_t xid) return NULL; } - -/* - * Retrieve the current NSM state - */ -static unsigned int -nsm_get_state(int update) -{ - char newfile[PATH_MAX]; - int fd, state; - - if ((fd = open(_SM_STATE_PATH, O_RDONLY)) < 0) { - if (!opt_quiet) { - nsm_log(LOG_WARNING, "%s: %m", _SM_STATE_PATH); - nsm_log(LOG_WARNING, "Creating %s, set initial state 1", - _SM_STATE_PATH); - } - state = 1; - update = 1; - } else { - if (read(fd, &state, sizeof(state)) != sizeof(state)) { - nsm_log(LOG_WARNING, - "%s: bad file size, setting state = 1", - _SM_STATE_PATH); - state = 1; - update = 1; - } else { - if (!(state & 1)) - state += 1; - } - close(fd); - } - - if (update) { - state += 2; - snprintf(newfile, sizeof(newfile), - "%s.new", _SM_STATE_PATH); - if ((fd = open(newfile, O_CREAT|O_WRONLY, 0644)) < 0) { - nsm_log(LOG_ERR, "Cannot create %s: %m", newfile); - exit(1); - } - if (write(fd, &state, sizeof(state)) != sizeof(state)) { - nsm_log(LOG_ERR, - "Failed to write state to %s", newfile); - exit(1); - } - close(fd); - if (rename(newfile, _SM_STATE_PATH) < 0) { - nsm_log(LOG_ERR, - "Cannot create %s: %m", _SM_STATE_PATH); - exit(1); - } - sync(); - } - - return state; -} - -/* - * Log a message - */ -static void -nsm_log(int fac, const char *fmt, ...) -{ - va_list ap; - - if (fac == LOG_DEBUG && !opt_debug) - return; - - va_start(ap, fmt); - if (log_syslog) - vsyslog(fac, fmt, ap); - else { - vfprintf(stderr, fmt, ap); - fputs("\n", stderr); - } - va_end(ap); -} - /* * Record pid in /var/run/sm-notify.pid * This file should remain until a reboot, even if the @@ -776,54 +621,20 @@ nsm_log(int fac, const char *fmt, ...) static int record_pid(void) { char pid[20]; + ssize_t len; int fd; - snprintf(pid, 20, "%d\n", getpid()); + (void)snprintf(pid, sizeof(pid), "%d\n", (int)getpid()); fd = open("/var/run/sm-notify.pid", O_CREAT|O_EXCL|O_WRONLY, 0600); if (fd < 0) return 0; - write(fd, pid, strlen(pid)); - close(fd); - return 1; -} - -/* Drop privileges to match owner of state-directory - * (in case a reply triggers some unknown bug). - */ -static void drop_privs(void) -{ - struct stat st; - - if (stat(_SM_DIR_PATH, &st) == -1 && - stat(_SM_BASE_PATH, &st) == -1) { - st.st_uid = 0; - st.st_gid = 0; - } - - if (st.st_uid == 0) { - nsm_log(LOG_WARNING, - "sm-notify running as root. chown %s to choose different user", - _SM_DIR_PATH); - return; - } - setgroups(0, NULL); - if (setgid(st.st_gid) == -1 - || setuid(st.st_uid) == -1) { - nsm_log(LOG_ERR, "Fail to drop privileges"); - exit(1); + len = write(fd, pid, strlen(pid)); + if ((len < 0) || ((size_t)len != strlen(pid))) { + xlog_warn("Writing to pid file failed: errno %d (%m)", + errno); } -} - -static void set_kernel_nsm_state(int state) -{ - int fd; - fd = open("/proc/sys/fs/nfs/nsm_local_state",O_WRONLY); - if (fd >= 0) { - char buf[20]; - snprintf(buf, sizeof(buf), "%d", state); - write(fd, buf, strlen(buf)); - close(fd); - } + (void)close(fd); + return 1; }