fix logic

Author: null

proxy/hysteria/client.go

@@ -223,15 +223,9 @@ func (w *UDPWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
 			for _, fMsg := range fMsgs {
 				err := w.sendMsg(&fMsg)
 				if err != nil {
-					b.Release()
-					buf.ReleaseMulti(mb)
-					return err
+					break
 				}
 			}
-		} else {
-			b.Release()
-			buf.ReleaseMulti(mb)
-			return err
 		}
 
 		b.Release()

proxy/hysteria/server.go

@@ -170,15 +170,6 @@ func (s *Server) Process(ctx context.Context, network net.Network, conn stat.Con
 		}
 		common.Must(conn.SetReadDeadline(time.Time{}))
 
-		bufferedWriter := buf.NewBufferedWriter(buf.NewWriter(conn))
-		err = WriteTCPResponse(bufferedWriter, true, "")
-		if err != nil {
-			return errors.New("failed to write response").Base(err)
-		}
-		if err := bufferedWriter.SetBuffered(false); err != nil {
-			return err
-		}
-
 		dest := common.Must2(net.ParseDestination("tcp:" + addr))
 		ctx = log.ContextWithAccessMessage(ctx, &log.AccessMessage{
 			From:   conn.RemoteAddr(),
@@ -189,6 +180,15 @@ func (s *Server) Process(ctx context.Context, network net.Network, conn stat.Con
 		})
 		errors.LogInfo(ctx, "tunnelling request to ", dest)
 
+		bufferedWriter := buf.NewBufferedWriter(buf.NewWriter(conn))
+		err = WriteTCPResponse(bufferedWriter, true, "")
+		if err != nil {
+			return errors.New("failed to write response").Base(err)
+		}
+		if err := bufferedWriter.SetBuffered(false); err != nil {
+			return err
+		}
+
 		return dispatcher.DispatchLink(ctx, dest, &transport.Link{
 			Reader: buf.NewReader(conn),
 			Writer: bufferedWriter,

transport/internet/hysteria/conn.go

@@ -35,14 +35,14 @@ func (i *interConn) Write(b []byte) (int, error) {
 	if i.client {
 		i.mutex.Lock()
 		if i.client {
-			i.client = false
 			buf := make([]byte, 0, quicvarint.Len(FrameTypeTCPRequest)+len(b))
 			buf = quicvarint.Append(buf, FrameTypeTCPRequest)
 			buf = append(buf, b...)
 			_, err := i.stream.Write(buf)
 			if err != nil {
 				return 0, err
 			}
+			i.client = false
 			return len(b), nil
 		}
 		i.mutex.Unlock()