wine4/wine/dlls/wsock32/tests/wsock32_main.c - diff

Return to wsock32_main.c CVS log

Up to [RizwankCVS] / wine4 / wine / dlls / wsock32 / tests

Diff for /wine4/wine/dlls/wsock32/tests/wsock32_main.c between version 1.4 and 1.12

version 1.4, 2005/02/20 22:33:10

version 1.12, 2005/02/22 05:15:01

Line 24

#include <winsock.h>

#include <wtypes.h>

#include <winerror.h>

#include <string.h>

#ifndef STANDALONE

#include "wine/test.h"

Line 42

Line 43

#define todo_wine

#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 {

int sock;

int serverSock;

int type;

int serverType;

int serverPort;

int clientPort[NUM_CLIENTS];

};

Line 56

Line 62

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

void BlockingClient();

void BlockingClient(int *serverPort);

void BlockingServer();

void BlockingServer(int *port);

static void test_ClientServerBlocking_1(void);

static void test_Startup(void);

void BlockingClient()

// StartNetworkApp creates socket sock of type type and returns assigned port number in addr.

void StartNetworkApp(int type, SOCKET *sock, SOCKADDR_IN *addr)

{

SOCKET sock;

SOCKADDR_IN tmpAddr;

SOCKADDR_IN client;

int tmpAddrSize;

sock = socket(AF_INET, SOCK_DGRAM, 0);

*sock = socket(AF_INET, type, 0);

if (sock == INVALID_SOCKET) {

if (*sock == INVALID_SOCKET) {

ok( 0 , "Error in socket()");

WSACleanup();

exit(0);

}

trace("socket() ok\n");

client.sin_family = AF_INET;

addr->sin_family = AF_INET;

client.sin_addr.s_addr = INADDR_ANY;

addr->sin_addr.s_addr = INADDR_ANY;

client.sin_port = htons(0);

addr->sin_port = htons(0);

bind(sock, (const SOCKADDR *) &client, sizeof(client));

if( bind(*sock, (const SOCKADDR *) addr, sizeof(*addr)) ) {

ok( 0 , "Error binding client to socket");

WSACleanup();

exit(0);

}

// get port number

tmpAddrSize = sizeof(tmpAddr);

getsockname(*sock, (SOCKADDR *) &tmpAddr, &tmpAddrSize);

addr->sin_port = tmpAddr.sin_port;

}

//while(1)

void BlockingClient(int *serverPort)

{

printf("Thread 1\n");

SOCKET sock;

SOCKADDR_IN client, server;

HOSTENT *hp;

char* msg="Some message to send to server";

StartNetworkApp(SOCK_STREAM, &sock, &client);

hp = gethostbyname("localhost");

while(*serverPort == 0) ;

// 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");

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

{

trace("blocking server\n");

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;

SOCKADDR_IN server;

int ListenReturn;

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

free(ServerThreads);

free(ServerThreadIDs);

free(ConnectedSockets);

trace("blocking server done\n");

}

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;

//trace("In test_ClientServerBlocking_1\n");

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

}

static void test_Startup(void)

Line 126

Line 242

trace("test 2 of 2:\n");

test_ClientServerBlocking_1();

trace("all tests done\n");

while (clientsDone != NUM_CLIENTS)

;

}

Legend:

Removed from v.1.4
changed lines
	Added in v.1.12

Rizwan Kassim