sower/proxy/server.go

package proxy

import (
	"net"
	"strings"

	"github.com/golang/glog"
	"github.com/wweir/sower/proxy/parser"
	"github.com/wweir/sower/proxy/shadow"
	"github.com/wweir/sower/proxy/transport"
)

func StartServer(tran transport.Transport, port, cipher, password string) {
	if port == "" {
		glog.Fatalln("port must set")
	}
	if !strings.HasPrefix(port, ":") {
		port = ":" + port
	}

	connCh, err := tran.Listen(port)
	if err != nil {
		glog.Fatalf("listen %v fail: %s", port, err)
	}

	glog.Infoln("Server started.")
	for {
		go handle(<-connCh, cipher, password)
	}
}

func handle(conn net.Conn, cipher, password string) {
	conn = shadow.Shadow(conn, cipher, password)
	conn, host, port, err := parser.ParseAddr(conn)
	if err != nil {
		conn.Close()
		glog.Warningln(err)
		return
	}
	glog.V(1).Infof("new conn from %s to %s:%s", conn.RemoteAddr(), host, port)

	rc, err := net.Dial("tcp", net.JoinHostPort(host, port))
	if err != nil {
		conn.Close()
		glog.Warningln(err)
		return
	}
	rc.(*net.TCPConn).SetKeepAlive(true)

	relay(rc, conn)
}