Archives/Cloak
2
0
Fork 0
mirror of https://github.com/cbeuw/Cloak.git synced 2024-11-15 18:13:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Optimise deobfs

Browse Source
This commit is contained in:
Qian Wang 2019-08-07 17:53:34 +01:00
parent e3de2b9852
commit 586cc5ebb6
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
internal/multiplex/obfs.go
View File

@ -74,12 +74,14 @@ func MakeObfs(salsaKey [32]byte, payloadCipher cipher.AEAD) Obfser {
func MakeDeobfs(salsaKey [32]byte, payloadCipher cipher.AEAD) Deobfser { func MakeDeobfs(salsaKey [32]byte, payloadCipher cipher.AEAD) Deobfser {
deobfs := func(in []byte) (*Frame, error) { deobfs := func(in []byte) (*Frame, error) {
if len(in) < 5+HEADER_LEN+8 { if len(in) < 5+HEADER_LEN+8 {
return nil, errors.New("Input cannot be shorter than 33 bytes") return nil, errors.New("Input cannot be shorter than 25 bytes")
} }
peeled := in[5:]
header := peeled[0:12] peeled := make([]byte, len(in)-5)
payload := peeled[12:] copy(peeled, in[5:])
header := peeled[:12]
pldWithOverHead := peeled[12:] // plaintext + potential overhead
nonce := peeled[len(peeled)-8:] nonce := peeled[len(peeled)-8:]
salsa20.XORKeyStream(header, header, nonce, &salsaKey) salsa20.XORKeyStream(header, header, nonce, &salsaKey)
@ -89,20 +91,25 @@ func MakeDeobfs(salsaKey [32]byte, payloadCipher cipher.AEAD) Deobfser {
closing := header[8] closing := header[8]
extraLen := header[9] extraLen := header[9]
outputLen := len(payload) - int(extraLen) usefulPayloadLen := len(pldWithOverHead) - int(extraLen)
if outputLen < 0 { if usefulPayloadLen < 0 {
return nil, errors.New("extra length is greater than total payload length") return nil, errors.New("extra length is greater than total pldWithOverHead length")
} }
outputPayload := make([]byte, outputLen)
var outputPayload []byte
if payloadCipher == nil { if payloadCipher == nil {
copy(outputPayload, payload) if extraLen == 0 {
outputPayload = pldWithOverHead
} else { } else {
plaintext, err := payloadCipher.Open(nil, header, payload, nil) outputPayload = pldWithOverHead[:usefulPayloadLen]
}
} else {
_, err := payloadCipher.Open(pldWithOverHead[:0], header, pldWithOverHead, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
copy(outputPayload, plaintext) outputPayload = pldWithOverHead[:usefulPayloadLen]
} }
ret := &Frame{ ret := &Frame{