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

Debug direct connection

Browse Source
This commit is contained in:
9seconds
2018-07-07 20:45:10 +03:00
parent f82ff1f6fe
commit cd63483503
6 changed files with 32 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
proxy/direct.go
+4 -6
View File
@@ -21,25 +21,23 @@ func NewProxyDirect(conf *config.Config) *Proxy {
return &Proxy{
conf: conf,
acceptCallback: func(ctx context.Context, cancel context.CancelFunc, clientSocket net.Conn,
connID string, wait *sync.WaitGroup, conf *config.Config) error {
connID string, wait *sync.WaitGroup, conf *config.Config) (io.Closer, io.Closer, error) {
client, opts, err := client.DirectInit(ctx, cancel, clientSocket, connID, conf)
if err != nil {
return errors.Annotate(err, "Cannot initialize client connection")
return nil, nil, errors.Annotate(err, "Cannot initialize client connection")
}
defer client.Close()
server, err := directTelegramStream(ctx, cancel, opts, connID, tg)
if err != nil {
return errors.Annotate(err, "Cannot initialize telegram connection")
return client, nil, errors.Annotate(err, "Cannot initialize telegram connection")
}
defer server.Close()
wait.Add(2)
go directPipe(client, server, wait)
go directPipe(server, client, wait)
return nil
return client, server, nil
},
}
}