func (receiver *socksReceiver) handleConnection(conn net.Conn) { data, host, err := receiver.request(conn) if err != nil { logger.Errorf("Server:---->cannot find host %s\n", err) return } cli := client.NewClient(host) if cli == nil { logger.Error("handle connection error,cannot create client.") return } defer func() { //fixe bug when cli is nil if cli != nil { cli.Close() } conn.Close() }() if _, err := cli.Conn.Write(data); err != nil { logger.Error("Server:----->write data error") return } c.SetTimeout(cli.Conn.SetReadDeadline, receiver.cfg.Timeout) c.SetTimeout(conn.SetReadDeadline, receiver.cfg.Timeout) //for comment go io.Copy(cli.Conn, conn) io.Copy(conn, cli.Conn) }
func (receiver *receiver) handleConnection(conn net.Conn) { var err error if err = receiver.handshake(conn); err != nil { logger.Errorf("Socks5--->handshake error %s\n", err) return } //set default timeout cfg.SetTimeout(conn.SetReadDeadline, receiver.config.Timeout) //conn.SetReadDeadline(time.Now().Add(time.Duration(s.config.Timeout) * time.Second)) rawAddr, host, domian, err := receiver.request(conn) if err != nil { logger.Errorf("Socks5--->get host failed %s\n", err) return } _, err = conn.Write([]byte{0x05, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x08, 0x43}) if err != nil { logger.Errorf("Socks5--->error send data to client %s\n", err) return } //need connect to remote server or not if domian && !cfg.IsInWhiteList(strings.Split(host, ":")[0]) { result := cfg.ParserDomain(strings.Split(host, ":")[0]) if result { handleChina(conn, host) return } } else if !domian { result := cfg.ParserIP(strings.Split(host, ":")[0]) if result { handleChina(conn, host) return } } ci, flag, err := client.NewRemoteServer(receiver.config).ChooseServer() if err != nil { logger.Errorf("Socks5--->connect to server failed %s", err) return } if flag { cli := ci.(*client.SSLClient) cfg.SetTimeout(cli.Conn.SetReadDeadline, receiver.config.Timeout) defer func() { cli.Conn.Close() conn.Close() }() if _, err = cli.Conn.Write(rawAddr); err != nil { logger.Errorf("Socks5--->handle error message when write data %s \n", err) return } pipe(&cli.Conn, conn) return } cli := ci.(*client.Client) cfg.SetTimeout(cli.Conn.SetReadDeadline, receiver.config.Timeout) //cli.Conn.SetReadDeadline(time.Now().Add(time.Duration(s.config.Timeout) * time.Second)) defer func() { cli.Conn.Close() conn.Close() }() if _, err = cli.Conn.Write(rawAddr); err != nil { logger.Errorf("Socks5--->handle error message when write data %s \n", err) return } pipe(cli.Conn, conn) }