From 42417cab175c46ecbef9b61ea9e7d29d28845f42 Mon Sep 17 00:00:00 2001
From: Darien Raymond <admin@v2ray.com>
Date: Mon, 12 Dec 2016 21:58:13 +0100
Subject: [PATCH] fix compile warning

---
 transport/internet/headers/srtp/srtp.go | 4 ++--
 transport/internet/headers/utp/utp.go   | 5 +++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/transport/internet/headers/srtp/srtp.go b/transport/internet/headers/srtp/srtp.go
index 5290a87bc..1d9e8fba4 100644
--- a/transport/internet/headers/srtp/srtp.go
+++ b/transport/internet/headers/srtp/srtp.go
@@ -19,8 +19,8 @@ func (v *SRTP) Size() int {
 
 func (v *SRTP) Write(b []byte) (int, error) {
 	v.number++
-	b = serial.Uint16ToBytes(v.number, b[:0])
-	b = serial.Uint16ToBytes(v.number, b)
+	serial.Uint16ToBytes(v.number, b[:0])
+	serial.Uint16ToBytes(v.number, b[:2])
 	return 4, nil
 }
 
diff --git a/transport/internet/headers/utp/utp.go b/transport/internet/headers/utp/utp.go
index 7dab46c33..24aac55df 100644
--- a/transport/internet/headers/utp/utp.go
+++ b/transport/internet/headers/utp/utp.go
@@ -19,8 +19,9 @@ func (v *UTP) Size() int {
 }
 
 func (v *UTP) Write(b []byte) (int, error) {
-	b = serial.Uint16ToBytes(v.connectionId, b[:0])
-	b = append(b, v.header, v.extension)
+	serial.Uint16ToBytes(v.connectionId, b[:0])
+	b[2] = v.header
+	b[3] = v.extension
 	return 4, nil
 }