1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 01:57:12 -05:00

checksum for response command

This commit is contained in:
v2ray 2016-01-22 17:56:03 +01:00
parent 24b300d75d
commit 790d37bf90
3 changed files with 30 additions and 2 deletions

View File

@ -10,6 +10,14 @@ func (this BytesLiteral) Value() []byte {
return []byte(this) return []byte(this)
} }
func (this BytesLiteral) Uint32Value() uint32 {
value := this.Value()
return uint32(value[0])<<24 +
uint32(value[1])<<16 +
uint32(value[2])<<8 +
uint32(value[3])
}
func (this BytesLiteral) Int64Value() int64 { func (this BytesLiteral) Int64Value() int64 {
value := this.Value() value := this.Value()
return int64(value[0])<<56 + return int64(value[0])<<56 +

View File

@ -1,6 +1,8 @@
package inbound package inbound
import ( import (
"hash/fnv"
"github.com/v2ray/v2ray-core/common/alloc" "github.com/v2ray/v2ray-core/common/alloc"
"github.com/v2ray/v2ray-core/common/log" "github.com/v2ray/v2ray-core/common/log"
"github.com/v2ray/v2ray-core/common/serial" "github.com/v2ray/v2ray-core/common/serial"
@ -38,10 +40,14 @@ func (this *VMessInboundHandler) generateCommand(buffer *alloc.Buffer) {
} }
} }
if commandBytes.Len() > 256 { if cmd == 0 || commandBytes.Len()+4 > 256 {
buffer.AppendBytes(byte(0), byte(0)) buffer.AppendBytes(byte(0), byte(0))
} else { } else {
buffer.AppendBytes(cmd, byte(commandBytes.Len())) buffer.AppendBytes(cmd, byte(commandBytes.Len()+4))
fnv1hash := fnv.New32a()
fnv1hash.Write(commandBytes.Value)
hashValue := fnv1hash.Sum32()
buffer.AppendBytes(byte(hashValue>>24), byte(hashValue>>16), byte(hashValue>>8), byte(hashValue))
buffer.Append(commandBytes.Value) buffer.Append(commandBytes.Value)
} }
} }

View File

@ -1,8 +1,11 @@
package outbound package outbound
import ( import (
"hash/fnv"
"github.com/v2ray/v2ray-core/common/log" "github.com/v2ray/v2ray-core/common/log"
v2net "github.com/v2ray/v2ray-core/common/net" v2net "github.com/v2ray/v2ray-core/common/net"
"github.com/v2ray/v2ray-core/common/serial"
"github.com/v2ray/v2ray-core/proxy/vmess" "github.com/v2ray/v2ray-core/proxy/vmess"
"github.com/v2ray/v2ray-core/proxy/vmess/command" "github.com/v2ray/v2ray-core/proxy/vmess/command"
) )
@ -14,6 +17,17 @@ func (this *VMessOutboundHandler) handleSwitchAccount(cmd *command.SwitchAccount
} }
func (this *VMessOutboundHandler) handleCommand(dest v2net.Destination, cmdId byte, data []byte) { func (this *VMessOutboundHandler) handleCommand(dest v2net.Destination, cmdId byte, data []byte) {
if len(data) < 4 {
return
}
fnv1hash := fnv.New32a()
fnv1hash.Write(data[4:])
actualHashValue := fnv1hash.Sum32()
expectedHashValue := serial.BytesLiteral(data[:4]).Uint32Value()
if actualHashValue != expectedHashValue {
return
}
data = data[4:]
cmd, err := command.CreateResponseCommand(cmdId) cmd, err := command.CreateResponseCommand(cmdId)
if err != nil { if err != nil {
log.Warning("VMessOut: Unknown response command (", cmdId, "): ", err) log.Warning("VMessOut: Unknown response command (", cmdId, "): ", err)