Browse Source

Remove unnecessary slicing

V2Ray 10 years ago
parent
commit
d2e69ed6f8
2 changed files with 8 additions and 8 deletions
  1. 4 4
      proxy/vmess/vmessin.go
  2. 4 4
      proxy/vmess/vmessin_udp.go

+ 4 - 4
proxy/vmess/vmessin.go

@@ -87,8 +87,8 @@ func (handler *VMessInboundHandler) HandleConnection(connection net.Conn) error
 
 	go handleInput(request, connReader, input, &readFinish)
 
-	responseKey := md5.Sum(request.RequestKey[:])
-	responseIV := md5.Sum(request.RequestIV[:])
+	responseKey := md5.Sum(request.RequestKey)
+	responseIV := md5.Sum(request.RequestIV)
 
 	responseWriter, err := v2io.NewAesEncryptWriter(responseKey[:], responseIV[:], connection)
 	if err != nil {
@@ -97,7 +97,7 @@ func (handler *VMessInboundHandler) HandleConnection(connection net.Conn) error
 
 	// Optimize for small response packet
 	buffer := make([]byte, 0, 2*1024)
-	buffer = append(buffer, request.ResponseHeader[:]...)
+	buffer = append(buffer, request.ResponseHeader...)
 
 	if data, open := <-output; open {
 		buffer = append(buffer, data...)
@@ -118,7 +118,7 @@ func handleInput(request *protocol.VMessRequest, reader io.Reader, input chan<-
 	defer close(input)
 	defer finish.Unlock()
 
-	requestReader, err := v2io.NewAesDecryptReader(request.RequestKey[:], request.RequestIV[:], reader)
+	requestReader, err := v2io.NewAesDecryptReader(request.RequestKey, request.RequestIV, reader)
 	if err != nil {
 		log.Error("VMessIn: Failed to create decrypt reader: %v", err)
 		return

+ 4 - 4
proxy/vmess/vmessin_udp.go

@@ -49,7 +49,7 @@ func (handler *VMessInboundHandler) AcceptPackets(conn *net.UDPConn) {
 			continue
 		}
 
-		cryptReader, err := v2io.NewAesDecryptReader(request.RequestKey[:], request.RequestIV[:], reader)
+		cryptReader, err := v2io.NewAesDecryptReader(request.RequestKey, request.RequestIV, reader)
 		if err != nil {
 			log.Error("VMessIn: Failed to create decrypt reader: %v", err)
 			continue
@@ -71,8 +71,8 @@ func (handler *VMessInboundHandler) handlePacket(conn *net.UDPConn, request *pro
 	ray := handler.vPoint.DispatchToOutbound(packet)
 	close(ray.InboundInput())
 
-	responseKey := md5.Sum(request.RequestKey[:])
-	responseIV := md5.Sum(request.RequestIV[:])
+	responseKey := md5.Sum(request.RequestKey)
+	responseIV := md5.Sum(request.RequestIV)
 
 	buffer := bytes.NewBuffer(make([]byte, 0, bufferSize))
 
@@ -81,7 +81,7 @@ func (handler *VMessInboundHandler) handlePacket(conn *net.UDPConn, request *pro
 		log.Error("VMessIn: Failed to create encrypt writer: %v", err)
 		return
 	}
-	responseWriter.Write(request.ResponseHeader[:])
+	responseWriter.Write(request.ResponseHeader)
 
 	hasData := false