mirror of https://github.com/ginuerzh/gost
rui.zheng
9 years ago
7 changed files with 164 additions and 369 deletions
@ -0,0 +1,26 @@ |
|||
package gost |
|||
|
|||
import ( |
|||
"net" |
|||
) |
|||
|
|||
type Handler interface { |
|||
Handle(net.Conn) |
|||
} |
|||
|
|||
type DefaultHandler struct { |
|||
server Server |
|||
} |
|||
|
|||
func (h *DefaultHandler) Handle(conn net.Conn) { |
|||
var handler Handler |
|||
|
|||
switch h.server.Options().BaseOptions().Protocol { |
|||
case "http": |
|||
case "ss": // shadowsocks
|
|||
handler = ShadowHandler(h.server) |
|||
|
|||
} |
|||
|
|||
handler.Handle(conn) |
|||
} |
|||
@ -1,165 +0,0 @@ |
|||
package ssocks |
|||
|
|||
import ( |
|||
"encoding/binary" |
|||
"fmt" |
|||
"io" |
|||
"net" |
|||
"net/url" |
|||
"strconv" |
|||
"time" |
|||
|
|||
"github.com/go-log/log" |
|||
|
|||
"github.com/ginuerzh/gost" |
|||
ss "github.com/shadowsocks/shadowsocks-go/shadowsocks" |
|||
) |
|||
|
|||
// Due to in/out byte length is inconsistent of the shadowsocks.Conn.Write,
|
|||
// we wrap around it to make io.Copy happy
|
|||
type shadowConn struct { |
|||
conn net.Conn |
|||
} |
|||
|
|||
func NewConn(conn net.Conn) net.Conn { |
|||
return &shadowConn{conn: conn} |
|||
} |
|||
|
|||
func (c *shadowConn) Read(b []byte) (n int, err error) { |
|||
return c.conn.Read(b) |
|||
} |
|||
|
|||
func (c *shadowConn) Write(b []byte) (n int, err error) { |
|||
n = len(b) // force byte length consistent
|
|||
_, err = c.conn.Write(b) |
|||
return |
|||
} |
|||
|
|||
func (c *shadowConn) Close() error { |
|||
return c.conn.Close() |
|||
} |
|||
|
|||
func (c *shadowConn) LocalAddr() net.Addr { |
|||
return c.conn.LocalAddr() |
|||
} |
|||
|
|||
func (c *shadowConn) RemoteAddr() net.Addr { |
|||
return c.conn.RemoteAddr() |
|||
} |
|||
|
|||
func (c *shadowConn) SetDeadline(t time.Time) error { |
|||
return c.conn.SetDeadline(t) |
|||
} |
|||
|
|||
func (c *shadowConn) SetReadDeadline(t time.Time) error { |
|||
return c.conn.SetReadDeadline(t) |
|||
} |
|||
|
|||
func (c *shadowConn) SetWriteDeadline(t time.Time) error { |
|||
return c.conn.SetWriteDeadline(t) |
|||
} |
|||
|
|||
const ( |
|||
idType = 0 // address type index
|
|||
idIP0 = 1 // ip addres start index
|
|||
idDmLen = 1 // domain address length index
|
|||
idDm0 = 2 // domain address start index
|
|||
|
|||
typeIPv4 = 1 // type is ipv4 address
|
|||
typeDm = 3 // type is domain address
|
|||
typeIPv6 = 4 // type is ipv6 address
|
|||
|
|||
lenIPv4 = net.IPv4len + 2 // ipv4 + 2port
|
|||
lenIPv6 = net.IPv6len + 2 // ipv6 + 2port
|
|||
lenDmBase = 2 // 1addrLen + 2port, plus addrLen
|
|||
lenHmacSha1 = 10 |
|||
) |
|||
|
|||
type ShadowServer struct { |
|||
conn net.Conn |
|||
base gost.Server |
|||
} |
|||
|
|||
func NewServer(conn net.Conn, cipher *url.Userinfo, base gost.Server) (*ShadowServer, error) { |
|||
method := cipher.Username() |
|||
password, _ := cipher.Password() |
|||
cp, err := ss.NewCipher(method, password) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
return &ShadowServer{conn: ss.NewConn(conn, cp), base: base}, nil |
|||
} |
|||
|
|||
func (s *ShadowServer) Serve() error { |
|||
log.Logf("[ss] %s - %s", s.conn.RemoteAddr(), s.conn.LocalAddr()) |
|||
|
|||
addr, err := s.getRequest() |
|||
if err != nil { |
|||
log.Logf("[ss] %s - %s : %s", s.conn.RemoteAddr(), s.conn.LocalAddr(), err) |
|||
return err |
|||
} |
|||
log.Logf("[ss] %s -> %s", s.conn.RemoteAddr(), addr) |
|||
|
|||
cc, err := s.base.Chain().Dial(addr) |
|||
if err != nil { |
|||
log.Logf("[ss] %s -> %s : %s", s.conn.RemoteAddr(), addr, err) |
|||
return err |
|||
} |
|||
defer cc.Close() |
|||
|
|||
log.Logf("[ss] %s <-> %s", s.conn.RemoteAddr(), addr) |
|||
defer log.Logf("[ss] %s >-< %s", s.conn.RemoteAddr(), addr) |
|||
|
|||
return gost.Transport(&shadowConn{conn: s.conn}, cc) |
|||
} |
|||
|
|||
// This function is copied from shadowsocks library with some modification.
|
|||
func (s *ShadowServer) getRequest() (host string, err error) { |
|||
// buf size should at least have the same size with the largest possible
|
|||
// request size (when addrType is 3, domain name has at most 256 bytes)
|
|||
// 1(addrType) + 1(lenByte) + 256(max length address) + 2(port)
|
|||
buf := make([]byte, gost.SmallBufferSize) |
|||
|
|||
// read till we get possible domain length field
|
|||
s.conn.SetReadDeadline(time.Now().Add(30 * time.Second)) |
|||
if _, err = io.ReadFull(s.conn, buf[:idType+1]); err != nil { |
|||
return |
|||
} |
|||
|
|||
var reqStart, reqEnd int |
|||
addrType := buf[idType] |
|||
switch addrType & ss.AddrMask { |
|||
case typeIPv4: |
|||
reqStart, reqEnd = idIP0, idIP0+lenIPv4 |
|||
case typeIPv6: |
|||
reqStart, reqEnd = idIP0, idIP0+lenIPv6 |
|||
case typeDm: |
|||
if _, err = io.ReadFull(s.conn, buf[idType+1:idDmLen+1]); err != nil { |
|||
return |
|||
} |
|||
reqStart, reqEnd = idDm0, int(idDm0+buf[idDmLen]+lenDmBase) |
|||
default: |
|||
err = fmt.Errorf("addr type %d not supported", addrType&ss.AddrMask) |
|||
return |
|||
} |
|||
|
|||
if _, err = io.ReadFull(s.conn, buf[reqStart:reqEnd]); err != nil { |
|||
return |
|||
} |
|||
|
|||
// Return string for typeIP is not most efficient, but browsers (Chrome,
|
|||
// Safari, Firefox) all seems using typeDm exclusively. So this is not a
|
|||
// big problem.
|
|||
switch addrType & ss.AddrMask { |
|||
case typeIPv4: |
|||
host = net.IP(buf[idIP0 : idIP0+net.IPv4len]).String() |
|||
case typeIPv6: |
|||
host = net.IP(buf[idIP0 : idIP0+net.IPv6len]).String() |
|||
case typeDm: |
|||
host = string(buf[idDm0 : idDm0+buf[idDmLen]]) |
|||
} |
|||
// parse port
|
|||
port := binary.BigEndian.Uint16(buf[reqEnd-2 : reqEnd]) |
|||
host = net.JoinHostPort(host, strconv.Itoa(int(port))) |
|||
return |
|||
} |
|||
Loading…
Reference in new issue