X-Git-Url: https://code.citadel.org/?a=blobdiff_plain;f=citadel%2Fdomain.c;h=3d055c671a55e7d53be9c9cc2e6d631d5a150607;hb=f0e4b5bde1396c4282a085d3e2af12ce9c0cf223;hp=69376cbb0b5d16a541c47479917a3b1725704f10;hpb=1f79d7b9a6f60f331ea0c90e9eb4ef7c40ee7aa5;p=citadel.git

diff --git a/citadel/domain.c b/citadel/domain.c
index 69376cbb0..3d055c671 100644
--- a/citadel/domain.c
+++ b/citadel/domain.c
@@ -1,35 +1,46 @@
 /*
- * $Id$
- *
  * DNS lookup for SMTP sender
  *
+ * Copyright (c) 1987-2011 by the citadel.org team
+ *
+ * This program is open source software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License, version 3.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
  */
 
-#include <stdlib.h>
-#include <unistd.h>
-#include <string.h>
-#include <netinet/in.h>
+#include "sysdep.h"
 #include <stdio.h>
+#include <syslog.h>
+#ifdef HAVE_RESOLV_H
 #include <arpa/nameser.h>
+#ifdef HAVE_ARPA_NAMESER_COMPAT_H
+#include <arpa/nameser_compat.h>
+#endif
 #include <resolv.h>
+#endif
+#include <libcitadel.h>
 #include "sysdep_decls.h"
 #include "citadel.h"
 #include "domain.h"
-#include "server.h"
-#include "tools.h"
 #include "internet_addressing.h"
 
 
 /*
- * get_smarthosts() checks the Internet configuration for "smarthost"
+ * get_hosts() checks the Internet configuration for various types of
  * entries and returns them in the same format as getmx() does -- fill the
  * buffer with a delimited list of hosts and return the number of hosts.
+ * 
+ * This is used to fetch MX smarthosts, SpamAssassin hosts, etc.
  */
-int get_smarthosts(char *mxbuf) {
+int get_hosts(char *mxbuf, char *rectype) {
 	int config_lines;
 	int i;
-	char buf[SIZ];
-	char host[SIZ], type[SIZ];
+	char buf[256];
+	char host[256], type[256];
 	int total_smarthosts = 0;
 
 	if (inetcfg == NULL) return(0);
@@ -37,11 +48,11 @@ int get_smarthosts(char *mxbuf) {
 
 	config_lines = num_tokens(inetcfg, '\n');
 	for (i=0; i<config_lines; ++i) {
-		extract_token(buf, inetcfg, i, '\n');
-		extract_token(host, buf, 0, '|');
-		extract_token(type, buf, 1, '|');
+		extract_token(buf, inetcfg, i, '\n', sizeof buf);
+		extract_token(host, buf, 0, '|', sizeof host);
+		extract_token(type, buf, 1, '|', sizeof type);
 
-		if (!strcasecmp(type, "smarthost")) {
+		if (!strcasecmp(type, rectype)) {
 			strcat(mxbuf, host);
 			strcat(mxbuf, "|");
 			++total_smarthosts;
@@ -52,36 +63,30 @@ int get_smarthosts(char *mxbuf) {
 }
 
 
-
-
 /*
- * sort_mxrecs()
- *
- * Sort a pile of MX records (struct mx, definted in domain.h) by preference
- *
+ * Compare the preference of two MX records.  First check by the actual
+ * number listed in the MX record.  If they're identical, randomize the
+ * result.
  */
-void sort_mxrecs(struct mx *mxrecs, int num_mxrecs) {
-	int a, b;
-	struct mx hold1, hold2;
-
-	if (num_mxrecs < 2) return;
+int mx_compare_pref(const void *mx1, const void *mx2) {
+	int pref1;
+	int pref2;
 
-	/* do the sort */
-	for (a = num_mxrecs - 2; a >= 0; --a) {
-		for (b = 0; b <= a; ++b) {
-			if (mxrecs[b].pref > mxrecs[b+1].pref) {
+	pref1 = ((const struct mx *)mx1)->pref;
+	pref2 = ((const struct mx *)mx2)->pref;
 
-				memcpy(&hold1, &mxrecs[b], sizeof(struct mx));
-				memcpy(&hold2, &mxrecs[b+1], sizeof(struct mx));
-				memcpy(&mxrecs[b], &hold2, sizeof(struct mx));
-				memcpy(&mxrecs[b+1], &hold1, sizeof(struct mx));
-			}
-		}
+	if (pref1 > pref2) {
+		return(1);
+	}
+	else if (pref1 < pref2) {
+		return(0);
+	}
+	else {
+		return(rand() % 2);
 	}
 }
 
 
-
 /* 
  * getmx()
  *
@@ -93,10 +98,14 @@ void sort_mxrecs(struct mx *mxrecs, int num_mxrecs) {
  *
  */
 int getmx(char *mxbuf, char *dest) {
+
+#ifdef HAVE_RESOLV_H
 	union {
 			u_char bytes[1024];
 			HEADER header;
     } answer;
+#endif
+
 	int ret;
 	unsigned char *startptr, *endptr, *ptr;
 	char expanded_buf[1024];
@@ -110,18 +119,20 @@ int getmx(char *mxbuf, char *dest) {
 	/* If we're configured to send all mail to a smart-host, then our
 	 * job here is really easy.
 	 */
-	n = get_smarthosts(mxbuf);
+	n = get_hosts(mxbuf, "smarthost");
 	if (n > 0) return(n);
 
 	/*
 	 * No smart-host?  Look up the best MX for a site.
+	 * Make a call to the resolver library.
 	 */
+
 	ret = res_query(
 		dest,
 		C_IN, T_MX, (unsigned char *)answer.bytes, sizeof(answer)  );
 
 	if (ret < 0) {
-		mxrecs = mallok(sizeof(struct mx));
+		mxrecs = malloc(sizeof(struct mx));
 		mxrecs[0].pref = 0;
 		strcpy(mxrecs[0].host, dest);
 		num_mxrecs = 1;
@@ -138,7 +149,7 @@ int getmx(char *mxbuf, char *dest) {
 	
 		for (qdcount = ntohs(answer.header.qdcount); qdcount--; ptr += ret + QFIXEDSZ) {
 			if ((ret = dn_skipname(ptr, endptr)) < 0) {
-				lprintf(9, "dn_skipname error\n");
+				syslog(LOG_DEBUG, "dn_skipname error\n");
 				return(0);
 			}
 		}
@@ -174,10 +185,10 @@ int getmx(char *mxbuf, char *dest) {
 	
 				++num_mxrecs;
 				if (mxrecs == NULL) {
-					mxrecs = mallok(sizeof(struct mx));
+					mxrecs = malloc(sizeof(struct mx));
 				}
 				else {
-					mxrecs = reallok(mxrecs,
+					mxrecs = realloc(mxrecs,
 					    (sizeof(struct mx) * num_mxrecs) );
 				}
 	
@@ -188,13 +199,20 @@ int getmx(char *mxbuf, char *dest) {
 		}
 	}
 
-	sort_mxrecs(mxrecs, num_mxrecs);
+	/* Sort the MX records by preference */
+	if (num_mxrecs > 1) {
+		qsort(mxrecs, num_mxrecs, sizeof(struct mx), mx_compare_pref);
+	}
 
 	strcpy(mxbuf, "");
 	for (n=0; n<num_mxrecs; ++n) {
 		strcat(mxbuf, mxrecs[n].host);
 		strcat(mxbuf, "|");
 	}
-	phree(mxrecs);
+	free(mxrecs);
+
+	/* Append any fallback smart hosts we have configured.
+	 */
+	num_mxrecs += get_hosts(&mxbuf[strlen(mxbuf)], "fallbackhost");
 	return(num_mxrecs);
 }