Forráskód Böngészése

fix lint warnings

Darien Raymond 8 éve
szülő
commit
7d31c0641b
2 módosított fájl, 7 hozzáadás és 2 törlés
  1. 4 1
      proxy/freedom/freedom.go
  2. 3 1
      proxy/http/server.go

+ 4 - 1
proxy/freedom/freedom.go

@@ -96,7 +96,10 @@ func (v *Handler) Dispatch(destination v2net.Destination, payload *buf.Buffer, r
 	output := ray.OutboundOutput()
 
 	if !payload.IsEmpty() {
-		conn.Write(payload.Bytes())
+		if _, err := conn.Write(payload.Bytes()); err != nil {
+			log.Warning("Freedom: Failed to write to destination: ", destination, ": ", err)
+			return
+		}
 	}
 
 	go func() {

+ 3 - 1
proxy/http/server.go

@@ -223,7 +223,7 @@ func (v *Server) GenerateResponse(statusCode int, status string) *http.Response
 		Header:        hdr,
 		Body:          nil,
 		ContentLength: 0,
-		Close:         false,
+		Close:         true,
 	}
 }
 
@@ -247,6 +247,8 @@ func (v *Server) handlePlainHTTP(request *http.Request, session *proxy.SessionIn
 	go func() {
 		defer finish.Done()
 		requestWriter := bufio.NewWriter(buf.NewBytesWriter(ray.InboundInput()))
+		defer requestWriter.Release()
+
 		err := request.Write(requestWriter)
 		if err != nil {
 			log.Warning("HTTP: Failed to write request: ", err)