Initial revision

1 files changed, 126 insertions, 0 deletions

diff --git a/utils/statd/state.c b/utils/statd/state.c
new file mode 100644
index 0000000..101c00b
--- /dev/null
+++ b/utils/statd/state.c
@@ -0,0 +1,126 @@
+/*
+ * Copyright (C) 1995-1997, 1999 Jeffrey A. Uphoff
+ * Modified by Olaf Kirch, 1996.
+ * Modified by H.J. Lu, 1998.
+ *
+ * NSM for Linux.
+ */
+
+#include "config.h"
+#include <dirent.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <netdb.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include "statd.h"
+
+
+/* 
+ * Most NSM's keep the status number in an ASCII file.  I'm keeping it
+ * as an int (4-byte binary) for now...
+ */
+void
+change_state (void)
+{
+  int fd, size;
+  extern short int restart;
+  
+  if ((fd = open (SM_STAT_PATH, O_CREAT | O_RDWR, S_IRUSR | S_IWUSR)) == -1)
+    die ("open (%s): %s", SM_STAT_PATH, strerror (errno));
+
+  if ((size = read (fd, &MY_STATE, sizeof MY_STATE)) == -1)
+    die ("read (%s): %s", SM_STAT_PATH, strerror (errno));
+
+  if (size != 0 && size != sizeof MY_STATE) {
+    log (L_ERROR, "Error in status file format...correcting.");
+
+    if (close (fd) == -1)
+      die ("close (%s): %s", SM_STAT_PATH, strerror (errno));
+
+    if ((fd = creat (SM_STAT_PATH, S_IRUSR | S_IWUSR)) == -1)
+      die ("creat (%s): %s", SM_STAT_PATH, strerror (errno));
+  }
+  log (L_DEBUG, "New state: %u", (++MY_STATE % 2) ? MY_STATE : ++MY_STATE);
+
+  if (lseek (fd, 0, SEEK_SET) == -1)
+    die ("lseek (%s): %s", SM_STAT_PATH, strerror (errno));
+
+  if (write (fd, &MY_STATE, sizeof MY_STATE) != sizeof MY_STATE)
+    die ("write (%s): %s", SM_STAT_PATH, strerror (errno));
+
+  if (fsync (fd) == -1)
+    log (L_ERROR, "fsync (%s): %s", SM_STAT_PATH, strerror (errno));
+
+  if (close (fd) == -1)
+    log (L_ERROR, "close (%s): %s", SM_STAT_PATH, strerror (errno));
+
+  if (!restart) {
+    char fullhost[SM_MAXSTRLEN + 1];
+    struct hostent *hostinfo;
+
+    if (gethostname (fullhost, SM_MAXSTRLEN) == -1)
+      die ("gethostname: %s", strerror (errno));
+
+    if ((hostinfo = gethostbyname (fullhost)) == NULL)
+      log (L_ERROR, "gethostbyname error for %s", fullhost);
+    else {
+      strncpy (fullhost, hostinfo->h_name, sizeof (fullhost) - 1);
+      fullhost[sizeof (fullhost) - 1] = '\0';
+    }
+
+    MY_NAME = xstrdup (fullhost);
+  }
+}
+
+
+/* 
+ * Fairly traditional use of two directories for this.
+ */
+void 
+shuffle_dirs (void)
+{
+  DIR *nld;
+  struct dirent *de;
+  struct stat st;
+  char *src, *dst;
+  int len1, len2, len;
+  
+  if (stat (SM_DIR, &st) == -1 && errno != ENOENT)
+    die ("stat (%s): %s", SM_DIR, strerror (errno));
+
+  if (!S_ISDIR (st.st_mode))
+    if (mkdir (SM_DIR, S_IRWXU) == -1)
+      die ("mkdir (%s): %s", SM_DIR, strerror (errno));
+
+  memset (&st, 0, sizeof st);
+
+  if (stat (SM_BAK_DIR, &st) == -1 && errno != ENOENT)
+    die ("stat (%s): %s", SM_BAK_DIR, strerror (errno));
+
+  if (!S_ISDIR (st.st_mode))
+    if (mkdir (SM_BAK_DIR, S_IRWXU) == -1)
+      die ("mkdir (%s): %s", SM_BAK_DIR, strerror (errno));
+
+  if (!(nld = opendir (SM_DIR)))
+    die ("opendir (%s): %s", SM_DIR, strerror (errno));
+
+  len1=strlen(SM_DIR);
+  len2=strlen(SM_BAK_DIR);
+  while ((de = readdir (nld))) {
+    if (de->d_name[0] == '.')
+      continue;
+    len=strlen(de->d_name);
+    src=xmalloc(len1+len+2);
+    dst=xmalloc(len2+len+2);
+    sprintf (src, "%s/%s", SM_DIR, de->d_name);
+    sprintf (dst, "%s/%s", SM_BAK_DIR, de->d_name);
+    if (rename (src, dst) == -1)
+      die ("rename (%s to %s): %s", SM_DIR, SM_BAK_DIR, strerror (errno));
+    free(src);
+    free(dst);
+  }
+  if (closedir (nld) == -1)
+    log (L_ERROR, "closedir (%s): %s", SM_DIR, strerror (errno));
+}