]>
jfr.im git - irc/rqf/shadowircd.git/blob - libratbox/src/helper.c
2 * ircd-ratbox: A slightly useful ircd
3 * helper.c: Starts and deals with ircd helpers
5 * Copyright (C) 2006 Aaron Sethman <androsyn@ratbox.org>
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation; either version 2 of the License, or
10 * (at your option) any later version.
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with this program; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
22 * $Id: helper.c 25038 2008-01-23 16:03:08Z androsyn $
24 #include <libratbox_config.h>
25 #include <ratbox_lib.h>
26 #include <commio-int.h>
37 rb_helper_cb
*read_cb
;
38 rb_helper_cb
*error_cb
;
42 /* setup all the stuff a new child needs */
44 rb_helper_child(rb_helper_cb
*read_cb
, rb_helper_cb
*error_cb
, log_cb
*ilog
, restart_cb
*irestart
, die_cb
*idie
,
45 int maxcon
, size_t lb_heap_size
, size_t dh_size
, size_t fd_heap_size
)
50 char *tifd
, *tofd
, *tmaxfd
;
54 tmaxfd
= getenv("MAXFD");
56 if(tifd
== NULL
|| tofd
== NULL
|| tmaxfd
== NULL
)
59 helper
= rb_malloc(sizeof(rb_helper
));
60 ifd
= (int)strtol(tifd
, NULL
, 10);
61 ofd
= (int)strtol(tofd
, NULL
, 10);
62 maxfd
= (int)strtol(tmaxfd
, NULL
, 10);
65 for(x
= 0; x
< maxfd
; x
++)
67 if(x
!= ifd
&& x
!= ofd
)
70 x
= open("/dev/null", O_RDWR
);
71 if(ifd
!= 0 && ofd
!= 0)
73 if(ifd
!= 1 && ofd
!= 1)
75 if(ifd
!= 2 && ofd
!= 2)
77 if(x
> 2) /* don't undo what we just did */
80 x
= 0; /* shut gcc up */
83 rb_lib_init(ilog
, irestart
, idie
, 0, maxfd
, dh_size
, fd_heap_size
);
84 rb_linebuf_init(lb_heap_size
);
85 rb_linebuf_newbuf(&helper
->sendq
);
86 rb_linebuf_newbuf(&helper
->recvq
);
88 helper
->ifd
= rb_open(ifd
, RB_FD_PIPE
, "incoming connection");
89 helper
->ofd
= rb_open(ofd
, RB_FD_PIPE
, "outgoing connection");
90 rb_set_nb(helper
->ifd
);
91 rb_set_nb(helper
->ofd
);
93 helper
->read_cb
= read_cb
;
94 helper
->error_cb
= error_cb
;
100 * starts a new ircd helper
101 * note that this function doesn't start doing reading..thats the job of the caller
105 rb_helper_start(const char *name
, const char *fullpath
, rb_helper_cb
*read_cb
, rb_helper_cb
*error_cb
)
115 if(access(fullpath
, X_OK
) == -1)
118 helper
= rb_malloc(sizeof(rb_helper
));
120 rb_snprintf(buf
, sizeof(buf
), "%s helper - read", name
);
121 if(rb_pipe(&in_f
[0], &in_f
[1], buf
) < 0)
126 rb_snprintf(buf
, sizeof(buf
), "%s helper - write", name
);
127 if(rb_pipe(&out_f
[0], &out_f
[1], buf
) < 0)
133 rb_snprintf(fx
, sizeof(fx
), "%d", rb_get_fd(in_f
[1]));
134 rb_snprintf(fy
, sizeof(fy
), "%d", rb_get_fd(out_f
[0]));
141 setenv("IFD", fy
, 1);
142 setenv("OFD", fx
, 1);
143 setenv("MAXFD", "256", 1);
145 rb_snprintf(buf
, sizeof(buf
), "-ircd %s daemon", name
);
150 SetHandleInformation((HANDLE
)rb_get_fd(in_f
[1]), HANDLE_FLAG_INHERIT
, 1);
151 SetHandleInformation((HANDLE
)rb_get_fd(out_f
[0]), HANDLE_FLAG_INHERIT
, 1);
154 pid
= rb_spawn_process(fullpath
, (const char **)parv
);
169 rb_linebuf_newbuf(&helper
->sendq
);
170 rb_linebuf_newbuf(&helper
->recvq
);
172 helper
->ifd
= in_f
[0];
173 helper
->ofd
= out_f
[1];
174 helper
->read_cb
= read_cb
;
175 helper
->error_cb
= error_cb
;
176 helper
->fork_count
= 0;
184 rb_helper_restart(rb_helper
*helper
)
186 helper
->error_cb(helper
);
191 rb_helper_write_sendq(rb_fde_t
*F
, void *helper_ptr
)
193 rb_helper
*helper
= helper_ptr
;
196 if(rb_linebuf_len(&helper
->sendq
) > 0)
198 while((retlen
= rb_linebuf_flush(F
, &helper
->sendq
)) > 0)
200 if(retlen
== 0 || (retlen
< 0 && !rb_ignore_errno(errno
))) {
201 rb_helper_restart(helper
);
206 if(rb_linebuf_len(&helper
->sendq
) > 0)
207 rb_setselect(helper
->ofd
, RB_SELECT_WRITE
, rb_helper_write_sendq
, helper
);
211 rb_helper_write_queue(rb_helper
*helper
, const char *format
, ...)
214 va_start(ap
, format
);
215 rb_linebuf_putmsg(&helper
->sendq
, format
, &ap
, NULL
);
220 rb_helper_write_flush(rb_helper
*helper
)
222 rb_helper_write_sendq(helper
->ofd
, helper
);
227 rb_helper_write(rb_helper
*helper
, const char *format
, ...)
230 va_start(ap
, format
);
231 rb_linebuf_putmsg(&helper
->sendq
, format
, &ap
, NULL
);
233 rb_helper_write_flush(helper
);
237 rb_helper_read_cb(rb_fde_t
*F
, void *data
)
239 rb_helper
*helper
= (rb_helper
*)data
;
240 static char buf
[32768];
245 while((length
= rb_read(helper
->ifd
, buf
, sizeof(buf
))) > 0)
247 rb_linebuf_parse(&helper
->recvq
, buf
, length
, 0);
248 helper
->read_cb(helper
);
251 if(length
== 0 || (length
< 0 && !rb_ignore_errno(errno
)))
253 rb_helper_restart(helper
);
257 rb_setselect(helper
->ifd
, RB_SELECT_READ
, rb_helper_read_cb
, helper
);
261 rb_helper_run(rb_helper
*helper
)
265 rb_helper_read_cb(helper
->ifd
, helper
);
270 rb_helper_close(rb_helper
*helper
)
275 rb_close(helper
->ifd
);
276 rb_close(helper
->ofd
);
281 rb_helper_read(rb_helper
*helper
, void *buf
, size_t bufsize
)
283 return rb_linebuf_get(&helper
->recvq
, buf
, bufsize
, LINEBUF_COMPLETE
, LINEBUF_PARSED
);
287 rb_helper_loop(rb_helper
*helper
, long delay
)
289 rb_helper_run(helper
);