]> jfr.im git - irc/quakenet/newserv.git/blame - nterfacer/nterfacer.c
CHANSERV: tell user when they can't attempts to auth any more, and drop max attempts...
[irc/quakenet/newserv.git] / nterfacer / nterfacer.c
CommitLineData
18f8bd28
CP
1/*
2 nterfacer
d4897ed9 3 Copyright (C) 2004-2007 Chris Porter.
18f8bd28
CP
4*/
5
6#include <stdio.h>
7#include <stdlib.h>
8#include <stdarg.h>
9#include <sys/poll.h>
10#include <sys/types.h>
11#include <sys/socket.h>
12#include <errno.h>
13#include <unistd.h>
14#include <sys/ioctl.h>
15#include <netdb.h>
16#include <string.h>
9eed0e10 17#include <strings.h>
9c312e03
CP
18#include <netinet/in.h>
19#include <arpa/inet.h>
18f8bd28
CP
20
21#include "../lib/sstring.h"
22#include "../lib/irc_string.h"
23#include "../core/config.h"
24#include "../core/events.h"
87698d77 25#include "../lib/version.h"
dae6d3d7 26#include "../core/schedule.h"
18f8bd28
CP
27
28#include "nterfacer.h"
29#include "logging.h"
30
80d91dd6 31MODULE_VERSION("1.1p" PROTOCOL_VERSION);
87698d77 32
18f8bd28
CP
33struct service_node *tree = NULL;
34struct esocket_events nterfacer_events;
35struct esocket *nterfacer_sock;
36struct rline *rlines = NULL;
37unsigned short nterfacer_token = BLANK_TOKEN;
38struct nterface_auto_log *nrl;
39
c86a0114 40struct service_node *ping = NULL;
4e4920a4
CP
41int accept_fd = -1;
42struct permitted *permits;
43int permit_count = 0;
6cf2d6a5
CP
44
45int ping_handler(struct rline *ri, int argc, char **argv);
2a6be5fe 46static void nterfacer_sendcallback(struct rline *ri, int error, char *buf);
6cf2d6a5 47
18f8bd28
CP
48void _init(void) {
49 int loaded;
50 int debug_mode = getcopyconfigitemintpositive("nterfacer", "debug", 0);
51
52 nrl = nterface_open_log("nterfacer", "logs/nterfacer.log", debug_mode);
53
54 loaded = load_permits();
2a6be5fe
CP
55 nterface_log(nrl, NL_INFO, "Loaded %d permit%s successfully.", loaded, loaded==1?"":"s");
56
57 if(!loaded)
18f8bd28 58 return;
18f8bd28
CP
59
60 nterfacer_events.on_accept = nterfacer_accept_event;
61 nterfacer_events.on_line = nterfacer_line_event;
62 nterfacer_events.on_disconnect = NULL;
63
64 nterfacer_token = esocket_token();
65
6cf2d6a5
CP
66 ping = register_service("nterfacer");
67 if(!ping) {
68 MemError();
69 } else {
70 register_handler(ping, "ping", 0, ping_handler);
71 }
72
18f8bd28
CP
73 accept_fd = setup_listening_socket();
74 if(accept_fd == -1) {
75 nterface_log(nrl, NL_ERROR, "Unable to setup listening socket!");
76 } else {
77 nterfacer_sock = esocket_add(accept_fd, ESOCKET_UNIX_DOMAIN, &nterfacer_events, nterfacer_token);
78 }
79
80 /* the main unix domain socket must NOT have a disconnect event. */
81 nterfacer_events.on_disconnect = nterfacer_disconnect_event;
82}
83
504f62c7 84void free_handler(struct handler *hp) {
f435bd22
CP
85 struct rline *li, *pi = NULL;
86
87 for(li=rlines;li;) {
88 if(li->handler == hp) {
89 if(li->socket) {
90 esocket_write_line(li->socket, "%d,OE%d,%s", li->id, BF_UNLOADED, "Service was unloaded.");
91 } else if(li->callback) {
2a6be5fe 92 nterfacer_sendcallback(li, BF_UNLOADED, "Service was unloaded.");
f435bd22
CP
93 }
94 if(pi) {
95 pi->next = li->next;
96 ntfree(li);
97 li = pi->next;
98 } else {
99 rlines = li->next;
100 ntfree(li);
101 li = rlines;
102 }
103 } else {
104 pi=li,li=li->next;
105 }
106 }
107
504f62c7 108 freesstring(hp->command);
5ab07198 109 ntfree(hp);
504f62c7
CP
110}
111
18f8bd28
CP
112void free_handlers(struct service_node *tp) {
113 struct handler *hp, *lp;
114
115 for(hp=tp->handlers;hp;) {
116 lp = hp;
117 hp = hp->next;
504f62c7 118 free_handler(lp);
18f8bd28
CP
119 }
120
121 tp->handlers = NULL;
122}
123
124void _fini(void) {
125 struct service_node *tp, *lp;
126 int i;
127
6cf2d6a5
CP
128 if(ping)
129 deregister_service(ping);
130
18f8bd28
CP
131 for(tp=tree;tp;) {
132 lp = tp;
133 tp = tp->next;
134 free_handlers(lp);
5ab07198 135 ntfree(lp);
18f8bd28
CP
136 }
137 tree = NULL;
138
139 if((accept_fd != -1) && nterfacer_sock) {
140 esocket_clean_by_token(nterfacer_token);
141 nterfacer_sock = NULL;
142 accept_fd = -1;
143 }
144
145 if(permits && permit_count) {
146 for(i=0;i<permit_count;i++) {
147 freesstring(permits[i].hostname);
148 freesstring(permits[i].password);
149 }
5ab07198 150 ntfree(permits);
18f8bd28
CP
151 permit_count = 0;
152 permits = NULL;
153 }
154
155 nrl = nterface_close_log(nrl);
f8d5cfcf 156 nscheckfreeall(POOL_NTERFACER);
18f8bd28
CP
157}
158
159int load_permits(void) {
8238d9c1 160 int loaded_lines = 0, i, j;
18f8bd28
CP
161 struct permitted *new_permits, *resized, *item;
162 struct hostent *host;
8238d9c1
CP
163 array *hostnamesa, *passwordsa;
164 sstring **hostnames, **passwords;
18f8bd28 165
8238d9c1
CP
166 hostnamesa = getconfigitems("nterfacer", "hostname");
167 passwordsa = getconfigitems("nterfacer", "password");
2a6be5fe 168 if(!hostnamesa || !passwordsa)
18f8bd28 169 return 0;
8238d9c1
CP
170 if(hostnamesa->cursi != passwordsa->cursi) {
171 nterface_log(nrl, NL_ERROR, "Different number of hostnames/passwords in config file.");
172 return 0;
173 }
174
175 hostnames = (sstring **)hostnamesa->content;
176 passwords = (sstring **)passwordsa->content;
18f8bd28 177
5ab07198
CP
178 new_permits = ntmalloc(hostnamesa->cursi * sizeof(struct permitted));
179 memset(new_permits, 0, hostnamesa->cursi * sizeof(struct permitted));
18f8bd28
CP
180 item = new_permits;
181
8238d9c1
CP
182 for(i=0;i<hostnamesa->cursi;i++) {
183 item->hostname = getsstring(hostnames[i]->content, hostnames[i]->length);
18f8bd28
CP
184
185 host = gethostbyname(item->hostname->content);
186 if (!host) {
8238d9c1 187 nterface_log(nrl, NL_WARNING, "Couldn't resolve hostname: %s (item %d).", item->hostname->content, i + 1);
18f8bd28
CP
188 freesstring(item->hostname);
189 continue;
190 }
191
00c5dbcb 192 item->ihost = (*(struct in_addr *)host->h_addr_list[0]).s_addr;
18f8bd28
CP
193 for(j=0;j<loaded_lines;j++) {
194 if(new_permits[j].ihost == item->ihost) {
8238d9c1 195 nterface_log(nrl, NL_WARNING, "Host with items %d and %d is identical, dropping item %d.", j + 1, i + 1, i + 1);
18f8bd28
CP
196 host = NULL;
197 }
198 }
199
200 if(!host) {
201 freesstring(item->hostname);
202 continue;
203 }
204
8238d9c1 205 item->password = getsstring(passwords[i]->content, passwords[i]->length);
18f8bd28
CP
206 nterface_log(nrl, NL_DEBUG, "Loaded permit, hostname: %s.", item->hostname->content);
207
208 item++;
209 loaded_lines++;
210 }
211
212 if(!loaded_lines) {
5ab07198 213 ntfree(new_permits);
18f8bd28
CP
214 return 0;
215 }
216
5ab07198 217 resized = ntrealloc(new_permits, sizeof(struct permitted) * loaded_lines);
18f8bd28
CP
218 if(!resized) {
219 MemError();
5ab07198 220 ntfree(new_permits);
18f8bd28
CP
221 return 0;
222 }
223 permits = resized;
224 permit_count = loaded_lines;
225
226 return permit_count;
227}
228
229int setup_listening_socket(void) {
230 struct sockaddr_in sin;
231 int fd;
232 unsigned int opt = 1;
233
234 fd = socket(AF_INET, SOCK_STREAM, 0);
235
236 /* also shamelessly ripped from proxyscan */
237 if(fd == -1) {
238 nterface_log(nrl, NL_ERROR, "Unable to open listening socket (%d).", errno);
239 return -1;
240 }
241
242 if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (const char *) &opt, sizeof(opt)) != 0) {
243 nterface_log(nrl, NL_ERROR, "Unable to set SO_REUSEADDR on listen socket.");
244 return -1;
245 }
246
247 /* Initialiase the addresses */
248 memset(&sin, 0, sizeof(sin));
249 sin.sin_family = AF_INET;
084e76d8 250 sin.sin_port = htons(getcopyconfigitemintpositive("nterfacer", "port", NTERFACER_PORT));
18f8bd28
CP
251
252 if(bind(fd, (struct sockaddr *) &sin, sizeof(sin))) {
253 nterface_log(nrl, NL_ERROR, "Unable to bind listen socket (%d).", errno);
254 return -1;
255 }
256
257 listen(fd, 5);
258
97eb3b6f 259 if(ioctl(fd, FIONBIO, &opt)) {
18f8bd28
CP
260 nterface_log(nrl, NL_ERROR, "Unable to set listen socket non-blocking.");
261 return -1;
262 }
263
264 return fd;
265}
266
267struct service_node *register_service(char *name) {
5ab07198 268 struct service_node *np = ntmalloc(sizeof(service_node));
18f8bd28
CP
269 MemCheckR(np, NULL);
270
271 np->name = getsstring(name, strlen(name));
272 if(!np->name) {
273 MemError();
5ab07198 274 ntfree(np);
18f8bd28
CP
275 return NULL;
276 }
277
278 np->handlers = NULL;
279 np->next = tree;
280 tree = np;
281
282 return np;
283}
284
285struct handler *register_handler(struct service_node *service, char *command, int args, handler_function fp) {
5ab07198 286 struct handler *hp = ntmalloc(sizeof(handler));
18f8bd28
CP
287 MemCheckR(hp, NULL);
288
289 hp->command = getsstring(command, strlen(command));
290 if(!hp->command) {
291 MemError();
5ab07198 292 ntfree(hp);
18f8bd28
CP
293 return NULL;
294 }
295
a690fb9f 296 hp->function = fp;
18f8bd28
CP
297 hp->args = args;
298
299 hp->next = service->handlers;
504f62c7 300 hp->service = service;
18f8bd28
CP
301 service->handlers = hp;
302
303 return hp;
304}
305
504f62c7
CP
306void deregister_handler(struct handler *hl) {
307 struct service_node *service = (struct service_node *)hl->service;
308 struct handler *np, *lp = NULL;
309 for(np=service->handlers;np;lp=np,np=np->next) {
310 if(hl == np) {
311 if(lp) {
312 lp->next = np->next;
313 } else {
314 service->handlers = np->next;
315 }
316 free_handler(np);
317 return;
318 }
319 }
320}
321
18f8bd28
CP
322void deregister_service(struct service_node *service) {
323 struct service_node *sp, *lp = NULL;
18f8bd28
CP
324
325 for(sp=tree;sp;lp=sp,sp=sp->next) {
326 if(sp == service) {
327 if(lp) {
328 lp->next = sp->next;
329 } else {
330 tree = sp->next;
331 }
332 break;
333 }
334 }
335
336 if(!sp) /* already freed */
337 return;
338
339 free_handlers(service);
340
18f8bd28
CP
341 freesstring(service->name);
342
5ab07198 343 ntfree(service);
18f8bd28
CP
344}
345
346void nterfacer_accept_event(struct esocket *socket) {
347 struct sockaddr_in sin;
348 unsigned int addrsize = sizeof(sin);
349 int newfd = accept(socket->fd, (struct sockaddr *)&sin, &addrsize), i;
350 struct sconnect *temp;
351 struct permitted *item = NULL;
352 struct esocket *newsocket;
084e76d8 353 unsigned int opt = 1;
18f8bd28
CP
354
355 if(newfd == -1) {
356 nterface_log(nrl, NL_WARNING, "Unable to accept nterfacer fd!");
357 return;
358 }
359
97eb3b6f
CP
360 if(ioctl(newfd, FIONBIO, &opt)) {
361 nterface_log(nrl, NL_ERROR, "Unable to set accepted socket non-blocking.");
3f15de97 362 return;
084e76d8
CP
363 }
364
18f8bd28
CP
365 for(i=0;i<permit_count;i++) {
366 if(permits[i].ihost == sin.sin_addr.s_addr) {
367 item = &permits[i];
368 break;
369 }
370 }
371
372 if(!item) {
9c312e03 373 nterface_log(nrl, NL_INFO, "Unauthorised connection from %s closed", inet_ntoa(sin.sin_addr));
18f8bd28
CP
374 close(newfd);
375 return;
376 }
377
5ab07198 378 temp = (struct sconnect *)ntmalloc(sizeof(struct sconnect));
18f8bd28
CP
379 if(!temp) {
380 MemError();
381 close(newfd);
382 return;
383 }
384
385 /* do checks on hostname first */
386
387 newsocket = esocket_add(newfd, ESOCKET_UNIX_DOMAIN_CONNECTED, &nterfacer_events, nterfacer_token);
388 if(!newsocket) {
5ab07198 389 ntfree(temp);
18f8bd28
CP
390 close(newfd);
391 return;
392 }
393 newsocket->tag = temp;
394
395 nterface_log(nrl, NL_INFO, "New connection from %s.", item->hostname->content);
396
397 temp->status = SS_IDLE;
398 temp->permit = item;
399
400 esocket_write_line(newsocket, "nterfacer " PROTOCOL_VERSION);
401}
402
80d91dd6 403void derive_key(unsigned char *out, char *password, char *segment, unsigned char *noncea, unsigned char *nonceb, unsigned char *extra, int extralen) {
1ac1d43c
CP
404 SHA256_CTX c;
405 SHA256_Init(&c);
406 SHA256_Update(&c, (unsigned char *)password, strlen(password));
407 SHA256_Update(&c, (unsigned char *)":", 1);
408 SHA256_Update(&c, (unsigned char *)segment, strlen(segment));
409 SHA256_Update(&c, (unsigned char *)":", 1);
410 SHA256_Update(&c, noncea, 16);
411 SHA256_Update(&c, (unsigned char *)":", 1);
412 SHA256_Update(&c, nonceb, 16);
80d91dd6
CP
413 SHA256_Update(&c, (unsigned char *)":", 1);
414 SHA256_Update(&c, extra, extralen);
1ac1d43c
CP
415 SHA256_Final(out, &c);
416
417 SHA256_Init(&c);
418 SHA256_Update(&c, out, 32);
419 SHA256_Final(out, &c);
420}
421
18f8bd28
CP
422int nterfacer_line_event(struct esocket *sock, char *newline) {
423 struct sconnect *socket = sock->tag;
1ac1d43c
CP
424 char *response, *theirnonceh = NULL, *theirivh = NULL;
425 unsigned char theirnonce[16], theiriv[16];
18f8bd28
CP
426 int number, reason;
427
428 switch(socket->status) {
429 case SS_IDLE:
430 if(strcasecmp(newline, ANTI_FULL_VERSION)) {
431 nterface_log(nrl, NL_INFO, "Protocol mismatch from %s: %s", socket->permit->hostname->content, newline);
432 return 1;
433 } else {
1ac1d43c
CP
434 unsigned char challenge[32];
435 char ivhex[16 * 2 + 1], noncehex[16 * 2 + 1];
18f8bd28 436
1ac1d43c
CP
437 if(!get_entropy(challenge, 32) || !get_entropy(socket->iv, 16)) {
438 nterface_log(nrl, NL_ERROR, "Unable to open challenge/IV entropy bin!");
18f8bd28
CP
439 return 1;
440 }
441
1ac1d43c
CP
442 int_to_hex(challenge, socket->challenge, 32);
443 int_to_hex(socket->iv, ivhex, 16);
444
445 memcpy(socket->response, challenge_response(socket->challenge, socket->permit->password->content), sizeof(socket->response));
18f8bd28
CP
446 socket->response[sizeof(socket->response) - 1] = '\0'; /* just in case */
447
448 socket->status = SS_VERSIONED;
449 if(!generate_nonce(socket->ournonce, 1)) {
450 nterface_log(nrl, NL_ERROR, "Unable to generate nonce!");
451 return 1;
452 }
1ac1d43c 453 int_to_hex(socket->ournonce, noncehex, 16);
18f8bd28 454
1ac1d43c 455 if(esocket_write_line(sock, "%s %s %s", socket->challenge, ivhex, noncehex))
701500af
CP
456 return BUF_ERROR;
457 return 0;
18f8bd28
CP
458 }
459 break;
460 case SS_VERSIONED:
461 for(response=newline;*response;response++) {
462 if((*response == ' ') && (*(response + 1))) {
463 *response = '\0';
1ac1d43c 464 theirivh = response + 1;
18f8bd28
CP
465 break;
466 }
467 }
468
1ac1d43c
CP
469 if(theirivh) {
470 for(response=theirivh;*response;response++) {
471 if((*response == ' ') && (*(response + 1))) {
472 *response = '\0';
473 theirnonceh = response + 1;
474 break;
475 }
476 }
477 }
478
479 if(!theirivh || (strlen(theirivh) != 32) || !hex_to_int(theirivh, theiriv, sizeof(theiriv)) ||
480 !theirnonceh || (strlen(theirnonceh) != 32) || !hex_to_int(theirnonceh, theirnonce, sizeof(theirnonce))) {
18f8bd28
CP
481 nterface_log(nrl, NL_INFO, "Protocol error drop: %s", socket->permit->hostname->content);
482 return 1;
483 }
484
485 if(!memcmp(socket->ournonce, theirnonce, sizeof(theirnonce))) {
486 nterface_log(nrl, NL_INFO, "Bad nonce drop: %s", socket->permit->hostname->content);
487 return 1;
488 }
489
490 if(!strncasecmp(newline, socket->response, sizeof(socket->response))) {
80d91dd6
CP
491 unsigned char theirkey[32], ourkey[32];
492
e66c4243 493 derive_key(ourkey, socket->permit->password->content, socket->challenge, socket->ournonce, theirnonce, (unsigned char *)"SERVER", 6);
18f8bd28 494
e66c4243 495 derive_key(theirkey, socket->permit->password->content, socket->response, theirnonce, socket->ournonce, (unsigned char *)"CLIENT", 6);
18f8bd28
CP
496 nterface_log(nrl, NL_INFO, "Authed: %s", socket->permit->hostname->content);
497 socket->status = SS_AUTHENTICATED;
80d91dd6 498 switch_buffer_mode(sock, ourkey, socket->iv, theirkey, theiriv);
1ac1d43c 499
80d91dd6 500 if(esocket_write_line(sock, "Oauth"))
701500af 501 return BUF_ERROR;
18f8bd28
CP
502 } else {
503 nterface_log(nrl, NL_INFO, "Bad CR drop: %s", socket->permit->hostname->content);
504
505 return 1;
506 }
507 break;
508 case SS_AUTHENTICATED:
509 nterface_log(nrl, NL_INFO|NL_LOG_ONLY, "L(%s): %s", socket->permit->hostname->content, newline);
510 reason = nterfacer_new_rline(newline, sock, &number);
511 if(reason) {
5757e1ef
CP
512 if(reason == RE_SOCKET_ERROR)
513 return BUF_ERROR;
18f8bd28 514 if(reason != RE_BAD_LINE) {
a690fb9f
CP
515 if(esocket_write_line(sock, "%d,E%d,%s", number, reason, request_error(reason)))
516 return BUF_ERROR;
517 return 0;
18f8bd28
CP
518 } else {
519 return 1;
520 }
521 }
522 break;
523 }
524
525 return 0;
526}
527
528int nterfacer_new_rline(char *line, struct esocket *socket, int *number) {
1fbb1306 529 char *sp, *p, *parsebuf = NULL, *pp, commandbuf[MAX_BUFSIZE], *args[MAX_ARGS], *newp;
18f8bd28
CP
530 int argcount;
531 struct service_node *service;
532 struct rline *prequest;
533 struct handler *hl;
534 int re;
535
536 if(!line || !line[0] || (line[0] == ','))
537 return 0;
538
539 for(sp=line;*sp;sp++)
540 if(*sp == ',')
541 break;
542
543 if(!*sp || !*(sp + 1))
544 return RE_BAD_LINE;
545
546 *sp = '\0';
547
548 for(service=tree;service;service=service->next)
549 if(!strcmp(service->name->content, line))
550 break;
551
552 for(p=sp+1;*p;p++)
553 if(*p == ',')
554 break;
555
556 if(!*p || !(p + 1))
557 return RE_BAD_LINE;
558
559 *p = '\0';
560 *number = positive_atoi(sp + 1);
561
562 if((*number < 1) || (*number > 0xffff))
563 return RE_BAD_LINE;
564
565 if (!service) {
566 nterface_log(nrl, NL_DEBUG, "Unable to find service: %s", line);
567 return RE_SERVICER_NOT_FOUND;
568 }
569
570 newp = commandbuf;
571
572 for(pp=p+1;*pp;pp++) {
573 if((*pp == '\\') && *(pp + 1)) {
574 if(*(pp + 1) == ',') {
575 *newp++ = ',';
576 } else if(*(pp + 1) == '\\') {
577 *newp++ = '\\';
578 }
579 pp++;
580 } else if(*pp == ',') {
581 break;
582 } else {
583 *newp++ = *pp;
584 }
585 }
586
587 if(*pp == '\0') { /* if we're at the end already, we have no arguments */
588 argcount = 0;
589 } else {
590 argcount = 1; /* we have a comma, so we have one already */
591 }
592
593 *newp = '\0';
594
595 for(hl=service->handlers;hl;hl=hl->next)
596 if(!strncmp(hl->command->content, commandbuf, sizeof(commandbuf)))
597 break;
598
599 if(!hl)
600 return RE_COMMAND_NOT_FOUND;
601
602 if(argcount) {
5ab07198 603 parsebuf = (char *)ntmalloc(strlen(pp) + 1);
18f8bd28 604 MemCheckR(parsebuf, RE_MEM_ERROR);
18f8bd28
CP
605
606 for(newp=args[0]=parsebuf,pp++;*pp;pp++) {
607 if((*pp == '\\') && *(pp + 1)) {
608 if(*(pp + 1) == ',') {
609 *newp++ = ',';
610 } else if(*(pp + 1) == '\\') {
611 *newp++ = '\\';
612 }
613 pp++;
614 } else if(*pp == ',') {
615 *newp++ = '\0';
616 args[argcount++] = newp;
617 if(argcount > MAX_ARGS) {
5ab07198 618 ntfree(parsebuf);
18f8bd28
CP
619 return RE_TOO_MANY_ARGS;
620 }
621 } else {
622 *newp++ = *pp;
623 }
624 }
625 *newp = '\0';
626 }
627 if(argcount < hl->args) {
1fbb1306 628 if(argcount && parsebuf)
5ab07198 629 ntfree(parsebuf);
18f8bd28
CP
630 return RE_WRONG_ARG_COUNT;
631 }
632
5ab07198 633 prequest = (struct rline *)ntmalloc(sizeof(struct rline));
18f8bd28
CP
634 if(!prequest) {
635 MemError();
1fbb1306 636 if(argcount && parsebuf)
5ab07198 637 ntfree(parsebuf);
18f8bd28
CP
638 return RE_MEM_ERROR;
639 }
640
641 prequest->service = service;
642 prequest->handler = hl;
95ee3dac 643 prequest->buf[0] = '\0';
18f8bd28
CP
644 prequest->curpos = prequest->buf;
645 prequest->tag = NULL;
646 prequest->id = *number;
647 prequest->next = rlines;
648 prequest->socket = socket;
e5a7c318 649 prequest->callback = NULL;
18f8bd28
CP
650
651 rlines = prequest;
a690fb9f 652 re = (hl->function)(prequest, argcount, args);
18f8bd28 653
1fbb1306 654 if(argcount && parsebuf)
5ab07198 655 ntfree(parsebuf);
18f8bd28
CP
656
657 return re;
658}
659
660void nterfacer_disconnect_event(struct esocket *sock) {
661 struct sconnect *socket = sock->tag;
662 struct rline *li;
663 /* not tested */
664
665 nterface_log(nrl, NL_INFO, "Disconnected from %s.", socket->permit->hostname->content);
666
667 /* not tested */
668 for(li=rlines;li;li=li->next)
2679891b 669 if(li->socket && (li->socket->tag == socket))
18f8bd28
CP
670 li->socket = NULL;
671
5ab07198 672 ntfree(socket);
18f8bd28
CP
673}
674
675int ri_append(struct rline *li, char *format, ...) {
676 char buf[MAX_BUFSIZE], escapedbuf[MAX_BUFSIZE * 2 + 1], *p, *tp;
677 int sizeleft = sizeof(li->buf) - (li->curpos - li->buf);
678 va_list ap;
679
680 va_start(ap, format);
681
682 if(vsnprintf(buf, sizeof(buf), format, ap) >= sizeof(buf)) {
683 va_end(ap);
684 return BF_OVER;
685 }
686
687 va_end(ap);
688
689 for(tp=escapedbuf,p=buf;*p||(*tp='\0');*tp++=*p++)
690 if((*p == ',') || (*p == '\\'))
691 *tp++ = '\\';
692
693 if(sizeleft > 0) {
694 if(li->curpos == li->buf) {
695 li->curpos+=snprintf(li->curpos, sizeleft, "%s", escapedbuf);
696 } else {
697 li->curpos+=snprintf(li->curpos, sizeleft, ",%s", escapedbuf);
698 }
699 }
700
701 if(sizeof(li->buf) - (li->curpos - li->buf) > 0) {
702 return BF_OK;
703 } else {
704 return BF_OVER;
705 }
706}
707
708int ri_error(struct rline *li, int error_code, char *format, ...) {
709 char buf[MAX_BUFSIZE], escapedbuf[MAX_BUFSIZE * 2 + 1], *p, *tp;
710 struct rline *pp, *lp = NULL;
711 va_list ap;
a690fb9f 712 int retval = RE_OK;
18f8bd28 713
e5a7c318 714 if(li->socket || li->callback) {
18f8bd28
CP
715 va_start(ap, format);
716 vsnprintf(buf, sizeof(buf), format, ap);
717 va_end(ap);
718
719 for(tp=escapedbuf,p=buf;*p||(*tp='\0');*tp++=*p++)
720 if((*p == ',') || (*p == '\\'))
721 *tp++ = '\\';
e5a7c318
CP
722 if(li->socket) {
723 if(esocket_write_line(li->socket, "%d,OE%d,%s", li->id, error_code, escapedbuf))
724 retval = RE_SOCKET_ERROR;
725 } else {
726 if(error_code == 0) /* :P */
727 error_code = -10000;
18f8bd28 728
2a6be5fe 729 nterfacer_sendcallback(li, error_code, escapedbuf);
e5a7c318 730 }
18f8bd28
CP
731 }
732
733 for(pp=rlines;pp;lp=pp,pp=pp->next) {
734 if(pp == li) {
735 if(lp) {
736 lp->next = li->next;
737 } else {
738 rlines = li->next;
739 }
5ab07198 740 ntfree(li);
18f8bd28
CP
741 break;
742 }
743 }
744
a690fb9f 745 return retval;
18f8bd28
CP
746}
747
748int ri_final(struct rline *li) {
749 struct rline *pp, *lp = NULL;
5757e1ef 750 int retval = RE_OK;
18f8bd28 751
e5a7c318 752 if(li->socket) {
5757e1ef
CP
753 if(esocket_write_line(li->socket, "%d,OO%s", li->id, li->buf))
754 retval = RE_SOCKET_ERROR;
e5a7c318 755 } else if(li->callback) {
2a6be5fe 756 nterfacer_sendcallback(li, 0, li->buf);
e5a7c318 757 }
5757e1ef 758
18f8bd28
CP
759 for(pp=rlines;pp;lp=pp,pp=pp->next) {
760 if(pp == li) {
761 if(lp) {
762 lp->next = li->next;
763 } else {
764 rlines = li->next;
765 }
5ab07198 766 ntfree(li);
18f8bd28
CP
767 break;
768 }
769 }
770
5757e1ef 771 return retval;
18f8bd28 772}
6cf2d6a5
CP
773
774int ping_handler(struct rline *ri, int argc, char **argv) {
775 ri_append(ri, "OK");
776 return ri_final(ri);
777}
e5a7c318 778
dae6d3d7
CP
779struct sched_rline {
780 rline rl;
781 int argc;
782 handler *hl;
783 void *schedule;
784 char argv[];
785};
786
787static const int XMAXARGS = 50;
788
789static void execrline(void *arg) {
790 struct sched_rline *sr = arg;
791 int re, i;
792 char *argv[XMAXARGS], *buf;
793
794 sr->schedule = NULL;
795
796 buf = sr->argv;
797 for(i=0;i<sr->argc;i++) {
dae6d3d7
CP
798 argv[i] = buf;
799 buf+=strlen(buf) + 1;
800 }
801
802 re = (sr->hl->function)(&sr->rl, sr->argc, argv);
803
804 if(re)
805 Error("nterfacer", ERR_WARNING, "sendline: error occured calling %p %d: %s", sr->hl->function, re, request_error(re));
806}
807
e5a7c318
CP
808void *nterfacer_sendline(char *service, char *command, int argc, char **argv, rline_callback callback, void *tag) {
809 struct service_node *servicep;
810 struct rline *prequest;
dae6d3d7 811 struct sched_rline *sr;
e5a7c318 812 struct handler *hl;
dae6d3d7
CP
813 int totallen, i;
814 char *buf;
e5a7c318
CP
815
816 for(servicep=tree;servicep;servicep=servicep->next)
817 if(!strcmp(servicep->name->content, service))
818 break;
819
dae6d3d7
CP
820 if(argc > XMAXARGS)
821 Error("nterfacer", ERR_STOP, "Over maximum arguments.");
822
e5a7c318
CP
823 if(!servicep) {
824 Error("nterfacer", ERR_WARNING, "sendline: service not found: %s", service);
825 return NULL;
826 }
827
828 for(hl=servicep->handlers;hl;hl=hl->next)
829 if(!strcmp(hl->command->content, command))
830 break;
831
832 if(!hl) {
833 Error("nterfacer", ERR_WARNING, "sendline: command not found: %s", command);
834 return NULL;
835 }
836
837 if(argc < hl->args) {
838 Error("nterfacer", ERR_WARNING, "sendline: wrong number of arguments: %s", command);
839 return NULL;
840 }
841
dae6d3d7
CP
842 /* we have to create a copy of the arguments for reentrancy reasons, grr */
843 totallen = 0;
844 for(i=0;i<argc;i++)
845 totallen+=strlen(argv[i]) + 1;
846
847 /* HACKY but allows existing code to still work */
848 sr = (struct sched_rline *)ntmalloc(sizeof(struct sched_rline) + totallen);
849 if(!sr) {
e5a7c318
CP
850 MemError();
851 return NULL;
852 }
dae6d3d7
CP
853 prequest = &sr->rl;
854
855 sr->argc = argc;
856 buf = sr->argv;
857 for(i=0;i<argc;i++) {
858 size_t len = strlen(argv[i]) + 1;
859 memcpy(buf, argv[i], len);
860 buf+=len;
861 }
862 sr->hl = hl;
e5a7c318
CP
863
864 prequest->service = servicep;
865 prequest->handler = hl;
866 prequest->buf[0] = '\0';
867 prequest->curpos = prequest->buf;
868 prequest->tag = tag;
869 prequest->id = 0;
e5a7c318
CP
870 prequest->socket = NULL;
871 prequest->callback = callback;
872
dae6d3d7 873 prequest->next = rlines;
e5a7c318 874 rlines = prequest;
e5a7c318 875
dae6d3d7 876 scheduleoneshot(time(NULL), execrline, sr);
e5a7c318 877
dae6d3d7 878 return (void *)sr;
e5a7c318
CP
879}
880
881void nterfacer_freeline(void *tag) {
dae6d3d7 882 struct sched_rline *prequest = tag;
e5a7c318 883
dae6d3d7
CP
884 prequest->rl.callback = NULL;
885 if(prequest->schedule)
886 deleteschedule(prequest->schedule, execrline, NULL);
e5a7c318 887}
2a6be5fe
CP
888
889#define MAX_LINES 8192
890
891/* this bites */
892static void nterfacer_sendcallback(struct rline *ri, int error, char *buf) {
893 char *lines[MAX_LINES+1];
819da112 894 char newbuf[MAX_BUFSIZE+5];
2a6be5fe
CP
895 char *s, *d, *laststart;
896 int linec = 0;
897
819da112 898 for(s=buf,laststart=d=newbuf;*s;s++) {
2a6be5fe
CP
899 if((*s == '\\') && *(s + 1)) {
900 if(*(s + 1) == ',') {
901 *d++ = ',';
902 } else if(*(s + 1) == '\\') {
903 *d++ = '\\';
904 }
905 s++;
906 } else if(*s == ',') {
907 *d++ = '\0';
908 if(linec >= MAX_LINES - 5) {
909 nterfacer_sendcallback(ri, BF_OVER, "Buffer overflow.");
910 return;
911 }
912
913 lines[linec++] = laststart;
819da112 914 laststart = d;
2a6be5fe
CP
915 } else {
916 *d++ = *s;
917 }
918 }
819da112 919 *d = '\0';
2a6be5fe
CP
920 lines[linec++] = laststart;
921
922 ri->callback(error, linec, lines, ri->tag);
923}