See protocol details in rss.cpp.
Signed-off-by: Michael Goldish<mgoldish@xxxxxxxxxx>
---
client/tests/kvm/deps/rss.cpp | 1429 ++++++++++++++++++++++++++++-------------
1 files changed, 970 insertions(+), 459 deletions(-)
diff --git a/client/tests/kvm/deps/rss.cpp b/client/tests/kvm/deps/rss.cpp
index 66d9a5b..b2f6049 100644
--- a/client/tests/kvm/deps/rss.cpp
+++ b/client/tests/kvm/deps/rss.cpp
@@ -1,459 +1,970 @@
-// Simple remote shell server
-// Author: Michael Goldish<mgoldish@xxxxxxxxxx>
-// Much of the code here was adapted from Microsoft code samples.
-
-// Usage: rss.exe [port]
-// If no port is specified the default is 22.
-
-#define _WIN32_WINNT 0x0500
-
-#include<windows.h>
-#include<winsock2.h>
-#include<stdio.h>
-
-#pragma comment(lib, "ws2_32.lib")
-
-int port = 22;
-
-HWND hMainWindow = NULL;
-HWND hTextBox = NULL;
-
-struct client_info {
- SOCKET socket;
- sockaddr_in addr;
- int pid;
- HWND hwnd;
- HANDLE hJob;
- HANDLE hChildOutputRead;
- HANDLE hThreadChildToSocket;
-};
-
-void ExitOnError(char *message, BOOL winsock = 0)
-{
- LPVOID system_message;
- char buffer[512];
-
- int error_code;
- if (winsock)
- error_code = WSAGetLastError();
- else
- error_code = GetLastError();
-
- WSACleanup();
-
- FormatMessage(
- FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM,
- NULL, error_code, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
- (LPTSTR)&system_message, 0, NULL);
-
- sprintf(buffer,
- "%s!\n"
- "Error code = %d\n"
- "Error message = %s",
- message, error_code, (char *)system_message);
-
- MessageBox(hMainWindow, buffer, "Error", MB_OK | MB_ICONERROR);
-
- LocalFree(system_message);
- ExitProcess(1);
-}
-
-void AppendMessage(char *message)
-{
- int length = GetWindowTextLength(hTextBox);
- SendMessage(hTextBox, EM_SETSEL, (WPARAM)length, (LPARAM)length);
- SendMessage(hTextBox, EM_REPLACESEL, (WPARAM)FALSE, (LPARAM)message);
-}
-
-void FormatStringForPrinting(char *dst, char *src, int size)
-{
- int j = 0;
-
- for (int i = 0; i< size&& src[i]; i++) {
- if (src[i] == '\n') {
- dst[j++] = '\\';
- dst[j++] = 'n';
- } else if (src[i] == '\r') {
- dst[j++] = '\\';
- dst[j++] = 'r';
- } else if (src[i] == '\t') {
- dst[j++] = '\\';
- dst[j++] = 't';
- } else if (src[i] == '\\') {
- dst[j++] = '\\';
- dst[j++] = '\\';
- } else dst[j++] = src[i];
- }
- dst[j] = 0;
-}
-
-char* GetClientIPAddress(client_info *ci)
-{
- char *address = inet_ntoa(ci->addr.sin_addr);
- if (address)
- return address;
- else
- return "unknown";
-}
-
-DWORD WINAPI ChildToSocket(LPVOID client_info_ptr)
-{
- char buffer[1024], message[1024];
- client_info ci;
- DWORD bytes_read;
- int bytes_sent;
-
- memcpy(&ci, client_info_ptr, sizeof(ci));
-
- while (1) {
- // Read data from the child's STDOUT/STDERR pipes
- if (!ReadFile(ci.hChildOutputRead,
- buffer, sizeof(buffer),
-&bytes_read, NULL) || !bytes_read) {
- if (GetLastError() == ERROR_BROKEN_PIPE)
- break; // Pipe done -- normal exit path
- else
- ExitOnError("ReadFile failed"); // Something bad happened
- }
- // Send data to the client
- bytes_sent = send(ci.socket, buffer, bytes_read, 0);
- /*
- // Make sure all the data was sent
- if (bytes_sent != bytes_read) {
- sprintf(message,
- "ChildToSocket: bytes read (%d) != bytes sent (%d)",
- bytes_read, bytes_sent);
- ExitOnError(message, 1);
- }
- */
- }
-
- AppendMessage("Child exited\r\n");
- shutdown(ci.socket, SD_BOTH);
-
- return 0;
-}
-
-DWORD WINAPI SocketToChild(LPVOID client_info_ptr)
-{
- char buffer[256], formatted_buffer[768];
- char message[1024], client_info_str[256];
- client_info ci;
- DWORD bytes_written;
- int bytes_received;
-
- memcpy(&ci, client_info_ptr, sizeof(ci));
-
- sprintf(client_info_str, "address %s, port %d",
- GetClientIPAddress(&ci), ci.addr.sin_port);
-
- sprintf(message, "New client connected (%s)\r\n", client_info_str);
- AppendMessage(message);
-
- while (1) {
- // Receive data from the socket
- ZeroMemory(buffer, sizeof(buffer));
- bytes_received = recv(ci.socket, buffer, sizeof(buffer), 0);
- if (bytes_received<= 0)
- break;
- // Report the data received
- FormatStringForPrinting(formatted_buffer, buffer, sizeof(buffer));
- sprintf(message, "Client (%s) entered text: \"%s\"\r\n",
- client_info_str, formatted_buffer);
- AppendMessage(message);
- // Send the data as a series of WM_CHAR messages to the console window
- for (int i=0; i<bytes_received; i++) {
- SendMessage(ci.hwnd, WM_CHAR, (WPARAM)buffer[i], 0);
- SendMessage(ci.hwnd, WM_SETFOCUS, 0, 0);
- }
- }
-
- sprintf(message, "Client disconnected (%s)\r\n", client_info_str);
- AppendMessage(message);
-
- // Attempt to terminate the child's process tree:
- // Using taskkill (where available)
- sprintf(buffer, "taskkill /PID %d /T /F", ci.pid);
- system(buffer);
- // .. and using TerminateJobObject()
- TerminateJobObject(ci.hJob, 0);
- // Wait for the ChildToSocket thread to terminate
- WaitForSingleObject(ci.hThreadChildToSocket, 10000);
- // In case the thread refuses to exit -- terminate it
- TerminateThread(ci.hThreadChildToSocket, 0);
- // Close the socket
- shutdown(ci.socket, SD_BOTH);
- closesocket(ci.socket);
-
- // Close unnecessary handles
- CloseHandle(ci.hJob);
- CloseHandle(ci.hThreadChildToSocket);
- CloseHandle(ci.hChildOutputRead);
-
- AppendMessage("SocketToChild thread exited\r\n");
-
- return 0;
-}
-
-void PrepAndLaunchRedirectedChild(client_info *ci,
- HANDLE hChildStdOut,
- HANDLE hChildStdErr)
-{
- PROCESS_INFORMATION pi;
- STARTUPINFO si;
-
- // Allocate a new console for the child
- HWND hwnd = GetForegroundWindow();
- FreeConsole();
- AllocConsole();
- ShowWindow(GetConsoleWindow(), SW_HIDE);
- if (hwnd)
- SetForegroundWindow(hwnd);
-
- // Set up the start up info struct.
- ZeroMemory(&si, sizeof(STARTUPINFO));
- si.cb = sizeof(STARTUPINFO);
- si.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
- si.hStdOutput = hChildStdOut;
- si.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
- si.hStdError = hChildStdErr;
- // Use this if you want to hide the child:
- si.wShowWindow = SW_HIDE;
- // Note that dwFlags must include STARTF_USESHOWWINDOW if you want to
- // use the wShowWindow flags.
-
- // Launch the process that you want to redirect.
- if (!CreateProcess(NULL, "cmd.exe", NULL, NULL, TRUE,
- 0, NULL, "C:\\",&si,&pi))
- ExitOnError("CreateProcess failed");
-
- // Close any unnecessary handles.
- if (!CloseHandle(pi.hThread))
- ExitOnError("CloseHandle failed");
-
- // Keep the process ID
- ci->pid = pi.dwProcessId;
- // Assign the process to a newly created JobObject
- ci->hJob = CreateJobObject(NULL, NULL);
- AssignProcessToJobObject(ci->hJob, pi.hProcess);
- // Keep the console window's handle
- ci->hwnd = GetConsoleWindow();
-
- // Detach from the child's console
- FreeConsole();
-}
-
-void SpawnSession(client_info *ci)
-{
- HANDLE hOutputReadTmp, hOutputRead, hOutputWrite;
- HANDLE hErrorWrite;
- SECURITY_ATTRIBUTES sa;
-
- // Set up the security attributes struct.
- sa.nLength = sizeof(SECURITY_ATTRIBUTES);
- sa.lpSecurityDescriptor = NULL;
- sa.bInheritHandle = TRUE;
-
- // Create the child output pipe.
- if (!CreatePipe(&hOutputReadTmp,&hOutputWrite,&sa, 0))
- ExitOnError("CreatePipe failed");
-
- // Create a duplicate of the output write handle for the std error
- // write handle. This is necessary in case the child application
- // closes one of its std output handles.
- if (!DuplicateHandle(GetCurrentProcess(), hOutputWrite,
- GetCurrentProcess(),&hErrorWrite, 0,
- TRUE, DUPLICATE_SAME_ACCESS))
- ExitOnError("DuplicateHandle failed");
-
- // Create new output read handle and the input write handles. Set
- // the Properties to FALSE. Otherwise, the child inherits the
- // properties and, as a result, non-closeable handles to the pipes
- // are created.
- if (!DuplicateHandle(GetCurrentProcess(), hOutputReadTmp,
- GetCurrentProcess(),
-&hOutputRead, // Address of new handle.
- 0, FALSE, // Make it uninheritable.
- DUPLICATE_SAME_ACCESS))
- ExitOnError("DuplicateHandle failed");
-
- // Close inheritable copies of the handles you do not want to be
- // inherited.
- if (!CloseHandle(hOutputReadTmp))
- ExitOnError("CloseHandle failed");
-
- PrepAndLaunchRedirectedChild(ci, hOutputWrite, hErrorWrite);
-
- ci->hChildOutputRead = hOutputRead;
-
- // Close pipe handles (do not continue to modify the parent).
- // You need to make sure that no handles to the write end of the
- // output pipe are maintained in this process or else the pipe will
- // not close when the child process exits and the ReadFile will hang.
- if (!CloseHandle(hOutputWrite)) ExitOnError("CloseHandle failed");
- if (!CloseHandle(hErrorWrite)) ExitOnError("CloseHandle failed");
-}
-
-DWORD WINAPI ListenThread(LPVOID param)
-{
- WSADATA wsaData;
- SOCKET ListenSocket = INVALID_SOCKET;
- sockaddr_in addr;
- int result, addrlen;
- client_info ci;
- HANDLE hThread;
-
- // Initialize Winsock
- result = WSAStartup(MAKEWORD(2,2),&wsaData);
- if (result)
- ExitOnError("Winsock initialization failed");
-
- // Create socket
- ListenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
- if (ListenSocket == INVALID_SOCKET)
- ExitOnError("Socket creation failed", 1);
-
- // Bind the socket
- addr.sin_family = AF_INET;
- addr.sin_addr.s_addr = htonl(INADDR_ANY);
- addr.sin_port = htons(port);
-
- result = bind(ListenSocket, (sockaddr *)&addr, sizeof(addr));
- if (result == SOCKET_ERROR)
- ExitOnError("bind failed", 1);
-
- // Start listening for incoming connections
- result = listen(ListenSocket, SOMAXCONN);
- if (result == SOCKET_ERROR)
- ExitOnError("listen failed", 1);
-
- // Inform the user
- AppendMessage("Waiting for clients to connect...\r\n");
-
- while (1) {
- addrlen = sizeof(ci.addr);
- ci.socket = accept(ListenSocket, (sockaddr *)&ci.addr,&addrlen);
- if (ci.socket == INVALID_SOCKET) {
- if (WSAGetLastError() == WSAEINTR)
- break;
- else
- ExitOnError("accept failed", 1);
- }
-
- // Under heavy load, spawning cmd.exe might take a while, so tell the
- // client to be patient
- char *message = "Please wait...\r\n";
- send(ci.socket, message, strlen(message), 0);
- // Spawn a new redirected cmd.exe process
- SpawnSession(&ci);
- // Start transferring data from the child process to the client
- hThread = CreateThread(NULL, 0, ChildToSocket, (LPVOID)&ci, 0, NULL);
- if (!hThread)
- ExitOnError("Could not create ChildToSocket thread");
- ci.hThreadChildToSocket = hThread;
- // ... and from the client to the child process
- hThread = CreateThread(NULL, 0, SocketToChild, (LPVOID)&ci, 0, NULL);
- if (!hThread)
- ExitOnError("Could not create SocketToChild thread");
- }
-
- return 0;
-}
-
-LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
-{
- RECT rect;
- HANDLE hListenThread;
-
- switch (msg) {
- case WM_CREATE:
- // Create text box
- GetClientRect(hwnd,&rect);
- hTextBox = CreateWindowEx(WS_EX_CLIENTEDGE,
- "EDIT", "",
- WS_CHILD|WS_VISIBLE|WS_VSCROLL|
- ES_MULTILINE|ES_AUTOVSCROLL,
- 20, 20,
- rect.right - 40,
- rect.bottom - 40,
- hwnd,
- NULL,
- GetModuleHandle(NULL),
- NULL);
- if (!hTextBox)
- ExitOnError("Could not create text box");
-
- // Set the font
- SendMessage(hTextBox, WM_SETFONT,
- (WPARAM)GetStockObject(DEFAULT_GUI_FONT),
- MAKELPARAM(FALSE, 0));
-
- // Start the listening thread
- hListenThread =
- CreateThread(NULL, 0, ListenThread, NULL, 0, NULL);
- if (!hListenThread)
- ExitOnError("Could not create server thread");
- break;
-
- case WM_DESTROY:
- WSACleanup();
- PostQuitMessage(0);
- break;
-
- default:
- return DefWindowProc(hwnd, msg, wParam, lParam);
- }
-
- return 0;
-}
-
-int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
- LPSTR lpCmdLine, int nShowCmd)
-{
- WNDCLASSEX wc;
- MSG msg;
-
- if (strlen(lpCmdLine))
- sscanf(lpCmdLine, "%d",&port);
-
- // Make sure the firewall is disabled
- system("netsh firewall set opmode disable");
-
- // Create the window class
- wc.cbSize = sizeof(WNDCLASSEX);
- wc.style = CS_HREDRAW | CS_VREDRAW;
- wc.lpfnWndProc = WndProc;
- wc.cbClsExtra = 0;
- wc.cbWndExtra = 0;
- wc.hInstance = hInstance;
- wc.hIcon = LoadIcon(NULL, IDI_APPLICATION);
- wc.hIconSm = LoadIcon(NULL, IDI_APPLICATION);
- wc.hbrBackground = (HBRUSH)(COLOR_BTNFACE + 1);
- wc.lpszMenuName = NULL;
- wc.lpszClassName = "RemoteShellServerWindowClass";
- wc.hCursor = LoadCursor(NULL, IDC_ARROW);
-
- if (!RegisterClassEx(&wc))
- ExitOnError("Could not register window class");
-
- // Create the main window
- hMainWindow =
- CreateWindow("RemoteShellServerWindowClass",
- "Remote Shell Server",
- WS_OVERLAPPED|WS_CAPTION|WS_SYSMENU|WS_MINIMIZEBOX,
- 20, 20, 500, 300,
- NULL, NULL, hInstance, NULL);
- if (!hMainWindow)
- ExitOnError("Could not create window");
-
- ShowWindow(hMainWindow, SW_SHOWMINNOACTIVE);
- UpdateWindow(hMainWindow);
-
- // Main message loop
- while (GetMessage(&msg, NULL, 0, 0)) {
- TranslateMessage(&msg);
- DispatchMessage(&msg);
- }
-
- ExitProcess(0);
-}
+// Simple remote shell server (and file transfer server)
+// Author: Michael Goldish<mgoldish@xxxxxxxxxx>
+// Much of the code here was adapted from Microsoft code samples.
+
+// Usage: rss.exe [shell port] [file transfer port]
+// If no shell port is specified the default is 22.
+// If no file transfer port is specified the default is 23.
+
+// Definitions:
+// A 'msg' is a 32 bit integer.
+// A 'packet' is a 64 bit unsigned integer followed by a string of bytes.
+// The 64 bit integer indicates the length of the string.
+
+// Protocol for file transfers:
+//
+// When uploading files to the server:
+// 1. The client connects.
+// 2. The server sends RSS_MAGIC.
+// 3. The client sends RSS_SET_PATH, followed by a packet (as defined above)
+// containing the path (in the server's filesystem) where files are to be
+// stored. The 'current path' is set to the path given.
+// Uploading a file (optional, can be repeated many times):
+// 4. The client sends RSS_CREATE_FILE, followed by a packet containing the
+// filename (filename only, without a path), followed by a packet
+// containing the file's contents. The file is stored in the current
+// path.
+// Uploading a directory (optional, can be repeated many times):
+// 5. The client sends RSS_CREATE_DIR, followed by a packet containing the
+// name of the directory to be created (directory name only, without a
+// path). The directory is created in the current path. Then, the
+// 'current path' becomes the path of that new directory.
+// 6. Steps 4-5 may be repeated to upload files and directories to the new
+// directory.
+// 7. The client sends RSS_LEAVE_DIR, and the current path is set to the path
+// of its parent directory (e.g. C:\Foobar -> C:\).
+// 8. The client sends RSS_DONE and waits for a response.
+// 9. The server sends RSS_OK to indicate that it's still listening.
+// 10. Steps 3-9 are repeated as many times as necessary.
+// If a critical error occurs at any time, the server may send RSS_ERROR
+// followed by a packet containing an error message, and the connection is
+// closed.
+//
+// When downloading files from the server:
+// 1. The client connects.
+// 2. The server sends RSS_MAGIC.
+// 3. The client sends RSS_SET_PATH, followed by a packet (as defined above)
+// containing a path (in the server's filesystem) or a wildcard pattern
+// indicating the files/directories the client wants to download.
+// The server then searches the given path. For every file found:
+// 4. The server sends RSS_CREATE_FILE, followed by a packet containing the
+// filename (filename only, without a path), followed by a packet
+// containing the file's contents.
+// For every directory found:
+// 5. The server sends RSS_CREATE_DIR, followed by a packet containing the
+// name of the directory to be created (directory name only, without a
+// path).
+// 6. Steps 4-5 are repeated as many times as necessary.
+// 7. The server sends RSS_LEAVE_DIR.
+// 8. The server sends RSS_DONE.
+// 9. Steps 3-8 are repeated as many times as necessary.
+// If a critical error occurs, the server may send RSS_ERROR followed by a
+// packet containing an error message, and the connection is closed.
+// RSS_ERROR may only be sent when the client expects a msg.
+
+#define _WIN32_WINNT 0x0500
+
+#include<winsock2.h>
+#include<windows.h>
+#include<stdio.h>
+#include<stdarg.h>
+#include<shlwapi.h>
+
+#pragma comment(lib, "ws2_32.lib")
+#pragma comment(lib, "shlwapi.lib")
+
+#define TEXTBOX_LIMIT 32767
+
+// Constants for file transfer server
+#define RSS_MAGIC 0x525353
+#define RSS_OK 1
+#define RSS_ERROR 2
+#define RSS_UPLOAD 3
+#define RSS_DOWNLOAD 4
+#define RSS_SET_PATH 5
+#define RSS_CREATE_FILE 6
+#define RSS_CREATE_DIR 7
+#define RSS_LEAVE_DIR 8
+#define RSS_DONE 9
+
+// Globals
+int shell_port = 22;
+int file_transfer_port = 23;
+
+HWND hMainWindow = NULL;
+HWND hTextBox = NULL;
+
+struct client_info {
+ SOCKET socket;
+ sockaddr_in addr;
+ char addr_str[256];
+ int pid;
+ HWND hwnd;
+ HANDLE hJob;
+ HANDLE hChildOutputRead;
+ HANDLE hThreadChildToSocket;
+};
+
+void ExitOnError(char *message, BOOL winsock = FALSE)
+{
+ LPVOID system_message;
+ char buffer[512];
+ int error_code;
+
+ if (winsock)
+ error_code = WSAGetLastError();
+ else
+ error_code = GetLastError();
+ WSACleanup();
+
+ FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
+ FORMAT_MESSAGE_FROM_SYSTEM,
+ NULL,
+ error_code,
+ MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
+ (LPTSTR)&system_message,
+ 0,
+ NULL);
+ sprintf(buffer,
+ "%s!\n"
+ "Error code = %d\n"
+ "Error message = %s",
+ message, error_code, (char *)system_message);
+ MessageBox(hMainWindow, buffer, "Error", MB_OK | MB_ICONERROR);
+
+ LocalFree(system_message);
+ ExitProcess(1);
+}
+
+void _vAppendMessage(char *message, va_list args)
+{
+ char str[512] = {0};
+
+ vsnprintf(str, sizeof(str) - 1, message, args);
+ strncat(str, "\r\n", sizeof(str) - 1 - strlen(str));
+
+ int length = GetWindowTextLength(hTextBox);
+ SendMessage(hTextBox, EM_SETSEL, (WPARAM)length, (LPARAM)length);
+ SendMessage(hTextBox, EM_REPLACESEL, (WPARAM)FALSE, (LPARAM)str);
+}
+
+// Append text to the textbox
+// (if the textbox is full, do nothing)
+void _AppendMessage(char *message, ...)
+{
+ va_list args;
+
+ va_start(args, message);
+ _vAppendMessage(message, args);
+ va_end(args);
+}
+
+// Append text to the textbox
+// (if the textbox is full or nearly full, remove old text first)
+void AppendMessage(char *message, ...)
+{
+ va_list args;
+
+ int length = GetWindowTextLength(hTextBox);
+ if (length> TEXTBOX_LIMIT - 512) {
+ SendMessage(hTextBox, EM_SETSEL, (WPARAM)0,
+ (LPARAM)(TEXTBOX_LIMIT / 2));
+ SendMessage(hTextBox, EM_REPLACESEL, (WPARAM)FALSE, (LPARAM)"...");
+ if (length == TEXTBOX_LIMIT) _AppendMessage("...");
+ }
+ va_start(args, message);
+ _vAppendMessage(message, args);
+ va_end(args);
+}
+
+void FormatStringForPrinting(char *dst, char *src, int size)
+{
+ int j = 0;
+
+ for (int i = 0; i< size&& src[i]; i++) {
+ if (src[i] == '\n') {
+ dst[j++] = '\\';
+ dst[j++] = 'n';
+ } else if (src[i] == '\r') {
+ dst[j++] = '\\';
+ dst[j++] = 'r';
+ } else if (src[i] == '\t') {
+ dst[j++] = '\\';
+ dst[j++] = 't';
+ } else if (src[i] == '\\') {
+ dst[j++] = '\\';
+ dst[j++] = '\\';
+ } else dst[j++] = src[i];
+ }
+ dst[j] = 0;
+}
+
+char* GetClientIPAddress(client_info *ci)
+{
+ char *address = inet_ntoa(ci->addr.sin_addr);
+ if (address)
+ return address;
+ else
+ return "unknown";
+}
+
+// Read a small chunk of data into a buffer
+BOOL Receive(SOCKET socket, char *buffer, int len)
+{
+ while (len> 0) {
+ int bytes_received = recv(socket, buffer, len, 0);
+ if (bytes_received<= 0)
+ return FALSE;
+ buffer += bytes_received;
+ len -= bytes_received;
+ }
+ return TRUE;
+}
+
+// Send data from a buffer
+BOOL Send(SOCKET socket, char *buffer, int len)
+{
+ while (len> 0) {
+ int bytes_sent = send(socket, buffer, len, 0);
+ if (bytes_sent<= 0)
+ return FALSE;
+ buffer += bytes_sent;
+ len -= bytes_sent;
+ }
+ return TRUE;
+}
+
+/*-------------
+ * Shell server
+ *-------------*/
+
+DWORD WINAPI ChildToSocket(LPVOID client_info_ptr)
+{
+ client_info *ci = (client_info *)client_info_ptr;
+ char buffer[1024];
+ DWORD bytes_read;
+
+ while (1) {
+ // Read data from the child's STDOUT/STDERR pipes
+ if (!ReadFile(ci->hChildOutputRead,
+ buffer, sizeof(buffer),
+&bytes_read, NULL) || !bytes_read) {
+ if (GetLastError() == ERROR_BROKEN_PIPE)
+ break; // Pipe done -- normal exit path
+ else
+ ExitOnError("ReadFile failed"); // Something bad happened
+ }
+ // Send data to the client
+ Send(ci->socket, buffer, bytes_read);
+ }
+
+ AppendMessage("Child exited");
+ closesocket(ci->socket);
+ return 0;
+}
+
+DWORD WINAPI SocketToChild(LPVOID client_info_ptr)
+{
+ client_info *ci = (client_info *)client_info_ptr;
+ char buffer[256], formatted_buffer[768];
+ int bytes_received;
+
+ AppendMessage("Shell server: new client connected (%s)", ci->addr_str);
+
+ while (1) {
+ // Receive data from the socket
+ ZeroMemory(buffer, sizeof(buffer));
+ bytes_received = recv(ci->socket, buffer, sizeof(buffer), 0);
+ if (bytes_received<= 0)
+ break;
+ // Report the data received
+ FormatStringForPrinting(formatted_buffer, buffer, sizeof(buffer));
+ _AppendMessage("Client (%s) entered text: \"%s\"",
+ ci->addr_str, formatted_buffer);
+ // Send the data as a series of WM_CHAR messages to the console window
+ for (int i = 0; i< bytes_received; i++) {
+ SendMessage(ci->hwnd, WM_CHAR, (WPARAM)buffer[i], 0);
+ SendMessage(ci->hwnd, WM_SETFOCUS, 0, 0);
+ }
+ }
+
+ AppendMessage("Shell server: client disconnected (%s)", ci->addr_str);
+
+ // Attempt to terminate the child's process tree:
+ // Using taskkill (where available)
+ sprintf(buffer, "taskkill /PID %d /T /F", ci->pid);
+ system(buffer);
+ // .. and using TerminateJobObject()
+ TerminateJobObject(ci->hJob, 0);
+ // Wait for the ChildToSocket thread to terminate
+ WaitForSingleObject(ci->hThreadChildToSocket, 10000);
+ // In case the thread refuses to exit, terminate it
+ TerminateThread(ci->hThreadChildToSocket, 0);
+ // Close the socket
+ closesocket(ci->socket);
+
+ // Free resources
+ CloseHandle(ci->hJob);
+ CloseHandle(ci->hThreadChildToSocket);
+ CloseHandle(ci->hChildOutputRead);
+ free(ci);
+
+ AppendMessage("SocketToChild thread exited");
+ return 0;
+}
+
+void PrepAndLaunchRedirectedChild(client_info *ci,
+ HANDLE hChildStdOut,
+ HANDLE hChildStdErr)
+{
+ PROCESS_INFORMATION pi;
+ STARTUPINFO si;
+
+ // Allocate a new console for the child
+ HWND hwnd = GetForegroundWindow();
+ FreeConsole();
+ AllocConsole();
+ ShowWindow(GetConsoleWindow(), SW_HIDE);
+ if (hwnd)
+ SetForegroundWindow(hwnd);
+
+ // Set up the start up info struct.
+ ZeroMemory(&si, sizeof(STARTUPINFO));
+ si.cb = sizeof(STARTUPINFO);
+ si.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
+ si.hStdOutput = hChildStdOut;
+ si.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
+ si.hStdError = hChildStdErr;
+ // Use this if you want to hide the child:
+ si.wShowWindow = SW_HIDE;
+ // Note that dwFlags must include STARTF_USESHOWWINDOW if you want to
+ // use the wShowWindow flags.
+
+ // Launch the process that you want to redirect.
+ if (!CreateProcess(NULL, "cmd.exe", NULL, NULL, TRUE,
+ 0, NULL, "C:\\",&si,&pi))
+ ExitOnError("CreateProcess failed");
+
+ // Close any unnecessary handles.
+ if (!CloseHandle(pi.hThread))
+ ExitOnError("CloseHandle failed");
+
+ // Keep the process ID
+ ci->pid = pi.dwProcessId;
+ // Assign the process to a newly created JobObject
+ ci->hJob = CreateJobObject(NULL, NULL);
+ AssignProcessToJobObject(ci->hJob, pi.hProcess);
+ // Keep the console window's handle
+ ci->hwnd = GetConsoleWindow();
+
+ // Detach from the child's console
+ FreeConsole();
+}
+
+void SpawnSession(client_info *ci)
+{
+ HANDLE hOutputReadTmp, hOutputRead, hOutputWrite;
+ HANDLE hErrorWrite;
+ SECURITY_ATTRIBUTES sa;
+
+ // Set up the security attributes struct.
+ sa.nLength = sizeof(SECURITY_ATTRIBUTES);
+ sa.lpSecurityDescriptor = NULL;
+ sa.bInheritHandle = TRUE;
+
+ // Create the child output pipe.
+ if (!CreatePipe(&hOutputReadTmp,&hOutputWrite,&sa, 0))
+ ExitOnError("CreatePipe failed");
+
+ // Create a duplicate of the output write handle for the std error
+ // write handle. This is necessary in case the child application
+ // closes one of its std output handles.
+ if (!DuplicateHandle(GetCurrentProcess(), hOutputWrite,
+ GetCurrentProcess(),&hErrorWrite, 0,
+ TRUE, DUPLICATE_SAME_ACCESS))
+ ExitOnError("DuplicateHandle failed");
+
+ // Create new output read handle and the input write handles. Set
+ // the Properties to FALSE. Otherwise, the child inherits the
+ // properties and, as a result, non-closeable handles to the pipes
+ // are created.
+ if (!DuplicateHandle(GetCurrentProcess(), hOutputReadTmp,
+ GetCurrentProcess(),
+&hOutputRead, // Address of new handle.
+ 0, FALSE, // Make it uninheritable.
+ DUPLICATE_SAME_ACCESS))
+ ExitOnError("DuplicateHandle failed");
+
+ // Close inheritable copies of the handles you do not want to be
+ // inherited.
+ if (!CloseHandle(hOutputReadTmp))
+ ExitOnError("CloseHandle failed");
+
+ PrepAndLaunchRedirectedChild(ci, hOutputWrite, hErrorWrite);
+
+ ci->hChildOutputRead = hOutputRead;
+
+ // Close pipe handles (do not continue to modify the parent).
+ // You need to make sure that no handles to the write end of the
+ // output pipe are maintained in this process or else the pipe will
+ // not close when the child process exits and the ReadFile will hang.
+ if (!CloseHandle(hOutputWrite)) ExitOnError("CloseHandle failed");
+ if (!CloseHandle(hErrorWrite)) ExitOnError("CloseHandle failed");
+}
+
+SOCKET PrepListenSocket(int port)
+{
+ sockaddr_in addr;
+ linger l;
+ int result;
+
+ // Create socket
+ SOCKET ListenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+ if (ListenSocket == INVALID_SOCKET)
+ ExitOnError("Socket creation failed", TRUE);
+
+ // Enable lingering
+ l.l_linger = 10;
+ l.l_onoff = 1;
+ setsockopt(ListenSocket, SOL_SOCKET, SO_LINGER, (char *)&l, sizeof(l));
+
+ // Bind the socket
+ addr.sin_family = AF_INET;
+ addr.sin_addr.s_addr = htonl(INADDR_ANY);
+ addr.sin_port = htons(port);
+
+ result = bind(ListenSocket, (sockaddr *)&addr, sizeof(addr));
+ if (result == SOCKET_ERROR)
+ ExitOnError("bind failed", TRUE);
+
+ // Start listening for incoming connections
+ result = listen(ListenSocket, SOMAXCONN);
+ if (result == SOCKET_ERROR)
+ ExitOnError("listen failed", TRUE);
+
+ return ListenSocket;
+}
+
+DWORD WINAPI ShellListenThread(LPVOID param)
+{
+ client_info _ci, *ci;
+ HANDLE hThread;
+
+ SOCKET ListenSocket = PrepListenSocket(shell_port);
+
+ // Inform the user
+ AppendMessage("Shell server: waiting for clients to connect...");
+
+ while (1) {
+ int addrlen = sizeof(_ci.addr);
+ _ci.socket = accept(ListenSocket, (sockaddr *)&_ci.addr,&addrlen);
+ if (_ci.socket == INVALID_SOCKET) {
+ if (WSAGetLastError() == WSAEINTR)
+ break;
+ else
+ ExitOnError("accept failed", TRUE);
+ }
+
+ if (!(ci = (client_info *)malloc(sizeof(client_info))))
+ ExitOnError("Could not allocate client_info struct");
+ memcpy(ci,&_ci, sizeof(client_info));
+ sprintf(ci->addr_str, "%s:%d", GetClientIPAddress(ci),
+ ci->addr.sin_port);
+
+ // Under heavy load, spawning cmd.exe might take a while, so tell the
+ // client to be patient
+ char *message = "Please wait...\r\n";
+ Send(ci->socket, message, strlen(message));
+ // Spawn a new redirected cmd.exe process
+ SpawnSession(ci);
+ // Start transferring data from the child process to the client
+ hThread = CreateThread(NULL, 0, ChildToSocket, (LPVOID)ci, 0, NULL);
+ if (!hThread)
+ ExitOnError("Could not create ChildToSocket thread");
+ ci->hThreadChildToSocket = hThread;
+ // ... and from the client to the child process
+ hThread = CreateThread(NULL, 0, SocketToChild, (LPVOID)ci, 0, NULL);
+ if (!hThread)
+ ExitOnError("Could not create SocketToChild thread");
+ }
+
+ return 0;
+}
+
+/*---------------------
+ * File transfer server
+ *---------------------*/
+
+#define BUFSIZE 65536
+
+// Receive up to 4GB into a file
+BOOL ReceiveIntoFile(SOCKET socket, char *filename, DWORD len)
+{
+ char *buffer = (char *)malloc(BUFSIZE);
+ if (!buffer) return FALSE;
+
+ FILE *fp = fopen(filename, "wb");
+ if (!fp) {
+ free(buffer);
+ return FALSE;
+ }
+
+ while (len> 0) {
+ int bytes_received = recv(socket, buffer, min(BUFSIZE, len), 0);
+ if (bytes_received<= 0)
+ break;
+ if (fwrite(buffer, bytes_received, 1, fp)< 1)
+ break;
+ len -= bytes_received;
+ }
+
+ fclose(fp);
+ free(buffer);
+ return len == 0;
+}
+
+// Send data from a file (unlimited size)
+BOOL SendFromFile(SOCKET socket, char *filename)
+{
+ char *buffer = (char *)malloc(BUFSIZE);
+ if (!buffer) return FALSE;
+
+ FILE *fp = fopen(filename, "rb");
+ if (!fp) {
+ free(buffer);
+ return FALSE;
+ }
+
+ while (!feof(fp)) {
+ int bytes_read = fread(buffer, 1, BUFSIZE, fp);
+ if (bytes_read<= 0)
+ break;
+ if (!Send(socket, buffer, bytes_read)) {
+ fclose(fp);
+ free(buffer);
+ return FALSE;
+ }
+ }
+
+ BOOL success = feof(fp);
+ fclose(fp);
+ free(buffer);
+ return success;
+}
+
+BOOL ReceivePacket(SOCKET socket, char *buffer, DWORD max_size)
+{
+ DWORD packet_size = 0, padding;
+
+ if (!Receive(socket, (char *)&packet_size, 4)) return FALSE;
+ if (!Receive(socket, (char *)&padding, 4)) return FALSE;
+ if (packet_size> max_size) return FALSE;
+ return Receive(socket, buffer, packet_size);
+}
+
+BOOL ReceiveStrPacket(SOCKET socket, char *buffer, DWORD max_size)
+{
+ memset(buffer, 0, max_size);
+ return ReceivePacket(socket, buffer, max_size - 1);
+}
+
+BOOL SendPacket(SOCKET socket, char *buffer, DWORD len)
+{
+ DWORD padding = 0;
+
+ if (!Send(socket, (char *)&len, 4)) return FALSE;
+ if (!Send(socket, (char *)&padding, 4)) return FALSE;
+ return Send(socket, buffer, len);
+}
+
+BOOL ReceivePacketIntoFile(SOCKET socket, char *filename)
+{
+ DWORD packet_size = 0, padding;
+
+ if (!Receive(socket, (char *)&packet_size, 4)) return FALSE;
+ if (!Receive(socket, (char *)&padding, 4)) return FALSE;
+ return ReceiveIntoFile(socket, filename, packet_size);
+}
+
+BOOL SendPacketFromFile(SOCKET socket, char *filename,
+ DWORD size_low, DWORD size_high)
+{
+ if (!Send(socket, (char *)&size_low, 4)) return FALSE;
+ if (!Send(socket, (char *)&size_high, 4)) return FALSE;
+ return SendFromFile(socket, filename);
+}
+
+BOOL SendMsg(SOCKET socket, int msg)
+{
+ return Send(socket, (char *)&msg, 4);
+}
+
+int TerminateTransfer(client_info *ci, char *message)
+{
+ AppendMessage(message);
+ AppendMessage("File transfer server: client disconnected (%s)",
+ ci->addr_str);
+ closesocket(ci->socket);
+ free(ci);
+ return 0;
+}
+
+int TerminateWithError(client_info *ci, char *message)
+{
+ SendMsg(ci->socket, RSS_ERROR);
+ SendPacket(ci->socket, message, strlen(message));
+ return TerminateTransfer(ci, message);
+}
+
+int ReceiveThread(client_info *ci)
+{
+ char path[512], filename[512];
+ int msg = 0;
+
+ AppendMessage("Client (%s) wants to upload files", ci->addr_str);
+
+ while (1) {
+ if (!Receive(ci->socket, (char *)&msg, 4))
+ return TerminateTransfer(ci, "Could not receive further msgs");
+
+ switch (msg) {
+ case RSS_SET_PATH:
+ if (!ReceiveStrPacket(ci->socket, path, sizeof(path)))
+ return TerminateWithError(ci,
+ "RSS_SET_PATH: could not receive path");
+ _AppendMessage("Client (%s) set path to %s", ci->addr_str, path);
+ break;
+
+ case RSS_CREATE_FILE:
+ if (!ReceiveStrPacket(ci->socket, filename, sizeof(filename)))
+ return TerminateWithError(ci,
+ "RSS_CREATE_FILE: could not receive filename");
+ if (PathIsDirectory(path))
+ PathAppend(path, filename);
+ _AppendMessage("Client (%s) is uploading %s", ci->addr_str, path);
+ if (!ReceivePacketIntoFile(ci->socket, path))
+ return TerminateWithError(ci,
+ "RSS_CREATE_FILE: error receiving or writing file "
+ "contents");
+ PathAppend(path, "..");
+ break;
+
+ case RSS_CREATE_DIR:
+ if (!ReceiveStrPacket(ci->socket, filename, sizeof(filename)))
+ return TerminateWithError(ci,
+ "RSS_CREATE_DIR: could not receive dirname");
+ if (PathIsDirectory(path))
+ PathAppend(path, filename);
+ _AppendMessage("Entering dir %s", path);
+ if (PathFileExists(path)) {
+ if (!PathIsDirectory(path))
+ return TerminateWithError(ci,
+ "RSS_CREATE_DIR: path exists and is not a directory");
+ } else {
+ if (!CreateDirectory(path, NULL))
+ return TerminateWithError(ci,
+ "RSS_CREATE_DIR: could not create directory");
+ }
+ break;
+
+ case RSS_LEAVE_DIR:
+ PathAppend(path, "..");
+ _AppendMessage("Returning to dir %s", path);
+ break;
+
+ case RSS_DONE:
+ if (!SendMsg(ci->socket, RSS_OK))
+ return TerminateTransfer(ci,
+ "RSS_DONE: could not send OK msg");
+ break;
+
+ default:
+ return TerminateWithError(ci, "Received unexpected msg");
+ }
+ }
+}
+
+// Given a path or a pattern with wildcards, send files or directory trees to
+// the client
+int SendFiles(client_info *ci, char *pattern)
+{
+ char path[MAX_PATH];
+ WIN32_FIND_DATA ffd;
+ FILE *fp;
+
+ HANDLE hFind = FindFirstFile(pattern,&ffd);
+ if (hFind == INVALID_HANDLE_VALUE) {
+ // If a weird error occurred (like failure to list directory contents
+ // due to insufficient permissions) print a warning and continue.
+ if (GetLastError() != ERROR_FILE_NOT_FOUND)
+ AppendMessage("WARNING: FindFirstFile failed on pattern %s",
+ pattern);
+ return 1;
+ }
+
+ strncpy(path, pattern, sizeof(path) - 1);
+ PathAppend(path, "..");
+
+ do {
+ if (ffd.dwFileAttributes& FILE_ATTRIBUTE_REPARSE_POINT)
+ continue;
+ if (ffd.dwFileAttributes& FILE_ATTRIBUTE_DIRECTORY) {
+ // Directory
+ if (!strcmp(ffd.cFileName, ".") || !strcmp(ffd.cFileName, ".."))
+ continue;
+ PathAppend(path, ffd.cFileName);
+ _AppendMessage("Entering dir %s", path);
+ PathAppend(path, "*");
+ if (!SendMsg(ci->socket, RSS_CREATE_DIR)) {
+ FindClose(hFind);
+ return TerminateTransfer(ci,
+ "Could not send RSS_CREATE_DIR msg");
+ }
+ if (!SendPacket(ci->socket, ffd.cFileName,
+ strlen(ffd.cFileName))) {
+ FindClose(hFind);
+ return TerminateTransfer(ci, "Could not send dirname");
+ }
+ if (!SendFiles(ci, path)) {
+ FindClose(hFind);
+ return 0;
+ }
+ if (!SendMsg(ci->socket, RSS_LEAVE_DIR)) {
+ FindClose(hFind);
+ return TerminateTransfer(ci,
+ "Could not send RSS_LEAVE_DIR msg");
+ }
+ PathAppend(path, "..");
+ PathAppend(path, "..");
+ _AppendMessage("Returning to dir %s", path);
+ } else {
+ // File
+ PathAppend(path, ffd.cFileName);
+ _AppendMessage("Client (%s) is downloading %s",
+ ci->addr_str, path);
+ // Make sure the file is readable
+ fp = fopen(path, "rb");
+ if (fp) fclose(fp);
+ else {
+ AppendMessage("WARNING: could not read file %s", path);
+ PathAppend(path, "..");
+ continue;
+ }
+ if (!SendMsg(ci->socket, RSS_CREATE_FILE)) {
+ FindClose(hFind);
+ return TerminateTransfer(ci,
+ "Could not send RSS_CREATE_FILE msg");
+ }
+ if (!SendPacket(ci->socket, ffd.cFileName,
+ strlen(ffd.cFileName))) {
+ FindClose(hFind);
+ return TerminateTransfer(ci, "Could not send filename");
+ }
+ if (!SendPacketFromFile(ci->socket, path,
+ ffd.nFileSizeLow, ffd.nFileSizeHigh)) {
+ FindClose(hFind);
+ return TerminateTransfer(ci, "Could not send file contents");
+ }
+ PathAppend(path, "..");
+ }
+ } while (FindNextFile(hFind,&ffd));
+
+ if (GetLastError() == ERROR_NO_MORE_FILES) {
+ FindClose(hFind);
+ return 1;
+ } else {
+ FindClose(hFind);
+ return TerminateWithError(ci, "FindNextFile failed");
+ }
+}
+
+int SendThread(client_info *ci)
+{
+ char pattern[512];
+ int msg = 0;
+
+ AppendMessage("Client (%s) wants to download files", ci->addr_str);
+
+ while (1) {
+ if (!Receive(ci->socket, (char *)&msg, 4))
+ return TerminateTransfer(ci, "Could not receive further msgs");
+
+ switch (msg) {
+ case RSS_SET_PATH:
+ if (!ReceiveStrPacket(ci->socket, pattern, sizeof(pattern)))
+ return TerminateWithError(ci,
+ "RSS_SET_PATH: could not receive path");
+ AppendMessage("Client (%s) asked for %s", ci->addr_str, pattern);
+ PathRemoveBackslash(pattern);
+ if (!SendFiles(ci, pattern))
+ return 0;
+ if (!SendMsg(ci->socket, RSS_DONE))
+ return TerminateTransfer(ci,
+ "RSS_SET_PATH: could not send RSS_DONE msg");
+ break;
+
+ default:
+ return TerminateWithError(ci, "Received unexpected msg");
+ }
+ }
+}
+
+DWORD WINAPI TransferThreadEntry(LPVOID client_info_ptr)
+{
+ client_info *ci = (client_info *)client_info_ptr;
+ int msg = 0;
+
+ AppendMessage("File transfer server: new client connected (%s)",
+ ci->addr_str);
+
+ if (!SendMsg(ci->socket, RSS_MAGIC))
+ return TerminateTransfer(ci, "Could not send greeting message");
+ if (!Receive(ci->socket, (char *)&msg, 4))
+ return TerminateTransfer(ci, "Error receiving msg");
+
+ if (msg == RSS_UPLOAD)
+ return ReceiveThread(ci);
+ else if (msg == RSS_DOWNLOAD)
+ return SendThread(ci);
+ return TerminateWithError(ci, "Received unexpected msg");
+}
+
+DWORD WINAPI FileTransferListenThread(LPVOID param)
+{
+ client_info _ci, *ci;
+ HANDLE hThread;
+
+ SOCKET ListenSocket = PrepListenSocket(file_transfer_port);
+
+ // Inform the user
+ AppendMessage("File transfer server: waiting for clients to connect...");
+
+ while (1) {
+ int addrlen = sizeof(_ci.addr);
+ _ci.socket = accept(ListenSocket, (sockaddr *)&_ci.addr,&addrlen);
+ if (_ci.socket == INVALID_SOCKET) {
+ if (WSAGetLastError() == WSAEINTR)
+ break;
+ else
+ ExitOnError("accept failed", TRUE);
+ }
+
+ if (!(ci = (client_info *)malloc(sizeof(client_info))))
+ ExitOnError("Could not allocate client_info struct");
+ memcpy(ci,&_ci, sizeof(client_info));
+ sprintf(ci->addr_str, "%s:%d", GetClientIPAddress(ci),
+ ci->addr.sin_port);
+ hThread = CreateThread(NULL, 0, TransferThreadEntry, (LPVOID)ci, 0,
+ NULL);
+ if (!hThread)
+ ExitOnError("Could not create MainTransferThread");
+ }
+
+ return 0;
+}
+
+/*--------------------
+ * WndProc and WinMain
+ *--------------------*/
+
+LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
+{
+ RECT rect;
+ WSADATA wsaData;
+
+ switch (msg) {
+ case WM_CREATE:
+ // Create text box
+ GetClientRect(hwnd,&rect);
+ hTextBox = CreateWindowEx(WS_EX_CLIENTEDGE,
+ "EDIT", "",
+ WS_CHILD | WS_VISIBLE | WS_VSCROLL |
+ ES_MULTILINE | ES_AUTOVSCROLL,
+ 20, 20,
+ rect.right - 40,
+ rect.bottom - 40,
+ hwnd,
+ NULL,
+ GetModuleHandle(NULL),
+ NULL);
+ if (!hTextBox)
+ ExitOnError("Could not create text box");
+ // Set font
+ SendMessage(hTextBox, WM_SETFONT,
+ (WPARAM)GetStockObject(DEFAULT_GUI_FONT),
+ MAKELPARAM(FALSE, 0));
+ // Set size limit
+ SendMessage(hTextBox, EM_LIMITTEXT, (WPARAM)TEXTBOX_LIMIT, (LPARAM)0);
+ // Initialize Winsock
+ if (WSAStartup(MAKEWORD(2,2),&wsaData))
+ ExitOnError("Winsock initialization failed");
+ // Start the listening threads
+ if (!CreateThread(NULL, 0, ShellListenThread, NULL, 0, NULL))
+ ExitOnError("Could not create shell server thread");
+ if (!CreateThread(NULL, 0, FileTransferListenThread, NULL, 0, NULL))
+ ExitOnError("Could not create file transfer server thread");
+ break;
+
+ case WM_DESTROY:
+ if (WSACleanup())
+ ExitOnError("WSACleanup failed");
+ PostQuitMessage(0);
+ break;
+
+ default:
+ return DefWindowProc(hwnd, msg, wParam, lParam);
+ }
+
+ return 0;
+}
+
+int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
+ LPSTR lpCmdLine, int nShowCmd)
+{
+ WNDCLASSEX wc;
+ MSG msg;
+
+ if (strlen(lpCmdLine))
+ sscanf(lpCmdLine, "%d %d",&shell_port,&file_transfer_port);
+
+ // Make sure the firewall is disabled
+ system("netsh firewall set opmode disable");
+
+ // Create the window class
+ wc.cbSize = sizeof(WNDCLASSEX);
+ wc.style = CS_HREDRAW | CS_VREDRAW;
+ wc.lpfnWndProc = WndProc;
+ wc.cbClsExtra = 0;
+ wc.cbWndExtra = 0;
+ wc.hInstance = hInstance;
+ wc.hIcon = LoadIcon(NULL, IDI_APPLICATION);
+ wc.hIconSm = LoadIcon(NULL, IDI_APPLICATION);
+ wc.hbrBackground = (HBRUSH)(COLOR_BTNFACE + 1);
+ wc.lpszMenuName = NULL;
+ wc.lpszClassName = "RemoteShellServerWindowClass";
+ wc.hCursor = LoadCursor(NULL, IDC_ARROW);
+
+ if (!RegisterClassEx(&wc))
+ ExitOnError("Could not register window class");
+
+ // Create the main window
+ hMainWindow =
+ CreateWindow("RemoteShellServerWindowClass",
+ "Remote Shell Server",
+ WS_OVERLAPPED | WS_CAPTION | WS_SYSMENU | WS_MINIMIZEBOX,
+ 20, 20, 600, 400,
+ NULL, NULL, hInstance, NULL);
+ if (!hMainWindow)
+ ExitOnError("Could not create window");
+
+ ShowWindow(hMainWindow, SW_SHOWMINNOACTIVE);
+ UpdateWindow(hMainWindow);
+
+ // Main message loop
+ while (GetMessage(&msg, NULL, 0, 0)) {
+ TranslateMessage(&msg);
+ DispatchMessage(&msg);
+ }
+
+ ExitProcess(0);
+}
