Quellcode durchsuchen

Show real IP of the telegram endpoint in event stream

tags/v2.2.2^2^2
9seconds vor 1 Monat
Ursprung
Commit
913a38d13a
1 geänderte Dateien mit 8 neuen und 1 gelöschten Zeilen
  1. 8
    1
      mtglib/proxy.go

+ 8
- 1
mtglib/proxy.go Datei anzeigen

@@ -259,9 +259,16 @@ func (p *Proxy) doTelegramCall(ctx *streamContext) error {
259 259
 		ctx:      ctx,
260 260
 	}
261 261
 
262
+	telegramHost, _, err := net.SplitHostPort(foundAddr.Address)
263
+	if err != nil {
264
+		conn.Close() //nolint: errcheck
265
+
266
+		return fmt.Errorf("cannot parse telegram address %s: %w", foundAddr.Address, err)
267
+	}
268
+
262 269
 	p.eventStream.Send(ctx,
263 270
 		NewEventConnectedToDC(ctx.streamID,
264
-			conn.RemoteAddr().(*net.TCPAddr).IP, //nolint: forcetypeassert
271
+			net.ParseIP(telegramHost),
265 272
 			ctx.dc),
266 273
 	)
267 274
 

Laden…
Abbrechen
Speichern