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

Merge pull request #379 from 9seconds/fix-telegram-ips

Show ip of telegram endpoints in event stream
tags/v2.2.2^2^2
Sergei Arkhipov 1 месяц назад
Родитель
Сommit
b153a55149
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 8 добавлений и 1 удалений
  1. 8
    1
      mtglib/proxy.go

+ 8
- 1
mtglib/proxy.go Просмотреть файл

259
 		ctx:      ctx,
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
 	p.eventStream.Send(ctx,
269
 	p.eventStream.Send(ctx,
263
 		NewEventConnectedToDC(ctx.streamID,
270
 		NewEventConnectedToDC(ctx.streamID,
264
-			conn.RemoteAddr().(*net.TCPAddr).IP, //nolint: forcetypeassert
271
+			net.ParseIP(telegramHost),
265
 			ctx.dc),
272
 			ctx.dc),
266
 	)
273
 	)
267
 
274
 

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