logos-messaging/noise
1
0
Fork 0
mirror of https://github.com/logos-messaging/noise.git synced 2026-01-02 14:13:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove an alloc by pre-allocating the dest slice

Browse Source
This commit is contained in:
Jonathan Rudenberg 2014-07-03 21:54:44 -07:00
parent 903cbf641c
commit e8ef3d40dd
No known key found for this signature in database
GPG Key ID: E38D8C6BAA8C49AA
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
box/box.go
View File

@ -90,6 +90,13 @@ func (c *Crypter) Encrypt(dst []byte, ephKey *Key, plaintext []byte, padLen int)
}
ephKey = &k
}
dstPrefixLen := len(dst)
// Allocate a new slice that can fit the full encrypted box if the current dst doesn't fit
if encLen := c.EncryptedLen(len(plaintext) + padLen); cap(dst)-len(dst) < encLen {
newDst := make([]byte, len(dst), len(dst)+encLen)
copy(newDst, dst)
dst = newDst
}
dh1 := c.Cipher.DH(ephKey.Private, c.ReceiverKey.Public)
dh2 := c.Cipher.DH(c.SenderKey.Private, c.ReceiverKey.Public)
@ -100,8 +107,13 @@ func (c *Crypter) Encrypt(dst []byte, ephKey *Key, plaintext []byte, padLen int)
cc1 := c.Cipher.NewCipher(cv1)
cc2 := c.Cipher.NewCipher(c.ChainVar)
header := cc1.Encrypt(ephKey.Public, ephKey.Public, c.SenderKey.Public)
return noiseBody(cc2, header, padLen, plaintext, header), nil
dst = append(dst, ephKey.Public...)
dst = cc1.Encrypt(dst, ephKey.Public, c.SenderKey.Public)
return noiseBody(cc2, dst, padLen, plaintext, dst[dstPrefixLen:]), nil
}
func (c *Crypter) EncryptedLen(n int) int {
return n + (2 * c.Cipher.DHLen()) + (2 * c.Cipher.MACLen()) + 4
}
func (c *Crypter) Decrypt(ciphertext []byte) ([]byte, error) {