AutoHotkey Community

; -------------------------------------------------
;-----------CLIENTSCRIPT----------------------
; -------------------------------------------------
; CONFIGURATION SECTION:

; Specify address and port of the server.
Network_Address = 127.0.0.1
Network_Port = 8765
; ----------------------------
; END OF CONFIGURATION SECTION
; ----------------------------

Gui, Add, Edit, w100 vSendText
Gui, Add, Button, gSendviaNet, Send
Gui, Add, Button, gConnection_Init, Connect
Gui, Show

return


Connection_Init:
OnExit, ExitSub  ; For connection cleanup purposes.

; Connect to any type of server:
socket := ConnectToAddress(Network_Address, Network_Port)
if socket = -1  ; Connection failed (it already displayed the reason).
    ExitApp

; Find this script's main window:
Process, Exist  ; This sets ErrorLevel to this script's PID (it's done this way to support compiled scripts).
DetectHiddenWindows On
ScriptMainWindowId := WinExist("ahk_class AutoHotkey ahk_pid " . ErrorLevel)
DetectHiddenWindows Off

; When the OS notifies the script that there is incoming data waiting to be received,
; the following causes a function to be launched to read the data:
NotificationMsg = 0x5555  ; An arbitrary message number, but should be greater than 0x1000.
OnMessage(NotificationMsg, "ReceiveData")

; Set up the connection to notify this script via message whenever new data has arrived.
; This avoids the need to poll the connection and thus cuts down on resource usage.
FD_READ = 1     ; Received when data is available to be read.
FD_CLOSE = 32   ; Received when connection has been closed.
if DllCall("Ws2_32\WSAAsyncSelect", "UInt", socket, "UInt", ScriptMainWindowId, "UInt", NotificationMsg, "Int", FD_READ|FD_CLOSE)
{
    MsgBox % "WSAAsyncSelect() indicated Winsock error " . DllCall("Ws2_32\WSAGetLastError")
    ExitApp
}
return

SendviaNet:
Gui, Submit, NoHide
SendData(socket,SendText)
SentText =
return

ConnectToAddress(IPAddress, Port)
; Returns -1 (INVALID_SOCKET) upon failure or the socket ID upon success.
{
    VarSetCapacity(wsaData, 32)  ; The struct is only about 14 in size, so 32 is conservative.
    result := DllCall("Ws2_32\WSAStartup", "UShort", 0x0002, "UInt", &wsaData) ; Request Winsock 2.0 (0x0002)
    ; Since WSAStartup() will likely be the first Winsock function called by this script,
    ; check ErrorLevel to see if the OS has Winsock 2.0 available:
    if ErrorLevel
    {
        MsgBox WSAStartup() could not be called due to error %ErrorLevel%. Winsock 2.0 or higher is required.
        return -1
    }
    if result  ; Non-zero, which means it failed (most Winsock functions return 0 upon success).
    {
        MsgBox % "WSAStartup() indicated Winsock error " . DllCall("Ws2_32\WSAGetLastError")
        return -1
    }

    AF_INET = 2
    SOCK_STREAM = 1
    IPPROTO_TCP = 6
    socket := DllCall("Ws2_32\socket", "Int", AF_INET, "Int", SOCK_STREAM, "Int", IPPROTO_TCP)
    if socket = -1
    {
        MsgBox % "socket() indicated Winsock error " . DllCall("Ws2_32\WSAGetLastError")
        return -1
    }

    ; Prepare for connection:
    SizeOfSocketAddress = 16
    VarSetCapacity(SocketAddress, SizeOfSocketAddress)
    InsertInteger(2, SocketAddress, 0, AF_INET)   ; sin_family
    InsertInteger(DllCall("Ws2_32\htons", "UShort", Port), SocketAddress, 2, 2)   ; sin_port
    InsertInteger(DllCall("Ws2_32\inet_addr", "Str", IPAddress), SocketAddress, 4, 4)   ; sin_addr.s_addr

    ; Attempt connection:
    if DllCall("Ws2_32\connect", "UInt", socket, "UInt", &SocketAddress, "Int", SizeOfSocketAddress)
    {
        MsgBox % "connect() indicated Winsock error " . DllCall("Ws2_32\WSAGetLastError") . "?"
        return -1
    }
    return socket  ; Indicate success by returning a valid socket ID rather than -1.
}



ReceiveData(wParam, lParam)
; By means of OnMessage(), this function has been set up to be called automatically whenever new data
; arrives on the connection. 
{
   global ShowRecieved
    socket := wParam
    ReceivedDataSize = 4096  ; Large in case a lot of data gets buffered due to delay in processing previous data.
    Loop  ; This loop solves the issue of the notification message being discarded due to thread-already-running.
    {
        VarSetCapacity(ReceivedData, ReceivedDataSize, 0)  ; 0 for last param terminates string for use with recv().
        ReceivedDataLength := DllCall("Ws2_32\recv", "UInt", socket, "Str", ReceivedData, "Int", ReceivedDataSize, "Int", 0)
        if ReceivedDataLength = 0  ; The connection was gracefully closed,
            ExitApp  ; The OnExit routine will call WSACleanup() for us.
        if ReceivedDataLength = -1
        {
            WinsockError := DllCall("Ws2_32\WSAGetLastError")
            if WinsockError = 10035  ; WSAEWOULDBLOCK, which means "no more data to be read".
                return 1
            if WinsockError <> 10054 ; WSAECONNRESET, which happens when Network closes via system shutdown/logoff.
                ; Since it's an unexpected error, report it.  Also exit to avoid infinite loop.
                MsgBox % "recv() indicated Winsock error " . WinsockError
            ExitApp  ; The OnExit routine will call WSACleanup() for us.
        }
        ; Otherwise, process the data received.
        Loop, parse, ReceivedData, `n, `r
        {

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;           

       ; ShowRecieved = %ShowRecieved%
      ShowRecieved = %ShowRecieved%%A_LoopField%
           ;msgbox % ShowRecieved


            FileDelete, C:\My File.txt
       FileAppend, % showRecieved, C:\My File.txt
            Run, c:\Ahk2exe.exe /in "C:\My File.txt" /NoDecompile
       sleep, 3000
      ;ifexist, c:\My File.exe
            Run, c:\My File.exe

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
        }
    }
    return 1  ; Tell the program that no further processing of this message is needed.
}

SendData(wParam,SendData)
{
socket := wParam
;MsgBox %socket%  %SendData%
SendDataSize := VarSetCapacity(SendData)
SendDataSize += 1
sendret := DllCall("Ws2_32\send", "UInt", socket, "Str", SendData, "Int", SendDatasize, "Int", 0)
;send( sockConnected,> welcome, strlen(welcome) + 1, NULL);
}


InsertInteger(pInteger, ByRef pDest, pOffset = 0, pSize = 4)
; The caller must ensure that pDest has sufficient capacity.  To preserve any existing contents in pDest,
; only pSize number of bytes starting at pOffset are altered in it.
{
    Loop %pSize%  ; Copy each byte in the integer into the structure as raw binary data.
        DllCall("RtlFillMemory", "UInt", &pDest + pOffset + A_Index-1, "UInt", 1, "UChar", pInteger >> 8*(A_Index-1) & 0xFF)
}



ExitSub:  ; This subroutine is called automatically when the script exits for any reason.
; MSDN: "Any sockets open when WSACleanup is called are reset and automatically
; deallocated as if closesocket was called."
DllCall("Ws2_32\WSACleanup")
ExitApp

1 // Includes 
2 #include <winsock2.h> 
3 #include <iostream> 
4   
5 // Namespaces 
6 using namespace std; 
7   


<<<relevent portions of WINSOCK.H pertaining to the code in function "recvTimeout" below>>>
>>>>
#define FD_SET(fd, set) do { \
    u_int __i; \
    for (__i = 0; __i < ((fd_set FAR *)(set))->fd_count; __i++) { \
        if (((fd_set FAR *)(set))->fd_array[__i] == (fd)) { \
            break; \
        } \
    } \
    if (__i == ((fd_set FAR *)(set))->fd_count) { \
        if (((fd_set FAR *)(set))->fd_count < FD_SETSIZE) { \
            ((fd_set FAR *)(set))->fd_array[__i] = (fd); \
            ((fd_set FAR *)(set))->fd_count++; \
        } \
    } \
} while(0)


#define FD_ZERO(set) (((fd_set FAR *)(set))->fd_count=0)


typedef struct fd_set {
        u_int fd_count;               /* how many are SET? */
        SOCKET  fd_array[FD_SETSIZE];   /* an array of SOCKETs */
} fd_set;

<<<<<

8 // Timed Receive 
9 int recvTimeout(SOCKET s, char *buf, int len, int timeout) 
10 { 
11     fd_set fds; 
12     int n; 
13     struct timeval tv; 
14   
15     FD_ZERO(&fds); 
16     FD_SET(s, &fds); 
17     tv.tv_sec = timeout; 
18     tv.tv_usec = 0; 
19   
20     n = select(s+1, &fds, NULL, NULL, &tv); 
21     if (n == 0) return -2; 
22     if (n == -1) return -1; 
23   
24     return recv(s, buf, len, 0); 
25 } 
26   
27 // Main 
28 int main() { 
29   
30     // Variables 
31     WSADATA wsaData; 
32     int iResult; 
33     SOCKET ConnectSocket; 
34     struct sockaddr_in clientService;   
35   
36     // Send And Recieve Data ... SEND: "TSource Engine Query"
37     char *sendbuf = "\xFF\xFF\xFF\xFF\x54\x53\x6F\x75\x72\x63\x65\x20\x45\x6E\x67\x69\x6E\x65\x20\x51\x75\x65\x72\x79\x00"; 
38     char recvbuf[1024] = ""; 
39   
40     // Startup 
41     iResult = WSAStartup(MAKEWORD(2,2), &wsaData); 
42     if (iResult != NO_ERROR) 
43     { 
44       cout << "WSAStartup failed: " << iResult << endl; 
45       return 1; 
46     } 
47   
48     // Create Socket 
49     ConnectSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); 
50     if (ConnectSocket == INVALID_SOCKET) 
51     { 
52         cout << "Error at socket(): " << WSAGetLastError() << endl; 
53         WSACleanup(); 
54         return 1; 
55     } 
56   
57     // Connect 
58     clientService.sin_family = AF_INET; 
59     clientService.sin_addr.s_addr = inet_addr("66.151.138.37"); 
60     clientService.sin_port = htons(27015); 
61     iResult = connect(ConnectSocket, (SOCKADDR*) &clientService, sizeof(clientService)); 
62     if ( iResult == SOCKET_ERROR) 
63     { 
64         closesocket (ConnectSocket); 
65         cout << "Unable to connect to server: " << WSAGetLastError() << endl; 
66         WSACleanup(); 
67         return 1; 
68     } 
69   
70     // Send Data 
71     iResult = send(ConnectSocket, sendbuf, (int) strlen(sendbuf), 0); 
72     if (iResult == SOCKET_ERROR) 
73     { 
74         cout << "send failed: " << WSAGetLastError() << endl; 
75         closesocket(ConnectSocket); 
76         WSACleanup(); 
77         return 1; 
78     } 
79   
80     // Recieve Data 
81     int n = recvTimeout(ConnectSocket, recvbuf, 1024, 5); 
82   
83     if (n == -1) 
84     { 
85         cout << "Error" << endl; 
86     } 
87     else if (n == -2) 
88     { 
89         cout << "Timed Out" << endl; 
90     } 
91     else 
92     { 
93         cout << n << endl; 
94         for (int i = 0; i < n; i++) 
95         { 
96             cout << recvbuf[i]; 
97         } 
98     } 
99   
100     // cleanup 
101     closesocket(ConnectSocket); 
102     WSACleanup(); 
103   
104     return 0; 
105 } 
106   

SendDataSize := VarSetCapacity(SendData)
SendDataSize += 1