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.2 by pcg, Fri Mar 21 23:17:01 2003 UTC vs.
Revision 1.19 by pcg, Wed Mar 23 21:55:39 2005 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-2005 Marc Lehmann <gvpe@schmorp.de>
3 4
4 Most 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
7 it under the terms of the GNU General Public License as published by 10 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or 11 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version. 12 (at your 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,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of 15 but 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
14 GNU General Public License for more details. 17 GNU General 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
17 along with this program; if not, write to the Free Software 20 along with gvpe; if not, write to the Free Software
18 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 21 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19*/ 22*/
20 23
21#include "config.h" 24#include "config.h"
22 25
23#include <cstdio> 26#include <cstdio>
27#include <cstdlib>
24#include <cstring> 28#include <cstring>
25 29
26#include <errno.h> 30#include <errno.h>
27#include <signal.h> 31#include <signal.h>
28#include <sys/types.h> 32#include <sys/types.h>
33#include <sys/wait.h>
29#include <unistd.h> 34#include <unistd.h>
30#include <time.h> 35#include <time.h>
31#include <sys/socket.h>
32#include <netinet/in.h>
33#include <arpa/inet.h>
34 36
35#include <sys/mman.h> 37#include "netcompat.h"
36
37#include <openssl/rand.h>
38#include <openssl/rsa.h>
39#include <openssl/pem.h>
40#include <openssl/evp.h>
41 38
42#include "gettext.h" 39#include "gettext.h"
43#include "pidfile.h" 40#include "pidfile.h"
44#include "dropin.h" 41#include "dropin.h"
45 42
46#include "global.h" 43#include "global.h"
47#include "conf.h" 44#include "conf.h"
45#include "util.h"
48#include "slog.h" 46#include "slog.h"
49#include "protocol.h"
50 47
51int 48int
52write_pidfile (void) 49write_pidfile (void)
53{ 50{
54 int pid; 51 int pid;
55 52
56 pid = check_pid (pidfilename); 53 pid = check_pid (conf.pidfilename);
57 54
58 if (pid) 55 if (pid)
59 { 56 {
60 fprintf (stderr, _("A vped is already running with pid %d.\n"), pid); 57 fprintf (stderr, _("A gvpe daemon is already running with pid %d.\n"), pid);
61 return 1; 58 return 1;
62 } 59 }
63 60
64 /* if it's locked, write-protected, or whatever */ 61 /* if it's locked, write-protected, or whatever */
65 if (!write_pid (pidfilename)) 62 if (!write_pid (conf.pidfilename))
66 return 1; 63 return 1;
67 64
68 return 0; 65 return 0;
69} 66}
70 67
71int 68int
72kill_other (int signal) 69kill_other (int signal)
73{ 70{
74 int pid; 71 int pid;
75 72
76 pid = read_pid (pidfilename); 73 pid = read_pid (conf.pidfilename);
77 74
78 if (!pid) 75 if (!pid)
79 { 76 {
80 fprintf (stderr, _("No other vped is running.\n")); 77 fprintf (stderr, _("No other gvpe daemon is running.\n"));
81 return 1; 78 return 1;
82 } 79 }
83 80
84 errno = 0; /* No error, sometimes errno is only changed on error */ 81 errno = 0; /* No error, sometimes errno is only changed on error */
85 82
86 /* ESRCH is returned when no process with that pid is found */ 83 /* ESRCH is returned when no process with that pid is found */
87 if (kill (pid, signal) && errno == ESRCH) 84 if (kill (pid, signal) && errno == ESRCH)
88 { 85 {
89 fprintf (stderr, _("The vped is no longer running. ")); 86 fprintf (stderr, _("The gvpe daemon is no longer running. "));
90 87
91 fprintf (stderr, _("Removing stale lock file.\n")); 88 fprintf (stderr, _("Removing stale lock file.\n"));
92 remove_pid (pidfilename); 89 remove_pid (conf.pidfilename);
93 } 90 }
94 91
95 return 0; 92 return 0;
96} 93}
97 94
117 return -1; 114 return -1;
118 } 115 }
119 116
120 /* Now UPDATE the pid in the pidfile, because we changed it... */ 117 /* Now UPDATE the pid in the pidfile, because we changed it... */
121 118
122 if (!write_pid (pidfilename)) 119 if (!write_pid (conf.pidfilename))
123 return -1; 120 return -1;
124 121
125 log_to (LOGTO_SYSLOG); 122 log_to (LOGTO_SYSLOG);
126 } 123 }
127 else 124 else
128 log_to (LOGTO_SYSLOG | LOGTO_STDERR); 125 log_to (LOGTO_SYSLOG | LOGTO_STDERR);
129 126
130 slog (L_INFO, _("vped %s (%s %s) starting"), VERSION, __DATE__, __TIME__); 127 slog (L_INFO, _("gvpe daemon %s (%s %s) starting"), VERSION, __DATE__, __TIME__);
131 128
132 return 0; 129 return 0;
133} 130}
134 131
132bool run_script (const run_script_cb &cb, bool wait)
133{
134 int pid;
135
136 if ((pid = fork ()) == 0)
137 {
138 execl ("/bin/sh", "/bin/sh", "-c", cb (), (char *) 0);
139 exit (EXIT_FAILURE);
140 }
141 else if (pid > 0)
142 {
143 if (wait)
144 {
145 int status;
146
147 if (waitpid (pid, &status, 0) < 0)
148 {
149 slog (L_WARN, _("waiting for an external command failed: %s."),
150 strerror (errno));
151 return false;
152 }
153 else if (!WIFEXITED (status) || WEXITSTATUS (status) != EXIT_SUCCESS)
154 {
155 slog (L_WARN, _("external command returned with exit status %d (%04x)."),
156 WEXITSTATUS (status), status);
157 return false;
158 }
159 }
160 }
161
162 return true;
163}
164
165#if ENABLE_HTTP_PROXY
166// works like strdup
167u8 *
168base64_encode (const u8 *data, unsigned int len)
169{
170 const static char base64[65] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
171
172 unsigned int t, i;
173 const u8 *end = data + len;
174 u8 *res = new u8 [4 * ((len + 2) / 3) + 1];
175 u8 *out = res;
176
177 while (data <= end - 3)
178 {
179 t = (((data[0] << 8) | data[1]) << 8) | data[2];
180 data += 3;
181
182 *out++ = base64[(t >> 18) & 0x3f];
183 *out++ = base64[(t >> 12) & 0x3f];
184 *out++ = base64[(t >> 6) & 0x3f];
185 *out++ = base64[(t ) & 0x3f];
186 }
187
188 for (t = 0, i = 0; data < end; i++)
189 t = (t << 8) | *data++;
190
191 switch (i)
192 {
193 case 2:
194 *out++ = base64[(t >> 10) & 0x3f];
195 *out++ = base64[(t >> 4) & 0x3f];
196 *out++ = base64[(t << 2) & 0x3f];
197 *out++ = '=';
198 break;
199 case 1:
200 *out++ = base64[(t >> 2) & 0x3f];
201 *out++ = base64[(t << 4) & 0x3f];
202 *out++ = '=';
203 *out++ = '=';
204 break;
205 }
206
207 *out++ = 0;
208
209 return res;
210}
211#endif
212
135void 213void
136make_names (void) 214id2mac (unsigned int id, void *m)
137{ 215{
138 if (!pidfilename) 216 mac &p = *(mac *)m;
139 pidfilename = LOCALSTATEDIR "/run/vped.pid";
140 217
141 if (!confbase) 218 if (id)
142 asprintf (&confbase, "%s/vpe", CONFDIR); 219 {
220 p[0] = 0xfe;
221 p[1] = 0xfd;
222 p[2] = 0x80;
223 p[3] = 0x00;
224 p[4] = id >> 8;
225 p[5] = id;
226 }
227 else
228 {
229 p[0] = 0xff;
230 p[1] = 0xff;
231 p[2] = 0xff;
232 p[3] = 0xff;
233 p[4] = 0xff;
234 p[5] = 0xff;
235 }
143} 236}
144 237
145sockinfo::operator const char *()
146{
147 static char hostport[15 + 1 + 5 + 1];
148 in_addr ia = { host };
149
150 sprintf (hostport, "%.15s:%d", inet_ntoa (ia), ntohs (port) & 0xffff);
151
152 return hostport;
153}
154

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines