Determine clinet IP from TCP connection

[woleium]

Hello everyone!

I can't work out how to determine the connecting client IP when using the TCPStartup() function

I'd like to limit connections based on IP, so if someone connects with the wrong IP the session is disconnected.

Any ideas?

Edited May 18, 2009 by woleium

[FreeFry]

Have you tried searching the forum?

I'm fairly sure I've seen a function that does what you want somewhere in here.

Edit:

Here you go:

Func _TCPSocketToIP(ByRef $ivOSocket)
    Local $stvSockAddr = DllStructCreate("short;ushort;ptr;char[8]")
    Local $avResult = DllCall("Ws2_32.dll", "int", "getpeername", "int", $ivOSocket, "ptr", DllStructGetPtr($stvSockAddr), "ptr*", DllStructGetSize($stvSockAddr))
    If Not IsArray($avResult) Or (IsArray($avResult) And $avResult[0]) Then Return(SetError(1, 0, 0)); Something went wrong
    $avResult = DllCall("Ws2_32.dll", "str*", "inet_ntoa", "ptr", DllStructGetData($stvSockAddr, 3))
    If Not IsArray($avResult) Or (IsArray($avResult) And Not $avResult[0]) Then Return(SetError(2, 0, 0)); Something else went wrong
    Return $avResult[0]
EndFunc

An example of how to use it:

TCPStartup()

Dim $ioSocket = TCPConnect(TCPNameToIP("www.google.com"), 80)
If $ioSocket = -1 Then Exit; Could not open socket

ConsoleWrite(_TCPSocketToIP($ioSocket) & @LF)

TCPCloseSocket($ioSocket)
TCPShutdown()

Func _TCPSocketToIP(ByRef $ivOSocket)
    Local $stvSockAddr = DllStructCreate("short;ushort;ptr;char[8]")
    Local $avResult = DllCall("Ws2_32.dll", "int", "getpeername", "int", $ivOSocket, "ptr", DllStructGetPtr($stvSockAddr), "ptr*", DllStructGetSize($stvSockAddr))
    If Not IsArray($avResult) Or (IsArray($avResult) And $avResult[0]) Then Return(SetError(1, 0, 0)); Something went wrong
    $avResult = DllCall("Ws2_32.dll", "str*", "inet_ntoa", "ptr", DllStructGetData($stvSockAddr, 3))
    If Not IsArray($avResult) Or (IsArray($avResult) And Not $avResult[0]) Then Return(SetError(2, 0, 0)); Something else went wrong
    Return $avResult[0]
EndFunc

Edited May 18, 2009 by FreeFry

[woleium]

Have you tried searching the forum?

I'm fairly sure I've seen a function that does what you want somewhere in here.

Yep, I looked for ages... I must have been searching for the wrong thing. Thank you very much :-)