mirror of
https://github.com/nadoo/glider.git
synced 2025-02-23 17:35:40 +08:00
127 lines
2.2 KiB
Go
127 lines
2.2 KiB
Go
package mixed
|
|
|
|
import (
|
|
"bytes"
|
|
"net"
|
|
"net/url"
|
|
|
|
"github.com/nadoo/glider/common/conn"
|
|
"github.com/nadoo/glider/common/log"
|
|
"github.com/nadoo/glider/proxy"
|
|
"github.com/nadoo/glider/proxy/http"
|
|
"github.com/nadoo/glider/proxy/socks5"
|
|
)
|
|
|
|
// https://www.ietf.org/rfc/rfc2616.txt, http methods must be uppercase
|
|
var httpMethods = [...][]byte{
|
|
[]byte("GET"),
|
|
[]byte("POST"),
|
|
[]byte("PUT"),
|
|
[]byte("DELETE"),
|
|
[]byte("CONNECT"),
|
|
[]byte("HEAD"),
|
|
[]byte("OPTIONS"),
|
|
[]byte("TRACE"),
|
|
}
|
|
|
|
// MixedProxy struct
|
|
type MixedProxy struct {
|
|
dialer proxy.Dialer
|
|
addr string
|
|
|
|
http *http.HTTP
|
|
socks5 *socks5.SOCKS5
|
|
}
|
|
|
|
func init() {
|
|
proxy.RegisterServer("mixed", NewMixedProxyServer)
|
|
}
|
|
|
|
// NewMixedProxy returns a mixed proxy
|
|
func NewMixedProxy(s string, dialer proxy.Dialer) (*MixedProxy, error) {
|
|
u, err := url.Parse(s)
|
|
if err != nil {
|
|
log.F("parse err: %s", err)
|
|
return nil, err
|
|
}
|
|
|
|
p := &MixedProxy{
|
|
dialer: dialer,
|
|
addr: u.Host,
|
|
}
|
|
|
|
p.http, _ = http.NewHTTP(s, dialer)
|
|
p.socks5, _ = socks5.NewSOCKS5(s, dialer)
|
|
|
|
return p, nil
|
|
}
|
|
|
|
// NewMixedProxyServer returns a mixed proxy server
|
|
func NewMixedProxyServer(s string, dialer proxy.Dialer) (proxy.Server, error) {
|
|
return NewMixedProxy(s, dialer)
|
|
}
|
|
|
|
// ListenAndServe .
|
|
func (p *MixedProxy) ListenAndServe() {
|
|
go p.socks5.ListenAndServeUDP()
|
|
|
|
l, err := net.Listen("tcp", p.addr)
|
|
if err != nil {
|
|
log.F("[mixed] failed to listen on %s: %v", p.addr, err)
|
|
return
|
|
}
|
|
|
|
log.F("[mixed] listening TCP on %s", p.addr)
|
|
|
|
for {
|
|
c, err := l.Accept()
|
|
if err != nil {
|
|
log.F("[mixed] failed to accept: %v", err)
|
|
continue
|
|
}
|
|
|
|
go p.Serve(c)
|
|
}
|
|
}
|
|
|
|
// Serve .
|
|
func (p *MixedProxy) Serve(c net.Conn) {
|
|
defer c.Close()
|
|
|
|
if c, ok := c.(*net.TCPConn); ok {
|
|
c.SetKeepAlive(true)
|
|
}
|
|
|
|
cc := conn.NewConn(c)
|
|
|
|
if p.socks5 != nil {
|
|
head, err := cc.Peek(1)
|
|
if err != nil {
|
|
log.F("[mixed] socks5 peek error: %s", err)
|
|
return
|
|
}
|
|
|
|
// check socks5, client send socksversion: 5 as the first byte
|
|
if head[0] == socks5.Version {
|
|
p.socks5.Serve(cc)
|
|
return
|
|
}
|
|
}
|
|
|
|
if p.http != nil {
|
|
head, err := cc.Peek(8)
|
|
if err != nil {
|
|
log.F("[mixed] http peek error: %s", err)
|
|
return
|
|
}
|
|
|
|
for _, method := range httpMethods {
|
|
if bytes.HasPrefix(head, method) {
|
|
p.http.Serve(cc)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|