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