Remove mtproto pool

This commit is contained in:
9seconds
2018-07-02 15:44:52 +03:00
parent 5ea0c4f7bf
commit 6fa5f31ca8
10 changed files with 23 additions and 118 deletions
-6
View File
@@ -1,12 +1,6 @@
package rpc
import "bytes"
const (
RPCNonceSeqNo = -2
RPCHandshakeSeqNo = -1
)
type RPC interface {
Bytes() *bytes.Buffer
}
+2 -2
View File
@@ -25,7 +25,7 @@ var (
type RPCHandshakeRequest struct {
}
func (r *RPCHandshakeRequest) Bytes() *bytes.Buffer {
func (r *RPCHandshakeRequest) Bytes() []byte {
buf := &bytes.Buffer{}
buf.Grow(rpcHandshakeRequestLength)
@@ -34,7 +34,7 @@ func (r *RPCHandshakeRequest) Bytes() *bytes.Buffer {
buf.Write(rpcHandshakeSenderPID[:])
buf.Write(rpcHandshakePeerPID[:])
return buf
return buf.Bytes()
}
func init() {
+2 -2
View File
@@ -15,7 +15,7 @@ type RPCHandshakeResponse struct {
PeerPID [rpcHandshakePeerPIDLength]byte
}
func (r *RPCHandshakeResponse) Bytes() *bytes.Buffer {
func (r *RPCHandshakeResponse) Bytes() []byte {
buf := &bytes.Buffer{}
buf.Grow(rpcHandshakeResponseLength)
@@ -24,7 +24,7 @@ func (r *RPCHandshakeResponse) Bytes() *bytes.Buffer {
buf.Write(r.SenderPID[:])
buf.Write(r.PeerPID[:])
return buf
return buf.Bytes()
}
func (r *RPCHandshakeResponse) Valid(req *RPCHandshakeRequest) error {
+2 -2
View File
@@ -31,7 +31,7 @@ type RPCNonceRequest struct {
Nonce [rpcNonceLength]byte
}
func (r *RPCNonceRequest) Bytes() *bytes.Buffer {
func (r *RPCNonceRequest) Bytes() []byte {
buf := &bytes.Buffer{}
buf.Grow(rpcNonceRequestLength)
@@ -41,7 +41,7 @@ func (r *RPCNonceRequest) Bytes() *bytes.Buffer {
buf.Write(r.CryptoTS[:])
buf.Write(r.Nonce[:])
return buf
return buf.Bytes()
}
func NewRPCNonceRequest(proxySecret []byte) (*RPCNonceRequest, error) {
+2 -2
View File
@@ -15,7 +15,7 @@ type RPCNonceResponse struct {
Crypto [rpcNonceCryptoAESLength]byte
}
func (r *RPCNonceResponse) Bytes() *bytes.Buffer {
func (r *RPCNonceResponse) Bytes() []byte {
buf := &bytes.Buffer{}
buf.Grow(rpcNonceResponseLength)
@@ -25,7 +25,7 @@ func (r *RPCNonceResponse) Bytes() *bytes.Buffer {
buf.Write(r.CryptoTS[:])
buf.Write(r.Nonce[:])
return buf
return buf.Bytes()
}
func (r *RPCNonceResponse) Valid(req *RPCNonceRequest) error {
+3 -3
View File
@@ -32,8 +32,8 @@ type RPCProxyRequest struct {
Extras *mtproto.Extras
}
func (r *RPCProxyRequest) Bytes() *bytes.Buffer {
buf := mtproto.GetBuffer()
func (r *RPCProxyRequest) Bytes() []byte {
buf := &bytes.Buffer{}
flags := r.Flags
if r.Extras.QuickAck {
@@ -62,7 +62,7 @@ func (r *RPCProxyRequest) Bytes() *bytes.Buffer {
buf.Write(messageBytes)
}
return buf
return buf.Bytes()
}
func NewRPCProxyRequest(connectionType mtproto.ConnectionType, local, remote *net.TCPAddr, adTag []byte, extras *mtproto.Extras) (*RPCProxyRequest, error) {