| version 1.5, 2005/02/21 19:30:18 | version 1.12, 2005/02/22 05:15:01 | 
|  |  | 
| #include <winsock.h> | #include <winsock.h> | 
| #include <wtypes.h> | #include <wtypes.h> | 
| #include <winerror.h> | #include <winerror.h> | 
|  | #include <string.h> | 
|  |  | 
| #ifndef STANDALONE | #ifndef STANDALONE | 
| #include "wine/test.h" | #include "wine/test.h" | 
|  |  | 
| #define todo_wine | #define todo_wine | 
| #endif | #endif | 
|  |  | 
| #define NUM_CLIENTS 5 | #define NUM_CLIENTS 1 | 
|  | // amount of data to transfer from each client to server | 
|  | #define TRANSFER_SIZE 1000000 | 
|  |  | 
|  | int clientsDone = 0; | 
|  | int sizeofSOCKADDR_IN = sizeof(SOCKADDR_IN); | 
|  |  | 
| struct TestParams { | struct TestParams { | 
| int serverSock; | int serverSock; | 
|  |  | 
| int clientNum; // 1...NUM_CLIENTS | int clientNum; // 1...NUM_CLIENTS | 
| }; | }; | 
|  |  | 
|  | struct ServerThread { | 
|  | HANDLE* ServerThread; | 
|  | DWORD* ServerThreadID; | 
|  | SOCKET Socket; // socket to communicate with client | 
|  | SOCKADDR_IN Client; // client info | 
|  | }; | 
|  |  | 
| static void test_Startup(void); | static void test_Startup(void); | 
| void BlockingClient(); | void BlockingClient(int *serverPort); | 
| void BlockingServer(); | void BlockingServer(int *port); | 
| static void test_ClientServerBlocking_1(void); | static void test_ClientServerBlocking_1(void); | 
| static void test_Startup(void); | static void test_Startup(void); | 
|  |  | 
|  |  | 
| addr->sin_port = tmpAddr.sin_port; | addr->sin_port = tmpAddr.sin_port; | 
| } | } | 
|  |  | 
| void BlockingClient() | void BlockingClient(int *serverPort) | 
| { | { | 
| SOCKET sock; | SOCKET sock; | 
| SOCKADDR_IN client; | SOCKADDR_IN client, server; | 
|  | HOSTENT *hp; | 
|  | char* msg="Some message to send to server"; | 
|  |  | 
| StartNetworkApp(SOCK_STREAM, &sock, &client); | StartNetworkApp(SOCK_STREAM, &sock, &client); | 
|  |  | 
|  | hp = gethostbyname("localhost"); | 
|  |  | 
|  | while(*serverPort == 0) ; | 
|  |  | 
| // network code here | // network code here | 
|  | server.sin_family = AF_INET; | 
|  | server.sin_addr = *(struct in_addr *) hp->h_addr; | 
|  | server.sin_port = *serverPort; | 
|  |  | 
|  | if(connect(sock, (struct sockaddr *)&server, sizeof(struct sockaddr)) < 0){ | 
|  | ok( 0 , "Error connecting client to socket\n"); | 
|  | WSACleanup(); | 
|  | exit(0); | 
|  | } | 
|  |  | 
|  | if((send(sock, msg, strlen(msg), 0))==-1){ | 
|  | ok( 0 , "Error sending message\n"); | 
|  | WSACleanup(); | 
|  | exit(0); | 
|  | } | 
|  |  | 
| trace("blocking client done\n"); | trace("blocking client done\n"); | 
|  | clientsDone++; | 
|  | } | 
|  |  | 
|  | void ProcessConnection(SOCKET *ConnectedSocket) | 
|  | { | 
|  | // this will handle all connections to the server, it's in its own function to allow for multithreading | 
|  | int bClosed; | 
|  | bClosed = close(ConnectedSocket); | 
|  | //ok(bClosed,"Error closing socket"); | 
| } | } | 
|  |  | 
| void BlockingServer() | void BlockingServer(int *port) // listens for incoming connections and accepts up to NUM_CLIENTS connections at once | 
| { | { | 
|  | struct ServerThread *Threads; | 
|  | HANDLE* ServerThreads; // the handles for the threads that process connections | 
|  | DWORD* ServerThreadIDs; // the thread ids for the threads that process connections | 
|  | SOCKET* ConnectedSockets; // the threads created by accept() that get sent to the processing function | 
|  | int ThreadIndex = 0; | 
|  |  | 
| SOCKET sock; | SOCKET sock; | 
| SOCKADDR_IN server; | SOCKADDR_IN server; | 
|  | int ListenReturn; | 
|  |  | 
| StartNetworkApp(SOCK_STREAM, &sock, &server); | StartNetworkApp(SOCK_STREAM, &sock, &server); | 
|  | *port = server.sin_port; | 
|  |  | 
|  | Threads = malloc(sizeof(struct ServerThread) * NUM_CLIENTS); | 
|  | memset(Threads, 0, sizeof(struct ServerThread) * NUM_CLIENTS); | 
|  |  | 
|  | ServerThreads = malloc(sizeof(HANDLE) * NUM_CLIENTS); | 
|  | memset(ServerThreads, 0, sizeof(HANDLE) * NUM_CLIENTS); | 
|  |  | 
|  | ServerThreadIDs = malloc(sizeof(DWORD) * NUM_CLIENTS); | 
|  | memset(ServerThreadIDs, 0, sizeof(DWORD) * NUM_CLIENTS); | 
|  |  | 
|  | ConnectedSockets = malloc(sizeof(DWORD) * NUM_CLIENTS); | 
|  | memset(ConnectedSockets, 0, sizeof(DWORD) * NUM_CLIENTS); | 
|  |  | 
|  | //SOCKADDR_IN RemoteAddress; | 
|  |  | 
|  | ListenReturn = listen(sock, 5); | 
|  | ok(ListenReturn != SOCKET_ERROR, "error listening on socket"); | 
|  |  | 
|  | for (ThreadIndex = 0; ThreadIndex < NUM_CLIENTS; ThreadIndex++) | 
|  | { | 
|  | ConnectedSockets[ThreadIndex] = accept(sock, (SOCKADDR *) &Threads[ThreadIndex].Client, &sizeofSOCKADDR_IN); // this can be modified to include the address of the remote socket | 
|  | ok(ConnectedSockets[ThreadIndex] != INVALID_SOCKET, "error accepting socket"); | 
|  |  | 
|  | ServerThreads[ThreadIndex] = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) &ProcessConnection, &ConnectedSockets[ThreadIndex], 0, &ServerThreadIDs[ThreadIndex]); | 
|  | } | 
|  |  | 
| // network code here | // network code here | 
|  |  | 
|  | free(ServerThreads); | 
|  | free(ServerThreadIDs); | 
|  | free(ConnectedSockets); | 
| trace("blocking server done\n"); | trace("blocking server done\n"); | 
| } | } | 
|  |  | 
| static void test_ClientServerBlocking_1(void) | static void test_ClientServerBlocking_1(void) | 
| { | { | 
| HANDLE Thread1, Thread2; | int ThreadIndex = 0; | 
| DWORD ThreadId1, ThreadId2; | int serverPort = 0; // I think the server port would work better as a #DEFINE rather than a variable that gets passed around everywhere | 
| Thread1 = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) &BlockingClient, NULL, 0, &ThreadId1); | HANDLE ServerThread; | 
| Thread2 = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) &BlockingServer, NULL, 0, &ThreadId2); | DWORD ServerThreadId; | 
|  | DWORD *ClientThreadIds; | 
|  | HANDLE *ClientThreads; | 
|  |  | 
|  | ServerThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) &BlockingClient, &serverPort, 0, &ServerThreadId); | 
|  |  | 
|  | ClientThreads = malloc(sizeof(HANDLE) * NUM_CLIENTS); | 
|  | memset(ClientThreads, 0, sizeof(HANDLE) * NUM_CLIENTS); | 
|  |  | 
|  | ClientThreadIds = malloc(sizeof(DWORD) * NUM_CLIENTS); | 
|  | memset(ClientThreadIds, 0, sizeof(DWORD) * NUM_CLIENTS); | 
|  |  | 
|  | for(ThreadIndex = 0; ThreadIndex < NUM_CLIENTS; ThreadIndex++) { | 
|  | ClientThreads[ThreadIndex] = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) &BlockingServer, &serverPort, 0, &ClientThreadIds[ThreadIndex]); | 
|  | } | 
|  |  | 
| trace("test_ClientServerBlocking_1 done\n"); | trace("test_ClientServerBlocking_1 done\n"); | 
| } | } | 
|  |  | 
|  |  | 
| trace("test 2 of 2:\n"); | trace("test 2 of 2:\n"); | 
| test_ClientServerBlocking_1(); | test_ClientServerBlocking_1(); | 
| trace("all tests done\n"); | trace("all tests done\n"); | 
|  | while (clientsDone != NUM_CLIENTS) | 
|  | ; | 
| } | } |