ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/gvpe/src/util.C
(Generate patch)

Comparing gvpe/src/util.C (file contents):
Revision 1.4 by pcg, Wed Apr 2 03:06:22 2003 UTC vs.
Revision 1.23 by pcg, Thu Aug 7 19:07:03 2008 UTC

1/* 1/*
2 util.C -- process management and other utility functions 2 util.C -- process management and other utility functions
3 Copyright (C) 2003-2008 Marc Lehmann <gvpe@schmorp.de>
3 4
4 Some of these are taken from tinc, see the AUTHORS file. 5 Some of these are taken from tinc, see the AUTHORS file.
5 6
7 This file is part of GVPE.
8
6 This program is free software; you can redistribute it and/or modify 9 GVPE is free software; you can redistribute it and/or modify it
7 it under the terms of the GNU General Public License as published by 10 under the terms of the GNU General Public License as published by the
8 the Free Software Foundation; either version 2 of the License, or 11 Free Software Foundation; either version 3 of the License, or (at your
9 (at your option) any later version. 12 option) any later version.
10 13
11 This program is distributed in the hope that it will be useful, 14 This program is distributed in the hope that it will be useful, but
12 but WITHOUT ANY WARRANTY; without even the implied warranty of 15 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
14 GNU General Public License for more details. 17 Public License for more details.
15 18
16 You should have received a copy of the GNU General Public License 19 You should have received a copy of the GNU General Public License along
17 along with this program; if not, write to the Free Software 20 with this program; if not, see <http://www.gnu.org/licenses/>.
18 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 21
22 Additional permission under GNU GPL version 3 section 7
23
24 If you modify this Program, or any covered work, by linking or
25 combining it with the OpenSSL project's OpenSSL library (or a modified
26 version of that library), containing parts covered by the terms of the
27 OpenSSL or SSLeay licenses, the licensors of this Program grant you
28 additional permission to convey the resulting work. Corresponding
29 Source for a non-source form of such a combination shall include the
30 source code for the parts of OpenSSL used as well as that of the
31 covered work.
19*/ 32*/
20 33
21#include "config.h" 34#include "config.h"
22 35
23#include <cstdio> 36#include <cstdio>
37#include <cstdlib>
24#include <cstring> 38#include <cstring>
25 39
26#include <errno.h> 40#include <errno.h>
27#include <signal.h> 41#include <signal.h>
28#include <sys/types.h> 42#include <sys/types.h>
29#include <sys/wait.h> 43#include <sys/wait.h>
30#include <unistd.h> 44#include <unistd.h>
31#include <time.h> 45#include <time.h>
32#include <sys/socket.h>
33#include <netinet/in.h>
34#include <arpa/inet.h>
35 46
36 47#include "netcompat.h"
37#include <sys/mman.h>
38 48
39#include "gettext.h" 49#include "gettext.h"
40#include "pidfile.h" 50#include "pidfile.h"
41#include "dropin.h" 51#include "dropin.h"
42 52
43#include "global.h" 53#include "global.h"
44#include "conf.h" 54#include "conf.h"
55#include "util.h"
45#include "slog.h" 56#include "slog.h"
46#include "protocol.h"
47 57
48int 58int
49write_pidfile (void) 59write_pidfile (void)
50{ 60{
51 int pid; 61 int pid;
52 62
53 pid = check_pid (pidfilename); 63 pid = check_pid (conf.pidfilename);
54 64
55 if (pid) 65 if (pid)
56 { 66 {
57 fprintf (stderr, _("A vped is already running with pid %d.\n"), pid); 67 fprintf (stderr, _("A gvpe daemon is already running with pid %d.\n"), pid);
58 return 1; 68 return 1;
59 } 69 }
60 70
61 /* if it's locked, write-protected, or whatever */ 71 /* if it's locked, write-protected, or whatever */
62 if (!write_pid (pidfilename)) 72 if (!write_pid (conf.pidfilename))
63 return 1; 73 return 1;
64 74
65 return 0; 75 return 0;
66} 76}
67 77
68int 78int
69kill_other (int signal) 79kill_other (int signal)
70{ 80{
71 int pid; 81 int pid;
72 82
73 pid = read_pid (pidfilename); 83 pid = read_pid (conf.pidfilename);
74 84
75 if (!pid) 85 if (!pid)
76 { 86 {
77 fprintf (stderr, _("No other vped is running.\n")); 87 fprintf (stderr, _("No other gvpe daemon is running.\n"));
78 return 1; 88 return 1;
79 } 89 }
80 90
81 errno = 0; /* No error, sometimes errno is only changed on error */ 91 errno = 0; /* No error, sometimes errno is only changed on error */
82 92
83 /* ESRCH is returned when no process with that pid is found */ 93 /* ESRCH is returned when no process with that pid is found */
84 if (kill (pid, signal) && errno == ESRCH) 94 if (kill (pid, signal) && errno == ESRCH)
85 { 95 {
86 fprintf (stderr, _("The vped is no longer running. ")); 96 fprintf (stderr, _("The gvpe daemon is no longer running. "));
87 97
88 fprintf (stderr, _("Removing stale lock file.\n")); 98 fprintf (stderr, _("Removing stale lock file.\n"));
89 remove_pid (pidfilename); 99 remove_pid (conf.pidfilename);
90 } 100 }
91 101
92 return 0; 102 return 0;
93} 103}
94 104
… …
114 return -1; 124 return -1;
115 } 125 }
116 126
117 /* Now UPDATE the pid in the pidfile, because we changed it... */ 127 /* Now UPDATE the pid in the pidfile, because we changed it... */
118 128
119 if (!write_pid (pidfilename)) 129 if (!write_pid (conf.pidfilename))
120 return -1; 130 return -1;
121 131
122 log_to (LOGTO_SYSLOG); 132 log_to (LOGTO_SYSLOG);
123 } 133 }
124 else 134 else
125 log_to (LOGTO_SYSLOG | LOGTO_STDERR); 135 log_to (LOGTO_SYSLOG | LOGTO_STDERR);
126 136
127 slog (L_INFO, _("vped %s (%s %s) starting"), VERSION, __DATE__, __TIME__); 137 slog (L_INFO, _("gvpe daemon %s (%s %s) starting"), VERSION, __DATE__, __TIME__);
128 138
129 return 0; 139 return 0;
130} 140}
131 141
132void 142pid_t
133make_names (void)
134{
135 if (!pidfilename)
136 pidfilename = LOCALSTATEDIR "/run/vped.pid";
137
138 if (!confbase)
139 asprintf (&confbase, "%s/vpe", CONFDIR);
140}
141
142void run_script (const run_script_cb &cb, bool wait) 143run_script (const run_script_cb &cb, bool wait)
143{ 144{
144 int pid; 145 sigset_t oldset;
145 146
146 if ((pid = fork ()) == 0) 147 if (wait)
148 {
149 sigset_t sigchld;
150 sigemptyset (&sigchld);
151 sigaddset (&sigchld, SIGCHLD);
152 sigprocmask (SIG_BLOCK, &sigchld, &oldset);
147 { 153 }
148 char *filename; 154
149 asprintf (&filename, "%s/%s", confbase, cb(0)); 155 pid_t pid = fork ();
150 execl (filename, filename, (char *) 0); 156
151 exit (255); 157 if (pid == 0)
158 {
159 sigprocmask (SIG_SETMASK, &oldset, 0);
160
161 execl ("/bin/sh", "/bin/sh", "-c", cb (), (char *) 0);
162 exit (EXIT_FAILURE);
152 } 163 }
153 else if (pid > 0) 164 else if (pid > 0)
154 { 165 {
155 if (wait) 166 if (wait)
156 { 167 {
168 int status;
157 waitpid (pid, 0, 0); 169 int res = waitpid (pid, &status, 0);
158 /* TODO: check status */ 170
171 sigprocmask (SIG_SETMASK, &oldset, 0);
172
173 if (res < 0)
174 {
175 slog (L_WARN, _("waiting for an external command failed: %s."),
176 strerror (errno));
177 return 0;
178 }
179 else if (!WIFEXITED (status) || WEXITSTATUS (status) != EXIT_SUCCESS)
180 {
181 slog (L_WARN, _("external command returned with exit status %d (%04x)."),
182 WEXITSTATUS (status), status);
183 return 0;
184 }
159 } 185 }
160 } 186 }
187 else
188 {
189 slog (L_ERR, _("unable to fork, exiting: %s"), strerror (errno));
190 exit (EXIT_FAILURE);
191 }
192
193 return pid;
161} 194}
195
196#if ENABLE_HTTP_PROXY
197// works like strdup
198u8 *
199base64_encode (const u8 *data, unsigned int len)
200{
201 const static char base64[65] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
202
203 unsigned int t, i;
204 const u8 *end = data + len;
205 u8 *res = new u8 [4 * ((len + 2) / 3) + 1];
206 u8 *out = res;
207
208 while (data <= end - 3)
209 {
210 t = (((data[0] << 8) | data[1]) << 8) | data[2];
211 data += 3;
212
213 *out++ = base64[(t >> 18) & 0x3f];
214 *out++ = base64[(t >> 12) & 0x3f];
215 *out++ = base64[(t >> 6) & 0x3f];
216 *out++ = base64[(t ) & 0x3f];
217 }
218
219 for (t = 0, i = 0; data < end; i++)
220 t = (t << 8) | *data++;
221
222 switch (i)
223 {
224 case 2:
225 *out++ = base64[(t >> 10) & 0x3f];
226 *out++ = base64[(t >> 4) & 0x3f];
227 *out++ = base64[(t << 2) & 0x3f];
228 *out++ = '=';
229 break;
230 case 1:
231 *out++ = base64[(t >> 2) & 0x3f];
232 *out++ = base64[(t << 4) & 0x3f];
233 *out++ = '=';
234 *out++ = '=';
235 break;
236 }
237
238 *out++ = 0;
239
240 return res;
241}
242#endif
243
244void
245id2mac (unsigned int id, void *m)
246{
247 mac &p = *(mac *)m;
248
249 if (id)
250 {
251 p[0] = 0xfe;
252 p[1] = 0xfd;
253 p[2] = 0x80;
254 p[3] = 0x00;
255 p[4] = id >> 8;
256 p[5] = id;
257 }
258 else
259 {
260 p[0] = 0xff;
261 p[1] = 0xff;
262 p[2] = 0xff;
263 p[3] = 0xff;
264 p[4] = 0xff;
265 p[5] = 0xff;
266 }
267}
268

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines