]>
jfr.im git - solanum.git/blob - librb/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
24 #include <librb_config.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
,
45 restart_cb
* irestart
, die_cb
* idie
, size_t lb_heap_size
,
46 size_t dh_size
, size_t fd_heap_size
)
51 char *tifd
, *tofd
, *tmaxfd
;
55 tmaxfd
= getenv("MAXFD");
57 if(tifd
== NULL
|| tofd
== NULL
|| tmaxfd
== NULL
)
60 helper
= rb_malloc(sizeof(rb_helper
));
61 ifd
= (int)strtol(tifd
, NULL
, 10);
62 ofd
= (int)strtol(tofd
, NULL
, 10);
63 maxfd
= (int)strtol(tmaxfd
, NULL
, 10);
66 for(x
= 0; x
< maxfd
; x
++)
68 if(x
!= ifd
&& x
!= ofd
)
71 x
= open("/dev/null", O_RDWR
);
72 if(ifd
!= 0 && ofd
!= 0)
74 if(ifd
!= 1 && ofd
!= 1)
76 if(ifd
!= 2 && ofd
!= 2)
78 if(x
> 2) /* don't undo what we just did */
81 (void) x
; /* shut gcc up */
84 rb_lib_init(ilog
, irestart
, idie
, 0, maxfd
, dh_size
, fd_heap_size
);
85 rb_linebuf_init(lb_heap_size
);
86 rb_linebuf_newbuf(&helper
->sendq
);
87 rb_linebuf_newbuf(&helper
->recvq
);
89 helper
->ifd
= rb_open(ifd
, RB_FD_PIPE
, "incoming connection");
90 helper
->ofd
= rb_open(ofd
, RB_FD_PIPE
, "outgoing connection");
91 rb_set_nb(helper
->ifd
);
92 rb_set_nb(helper
->ofd
);
94 helper
->read_cb
= read_cb
;
95 helper
->error_cb
= error_cb
;
101 * starts a new ircd helper
102 * note that this function doesn't start doing reading..thats the job of the caller
106 rb_helper_start(const char *name
, const char *fullpath
, rb_helper_cb
* read_cb
,
107 rb_helper_cb
* error_cb
)
117 if(access(fullpath
, X_OK
) == -1)
120 helper
= rb_malloc(sizeof(rb_helper
));
122 snprintf(buf
, sizeof(buf
), "%s helper - read", name
);
123 if(rb_pipe(&in_f
[0], &in_f
[1], buf
) < 0)
128 snprintf(buf
, sizeof(buf
), "%s helper - write", name
);
129 if(rb_pipe(&out_f
[0], &out_f
[1], buf
) < 0)
135 snprintf(fx
, sizeof(fx
), "%d", rb_get_fd(in_f
[1]));
136 snprintf(fy
, sizeof(fy
), "%d", rb_get_fd(out_f
[0]));
143 rb_setenv("IFD", fy
, 1);
144 rb_setenv("OFD", fx
, 1);
145 rb_setenv("MAXFD", "256", 1);
147 snprintf(buf
, sizeof(buf
), "-ircd %s daemon", name
);
151 rb_clear_cloexec(in_f
[1]);
152 rb_clear_cloexec(out_f
[0]);
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
)))
202 rb_helper_restart(helper
);
207 if(rb_linebuf_len(&helper
->sendq
) > 0)
208 rb_setselect(helper
->ofd
, RB_SELECT_WRITE
, rb_helper_write_sendq
, helper
);
212 rb_helper_write_queue(rb_helper
*helper
, const char *format
, ...)
215 rb_strf_t strings
= { .format
= format
, .format_args
= &ap
, .next
= NULL
};
217 va_start(ap
, format
);
218 rb_linebuf_put(&helper
->sendq
, &strings
);
223 rb_helper_write_flush(rb_helper
*helper
)
225 rb_helper_write_sendq(helper
->ofd
, helper
);
230 rb_helper_write(rb_helper
*helper
, const char *format
, ...)
233 rb_strf_t strings
= { .format
= format
, .format_args
= &ap
, .next
= NULL
};
235 va_start(ap
, format
);
236 rb_linebuf_put(&helper
->sendq
, &strings
);
239 rb_helper_write_flush(helper
);
243 rb_helper_read_cb(rb_fde_t
*F
__attribute__((unused
)), void *data
)
245 rb_helper
*helper
= (rb_helper
*)data
;
246 static char buf
[32768];
251 while((length
= rb_read(helper
->ifd
, buf
, sizeof(buf
))) > 0)
253 rb_linebuf_parse(&helper
->recvq
, buf
, length
, 0);
254 helper
->read_cb(helper
);
257 if(length
== 0 || (length
< 0 && !rb_ignore_errno(errno
)))
259 rb_helper_restart(helper
);
263 rb_setselect(helper
->ifd
, RB_SELECT_READ
, rb_helper_read_cb
, helper
);
267 rb_helper_run(rb_helper
*helper
)
271 rb_helper_read_cb(helper
->ifd
, helper
);
276 rb_helper_close(rb_helper
*helper
)
280 rb_kill(helper
->pid
, SIGKILL
);
281 rb_close(helper
->ifd
);
282 rb_close(helper
->ofd
);
287 rb_helper_read(rb_helper
*helper
, void *buf
, size_t bufsize
)
289 return rb_linebuf_get(&helper
->recvq
, buf
, bufsize
, LINEBUF_COMPLETE
, LINEBUF_PARSED
);
293 rb_helper_loop(rb_helper
*helper
, long delay
)
295 rb_helper_run(helper
);