1 /******************************************************************************/
2 /* pfixtools: a collection of postfix related tools */
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
34 * Copyright © 2008 Florent Bruneau
46 #define DAEMON_NAME "pfix-srsd"
47 #define DEFAULT_ENCODER_PORT 10001
48 #define DEFAULT_DECODER_PORT 10002
49 #define RUNAS_USER "nobody"
50 #define RUNAS_GROUP "nogroup"
54 typedef struct srs_config_t {
63 static const char* const decoder_ptr = "decoder";
64 static const char* const encoder_ptr = "encoder";
66 static void *srsd_new_decoder(void)
68 return (void*)decoder_ptr;
71 static void *srsd_new_encoder(void)
73 return (void*)encoder_ptr;
76 static void *srsd_starter(server_t *server)
81 int start_listener(int port, bool decoder)
83 return start_server(port, decoder ? srsd_new_decoder : srsd_new_encoder, NULL);
90 void urldecode(char *s, char *end)
95 if (*p == '%' && end - p >= 3) {
96 int h = (hexval(p[1]) << 4) | hexval(p[2]);
110 int process_srs(server_t *srsd, void* vconfig)
112 srs_config_t* config = vconfig;
113 int res = buffer_read(&srsd->ibuf, srsd->fd, -1);
115 if ((res < 0 && errno != EINTR && errno != EAGAIN) || res == 0)
118 while (srsd->ibuf.len > 4) {
119 char buf[BUFSIZ], *p, *q, *nl;
122 nl = strchr(srsd->ibuf.data + 4, '\n');
124 if (srsd->ibuf.len > BUFSIZ) {
125 syslog(LOG_ERR, "unreasonnable amount of data without a \\n");
128 if (srsd->obuf.len) {
129 epoll_modify(srsd->fd, EPOLLIN | EPOLLOUT, srsd);
134 if (strncmp("get ", srsd->ibuf.data, 4)) {
135 syslog(LOG_ERR, "bad request, not starting with \"get \"");
139 for (p = srsd->ibuf.data + 4; p < nl && isspace(*p); p++);
140 for (q = nl++; q >= p && isspace(*q); *q-- = '\0');
143 buffer_addstr(&srsd->obuf, "400 empty request ???\n");
144 syslog(LOG_WARNING, "empty request");
150 if (srsd->data == (void*)decoder_ptr) {
151 err = srs_reverse(config->srs, buf, ssizeof(buf), p);
153 err = srs_forward(config->srs, buf, ssizeof(buf), p, config->domain);
157 buffer_addstr(&srsd->obuf, "200 ");
158 buffer_addstr(&srsd->obuf, buf);
160 switch (SRS_ERROR_TYPE(err)) {
161 case SRS_ERRTYPE_SRS:
162 case SRS_ERRTYPE_SYNTAX:
163 buffer_addstr(&srsd->obuf, "500 ");
166 buffer_addstr(&srsd->obuf, "400 ");
169 buffer_addstr(&srsd->obuf, srs_strerror(err));
171 buffer_addch(&srsd->obuf, '\n');
174 buffer_consume(&srsd->ibuf, nl - srsd->ibuf.data);
176 if (srsd->obuf.len) {
177 epoll_modify(srsd->fd, EPOLLIN | EPOLLOUT, srsd);
186 static srs_config_t config = {
191 /** overload srs_free since the lib is not properly maintained.
193 inline void srs_free(srs_t* srs)
196 for (i = 0; i < srs->numsecrets; i++) {
197 memset(srs->secrets[i], 0, strlen(srs->secrets[i]));
198 free(srs->secrets[i]);
199 srs->secrets[i] = '\0';
207 static void config_shutdown(void)
210 srs_free(config.srs);
215 module_exit(config_shutdown);
217 static srs_t *srs_read_secrets(const char *sfile)
224 f = fopen(sfile, "r");
232 while (fgets(buf, sizeof(buf), f)) {
236 if (n == sizeof(buf) - 1 && buf[n - 1] != '\n') {
237 syslog(LOG_CRIT, "%s:%d: line too long", sfile, lineno);
241 srs_add_secret(srs, skipspaces(buf));
245 syslog(LOG_CRIT, "%s: empty file, no secrets", sfile);
258 /* administrivia {{{1
263 fputs("usage: "DAEMON_NAME" [options] domain secrets\n"
266 " -e <port> port to listen to for encoding requests\n"
267 " (default: "STR(DEFAULT_ENCODER_PORT)")\n"
268 " -d <port> port to listen to for decoding requests\n"
269 " (default: "STR(DEFAULT_DECODER_PORT)")\n"
270 " -p <pidfile> file to write our pid to\n"
271 " -u unsafe mode: don't drop privilegies\n"
272 " -f stay in foreground\n"
279 int main(int argc, char *argv[])
282 bool daemonize = true;
283 int port_enc = DEFAULT_ENCODER_PORT;
284 int port_dec = DEFAULT_DECODER_PORT;
285 const char *pidfile = NULL;
287 for (int c = 0; (c = getopt(argc, argv, "hfu" "e:d:p:")) >= 0; ) {
290 port_enc = atoi(optarg);
296 port_dec = atoi(optarg);
310 if (argc - optind != 2) {
315 config.domain = argv[optind];
316 config.srs = srs_read_secrets(argv[optind + 1]);
318 || common_setup(pidfile, unsafe, RUNAS_USER, RUNAS_GROUP,
319 daemonize) != EXIT_SUCCESS
320 || start_listener(port_enc, false) < 0
321 || start_listener(port_dec, true) < 0) {
324 return server_loop(srsd_starter, NULL, process_srs, NULL, &config);