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

Small changes for abridged/intermediate

Browse Source
This commit is contained in:
9seconds
2018-07-05 16:37:25 +03:00
parent 4210cf014e
commit 154ff70bf7
3 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mtproto/wrappers/intermediate.go
+7 -4
View File
@@ -23,18 +23,21 @@ type IntermediateReadWriteCloserWithAddr struct {
func (i *IntermediateReadWriteCloserWithAddr) Read(p []byte) (int, error) {
return i.BufferedRead(p, func() error {
var length uint32
if err := binary.Read(i.conn, binary.LittleEndian, &length); err != nil {
buf := &bytes.Buffer{}
buf.Grow(4)
if _, err := io.CopyN(buf, i.conn, 4); err != nil {
return errors.Annotate(err, "Cannot read message length")
}
length := binary.LittleEndian.Uint32(buf.Bytes())
buf.Reset()
buf.Grow(int(length))
if length > intermediateQuickAckLength {
i.opts.QuickAck = true
length -= intermediateQuickAckLength
}
buf := &bytes.Buffer{}
buf.Grow(int(length))
if _, err := io.CopyN(buf, i.conn, int64(length)); err != nil {
return errors.Annotate(err, "Cannot read the message")
}