adamc@436
|
1 #define _XOPEN_SOURCE
|
adamc@436
|
2
|
adamc@117
|
3 #include <stdlib.h>
|
adamc@102
|
4 #include <stdio.h>
|
adamc@117
|
5 #include <string.h>
|
adamc@457
|
6 #include <strings.h>
|
adamc@106
|
7 #include <ctype.h>
|
adamc@167
|
8 #include <setjmp.h>
|
adamc@167
|
9 #include <stdarg.h>
|
adamc@102
|
10
|
adamc@667
|
11 #include <pthread.h>
|
adamc@667
|
12
|
adamc@102
|
13 #include "types.h"
|
adamc@102
|
14
|
adamc@311
|
15 uw_unit uw_unit_v = {};
|
adamc@102
|
16
|
adamc@667
|
17
|
adamc@667
|
18 // Socket extras
|
adamc@667
|
19
|
adamc@667
|
20 int uw_really_send(int sock, const void *buf, size_t len) {
|
adamc@667
|
21 while (len > 0) {
|
adamc@667
|
22 size_t n = send(sock, buf, len, 0);
|
adamc@667
|
23
|
adamc@667
|
24 if (n < 0)
|
adamc@667
|
25 return n;
|
adamc@667
|
26
|
adamc@667
|
27 buf += n;
|
adamc@667
|
28 len -= n;
|
adamc@667
|
29 }
|
adamc@667
|
30
|
adamc@667
|
31 return 0;
|
adamc@667
|
32 }
|
adamc@667
|
33
|
adamc@667
|
34
|
adamc@667
|
35 // Buffers
|
adamc@667
|
36
|
adamc@667
|
37 typedef struct {
|
adamc@667
|
38 char *start, *front, *back;
|
adamc@667
|
39 } buf;
|
adamc@667
|
40
|
adamc@667
|
41 static void buf_init(buf *b, size_t s) {
|
adamc@667
|
42 b->front = b->start = malloc(s);
|
adamc@667
|
43 b->back = b->front + s;
|
adamc@667
|
44 }
|
adamc@667
|
45
|
adamc@667
|
46 static void buf_free(buf *b) {
|
adamc@667
|
47 free(b->start);
|
adamc@667
|
48 }
|
adamc@667
|
49
|
adamc@667
|
50 static void buf_reset(buf *b) {
|
adamc@667
|
51 b->front = b->start;
|
adamc@667
|
52 }
|
adamc@667
|
53
|
adamc@667
|
54 static void buf_check(buf *b, size_t extra) {
|
adamc@667
|
55 if (b->back - b->front < extra) {
|
adamc@667
|
56 size_t desired = b->front - b->start + extra, next;
|
adamc@667
|
57 char *new_heap;
|
adamc@667
|
58
|
adamc@667
|
59 next = b->back - b->start;
|
adamc@667
|
60 if (next == 0)
|
adamc@667
|
61 next = 1;
|
adamc@667
|
62 for (; next < desired; next *= 2);
|
adamc@667
|
63
|
adamc@667
|
64 new_heap = realloc(b->start, next);
|
adamc@667
|
65 b->front = new_heap + (b->front - b->start);
|
adamc@667
|
66 b->back = new_heap + next;
|
adamc@667
|
67 b->start = new_heap;
|
adamc@667
|
68 }
|
adamc@667
|
69 }
|
adamc@667
|
70
|
adamc@667
|
71 static size_t buf_used(buf *b) {
|
adamc@667
|
72 return b->front - b->start;
|
adamc@667
|
73 }
|
adamc@667
|
74
|
adamc@667
|
75 static size_t buf_avail(buf *b) {
|
adamc@667
|
76 return b->back - b->start;
|
adamc@667
|
77 }
|
adamc@667
|
78
|
adamc@667
|
79
|
adamc@667
|
80 // Cross-request state
|
adamc@667
|
81
|
adamc@667
|
82 typedef enum { UNUSED, USED } usage;
|
adamc@667
|
83
|
adamc@667
|
84 typedef struct client {
|
adamc@667
|
85 size_t id;
|
adamc@667
|
86 usage mode;
|
adamc@667
|
87 union {
|
adamc@667
|
88 struct client *next;
|
adamc@667
|
89 struct {
|
adamc@667
|
90 pthread_mutex_t lock;
|
adamc@667
|
91 int pass;
|
adamc@667
|
92 buf msgs;
|
adamc@667
|
93 int sock;
|
adamc@667
|
94 time_t last_contact;
|
adamc@667
|
95 } used;
|
adamc@667
|
96 } data;
|
adamc@667
|
97 } client;
|
adamc@667
|
98
|
adamc@667
|
99 static client **clients, *clients_free;
|
adamc@667
|
100 static size_t n_clients;
|
adamc@667
|
101
|
adamc@667
|
102 static pthread_mutex_t clients_mutex = PTHREAD_MUTEX_INITIALIZER;
|
adamc@667
|
103
|
adamc@667
|
104 void uw_global_init() {
|
adamc@667
|
105 srand(time(NULL) ^ getpid());
|
adamc@667
|
106
|
adamc@667
|
107 clients = malloc(0);
|
adamc@667
|
108 }
|
adamc@667
|
109
|
adamc@667
|
110 void uw_global_free() {
|
adamc@667
|
111 size_t i;
|
adamc@667
|
112
|
adamc@667
|
113 for (i = 0; i < n_clients; ++i)
|
adamc@667
|
114 free(clients[i]);
|
adamc@667
|
115
|
adamc@667
|
116 free(clients);
|
adamc@667
|
117 }
|
adamc@667
|
118
|
adamc@667
|
119 static client *uw_new_client() {
|
adamc@667
|
120 client *c;
|
adamc@667
|
121
|
adamc@667
|
122 pthread_mutex_lock(&clients_mutex);
|
adamc@667
|
123
|
adamc@667
|
124 if (clients_free) {
|
adamc@667
|
125 c = clients_free;
|
adamc@667
|
126 clients_free = clients_free->data.next;
|
adamc@667
|
127 }
|
adamc@667
|
128 else {
|
adamc@667
|
129 ++n_clients;
|
adamc@667
|
130 clients = realloc(clients, sizeof(client) * n_clients);
|
adamc@667
|
131 c = malloc(sizeof(client));
|
adamc@667
|
132 c->id = n_clients-1;
|
adamc@667
|
133 clients[n_clients-1] = c;
|
adamc@667
|
134 }
|
adamc@667
|
135
|
adamc@667
|
136 c->mode = USED;
|
adamc@667
|
137 pthread_mutex_init(&c->data.used.lock, NULL);
|
adamc@667
|
138 c->data.used.pass = rand();
|
adamc@667
|
139 c->data.used.sock = -1;
|
adamc@667
|
140 c->data.used.last_contact = time(NULL);
|
adamc@667
|
141 buf_init(&c->data.used.msgs, 0);
|
adamc@667
|
142
|
adamc@667
|
143 pthread_mutex_unlock(&clients_mutex);
|
adamc@667
|
144
|
adamc@667
|
145 return c;
|
adamc@667
|
146 }
|
adamc@667
|
147
|
adamc@667
|
148 static const char begin_msgs[] = "HTTP/1.1 200 OK\r\nContent-type: text/plain\r\n\r\n";
|
adamc@667
|
149
|
adamc@667
|
150 void uw_client_connect(size_t id, int pass, int sock) {
|
adamc@667
|
151 client *c;
|
adamc@667
|
152
|
adamc@667
|
153 pthread_mutex_lock(&clients_mutex);
|
adamc@667
|
154
|
adamc@667
|
155 if (id >= n_clients) {
|
adamc@667
|
156 pthread_mutex_unlock(&clients_mutex);
|
adamc@667
|
157 close(sock);
|
adamc@667
|
158 fprintf(stderr, "Out-of-bounds client request (%d)\n", (int)id);
|
adamc@667
|
159 return;
|
adamc@667
|
160 }
|
adamc@667
|
161
|
adamc@667
|
162 c = clients[id];
|
adamc@667
|
163
|
adamc@667
|
164 if (c->mode != USED) {
|
adamc@667
|
165 pthread_mutex_unlock(&clients_mutex);
|
adamc@667
|
166 close(sock);
|
adamc@667
|
167 fprintf(stderr, "Client request for unused ID (%d)\n", (int)id);
|
adamc@667
|
168 return;
|
adamc@667
|
169 }
|
adamc@667
|
170
|
adamc@667
|
171 pthread_mutex_lock(&c->data.used.lock);
|
adamc@667
|
172
|
adamc@667
|
173 if (pass != c->data.used.pass) {
|
adamc@667
|
174 pthread_mutex_unlock(&c->data.used.lock);
|
adamc@667
|
175 pthread_mutex_unlock(&clients_mutex);
|
adamc@667
|
176 close(sock);
|
adamc@667
|
177 fprintf(stderr, "Wrong client password (%d)\n", (int)id);
|
adamc@667
|
178 return;
|
adamc@667
|
179 }
|
adamc@667
|
180
|
adamc@667
|
181 if (c->data.used.sock != -1) {
|
adamc@667
|
182 pthread_mutex_unlock(&c->data.used.lock);
|
adamc@667
|
183 pthread_mutex_unlock(&clients_mutex);
|
adamc@667
|
184 close(sock);
|
adamc@667
|
185 fprintf(stderr, "Duplicate client connection (%d)\n", (int)id);
|
adamc@667
|
186 return;
|
adamc@667
|
187 }
|
adamc@667
|
188
|
adamc@667
|
189 c->data.used.last_contact = time(NULL);
|
adamc@667
|
190
|
adamc@667
|
191 if (buf_used(&c->data.used.msgs) > 0) {
|
adamc@667
|
192 uw_really_send(sock, begin_msgs, sizeof(begin_msgs) - 1);
|
adamc@667
|
193 uw_really_send(sock, c->data.used.msgs.start, buf_used(&c->data.used.msgs));
|
adamc@667
|
194 close(sock);
|
adamc@667
|
195 }
|
adamc@667
|
196 else {
|
adamc@667
|
197 uw_really_send(sock, begin_msgs, sizeof(begin_msgs) - 1);
|
adamc@667
|
198 uw_really_send(sock, "Hi!", 3);
|
adamc@667
|
199 close(sock);
|
adamc@667
|
200 //c->data.used.sock = sock;
|
adamc@667
|
201 }
|
adamc@667
|
202
|
adamc@667
|
203 pthread_mutex_unlock(&c->data.used.lock);
|
adamc@667
|
204 pthread_mutex_unlock(&clients_mutex);
|
adamc@667
|
205 }
|
adamc@667
|
206
|
adamc@667
|
207 static void uw_free_client(client *c) {
|
adamc@667
|
208 printf("Freeing client %d\n", c->id);
|
adamc@667
|
209
|
adamc@667
|
210 if (c->mode == USED && c->data.used.sock != -1)
|
adamc@667
|
211 close(c->data.used.sock);
|
adamc@667
|
212
|
adamc@667
|
213 pthread_mutex_destroy(&c->data.used.lock);
|
adamc@667
|
214 buf_free(&c->data.used.msgs);
|
adamc@667
|
215 c->mode = UNUSED;
|
adamc@667
|
216 c->data.next = clients_free;
|
adamc@667
|
217 clients_free = c;
|
adamc@667
|
218 }
|
adamc@667
|
219
|
adamc@667
|
220 void uw_prune_clients(time_t timeout) {
|
adamc@667
|
221 size_t i;
|
adamc@667
|
222 time_t cutoff;
|
adamc@667
|
223
|
adamc@667
|
224 cutoff = time(NULL) - timeout;
|
adamc@667
|
225
|
adamc@667
|
226 pthread_mutex_lock(&clients_mutex);
|
adamc@667
|
227
|
adamc@667
|
228 for (i = 0; i < n_clients; ++i) {
|
adamc@667
|
229 if (clients[i]->mode == USED && clients[i]->data.used.last_contact < cutoff)
|
adamc@667
|
230 uw_free_client(clients[i]);
|
adamc@667
|
231 }
|
adamc@667
|
232
|
adamc@667
|
233 pthread_mutex_unlock(&clients_mutex);
|
adamc@667
|
234 }
|
adamc@667
|
235
|
adamc@667
|
236
|
adamc@667
|
237 // Single-request state
|
adamc@667
|
238
|
adamc@167
|
239 #define ERROR_BUF_LEN 1024
|
adamc@167
|
240
|
adamc@323
|
241 typedef struct regions {
|
adamc@323
|
242 struct regions *next;
|
adamc@323
|
243 } regions;
|
adamc@323
|
244
|
adamc@425
|
245 typedef struct {
|
adamc@425
|
246 void (*func)(void*);
|
adamc@425
|
247 void *arg;
|
adamc@425
|
248 } cleanup;
|
adamc@425
|
249
|
adamc@311
|
250 struct uw_context {
|
adamc@458
|
251 char *headers, *headers_end;
|
adamc@457
|
252
|
adamc@666
|
253 buf outHeaders, page, heap, script;
|
adamc@144
|
254 char **inputs;
|
adamc@167
|
255
|
adamc@565
|
256 int source_count;
|
adamc@562
|
257
|
adamc@272
|
258 void *db;
|
adamc@272
|
259
|
adamc@167
|
260 jmp_buf jmp_buf;
|
adamc@167
|
261
|
adamc@323
|
262 regions *regions;
|
adamc@323
|
263
|
adamc@425
|
264 cleanup *cleanup, *cleanup_front, *cleanup_back;
|
adamc@425
|
265
|
adamc@667
|
266 const char *script_header, *url_prefix;
|
adamc@643
|
267
|
adamc@167
|
268 char error_message[ERROR_BUF_LEN];
|
adamc@117
|
269 };
|
adamc@117
|
270
|
adamc@311
|
271 extern int uw_inputs_len;
|
adamc@144
|
272
|
adamc@562
|
273 uw_context uw_init(size_t outHeaders_len, size_t script_len, size_t page_len, size_t heap_len) {
|
adamc@311
|
274 uw_context ctx = malloc(sizeof(struct uw_context));
|
adamc@136
|
275
|
adamc@458
|
276 ctx->headers = ctx->headers_end = NULL;
|
adamc@457
|
277
|
adamc@666
|
278 buf_init(&ctx->outHeaders, outHeaders_len);
|
adamc@666
|
279 buf_init(&ctx->page, page_len);
|
adamc@666
|
280 buf_init(&ctx->heap, heap_len);
|
adamc@666
|
281 buf_init(&ctx->script, script_len);
|
adamc@667
|
282 ctx->script.start[0] = 0;
|
adamc@136
|
283
|
adamc@311
|
284 ctx->inputs = calloc(uw_inputs_len, sizeof(char *));
|
adamc@144
|
285
|
adamc@272
|
286 ctx->db = NULL;
|
adamc@272
|
287
|
adamc@323
|
288 ctx->regions = NULL;
|
adamc@323
|
289
|
adamc@425
|
290 ctx->cleanup_front = ctx->cleanup_back = ctx->cleanup = malloc(0);
|
adamc@425
|
291
|
adamc@643
|
292 ctx->script_header = "";
|
adamc@667
|
293 ctx->url_prefix = "/";
|
adamc@643
|
294
|
adamc@167
|
295 ctx->error_message[0] = 0;
|
adamc@167
|
296
|
adamc@565
|
297 ctx->source_count = 0;
|
adamc@562
|
298
|
adamc@117
|
299 return ctx;
|
adamc@106
|
300 }
|
adamc@106
|
301
|
adamc@311
|
302 void uw_set_db(uw_context ctx, void *db) {
|
adamc@272
|
303 ctx->db = db;
|
adamc@272
|
304 }
|
adamc@272
|
305
|
adamc@311
|
306 void *uw_get_db(uw_context ctx) {
|
adamc@272
|
307 return ctx->db;
|
adamc@272
|
308 }
|
adamc@272
|
309
|
adamc@311
|
310 void uw_free(uw_context ctx) {
|
adamc@666
|
311 buf_free(&ctx->outHeaders);
|
adamc@666
|
312 buf_free(&ctx->script);
|
adamc@666
|
313 buf_free(&ctx->page);
|
adamc@666
|
314 buf_free(&ctx->heap);
|
adamc@144
|
315 free(ctx->inputs);
|
adamc@425
|
316 free(ctx->cleanup);
|
adamc@136
|
317 free(ctx);
|
adamc@136
|
318 }
|
adamc@136
|
319
|
adamc@562
|
320 void uw_reset_keep_error_message(uw_context ctx) {
|
adamc@666
|
321 buf_reset(&ctx->outHeaders);
|
adamc@666
|
322 buf_reset(&ctx->script);
|
adamc@667
|
323 ctx->script.start[0] = 0;
|
adamc@666
|
324 buf_reset(&ctx->page);
|
adamc@666
|
325 buf_reset(&ctx->heap);
|
adamc@323
|
326 ctx->regions = NULL;
|
adamc@425
|
327 ctx->cleanup_front = ctx->cleanup;
|
adamc@565
|
328 ctx->source_count = 0;
|
adamc@167
|
329 }
|
adamc@167
|
330
|
adamc@562
|
331 void uw_reset_keep_request(uw_context ctx) {
|
adamc@562
|
332 uw_reset_keep_error_message(ctx);
|
adamc@562
|
333 ctx->error_message[0] = 0;
|
adamc@167
|
334 }
|
adamc@167
|
335
|
adamc@311
|
336 void uw_reset(uw_context ctx) {
|
adamc@311
|
337 uw_reset_keep_request(ctx);
|
adamc@311
|
338 memset(ctx->inputs, 0, uw_inputs_len * sizeof(char *));
|
adamc@144
|
339 }
|
adamc@144
|
340
|
adamc@311
|
341 void uw_db_init(uw_context);
|
adamc@311
|
342 void uw_handle(uw_context, char *);
|
adamc@167
|
343
|
adamc@311
|
344 failure_kind uw_begin_init(uw_context ctx) {
|
adamc@272
|
345 int r = setjmp(ctx->jmp_buf);
|
adamc@272
|
346
|
adamc@272
|
347 if (r == 0)
|
adamc@311
|
348 uw_db_init(ctx);
|
adamc@272
|
349
|
adamc@272
|
350 return r;
|
adamc@272
|
351 }
|
adamc@272
|
352
|
adamc@458
|
353 void uw_set_headers(uw_context ctx, char *headers) {
|
adamc@458
|
354 char *s = headers, *s2;
|
adamc@458
|
355 ctx->headers = headers;
|
adamc@458
|
356
|
adamc@458
|
357 while (s2 = strchr(s, '\r')) {
|
adamc@458
|
358 s = s2;
|
adamc@458
|
359
|
adamc@458
|
360 if (s[1] == 0)
|
adamc@458
|
361 break;
|
adamc@458
|
362
|
adamc@458
|
363 *s = 0;
|
adamc@458
|
364 s += 2;
|
adamc@458
|
365 }
|
adamc@458
|
366
|
adamc@458
|
367 ctx->headers_end = s;
|
adamc@458
|
368 }
|
adamc@458
|
369
|
adamc@458
|
370 failure_kind uw_begin(uw_context ctx, char *path) {
|
adamc@190
|
371 int r = setjmp(ctx->jmp_buf);
|
adamc@190
|
372
|
adamc@190
|
373 if (r == 0)
|
adamc@311
|
374 uw_handle(ctx, path);
|
adamc@167
|
375
|
adamc@190
|
376 return r;
|
adamc@167
|
377 }
|
adamc@167
|
378
|
adamc@311
|
379 __attribute__((noreturn)) void uw_error(uw_context ctx, failure_kind fk, const char *fmt, ...) {
|
adamc@425
|
380 cleanup *cl;
|
adamc@425
|
381
|
adamc@167
|
382 va_list ap;
|
adamc@167
|
383 va_start(ap, fmt);
|
adamc@167
|
384
|
adamc@167
|
385 vsnprintf(ctx->error_message, ERROR_BUF_LEN, fmt, ap);
|
adamc@167
|
386
|
adamc@425
|
387 for (cl = ctx->cleanup; cl < ctx->cleanup_front; ++cl)
|
adamc@425
|
388 cl->func(cl->arg);
|
adamc@425
|
389
|
adamc@425
|
390 ctx->cleanup_front = ctx->cleanup;
|
adamc@425
|
391
|
adamc@190
|
392 longjmp(ctx->jmp_buf, fk);
|
adamc@167
|
393 }
|
adamc@167
|
394
|
adamc@425
|
395 void uw_push_cleanup(uw_context ctx, void (*func)(void *), void *arg) {
|
adamc@425
|
396 if (ctx->cleanup_front >= ctx->cleanup_back) {
|
adamc@425
|
397 int len = ctx->cleanup_back - ctx->cleanup, newLen;
|
adamc@425
|
398 if (len == 0)
|
adamc@425
|
399 newLen = 1;
|
adamc@425
|
400 else
|
adamc@428
|
401 newLen = len * 2;
|
adamc@470
|
402 ctx->cleanup = realloc(ctx->cleanup, newLen * sizeof(cleanup));
|
adamc@425
|
403 ctx->cleanup_front = ctx->cleanup + len;
|
adamc@425
|
404 ctx->cleanup_back = ctx->cleanup + newLen;
|
adamc@425
|
405 }
|
adamc@425
|
406
|
adamc@425
|
407 ctx->cleanup_front->func = func;
|
adamc@425
|
408 ctx->cleanup_front->arg = arg;
|
adamc@425
|
409 ++ctx->cleanup_front;
|
adamc@425
|
410 }
|
adamc@425
|
411
|
adamc@425
|
412 void uw_pop_cleanup(uw_context ctx) {
|
adamc@425
|
413 if (ctx->cleanup_front == ctx->cleanup)
|
adamc@425
|
414 uw_error(ctx, FATAL, "Attempt to pop from empty cleanup action stack");
|
adamc@425
|
415
|
adamc@425
|
416 --ctx->cleanup_front;
|
adamc@425
|
417 ctx->cleanup_front->func(ctx->cleanup_front->arg);
|
adamc@425
|
418 }
|
adamc@425
|
419
|
adamc@311
|
420 char *uw_error_message(uw_context ctx) {
|
adamc@167
|
421 return ctx->error_message;
|
adamc@167
|
422 }
|
adamc@167
|
423
|
adamc@311
|
424 int uw_input_num(char*);
|
adamc@144
|
425
|
adamc@311
|
426 void uw_set_input(uw_context ctx, char *name, char *value) {
|
adamc@311
|
427 int n = uw_input_num(name);
|
adamc@144
|
428
|
adamc@169
|
429 if (n < 0)
|
adamc@311
|
430 uw_error(ctx, FATAL, "Bad input name %s", name);
|
adamc@144
|
431
|
adamc@311
|
432 if (n >= uw_inputs_len)
|
adamc@311
|
433 uw_error(ctx, FATAL, "For input name %s, index %d is out of range", name, n);
|
adamc@169
|
434
|
adamc@144
|
435 ctx->inputs[n] = value;
|
adamc@144
|
436
|
adamc@282
|
437 //printf("[%d] %s = %s\n", n, name, value);
|
adamc@144
|
438 }
|
adamc@144
|
439
|
adamc@311
|
440 char *uw_get_input(uw_context ctx, int n) {
|
adamc@169
|
441 if (n < 0)
|
adamc@311
|
442 uw_error(ctx, FATAL, "Negative input index %d", n);
|
adamc@311
|
443 if (n >= uw_inputs_len)
|
adamc@311
|
444 uw_error(ctx, FATAL, "Out-of-bounds input index %d", n);
|
adamc@282
|
445 //printf("[%d] = %s\n", n, ctx->inputs[n]);
|
adamc@144
|
446 return ctx->inputs[n];
|
adamc@138
|
447 }
|
adamc@138
|
448
|
adamc@311
|
449 char *uw_get_optional_input(uw_context ctx, int n) {
|
adamc@190
|
450 if (n < 0)
|
adamc@311
|
451 uw_error(ctx, FATAL, "Negative input index %d", n);
|
adamc@311
|
452 if (n >= uw_inputs_len)
|
adamc@311
|
453 uw_error(ctx, FATAL, "Out-of-bounds input index %d", n);
|
adamc@426
|
454 //printf("[%d] = %s\n", n, ctx->inputs[n]);
|
adamc@190
|
455 return (ctx->inputs[n] == NULL ? "" : ctx->inputs[n]);
|
adamc@190
|
456 }
|
adamc@190
|
457
|
adamc@643
|
458 void uw_set_script_header(uw_context ctx, const char *s) {
|
adamc@643
|
459 ctx->script_header = s;
|
adamc@643
|
460 }
|
adamc@643
|
461
|
adamc@667
|
462 void uw_set_url_prefix(uw_context ctx, const char *s) {
|
adamc@667
|
463 ctx->url_prefix = s;
|
adamc@667
|
464 }
|
adamc@667
|
465
|
adamc@667
|
466
|
adamc@667
|
467 static void buf_check_ctx(uw_context ctx, buf *b, size_t extra, const char *desc) {
|
adamc@666
|
468 if (b->back - b->front < extra) {
|
adamc@666
|
469 size_t desired = b->front - b->start + extra, next;
|
adamc@136
|
470 char *new_heap;
|
adamc@136
|
471
|
adamc@666
|
472 next = b->back - b->start;
|
adamc@317
|
473 if (next == 0)
|
adamc@317
|
474 next = 1;
|
adamc@317
|
475 for (; next < desired; next *= 2);
|
adamc@136
|
476
|
adamc@666
|
477 new_heap = realloc(b->start, next);
|
adamc@666
|
478 b->front = new_heap + (b->front - b->start);
|
adamc@666
|
479 b->back = new_heap + next;
|
adamc@136
|
480
|
adamc@667
|
481 if (new_heap != b->start) {
|
adamc@666
|
482 b->start = new_heap;
|
adamc@666
|
483 uw_error(ctx, UNLIMITED_RETRY, "Couldn't allocate new %s contiguously", desc);
|
adamc@136
|
484 }
|
adamc@136
|
485
|
adamc@666
|
486 b->start = new_heap;
|
adamc@136
|
487 }
|
adamc@136
|
488 }
|
adamc@136
|
489
|
adamc@666
|
490 static void uw_check_heap(uw_context ctx, size_t extra) {
|
adamc@667
|
491 buf_check_ctx(ctx, &ctx->heap, extra, "heap chunk");
|
adamc@666
|
492 }
|
adamc@666
|
493
|
adamc@311
|
494 void *uw_malloc(uw_context ctx, size_t len) {
|
adamc@136
|
495 void *result;
|
adamc@136
|
496
|
adamc@311
|
497 uw_check_heap(ctx, len);
|
adamc@136
|
498
|
adamc@666
|
499 result = ctx->heap.front;
|
adamc@666
|
500 ctx->heap.front += len;
|
adamc@136
|
501 return result;
|
adamc@117
|
502 }
|
adamc@117
|
503
|
adamc@323
|
504 void uw_begin_region(uw_context ctx) {
|
adamc@666
|
505 regions *r = (regions *) ctx->heap.front;
|
adamc@323
|
506
|
adamc@323
|
507 uw_check_heap(ctx, sizeof(regions));
|
adamc@323
|
508
|
adamc@666
|
509 ctx->heap.front += sizeof(regions);
|
adamc@323
|
510
|
adamc@323
|
511 r->next = ctx->regions;
|
adamc@323
|
512 ctx->regions = r;
|
adamc@323
|
513 }
|
adamc@323
|
514
|
adamc@323
|
515 void uw_end_region(uw_context ctx) {
|
adamc@323
|
516 regions *r = ctx->regions;
|
adamc@323
|
517
|
adamc@323
|
518 if (r == NULL)
|
adamc@323
|
519 uw_error(ctx, FATAL, "Region stack underflow");
|
adamc@323
|
520
|
adamc@666
|
521 ctx->heap.front = (char *) r;
|
adamc@323
|
522 ctx->regions = r->next;
|
adamc@323
|
523 }
|
adamc@323
|
524
|
adamc@324
|
525 void uw_memstats(uw_context ctx) {
|
adamc@666
|
526 printf("Headers: %d/%d\n", buf_used(&ctx->outHeaders), buf_avail(&ctx->outHeaders));
|
adamc@666
|
527 printf("Script: %d/%d\n", buf_used(&ctx->script), buf_avail(&ctx->script));
|
adamc@666
|
528 printf("Page: %d/%d\n", buf_used(&ctx->page), buf_avail(&ctx->page));
|
adamc@666
|
529 printf("Heap: %d/%d\n", buf_used(&ctx->heap), buf_avail(&ctx->heap));
|
adamc@324
|
530 }
|
adamc@324
|
531
|
adamc@311
|
532 int uw_send(uw_context ctx, int sock) {
|
adamc@666
|
533 int n = uw_really_send(sock, ctx->outHeaders.start, ctx->outHeaders.front - ctx->outHeaders.start);
|
adamc@462
|
534
|
adamc@462
|
535 if (n < 0)
|
adamc@462
|
536 return n;
|
adamc@462
|
537
|
adamc@462
|
538 n = uw_really_send(sock, "\r\n", 2);
|
adamc@462
|
539
|
adamc@462
|
540 if (n < 0)
|
adamc@462
|
541 return n;
|
adamc@462
|
542
|
adamc@666
|
543 return uw_really_send(sock, ctx->page.start, ctx->page.front - ctx->page.start);
|
adamc@462
|
544 }
|
adamc@462
|
545
|
adamc@462
|
546 static void uw_check_headers(uw_context ctx, size_t extra) {
|
adamc@667
|
547 buf_check(&ctx->outHeaders, extra);
|
adamc@462
|
548 }
|
adamc@462
|
549
|
adamc@462
|
550 void uw_write_header(uw_context ctx, uw_Basis_string s) {
|
adamc@462
|
551 int len = strlen(s);
|
adamc@462
|
552
|
adamc@462
|
553 uw_check_headers(ctx, len + 1);
|
adamc@666
|
554 strcpy(ctx->outHeaders.front, s);
|
adamc@666
|
555 ctx->outHeaders.front += len;
|
adamc@562
|
556 }
|
adamc@562
|
557
|
adamc@562
|
558 static void uw_check_script(uw_context ctx, size_t extra) {
|
adamc@667
|
559 buf_check(&ctx->script, extra);
|
adamc@562
|
560 }
|
adamc@562
|
561
|
adamc@562
|
562 void uw_write_script(uw_context ctx, uw_Basis_string s) {
|
adamc@562
|
563 int len = strlen(s);
|
adamc@562
|
564
|
adamc@562
|
565 uw_check_script(ctx, len + 1);
|
adamc@666
|
566 strcpy(ctx->script.front, s);
|
adamc@666
|
567 ctx->script.front += len;
|
adamc@562
|
568 }
|
adamc@562
|
569
|
adamc@645
|
570 const char *uw_Basis_get_script(uw_context ctx, uw_unit u) {
|
adamc@667
|
571 if (ctx->script_header[0] == 0)
|
adamc@667
|
572 return "";
|
adamc@667
|
573 else {
|
adamc@667
|
574 int pass;
|
adamc@667
|
575 client *c = uw_new_client(&pass);
|
adamc@565
|
576
|
adamc@667
|
577 char *r = uw_malloc(ctx, strlen(ctx->script_header) + 56 + 2 * INTS_MAX + buf_used(&ctx->script)
|
adamc@667
|
578 + strlen(ctx->url_prefix));
|
adamc@667
|
579 sprintf(r, "%s<script>client_id=%d;client_pass=%d;url_prefix=\"%s\";%s</script>",
|
adamc@667
|
580 ctx->script_header, (int)c->id, c->data.used.pass, ctx->url_prefix, ctx->script.start);
|
adamc@565
|
581 return r;
|
adamc@565
|
582 }
|
adamc@565
|
583 }
|
adamc@565
|
584
|
adamc@667
|
585 const char *uw_Basis_get_listener(uw_context ctx, uw_unit u) {
|
adamc@667
|
586 if (ctx->script_header[0] == 0)
|
adamc@667
|
587 return "";
|
adamc@667
|
588 else
|
adamc@667
|
589 return " onload='listener()'";
|
adamc@667
|
590 }
|
adamc@667
|
591
|
adamc@574
|
592 uw_Basis_string uw_Basis_jsifyString(uw_context ctx, uw_Basis_string s) {
|
adamc@574
|
593 char *r, *s2;
|
adamc@574
|
594
|
adamc@574
|
595 uw_check_heap(ctx, strlen(s) * 4 + 2);
|
adamc@574
|
596
|
adamc@666
|
597 r = s2 = ctx->heap.front;
|
adamc@574
|
598 *s2++ = '"';
|
adamc@574
|
599
|
adamc@574
|
600 for (; *s; s++) {
|
adamc@574
|
601 char c = *s;
|
adamc@574
|
602
|
adamc@574
|
603 switch (c) {
|
adamc@574
|
604 case '"':
|
adamc@574
|
605 strcpy(s2, "\\\"");
|
adamc@574
|
606 s2 += 2;
|
adamc@574
|
607 break;
|
adamc@574
|
608 case '\\':
|
adamc@574
|
609 strcpy(s2, "\\\\");
|
adamc@574
|
610 s2 += 2;
|
adamc@574
|
611 break;
|
adamc@574
|
612 default:
|
adamc@574
|
613 if (isprint(c))
|
adamc@574
|
614 *s2++ = c;
|
adamc@574
|
615 else {
|
adamc@574
|
616 sprintf(s2, "\\%3o", c);
|
adamc@574
|
617 s2 += 4;
|
adamc@574
|
618 }
|
adamc@574
|
619 }
|
adamc@574
|
620 }
|
adamc@574
|
621
|
adamc@574
|
622 strcpy(s2, "\"");
|
adamc@666
|
623 ctx->heap.front = s2 + 1;
|
adamc@574
|
624 return r;
|
adamc@574
|
625 }
|
adamc@574
|
626
|
adamc@574
|
627 uw_Basis_string uw_Basis_jsifyString_ws(uw_context ctx, uw_Basis_string s) {
|
adamc@574
|
628 char *r, *s2;
|
adamc@574
|
629
|
adamc@574
|
630 uw_check_script(ctx, strlen(s) * 4 + 2);
|
adamc@574
|
631
|
adamc@666
|
632 r = s2 = ctx->script.front;
|
adamc@574
|
633 *s2++ = '"';
|
adamc@574
|
634
|
adamc@574
|
635 for (; *s; s++) {
|
adamc@574
|
636 char c = *s;
|
adamc@574
|
637
|
adamc@574
|
638 switch (c) {
|
adamc@577
|
639 case '\'':
|
adamc@574
|
640 strcpy(s2, "\\\"");
|
adamc@574
|
641 s2 += 2;
|
adamc@574
|
642 break;
|
adamc@574
|
643 case '\\':
|
adamc@574
|
644 strcpy(s2, "\\\\");
|
adamc@574
|
645 s2 += 2;
|
adamc@574
|
646 break;
|
adamc@574
|
647 default:
|
adamc@574
|
648 if (isprint(c))
|
adamc@574
|
649 *s2++ = c;
|
adamc@574
|
650 else {
|
adamc@574
|
651 sprintf(s2, "\\%3o", c);
|
adamc@574
|
652 s2 += 4;
|
adamc@574
|
653 }
|
adamc@574
|
654 }
|
adamc@574
|
655 }
|
adamc@574
|
656
|
adamc@574
|
657 strcpy(s2, "\"");
|
adamc@666
|
658 ctx->script.front = s2 + 1;
|
adamc@574
|
659 return r;
|
adamc@574
|
660 }
|
adamc@574
|
661
|
adamc@577
|
662 uw_Basis_int uw_Basis_new_client_source(uw_context ctx, uw_Basis_string s) {
|
adamc@577
|
663 int len;
|
adamc@577
|
664 size_t s_len = strlen(s);
|
adamc@562
|
665
|
adamc@577
|
666 uw_check_script(ctx, 12 + INTS_MAX + s_len);
|
adamc@666
|
667 sprintf(ctx->script.front, "var s%d=sc(%n", ctx->source_count, &len);
|
adamc@666
|
668 ctx->script.front += len;
|
adamc@666
|
669 strcpy(ctx->script.front, s);
|
adamc@666
|
670 ctx->script.front += s_len;
|
adamc@666
|
671 strcpy(ctx->script.front, ");");
|
adamc@666
|
672 ctx->script.front += 2;
|
adamc@562
|
673
|
adamc@565
|
674 return ctx->source_count++;
|
adamc@117
|
675 }
|
adamc@117
|
676
|
adamc@577
|
677 uw_unit uw_Basis_set_client_source(uw_context ctx, uw_Basis_int n, uw_Basis_string s) {
|
adamc@577
|
678 int len;
|
adamc@577
|
679 size_t s_len = strlen(s);
|
adamc@577
|
680
|
adamc@577
|
681 uw_check_script(ctx, 6 + INTS_MAX + s_len);
|
adamc@666
|
682 sprintf(ctx->script.front, "s%d.v=%n", (int)n, &len);
|
adamc@666
|
683 ctx->script.front += len;
|
adamc@666
|
684 strcpy(ctx->script.front, s);
|
adamc@666
|
685 ctx->script.front += s_len;
|
adamc@666
|
686 strcpy(ctx->script.front, ";");
|
adamc@666
|
687 ctx->script.front++;
|
adamc@577
|
688
|
adamc@577
|
689 return uw_unit_v;
|
adamc@577
|
690 }
|
adamc@577
|
691
|
adamc@311
|
692 static void uw_check(uw_context ctx, size_t extra) {
|
adamc@667
|
693 buf_check(&ctx->page, extra);
|
adamc@117
|
694 }
|
adamc@117
|
695
|
adamc@311
|
696 static void uw_writec_unsafe(uw_context ctx, char c) {
|
adamc@666
|
697 *(ctx->page.front)++ = c;
|
adamc@117
|
698 }
|
adamc@117
|
699
|
adamc@311
|
700 void uw_writec(uw_context ctx, char c) {
|
adamc@311
|
701 uw_check(ctx, 1);
|
adamc@311
|
702 uw_writec_unsafe(ctx, c);
|
adamc@117
|
703 }
|
adamc@117
|
704
|
adamc@311
|
705 static void uw_write_unsafe(uw_context ctx, const char* s) {
|
adamc@117
|
706 int len = strlen(s);
|
adamc@666
|
707 memcpy(ctx->page.front, s, len);
|
adamc@666
|
708 ctx->page.front += len;
|
adamc@117
|
709 }
|
adamc@117
|
710
|
adamc@311
|
711 void uw_write(uw_context ctx, const char* s) {
|
adamc@311
|
712 uw_check(ctx, strlen(s) + 1);
|
adamc@311
|
713 uw_write_unsafe(ctx, s);
|
adamc@666
|
714 *ctx->page.front = 0;
|
adamc@102
|
715 }
|
adamc@106
|
716
|
adamc@311
|
717 char *uw_Basis_attrifyInt(uw_context ctx, uw_Basis_int n) {
|
adamc@136
|
718 char *result;
|
adamc@136
|
719 int len;
|
adamc@311
|
720 uw_check_heap(ctx, INTS_MAX);
|
adamc@666
|
721 result = ctx->heap.front;
|
adamc@276
|
722 sprintf(result, "%lld%n", n, &len);
|
adamc@666
|
723 ctx->heap.front += len+1;
|
adamc@136
|
724 return result;
|
adamc@106
|
725 }
|
adamc@106
|
726
|
adamc@311
|
727 char *uw_Basis_attrifyFloat(uw_context ctx, uw_Basis_float n) {
|
adamc@136
|
728 char *result;
|
adamc@136
|
729 int len;
|
adamc@311
|
730 uw_check_heap(ctx, FLOATS_MAX);
|
adamc@666
|
731 result = ctx->heap.front;
|
adamc@136
|
732 sprintf(result, "%g%n", n, &len);
|
adamc@666
|
733 ctx->heap.front += len+1;
|
adamc@136
|
734 return result;
|
adamc@106
|
735 }
|
adamc@106
|
736
|
adamc@311
|
737 char *uw_Basis_attrifyString(uw_context ctx, uw_Basis_string s) {
|
adamc@136
|
738 int len = strlen(s);
|
adamc@136
|
739 char *result, *p;
|
adamc@311
|
740 uw_check_heap(ctx, len * 6 + 1);
|
adamc@136
|
741
|
adamc@666
|
742 result = p = ctx->heap.front;
|
adamc@136
|
743
|
adamc@136
|
744 for (; *s; s++) {
|
adamc@136
|
745 char c = *s;
|
adamc@136
|
746
|
adamc@136
|
747 if (c == '"') {
|
adamc@136
|
748 strcpy(p, """);
|
adamc@136
|
749 p += 6;
|
adamc@136
|
750 } else if (c == '&') {
|
adamc@136
|
751 strcpy(p, "&");
|
adamc@136
|
752 p += 5;
|
adamc@136
|
753 }
|
adamc@136
|
754 else if (isprint(c))
|
adamc@136
|
755 *p++ = c;
|
adamc@136
|
756 else {
|
adamc@136
|
757 int len2;
|
adamc@136
|
758 sprintf(p, "&#%d;%n", c, &len2);
|
adamc@136
|
759 p += len2;
|
adamc@136
|
760 }
|
adamc@136
|
761 }
|
adamc@136
|
762
|
adamc@137
|
763 *p++ = 0;
|
adamc@666
|
764 ctx->heap.front = p;
|
adamc@136
|
765 return result;
|
adamc@106
|
766 }
|
adamc@106
|
767
|
adamc@311
|
768 static void uw_Basis_attrifyInt_w_unsafe(uw_context ctx, uw_Basis_int n) {
|
adamc@117
|
769 int len;
|
adamc@117
|
770
|
adamc@666
|
771 sprintf(ctx->page.front, "%lld%n", n, &len);
|
adamc@666
|
772 ctx->page.front += len;
|
adamc@106
|
773 }
|
adamc@106
|
774
|
adamc@428
|
775 uw_unit uw_Basis_attrifyInt_w(uw_context ctx, uw_Basis_int n) {
|
adamc@311
|
776 uw_check(ctx, INTS_MAX);
|
adamc@311
|
777 uw_Basis_attrifyInt_w_unsafe(ctx, n);
|
adamc@428
|
778
|
adamc@428
|
779 return uw_unit_v;
|
adamc@106
|
780 }
|
adamc@106
|
781
|
adamc@428
|
782 uw_unit uw_Basis_attrifyFloat_w(uw_context ctx, uw_Basis_float n) {
|
adamc@117
|
783 int len;
|
adamc@117
|
784
|
adamc@311
|
785 uw_check(ctx, FLOATS_MAX);
|
adamc@666
|
786 sprintf(ctx->page.front, "%g%n", n, &len);
|
adamc@666
|
787 ctx->page.front += len;
|
adamc@428
|
788
|
adamc@428
|
789 return uw_unit_v;
|
adamc@117
|
790 }
|
adamc@117
|
791
|
adamc@428
|
792 uw_unit uw_Basis_attrifyString_w(uw_context ctx, uw_Basis_string s) {
|
adamc@311
|
793 uw_check(ctx, strlen(s) * 6);
|
adamc@117
|
794
|
adamc@106
|
795 for (; *s; s++) {
|
adamc@106
|
796 char c = *s;
|
adamc@106
|
797
|
adamc@106
|
798 if (c == '"')
|
adamc@311
|
799 uw_write_unsafe(ctx, """);
|
adamc@136
|
800 else if (c == '&')
|
adamc@311
|
801 uw_write_unsafe(ctx, "&");
|
adamc@106
|
802 else if (isprint(c))
|
adamc@311
|
803 uw_writec_unsafe(ctx, c);
|
adamc@106
|
804 else {
|
adamc@311
|
805 uw_write_unsafe(ctx, "&#");
|
adamc@311
|
806 uw_Basis_attrifyInt_w_unsafe(ctx, c);
|
adamc@311
|
807 uw_writec_unsafe(ctx, ';');
|
adamc@106
|
808 }
|
adamc@106
|
809 }
|
adamc@428
|
810
|
adamc@428
|
811 return uw_unit_v;
|
adamc@106
|
812 }
|
adamc@120
|
813
|
adamc@120
|
814
|
adamc@311
|
815 char *uw_Basis_urlifyInt(uw_context ctx, uw_Basis_int n) {
|
adamc@137
|
816 int len;
|
adamc@137
|
817 char *r;
|
adamc@137
|
818
|
adamc@311
|
819 uw_check_heap(ctx, INTS_MAX);
|
adamc@666
|
820 r = ctx->heap.front;
|
adamc@276
|
821 sprintf(r, "%lld%n", n, &len);
|
adamc@666
|
822 ctx->heap.front += len+1;
|
adamc@137
|
823 return r;
|
adamc@120
|
824 }
|
adamc@120
|
825
|
adamc@311
|
826 char *uw_Basis_urlifyFloat(uw_context ctx, uw_Basis_float n) {
|
adamc@137
|
827 int len;
|
adamc@137
|
828 char *r;
|
adamc@137
|
829
|
adamc@311
|
830 uw_check_heap(ctx, FLOATS_MAX);
|
adamc@666
|
831 r = ctx->heap.front;
|
adamc@137
|
832 sprintf(r, "%g%n", n, &len);
|
adamc@666
|
833 ctx->heap.front += len+1;
|
adamc@137
|
834 return r;
|
adamc@120
|
835 }
|
adamc@120
|
836
|
adamc@311
|
837 char *uw_Basis_urlifyString(uw_context ctx, uw_Basis_string s) {
|
adamc@137
|
838 char *r, *p;
|
adamc@137
|
839
|
adamc@311
|
840 uw_check_heap(ctx, strlen(s) * 3 + 1);
|
adamc@137
|
841
|
adamc@666
|
842 for (r = p = ctx->heap.front; *s; s++) {
|
adamc@137
|
843 char c = *s;
|
adamc@137
|
844
|
adamc@137
|
845 if (c == ' ')
|
adamc@137
|
846 *p++ = '+';
|
adamc@137
|
847 else if (isalnum(c))
|
adamc@137
|
848 *p++ = c;
|
adamc@137
|
849 else {
|
adamc@137
|
850 sprintf(p, "%%%02X", c);
|
adamc@137
|
851 p += 3;
|
adamc@137
|
852 }
|
adamc@137
|
853 }
|
adamc@137
|
854
|
adamc@137
|
855 *p++ = 0;
|
adamc@666
|
856 ctx->heap.front = p;
|
adamc@137
|
857 return r;
|
adamc@120
|
858 }
|
adamc@120
|
859
|
adamc@311
|
860 char *uw_Basis_urlifyBool(uw_context ctx, uw_Basis_bool b) {
|
adamc@311
|
861 if (b == uw_Basis_False)
|
adamc@186
|
862 return "0";
|
adamc@186
|
863 else
|
adamc@186
|
864 return "1";
|
adamc@186
|
865 }
|
adamc@186
|
866
|
adamc@311
|
867 static void uw_Basis_urlifyInt_w_unsafe(uw_context ctx, uw_Basis_int n) {
|
adamc@120
|
868 int len;
|
adamc@120
|
869
|
adamc@666
|
870 sprintf(ctx->page.front, "%lld%n", n, &len);
|
adamc@666
|
871 ctx->page.front += len;
|
adamc@120
|
872 }
|
adamc@120
|
873
|
adamc@428
|
874 uw_unit uw_Basis_urlifyInt_w(uw_context ctx, uw_Basis_int n) {
|
adamc@311
|
875 uw_check(ctx, INTS_MAX);
|
adamc@311
|
876 uw_Basis_urlifyInt_w_unsafe(ctx, n);
|
adamc@428
|
877
|
adamc@428
|
878 return uw_unit_v;
|
adamc@120
|
879 }
|
adamc@120
|
880
|
adamc@428
|
881 uw_unit uw_Basis_urlifyFloat_w(uw_context ctx, uw_Basis_float n) {
|
adamc@120
|
882 int len;
|
adamc@120
|
883
|
adamc@311
|
884 uw_check(ctx, FLOATS_MAX);
|
adamc@666
|
885 sprintf(ctx->page.front, "%g%n", n, &len);
|
adamc@666
|
886 ctx->page.front += len;
|
adamc@428
|
887
|
adamc@428
|
888 return uw_unit_v;
|
adamc@120
|
889 }
|
adamc@120
|
890
|
adamc@488
|
891 uw_Basis_string uw_Basis_urlifyTime(uw_context ctx, uw_Basis_time t) {
|
adamc@488
|
892 return uw_Basis_urlifyInt(ctx, t);
|
adamc@488
|
893 }
|
adamc@488
|
894
|
adamc@428
|
895 uw_unit uw_Basis_urlifyString_w(uw_context ctx, uw_Basis_string s) {
|
adamc@311
|
896 uw_check(ctx, strlen(s) * 3);
|
adamc@120
|
897
|
adamc@120
|
898 for (; *s; s++) {
|
adamc@120
|
899 char c = *s;
|
adamc@120
|
900
|
adamc@120
|
901 if (c == ' ')
|
adamc@311
|
902 uw_writec_unsafe(ctx, '+');
|
adamc@120
|
903 else if (isalnum(c))
|
adamc@311
|
904 uw_writec_unsafe(ctx, c);
|
adamc@120
|
905 else {
|
adamc@666
|
906 sprintf(ctx->page.front, "%%%02X", c);
|
adamc@666
|
907 ctx->page.front += 3;
|
adamc@120
|
908 }
|
adamc@120
|
909 }
|
adamc@428
|
910
|
adamc@428
|
911 return uw_unit_v;
|
adamc@120
|
912 }
|
adamc@120
|
913
|
adamc@428
|
914 uw_unit uw_Basis_urlifyBool_w(uw_context ctx, uw_Basis_bool b) {
|
adamc@311
|
915 if (b == uw_Basis_False)
|
adamc@311
|
916 uw_writec(ctx, '0');
|
adamc@186
|
917 else
|
adamc@311
|
918 uw_writec(ctx, '1');
|
adamc@428
|
919
|
adamc@428
|
920 return uw_unit_v;
|
adamc@186
|
921 }
|
adamc@186
|
922
|
adamc@120
|
923
|
adamc@311
|
924 static char *uw_unurlify_advance(char *s) {
|
adamc@144
|
925 char *new_s = strchr(s, '/');
|
adamc@120
|
926
|
adamc@120
|
927 if (new_s)
|
adamc@120
|
928 *new_s++ = 0;
|
adamc@120
|
929 else
|
adamc@144
|
930 new_s = strchr(s, 0);
|
adamc@144
|
931
|
adamc@144
|
932 return new_s;
|
adamc@144
|
933 }
|
adamc@144
|
934
|
adamc@311
|
935 uw_Basis_int uw_Basis_unurlifyInt(uw_context ctx, char **s) {
|
adamc@311
|
936 char *new_s = uw_unurlify_advance(*s);
|
adamc@311
|
937 uw_Basis_int r;
|
adamc@120
|
938
|
adamc@276
|
939 r = atoll(*s);
|
adamc@120
|
940 *s = new_s;
|
adamc@120
|
941 return r;
|
adamc@120
|
942 }
|
adamc@120
|
943
|
adamc@311
|
944 uw_Basis_float uw_Basis_unurlifyFloat(uw_context ctx, char **s) {
|
adamc@311
|
945 char *new_s = uw_unurlify_advance(*s);
|
adamc@311
|
946 uw_Basis_float r;
|
adamc@120
|
947
|
adamc@120
|
948 r = atof(*s);
|
adamc@120
|
949 *s = new_s;
|
adamc@120
|
950 return r;
|
adamc@120
|
951 }
|
adamc@120
|
952
|
adamc@488
|
953 uw_Basis_time uw_Basis_unurlifyTime(uw_context ctx, char **s) {
|
adamc@488
|
954 return uw_Basis_unurlifyInt(ctx, s);
|
adamc@488
|
955 }
|
adamc@488
|
956
|
adamc@311
|
957 static uw_Basis_string uw_unurlifyString_to(uw_context ctx, char *r, char *s) {
|
adamc@144
|
958 char *s1, *s2;
|
adamc@144
|
959 int n;
|
adamc@136
|
960
|
adamc@144
|
961 for (s1 = r, s2 = s; *s2; ++s1, ++s2) {
|
adamc@136
|
962 char c = *s2;
|
adamc@136
|
963
|
adamc@136
|
964 switch (c) {
|
adamc@136
|
965 case '+':
|
adamc@136
|
966 *s1 = ' ';
|
adamc@136
|
967 break;
|
adamc@136
|
968 case '%':
|
adamc@169
|
969 if (s2[1] == 0)
|
adamc@311
|
970 uw_error(ctx, FATAL, "Missing first character of escaped URL byte");
|
adamc@169
|
971 if (s2[2] == 0)
|
adamc@311
|
972 uw_error(ctx, FATAL, "Missing second character of escaped URL byte");
|
adamc@169
|
973 if (sscanf(s2+1, "%02X", &n) != 1)
|
adamc@311
|
974 uw_error(ctx, FATAL, "Invalid escaped URL byte starting at: %s", s2);
|
adamc@136
|
975 *s1 = n;
|
adamc@136
|
976 s2 += 2;
|
adamc@136
|
977 break;
|
adamc@136
|
978 default:
|
adamc@136
|
979 *s1 = c;
|
adamc@136
|
980 }
|
adamc@136
|
981 }
|
adamc@136
|
982 *s1++ = 0;
|
adamc@144
|
983 return s1;
|
adamc@144
|
984 }
|
adamc@144
|
985
|
adamc@311
|
986 uw_Basis_bool uw_Basis_unurlifyBool(uw_context ctx, char **s) {
|
adamc@311
|
987 char *new_s = uw_unurlify_advance(*s);
|
adamc@311
|
988 uw_Basis_bool r;
|
adamc@186
|
989
|
adamc@186
|
990 if (*s[0] == 0 || !strcmp(*s, "0") || !strcmp(*s, "off"))
|
adamc@311
|
991 r = uw_Basis_False;
|
adamc@186
|
992 else
|
adamc@311
|
993 r = uw_Basis_True;
|
adamc@186
|
994
|
adamc@186
|
995 *s = new_s;
|
adamc@186
|
996 return r;
|
adamc@186
|
997 }
|
adamc@186
|
998
|
adamc@311
|
999 uw_Basis_string uw_Basis_unurlifyString(uw_context ctx, char **s) {
|
adamc@311
|
1000 char *new_s = uw_unurlify_advance(*s);
|
adamc@144
|
1001 char *r, *s1, *s2;
|
adamc@144
|
1002 int len, n;
|
adamc@144
|
1003
|
adamc@200
|
1004 len = strlen(*s);
|
adamc@311
|
1005 uw_check_heap(ctx, len + 1);
|
adamc@144
|
1006
|
adamc@666
|
1007 r = ctx->heap.front;
|
adamc@666
|
1008 ctx->heap.front = uw_unurlifyString_to(ctx, ctx->heap.front, *s);
|
adamc@136
|
1009 *s = new_s;
|
adamc@136
|
1010 return r;
|
adamc@120
|
1011 }
|
adamc@135
|
1012
|
adamc@135
|
1013
|
adamc@311
|
1014 char *uw_Basis_htmlifyInt(uw_context ctx, uw_Basis_int n) {
|
adamc@286
|
1015 int len;
|
adamc@286
|
1016 char *r;
|
adamc@286
|
1017
|
adamc@311
|
1018 uw_check_heap(ctx, INTS_MAX);
|
adamc@666
|
1019 r = ctx->heap.front;
|
adamc@286
|
1020 sprintf(r, "%lld%n", n, &len);
|
adamc@666
|
1021 ctx->heap.front += len+1;
|
adamc@286
|
1022 return r;
|
adamc@286
|
1023 }
|
adamc@286
|
1024
|
adamc@428
|
1025 uw_unit uw_Basis_htmlifyInt_w(uw_context ctx, uw_Basis_int n) {
|
adamc@286
|
1026 int len;
|
adamc@286
|
1027
|
adamc@311
|
1028 uw_check(ctx, INTS_MAX);
|
adamc@666
|
1029 sprintf(ctx->page.front, "%lld%n", n, &len);
|
adamc@666
|
1030 ctx->page.front += len;
|
adamc@428
|
1031
|
adamc@428
|
1032 return uw_unit_v;
|
adamc@286
|
1033 }
|
adamc@286
|
1034
|
adamc@311
|
1035 char *uw_Basis_htmlifyFloat(uw_context ctx, uw_Basis_float n) {
|
adamc@286
|
1036 int len;
|
adamc@286
|
1037 char *r;
|
adamc@286
|
1038
|
adamc@311
|
1039 uw_check_heap(ctx, FLOATS_MAX);
|
adamc@666
|
1040 r = ctx->heap.front;
|
adamc@286
|
1041 sprintf(r, "%g%n", n, &len);
|
adamc@666
|
1042 ctx->heap.front += len+1;
|
adamc@286
|
1043 return r;
|
adamc@286
|
1044 }
|
adamc@286
|
1045
|
adamc@428
|
1046 uw_unit uw_Basis_htmlifyFloat_w(uw_context ctx, uw_Basis_float n) {
|
adamc@286
|
1047 int len;
|
adamc@286
|
1048
|
adamc@311
|
1049 uw_check(ctx, FLOATS_MAX);
|
adamc@666
|
1050 sprintf(ctx->page.front, "%g%n", n, &len);
|
adamc@666
|
1051 ctx->page.front += len;
|
adamc@428
|
1052
|
adamc@428
|
1053 return uw_unit_v;
|
adamc@286
|
1054 }
|
adamc@286
|
1055
|
adamc@311
|
1056 char *uw_Basis_htmlifyString(uw_context ctx, uw_Basis_string s) {
|
adamc@137
|
1057 char *r, *s2;
|
adamc@137
|
1058
|
adamc@311
|
1059 uw_check_heap(ctx, strlen(s) * 5 + 1);
|
adamc@137
|
1060
|
adamc@666
|
1061 for (r = s2 = ctx->heap.front; *s; s++) {
|
adamc@137
|
1062 char c = *s;
|
adamc@137
|
1063
|
adamc@137
|
1064 switch (c) {
|
adamc@137
|
1065 case '<':
|
adamc@137
|
1066 strcpy(s2, "<");
|
adamc@137
|
1067 s2 += 4;
|
adamc@137
|
1068 break;
|
adamc@137
|
1069 case '&':
|
adamc@137
|
1070 strcpy(s2, "&");
|
adamc@137
|
1071 s2 += 5;
|
adamc@137
|
1072 break;
|
adamc@137
|
1073 default:
|
adamc@137
|
1074 if (isprint(c))
|
adamc@137
|
1075 *s2++ = c;
|
adamc@137
|
1076 else {
|
adamc@137
|
1077 int len2;
|
adamc@137
|
1078 sprintf(s2, "&#%d;%n", c, &len2);
|
adamc@137
|
1079 s2 += len2;
|
adamc@137
|
1080 }
|
adamc@137
|
1081 }
|
adamc@137
|
1082 }
|
adamc@137
|
1083
|
adamc@137
|
1084 *s2++ = 0;
|
adamc@666
|
1085 ctx->heap.front = s2;
|
adamc@137
|
1086 return r;
|
adamc@135
|
1087 }
|
adamc@135
|
1088
|
adamc@428
|
1089 uw_unit uw_Basis_htmlifyString_w(uw_context ctx, uw_Basis_string s) {
|
adamc@321
|
1090 uw_check(ctx, strlen(s) * 6);
|
adamc@135
|
1091
|
adamc@135
|
1092 for (; *s; s++) {
|
adamc@135
|
1093 char c = *s;
|
adamc@135
|
1094
|
adamc@135
|
1095 switch (c) {
|
adamc@135
|
1096 case '<':
|
adamc@311
|
1097 uw_write_unsafe(ctx, "<");
|
adamc@135
|
1098 break;
|
adamc@135
|
1099 case '&':
|
adamc@311
|
1100 uw_write_unsafe(ctx, "&");
|
adamc@135
|
1101 break;
|
adamc@135
|
1102 default:
|
adamc@135
|
1103 if (isprint(c))
|
adamc@311
|
1104 uw_writec_unsafe(ctx, c);
|
adamc@135
|
1105 else {
|
adamc@311
|
1106 uw_write_unsafe(ctx, "&#");
|
adamc@311
|
1107 uw_Basis_attrifyInt_w_unsafe(ctx, c);
|
adamc@311
|
1108 uw_writec_unsafe(ctx, ';');
|
adamc@135
|
1109 }
|
adamc@135
|
1110 }
|
adamc@135
|
1111 }
|
adamc@428
|
1112
|
adamc@428
|
1113 return uw_unit_v;
|
adamc@135
|
1114 }
|
adamc@180
|
1115
|
adamc@311
|
1116 uw_Basis_string uw_Basis_htmlifyBool(uw_context ctx, uw_Basis_bool b) {
|
adamc@311
|
1117 if (b == uw_Basis_False)
|
adamc@286
|
1118 return "False";
|
adamc@286
|
1119 else
|
adamc@286
|
1120 return "True";
|
adamc@286
|
1121 }
|
adamc@286
|
1122
|
adamc@428
|
1123 uw_unit uw_Basis_htmlifyBool_w(uw_context ctx, uw_Basis_bool b) {
|
adamc@311
|
1124 if (b == uw_Basis_False) {
|
adamc@311
|
1125 uw_check(ctx, 6);
|
adamc@666
|
1126 strcpy(ctx->page.front, "False");
|
adamc@666
|
1127 ctx->page.front += 5;
|
adamc@286
|
1128 } else {
|
adamc@311
|
1129 uw_check(ctx, 5);
|
adamc@666
|
1130 strcpy(ctx->page.front, "True");
|
adamc@666
|
1131 ctx->page.front += 4;
|
adamc@286
|
1132 }
|
adamc@428
|
1133
|
adamc@428
|
1134 return uw_unit_v;
|
adamc@286
|
1135 }
|
adamc@286
|
1136
|
adamc@436
|
1137 #define TIME_FMT "%x %X"
|
adamc@438
|
1138 #define TIME_FMT_PG "%Y-%m-%d %T"
|
adamc@436
|
1139
|
adamc@436
|
1140 uw_Basis_string uw_Basis_htmlifyTime(uw_context ctx, uw_Basis_time t) {
|
adamc@436
|
1141 size_t len;
|
adamc@436
|
1142 char *r;
|
adamc@436
|
1143 struct tm stm;
|
adamc@436
|
1144
|
adamc@436
|
1145 if (localtime_r(&t, &stm)) {
|
adamc@436
|
1146 uw_check_heap(ctx, TIMES_MAX);
|
adamc@666
|
1147 r = ctx->heap.front;
|
adamc@436
|
1148 len = strftime(r, TIMES_MAX, TIME_FMT, &stm);
|
adamc@666
|
1149 ctx->heap.front += len+1;
|
adamc@436
|
1150 return r;
|
adamc@436
|
1151 } else
|
adamc@436
|
1152 return "<i>Invalid time</i>";
|
adamc@436
|
1153 }
|
adamc@436
|
1154
|
adamc@436
|
1155 uw_unit uw_Basis_htmlifyTime_w(uw_context ctx, uw_Basis_time t) {
|
adamc@436
|
1156 size_t len;
|
adamc@436
|
1157 char *r;
|
adamc@436
|
1158 struct tm stm;
|
adamc@436
|
1159
|
adamc@436
|
1160 if (localtime_r(&t, &stm)) {
|
adamc@436
|
1161 uw_check(ctx, TIMES_MAX);
|
adamc@666
|
1162 r = ctx->page.front;
|
adamc@436
|
1163 len = strftime(r, TIMES_MAX, TIME_FMT, &stm);
|
adamc@666
|
1164 ctx->page.front += len;
|
adamc@436
|
1165 } else {
|
adamc@436
|
1166 uw_check(ctx, 20);
|
adamc@666
|
1167 strcpy(ctx->page.front, "<i>Invalid time</i>");
|
adamc@666
|
1168 ctx->page.front += 19;
|
adamc@436
|
1169 }
|
adamc@436
|
1170
|
adamc@436
|
1171 return uw_unit_v;
|
adamc@436
|
1172 }
|
adamc@436
|
1173
|
adamc@311
|
1174 uw_Basis_string uw_Basis_strcat(uw_context ctx, uw_Basis_string s1, uw_Basis_string s2) {
|
adamc@180
|
1175 int len = strlen(s1) + strlen(s2) + 1;
|
adamc@180
|
1176 char *s;
|
adamc@180
|
1177
|
adamc@311
|
1178 uw_check_heap(ctx, len);
|
adamc@180
|
1179
|
adamc@666
|
1180 s = ctx->heap.front;
|
adamc@180
|
1181
|
adamc@180
|
1182 strcpy(s, s1);
|
adamc@180
|
1183 strcat(s, s2);
|
adamc@666
|
1184 ctx->heap.front += len;
|
adamc@180
|
1185
|
adamc@180
|
1186 return s;
|
adamc@180
|
1187 }
|
adamc@278
|
1188
|
adamc@311
|
1189 uw_Basis_string uw_Basis_strdup(uw_context ctx, uw_Basis_string s1) {
|
adamc@278
|
1190 int len = strlen(s1) + 1;
|
adamc@278
|
1191 char *s;
|
adamc@278
|
1192
|
adamc@311
|
1193 uw_check_heap(ctx, len);
|
adamc@278
|
1194
|
adamc@666
|
1195 s = ctx->heap.front;
|
adamc@278
|
1196
|
adamc@278
|
1197 strcpy(s, s1);
|
adamc@666
|
1198 ctx->heap.front += len;
|
adamc@278
|
1199
|
adamc@278
|
1200 return s;
|
adamc@278
|
1201 }
|
adamc@280
|
1202
|
adamc@493
|
1203 uw_Basis_string uw_Basis_maybe_strdup(uw_context ctx, uw_Basis_string s1) {
|
adamc@493
|
1204 if (s1)
|
adamc@493
|
1205 return uw_Basis_strdup(ctx, s1);
|
adamc@493
|
1206 else
|
adamc@493
|
1207 return NULL;
|
adamc@493
|
1208 }
|
adamc@493
|
1209
|
adamc@280
|
1210
|
adamc@311
|
1211 char *uw_Basis_sqlifyInt(uw_context ctx, uw_Basis_int n) {
|
adamc@281
|
1212 int len;
|
adamc@281
|
1213 char *r;
|
adamc@281
|
1214
|
adamc@311
|
1215 uw_check_heap(ctx, INTS_MAX + 6);
|
adamc@666
|
1216 r = ctx->heap.front;
|
adamc@281
|
1217 sprintf(r, "%lld::int8%n", n, &len);
|
adamc@666
|
1218 ctx->heap.front += len+1;
|
adamc@281
|
1219 return r;
|
adamc@281
|
1220 }
|
adamc@281
|
1221
|
adamc@467
|
1222 char *uw_Basis_sqlifyIntN(uw_context ctx, uw_Basis_int *n) {
|
adamc@467
|
1223 if (n == NULL)
|
adamc@467
|
1224 return "NULL";
|
adamc@467
|
1225 else
|
adamc@467
|
1226 return uw_Basis_sqlifyInt(ctx, *n);
|
adamc@467
|
1227 }
|
adamc@467
|
1228
|
adamc@311
|
1229 char *uw_Basis_sqlifyFloat(uw_context ctx, uw_Basis_float n) {
|
adamc@281
|
1230 int len;
|
adamc@281
|
1231 char *r;
|
adamc@281
|
1232
|
adamc@311
|
1233 uw_check_heap(ctx, FLOATS_MAX + 8);
|
adamc@666
|
1234 r = ctx->heap.front;
|
adamc@281
|
1235 sprintf(r, "%g::float8%n", n, &len);
|
adamc@666
|
1236 ctx->heap.front += len+1;
|
adamc@281
|
1237 return r;
|
adamc@281
|
1238 }
|
adamc@281
|
1239
|
adamc@467
|
1240 char *uw_Basis_sqlifyFloatN(uw_context ctx, uw_Basis_float *n) {
|
adamc@467
|
1241 if (n == NULL)
|
adamc@467
|
1242 return "NULL";
|
adamc@467
|
1243 else
|
adamc@467
|
1244 return uw_Basis_sqlifyFloat(ctx, *n);
|
adamc@467
|
1245 }
|
adamc@467
|
1246
|
adamc@281
|
1247
|
adamc@311
|
1248 uw_Basis_string uw_Basis_sqlifyString(uw_context ctx, uw_Basis_string s) {
|
adamc@280
|
1249 char *r, *s2;
|
adamc@280
|
1250
|
adamc@311
|
1251 uw_check_heap(ctx, strlen(s) * 2 + 10);
|
adamc@280
|
1252
|
adamc@666
|
1253 r = s2 = ctx->heap.front;
|
adamc@280
|
1254 *s2++ = 'E';
|
adamc@280
|
1255 *s2++ = '\'';
|
adamc@280
|
1256
|
adamc@280
|
1257 for (; *s; s++) {
|
adamc@280
|
1258 char c = *s;
|
adamc@280
|
1259
|
adamc@280
|
1260 switch (c) {
|
adamc@280
|
1261 case '\'':
|
adamc@280
|
1262 strcpy(s2, "\\'");
|
adamc@280
|
1263 s2 += 2;
|
adamc@280
|
1264 break;
|
adamc@280
|
1265 case '\\':
|
adamc@280
|
1266 strcpy(s2, "\\\\");
|
adamc@280
|
1267 s2 += 2;
|
adamc@280
|
1268 break;
|
adamc@280
|
1269 default:
|
adamc@280
|
1270 if (isprint(c))
|
adamc@280
|
1271 *s2++ = c;
|
adamc@280
|
1272 else {
|
adamc@280
|
1273 sprintf(s2, "\\%3o", c);
|
adamc@280
|
1274 s2 += 4;
|
adamc@280
|
1275 }
|
adamc@280
|
1276 }
|
adamc@280
|
1277 }
|
adamc@280
|
1278
|
adamc@281
|
1279 strcpy(s2, "'::text");
|
adamc@666
|
1280 ctx->heap.front = s2 + 8;
|
adamc@280
|
1281 return r;
|
adamc@280
|
1282 }
|
adamc@281
|
1283
|
adamc@467
|
1284 uw_Basis_string uw_Basis_sqlifyStringN(uw_context ctx, uw_Basis_string s) {
|
adamc@467
|
1285 if (s == NULL)
|
adamc@467
|
1286 return "NULL";
|
adamc@467
|
1287 else
|
adamc@467
|
1288 return uw_Basis_sqlifyString(ctx, s);
|
adamc@467
|
1289 }
|
adamc@467
|
1290
|
adamc@311
|
1291 char *uw_Basis_sqlifyBool(uw_context ctx, uw_Basis_bool b) {
|
adamc@311
|
1292 if (b == uw_Basis_False)
|
adamc@281
|
1293 return "FALSE";
|
adamc@281
|
1294 else
|
adamc@281
|
1295 return "TRUE";
|
adamc@281
|
1296 }
|
adamc@282
|
1297
|
adamc@467
|
1298 char *uw_Basis_sqlifyBoolN(uw_context ctx, uw_Basis_bool *b) {
|
adamc@467
|
1299 if (b == NULL)
|
adamc@467
|
1300 return "NULL";
|
adamc@467
|
1301 else
|
adamc@467
|
1302 return uw_Basis_sqlifyBool(ctx, *b);
|
adamc@467
|
1303 }
|
adamc@467
|
1304
|
adamc@439
|
1305 char *uw_Basis_sqlifyTime(uw_context ctx, uw_Basis_time t) {
|
adamc@439
|
1306 size_t len;
|
adamc@439
|
1307 char *r;
|
adamc@439
|
1308 struct tm stm;
|
adamc@439
|
1309
|
adamc@439
|
1310 if (localtime_r(&t, &stm)) {
|
adamc@439
|
1311 uw_check_heap(ctx, TIMES_MAX);
|
adamc@666
|
1312 r = ctx->heap.front;
|
adamc@439
|
1313 len = strftime(r, TIMES_MAX, TIME_FMT, &stm);
|
adamc@666
|
1314 ctx->heap.front += len+1;
|
adamc@439
|
1315 return r;
|
adamc@439
|
1316 } else
|
adamc@439
|
1317 return "<Invalid time>";
|
adamc@439
|
1318 }
|
adamc@439
|
1319
|
adamc@467
|
1320 char *uw_Basis_sqlifyTimeN(uw_context ctx, uw_Basis_time *t) {
|
adamc@467
|
1321 if (t == NULL)
|
adamc@467
|
1322 return "NULL";
|
adamc@467
|
1323 else
|
adamc@467
|
1324 return uw_Basis_sqlifyTime(ctx, *t);
|
adamc@467
|
1325 }
|
adamc@467
|
1326
|
adamc@311
|
1327 char *uw_Basis_ensqlBool(uw_Basis_bool b) {
|
adamc@311
|
1328 static uw_Basis_int true = 1;
|
adamc@311
|
1329 static uw_Basis_int false = 0;
|
adamc@282
|
1330
|
adamc@311
|
1331 if (b == uw_Basis_False)
|
adamc@282
|
1332 return (char *)&false;
|
adamc@282
|
1333 else
|
adamc@282
|
1334 return (char *)&true;
|
adamc@282
|
1335 }
|
adamc@284
|
1336
|
adamc@311
|
1337 uw_Basis_string uw_Basis_intToString(uw_context ctx, uw_Basis_int n) {
|
adamc@284
|
1338 int len;
|
adamc@284
|
1339 char *r;
|
adamc@284
|
1340
|
adamc@311
|
1341 uw_check_heap(ctx, INTS_MAX);
|
adamc@666
|
1342 r = ctx->heap.front;
|
adamc@284
|
1343 sprintf(r, "%lld%n", n, &len);
|
adamc@666
|
1344 ctx->heap.front += len+1;
|
adamc@284
|
1345 return r;
|
adamc@284
|
1346 }
|
adamc@285
|
1347
|
adamc@311
|
1348 uw_Basis_string uw_Basis_floatToString(uw_context ctx, uw_Basis_float n) {
|
adamc@285
|
1349 int len;
|
adamc@285
|
1350 char *r;
|
adamc@285
|
1351
|
adamc@311
|
1352 uw_check_heap(ctx, FLOATS_MAX);
|
adamc@666
|
1353 r = ctx->heap.front;
|
adamc@285
|
1354 sprintf(r, "%g%n", n, &len);
|
adamc@666
|
1355 ctx->heap.front += len+1;
|
adamc@285
|
1356 return r;
|
adamc@285
|
1357 }
|
adamc@285
|
1358
|
adamc@311
|
1359 uw_Basis_string uw_Basis_boolToString(uw_context ctx, uw_Basis_bool b) {
|
adamc@311
|
1360 if (b == uw_Basis_False)
|
adamc@285
|
1361 return "False";
|
adamc@285
|
1362 else
|
adamc@285
|
1363 return "True";
|
adamc@285
|
1364 }
|
adamc@288
|
1365
|
adamc@436
|
1366 uw_Basis_string uw_Basis_timeToString(uw_context ctx, uw_Basis_time t) {
|
adamc@436
|
1367 size_t len;
|
adamc@436
|
1368 char *r;
|
adamc@436
|
1369 struct tm stm;
|
adamc@436
|
1370
|
adamc@436
|
1371 if (localtime_r(&t, &stm)) {
|
adamc@436
|
1372 uw_check_heap(ctx, TIMES_MAX);
|
adamc@666
|
1373 r = ctx->heap.front;
|
adamc@436
|
1374 len = strftime(r, TIMES_MAX, TIME_FMT, &stm);
|
adamc@666
|
1375 ctx->heap.front += len+1;
|
adamc@436
|
1376 return r;
|
adamc@436
|
1377 } else
|
adamc@436
|
1378 return "<Invalid time>";
|
adamc@436
|
1379 }
|
adamc@288
|
1380
|
adamc@311
|
1381 uw_Basis_int *uw_Basis_stringToInt(uw_context ctx, uw_Basis_string s) {
|
adamc@288
|
1382 char *endptr;
|
adamc@311
|
1383 uw_Basis_int n = strtoll(s, &endptr, 10);
|
adamc@288
|
1384
|
adamc@288
|
1385 if (*s != '\0' && *endptr == '\0') {
|
adamc@311
|
1386 uw_Basis_int *r = uw_malloc(ctx, sizeof(uw_Basis_int));
|
adamc@288
|
1387 *r = n;
|
adamc@288
|
1388 return r;
|
adamc@288
|
1389 } else
|
adamc@288
|
1390 return NULL;
|
adamc@288
|
1391 }
|
adamc@289
|
1392
|
adamc@311
|
1393 uw_Basis_float *uw_Basis_stringToFloat(uw_context ctx, uw_Basis_string s) {
|
adamc@289
|
1394 char *endptr;
|
adamc@311
|
1395 uw_Basis_float n = strtod(s, &endptr);
|
adamc@289
|
1396
|
adamc@289
|
1397 if (*s != '\0' && *endptr == '\0') {
|
adamc@311
|
1398 uw_Basis_float *r = uw_malloc(ctx, sizeof(uw_Basis_float));
|
adamc@289
|
1399 *r = n;
|
adamc@289
|
1400 return r;
|
adamc@289
|
1401 } else
|
adamc@289
|
1402 return NULL;
|
adamc@289
|
1403 }
|
adamc@289
|
1404
|
adamc@311
|
1405 uw_Basis_bool *uw_Basis_stringToBool(uw_context ctx, uw_Basis_string s) {
|
adamc@311
|
1406 static uw_Basis_bool true = uw_Basis_True;
|
adamc@311
|
1407 static uw_Basis_bool false = uw_Basis_False;
|
adamc@289
|
1408
|
adamc@289
|
1409 if (!strcasecmp (s, "True"))
|
adamc@289
|
1410 return &true;
|
adamc@289
|
1411 else if (!strcasecmp (s, "False"))
|
adamc@289
|
1412 return &false;
|
adamc@289
|
1413 else
|
adamc@289
|
1414 return NULL;
|
adamc@289
|
1415 }
|
adamc@292
|
1416
|
adamc@436
|
1417 uw_Basis_time *uw_Basis_stringToTime(uw_context ctx, uw_Basis_string s) {
|
adamc@438
|
1418 char *dot = strchr(s, '.'), *end = strchr(s, 0);
|
adamc@436
|
1419 struct tm stm;
|
adamc@436
|
1420
|
adamc@439
|
1421 if (dot) {
|
adamc@439
|
1422 *dot = 0;
|
adamc@439
|
1423 if (strptime(s, TIME_FMT_PG, &stm) == end) {
|
adamc@439
|
1424 *dot = '.';
|
adamc@439
|
1425 uw_Basis_time *r = uw_malloc(ctx, sizeof(uw_Basis_time));
|
adamc@439
|
1426 *r = mktime(&stm);
|
adamc@439
|
1427 return r;
|
adamc@439
|
1428 }
|
adamc@439
|
1429 else {
|
adamc@439
|
1430 *dot = '.';
|
adamc@439
|
1431 return NULL;
|
adamc@439
|
1432 }
|
adamc@436
|
1433 }
|
adamc@439
|
1434 else {
|
adamc@439
|
1435 if (strptime(s, TIME_FMT_PG, &stm) == end) {
|
adamc@439
|
1436 uw_Basis_time *r = uw_malloc(ctx, sizeof(uw_Basis_time));
|
adamc@439
|
1437 *r = mktime(&stm);
|
adamc@439
|
1438 return r;
|
adamc@439
|
1439 }
|
adamc@439
|
1440 else if (strptime(s, TIME_FMT, &stm) == end) {
|
adamc@439
|
1441 uw_Basis_time *r = uw_malloc(ctx, sizeof(uw_Basis_time));
|
adamc@439
|
1442 *r = mktime(&stm);
|
adamc@439
|
1443 return r;
|
adamc@439
|
1444 }
|
adamc@439
|
1445 else
|
adamc@439
|
1446 return NULL;
|
adamc@439
|
1447 }
|
adamc@436
|
1448 }
|
adamc@436
|
1449
|
adamc@311
|
1450 uw_Basis_int uw_Basis_stringToInt_error(uw_context ctx, uw_Basis_string s) {
|
adamc@292
|
1451 char *endptr;
|
adamc@311
|
1452 uw_Basis_int n = strtoll(s, &endptr, 10);
|
adamc@292
|
1453
|
adamc@292
|
1454 if (*s != '\0' && *endptr == '\0')
|
adamc@292
|
1455 return n;
|
adamc@292
|
1456 else
|
adamc@311
|
1457 uw_error(ctx, FATAL, "Can't parse int: %s", s);
|
adamc@292
|
1458 }
|
adamc@293
|
1459
|
adamc@311
|
1460 uw_Basis_float uw_Basis_stringToFloat_error(uw_context ctx, uw_Basis_string s) {
|
adamc@293
|
1461 char *endptr;
|
adamc@311
|
1462 uw_Basis_float n = strtod(s, &endptr);
|
adamc@293
|
1463
|
adamc@293
|
1464 if (*s != '\0' && *endptr == '\0')
|
adamc@293
|
1465 return n;
|
adamc@293
|
1466 else
|
adamc@311
|
1467 uw_error(ctx, FATAL, "Can't parse float: %s", s);
|
adamc@293
|
1468 }
|
adamc@293
|
1469
|
adamc@311
|
1470 uw_Basis_bool uw_Basis_stringToBool_error(uw_context ctx, uw_Basis_string s) {
|
adamc@296
|
1471 if (!strcasecmp(s, "T") || !strcasecmp (s, "True"))
|
adamc@311
|
1472 return uw_Basis_True;
|
adamc@296
|
1473 else if (!strcasecmp(s, "F") || !strcasecmp (s, "False"))
|
adamc@311
|
1474 return uw_Basis_False;
|
adamc@293
|
1475 else
|
adamc@311
|
1476 uw_error(ctx, FATAL, "Can't parse bool: %s", s);
|
adamc@293
|
1477 }
|
adamc@436
|
1478
|
adamc@436
|
1479 uw_Basis_time uw_Basis_stringToTime_error(uw_context ctx, uw_Basis_string s) {
|
adamc@438
|
1480 char *dot = strchr(s, '.'), *end = strchr(s, 0);
|
adamc@436
|
1481 struct tm stm = {};
|
adamc@436
|
1482
|
adamc@438
|
1483 if (dot) {
|
adamc@438
|
1484 *dot = 0;
|
adamc@438
|
1485 if (strptime(s, TIME_FMT_PG, &stm)) {
|
adamc@438
|
1486 *dot = '.';
|
adamc@438
|
1487 return mktime(&stm);
|
adamc@438
|
1488 }
|
adamc@438
|
1489 else {
|
adamc@438
|
1490 *dot = '.';
|
adamc@438
|
1491 uw_error(ctx, FATAL, "Can't parse time: %s", s);
|
adamc@438
|
1492 }
|
adamc@438
|
1493 }
|
adamc@438
|
1494 else {
|
adamc@439
|
1495 if (strptime(s, TIME_FMT_PG, &stm) == end)
|
adamc@439
|
1496 return mktime(&stm);
|
adamc@439
|
1497 else if (strptime(s, TIME_FMT, &stm) == end)
|
adamc@438
|
1498 return mktime(&stm);
|
adamc@438
|
1499 else
|
adamc@438
|
1500 uw_error(ctx, FATAL, "Can't parse time: %s", s);
|
adamc@438
|
1501 }
|
adamc@436
|
1502 }
|
adamc@457
|
1503
|
adamc@457
|
1504 uw_Basis_string uw_Basis_requestHeader(uw_context ctx, uw_Basis_string h) {
|
adamc@457
|
1505 int len = strlen(h);
|
adamc@457
|
1506 char *s = ctx->headers, *p;
|
adamc@457
|
1507
|
adamc@457
|
1508 while (p = strchr(s, ':')) {
|
adamc@457
|
1509 if (p - s == len && !strncasecmp(s, h, len)) {
|
adamc@458
|
1510 return p + 2;
|
adamc@457
|
1511 } else {
|
adamc@458
|
1512 if ((s = strchr(p, 0)) && s < ctx->headers_end)
|
adamc@458
|
1513 s += 2;
|
adamc@457
|
1514 else
|
adamc@457
|
1515 return NULL;
|
adamc@457
|
1516 }
|
adamc@457
|
1517 }
|
adamc@463
|
1518 }
|
adamc@458
|
1519
|
adamc@463
|
1520 uw_Basis_string uw_Basis_get_cookie(uw_context ctx, uw_Basis_string c) {
|
adamc@463
|
1521 int len = strlen(c);
|
adamc@666
|
1522 char *s = ctx->headers, *p = ctx->outHeaders.start;
|
adamc@474
|
1523
|
adamc@474
|
1524 while (p = strstr(p, "\nSet-Cookie: ")) {
|
adamc@474
|
1525 char *p2;
|
adamc@474
|
1526 p += 13;
|
adamc@474
|
1527 p2 = strchr(p, '=');
|
adamc@474
|
1528
|
adamc@474
|
1529 if (p2) {
|
adamc@474
|
1530 size_t sz = strcspn(p2+1, ";\r\n");
|
adamc@474
|
1531
|
adamc@474
|
1532 if (!strncasecmp(p, c, p2 - p)) {
|
adamc@474
|
1533 char *ret = uw_malloc(ctx, sz + 1);
|
adamc@474
|
1534 memcpy(ret, p2+1, sz);
|
adamc@474
|
1535 ret[sz] = 0;
|
adamc@474
|
1536 return ret;
|
adamc@474
|
1537 }
|
adamc@474
|
1538 }
|
adamc@474
|
1539 }
|
adamc@463
|
1540
|
adamc@463
|
1541 while (p = strchr(s, ':')) {
|
adamc@476
|
1542 if (!strncasecmp(s, "Cookie: ", 8)) {
|
adamc@476
|
1543 p += 2;
|
adamc@476
|
1544 while (1) {
|
adamc@476
|
1545 if (!strncmp(p, c, len)
|
adamc@476
|
1546 && p + len < ctx->headers_end && p[len] == '=')
|
adamc@476
|
1547 return p + 1 + len;
|
adamc@476
|
1548 else if (p = strchr(p, ';'))
|
adamc@476
|
1549 p += 2;
|
adamc@476
|
1550 else if ((s = strchr(s, 0)) && s < ctx->headers_end) {
|
adamc@476
|
1551 s += 2;
|
adamc@476
|
1552 break;
|
adamc@476
|
1553 }
|
adamc@476
|
1554 else
|
adamc@476
|
1555 return NULL;
|
adamc@476
|
1556 }
|
adamc@463
|
1557 } else {
|
adamc@463
|
1558 if ((s = strchr(p, 0)) && s < ctx->headers_end)
|
adamc@463
|
1559 s += 2;
|
adamc@463
|
1560 else
|
adamc@463
|
1561 return NULL;
|
adamc@463
|
1562 }
|
adamc@463
|
1563 }
|
adamc@457
|
1564 }
|
adamc@462
|
1565
|
adamc@466
|
1566 uw_unit uw_Basis_set_cookie(uw_context ctx, uw_Basis_string prefix, uw_Basis_string c, uw_Basis_string v) {
|
adamc@462
|
1567 uw_write_header(ctx, "Set-Cookie: ");
|
adamc@462
|
1568 uw_write_header(ctx, c);
|
adamc@462
|
1569 uw_write_header(ctx, "=");
|
adamc@462
|
1570 uw_write_header(ctx, v);
|
adamc@466
|
1571 uw_write_header(ctx, "; path=");
|
adamc@466
|
1572 uw_write_header(ctx, prefix);
|
adamc@462
|
1573 uw_write_header(ctx, "\r\n");
|
adamc@462
|
1574
|
adamc@462
|
1575 return uw_unit_v;
|
adamc@462
|
1576 }
|
adamc@667
|
1577
|
adamc@667
|
1578
|