TCP Issues

[yhu420]

Hello everyone,

I'm trying to create a small program that would allow me to send files across my network, therefore, I copy pasted the code available right here.

I modified the code a little to suit my needs, it is in the spoiler below:

expandcollapse popup

#Region ;**** Directives created by AutoIt3Wrapper_GUI ****
#AutoIt3Wrapper_Outfile_x64=server.exe
#EndRegion ;**** Directives created by AutoIt3Wrapper_GUI ****

#include <Misc.au3>
Global Const $WS_CAPTION = 0x00C00000
Global Const $WS_POPUP = 0x80000000
Global Const $WS_POPUPWINDOW = 0x80880000
Global Const $WS_EX_ACCEPTFILES = 0x00000010
Global Const $TCS_FIXEDWIDTH = 0x00000400
Global Const $PBS_SMOOTH = 1
Global Const $SS_CENTER = 1
Global Const $GUI_EVENT_CLOSE = -3
Global Const $GUI_SHOW = 16
Global Const $GUI_BKCOLOR_TRANSPARENT = -2
Global Const $GUI_DROPACCEPTED = 8
Global Const $ES_NUMBER = 8192

OnAutoItExitRegister("OnAutoItExit")
$GUI = GUICreate("TCP File Transfer - Intended server: " & @IPAddress1, 321, 228, -1, -1, BitOR($WS_POPUPWINDOW, $WS_CAPTION), $WS_EX_ACCEPTFILES)
GUICtrlCreateTab(10, 10, 300, 145, $TCS_FIXEDWIDTH)

GUICtrlCreateTabItem("Receiver")
GUICtrlSetState(-1, $GUI_SHOW)
$Receiver_Destination = GUICtrlCreateInput("", 24, 90, 187, 22)
GUICtrlCreateLabel("Select a destination directory. If one is not selected, it will be set to the desktop.", 24, 45, 272, 40)
GUICtrlSetFont(-1, 10)
$Receiver_Browse = GUICtrlCreateButton("Browse...", 221, 90, 75, 22, 0)
$Receiver_Wait = GUICtrlCreateButton("Wait for File", 24, 121, 272, 22, 0)

GUICtrlCreateTabItem("Sender")
GUICtrlCreateLabel("Select a file to send.", 29, 50, 262, 20)
GUICtrlSetFont(-1, 10)
$Sender_File = GUICtrlCreateInput("", 29, 80, 177, 22)
GUICtrlSetState(-1, $GUI_DROPACCEPTED)
$Sender_Browse = GUICtrlCreateButton("Browse...", 216, 80, 75, 22, 0)
$Sender_Chunk = GUICtrlCreateInput("5", 142, 113, 65, 22, $ES_NUMBER)
GUICtrlCreateLabel("Chunk Size in kb:", 29, 114, 107, 20)
GUICtrlSetFont(-1, 10)
GUICtrlSetLimit(-1, 2, 1)
$Sender_Send = GUICtrlCreateButton("Send", 216, 113, 75, 22, 0)
GUICtrlCreateTabItem("")

$Progress = GUICtrlCreateProgress(10, 192, 300, 25, $PBS_SMOOTH)
$Bytes_Label = GUICtrlCreateLabel("0kb / 0kb", 10, 168, 300, 20)
GUICtrlSetFont(-1, 10)
$Progress_Label = GUICtrlCreateLabel("0%", 10, 196, 300, 24, $SS_CENTER)
GUICtrlSetFont(-1, 12)
GUICtrlSetBkColor(-1, $GUI_BKCOLOR_TRANSPARENT)
GUISetState()

TCPStartup()
While 1
    $nMsg = GUIGetMsg()
    Switch $nMsg
        Case $GUI_EVENT_CLOSE
            Exit
        Case $Receiver_Browse
            $Dir = FileSelectFolder("Select a Destination Directory", @HomeDrive)
            If Not @error Then GUICtrlSetData($Receiver_Destination, $Dir)
            GUICtrlSetTip($Dir, GUICtrlRead($Receiver_Destination))
        Case $Receiver_Wait
            Select
                Case GUICtrlRead($Receiver_Destination) = ""
                    GUICtrlSetData($Receiver_Destination, @DesktopDir)
                Case DirGetSize(GUICtrlRead($Receiver_Destination)) = -1
                    GUICtrlSetData($Receiver_Destination, @DesktopDir)
            EndSelect
            _ReceiveFile(GUICtrlRead($Receiver_Destination), 5 * 1024 * 1024) ;file limit of 5 megabytes
            Switch @error
                Case 0
                    MsgBox(262208, "TCP Sender", "Download Complete")
                Case 1
                    MsgBox(16, "WSA:" & @error, "Unable to hook port!")
                Case 2
                    MsgBox(16, "WSA:" & @error, "Lost connection while receiving! Extended: " & @extended)
                Case 3
                    MsgBox(16, "WSA:" & @error, "Lost connection while receiving! Extended: " & @extended)
            EndSwitch
        Case $Sender_Browse
            $File = FileOpenDialog("Select a File To Send", @DesktopDir, "ALL (*.*)", 1 + 2)
            If Not @error Then GUICtrlSetData($Sender_File, $File)
            ;GUICtrlSetTip($Dir, GUICtrlRead($Sender_File))
        Case $Sender_Send
            Select
                Case Not FileExists(GUICtrlRead($Sender_File))
                    MsgBox(262192, "ERROR", "Invalid file!")
                Case GUICtrlRead($Sender_Chunk) > 20
                    MsgBox(262192, "ERROR", "Chunk size specified is too large!")
                Case GUICtrlRead($Sender_Chunk) < 5
                    MsgBox(262192, "ERROR", "Chunk size specified is too small!")
                Case Else
                    _SendFile(GUICtrlRead($Sender_File), GUICtrlRead($Sender_Chunk) * 1024)
                    Switch @error
                        Case 0
                            MsgBox(262208, "TCP Sender", "Upload Complete")
                        Case 1
                            MsgBox(16, "ERROR", "Bad file")
                        Case 2
                            MsgBox(16, "ERROR", "You can't send directories!")
                        Case 3
                            MsgBox(16, "WSA:" & @extended, "Unable to connect to the host!")
                        Case 4
                            MsgBox(16, "WSA:" & @error, "Lost connection while sending! Extended: " & @extended)
                        Case 5
                            MsgBox(16, "ERROR", "That's odd!" & @CRLF & 'I got: "' & @extended & '" from the socket!')
                        Case 6
                            MsgBox(16, "WSA:" & @error, "Lost connection while sending! Extended: " & @extended)
                        Case 7
                            MsgBox(16, "ERROR", "That's odd!" & @CRLF & 'I got: "' & @extended & '" from the socket!')
                    EndSwitch
            EndSelect
    EndSwitch
    Sleep(15)
WEnd

Func _SendFile($File, $MaxLen)
    Local $BytesRead = 0

    $Size = FileGetSize($File)
    If $Size = 0 Then Return SetError(1, 0, -1)

    ;get file name
    $Reg = StringRegExp($File, "(.)+\\((.)+)?", 3)
    ConsoleWrite($Reg & @CRLF)
    Select
        Case Not IsArray($Reg)
            $Name = $Reg
        Case UBound($Reg) < 2
            Return SetError(2, 0, -1)
        Case Else
            $Name = $Reg[1]
    EndSelect
    $IP = InputBox("IP Address", "What is the IP address of the server?", @IPAddress1)

    $Socket = TCPConnect($IP, 8081)
    If @error Then Return SetError(3, @error, -1)

    ;wait until something is sent
    Do
        $Receive = TCPRecv($Socket, 1000)
        ;If @error Then Return SetError(4, 0, -1)
        If @error Then
            ;MsgBox(0, "", "Error: " & @error & " in sending."); --> -1 error
            Return SetError(4, 0, -1)
        EndIf
        Sleep(10)
    Until $Receive <> ""

    If $Receive <> "Sending Data" Then Return SetError(5, $Receive, -1)

    ;Send Name and file size to receiver
    TCPSend($Socket, $Name & ":" & $Size)

    ;Wait for confirmation from receiver
    Do
        $Receive = TCPRecv($Socket, 1000)
        If @error Then SetError(6, 0, -1)
    Until $Receive <> ""

    If $Receive <> "Start Upload" Then SetError(7, $Receive, -1)

    ;open file to read in binary
    $FileHandle = FileOpen($File, 16)

    ;loop until the whole file is received
    While 1
        $Data = FileRead($FileHandle, $MaxLen)
        If @error Then ExitLoop

        $BytesRead += TCPSend($Socket, $Data)

        If GUICtrlRead($Progress) <> Round($BytesRead / $Size * 100) Then
            GUICtrlSetData($Progress, Round($BytesRead / $Size * 100))
            GUICtrlSetData($Bytes_Label, Round($BytesRead / 1024) & " kb / " & Round($Size / 1024) & " kb")
            GUICtrlSetData($Progress_Label, Round($BytesRead / $Size * 100) & "%")
        EndIf
    WEnd
    FileClose($FileHandle)
    TCPCloseSocket($Socket)

    Return SetError(0, 0, 1)
EndFunc   ;==>_SendFile

Func OnAutoItExit()
    TCPShutdown() ; Close the TCP service.
EndFunc   ;==>OnAutoItExit

Func _ReceiveFile($Destination, $MaxLen)

    Local $Return = False, $Bytes = 0

    $Listen = TCPListen(@IPAddress1, 8081)
    If @error Then Return SetError(1, 0, -1)

    ;ToolTip("Listening" & @CRLF & @IPAddress1 & @CRLF & 8081, 0, 0)

    ;wait for sender to connect
    Do
        $Socket = TCPAccept($Listen)
    Until $Socket <> -1

    ;If @error Then MsgBox(0, "", "Bug here")
    ;No bug here

    $Receive = TCPRecv($Socket, $MaxLen)
    ;If @error Then Return SetError(2, 0, -1)
    If @error Then
        ;MsgBox(0, "", "Error " & @error & " in receiving."); --> -1 error
        Return SetError(2, 0, -1)
    EndIf

    TCPSend($Socket, "Sending Data")

    ;get file data
    ;wait for file name and size to be sent
    Do
        $Receive = TCPRecv($Socket, 1000)
        If @error Then Return SetError(3, 0, -1)
        Sleep(5)
    Until $Receive <> ""
    $FileData = StringSplit($Receive, ":", 2)

    $Destination &= "\" & $FileData[0]

    $FileHandle = FileOpen($Destination, 16 + 2 + 8)

    ;initiate download
    TCPSend($Socket, "Start Upload")

    While 1
        $Data = TCPRecv($Socket, $MaxLen, 1) ;receive data in binary
        If @error Then ExitLoop

        $Bytes += BinaryLen($Data) ;count number of bytes sent

        FileWrite($FileHandle, $Data) ;recreate the file sent


        If GUICtrlRead($Progress) <> Round($Bytes / $FileData[1] * 100) Then
            GUICtrlSetData($Progress, Round($Bytes / $FileData[1] * 100))
            GUICtrlSetData($Bytes_Label, Round($Bytes / 1024) & " kb / " & Round($FileData[1] / 1024) & " kb")
            GUICtrlSetData($Progress_Label, Round($Bytes / $FileData[1] * 100) & "%")
        EndIf

    WEnd
    FileClose($FileHandle)
    TCPCloseSocket($Socket)
    TCPCloseSocket($Listen)
    ;ToolTip("")

    Return SetError(0, 0, 1)
EndFunc   ;==>_ReceiveFile

So I ran the program, clicked on "Wait for file" on my main computer, ran it again on my other computer, chose a fie, and I set the IP to the one of my first computer (192.168.x.x).

However, the I only get a loading time, and a WSA error 10060 (Time out).

I double checked with the examples of the official autoit doc, but unfortunately, I have no idea where my mistake is..

Do you guys have any idea?

Thanks

EDIT: removed unnecessary code

Edited December 5, 2015 by yhu420

[Jos]

Are you using a (windows) firewall and if so have you set it to allow port 8081?

Jos

[yhu420]

Yes, the first time I launched this program, it prompted me whether I should allow it or prevent it from using the ports. I clicked "Yes", and it never asked again.

I noticed two other things: I have VirtualBox installed, which makes @IPAddress2 my real IP on the network (I think).

The client/server example given in the TCPRecv doc works just fine. I think I will rather work from this.

[Jefrey]

Just to test if these computers can comunicate to each other, try:

1. Ping from one against another

2. Start a netcat server in one of them (nc -vv -l -p 8081) and try connecting from another (nc -vv 192.168.x.x 8081)