version 1.10, 2005/02/22 02:28:50
|
version 1.12, 2005/02/22 05:15:01
|
|
|
#define TRANSFER_SIZE 1000000 | #define TRANSFER_SIZE 1000000 |
| |
int clientsDone = 0; | 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(int *serverPort); | void BlockingClient(int *serverPort); |
void BlockingServer(int *port); | void BlockingServer(int *port); |
|
|
SOCKET sock; | SOCKET sock; |
SOCKADDR_IN client, server; | SOCKADDR_IN client, server; |
HOSTENT *hp; | HOSTENT *hp; |
|
char* msg="Some message to send to server"; |
|
|
StartNetworkApp(SOCK_STREAM, &sock, &client); | StartNetworkApp(SOCK_STREAM, &sock, &client); |
| |
hp = gethostbyname("localhost"); | hp = gethostbyname("localhost"); |
|
|
server.sin_addr = *(struct in_addr *) hp->h_addr; | server.sin_addr = *(struct in_addr *) hp->h_addr; |
server.sin_port = *serverPort; | 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++; | clientsDone++; |
} | } |
| |
void ProcessConnection(SOCKET ConnectedSocket) |
void ProcessConnection(SOCKET *ConnectedSocket) |
{ | { |
// this will handle all connections to the server, it's in its own function to allow for multithreading | // this will handle all connections to the server, it's in its own function to allow for multithreading |
int bClosed; | int bClosed; |
|
|
//ok(bClosed,"Error closing socket"); | //ok(bClosed,"Error closing socket"); |
} | } |
| |
void BlockingServer(int *port) |
void BlockingServer(int *port) // listens for incoming connections and accepts up to NUM_CLIENTS connections at once |
{ | { |
HANDLE* ServerThreads; |
struct ServerThread *Threads; |
DWORD* ServerThreadIDs; |
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; | int ThreadIndex = 0; |
| |
SOCKET sock, ConnectedSocket; |
SOCKET sock; |
SOCKADDR_IN server; | SOCKADDR_IN server; |
|
int ListenReturn; |
|
|
StartNetworkApp(SOCK_STREAM, &sock, &server); | StartNetworkApp(SOCK_STREAM, &sock, &server); |
*port = server.sin_port; | *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); | ServerThreads = malloc(sizeof(HANDLE) * NUM_CLIENTS); |
memset(ServerThreads, 0, sizeof(HANDLE) * NUM_CLIENTS); | memset(ServerThreads, 0, sizeof(HANDLE) * NUM_CLIENTS); |
| |
ServerThreadIDs = malloc(sizeof(DWORD) * NUM_CLIENTS); | ServerThreadIDs = malloc(sizeof(DWORD) * NUM_CLIENTS); |
memset(ServerThreadIDs, 0, 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; | //SOCKADDR_IN RemoteAddress; |
| |
// condition for how long we want the test to run goes here |
ListenReturn = listen(sock, 5); |
|
ok(ListenReturn != SOCKET_ERROR, "error listening on socket"); |
|
|
|
for (ThreadIndex = 0; ThreadIndex < NUM_CLIENTS; ThreadIndex++) |
{ | { |
ok(listen(sock, 5) != SOCKET_ERROR, "error listening on socket"); |
ConnectedSockets[ThreadIndex] = accept(sock, (SOCKADDR *) &Threads[ThreadIndex].Client, &sizeofSOCKADDR_IN); // this can be modified to include the address of the remote socket |
//ok(INVALID_SOCKET != (ConnectedSocket = accept(sock)), "error accepting socket"); // 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, &ConnectedSocket, 0, &ServerThreadIDs[ThreadIndex]); |
|
// the line above needs to be cleaned up, it currently copies the connected socket into the called function |
ServerThreads[ThreadIndex] = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) &ProcessConnection, &ConnectedSockets[ThreadIndex], 0, &ServerThreadIDs[ThreadIndex]); |
// and will then overwrite the local variable. I'm thinking maybe it needs an array of sockets to handle this |
|
// and then it can pass the address of the socket. |
|
} | } |
| |
// network code here | // network code here |
| |
free(ServerThreads); | free(ServerThreads); |
free(ServerThreadIDs); | free(ServerThreadIDs); |
|
free(ConnectedSockets); |
trace("blocking server done\n"); | trace("blocking server done\n"); |
} | } |
| |