1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
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));
}
|
