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

Diff Legend

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