Kaynağa Gözat

fix lint warnings

Darien Raymond 8 yıl önce
ebeveyn
işleme
721bea5d54
2 değiştirilmiş dosya ile 2 ekleme ve 2 silme
  1. 1 1
      proxy/http/server.go
  2. 1 1
      proxy/shadowsocks/protocol.go

+ 1 - 1
proxy/http/server.go

@@ -171,7 +171,7 @@ func (s *Server) handleConnect(ctx context.Context, request *http.Request, reade
 	if reader.Buffered() > 0 {
 		payload := buf.New()
 		common.Must(payload.Reset(func(b []byte) (int, error) {
-			return reader.Read(b)
+			return reader.Read(b[:reader.Buffered()])
 		}))
 		if err := ray.InboundInput().WriteMultiBuffer(buf.NewMultiBufferValue(payload)); err != nil {
 			return err

+ 1 - 1
proxy/shadowsocks/protocol.go

@@ -225,7 +225,7 @@ func WriteTCPResponse(request *protocol.RequestHeader, writer io.Writer) (buf.Wr
 	account := rawAccount.(*ShadowsocksAccount)
 
 	iv := make([]byte, account.Cipher.IVSize())
-	rand.Read(iv)
+	common.Must2(rand.Read(iv))
 	_, err = writer.Write(iv)
 	if err != nil {
 		return nil, newError("failed to write IV.").Base(err)