mirrors/wireproxy
1
0
Fork 0
mirror of https://github.com/whyvl/wireproxy.git synced 2025-04-29 19:01:42 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: migration to github.com/things-go/go-socks5

Browse source 
- preallocate config slices
- not used interfaces in consumer
- do not allocate new variables in loops
This commit is contained in:
Evsyukov Denis 2024-02-08 15:16:52 +03:00
parent bc9c2c8473
commit 229ad7a95d
No known key found for this signature in database
6 changed files with 74 additions and 71 deletions
Download patch file Download diff file Expand all files Collapse all files

21
http.go
View file

@ -62,7 +62,7 @@ func (s *HTTPServer) handleConn(req *http.Request, conn net.Conn) (peer net.Conn
_, err = conn.Write([]byte("HTTP/1.1 200 Connection established\r\n\r\n"))
if err != nil {
peer.Close()
_ = peer.Close()
peer = nil
}
@ -83,7 +83,7 @@ func (s *HTTPServer) handle(req *http.Request) (peer net.Conn, err error) {
err = req.Write(peer)
if err != nil {
peer.Close()
_ = peer.Close()
peer = nil
return peer, fmt.Errorf("conn write failed: %w", err)
}
@ -92,10 +92,8 @@ func (s *HTTPServer) handle(req *http.Request) (peer net.Conn, err error) {
}
func (s *HTTPServer) serve(conn net.Conn) error {
defer conn.Close()
var rd io.Reader = bufio.NewReader(conn)
req, err := http.ReadRequest(rd.(*bufio.Reader))
var rd = bufio.NewReader(conn)
req, err := http.ReadRequest(rd)
if err != nil {
return fmt.Errorf("read request failed: %w", err)
}
@ -124,11 +122,6 @@ func (s *HTTPServer) serve(conn net.Conn) error {
}
defer peer.Close()
go func() {
defer peer.Close()
defer conn.Close()
_, _ = io.Copy(conn, peer)
}()
_, err = io.Copy(peer, conn)
return err
@ -136,13 +129,14 @@ func (s *HTTPServer) serve(conn net.Conn) error {
// ListenAndServe is used to create a listener and serve on it
func (s *HTTPServer) ListenAndServe(network, addr string) error {
server, err := net.Listen("tcp", s.config.BindAddress)
server, err := net.Listen(network, addr)
if err != nil {
return fmt.Errorf("listen tcp failed: %w", err)
}
defer server.Close()
var conn net.Conn
for {
conn, err := server.Accept()
conn, err = server.Accept()
if err != nil {
return fmt.Errorf("accept request failed: %w", err)
}
@ -152,6 +146,7 @@ func (s *HTTPServer) ListenAndServe(network, addr string) error {
log.Println(err)
}
_ = conn.Close()
conn = nil
}(conn)
}
}