mirror of
https://github.com/ScuroNeko/mtg.git
synced 2026-08-31 23:24:01 +03:00
Add obfuscated2 server handshake
This commit is contained in:
+6
-52
@@ -2,66 +2,20 @@ package mtglib
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net"
|
||||
"time"
|
||||
)
|
||||
|
||||
type connStandard struct {
|
||||
conn net.Conn
|
||||
idleTimeout time.Duration
|
||||
}
|
||||
|
||||
func (c connStandard) Read(b []byte) (int, error) {
|
||||
if err := c.conn.SetReadDeadline(time.Now().Add(c.idleTimeout)); err != nil {
|
||||
return 0, fmt.Errorf("cannot set read deadline: %w", err)
|
||||
}
|
||||
|
||||
return c.conn.Read(b)
|
||||
}
|
||||
|
||||
func (c connStandard) Write(b []byte) (int, error) {
|
||||
if err := c.conn.SetWriteDeadline(time.Now().Add(c.idleTimeout)); err != nil {
|
||||
return 0, fmt.Errorf("cannot set write deadline: %w", err)
|
||||
}
|
||||
|
||||
return c.conn.Write(b)
|
||||
}
|
||||
|
||||
func (c connStandard) Close() error {
|
||||
return c.conn.Close()
|
||||
}
|
||||
|
||||
func (c connStandard) LocalAddr() net.Addr {
|
||||
return c.conn.LocalAddr()
|
||||
}
|
||||
|
||||
func (c connStandard) RemoteAddr() net.Addr {
|
||||
return c.conn.RemoteAddr()
|
||||
}
|
||||
|
||||
func (c connStandard) SetDeadline(t time.Time) error {
|
||||
return c.conn.SetDeadline(t)
|
||||
}
|
||||
|
||||
func (c connStandard) SetReadDeadline(t time.Time) error {
|
||||
return c.conn.SetReadDeadline(t)
|
||||
}
|
||||
|
||||
func (c connStandard) SetWriteDeadline(t time.Time) error {
|
||||
return c.conn.SetWriteDeadline(t)
|
||||
}
|
||||
|
||||
type connEventTraffic struct {
|
||||
connStandard
|
||||
type connTelegramTraffic struct {
|
||||
net.Conn
|
||||
|
||||
connID string
|
||||
stream EventStream
|
||||
ctx context.Context
|
||||
}
|
||||
|
||||
func (c connEventTraffic) Read(b []byte) (int, error) {
|
||||
n, err := c.connStandard.Read(b)
|
||||
func (c connTelegramTraffic) Read(b []byte) (int, error) {
|
||||
n, err := c.Conn.Read(b)
|
||||
|
||||
if n > 0 {
|
||||
c.stream.Send(c.ctx, EventTraffic{
|
||||
@@ -75,8 +29,8 @@ func (c connEventTraffic) Read(b []byte) (int, error) {
|
||||
return n, err
|
||||
}
|
||||
|
||||
func (c connEventTraffic) Write(b []byte) (int, error) {
|
||||
n, err := c.connStandard.Write(b)
|
||||
func (c connTelegramTraffic) Write(b []byte) (int, error) {
|
||||
n, err := c.Conn.Write(b)
|
||||
|
||||
if n > 0 {
|
||||
c.stream.Send(c.ctx, EventTraffic{
|
||||
|
||||
Reference in New Issue
Block a user