On Thu, 2010-06-24 at 14:25 +0300, Yaniv Kaul wrote:
> On 6/24/2010 2:03 PM, Michael Goldish wrote:
> > Enable RSS to send/receive files and directory trees (recursively).
>
> Are you slowly developing a competitor to STAF
> (http://staf.sourceforge.net/) ?
> And re. the file transfer protocol, why not use TFTP, instead of
> creating your own protocol?
> Y.
About this, some time ago, after analyzing all the options at the time,
we found out that having our own, small, maintainable application was a
better choice, since it fits fully into our current model of running
tests.
Also, it's very straightforward to get rss.exe installed and working
under pretty much all windows guests out there. It all boils down to a
matter of convenience.
> > See protocol details in rss.cpp.
> >
> > Signed-off-by: Michael Goldish<mgold...@redhat.com>
> > ---
> > 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<mgold...@redhat.com>
> > -// 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<mgold...@redhat.com>
> > +// 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);
> > +}
>
> --
> To unsubscribe from this list: send the line "unsubscribe kvm" in
> the body of a message to majord...@vger.kernel.org
> More majordomo info at http://vger.kernel.org/majordomo-info.html
--
To unsubscribe from this list: send the line "unsubscribe kvm" in
the body of a message to majord...@vger.kernel.org
More majordomo info at http://vger.kernel.org/majordomo-info.html
