2 * sslproc.c: An interface to wsockd
3 * Copyright (C) 2007 Aaron Sethman <androsyn@ratbox.org>
4 * Copyright (C) 2007 ircd-ratbox development team
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.
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.
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
37 static void ws_read_ctl(rb_fde_t
* F
, void *data
);
38 static int wsockd_count
;
42 typedef struct _ws_ctl_buf
47 rb_fde_t
*F
[MAXPASSFD
];
65 static rb_dlink_list wsock_daemons
;
67 static inline uint32_t
68 buf_to_uint32(char *buf
)
71 memcpy(&x
, buf
, sizeof(x
));
76 uint32_to_buf(char *buf
, uint32_t x
)
78 memcpy(buf
, &x
, sizeof(x
));
83 allocate_ws_daemon(rb_fde_t
* F
, rb_fde_t
* P
, int pid
)
87 if(F
== NULL
|| pid
< 0)
89 ctl
= rb_malloc(sizeof(ws_ctl_t
));
94 rb_dlinkAdd(ctl
, &ctl
->node
, &wsock_daemons
);
99 free_ws_daemon(ws_ctl_t
* ctl
)
102 ws_ctl_buf_t
*ctl_buf
;
107 RB_DLINK_FOREACH(ptr
, ctl
->readq
.head
)
110 for(x
= 0; x
< ctl_buf
->nfds
; x
++)
111 rb_close(ctl_buf
->F
[x
]);
113 rb_free(ctl_buf
->buf
);
117 RB_DLINK_FOREACH(ptr
, ctl
->writeq
.head
)
120 for(x
= 0; x
< ctl_buf
->nfds
; x
++)
121 rb_close(ctl_buf
->F
[x
]);
123 rb_free(ctl_buf
->buf
);
128 rb_dlinkDelete(&ctl
->node
, &wsock_daemons
);
132 static char *wsockd_path
;
134 static int wsockd_spin_count
= 0;
135 static time_t last_spin
;
136 static int wsockd_wait
= 0;
141 rb_dlink_node
*ptr
, *next
;
144 RB_DLINK_FOREACH_SAFE(ptr
, next
, wsock_daemons
.head
)
155 rb_kill(ctl
->pid
, SIGKILL
);
160 start_wsockd(ServerInfo
.wsockd_count
);
167 rb_dlink_node
*ptr
, *next
;
169 RB_DLINK_FOREACH_SAFE(ptr
, next
, wsock_daemons
.head
)
177 rb_kill(ctl
->pid
, SIGKILL
);
185 ws_dead(ws_ctl_t
* ctl
)
191 rb_kill(ctl
->pid
, SIGKILL
); /* make sure the process is really gone */
196 ilog(L_MAIN
, "wsockd helper died - attempting to restart");
197 sendto_realops_snomask(SNO_GENERAL
, L_ALL
, "wsockd helper died - attempting to restart");
203 ws_do_pipe(rb_fde_t
* F
, void *data
)
206 ws_ctl_t
*ctl
= data
;
207 retlen
= rb_write(F
, "0", 1);
208 if(retlen
== 0 || (retlen
< 0 && !rb_ignore_errno(errno
)))
213 rb_setselect(F
, RB_SELECT_READ
, ws_do_pipe
, data
);
217 restart_wsockd_event(void *unused
)
219 wsockd_spin_count
= 0;
222 if(ServerInfo
.wsockd_count
> get_wsockd_count())
224 int start
= ServerInfo
.wsockd_count
- get_wsockd_count();
225 ilog(L_MAIN
, "Attempting to restart wsockd processes");
226 sendto_realops_snomask(SNO_GENERAL
, L_ALL
, "Attempting to restart wsockd processes");
232 start_wsockd(int count
)
237 const char *suffix
= ".exe";
239 const char *suffix
= "";
242 char fullpath
[PATH_MAX
+ 1];
253 if(wsockd_spin_count
> 20 && (rb_current_time() - last_spin
< 5))
255 ilog(L_MAIN
, "wsockd helper is spinning - will attempt to restart in 1 minute");
256 sendto_realops_snomask(SNO_GENERAL
, L_ALL
,
257 "wsockd helper is spinning - will attempt to restart in 1 minute");
258 rb_event_add("restart_wsockd_event", restart_wsockd_event
, NULL
, 60);
264 last_spin
= rb_current_time();
266 if(wsockd_path
== NULL
)
268 snprintf(fullpath
, sizeof(fullpath
), "%s%cwsockd%s", ircd_paths
[IRCD_PATH_LIBEXEC
], RB_PATH_SEPARATOR
, suffix
);
270 if(access(fullpath
, X_OK
) == -1)
272 snprintf(fullpath
, sizeof(fullpath
), "%s%cbin%cwsockd%s",
273 ConfigFileEntry
.dpath
, RB_PATH_SEPARATOR
, RB_PATH_SEPARATOR
, suffix
);
274 if(access(fullpath
, X_OK
) == -1)
277 "Unable to execute wsockd%s in %s or %s/bin",
278 suffix
, ircd_paths
[IRCD_PATH_LIBEXEC
], ConfigFileEntry
.dpath
);
282 wsockd_path
= rb_strdup(fullpath
);
284 rb_strlcpy(buf
, "-ircd wsockd daemon", sizeof(buf
));
288 for(i
= 0; i
< count
; i
++)
291 if(rb_socketpair(AF_UNIX
, SOCK_DGRAM
, 0, &F1
, &F2
, "wsockd handle passing socket") == -1)
293 ilog(L_MAIN
, "Unable to create wsockd - rb_socketpair failed: %s", strerror(errno
));
297 rb_set_buffers(F1
, READBUF_SIZE
);
298 rb_set_buffers(F2
, READBUF_SIZE
);
299 snprintf(fdarg
, sizeof(fdarg
), "%d", rb_get_fd(F2
));
300 rb_setenv("CTL_FD", fdarg
, 1);
301 if(rb_pipe(&P1
, &P2
, "wsockd pipe") == -1)
303 ilog(L_MAIN
, "Unable to create wsockd - rb_pipe failed: %s", strerror(errno
));
306 snprintf(fdarg
, sizeof(fdarg
), "%d", rb_get_fd(P1
));
307 rb_setenv("CTL_PIPE", fdarg
, 1);
308 snprintf(s_pid
, sizeof(s_pid
), "%d", (int)getpid());
309 rb_setenv("CTL_PPID", s_pid
, 1);
312 SetHandleInformation((HANDLE
) rb_get_fd(F2
), HANDLE_FLAG_INHERIT
, 1);
313 SetHandleInformation((HANDLE
) rb_get_fd(P1
), HANDLE_FLAG_INHERIT
, 1);
316 pid
= rb_spawn_process(wsockd_path
, (const char **) parv
);
319 ilog(L_MAIN
, "Unable to create wsockd: %s\n", strerror(errno
));
329 ctl
= allocate_ws_daemon(F1
, P2
, pid
);
330 ws_read_ctl(ctl
->F
, ctl
);
338 ws_process_dead_fd(ws_ctl_t
* ctl
, ws_ctl_buf_t
* ctl_buf
)
340 struct Client
*client_p
;
344 if(ctl_buf
->buflen
< 6)
345 return; /* bogus message..drop it.. XXX should warn here */
347 fd
= buf_to_uint32(&ctl_buf
->buf
[1]);
348 rb_strlcpy(reason
, &ctl_buf
->buf
[5], sizeof(reason
));
349 client_p
= find_cli_connid_hash(fd
);
352 if(IsAnyServer(client_p
) || IsRegistered(client_p
))
354 /* read any last moment ERROR, QUIT or the like -- jilles */
355 if (!strcmp(reason
, "Remote host closed the connection"))
356 read_packet(client_p
->localClient
->F
, client_p
);
357 if (IsAnyDead(client_p
))
360 exit_client(client_p
, client_p
, &me
, reason
);
365 ws_process_cmd_recv(ws_ctl_t
* ctl
)
367 rb_dlink_node
*ptr
, *next
;
368 ws_ctl_buf_t
*ctl_buf
;
373 RB_DLINK_FOREACH_SAFE(ptr
, next
, ctl
->readq
.head
)
376 switch (*ctl_buf
->buf
)
379 ws_process_dead_fd(ctl
, ctl_buf
);
382 ilog(L_MAIN
, "Received invalid command from wsockd: %s", ctl_buf
->buf
);
383 sendto_realops_snomask(SNO_GENERAL
, L_ALL
, "Received invalid command from wsockd");
386 rb_dlinkDelete(ptr
, &ctl
->readq
);
387 rb_free(ctl_buf
->buf
);
395 ws_read_ctl(rb_fde_t
* F
, void *data
)
397 ws_ctl_buf_t
*ctl_buf
;
398 ws_ctl_t
*ctl
= data
;
405 ctl_buf
= rb_malloc(sizeof(ws_ctl_buf_t
));
406 ctl_buf
->buf
= rb_malloc(READSIZE
);
407 retlen
= rb_recv_fd_buf(ctl
->F
, ctl_buf
->buf
, READSIZE
, ctl_buf
->F
, 4);
408 ctl_buf
->buflen
= retlen
;
411 rb_free(ctl_buf
->buf
);
415 rb_dlinkAddTail(ctl_buf
, &ctl_buf
->node
, &ctl
->readq
);
419 if(retlen
== 0 || (retlen
< 0 && !rb_ignore_errno(errno
)))
424 ws_process_cmd_recv(ctl
);
425 rb_setselect(ctl
->F
, RB_SELECT_READ
, ws_read_ctl
, ctl
);
431 ws_ctl_t
*ctl
, *lowest
= NULL
;
434 RB_DLINK_FOREACH(ptr
, wsock_daemons
.head
)
446 if(ctl
->cli_count
< lowest
->cli_count
)
454 ws_write_ctl(rb_fde_t
* F
, void *data
)
456 ws_ctl_t
*ctl
= data
;
457 ws_ctl_buf_t
*ctl_buf
;
458 rb_dlink_node
*ptr
, *next
;
464 RB_DLINK_FOREACH_SAFE(ptr
, next
, ctl
->writeq
.head
)
467 /* in theory unix sock_dgram shouldn't ever short write this.. */
468 retlen
= rb_send_fd_buf(ctl
->F
, ctl_buf
->F
, ctl_buf
->nfds
, ctl_buf
->buf
, ctl_buf
->buflen
, ctl
->pid
);
471 rb_dlinkDelete(ptr
, &ctl
->writeq
);
472 for(x
= 0; x
< ctl_buf
->nfds
; x
++)
473 rb_close(ctl_buf
->F
[x
]);
474 rb_free(ctl_buf
->buf
);
478 if(retlen
== 0 || (retlen
< 0 && !rb_ignore_errno(errno
)))
485 rb_setselect(ctl
->F
, RB_SELECT_WRITE
, ws_write_ctl
, ctl
);
491 ws_cmd_write_queue(ws_ctl_t
* ctl
, rb_fde_t
** F
, int count
, const void *buf
, size_t buflen
)
493 ws_ctl_buf_t
*ctl_buf
;
500 ctl_buf
= rb_malloc(sizeof(ws_ctl_buf_t
));
501 ctl_buf
->buf
= rb_malloc(buflen
);
502 memcpy(ctl_buf
->buf
, buf
, buflen
);
503 ctl_buf
->buflen
= buflen
;
505 for(x
= 0; x
< count
&& x
< MAXPASSFD
; x
++)
507 ctl_buf
->F
[x
] = F
[x
];
509 ctl_buf
->nfds
= count
;
510 rb_dlinkAddTail(ctl_buf
, &ctl_buf
->node
, &ctl
->writeq
);
511 ws_write_ctl(ctl
->F
, ctl
);
515 start_wsockd_accept(rb_fde_t
* sslF
, rb_fde_t
* plainF
, uint32_t id
)
524 uint32_to_buf(&buf
[1], id
);
525 ctl
= which_wsockd();
529 ws_cmd_write_queue(ctl
, F
, 2, buf
, sizeof(buf
));
534 wsockd_decrement_clicount(ws_ctl_t
* ctl
)
540 if(ctl
->shutdown
&& !ctl
->cli_count
)
543 rb_kill(ctl
->pid
, SIGKILL
);
545 if(ctl
->dead
&& !ctl
->cli_count
)
552 cleanup_dead_ws(void *unused
)
554 rb_dlink_node
*ptr
, *next
;
556 RB_DLINK_FOREACH_SAFE(ptr
, next
, wsock_daemons
.head
)
559 if(ctl
->dead
&& !ctl
->cli_count
)
567 get_wsockd_count(void)
573 wsockd_foreach_info(void (*func
)(void *data
, pid_t pid
, int cli_count
, enum wsockd_status status
), void *data
)
575 rb_dlink_node
*ptr
, *next
;
577 RB_DLINK_FOREACH_SAFE(ptr
, next
, wsock_daemons
.head
)
580 func(data
, ctl
->pid
, ctl
->cli_count
,
581 ctl
->dead
? WSOCKD_DEAD
:
582 (ctl
->shutdown
? WSOCKD_SHUTDOWN
: WSOCKD_ACTIVE
));
589 rb_event_addish("cleanup_dead_ws", cleanup_dead_ws
, NULL
, 60);