From 7b01c4762ae242d2114cc6a59994f678b7d253fc Mon Sep 17 00:00:00 2001
From: Neil Brown <neilb@suse.de>
Date: Thu, 29 Mar 2007 13:07:48 +1000
Subject: [PATCH] statd - keep persistent state in sm/* files.

If statd dies and is restarted, it forgets what peers the kernel
is interested in monitoring, and so will not forward NOTIFY
requests properly.

With this patch the required information is recorded in the files
in /var/lib/nfs/sm/* so that a kill/restart does what you might
hope.

Signed-off-by: Neil Brown <neilb@suse.de>
---
 utils/statd/monitor.c | 77 ++++++++++++++++++++++++++++++++++++++++++-
 utils/statd/statd.c   |  6 ++++
 utils/statd/statd.h   |  1 +
 3 files changed, 83 insertions(+), 1 deletion(-)

diff --git a/utils/statd/monitor.c b/utils/statd/monitor.c
index 8ee0441..b452082 100644
--- a/utils/statd/monitor.c
+++ b/utils/statd/monitor.c
@@ -19,6 +19,7 @@
 #include <sys/stat.h>
 #include <errno.h>
 #include <arpa/inet.h>
+#include <dirent.h>
 #include "misc.h"
 #include "statd.h"
 #include "notlist.h"
@@ -26,6 +27,7 @@
 
 notify_list *		rtnl = NULL;	/* Run-time notify list. */
 
+#define LINELEN (4*(8+1)+SM_PRIV_SIZE*2+1)
 
 /*
  * Services SM_MON requests.
@@ -181,13 +183,28 @@ sm_mon_1_svc(struct mon *argp, struct svc_req *rqstp)
 
 	path=xmalloc(strlen(SM_DIR)+strlen(mon_name)+2);
 	sprintf(path, "%s/%s", SM_DIR, mon_name);
-	if ((fd = open(path, O_WRONLY|O_SYNC|O_CREAT, S_IRUSR|S_IWUSR)) < 0) {
+	if ((fd = open(path, O_WRONLY|O_SYNC|O_CREAT|O_APPEND,
+		       S_IRUSR|S_IWUSR)) < 0) {
 		/* Didn't fly.  We won't monitor. */
 		note(N_ERROR, "creat(%s) failed: %s", path, strerror (errno));
 		nlist_free(NULL, clnt);
 		free(path);
 		goto failure;
 	}
+	{
+		char buf[LINELEN + 1 + SM_MAXSTRLEN + 2];
+		char *e;
+		int i;
+		e = buf + sprintf(buf, "%08x %08x %08x %08x ",
+				  my_addr.s_addr, id->my_prog,
+				  id->my_vers, id->my_proc);
+		for (i=0; i<SM_PRIV_SIZE; i++)
+			e += sprintf(e, "%02x", 0xff & (argp->priv[i]));
+		if (e+1-buf != LINELEN) abort();
+		e += sprintf(e, " %s\n", mon_name);
+		write(fd, buf, e-buf);
+	}
+
 	free(path);
 	/* PRC: do the HA callout: */
 	ha_callout("add-client", mon_name, my_name, -1);
@@ -204,6 +221,64 @@ failure:
 	return (&result);
 }
 
+void load_state(void)
+{
+	DIR *d;
+	struct dirent *de;
+	char buf[LINELEN + 1 + SM_MAXSTRLEN + 2];
+
+	d = opendir(SM_DIR);
+	if (!d)
+		return;
+	while ((de = readdir(d))) {
+		char *path;
+		FILE *f;
+		int p;
+
+		if (de->d_name[0] == '.')
+			continue;
+		path = xmalloc(strlen(SM_DIR)+strlen(de->d_name)+2);
+		sprintf(path, "%s/%s", SM_DIR, de->d_name);
+		f = fopen(path, "r");
+		free(path);
+		if (f == NULL)
+			continue;
+		while (fgets(buf, sizeof(buf), f) != NULL) {
+			int addr, proc, prog, vers;
+			char priv[SM_PRIV_SIZE];
+			char *b;
+			int i;
+			notify_list	*clnt;
+
+			buf[sizeof(buf)-1] = 0;
+			b = strchr(buf, '\n');
+			if (b) *b = 0;
+			sscanf(buf, "%x %x %x %x ",
+			       &addr, &prog, &vers, &proc);
+			b = buf+36;
+			for (i=0; i<SM_PRIV_SIZE; i++) {
+				sscanf(b, "%2x", &p);
+				priv[i] = p;
+				b += 2;
+			}
+			b++;
+			clnt = nlist_new("127.0.0.1", b, 0);
+			if (!clnt)
+				break;
+			NL_ADDR(clnt).s_addr = addr;
+			NL_MY_PROG(clnt) = prog;
+			NL_MY_VERS(clnt) = vers;
+			NL_MY_PROC(clnt) = proc;
+			memcpy(NL_PRIV(clnt), priv, SM_PRIV_SIZE);
+			nlist_insert(&rtnl, clnt);
+		}
+		fclose(f);
+	}
+	closedir(d);
+}
+
+
+
 
 /*
  * Services SM_UNMON requests.
diff --git a/utils/statd/statd.c b/utils/statd/statd.c
index 6c5dbde..a44f39f 100644
--- a/utils/statd/statd.c
+++ b/utils/statd/statd.c
@@ -479,6 +479,12 @@ int main (int argc, char **argv)
 	/* Make sure we have a privilege port for calling into the kernel */
 	statd_get_socket();
 
+	/* If sm-notify didn't take all the state files, load
+	 * state information into our notify-list so we can
+	 * pass on any SM_NOTIFY that arrives
+	 */
+	load_state();
+
 	for (;;) {
 		pmap_unset (SM_PROG, SM_VERS);
 
diff --git a/utils/statd/statd.h b/utils/statd/statd.h
index 441eaaa..b7ea40b 100644
--- a/utils/statd/statd.h
+++ b/utils/statd/statd.h
@@ -54,6 +54,7 @@ extern int	process_reply(FD_SET_TYPE *);
 extern char *	xstrdup(const char *);
 extern void *	xmalloc(size_t);
 extern void	xunlink (char *, char *, short int);
+extern void	load_state(void);
 
 /*
  * Host status structure and macros.
-- 
2.39.2