[ Cmii ] [ Octopus ] - add all image tag back up; add socks5 proxy
This commit is contained in:
zeaslity
254
socks5_txthinking/client.go
Normal file
254
socks5_txthinking/client.go
Normal file
@@ -0,0 +1,254 @@
|
||||
package socks5
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Client is socks5 client wrapper
|
||||
type Client struct {
|
||||
Server string
|
||||
UserName string
|
||||
Password string
|
||||
// On cmd UDP, let server control the tcp and udp connection relationship
|
||||
TCPConn net.Conn
|
||||
UDPConn net.Conn
|
||||
RemoteAddress net.Addr
|
||||
TCPTimeout int
|
||||
UDPTimeout int
|
||||
Dst string
|
||||
}
|
||||
|
||||
// This is just create a client, you need to use Dial to create conn
|
||||
func NewClient(addr, username, password string, tcpTimeout, udpTimeout int) (*Client, error) {
|
||||
c := &Client{
|
||||
Server: addr,
|
||||
UserName: username,
|
||||
Password: password,
|
||||
TCPTimeout: tcpTimeout,
|
||||
UDPTimeout: udpTimeout,
|
||||
}
|
||||
return c, nil
|
||||
}
|
||||
|
||||
func (c *Client) Dial(network, addr string) (net.Conn, error) {
|
||||
return c.DialWithLocalAddr(network, "", addr, nil)
|
||||
}
|
||||
|
||||
// If you want to send address that expects to use to send UDP, just assign it to src, otherwise it will send zero address.
|
||||
// Recommend specifying the src address in a non-NAT environment, and leave it blank in other cases.
|
||||
func (c *Client) DialWithLocalAddr(network, src, dst string, remoteAddr net.Addr) (net.Conn, error) {
|
||||
c = &Client{
|
||||
Server: c.Server,
|
||||
UserName: c.UserName,
|
||||
Password: c.Password,
|
||||
TCPTimeout: c.TCPTimeout,
|
||||
UDPTimeout: c.UDPTimeout,
|
||||
Dst: dst,
|
||||
RemoteAddress: remoteAddr,
|
||||
}
|
||||
var err error
|
||||
if network == "tcp" {
|
||||
var laddr net.Addr
|
||||
if src != "" {
|
||||
laddr, err = net.ResolveTCPAddr("tcp", src)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
if err := c.Negotiate(laddr); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
a, h, p, err := ParseAddress(dst)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if a == ATYPDomain {
|
||||
h = h[1:]
|
||||
}
|
||||
if _, err := c.Request(NewRequest(CmdConnect, a, h, p)); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return c, nil
|
||||
}
|
||||
if network == "udp" {
|
||||
var laddr net.Addr
|
||||
if src != "" {
|
||||
laddr, err = net.ResolveTCPAddr("tcp", src)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
if err := c.Negotiate(laddr); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
a, h, p := ATYPIPv4, net.IPv4zero, []byte{0x00, 0x00}
|
||||
if src != "" {
|
||||
a, h, p, err = ParseAddress(src)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if a == ATYPDomain {
|
||||
h = h[1:]
|
||||
}
|
||||
}
|
||||
rp, err := c.Request(NewRequest(CmdUDP, a, h, p))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
c.UDPConn, err = DialUDP("udp", src, rp.Address())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if c.UDPTimeout != 0 {
|
||||
if err := c.UDPConn.SetDeadline(time.Now().Add(time.Duration(c.UDPTimeout) * time.Second)); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
return c, nil
|
||||
}
|
||||
return nil, errors.New("unsupport network")
|
||||
}
|
||||
|
||||
func (c *Client) Read(b []byte) (int, error) {
|
||||
if c.UDPConn == nil {
|
||||
return c.TCPConn.Read(b)
|
||||
}
|
||||
n, err := c.UDPConn.Read(b)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
d, err := NewDatagramFromBytes(b[0:n])
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
n = copy(b, d.Data)
|
||||
return n, nil
|
||||
}
|
||||
|
||||
func (c *Client) Write(b []byte) (int, error) {
|
||||
if c.UDPConn == nil {
|
||||
return c.TCPConn.Write(b)
|
||||
}
|
||||
a, h, p, err := ParseAddress(c.Dst)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if a == ATYPDomain {
|
||||
h = h[1:]
|
||||
}
|
||||
d := NewDatagram(a, h, p, b)
|
||||
b1 := d.Bytes()
|
||||
n, err := c.UDPConn.Write(b1)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if len(b1) != n {
|
||||
return 0, errors.New("not write full")
|
||||
}
|
||||
return len(b), nil
|
||||
}
|
||||
|
||||
func (c *Client) Close() error {
|
||||
if c.UDPConn == nil {
|
||||
return c.TCPConn.Close()
|
||||
}
|
||||
if c.TCPConn != nil {
|
||||
c.TCPConn.Close()
|
||||
}
|
||||
return c.UDPConn.Close()
|
||||
}
|
||||
|
||||
func (c *Client) LocalAddr() net.Addr {
|
||||
if c.UDPConn == nil {
|
||||
return c.TCPConn.LocalAddr()
|
||||
}
|
||||
return c.UDPConn.LocalAddr()
|
||||
}
|
||||
|
||||
func (c *Client) RemoteAddr() net.Addr {
|
||||
return c.RemoteAddress
|
||||
}
|
||||
|
||||
func (c *Client) SetDeadline(t time.Time) error {
|
||||
if c.UDPConn == nil {
|
||||
return c.TCPConn.SetDeadline(t)
|
||||
}
|
||||
return c.UDPConn.SetDeadline(t)
|
||||
}
|
||||
|
||||
func (c *Client) SetReadDeadline(t time.Time) error {
|
||||
if c.UDPConn == nil {
|
||||
return c.TCPConn.SetReadDeadline(t)
|
||||
}
|
||||
return c.UDPConn.SetReadDeadline(t)
|
||||
}
|
||||
|
||||
func (c *Client) SetWriteDeadline(t time.Time) error {
|
||||
if c.UDPConn == nil {
|
||||
return c.TCPConn.SetWriteDeadline(t)
|
||||
}
|
||||
return c.UDPConn.SetWriteDeadline(t)
|
||||
}
|
||||
|
||||
func (c *Client) Negotiate(laddr net.Addr) error {
|
||||
src := ""
|
||||
if laddr != nil {
|
||||
src = laddr.String()
|
||||
}
|
||||
var err error
|
||||
c.TCPConn, err = DialTCP("tcp", src, c.Server)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if c.TCPTimeout != 0 {
|
||||
if err := c.TCPConn.SetDeadline(time.Now().Add(time.Duration(c.TCPTimeout) * time.Second)); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
m := MethodNone
|
||||
if c.UserName != "" && c.Password != "" {
|
||||
m = MethodUsernamePassword
|
||||
}
|
||||
rq := NewNegotiationRequest([]byte{m})
|
||||
if _, err := rq.WriteTo(c.TCPConn); err != nil {
|
||||
return err
|
||||
}
|
||||
rp, err := NewNegotiationReplyFrom(c.TCPConn)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if rp.Method != m {
|
||||
return errors.New("Unsupport method")
|
||||
}
|
||||
if m == MethodUsernamePassword {
|
||||
urq := NewUserPassNegotiationRequest([]byte(c.UserName), []byte(c.Password))
|
||||
if _, err := urq.WriteTo(c.TCPConn); err != nil {
|
||||
return err
|
||||
}
|
||||
urp, err := NewUserPassNegotiationReplyFrom(c.TCPConn)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if urp.Status != UserPassStatusSuccess {
|
||||
return ErrUserPassAuth
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) Request(r *Request) (*Reply, error) {
|
||||
if _, err := r.WriteTo(c.TCPConn); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
rp, err := NewReplyFrom(c.TCPConn)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if rp.Rep != RepSuccess {
|
||||
return nil, errors.New("Host unreachable")
|
||||
}
|
||||
return rp, nil
|
||||
}
|
||||
Reference in New Issue
Block a user