citadel/domain.c

   1 /*
   2  * DNS lookup for SMTP sender
   3  *
   4  * Copyright (c) 1987-2017 by the citadel.org team
   5  *
   6  * This program is open source software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License, version 3.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  */
  14
  15 #include "sysdep.h"
  16 #include <stdio.h>
  17 #include <syslog.h>
  18 #ifdef HAVE_RESOLV_H
  19 #include <arpa/nameser.h>
  20 #ifdef HAVE_ARPA_NAMESER_COMPAT_H
  21 #include <arpa/nameser_compat.h>
  22 #endif
  23 #include <resolv.h>
  24 #endif
  25 #include <libcitadel.h>
  26 #include "sysdep_decls.h"
  27 #include "citadel.h"
  28 #include "domain.h"
  29 #include "internet_addressing.h"
  30
  31
  32 /*
  33  * get_hosts() checks the Internet configuration for various types of
  34  * entries and returns them in the same format as getmx() does -- fill the
  35  * buffer with a delimited list of hosts and return the number of hosts.
  36  *
  37  * This is used to fetch MX smarthosts, SpamAssassin hosts, etc.
  38  */
  39 int get_hosts(char *mxbuf, char *rectype) {
  40         int config_lines;
  41         int i;
  42         char buf[256];
  43         char host[256], type[256];
  44         int total_smarthosts = 0;
  45
  46         if (inetcfg == NULL) return(0);
  47         strcpy(mxbuf, "");
  48
  49         config_lines = num_tokens(inetcfg, '\n');
  50         for (i=0; i<config_lines; ++i) {
  51                 extract_token(buf, inetcfg, i, '\n', sizeof buf);
  52                 extract_token(host, buf, 0, '|', sizeof host);
  53                 extract_token(type, buf, 1, '|', sizeof type);
  54
  55                 if (!strcasecmp(type, rectype)) {
  56                         strcat(mxbuf, host);
  57                         strcat(mxbuf, "|");
  58                         ++total_smarthosts;
  59                 }
  60         }
  61
  62         return(total_smarthosts);
  63 }
  64
  65
  66 /*
  67  * Compare the preference of two MX records.  First check by the actual
  68  * number listed in the MX record.  If they're identical, randomize the
  69  * result.
  70  */
  71 int mx_compare_pref(const void *mx1, const void *mx2) {
  72         int pref1;
  73         int pref2;
  74
  75         pref1 = ((const struct mx *)mx1)->pref;
  76         pref2 = ((const struct mx *)mx2)->pref;
  77
  78         if (pref1 > pref2) {
  79                 return(1);
  80         }
  81         else if (pref1 < pref2) {
  82                 return(0);
  83         }
  84         else {
  85                 return(rand() % 2);
  86         }
  87 }
  88
  89
  90 /*
  91  * getmx()
  92  *
  93  * Return one or more MX's for a mail destination.
  94  *
  95  * Upon success, it fills 'mxbuf' with one or more MX hosts, separated by
  96  * vertical bar characters, and returns the number of hosts as its return
  97  * value.  If no MX's are found, it returns 0.
  98  *
  99  */
 100 int getmx(char *mxbuf, char *dest) {
 101
 102 #ifdef HAVE_RESOLV_H
 103         union {
 104                         u_char bytes[1024];
 105                         HEADER header;
 106     } answer;
 107 #endif
 108
 109         int ret;
 110         unsigned char *startptr, *endptr, *ptr;
 111         char expanded_buf[1024];
 112         unsigned short pref, type;
 113         int n = 0;
 114         int qdcount;
 115
 116         struct mx *mxrecs = NULL;
 117         int num_mxrecs = 0;
 118
 119         /* If we're configured to send all mail to a smart-host, then our
 120          * job here is really easy.
 121          */
 122         n = get_hosts(mxbuf, "smarthost");
 123         if (n > 0) return(n);
 124
 125         /*
 126          * No smart-host?  Look up the best MX for a site.
 127          * Make a call to the resolver library.
 128          */
 129
 130         ret = res_query(dest, C_IN, T_MX, (unsigned char *)answer.bytes, sizeof(answer));
 131
 132         if (ret < 0) {
 133                 mxrecs = malloc(sizeof(struct mx));
 134                 mxrecs[0].pref = 0;
 135                 strcpy(mxrecs[0].host, dest);
 136                 num_mxrecs = 1;
 137         }
 138         else {
 139
 140                 /* If we had to truncate, shrink the number to avoid fireworks */
 141                 if (ret > sizeof(answer)) {
 142                         ret = sizeof(answer);
 143                 }
 144
 145                 startptr = &answer.bytes[0];            // start and end of buffer
 146                 endptr = &answer.bytes[ret];
 147                 ptr = startptr + HFIXEDSZ;              // advance past header
 148
 149                 for (qdcount = ntohs(answer.header.qdcount); qdcount--; ptr += ret + QFIXEDSZ) {
 150                         if ((ret = dn_skipname(ptr, endptr)) < 0) {
 151                                 syslog(LOG_DEBUG, "domain: dn_skipname error");
 152                                 return(0);
 153                         }
 154                 }
 155
 156                 while(1) {
 157                         memset(expanded_buf, 0, sizeof(expanded_buf));
 158                         ret = dn_expand(startptr, endptr, ptr, expanded_buf, sizeof(expanded_buf));
 159                         if (ret < 0) break;
 160                         ptr += ret;
 161
 162                         GETSHORT(type, ptr);
 163                         ptr += INT16SZ + INT32SZ;
 164                         GETSHORT(n, ptr);
 165
 166                         if (type != T_MX) {
 167                                 ptr += n;
 168                         }
 169
 170                         else {
 171                                 GETSHORT(pref, ptr);
 172                                 ret = dn_expand(startptr, endptr, ptr, expanded_buf, sizeof(expanded_buf));
 173                                 ptr += ret;
 174
 175                                 ++num_mxrecs;
 176                                 if (mxrecs == NULL) {
 177                                         mxrecs = malloc(sizeof(struct mx));
 178                                 }
 179                                 else {
 180                                         mxrecs = realloc(mxrecs, (sizeof(struct mx) * num_mxrecs) );
 181                                 }
 182
 183                                 mxrecs[num_mxrecs - 1].pref = pref;
 184                                 strcpy(mxrecs[num_mxrecs - 1].host, expanded_buf);
 185                         }
 186                 }
 187         }
 188
 189         /* Sort the MX records by preference */
 190         if (num_mxrecs > 1) {
 191                 qsort(mxrecs, num_mxrecs, sizeof(struct mx), mx_compare_pref);
 192         }
 193
 194         strcpy(mxbuf, "");
 195         for (n=0; n<num_mxrecs; ++n) {
 196                 strcat(mxbuf, mxrecs[n].host);
 197                 strcat(mxbuf, "|");
 198         }
 199         free(mxrecs);
 200
 201         /*
 202          * Append any fallback smart hosts we have configured.
 203          */
 204         num_mxrecs += get_hosts(&mxbuf[strlen(mxbuf)], "fallbackhost");
 205         return(num_mxrecs);
 206 }