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

Fix infinite hangs

Browse Source
This commit is contained in:
9seconds
2018-07-05 20:46:44 +03:00
parent 53ef81a4fc
commit 5413a5b090
5 changed files with 18 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mtproto/wrappers/proxy_request.go
+5 -6
View File
@@ -29,15 +29,16 @@ func (p *ProxyRequestReadWriteCloserWithAddr) Read(buf []byte) (int, error) {
return errors.Annotate(err, "Cannot read RPC tag")
}
if bytes.Equal(ansBuf.Bytes(), rpc.TagCloseExt) {
switch {
case bytes.Equal(ansBuf.Bytes(), rpc.TagCloseExt):
return p.readCloseExt()
} else if bytes.Equal(ansBuf.Bytes(), rpc.TagProxyAns) {
case bytes.Equal(ansBuf.Bytes(), rpc.TagProxyAns):
return p.readProxyAns(buf)
} else if bytes.Equal(ansBuf.Bytes(), rpc.TagSimpleAck) {
case bytes.Equal(ansBuf.Bytes(), rpc.TagSimpleAck):
return p.readSimpleAck()
}
return nil
return errors.Errorf("Unknown RPC answer %s", ansBuf.Bytes())
})
}
@@ -80,8 +81,6 @@ func (p *ProxyRequestReadWriteCloserWithAddr) Write(raw []byte) (int, error) {
if _, err := p.conn.Write(p.req.Bytes(raw)); err != nil {
return 0, err
}
p.req.Options.SimpleAck = false
p.req.Options.QuickAck = false
return len(raw), nil
}