]> jfr.im git - irc/rqf/shadowircd.git/blob - src/sslproc.c
178fde32b7f3c10e4575b4a700276bc3cee5d642
[irc/rqf/shadowircd.git] / src / sslproc.c
1 /*
2 * sslproc.c: An interface to ssld
3 * Copyright (C) 2007 Aaron Sethman <androsyn@ratbox.org>
4 * Copyright (C) 2007 ircd-ratbox development team
5 *
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
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
19 * USA
20 *
21 * $Id$
22 */
23
24 #include <ratbox_lib.h>
25 #include "stdinc.h"
26
27
28 #include "s_conf.h"
29 #include "logger.h"
30 #include "listener.h"
31 #include "sslproc.h"
32 #include "s_serv.h"
33 #include "ircd.h"
34 #include "hash.h"
35 #include "client.h"
36 #include "send.h"
37 #include "packet.h"
38
39 #define ZIPSTATS_TIME 60
40
41 static void collect_zipstats(void *unused);
42 static void ssl_read_ctl(rb_fde_t * F, void *data);
43 static int ssld_count;
44
45 static char tmpbuf[READBUF_SIZE];
46 static char nul = '\0';
47
48 #define MAXPASSFD 4
49 #define READSIZE 1024
50 typedef struct _ssl_ctl_buf
51 {
52 rb_dlink_node node;
53 char *buf;
54 size_t buflen;
55 rb_fde_t *F[MAXPASSFD];
56 int nfds;
57 } ssl_ctl_buf_t;
58
59
60 struct _ssl_ctl
61 {
62 rb_dlink_node node;
63 int cli_count;
64 rb_fde_t *F;
65 rb_fde_t *P;
66 pid_t pid;
67 rb_dlink_list readq;
68 rb_dlink_list writeq;
69 uint8_t dead;
70 };
71
72 static void send_new_ssl_certs_one(ssl_ctl_t * ctl, const char *ssl_cert,
73 const char *ssl_private_key, const char *ssl_dh_params);
74 static void send_init_prng(ssl_ctl_t * ctl, prng_seed_t seedtype, const char *path);
75
76
77 static rb_dlink_list ssl_daemons;
78
79 static inline int32_t
80 buf_to_int32(char *buf)
81 {
82 int32_t x;
83 memcpy(&x, buf, sizeof(x));
84 return x;
85 }
86
87 static inline void
88 int32_to_buf(char *buf, int32_t x)
89 {
90 memcpy(buf, &x, sizeof(x));
91 return;
92 }
93
94
95 static inline uint16_t
96 buf_to_uint16(char *buf)
97 {
98 uint16_t x;
99 memcpy(&x, buf, sizeof(x));
100 return x;
101 }
102
103 static inline void
104 uint16_to_buf(char *buf, uint16_t x)
105 {
106 memcpy(buf, &x, sizeof(x));
107 return;
108 }
109
110
111 static ssl_ctl_t *
112 allocate_ssl_daemon(rb_fde_t * F, rb_fde_t * P, int pid)
113 {
114 ssl_ctl_t *ctl;
115
116 if(F == NULL || pid < 0)
117 return NULL;
118 ctl = rb_malloc(sizeof(ssl_ctl_t));
119 ctl->F = F;
120 ctl->P = P;
121 ctl->pid = pid;
122 ssld_count++;
123 rb_dlinkAdd(ctl, &ctl->node, &ssl_daemons);
124 return ctl;
125 }
126
127 static void
128 free_ssl_daemon(ssl_ctl_t * ctl)
129 {
130 rb_dlink_node *ptr;
131 ssl_ctl_buf_t *ctl_buf;
132 int x;
133 if(ctl->cli_count)
134 return;
135
136 RB_DLINK_FOREACH(ptr, ctl->readq.head)
137 {
138 ctl_buf = ptr->data;
139 for(x = 0; x < ctl_buf->nfds; x++)
140 rb_close(ctl_buf->F[x]);
141
142 rb_free(ctl_buf->buf);
143 rb_free(ctl_buf);
144 }
145
146 RB_DLINK_FOREACH(ptr, ctl->writeq.head)
147 {
148 ctl_buf = ptr->data;
149 for(x = 0; x < ctl_buf->nfds; x++)
150 rb_close(ctl_buf->F[x]);
151
152 rb_free(ctl_buf->buf);
153 rb_free(ctl_buf);
154 }
155 rb_close(ctl->F);
156 rb_close(ctl->P);
157 rb_dlinkDelete(&ctl->node, &ssl_daemons);
158 rb_free(ctl);
159 }
160
161 static char *ssld_path;
162
163 static int ssld_spin_count = 0;
164 static time_t last_spin;
165 static int ssld_wait = 0;
166
167
168 static void
169 ssl_killall(void)
170 {
171 rb_dlink_node *ptr, *next;
172 ssl_ctl_t *ctl;
173 RB_DLINK_FOREACH_SAFE(ptr, next, ssl_daemons.head)
174 {
175 ctl = ptr->data;
176 if(ctl->dead)
177 continue;
178 ctl->dead = 1;
179 ssld_count--;
180 rb_kill(ctl->pid, SIGKILL);
181 }
182 }
183
184 static void
185 ssl_dead(ssl_ctl_t * ctl)
186 {
187 if(ctl->dead)
188 return;
189
190 ctl->dead = 1;
191 ssld_count--;
192 rb_kill(ctl->pid, SIGKILL); /* make sure the process is really gone */
193 ilog(L_MAIN, "ssld helper died - attempting to restart");
194 sendto_realops_snomask(SNO_GENERAL, L_ALL, "ssld helper died - attempting to restart");
195 start_ssldaemon(1, ServerInfo.ssl_cert, ServerInfo.ssl_private_key, ServerInfo.ssl_dh_params);
196 }
197
198 static void
199 ssl_do_pipe(rb_fde_t * F, void *data)
200 {
201 int retlen;
202 ssl_ctl_t *ctl = data;
203 retlen = rb_write(F, "0", 1);
204 if(retlen == 0 || (retlen < 0 && !rb_ignore_errno(errno)))
205 {
206 ssl_dead(ctl);
207 return;
208 }
209 rb_setselect(F, RB_SELECT_READ, ssl_do_pipe, data);
210 }
211
212 static void
213 restart_ssld_event(void *unused)
214 {
215 ssld_spin_count = 0;
216 last_spin = 0;
217 ssld_wait = 0;
218 if(ServerInfo.ssld_count > get_ssld_count())
219 {
220 int start = ServerInfo.ssld_count - get_ssld_count();
221 ilog(L_MAIN, "Attempting to restart ssld processes");
222 sendto_realops_snomask(SNO_GENERAL, L_ALL, "Attempt to restart ssld processes");
223 start_ssldaemon(start, ServerInfo.ssl_cert, ServerInfo.ssl_private_key, ServerInfo.ssl_dh_params);
224 }
225 }
226
227 int
228 start_ssldaemon(int count, const char *ssl_cert, const char *ssl_private_key, const char *ssl_dh_params)
229 {
230 rb_fde_t *F1, *F2;
231 rb_fde_t *P1, *P2;
232 #ifdef _WIN32
233 const char *suffix = ".exe";
234 #else
235 const char *suffix = "";
236 #endif
237
238 char fullpath[PATH_MAX + 1];
239 char fdarg[6];
240 const char *parv[2];
241 char buf[128];
242 char s_pid[10];
243 pid_t pid;
244 int started = 0, i;
245
246 if(ssld_wait)
247 return 0;
248
249 if(ssld_spin_count > 20 && (rb_current_time() - last_spin < 5))
250 {
251 ilog(L_MAIN, "ssld helper is spinning - will attempt to restart in 1 minute");
252 sendto_realops_snomask(SNO_GENERAL, L_ALL,
253 "ssld helper is spinning - will attempt to restart in 1 minute");
254 rb_event_add("restart_ssld_event", restart_ssld_event, NULL, 60);
255 ssld_wait = 1;
256 return 0;
257 }
258
259 ssld_spin_count++;
260 last_spin = rb_current_time();
261
262 if(ssld_path == NULL)
263 {
264 rb_snprintf(fullpath, sizeof(fullpath), "%s/ssld%s", BINPATH, suffix);
265
266 if(access(fullpath, X_OK) == -1)
267 {
268 rb_snprintf(fullpath, sizeof(fullpath), "%s/bin/ssld%s",
269 ConfigFileEntry.dpath, suffix);
270 if(access(fullpath, X_OK) == -1)
271 {
272 ilog(L_MAIN,
273 "Unable to execute ssld%s in %s/bin or %s",
274 ConfigFileEntry.dpath, suffix, BINPATH);
275 return 0;
276 }
277 }
278 ssld_path = rb_strdup(fullpath);
279 }
280 rb_strlcpy(buf, "-ircd ssld daemon helper", sizeof(buf));
281 parv[0] = buf;
282 parv[1] = NULL;
283
284 for(i = 0; i < count; i++)
285 {
286 ssl_ctl_t *ctl;
287 if(rb_socketpair(AF_UNIX, SOCK_DGRAM, 0, &F1, &F2, "SSL/TLS handle passing socket") == -1)
288 {
289 ilog(L_MAIN, "Unable to create ssld - rb_socketpair failed: %s", strerror(errno));
290 return started;
291 }
292
293 rb_set_buffers(F1, READBUF_SIZE);
294 rb_set_buffers(F2, READBUF_SIZE);
295 rb_snprintf(fdarg, sizeof(fdarg), "%d", rb_get_fd(F2));
296 rb_setenv("CTL_FD", fdarg, 1);
297 if(rb_pipe(&P1, &P2, "SSL/TLS pipe") == -1)
298 {
299 ilog(L_MAIN, "Unable to create ssld - rb_pipe failed: %s", strerror(errno));
300 return started;
301 }
302 rb_snprintf(fdarg, sizeof(fdarg), "%d", rb_get_fd(P1));
303 rb_setenv("CTL_PIPE", fdarg, 1);
304 rb_snprintf(s_pid, sizeof(s_pid), "%d", (int)getpid());
305 rb_setenv("CTL_PPID", s_pid, 1);
306 #ifdef _WIN32
307 SetHandleInformation((HANDLE) rb_get_fd(F2), HANDLE_FLAG_INHERIT, 1);
308 SetHandleInformation((HANDLE) rb_get_fd(P1), HANDLE_FLAG_INHERIT, 1);
309 #endif
310
311 pid = rb_spawn_process(ssld_path, (const char **) parv);
312 if(pid == -1)
313 {
314 ilog(L_MAIN, "Unable to create ssld: %s\n", strerror(errno));
315 rb_close(F1);
316 rb_close(F2);
317 rb_close(P1);
318 rb_close(P2);
319 return started;
320 }
321 started++;
322 rb_close(F2);
323 rb_close(P1);
324 ctl = allocate_ssl_daemon(F1, P2, pid);
325 if(ssl_ok)
326 {
327 if(ConfigFileEntry.use_egd && (ConfigFileEntry.egdpool_path != NULL))
328 send_init_prng(ctl, RB_PRNG_EGD, ConfigFileEntry.egdpool_path);
329 else
330 send_init_prng(ctl, RB_PRNG_DEFAULT, NULL);
331 }
332 if(ssl_ok && ssl_cert != NULL && ssl_private_key != NULL)
333 send_new_ssl_certs_one(ctl, ssl_cert, ssl_private_key,
334 ssl_dh_params != NULL ? ssl_dh_params : "");
335 ssl_read_ctl(ctl->F, ctl);
336 ssl_do_pipe(P2, ctl);
337
338 }
339 return started;
340 }
341
342 static void
343 ssl_process_zipstats(ssl_ctl_t * ctl, ssl_ctl_buf_t * ctl_buf)
344 {
345 struct Client *server;
346 struct ZipStats *zips;
347 int parc;
348 char *parv[7];
349 parc = rb_string_to_array(ctl_buf->buf, parv, 6);
350 server = find_server(NULL, parv[1]);
351 if(server == NULL || server->localClient == NULL || !IsCapable(server, CAP_ZIP))
352 return;
353 if(server->localClient->zipstats == NULL)
354 server->localClient->zipstats = rb_malloc(sizeof(struct ZipStats));
355
356 zips = server->localClient->zipstats;
357
358 zips->in += strtoull(parv[2], NULL, 10);
359 zips->in_wire += strtoull(parv[3], NULL, 10);
360 zips->out += strtoull(parv[4], NULL, 10);
361 zips->out_wire += strtoull(parv[5], NULL, 10);
362
363 if(zips->in > 0)
364 zips->in_ratio = ((double) (zips->in - zips->in_wire) / (double) zips->in) * 100.00;
365 else
366 zips->in_ratio = 0;
367
368 if(zips->out > 0)
369 zips->out_ratio = ((double) (zips->out - zips->out_wire) / (double) zips->out) * 100.00;
370 else
371 zips->out_ratio = 0;
372 }
373
374 static void
375 ssl_process_dead_fd(ssl_ctl_t * ctl, ssl_ctl_buf_t * ctl_buf)
376 {
377 struct Client *client_p;
378 char reason[256];
379 int32_t fd;
380
381 if(ctl_buf->buflen < 6)
382 return; /* bogus message..drop it.. XXX should warn here */
383
384 fd = buf_to_int32(&ctl_buf->buf[1]);
385 rb_strlcpy(reason, &ctl_buf->buf[5], sizeof(reason));
386 client_p = find_cli_fd_hash(fd);
387 if(client_p == NULL)
388 return;
389 if(IsAnyServer(client_p) || IsRegistered(client_p))
390 {
391 /* read any last moment ERROR, QUIT or the like -- jilles */
392 if (!strcmp(reason, "Remote host closed the connection"))
393 read_packet(client_p->localClient->F, client_p);
394 if (IsAnyDead(client_p))
395 return;
396 }
397 if(IsAnyServer(client_p))
398 {
399 sendto_realops_snomask(SNO_GENERAL, is_remote_connect(client_p) && !IsServer(client_p) ? L_NETWIDE : L_ALL, "ssld error for %s: %s", client_p->name, reason);
400 ilog(L_SERVER, "ssld error for %s: %s", log_client_name(client_p, SHOW_IP), reason);
401 }
402 exit_client(client_p, client_p, &me, reason);
403 }
404
405 static void
406 ssl_process_certfp(ssl_ctl_t * ctl, ssl_ctl_buf_t * ctl_buf)
407 {
408 struct Client *client_p;
409 int32_t fd;
410 uint8_t *certfp;
411 char *certfp_string;
412 int i;
413
414 if(ctl_buf->buflen != 5 + RB_SSL_CERTFP_LEN)
415 return; /* bogus message..drop it.. XXX should warn here */
416
417 fd = buf_to_int32(&ctl_buf->buf[1]);
418 certfp = (uint8_t *)&ctl_buf->buf[5];
419 client_p = find_cli_fd_hash(fd);
420 if(client_p == NULL)
421 return;
422 rb_free(client_p->certfp);
423 certfp_string = rb_malloc(RB_SSL_CERTFP_LEN * 2 + 1);
424 for(i = 0; i < RB_SSL_CERTFP_LEN; i++)
425 rb_snprintf(certfp_string + 2 * i, 3, "%02x",
426 certfp[i]);
427 client_p->certfp = certfp_string;
428 }
429
430 static void
431 ssl_process_cmd_recv(ssl_ctl_t * ctl)
432 {
433 static const char *cannot_setup_ssl = "ssld cannot setup ssl, check your certificates and private key";
434 static const char *no_ssl_or_zlib = "ssld has neither SSL/TLS or zlib support killing all sslds";
435 rb_dlink_node *ptr, *next;
436 ssl_ctl_buf_t *ctl_buf;
437 if(ctl->dead)
438 return;
439 RB_DLINK_FOREACH_SAFE(ptr, next, ctl->readq.head)
440 {
441 ctl_buf = ptr->data;
442 switch (*ctl_buf->buf)
443 {
444 case 'N':
445 ssl_ok = 0; /* ssld says it can't do ssl/tls */
446 break;
447 case 'D':
448 ssl_process_dead_fd(ctl, ctl_buf);
449 break;
450 case 'F':
451 ssl_process_certfp(ctl, ctl_buf);
452 break;
453 case 'S':
454 ssl_process_zipstats(ctl, ctl_buf);
455 break;
456 case 'I':
457 ssl_ok = 0;
458 ilog(L_MAIN, cannot_setup_ssl);
459 sendto_realops_snomask(SNO_GENERAL, L_ALL, cannot_setup_ssl);
460 case 'U':
461 zlib_ok = 0;
462 ssl_ok = 0;
463 ilog(L_MAIN, no_ssl_or_zlib);
464 sendto_realops_snomask(SNO_GENERAL, L_ALL, no_ssl_or_zlib);
465 ssl_killall();
466 break;
467 case 'z':
468 zlib_ok = 0;
469 break;
470 default:
471 ilog(L_MAIN, "Received invalid command from ssld: %s", ctl_buf->buf);
472 sendto_realops_snomask(SNO_GENERAL, L_ALL, "Received invalid command from ssld");
473 break;
474 }
475 rb_dlinkDelete(ptr, &ctl->readq);
476 rb_free(ctl_buf->buf);
477 rb_free(ctl_buf);
478 }
479
480 }
481
482
483 static void
484 ssl_read_ctl(rb_fde_t * F, void *data)
485 {
486 ssl_ctl_buf_t *ctl_buf;
487 ssl_ctl_t *ctl = data;
488 int retlen;
489
490 if(ctl->dead)
491 return;
492 do
493 {
494 ctl_buf = rb_malloc(sizeof(ssl_ctl_buf_t));
495 ctl_buf->buf = rb_malloc(READSIZE);
496 retlen = rb_recv_fd_buf(ctl->F, ctl_buf->buf, READSIZE, ctl_buf->F, 4);
497 ctl_buf->buflen = retlen;
498 if(retlen <= 0)
499 {
500 rb_free(ctl_buf->buf);
501 rb_free(ctl_buf);
502 }
503 else
504 rb_dlinkAddTail(ctl_buf, &ctl_buf->node, &ctl->readq);
505 }
506 while(retlen > 0);
507
508 if(retlen == 0 || (retlen < 0 && !rb_ignore_errno(errno)))
509 {
510 ssl_dead(ctl);
511 return;
512 }
513 ssl_process_cmd_recv(ctl);
514 rb_setselect(ctl->F, RB_SELECT_READ, ssl_read_ctl, ctl);
515 }
516
517 static ssl_ctl_t *
518 which_ssld(void)
519 {
520 ssl_ctl_t *ctl, *lowest = NULL;
521 rb_dlink_node *ptr;
522
523 RB_DLINK_FOREACH(ptr, ssl_daemons.head)
524 {
525 ctl = ptr->data;
526 if(ctl->dead)
527 continue;
528 if(lowest == NULL)
529 {
530 lowest = ctl;
531 continue;
532 }
533 if(ctl->cli_count < lowest->cli_count)
534 lowest = ctl;
535 }
536 return (lowest);
537 }
538
539 static void
540 ssl_write_ctl(rb_fde_t * F, void *data)
541 {
542 ssl_ctl_t *ctl = data;
543 ssl_ctl_buf_t *ctl_buf;
544 rb_dlink_node *ptr, *next;
545 int retlen, x;
546
547 if(ctl->dead)
548 return;
549
550 RB_DLINK_FOREACH_SAFE(ptr, next, ctl->writeq.head)
551 {
552 ctl_buf = ptr->data;
553 /* in theory unix sock_dgram shouldn't ever short write this.. */
554 retlen = rb_send_fd_buf(ctl->F, ctl_buf->F, ctl_buf->nfds, ctl_buf->buf, ctl_buf->buflen, ctl->pid);
555 if(retlen > 0)
556 {
557 rb_dlinkDelete(ptr, &ctl->writeq);
558 for(x = 0; x < ctl_buf->nfds; x++)
559 rb_close(ctl_buf->F[x]);
560 rb_free(ctl_buf->buf);
561 rb_free(ctl_buf);
562
563 }
564 if(retlen == 0 || (retlen < 0 && !rb_ignore_errno(errno)))
565 {
566 ssl_dead(ctl);
567 return;
568 }
569 else
570 {
571 rb_setselect(ctl->F, RB_SELECT_WRITE, ssl_write_ctl, ctl);
572 }
573 }
574 }
575
576 static void
577 ssl_cmd_write_queue(ssl_ctl_t * ctl, rb_fde_t ** F, int count, const void *buf, size_t buflen)
578 {
579 ssl_ctl_buf_t *ctl_buf;
580 int x;
581
582 /* don't bother */
583 if(ctl->dead)
584 return;
585
586 ctl_buf = rb_malloc(sizeof(ssl_ctl_buf_t));
587 ctl_buf->buf = rb_malloc(buflen);
588 memcpy(ctl_buf->buf, buf, buflen);
589 ctl_buf->buflen = buflen;
590
591 for(x = 0; x < count && x < MAXPASSFD; x++)
592 {
593 ctl_buf->F[x] = F[x];
594 }
595 ctl_buf->nfds = count;
596 rb_dlinkAddTail(ctl_buf, &ctl_buf->node, &ctl->writeq);
597 ssl_write_ctl(ctl->F, ctl);
598 }
599
600
601 static void
602 send_new_ssl_certs_one(ssl_ctl_t * ctl, const char *ssl_cert, const char *ssl_private_key, const char *ssl_dh_params)
603 {
604 size_t len;
605
606 len = strlen(ssl_cert) + strlen(ssl_private_key) + strlen(ssl_dh_params) + 5;
607 if(len > sizeof(tmpbuf))
608 {
609 sendto_realops_snomask(SNO_GENERAL, L_ALL,
610 "Parameters for send_new_ssl_certs_one too long (%zu > %zu) to pass to ssld, not sending...",
611 len, sizeof(tmpbuf));
612 ilog(L_MAIN,
613 "Parameters for send_new_ssl_certs_one too long (%zu > %zu) to pass to ssld, not sending...",
614 len, sizeof(tmpbuf));
615 return;
616 }
617 len = rb_snprintf(tmpbuf, sizeof(tmpbuf), "K%c%s%c%s%c%s%c", nul, ssl_cert, nul,
618 ssl_private_key, nul, ssl_dh_params, nul);
619 ssl_cmd_write_queue(ctl, NULL, 0, tmpbuf, len);
620 }
621
622 static void
623 send_init_prng(ssl_ctl_t * ctl, prng_seed_t seedtype, const char *path)
624 {
625 size_t len;
626 const char *s;
627 uint8_t seed = (uint8_t) seedtype;
628
629 if(path == NULL)
630 s = "";
631 else
632 s = path;
633
634 len = strlen(s) + 3;
635 if(len > sizeof(tmpbuf))
636 {
637 sendto_realops_snomask(SNO_GENERAL, L_ALL,
638 "Parameters for send_init_prng too long (%zd > %zd) to pass to ssld, not sending...",
639 len, sizeof(tmpbuf));
640 ilog(L_MAIN,
641 "Parameters for send_init_prng too long (%zd > %zd) to pass to ssld, not sending...",
642 len, sizeof(tmpbuf));
643 return;
644
645 }
646 len = rb_snprintf(tmpbuf, sizeof(tmpbuf), "I%c%s%c", seed, s, nul);
647 ssl_cmd_write_queue(ctl, NULL, 0, tmpbuf, len);
648 }
649
650 void
651 send_new_ssl_certs(const char *ssl_cert, const char *ssl_private_key, const char *ssl_dh_params)
652 {
653 rb_dlink_node *ptr;
654 if(ssl_cert == NULL || ssl_private_key == NULL || ssl_dh_params == NULL)
655 {
656 ssl_ok = 0;
657 return;
658 }
659 RB_DLINK_FOREACH(ptr, ssl_daemons.head)
660 {
661 ssl_ctl_t *ctl = ptr->data;
662 send_new_ssl_certs_one(ctl, ssl_cert, ssl_private_key, ssl_dh_params);
663 }
664 }
665
666
667 ssl_ctl_t *
668 start_ssld_accept(rb_fde_t * sslF, rb_fde_t * plainF, int32_t id)
669 {
670 rb_fde_t *F[2];
671 ssl_ctl_t *ctl;
672 char buf[5];
673 F[0] = sslF;
674 F[1] = plainF;
675
676 buf[0] = 'A';
677 int32_to_buf(&buf[1], id);
678 ctl = which_ssld();
679 ctl->cli_count++;
680 ssl_cmd_write_queue(ctl, F, 2, buf, sizeof(buf));
681 return ctl;
682 }
683
684 ssl_ctl_t *
685 start_ssld_connect(rb_fde_t * sslF, rb_fde_t * plainF, int32_t id)
686 {
687 rb_fde_t *F[2];
688 ssl_ctl_t *ctl;
689 char buf[5];
690 F[0] = sslF;
691 F[1] = plainF;
692
693 buf[0] = 'C';
694 int32_to_buf(&buf[1], id);
695
696 ctl = which_ssld();
697 ctl->cli_count++;
698 ssl_cmd_write_queue(ctl, F, 2, buf, sizeof(buf));
699 return ctl;
700 }
701
702 void
703 ssld_decrement_clicount(ssl_ctl_t * ctl)
704 {
705 if(ctl == NULL)
706 return;
707
708 ctl->cli_count--;
709 if(ctl->dead && !ctl->cli_count)
710 {
711 free_ssl_daemon(ctl);
712 }
713 }
714
715 /*
716 * what we end up sending to the ssld process for ziplinks is the following
717 * Z[ourfd][level][RECVQ]
718 * Z = ziplinks command = buf[0]
719 * ourfd = Our end of the socketpair = buf[1..4]
720 * level = zip level buf[5]
721 * recvqlen = our recvq len = buf[6-7]
722 * recvq = any data we read prior to starting ziplinks
723 */
724 void
725 start_zlib_session(void *data)
726 {
727 struct Client *server = (struct Client *) data;
728 uint16_t recvqlen;
729 uint8_t level;
730 void *xbuf;
731
732 rb_fde_t *F[2];
733 rb_fde_t *xF1, *xF2;
734 char *buf;
735 char buf2[9];
736 void *recvq_start;
737
738 size_t hdr = (sizeof(uint8_t) * 2) + sizeof(int32_t);
739 size_t len;
740 int cpylen, left;
741
742 server->localClient->event = NULL;
743
744 recvqlen = rb_linebuf_len(&server->localClient->buf_recvq);
745
746 len = recvqlen + hdr;
747
748 if(len > READBUF_SIZE)
749 {
750 sendto_realops_snomask(SNO_GENERAL, L_ALL,
751 "ssld - attempted to pass message of %zd len, max len %d, giving up",
752 len, READBUF_SIZE);
753 ilog(L_MAIN, "ssld - attempted to pass message of %zd len, max len %d, giving up", len, READBUF_SIZE);
754 exit_client(server, server, server, "ssld readbuf exceeded");
755 return;
756 }
757
758 buf = rb_malloc(len);
759 level = ConfigFileEntry.compression_level;
760
761 int32_to_buf(&buf[1], rb_get_fd(server->localClient->F));
762 buf[5] = (char) level;
763
764 recvq_start = &buf[6];
765 server->localClient->zipstats = rb_malloc(sizeof(struct ZipStats));
766
767 xbuf = recvq_start;
768 left = recvqlen;
769
770 do
771 {
772 cpylen = rb_linebuf_get(&server->localClient->buf_recvq, xbuf, left, LINEBUF_PARTIAL, LINEBUF_RAW);
773 left -= cpylen;
774 xbuf = (void *) (((uintptr_t) xbuf) + cpylen);
775 }
776 while(cpylen > 0);
777
778 /* Pass the socket to ssld. */
779 *buf = 'Z';
780 if(rb_socketpair(AF_UNIX, SOCK_STREAM, 0, &xF1, &xF2, "Initial zlib socketpairs") == -1)
781 {
782 sendto_realops_snomask(SNO_GENERAL, L_ALL, "Error creating zlib socketpair - %s", strerror(errno));
783 ilog(L_MAIN, "Error creating zlib socketpairs - %s", strerror(errno));
784 exit_client(server, server, server, "Error creating zlib socketpair");
785 return;
786 }
787
788 if(IsSSL(server))
789 {
790 /* tell ssld the new connid for the ssl part*/
791 buf2[0] = 'Y';
792 int32_to_buf(&buf2[1], rb_get_fd(server->localClient->F));
793 int32_to_buf(&buf2[5], rb_get_fd(xF2));
794 ssl_cmd_write_queue(server->localClient->ssl_ctl, NULL, 0, buf2, sizeof(buf2));
795 }
796
797
798 F[0] = server->localClient->F;
799 F[1] = xF1;
800 del_from_cli_fd_hash(server);
801 server->localClient->F = xF2;
802 /* need to redo as what we did before isn't valid now */
803 int32_to_buf(&buf[1], rb_get_fd(server->localClient->F));
804 add_to_cli_fd_hash(server);
805
806 server->localClient->z_ctl = which_ssld();
807 server->localClient->z_ctl->cli_count++;
808 ssl_cmd_write_queue(server->localClient->z_ctl, F, 2, buf, len);
809 rb_free(buf);
810 }
811
812 static void
813 collect_zipstats(void *unused)
814 {
815 rb_dlink_node *ptr;
816 struct Client *target_p;
817 char buf[sizeof(uint8_t) + sizeof(int32_t) + HOSTLEN];
818 void *odata;
819 size_t len;
820 int32_t id;
821
822 buf[0] = 'S';
823 odata = buf + sizeof(uint8_t) + sizeof(int32_t);
824
825 RB_DLINK_FOREACH(ptr, serv_list.head)
826 {
827 target_p = ptr->data;
828 if(IsCapable(target_p, CAP_ZIP))
829 {
830 len = sizeof(uint8_t) + sizeof(uint32_t);
831
832 id = rb_get_fd(target_p->localClient->F);
833 int32_to_buf(&buf[1], rb_get_fd(target_p->localClient->F));
834 rb_strlcpy(odata, target_p->name, (sizeof(buf) - len));
835 len += strlen(odata) + 1; /* Get the \0 as well */
836 ssl_cmd_write_queue(target_p->localClient->z_ctl, NULL, 0, buf, len);
837 }
838 }
839 }
840
841 static void
842 cleanup_dead_ssl(void *unused)
843 {
844 rb_dlink_node *ptr, *next;
845 ssl_ctl_t *ctl;
846 RB_DLINK_FOREACH_SAFE(ptr, next, ssl_daemons.head)
847 {
848 ctl = ptr->data;
849 if(ctl->dead && !ctl->cli_count)
850 {
851 free_ssl_daemon(ctl);
852 }
853 }
854 }
855
856 int
857 get_ssld_count(void)
858 {
859 return ssld_count;
860 }
861
862 void
863 init_ssld(void)
864 {
865 rb_event_addish("collect_zipstats", collect_zipstats, NULL, ZIPSTATS_TIME);
866 rb_event_addish("cleanup_dead_ssld", cleanup_dead_ssl, NULL, 1200);
867 }