v1.2

Author: ilanyu

handle.go

@@ -5,19 +5,45 @@ import (
 	"net/url"
 	"net/http/httputil"
 	"log"
+	"net"
+	"context"
+	"time"
+	"io/ioutil"
 )
 
 type handle struct {
 	reverseProxy string
 }
 
 func (this *handle) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	log.Println(r.RemoteAddr + " " + r.Method + " " + r.URL.String() + " " + r.Proto + " " + r.UserAgent())
 	remote, err := url.Parse(this.reverseProxy)
 	if err != nil {
 		log.Fatalln(err)
 	}
+
+	dialer := &net.Dialer{
+		Timeout:   30 * time.Second,
+		KeepAlive: 30 * time.Second,
+		DualStack: true,
+	}
+	http.DefaultTransport.(*http.Transport).DialContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
+		if addr == "idea.lanyus.com:80" {
+			resp, err := http.Get("http://119.29.29.29/d?dn=idea.lanyus.com")
+			if err != nil {
+				log.Println(err)
+			}
+			defer resp.Body.Close()
+			res, err := ioutil.ReadAll(resp.Body)
+			if err != nil {
+				log.Println(err)
+			}
+			addr = string(res) + ":80"
+		}
+
+		return dialer.DialContext(ctx, network, addr)
+	}
 	proxy := httputil.NewSingleHostReverseProxy(remote)
 	r.Host = remote.Host
 	proxy.ServeHTTP(w, r)
-	log.Println(r.RemoteAddr + " " + r.Method + " " + r.URL.String() + " " + r.Proto + " " + r.UserAgent())
 }

main.go

@@ -15,6 +15,7 @@ func StartServer(bind string, remote string)  {
 	h := &handle{reverseProxy: remote}
 	srv.Addr = bind
 	srv.Handler = h
+
 	go func() {
 		if err := srv.ListenAndServe(); err != nil {
 			log.Fatalln("ListenAndServe: ", err)
@@ -33,7 +34,7 @@ func main() {
 	StartServer(cmd.bind, cmd.remote)
 	reader := bufio.NewReader(os.Stdin)
 	for {
-		str , err := reader.ReadString('\n')
+		str, err := reader.ReadString('\n')
 		if err != nil {
 			log.Println(err)
 		}