Init: first commit 🎉

proxy/http.go

@@ -0,0 +1,120 @@
+package proxy
+
+import (
+	"bufio"
+	"bytes"
+	"crypto/tls"
+	"fmt"
+	"io"
+	"net"
+	"net/http"
+	"net/http/httputil"
+	"strings"
+
+	"github.com/Dreamacro/clash/constant"
+
+	"github.com/riobard/go-shadowsocks2/socks"
+	log "github.com/sirupsen/logrus"
+)
+
+func NewHttpProxy(port string) {
+	server := &http.Server{
+		Addr: fmt.Sprintf(":%s", port),
+		Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			if r.Method == http.MethodConnect {
+				handleTunneling(w, r)
+			} else {
+				handleHTTP(w, r)
+			}
+		}),
+		// Disable HTTP/2.
+		TLSNextProto: make(map[string]func(*http.Server, *tls.Conn, http.Handler)),
+	}
+	log.Infof("HTTP proxy :%s", port)
+	server.ListenAndServe()
+}
+
+func handleHTTP(w http.ResponseWriter, r *http.Request) {
+	buf, _ := httputil.DumpRequestOut(r, true)
+	hijacker, ok := w.(http.Hijacker)
+	if !ok {
+		return
+	}
+	conn, rw, err := hijacker.Hijack()
+	if err != nil {
+		return
+	}
+	addr := r.Host
+	// padding default port
+	if !strings.Contains(addr, ":") {
+		addr += ":80"
+	}
+	tun.Add(NewHttp(addr, conn, rw, buf))
+}
+
+func handleTunneling(w http.ResponseWriter, r *http.Request) {
+	w.WriteHeader(http.StatusOK)
+	hijacker, ok := w.(http.Hijacker)
+	if !ok {
+		return
+	}
+	conn, rw, err := hijacker.Hijack()
+	if err != nil {
+		return
+	}
+	tun.Add(NewHttp(r.Host, conn, rw, []byte{}))
+}
+
+type HttpAdapter struct {
+	addr *constant.Addr
+	conn net.Conn
+	rw   *bufio.ReadWriter
+	r    io.Reader
+}
+
+func (h *HttpAdapter) Writer() io.Writer {
+	return h.conn
+}
+
+func (h *HttpAdapter) Reader() io.Reader {
+	return h.r
+}
+
+func (h *HttpAdapter) Close() {
+	h.conn.Close()
+}
+
+func (h *HttpAdapter) Addr() *constant.Addr {
+	return h.addr
+}
+
+func parseHttpAddr(target string) *constant.Addr {
+	host, port, _ := net.SplitHostPort(target)
+
+	var addType int
+	ip := net.ParseIP(host)
+	switch {
+	case ip == nil:
+		addType = socks.AtypDomainName
+	case ip.To4() == nil:
+		addType = socks.AtypIPv6
+	default:
+		addType = socks.AtypIPv4
+	}
+
+	return &constant.Addr{
+		AddrType: addType,
+		Host:     host,
+		Port:     port,
+	}
+}
+
+func NewHttp(host string, conn net.Conn, rw *bufio.ReadWriter, payload []byte) *HttpAdapter {
+	r := io.MultiReader(bytes.NewReader(payload), rw)
+	return &HttpAdapter{
+		conn: conn,
+		addr: parseHttpAddr(host),
+		rw:   rw,
+		r:    r,
+	}
+}

proxy/socks.go

@@ -0,0 +1,93 @@
+package proxy
+
+import (
+	"fmt"
+	"io"
+	"net"
+	"strconv"
+
+	"github.com/Dreamacro/clash/constant"
+	"github.com/Dreamacro/clash/tunnel"
+
+	"github.com/riobard/go-shadowsocks2/socks"
+	log "github.com/sirupsen/logrus"
+)
+
+var (
+	tun = tunnel.GetInstance()
+)
+
+func NewSocksProxy(port string) {
+	l, err := net.Listen("tcp", fmt.Sprintf(":%s", port))
+	defer l.Close()
+	if err != nil {
+		return
+	}
+	log.Infof("SOCKS proxy :%s", port)
+	for {
+		c, err := l.Accept()
+		if err != nil {
+			continue
+		}
+		go handleSocks(c)
+	}
+}
+
+func handleSocks(conn net.Conn) {
+	target, err := socks.Handshake(conn)
+	if err != nil {
+
+	}
+	conn.(*net.TCPConn).SetKeepAlive(true)
+	tun.Add(NewSocks(target, conn))
+}
+
+type SocksAdapter struct {
+	conn net.Conn
+	addr *constant.Addr
+}
+
+func (s *SocksAdapter) Writer() io.Writer {
+	return s.conn
+}
+
+func (s *SocksAdapter) Reader() io.Reader {
+	return s.conn
+}
+
+func (s *SocksAdapter) Close() {
+	s.conn.Close()
+}
+
+func (s *SocksAdapter) Addr() *constant.Addr {
+	return s.addr
+}
+
+func parseSocksAddr(target socks.Addr) *constant.Addr {
+	var host, port string
+
+	switch target[0] {
+	case socks.AtypDomainName:
+		host = string(target[2 : 2+target[1]])
+		port = strconv.Itoa((int(target[2+target[1]]) << 8) | int(target[2+target[1]+1]))
+	case socks.AtypIPv4:
+		host = net.IP(target[1 : 1+net.IPv4len]).String()
+		port = strconv.Itoa((int(target[1+net.IPv4len]) << 8) | int(target[1+net.IPv4len+1]))
+	case socks.AtypIPv6:
+		host = net.IP(target[1 : 1+net.IPv6len]).String()
+		port = strconv.Itoa((int(target[1+net.IPv6len]) << 8) | int(target[1+net.IPv6len+1]))
+	}
+
+	return &constant.Addr{
+		AddrType: int(target[0]),
+		Host:     host,
+		Port:     port,
+	}
+}
+
+func NewSocks(target socks.Addr, conn net.Conn) *SocksAdapter {
+	return &SocksAdapter{
+		conn: conn,
+		addr: parseSocksAddr(target),
+	}
+}