1 /******************************************************************************/
2 /* postlicyd: a postfix policy daemon with a lot of features */
4 /* ________________________________________________________________________ */
6 /* Redistribution and use in source and binary forms, with or without */
7 /* modification, are permitted provided that the following conditions */
10 /* 1. Redistributions of source code must retain the above copyright */
11 /* notice, this list of conditions and the following disclaimer. */
12 /* 2. Redistributions in binary form must reproduce the above copyright */
13 /* notice, this list of conditions and the following disclaimer in the */
14 /* documentation and/or other materials provided with the distribution. */
15 /* 3. The names of its contributors may not be used to endorse or promote */
16 /* products derived from this software without specific prior written */
19 /* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND */
20 /* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE */
21 /* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR */
22 /* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS */
23 /* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR */
24 /* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF */
25 /* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS */
26 /* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN */
27 /* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) */
28 /* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF */
29 /* THE POSSIBILITY OF SUCH DAMAGE. */
30 /******************************************************************************/
33 * Copyright © 2005-2007 Pierre Habouzit
37 #include <netinet/in.h>
38 #include <sys/epoll.h>
48 #define DAEMON_NAME "srsd"
49 #define DEFAULT_ENCODER_PORT 10000
50 #define DEFAULT_DECODER_PORT 10001
52 #define STR(x) __tostr(x)
54 /* srs encoder/decoder/listener worker {{{ */
56 typedef struct srsd_t {
57 unsigned listener : 1;
65 static srsd_t *srsd_new(void)
67 srsd_t *srsd = p_new(srsd_t, 1);
72 static void srsd_delete(srsd_t **srsd)
77 buffer_wipe(&(*srsd)->ibuf);
78 buffer_wipe(&(*srsd)->obuf);
83 int process_srs(srs_t *srs, const char *domain, srsd_t *srsd)
85 while (srsd->ibuf.len > 4) {
86 char buf[BUFSIZ], *p, *q, *nl;
89 if (strncmp("get ", srsd->ibuf.data, 4)) {
90 syslog(LOG_ERR, "bad request, not starting with \"get \"");
94 nl = strchr(srsd->ibuf.data + 4, '\n');
98 for (p = srsd->ibuf.data + 4; p < nl && isspace(*p); p++);
99 for (q = nl++; q >= p && isspace(*q); *q-- = '\0');
102 syslog(LOG_WARNING, "empty request");
107 err = srs_reverse(srs, buf, ssizeof(buf), p);
109 err = srs_forward(srs, buf, ssizeof(buf), p, domain);
113 buffer_addstr(&srsd->obuf, "200 ");
114 buffer_addstr(&srsd->obuf, buf);
115 buffer_addstr(&srsd->obuf, "\r\n");
117 switch (SRS_ERROR_TYPE(err)) {
118 case SRS_ERRTYPE_SRS:
119 case SRS_ERRTYPE_SYNTAX:
120 buffer_addstr(&srsd->obuf, "500 ");
123 buffer_addstr(&srsd->obuf, "400 ");
126 buffer_addstr(&srsd->obuf, srs_strerror(err));
127 buffer_addstr(&srsd->obuf, "\r\n");
131 buffer_consume(&srsd->ibuf, nl - srsd->ibuf.data);
137 int start_listener(int epollfd, int port, bool decoder)
139 struct sockaddr_in addr = {
140 .sin_family = AF_INET,
141 .sin_addr = { htonl(INADDR_LOOPBACK) },
143 struct epoll_event evt = { .events = EPOLLIN };
147 addr.sin_port = htons(port);
148 sock = tcp_listen((const struct sockaddr *)&addr, sizeof(addr));
153 evt.data.ptr = tmp = srsd_new();
155 tmp->decoder = decoder;
156 tmp->listener = true;
157 if (epoll_ctl(epollfd, EPOLL_CTL_ADD, sock, &evt) < 0) {
158 UNIXERR("epoll_ctl");
165 /* administrivia {{{ */
167 static int main_initialize(void)
169 openlog(DAEMON_NAME, LOG_PID, LOG_MAIL);
170 signal(SIGPIPE, SIG_IGN);
171 signal(SIGINT, &common_sighandler);
172 signal(SIGTERM, &common_sighandler);
173 signal(SIGHUP, &common_sighandler);
174 syslog(LOG_INFO, "Starting...");
178 static void main_shutdown(void)
180 syslog(LOG_INFO, cleanexit ? "Stopping..." : "Unclean exit...");
184 module_init(main_initialize);
185 module_exit(main_shutdown);
189 fputs("usage: "DAEMON_NAME" [ -e <port> ] [ -d <port> ] domain secrets\n"
191 " -e <port> port to listen to for encoding requests\n"
192 " (default: "STR(DEFAULT_ENCODER_PORT)")\n"
193 " -d <port> port to listen to for decoding requests\n"
194 " (default: "STR(DEFAULT_DECODER_PORT)")\n"
200 int main_loop(srs_t *srs, const char *domain, int port_enc, int port_dec)
202 int exitcode = EXIT_SUCCESS;
203 int epollfd = epoll_create(128);
206 UNIXERR("epoll_create");
207 exitcode = EXIT_FAILURE;
211 if (start_listener(epollfd, port_enc, false) < 0)
213 if (start_listener(epollfd, port_dec, true) < 0)
217 struct epoll_event evts[1024];
220 n = epoll_wait(epollfd, evts, countof(evts), -1);
222 if (errno != EAGAIN && errno != EINTR) {
223 UNIXERR("epoll_wait");
224 exitcode = EXIT_FAILURE;
231 srsd_t *srsd = evts[n].data.ptr;
233 if (srsd->listener) {
234 struct epoll_event evt = { .events = EPOLLIN };
238 sock = accept(srsd->fd, NULL, NULL);
244 evt.data.ptr = tmp = srsd_new();
245 tmp->decoder = srsd->decoder;
247 if (epoll_ctl(epollfd, EPOLL_CTL_ADD, sock, &evt) < 0) {
248 UNIXERR("epoll_ctl");
255 if (evts[n].events & EPOLLIN) {
256 int res = buffer_read(&srsd->ibuf, srsd->fd, -1);
258 if ((res < 0 && errno != EINTR && errno != EAGAIN)
265 if (process_srs(srs, domain, srsd) < 0) {
271 if ((evts[n].events & EPOLLOUT) && srsd->obuf.len) {
272 int res = write(srsd->fd, srsd->obuf.data, srsd->obuf.len);
274 if (res < 0 && errno != EINTR && errno != EAGAIN) {
280 buffer_consume(&srsd->obuf, res);
284 if (srsd->watchwr == !srsd->obuf.len) {
285 struct epoll_event evt = {
286 .events = EPOLLIN | (srsd->obuf.len ? EPOLLOUT : 0),
289 if (epoll_ctl(epollfd, EPOLL_CTL_MOD, srsd->fd, &evt) < 0) {
290 UNIXERR("epoll_ctl");
294 srsd->watchwr = srsd->obuf.len != 0;
306 static srs_t *srs_read_secrets(const char *sfile)
313 f = fopen(sfile, "r");
321 while (fgets(buf, sizeof(buf), f)) {
325 if (buf[n - 1] != '\n') {
326 syslog(LOG_CRIT, "%s:%d: line too long", sfile, lineno);
330 srs_add_secret(srs, buf);
334 syslog(LOG_CRIT, "%s: empty file, no secrets", sfile);
347 int main(int argc, char *argv[])
349 int port_enc = DEFAULT_ENCODER_PORT;
350 int port_dec = DEFAULT_DECODER_PORT;
354 if (atexit(common_shutdown)) {
355 fputs("Cannot hook my atexit function, quitting !\n", stderr);
360 for (int c = 0; (c = getopt(argc, argv, "he:d:")) >= 0; ) {
363 port_enc = atoi(optarg);
366 port_dec = atoi(optarg);
374 if (argc - optind != 2) {
379 srs = srs_read_secrets(argv[optind + 1]);
384 return main_loop(srs, argv[optind], port_enc, port_dec);