|
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) |
| |
; |
| } | } |