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.3 by pcg, Fri Mar 28 04:05:10 2003 UTC vs.
Revision 1.13 by pcg, Fri Apr 2 14:42:45 2004 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 Marc Lehmann <pcg@goof.com>
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
6 This program is free software; you can redistribute it and/or modify 7 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by 8 it under the terms of the GNU General Public License as published by
19*/ 20*/
20 21
21#include "config.h" 22#include "config.h"
22 23
23#include <cstdio> 24#include <cstdio>
25#include <cstdlib>
24#include <cstring> 26#include <cstring>
25 27
26#include <errno.h> 28#include <errno.h>
27#include <signal.h> 29#include <signal.h>
28#include <sys/types.h> 30#include <sys/types.h>
31#include <sys/wait.h>
29#include <unistd.h> 32#include <unistd.h>
30#include <time.h> 33#include <time.h>
31#include <sys/socket.h>
32#include <netinet/in.h>
33#include <arpa/inet.h>
34 34
35#include <sys/mman.h> 35#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 36
42#include "gettext.h" 37#include "gettext.h"
43#include "pidfile.h" 38#include "pidfile.h"
44#include "dropin.h" 39#include "dropin.h"
45 40
46#include "global.h" 41#include "global.h"
47#include "conf.h" 42#include "conf.h"
43#include "util.h"
48#include "slog.h" 44#include "slog.h"
49#include "protocol.h"
50 45
51int 46int
52write_pidfile (void) 47write_pidfile (void)
53{ 48{
54 int pid; 49 int pid;
55 50
56 pid = check_pid (pidfilename); 51 pid = check_pid (conf.pidfilename);
57 52
58 if (pid) 53 if (pid)
59 { 54 {
60 fprintf (stderr, _("A vped is already running with pid %d.\n"), pid); 55 fprintf (stderr, _("A vped is already running with pid %d.\n"), pid);
61 return 1; 56 return 1;
62 } 57 }
63 58
64 /* if it's locked, write-protected, or whatever */ 59 /* if it's locked, write-protected, or whatever */
65 if (!write_pid (pidfilename)) 60 if (!write_pid (conf.pidfilename))
66 return 1; 61 return 1;
67 62
68 return 0; 63 return 0;
69} 64}
70 65
71int 66int
72kill_other (int signal) 67kill_other (int signal)
73{ 68{
74 int pid; 69 int pid;
75 70
76 pid = read_pid (pidfilename); 71 pid = read_pid (conf.pidfilename);
77 72
78 if (!pid) 73 if (!pid)
79 { 74 {
80 fprintf (stderr, _("No other vped is running.\n")); 75 fprintf (stderr, _("No other vped is running.\n"));
81 return 1; 76 return 1;
87 if (kill (pid, signal) && errno == ESRCH) 82 if (kill (pid, signal) && errno == ESRCH)
88 { 83 {
89 fprintf (stderr, _("The vped is no longer running. ")); 84 fprintf (stderr, _("The vped is no longer running. "));
90 85
91 fprintf (stderr, _("Removing stale lock file.\n")); 86 fprintf (stderr, _("Removing stale lock file.\n"));
92 remove_pid (pidfilename); 87 remove_pid (conf.pidfilename);
93 } 88 }
94 89
95 return 0; 90 return 0;
96} 91}
97 92
117 return -1; 112 return -1;
118 } 113 }
119 114
120 /* Now UPDATE the pid in the pidfile, because we changed it... */ 115 /* Now UPDATE the pid in the pidfile, because we changed it... */
121 116
122 if (!write_pid (pidfilename)) 117 if (!write_pid (conf.pidfilename))
123 return -1; 118 return -1;
124 119
125 log_to (LOGTO_SYSLOG); 120 log_to (LOGTO_SYSLOG);
126 } 121 }
127 else 122 else
133} 128}
134 129
135void 130void
136make_names (void) 131make_names (void)
137{ 132{
138 if (!pidfilename) 133 if (!conf.pidfilename)
139 pidfilename = LOCALSTATEDIR "/run/vped.pid"; 134 conf.pidfilename = strdup (LOCALSTATEDIR "/run/vped.pid");
140 135
141 if (!confbase) 136 if (!confbase)
142 asprintf (&confbase, "%s/vpe", CONFDIR); 137 asprintf (&confbase, "%s/vpe", CONFDIR);
143} 138}
144 139
140void run_script (const run_script_cb &cb, bool wait)
141{
142 int pid;
143
144 if ((pid = fork ()) == 0)
145 {
146 char *filename;
147 asprintf (&filename, "%s/%s", confbase, cb());
148 execl (filename, filename, (char *) 0);
149 exit (126);
150 }
151 else if (pid > 0)
152 {
153 if (wait)
154 {
155 waitpid (pid, 0, 0);
156 /* TODO: check status */
157 }
158 }
159}
160
161#if ENABLE_HTTP_PROXY
162// works like strdup
163u8 *
164base64_encode (const u8 *data, unsigned int len)
165{
166 const static char base64[65] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
167
168 unsigned int t, i;
169 const u8 *end = data + len;
170 u8 *res = new u8 [4 * ((len + 2) / 3) + 1];
171 u8 *out = res;
172
173 while (data <= end - 3)
174 {
175 t = (((data[0] << 8) | data[1]) << 8) | data[2];
176 data += 3;
177
178 *out++ = base64[(t >> 18) & 0x3f];
179 *out++ = base64[(t >> 12) & 0x3f];
180 *out++ = base64[(t >> 6) & 0x3f];
181 *out++ = base64[(t ) & 0x3f];
182 }
183
184 for (t = 0, i = 0; data < end; i++)
185 t = (t << 8) | *data++;
186
187 switch (i)
188 {
189 case 2:
190 *out++ = base64[(t >> 10) & 0x3f];
191 *out++ = base64[(t >> 4) & 0x3f];
192 *out++ = base64[(t << 2) & 0x3f];
193 *out++ = '=';
194 break;
195 case 1:
196 *out++ = base64[(t >> 2) & 0x3f];
197 *out++ = base64[(t << 4) & 0x3f];
198 *out++ = '=';
199 *out++ = '=';
200 break;
201 }
202
203 *out++ = 0;
204
205 return res;
206}
207#endif
208
209void
210id2mac (unsigned int id, void *m)
211{
212 mac &p = *(mac *)m;
213
214 if (id)
215 {
216 p[0] = 0xfe;
217 p[1] = 0xfd;
218 p[2] = 0x80;
219 p[3] = 0x00;
220 p[4] = id >> 8;
221 p[5] = id;
222 }
223 else
224 {
225 p[0] = 0xff;
226 p[1] = 0xff;
227 p[2] = 0xff;
228 p[3] = 0xff;
229 p[4] = 0xff;
230 p[5] = 0xff;
231 }
232}
233

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines