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

Add buffered reader

Browse Source
This commit is contained in:
9seconds
2018-07-03 10:10:00 +03:00
parent 5f54a235d2
commit 586ffab01b
4 changed files with 100 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mtproto/rpc/rpc_proxy_request.go
+3 -7
View File
@@ -29,10 +29,9 @@ type RPCProxyRequest struct {
LocalIPPort [rpcProxyRequestIPPortLength]byte
ADTag []byte
Extras Extras
Message *bytes.Buffer
}
func (r *RPCProxyRequest) Bytes() []byte {
func (r *RPCProxyRequest) Bytes(message []byte) []byte {
buf := &bytes.Buffer{}
flags := r.Flags
@@ -40,8 +39,7 @@ func (r *RPCProxyRequest) Bytes() []byte {
flags |= RPCProxyRequestFlagsQuickAck
}
messageBytes := r.Message.Bytes()
if bytes.HasPrefix(messageBytes, rpcProxyRequestFlagsEncryptedPrefix[:]) {
if bytes.HasPrefix(message, rpcProxyRequestFlagsEncryptedPrefix[:]) {
flags |= RPCProxyRequestFlagsEncrypted
}
@@ -58,9 +56,7 @@ func (r *RPCProxyRequest) Bytes() []byte {
for i := 0; i < (buf.Len() % 4); i++ {
buf.WriteByte(0x00)
}
if r.Message != nil {
buf.Write(messageBytes)
}
buf.Write(message)
return buf.Bytes()
}