Allow client to issue more numerous requests on the same connection

mxrouter.go

@@ -30,7 +30,7 @@ import (
 )
 
 const (
-	version = "0.9.6"
+	version = "0.9.7"
 )
 
 var (
@@ -70,15 +70,31 @@ func ldapSearch(searchReq *ldap.SearchRequest, attempt int) (*ldap.SearchResult,
 	return result, err
 }
 
+func sendResponse(con net.Conn, respMsg string, respCode int) error {
+	response := fmt.Sprintf("%d %s\n", respCode, strings.Replace(respMsg, " ", "%20", -1))
+	_, err := con.Write([]byte(response))
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
 func handleConnection(connClt net.Conn, conLdap *ldap.Conn) {
 	var s string
 	var mxdom string
 	buf := make([]byte, 1024)
 
-	defer connClt.Close()
+	// It's client responsability to close connection
+	//defer connClt.Close()
+	for {
 		readlen, err := connClt.Read(buf)
 		if err != nil {
-		fmt.Println("Error reading:", err.Error())
+			// Dont notice if client closed connection
+			if err.Error() != "EOF" {
+				logstream.Err("Error reading:", err.Error())
+			}
+			return
 		}
 
 		// 1) Recupere le MX du domaine de l'adresse mail recue
@@ -88,20 +104,18 @@ func handleConnection(connClt net.Conn, conLdap *ldap.Conn) {
 			mxs, err := net.LookupMX(domain)
 			if err != nil {
 				logstream.Err(fmt.Sprintln("Error lookup mx: ", err))
-			s := strings.Replace(fmt.Sprintf("Error lookup MX: %s", err.Error()), " ", "%20", -1)
-			response := fmt.Sprintf("500 %s\n", s)
-			connClt.Write([]byte(response))
-			return
+				s := fmt.Sprintf("Error lookup MX: %s", err.Error())
+				sendResponse(connClt, s, 500)
+				continue
 			}
 			// 2) Requete LDAP pour voir si le domaine de second niveau du MX possede un routage particulier
 			// Protection contre ca : example.org.            72      IN      MX      0 .
 			// Considerons qu'il n'y aura jamais de "one letter TLD"
 			if len(mxs[0].Host) < 4 {
 				logstream.Err(fmt.Sprintln("No usable mx found"))
-			s := strings.Replace("No usable MX found", " ", "%20", -1)
-			resp := fmt.Sprintf("500 %s\n", s)
-			connClt.Write([]byte(resp))
-			return
+				s := "No usable MX found"
+				sendResponse(connClt, s, 500)
+				continue
 			}
 
 			mxslic := strings.Split(mxs[0].Host, ".")
@@ -119,29 +133,26 @@ func handleConnection(connClt net.Conn, conLdap *ldap.Conn) {
 			result, err := ldapSearch(searchReq, 0)
 			if err != nil {
 				logstream.Err(fmt.Sprintln("Error searching into LDAP: ", err))
-			s := strings.Replace(err.Error(), " ", "%20", -1)
-			response := fmt.Sprintf("500 %s\n", s)
-			connClt.Write([]byte(response))
-			return
+				s := err.Error()
+				sendResponse(connClt, s, 500)
+				continue
 			}
 
 			if len(result.Entries) != 1 {
 				if *debug {
 					logstream.Debug("Got no result, returning 500")
 				}
-			s := strings.Replace("No route defined", " ", "%20", -1)
-			response := fmt.Sprintf("500 %s\n", s)
-			connClt.Write([]byte(response))
-			return
+				s := "No route defined"
+				sendResponse(connClt, s, 500)
+				continue
 			}
 
 			// L'attribut n'existe pas
 			if len(result.Entries[0].Attributes) == 0 {
 				logstream.Err(fmt.Sprintf("Error searching into LDAP: Attribute %s not found for entry %s\n", *ldapAttr, result.Entries[0]))
-			s := strings.Replace(fmt.Sprintf("Attribute not found: %s", *ldapAttr), " ", "%20", -1)
-			response := fmt.Sprintf("500 %s\n", s)
-			connClt.Write([]byte(response))
-			return
+				s := fmt.Sprintf("Attribute not found: %s", *ldapAttr)
+				sendResponse(connClt, s, 500)
+				continue
 			}
 
 			if *debug {
@@ -149,19 +160,19 @@ func handleConnection(connClt net.Conn, conLdap *ldap.Conn) {
 			}
 			// Check if result is a FQDN, if not append defDomain
 			if strings.Contains(string(result.Entries[0].Attributes[0].Values[0]), ".") {
-			s = strings.Replace(fmt.Sprintf("FILTER relay:[%s]", result.Entries[0].Attributes[0].Values[0]), " ", "%20", -1)
+				s = fmt.Sprintf("FILTER relay:[%s]", result.Entries[0].Attributes[0].Values[0])
 			} else {
-			s = strings.Replace(fmt.Sprintf("FILTER relay:[%s]", fmt.Sprintf("%s.%s", result.Entries[0].Attributes[0].Values[0], *defDomain)), " ", "%20", -1)
+				s = fmt.Sprintf("FILTER relay:[%s]", fmt.Sprintf("%s.%s", result.Entries[0].Attributes[0].Values[0], *defDomain))
 			}
-		connClt.Write([]byte(fmt.Sprintf("200 %s\n", s)))
+			sendResponse(connClt, s, 200)
 		} else {
 			if *debug {
 				logstream.Debug("Incorrect input format : " + string(buf[:readlen-1]))
 			}
-		s := strings.Replace("Incorrect input format", " ", "%20", -1)
-		response := fmt.Sprintf("500 %s\n", s)
-		connClt.Write([]byte(response))
-		return
+			s := "Incorrect input format"
+			sendResponse(connClt, s, 500)
+			continue
+		}
 	}
 }
 
@@ -217,7 +228,7 @@ func main() {
 	ldapPass = fs.String("ldapPass", "", "LDAP user password (also via LDAPPASS env var)")
 	ldapAttr = fs.String("ldapAttr", "relayName", "LDAP attribute containing the relay name to return")
 	defDomain = fs.String("domain", "", "Domain to add to relay name if not a FQDN (also via DOMAIN env var)")
-	pidFilePath= fs.String("pidfile", "/var/run/mxrouter/mxrouter.pid", "PID File (also via PIDFILE env var)")
+	pidFilePath = fs.String("pidfile", "/var/run/mxrouter/mxrouter.pid", "PID File (also via PIDFILE env var)")
 	_ = fs.String("config", "", "config file (optional)")
 
 	// Surcharge de la fonction Usage()