opt sth

proxy/proxy.go

@@ -18,11 +18,15 @@ var (
 	hijackDNS []string
 )
 
-// Register Monitor
 func RegisterMonitor(m S.Monitor) {
 	monitor = m
 }
 
+func RegisterFakeDNS(d D.FakeDNS, h string) {
+	fakeDNS = d
+	hijackDNS = append(hijackDNS, strings.Split(h, ",")...)
+}
+
 // Session Operation
 func addSession(key interface{}, session *S.Session) {
 	if monitor != nil {
@@ -36,12 +40,6 @@ func removeSession(key interface{}) {
 	}
 }
 
-// Register FakeDNS
-func RegisterFakeDNS(d D.FakeDNS, h string) {
-	fakeDNS = d
-	hijackDNS = append(hijackDNS, strings.Split(h, ",")...)
-}
-
 // Check target if is hijacked address.
 func isHijacked(target *net.UDPAddr) bool {
 	if fakeDNS == nil {

proxy/udp.go

@@ -120,13 +120,13 @@ func (h *udpHandler) ReceiveTo(conn core.UDPConn, data []byte, addr *net.UDPAddr
 		resp, err := fakeDNS.Resolve(data)
 		if err != nil {
 			return fmt.Errorf("hijack DNS request error: %v", err)
-		} else {
-			if _, err = conn.WriteFrom(resp, addr); err != nil {
-				return fmt.Errorf("write dns answer failed: %v", err)
-			}
-			h.Close(conn)
-			return nil
 		}
+
+		if _, err = conn.WriteFrom(resp, addr); err != nil {
+			return fmt.Errorf("write dns answer failed: %v", err)
+		}
+		h.Close(conn)
+		return nil
 	}
 
 	var remoteAddr net.Addr