You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
139 lines
2.8 KiB
Go
139 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"github.com/fangdingjun/obfssh"
|
|
socks "github.com/fangdingjun/socks-go"
|
|
"net"
|
|
"net/url"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func updateProxyFromEnv(cfg *config) {
|
|
if cfg.Proxy.Scheme != "" && cfg.Proxy.Host != "" && cfg.Proxy.Port != 0 {
|
|
obfssh.Log(obfssh.DEBUG, "proxy already specified by config, not parse environment proxy")
|
|
return
|
|
}
|
|
proxyStr := os.Getenv("http_proxy")
|
|
if proxyStr == "" {
|
|
proxyStr = os.Getenv("https_proxy")
|
|
}
|
|
if proxyStr == "" {
|
|
return
|
|
}
|
|
u, err := url.Parse(proxyStr)
|
|
if err != nil {
|
|
obfssh.Log(obfssh.DEBUG, "parse proxy from environment failed: %s", err)
|
|
return
|
|
}
|
|
cfg.Proxy.Scheme = u.Scheme
|
|
host, port, err := net.SplitHostPort(u.Host)
|
|
if err != nil {
|
|
cfg.Proxy.Host = host
|
|
cfg.Proxy.Port = 8080
|
|
} else {
|
|
cfg.Proxy.Host = host
|
|
p, err := strconv.ParseInt(port, 10, 32)
|
|
if err == nil {
|
|
cfg.Proxy.Port = int(p)
|
|
} else {
|
|
cfg.Proxy.Port = 8080
|
|
}
|
|
}
|
|
}
|
|
|
|
func httpProxyHandshake(c net.Conn, host string, port int) error {
|
|
fmt.Fprintf(c, "CONNECT %s:%d HTTP/1.1\r\n", host, port)
|
|
fmt.Fprintf(c, "Host: %s:%d\r\n", host, port)
|
|
fmt.Fprintf(c, "User-Agent: go/1.7\r\n")
|
|
fmt.Fprintf(c, "\r\n")
|
|
|
|
r := bufio.NewReader(c)
|
|
|
|
// read status line
|
|
statusLine, err := r.ReadString('\n')
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if statusLine[0:4] != "HTTP" {
|
|
return err
|
|
}
|
|
|
|
status := strings.Fields(statusLine)[1]
|
|
|
|
statusCode, err := strconv.ParseInt(status, 10, 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if statusCode != 200 {
|
|
return fmt.Errorf("http status error %d", statusCode)
|
|
}
|
|
|
|
// read header
|
|
for {
|
|
h, err := r.ReadString('\n')
|
|
if err != nil {
|
|
return err
|
|
}
|
|
h1 := strings.Trim(h, " \r\n")
|
|
if h1 == "" {
|
|
break
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func dialHTTPProxy(host string, port int, p proxy) (net.Conn, error) {
|
|
c, err := net.Dial("tcp", fmt.Sprintf("%s:%d", p.Host, p.Port))
|
|
if err = httpProxyHandshake(c, host, port); err != nil {
|
|
c.Close()
|
|
return nil, err
|
|
}
|
|
return c, nil
|
|
}
|
|
|
|
func dialHTTPSProxy(host string, port int, p proxy) (net.Conn, error) {
|
|
hostname := p.Host
|
|
if p.SNI != "" {
|
|
hostname = p.SNI
|
|
}
|
|
tlsconfig := &tls.Config{
|
|
ServerName: hostname,
|
|
InsecureSkipVerify: p.Insecure,
|
|
}
|
|
|
|
c, err := tls.Dial("tcp", fmt.Sprintf("%s:%d", p.Host, p.Port), tlsconfig)
|
|
if err := c.Handshake(); err != nil {
|
|
c.Close()
|
|
return nil, err
|
|
}
|
|
|
|
if err = httpProxyHandshake(c, host, port); err != nil {
|
|
c.Close()
|
|
return nil, err
|
|
}
|
|
return c, nil
|
|
}
|
|
|
|
func dialSocks5Proxy(host string, port int, p proxy) (net.Conn, error) {
|
|
c, err := net.Dial("tcp", fmt.Sprintf("%s:%d", p.Host, p.Port))
|
|
if err != nil {
|
|
c.Close()
|
|
return nil, err
|
|
}
|
|
|
|
c1 := &socks.Client{Conn: c}
|
|
if err = c1.Connect(host, uint16(port)); err != nil {
|
|
c1.Close()
|
|
return nil, err
|
|
}
|
|
return c1, err
|
|
}
|