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.1 by pcg, Sat Mar 1 15:53:03 2003 UTC vs.
Revision 1.10 by pcg, Thu Oct 16 02:28:36 2003 UTC

1/* 1/*
2 util.C -- process management and other utility functions 2 util.C -- process management and other utility functions
3 3
4 Most of these are taken from tinc, see the AUTHORS file. 4 Some of these are taken from tinc, see the AUTHORS file.
5 5
6 This program is free software; you can redistribute it and/or modify 6 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 7 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 8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version. 9 (at your option) any later version.
… …
24#include <cstring> 24#include <cstring>
25 25
26#include <errno.h> 26#include <errno.h>
27#include <signal.h> 27#include <signal.h>
28#include <sys/types.h> 28#include <sys/types.h>
29#include <sys/wait.h>
29#include <unistd.h> 30#include <unistd.h>
30#include <time.h> 31#include <time.h>
31#include <sys/socket.h>
32#include <netinet/in.h>
33#include <arpa/inet.h>
34 32
35#include <sys/mman.h> 33#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 34
42#include "gettext.h" 35#include "gettext.h"
43#include "pidfile.h" 36#include "pidfile.h"
44#include "dropin.h" 37#include "dropin.h"
45 38
46#include "global.h" 39#include "global.h"
47#include "conf.h" 40#include "conf.h"
41#include "util.h"
48#include "slog.h" 42#include "slog.h"
49#include "protocol.h"
50
51time_t now;
52 43
53int 44int
54write_pidfile (void) 45write_pidfile (void)
55{ 46{
56 int pid; 47 int pid;
… …
142 133
143 if (!confbase) 134 if (!confbase)
144 asprintf (&confbase, "%s/vpe", CONFDIR); 135 asprintf (&confbase, "%s/vpe", CONFDIR);
145} 136}
146 137
147void pkt_queue::put (tap_packet *p) 138void run_script (const run_script_cb &cb, bool wait)
148{ 139{
149 if (queue[i]) 140 int pid;
150 { 141
151 delete queue[i]; 142 if ((pid = fork ()) == 0)
152 j = (j + 1) % QUEUEDEPTH;
153 } 143 {
154 144 char *filename;
155 queue[i] = p; 145 asprintf (&filename, "%s/%s", confbase, cb());
156 146 execl (filename, filename, (char *) 0);
157 i = (i + 1) % QUEUEDEPTH; 147 exit (255);
158}
159
160tap_packet *pkt_queue::get ()
161{
162 tap_packet *p = queue[j];
163
164 if (p)
165 { 148 }
166 queue[j] = 0; 149 else if (pid > 0)
167 j = (j + 1) % QUEUEDEPTH;
168 } 150 {
169 151 if (wait)
170 return p; 152 {
153 waitpid (pid, 0, 0);
154 /* TODO: check status */
155 }
156 }
171} 157}
172 158
173pkt_queue::pkt_queue () 159#if ENABLE_HTTP_PROXY
160// works like strdup
161u8 *
162base64_encode (const u8 *data, unsigned int len)
174{ 163{
175 memset (queue, 0, sizeof (queue)); 164 const static char base64[65] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
176 i = 0;
177 j = 0;
178}
179 165
180pkt_queue::~pkt_queue () 166 unsigned int t, i;
181{ 167 const u8 *end = data + len;
182 for (i = QUEUEDEPTH; --i > 0; ) 168 u8 *res = new u8 [4 * ((len + 2) / 3) + 1];
183 delete queue[i]; 169 u8 *out = res;
184}
185 170
186sockinfo::operator const char *() 171 while (data <= end - 3)
187{ 172 {
188 static char hostport[15 + 1 + 5 + 1]; 173 t = (((data[0] << 8) | data[1]) << 8) | data[2];
189 in_addr ia = { host }; 174 data += 3;
190
191 sprintf (hostport, "%.15s:%d", inet_ntoa (ia), ntohs (port) & 0xffff);
192
193 return hostport;
194}
195
196bool u32_rate_limiter::can (u32 host)
197{
198 iterator i;
199
200 for (i = begin (); i != end (); )
201 if (i->second <= now)
202 { 175
203 erase (i); 176 *out++ = base64[(t >> 18) & 0x3f];
204 i = begin (); 177 *out++ = base64[(t >> 12) & 0x3f];
178 *out++ = base64[(t >> 6) & 0x3f];
179 *out++ = base64[(t ) & 0x3f];
205 } 180 }
206 else
207 ++i;
208 181
209 i = find (host); 182 for (t = 0, i = 0; data < end; i++)
183 t = (t << 8) | *data++;
210 184
211 if (i != end ()) 185 switch (i)
212 return false; 186 {
187 case 2:
188 *out++ = base64[(t >> 10) & 0x3f];
189 *out++ = base64[(t >> 4) & 0x3f];
190 *out++ = base64[(t << 2) & 0x3f];
191 *out++ = '=';
192 break;
193 case 1:
194 *out++ = base64[(t >> 2) & 0x3f];
195 *out++ = base64[(t << 4) & 0x3f];
196 *out++ = '=';
197 *out++ = '=';
198 break;
199 }
213 200
214 insert (value_type (host, now + every)); 201 *out++ = 0;
215 202
216 return true; 203 return res;
217} 204}
205#endif
218 206
207void
208id2mac (unsigned int id, void *m)
209{
210 mac &p = *(mac *)m;
211
212 if (id)
213 {
214 p[0] = 0xfe;
215 p[1] = 0xfd;
216 p[2] = 0x80;
217 p[3] = 0x00;
218 p[4] = id >> 8;
219 p[5] = id;
220 }
221 else
222 {
223 p[0] = 0xff;
224 p[1] = 0xff;
225 p[2] = 0xff;
226 p[3] = 0xff;
227 p[4] = 0xff;
228 p[5] = 0xff;
229 }
230}
231

Diff Legend

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