ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-09-01 04:04:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add framerwc

Browse Source
This commit is contained in:
9seconds
2018-07-02 11:43:11 +03:00
parent 2c387cc213
commit 420159a7d3
2 changed files with 145 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wrappers/blockcipherrwc.go
+7 -5
View File
@@ -29,6 +29,7 @@ func (c *BlockCipherReadWriteCloser) Read(p []byte) (int, error) {
}
c.buf.Write(p[:n])
}
c.decryptor.CryptBlocks(c.buf.Bytes(), c.buf.Bytes())
return c.flush(p)
}
@@ -59,16 +60,17 @@ func (c *BlockCipherReadWriteCloser) flush(p []byte) (int, error) {
if c.buf.Len() < sizeToRead {
sizeToRead = c.buf.Len()
}
sizeToRead = aes.BlockSize * (sizeToRead / aes.BlockSize)
c.decryptor.CryptBlocks(p, c.buf.Bytes()[:sizeToRead])
data := c.buf.Bytes()
copy(p, data[:sizeToRead])
if sizeToRead == c.buf.Len() {
c.buf.Reset()
} else {
leftover := c.buf.Bytes()[sizeToRead:]
newBuf := getBuffer()
newBuf.Write(data[sizeToRead:])
putBuffer(c.buf)
c.buf = getBuffer()
c.buf.Write(leftover)
c.buf = newBuf
}
return sizeToRead, nil