Просмотр исходного кода

Now it is a time to start to make a handshake

tags/0.9
9seconds 7 лет назад
Родитель
Сommit
6db04980ff
4 измененных файлов: 14 добавлений и 9 удалений
  1. 0
    6
      mtproto/extras.go
  2. 5
    0
      mtproto/rpc/rpc.go
  3. 2
    2
      mtproto/rpc/rpc_proxy_request.go
  4. 7
    1
      telegram/middle.go

+ 0
- 6
mtproto/extras.go Просмотреть файл

1
-package mtproto
2
-
3
-type Extras struct {
4
-	QuickAck  bool
5
-	SimpleAck bool
6
-}

+ 5
- 0
mtproto/rpc/rpc.go Просмотреть файл

4
 	RPCNonceSeqNo     = -2
4
 	RPCNonceSeqNo     = -2
5
 	RPCHandshakeSeqNo = -1
5
 	RPCHandshakeSeqNo = -1
6
 )
6
 )
7
+
8
+type Extras struct {
9
+	QuickAck  bool
10
+	SimpleAck bool
11
+}

+ 2
- 2
mtproto/rpc/rpc_proxy_request.go Просмотреть файл

28
 	RemoteIPPort [rpcProxyRequestIPPortLength]byte
28
 	RemoteIPPort [rpcProxyRequestIPPortLength]byte
29
 	LocalIPPort  [rpcProxyRequestIPPortLength]byte
29
 	LocalIPPort  [rpcProxyRequestIPPortLength]byte
30
 	ADTag        []byte
30
 	ADTag        []byte
31
+	Extras       Extras
31
 	Message      *bytes.Buffer
32
 	Message      *bytes.Buffer
32
-	Extras       *mtproto.Extras
33
 }
33
 }
34
 
34
 
35
 func (r *RPCProxyRequest) Bytes() []byte {
35
 func (r *RPCProxyRequest) Bytes() []byte {
65
 	return buf.Bytes()
65
 	return buf.Bytes()
66
 }
66
 }
67
 
67
 
68
-func NewRPCProxyRequest(connectionType mtproto.ConnectionType, local, remote *net.TCPAddr, adTag []byte, extras *mtproto.Extras) (*RPCProxyRequest, error) {
68
+func NewRPCProxyRequest(connectionType mtproto.ConnectionType, local, remote *net.TCPAddr, adTag []byte, extras Extras) (*RPCProxyRequest, error) {
69
 	flags := RPCProxyRequestFlagsHasAdTag | RPCProxyRequestFlagsMagic | RPCProxyRequestFlagsExtMode2
69
 	flags := RPCProxyRequestFlagsHasAdTag | RPCProxyRequestFlagsMagic | RPCProxyRequestFlagsExtMode2
70
 
70
 
71
 	switch connectionType {
71
 	switch connectionType {

+ 7
- 1
telegram/middle.go Просмотреть файл

8
 	"go.uber.org/zap"
8
 	"go.uber.org/zap"
9
 
9
 
10
 	"github.com/9seconds/mtg/config"
10
 	"github.com/9seconds/mtg/config"
11
+	"github.com/9seconds/mtg/mtproto"
12
+	"github.com/9seconds/mtg/wrappers"
11
 )
13
 )
12
 
14
 
13
 type middleTelegram struct {
15
 type middleTelegram struct {
14
 	middleTelegramCaller
16
 	middleTelegramCaller
15
 }
17
 }
16
 
18
 
17
-func NewMiddleTelegram(conf *config.Config, logger *zap.SugaredLogger) *middleTelegram {
19
+func NewMiddleTelegram(conf *config.Config, logger *zap.SugaredLogger) Telegram {
18
 	tg := &middleTelegram{
20
 	tg := &middleTelegram{
19
 		middleTelegramCaller: middleTelegramCaller{
21
 		middleTelegramCaller: middleTelegramCaller{
20
 			baseTelegram: baseTelegram{
22
 			baseTelegram: baseTelegram{
35
 
37
 
36
 	return tg
38
 	return tg
37
 }
39
 }
40
+
41
+func (t *middleTelegram) Init(connOpts *mtproto.ConnectionOpts, conn wrappers.ReadWriteCloserWithAddr) (wrappers.ReadWriteCloserWithAddr, error) {
42
+	return nil, nil
43
+}

Загрузка…
Отмена
Сохранить