1 a248a3c1 2019-03-27 martijn #include <sys/socket.h>
3 a248a3c1 2019-03-27 martijn #include <err.h>
4 a248a3c1 2019-03-27 martijn #include <errno.h>
5 a248a3c1 2019-03-27 martijn #include <netdb.h>
6 a248a3c1 2019-03-27 martijn #include <stdlib.h>
7 a248a3c1 2019-03-27 martijn #include <stdio.h>
8 a248a3c1 2019-03-27 martijn #include <string.h>
9 a248a3c1 2019-03-27 martijn #include <unistd.h>
11 a248a3c1 2019-03-27 martijn #include "smtp_proc.h"
13 a248a3c1 2019-03-27 martijn static char **blacklists = NULL;
14 a248a3c1 2019-03-27 martijn static size_t nblacklists = 0;
16 a248a3c1 2019-03-27 martijn void usage(void);
17 a248a3c1 2019-03-27 martijn enum filter_decision dnsbl_connect(char *, int, time_t, char *, char *,
18 a248a3c1 2019-03-27 martijn uint64_t, uint64_t, struct smtp_filter_connect *);
21 a248a3c1 2019-03-27 martijn main(int argc, char *argv[])
24 a248a3c1 2019-03-27 martijn size_t i;
25 a248a3c1 2019-03-27 martijn char *line = NULL;
26 a248a3c1 2019-03-27 martijn size_t linesize = 0;
27 a248a3c1 2019-03-27 martijn ssize_t linelen;
28 a248a3c1 2019-03-27 martijn char *msgid, *token, *ip;
29 a248a3c1 2019-03-27 martijn const char *errstr;
30 a248a3c1 2019-03-27 martijn int lookup;
32 a248a3c1 2019-03-27 martijn if (pledge("stdio dns", NULL) == -1)
33 a248a3c1 2019-03-27 martijn err(1, "pledge");
35 a248a3c1 2019-03-27 martijn while ((ch = getopt(argc, argv, "b:f:r:")) != -1) {
36 a248a3c1 2019-03-27 martijn switch (ch) {
37 a248a3c1 2019-03-27 martijn case 'b':
38 a248a3c1 2019-03-27 martijn blacklists = reallocarray(blacklists, nblacklists + 1,
39 a248a3c1 2019-03-27 martijn sizeof(*blacklists));
40 a248a3c1 2019-03-27 martijn if (blacklists == NULL)
41 a248a3c1 2019-03-27 martijn err(1, NULL);
42 a248a3c1 2019-03-27 martijn if ((blacklists[nblacklists] = strdup(optarg)) == NULL)
43 a248a3c1 2019-03-27 martijn err(1, NULL);
44 a248a3c1 2019-03-27 martijn nblacklists++;
51 a248a3c1 2019-03-27 martijn if (nblacklists == 0)
52 a248a3c1 2019-03-27 martijn errx(1, "No blacklist specified");
54 a248a3c1 2019-03-27 martijn smtp_register_filter_connect(dnsbl_connect);
55 a248a3c1 2019-03-27 martijn smtp_run();
57 a248a3c1 2019-03-27 martijn return 0;
60 a248a3c1 2019-03-27 martijn enum filter_decision
61 a248a3c1 2019-03-27 martijn dnsbl_connect(char *type, int version, time_t tm, char *direction, char *phase,
62 a248a3c1 2019-03-27 martijn uint64_t reqid, uint64_t token, struct smtp_filter_connect *params)
64 a248a3c1 2019-03-27 martijn char query[255];
65 a248a3c1 2019-03-27 martijn char reply[1500];
66 a248a3c1 2019-03-27 martijn struct hostent *hent;
67 a248a3c1 2019-03-27 martijn u_char *addr;
68 57cbfccc 2019-03-27 martijn int i, try;
70 a248a3c1 2019-03-27 martijn addr = (u_char *)&(params->addr);
71 a248a3c1 2019-03-27 martijn for (i = 0; i < nblacklists; i++) {
72 a248a3c1 2019-03-27 martijn if (params->af == AF_INET) {
73 a248a3c1 2019-03-27 martijn if (snprintf(query, sizeof(query), "%u.%u.%u.%u.%s",
74 a248a3c1 2019-03-27 martijn addr[3], addr[2], addr[1], addr[0],
75 a248a3c1 2019-03-27 martijn blacklists[i]) >= sizeof(query))
76 a248a3c1 2019-03-27 martijn errx(1, "Can't create query, domain too long");
77 a248a3c1 2019-03-27 martijn } else if (params->af == AF_INET6) {
78 a248a3c1 2019-03-27 martijn if (snprintf(query, sizeof(query), "%hhx.%hhx.%hhx.%hhx"
79 a248a3c1 2019-03-27 martijn ".%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx"
80 a248a3c1 2019-03-27 martijn ".%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx"
81 a248a3c1 2019-03-27 martijn ".%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%hhx.%s",
82 a248a3c1 2019-03-27 martijn (u_char) (addr[15] & 0xf), (u_char) (addr[15] >> 4),
83 a248a3c1 2019-03-27 martijn (u_char) (addr[14] & 0xf), (u_char) (addr[14] >> 4),
84 a248a3c1 2019-03-27 martijn (u_char) (addr[13] & 0xf), (u_char) (addr[13] >> 4),
85 a248a3c1 2019-03-27 martijn (u_char) (addr[12] & 0xf), (u_char) (addr[12] >> 4),
86 a248a3c1 2019-03-27 martijn (u_char) (addr[11] & 0xf), (u_char) (addr[11] >> 4),
87 a248a3c1 2019-03-27 martijn (u_char) (addr[10] & 0xf), (u_char) (addr[10] >> 4),
88 a248a3c1 2019-03-27 martijn (u_char) (addr[9] & 0xf), (u_char) (addr[9] >> 4),
89 a248a3c1 2019-03-27 martijn (u_char) (addr[8] & 0xf), (u_char) (addr[8] >> 4),
90 a248a3c1 2019-03-27 martijn (u_char) (addr[7] & 0xf), (u_char) (addr[8] >> 4),
91 a248a3c1 2019-03-27 martijn (u_char) (addr[6] & 0xf), (u_char) (addr[7] >> 4),
92 a248a3c1 2019-03-27 martijn (u_char) (addr[5] & 0xf), (u_char) (addr[5] >> 4),
93 a248a3c1 2019-03-27 martijn (u_char) (addr[4] & 0xf), (u_char) (addr[4] >> 4),
94 a248a3c1 2019-03-27 martijn (u_char) (addr[3] & 0xf), (u_char) (addr[3] >> 4),
95 a248a3c1 2019-03-27 martijn (u_char) (addr[2] & 0xf), (u_char) (addr[2] >> 4),
96 a248a3c1 2019-03-27 martijn (u_char) (addr[1] & 0xf), (u_char) (addr[1] >> 4),
97 a248a3c1 2019-03-27 martijn (u_char) (addr[0] & 0xf), (u_char) (addr[0] >> 4),
98 a248a3c1 2019-03-27 martijn blacklists[i]) >= sizeof(query))
99 a248a3c1 2019-03-27 martijn errx(1, "Can't create query, domain too long");
101 a248a3c1 2019-03-27 martijn errx(1, "Invalid address family received");
103 05f20f1d 2019-03-28 martijn if ((hent = gethostbyname(query)) == NULL) {
104 05f20f1d 2019-03-28 martijn if (h_errno == HOST_NOT_FOUND)
106 05f20f1d 2019-03-28 martijn if (h_errno != TRY_AGAIN) {
107 57cbfccc 2019-03-27 martijn smtp_filter_disconnect(reqid, token,
108 05f20f1d 2019-03-28 martijn "Blacklist check failed");
109 57cbfccc 2019-03-27 martijn return FILTER_DISCONNECT;
111 05f20f1d 2019-03-28 martijn } else {
112 05f20f1d 2019-03-28 martijn smtp_filter_disconnect(reqid, token,
113 05f20f1d 2019-03-28 martijn "Listed at %s", blacklists[i]);
114 05f20f1d 2019-03-28 martijn return FILTER_DISCONNECT;
117 57cbfccc 2019-03-27 martijn return FILTER_PROCEED;
120 f8356fc3 2019-03-28 martijn __dead void
121 a248a3c1 2019-03-27 martijn usage(void)
123 a248a3c1 2019-03-27 martijn fprintf(stderr, "usage: %s [-b blacklist]\n",
124 a248a3c1 2019-03-27 martijn getprogname());
125 a248a3c1 2019-03-27 martijn exit(1);
