Bladeren bron

Minor cleanups

tags/0.9
9seconds 7 jaren geleden
bovenliggende
commit
daddf50390
3 gewijzigde bestanden met toevoegingen van 8 en 11 verwijderingen
  1. 1
    4
      mtproto/rpc/rpc_handshake_request.go
  2. 1
    2
      mtproto/rpc/rpc_nonce_request.go
  3. 6
    5
      telegram/middle.go

+ 1
- 4
mtproto/rpc/rpc_handshake_request.go Bestand weergeven

8
 	rpcHandshakeSenderPIDLength = 12
8
 	rpcHandshakeSenderPIDLength = 12
9
 	rpcHandshakePeerPIDLength   = rpcHandshakeSenderPIDLength
9
 	rpcHandshakePeerPIDLength   = rpcHandshakeSenderPIDLength
10
 
10
 
11
-	rpcHandshakeRequestLength = rpcHandshakeTagLength + rpcHandshakeFlagsLength +
12
-		rpcHandshakeSenderPIDLength + rpcHandshakePeerPIDLength
11
+	rpcHandshakeRequestLength = rpcHandshakeTagLength + rpcHandshakeFlagsLength + rpcHandshakeSenderPIDLength + rpcHandshakePeerPIDLength
13
 )
12
 )
14
 
13
 
15
 var (
14
 var (
18
 
17
 
19
 	rpcHandshakeTag   = [rpcHandshakeTagLength]byte{0xf5, 0xee, 0x82, 0x76}
18
 	rpcHandshakeTag   = [rpcHandshakeTagLength]byte{0xf5, 0xee, 0x82, 0x76}
20
 	rpcHandshakeFlags = [rpcHandshakeFlagsLength]byte{0x00, 0x00, 0x00, 0x00}
19
 	rpcHandshakeFlags = [rpcHandshakeFlagsLength]byte{0x00, 0x00, 0x00, 0x00}
21
-
22
-	rpcHandshakeBuffer *bytes.Buffer
23
 )
20
 )
24
 
21
 
25
 type RPCHandshakeRequest struct {
22
 type RPCHandshakeRequest struct {

+ 1
- 2
mtproto/rpc/rpc_nonce_request.go Bestand weergeven

16
 	rpcNonceTagLength         = 4
16
 	rpcNonceTagLength         = 4
17
 	rpcNonceCryptoAESLength   = 4
17
 	rpcNonceCryptoAESLength   = 4
18
 
18
 
19
-	rpcNonceRequestLength = rpcNonceTagLength + rpcNonceKeySelectorLength +
20
-		rpcNonceCryptoAESLength + rpcNonceCryptoTSLength + rpcNonceLength
19
+	rpcNonceRequestLength = rpcNonceTagLength + rpcNonceKeySelectorLength + rpcNonceCryptoAESLength + rpcNonceCryptoTSLength + rpcNonceLength
21
 )
20
 )
22
 
21
 
23
 var (
22
 var (

+ 6
- 5
telegram/middle.go Bestand weergeven

1
 package telegram
1
 package telegram
2
 
2
 
3
 import (
3
 import (
4
+	"io"
4
 	"io/ioutil"
5
 	"io/ioutil"
5
 	"net"
6
 	"net"
6
 	"net/http"
7
 	"net/http"
7
 	"sync"
8
 	"sync"
8
 
9
 
10
+	"github.com/juju/errors"
9
 	"go.uber.org/zap"
11
 	"go.uber.org/zap"
10
 
12
 
11
 	"github.com/9seconds/mtg/config"
13
 	"github.com/9seconds/mtg/config"
13
 	"github.com/9seconds/mtg/mtproto/rpc"
15
 	"github.com/9seconds/mtg/mtproto/rpc"
14
 	mtwrappers "github.com/9seconds/mtg/mtproto/wrappers"
16
 	mtwrappers "github.com/9seconds/mtg/mtproto/wrappers"
15
 	"github.com/9seconds/mtg/wrappers"
17
 	"github.com/9seconds/mtg/wrappers"
16
-	"github.com/juju/errors"
17
 )
18
 )
18
 
19
 
19
 type middleTelegram struct {
20
 type middleTelegram struct {
70
 	return nil, nil
71
 	return nil, nil
71
 }
72
 }
72
 
73
 
73
-func (t *middleTelegram) sendRPCNonceRequest(conn wrappers.ReadWriteCloserWithAddr) (*rpc.RPCNonceRequest, error) {
74
+func (t *middleTelegram) sendRPCNonceRequest(conn io.Writer) (*rpc.RPCNonceRequest, error) {
74
 	rpcNonceReq, err := rpc.NewRPCNonceRequest(t.proxySecret)
75
 	rpcNonceReq, err := rpc.NewRPCNonceRequest(t.proxySecret)
75
 	if err != nil {
76
 	if err != nil {
76
 		return nil, errors.Annotate(err, "Cannot create RPC nonce request")
77
 		return nil, errors.Annotate(err, "Cannot create RPC nonce request")
82
 	return rpcNonceReq, nil
83
 	return rpcNonceReq, nil
83
 }
84
 }
84
 
85
 
85
-func (t *middleTelegram) receiveRPCNonceResponse(conn wrappers.ReadWriteCloserWithAddr, req *rpc.RPCNonceRequest) (*rpc.RPCNonceResponse, error) {
86
+func (t *middleTelegram) receiveRPCNonceResponse(conn io.Reader, req *rpc.RPCNonceRequest) (*rpc.RPCNonceResponse, error) {
86
 	ans, err := ioutil.ReadAll(conn)
87
 	ans, err := ioutil.ReadAll(conn)
87
 	if err != nil {
88
 	if err != nil {
88
 		return nil, errors.Annotate(err, "Cannot read RPC nonce response")
89
 		return nil, errors.Annotate(err, "Cannot read RPC nonce response")
98
 	return rpcNonceResp, nil
99
 	return rpcNonceResp, nil
99
 }
100
 }
100
 
101
 
101
-func (t *middleTelegram) sendRPCHandshakeRequest(conn wrappers.ReadWriteCloserWithAddr) (*rpc.RPCHandshakeRequest, error) {
102
+func (t *middleTelegram) sendRPCHandshakeRequest(conn io.Writer) (*rpc.RPCHandshakeRequest, error) {
102
 	req := rpc.NewRPCHandshakeRequest()
103
 	req := rpc.NewRPCHandshakeRequest()
103
 	if _, err := conn.Write(req.Bytes()); err != nil {
104
 	if _, err := conn.Write(req.Bytes()); err != nil {
104
 		return nil, errors.Annotate(err, "Cannot send RPC handshake request")
105
 		return nil, errors.Annotate(err, "Cannot send RPC handshake request")
107
 	return req, nil
108
 	return req, nil
108
 }
109
 }
109
 
110
 
110
-func (t *middleTelegram) receiveRPCHandshakeResponse(conn wrappers.ReadWriteCloserWithAddr, req *rpc.RPCHandshakeRequest) (*rpc.RPCHandshakeResponse, error) {
111
+func (t *middleTelegram) receiveRPCHandshakeResponse(conn io.Reader, req *rpc.RPCHandshakeRequest) (*rpc.RPCHandshakeResponse, error) {
111
 	ans, err := ioutil.ReadAll(conn)
112
 	ans, err := ioutil.ReadAll(conn)
112
 	if err != nil {
113
 	if err != nil {
113
 		return nil, errors.Annotate(err, "Cannot read RPC handshake response")
114
 		return nil, errors.Annotate(err, "Cannot read RPC handshake response")

Laden…
Annuleren
Opslaan