ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-09-01 12:24:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

Propagate client addr with connection options

Browse Source
This commit is contained in:
9seconds
2018-07-02 16:14:11 +03:00
parent 09476cc467
commit 9f746aa699
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
client/direct.go
+1
View File
@@ -32,6 +32,7 @@ func DirectInit(conn net.Conn, conf *config.Config) (*mtproto.ConnectionOpts, wr
return nil, nil, errors.Annotate(err, "Cannot parse obfuscated frame") return nil, nil, errors.Annotate(err, "Cannot parse obfuscated frame")
} }
connOpts.ConnectionProto = mtproto.ConnectionProtocolAny connOpts.ConnectionProto = mtproto.ConnectionProtocolAny
connOpts.ClientAddr = conn.RemoteAddr().(*net.TCPAddr)
socket := wrappers.NewTimeoutRWC(conn) socket := wrappers.NewTimeoutRWC(conn)
socket = wrappers.NewStreamCipherRWC(socket, obfs2.Encryptor, obfs2.Decryptor) socket = wrappers.NewStreamCipherRWC(socket, obfs2.Encryptor, obfs2.Decryptor)
mtproto/connection_options.go
+2
View File
@@ -2,6 +2,7 @@ package mtproto
import ( import (
"bytes" "bytes"
"net"
"github.com/juju/errors" "github.com/juju/errors"
) )
@@ -18,6 +19,7 @@ type ConnectionOpts struct {
DC int16 DC int16
ConnectionType ConnectionType ConnectionType ConnectionType
ConnectionProto ConnectionProtocol ConnectionProto ConnectionProtocol
ClientAddr *net.TCPAddr
} }
// Different connection types which user requests from Telegram. // Different connection types which user requests from Telegram.