version 1.4, 2005/03/05 01:43:58
|
version 1.5, 2005/03/07 10:59:18
|
|
|
#include <stdio.h> | #include <stdio.h> |
#include "winsock2.h" | #include "winsock2.h" |
#include "windows.h" | #include "windows.h" |
|
|
#include <wine/test.h> | #include <wine/test.h> |
#include <winbase.h> | #include <winbase.h> |
| |
/* The #pragma directives offer a way for each compiler to offer machine- |
|
and operating-system-specific features while retaining overall |
|
compatibility with the C and C++ languages. |
|
The next line places a library-search record, ws2_32, in the object file. */ |
|
// #pragma comment ( lib, "ws2_32" ) |
|
|
|
/* constants */ | /* constants */ |
#define ASYNC_EVENT (WM_USER +5) // the message we'll use for our async notification | #define ASYNC_EVENT (WM_USER +5) // the message we'll use for our async notification |
#define PORT_NUM 27015 | #define PORT_NUM 27015 |
|
|
#define MAX_WAIT_TIME 30000 // 30 seconds expressed in milliseconds | #define MAX_WAIT_TIME 30000 // 30 seconds expressed in milliseconds |
#define NUM_THREADS_TO_WAIT 2 | #define NUM_THREADS_TO_WAIT 2 |
| |
|
// macros |
|
#define eq(received, expected, label, type) \ |
|
ok((received) == (expected), "%s: got " type " instead of " type "\n", (label),(received),(expected)) |
|
|
enum // define all the exit codes for a thread | enum // define all the exit codes for a thread |
{ | { |
TH_START_ERROR = 1001, | TH_START_ERROR = 1001, |
|
|
| |
/* static func prototypes */ | /* static func prototypes */ |
static int WSAStartup_w( WORD wVersionRequired, LPWSADATA lpWSAData ); | static int WSAStartup_w( WORD wVersionRequired, LPWSADATA lpWSAData ); |
|
static int WSAAsyncSelect_w( SOCKET s, HWND hWnd, unsigned int wMsg, long lEvent ); |
static SOCKET socket_w( int af, int type, int protocol ); | static SOCKET socket_w( int af, int type, int protocol ); |
|
static int bind_w( SOCKET sock, const struct sockaddr* name, int namelen ); |
|
static int listen_w( SOCKET sock, int backlog ); |
static LRESULT CALLBACK WndProcedure(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam); | static LRESULT CALLBACK WndProcedure(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam); |
static HWND WineCreateWindow( HINSTANCE hInstance); | static HWND WineCreateWindow( HINSTANCE hInstance); |
static void WineThreadCleanUp(); | static void WineThreadCleanUp(); |
static void WineCreateFile(); | static void WineCreateFile(); |
static DWORD WINAPI WineRunServer(); | static DWORD WINAPI WineRunServer(); |
static DWORD WINAPI WineRunClient(); | static DWORD WINAPI WineRunClient(); |
|
static void print_buf( BOOL sent, char* buf, int max ); |
| |
/* struct def */ | /* struct def */ |
typedef struct | typedef struct |
|
|
*/ | */ |
static int WSAStartup_w( WORD wVersionRequired, LPWSADATA lpWSAData ) | static int WSAStartup_w( WORD wVersionRequired, LPWSADATA lpWSAData ) |
{ | { |
|
|
int errCode = 0; | int errCode = 0; |
| |
// WSAVERNOTSUPPORTED test | // WSAVERNOTSUPPORTED test |
WSADATA tmpLpWSAData; | WSADATA tmpLpWSAData; |
errCode = WSAStartup( MAKEWORD( 0, 0 ), &tmpLpWSAData ); | errCode = WSAStartup( MAKEWORD( 0, 0 ), &tmpLpWSAData ); |
if( errCode != WSAVERNOTSUPPORTED ) { |
|
| |
// print out the Error Code to be thorough |
// check the output |
trace( "WSAVERNOTSUPPORTED test failed with error code: %d\n", errCode ); |
eq( errCode, WSAVERNOTSUPPORTED, "WSAStartup", "%d" ); |
errCode = 0; | errCode = 0; |
} |
|
| |
// WSEFAULT test | // WSEFAULT test |
errCode = WSAStartup( wVersionRequired, ( LPWSADATA ) 0 ); | errCode = WSAStartup( wVersionRequired, ( LPWSADATA ) 0 ); |
if( errCode != WSAEFAULT ) { |
eq( errCode, WSAEFAULT, "WSAStartup", "%d" ); |
|
|
// print out the Error Code to be thorough |
|
trace( "WSAFAULT test failed with error code: %d\n", errCode ); |
|
errCode = 0; | errCode = 0; |
} |
|
| |
// lastly regular case - should succeed | // lastly regular case - should succeed |
errCode = WSAStartup( wVersionRequired, lpWSAData ); | errCode = WSAStartup( wVersionRequired, lpWSAData ); |
if( errCode != 0 ) { |
eq( errCode, 0, "WSAStartup", "%d" ); |
|
return errCode; |
| |
// print out the Error Code to be thorough |
|
errCode = WSAGetLastError(); |
|
trace( "WSAStartup() failed with error code: %d\n", errCode ); |
|
} | } |
| |
return errCode; |
|
|
/* |
|
Parameters: |
|
sock |
|
[in] A descriptor that identifies the socket for which event notification is required. |
|
hwnd |
|
[in] A handle that identifies the window that will receive a message when a network event occurs. |
|
msg |
|
[in] A message to be received when a network event occurs. |
|
event |
|
[in] A bitmask that specifies a combination of network events in which the application is interested. |
|
|
|
Return Values: |
|
0 - success |
|
SOCKET_ERROR - Failure |
|
|
|
Error Code | Meaning |
|
WSANOTINITIALISED | A successful WSAStartup call must occur before using this function. |
|
WSAENETDOWN | The network subsystem failed. |
|
WSAEINVAL | One of the specified parameters was invalid, such as the window handle not referring to an |
|
| existing window, or the specified socket is in an invalid state. |
|
WSAEINPROGRESS | A blocking Windows Sockets 1.1 call is in progress, or the service provider is still processing |
|
| a callback function. |
|
WSAENOTSOCK | The descriptor is not a socket. |
|
*/ |
|
static int WSAAsyncSelect_w( SOCKET sock, HWND hwnd, unsigned int msg, long event ) |
|
{ |
|
|
|
BOOL unblocked = FALSE; |
|
|
|
// Test invalid socket |
|
int retVal = WSAAsyncSelect( -999, hwnd, msg, event ); |
|
ok( retVal == SOCKET_ERROR && WSAGetLastError() == WSAENOTSOCK, |
|
"WSAAsyncSelect: Got retVal=%d WSAGetLastError()=%d Expected retVal=%d WSAGetLastError()=%d\n", |
|
retVal, WSAGetLastError(), SOCKET_ERROR, WSAENOTSOCK ); |
|
|
|
if( retVal == 0 ) |
|
unblocked = TRUE; |
|
retVal = 0; |
|
|
|
// Test invalid window handler |
|
retVal = WSAAsyncSelect( sock, 0, msg, event ); |
|
ok( retVal == SOCKET_ERROR && WSAGetLastError() == WSAEINVAL, |
|
"WSAAsyncSelect: Got retVal=%d WSAGetLastError()=%d Expected retVal=%d WSAGetLastError()=%d\n", |
|
retVal, WSAGetLastError(), SOCKET_ERROR, WSAEINVAL ); |
|
|
|
if( retVal == 0 ) |
|
unblocked = TRUE; |
|
retVal = 0; |
|
|
|
// Test invalid event |
|
retVal = WSAAsyncSelect( sock, hwnd, msg, -999 ); |
|
ok( retVal == SOCKET_ERROR && WSAGetLastError() == WSAEINVAL, |
|
"WSAAsyncSelect: Got retVal=%d WSAGetLastError()=%d Expected retVal=%d WSAGetLastError()=%d\n", |
|
retVal, WSAGetLastError(), SOCKET_ERROR, WSAEINVAL ); |
|
|
|
if( retVal == 0 ) |
|
unblocked = TRUE; |
|
retVal = 0; |
|
|
|
// If blocked, make the socket unblocking again |
|
if( unblocked ){ |
|
|
|
trace( "Making socket blocking again\n" ); |
|
if( WSAAsyncSelect( sock, hwnd, msg, 0 ) == SOCKET_ERROR ){ |
|
trace( "WSAAsyncSelect failed to disable events %d\n", WSAGetLastError() ); |
|
return SOCKET_ERROR; |
|
} |
|
|
|
int mode = 0; |
|
if( ioctlsocket( sock, FIONBIO, (u_long FAR*) &mode ) != 0 ){ |
|
trace( "ioctlsocket failed with error code %d\n", WSAGetLastError() ); |
|
return SOCKET_ERROR; |
|
} |
|
|
|
} |
|
|
|
// normal case |
|
retVal = WSAAsyncSelect( sock, hwnd, msg, event ); |
|
eq( retVal, 0, "WSSAsyncSelect", "%d" ); |
|
return retVal; |
|
|
} | } |
| |
/* | /* |
|
Parameters: |
|
sock |
|
[in] Descriptor identifying an unbound socket. |
|
name |
|
[in] Address to assign to the socket from the sockaddr structure. |
|
namelen |
|
[in] Length of the value in the name parameter, in bytes. |
|
|
|
Return Values: |
|
0 - SUCCESS |
|
SOCKET_ERROR - FAILED |
|
|
|
Error code | Meaning |
|
WSANOTINITIALISED | A successful WSAStartup call must occur before using this function. |
|
WSAENETDOWN | The network subsystem has failed. |
|
WSAEACCES | Attempt to connect datagram socket to broadcast address failed because |
|
setsockopt option SO_BROADCAST is not enabled. |
|
WSAEADDRINUSE | A process on the computer is already bound to the same fully-qualified |
|
address and the socket has not been marked to allow address reuse with |
|
SO_REUSEADDR. For example, the IP address and port are bound in the |
|
af_inet case). (See the SO_REUSEADDR socket option under setsockopt.) |
|
WSAEADDRNOTAVAIL | The specified address is not a valid address for this computer. |
|
WSAEFAULT | The name or namelen parameter is not a valid part of the user address space, |
|
the namelen parameter is too small, the name parameter contains an incorrect |
|
address format for the associated address family, or the first two bytes of |
|
the memory block specified by name does not match the address family associated |
|
with the socket descriptor s. |
|
WSAEINPROGRESS | A blocking Windows Sockets 1.1 call is in progress, or the service provider is |
|
still processing a callback function. |
|
WSAEINVAL | The socket is already bound to an address. |
|
WSAENOBUFS | Not enough buffers available, too many connections. |
|
WSAENOTSOCK | The descriptor is not a socket. |
|
*/ |
|
static int bind_w( SOCKET sock, const struct sockaddr* name, int namelen ) |
|
{ |
|
int retVal = 0; |
|
|
|
// invalid socket |
|
retVal = bind( -999, name, namelen ); |
|
ok( retVal == SOCKET_ERROR && WSAGetLastError() == WSAENOTSOCK, |
|
"bind: Got retVal=%d WSAGetLastError()=%d Expected retVal=%d WSAGetLastError()=%d\n", |
|
retVal, WSAGetLastError(), SOCKET_ERROR, WSAENOTSOCK ); |
|
retVal = 0; |
|
|
|
// invalid address |
|
retVal = bind( sock, 0, namelen ); |
|
ok( retVal == SOCKET_ERROR && WSAGetLastError() == WSAEFAULT, |
|
"bind: Got retVal=%d WSAGetLastError()=%d Expected retVal=%d WSAGetLastError()=%d\n", |
|
retVal, WSAGetLastError(), SOCKET_ERROR, WSAEFAULT ); |
|
retVal = 0; |
|
|
|
// invalid arg |
|
retVal = bind( sock, name, -999 ); |
|
ok( retVal == SOCKET_ERROR && WSAGetLastError() == WSAEFAULT, |
|
"bind: Got retVal=%d WSAGetLastError()=%d Expected retVal=%d WSAGetLastError()=%d\n", |
|
retVal, WSAGetLastError(), SOCKET_ERROR, WSAEFAULT ); |
|
retVal = 0; |
|
|
|
// normal case |
|
retVal = bind( sock, name, namelen ); |
|
eq( retVal, 0, "bind", "%d" ); |
|
|
|
return retVal; |
|
} |
|
|
|
|
|
/* |
Wrapper for socket(). | Wrapper for socket(). |
socket() function creates a socket that is bound to a specific service provider. | socket() function creates a socket that is bound to a specific service provider. |
| |
|
|
*/ | */ |
static SOCKET socket_w( int af, int type, int protocol ) | static SOCKET socket_w( int af, int type, int protocol ) |
{ | { |
// Create a socket. |
|
SOCKET m_socket; |
SOCKET sock; |
| |
// WSAEAFNOSUPPORT test | // WSAEAFNOSUPPORT test |
m_socket = socket( -999, SOCK_STREAM, IPPROTO_TCP ); |
sock = socket( -999, SOCK_STREAM, IPPROTO_TCP ); |
if( m_socket != INVALID_SOCKET || WSAGetLastError() != WSAEAFNOSUPPORT ){ |
ok( sock == INVALID_SOCKET && WSAGetLastError() == WSAEAFNOSUPPORT, |
trace( "WSAEAFNOSUPPORT test failed: %d %d\n", m_socket, WSAGetLastError() ); |
"socket: Got sock=%d WSAGetLastError()=%d Expected sock=%d WSAGetLastError()=%d\n", |
} |
sock, WSAGetLastError(), INVALID_SOCKET, WSAEAFNOSUPPORT ); |
| |
// WSAEPROTONOSUPPORT test | // WSAEPROTONOSUPPORT test |
m_socket = socket( AF_INET, SOCK_STREAM, -999 ); |
sock = socket( AF_INET, SOCK_STREAM, -999 ); |
if( m_socket != INVALID_SOCKET || WSAGetLastError() != WSAEPROTONOSUPPORT ){ |
ok( sock == INVALID_SOCKET && WSAGetLastError() == WSAEPROTONOSUPPORT, |
trace( "WSAEPROTONOSUPPORT test failed: %d %d\n", m_socket, WSAGetLastError() ); |
"socket: Got sock=%d WSAGetLastError()=%d Expected sock=%d WSAGetLastError()=%d\n", |
} |
sock, WSAGetLastError(), INVALID_SOCKET, WSAEPROTONOSUPPORT ); |
| |
// WSAESOCKTNOSUPPORT test | // WSAESOCKTNOSUPPORT test |
m_socket = socket( AF_INET, -999, IPPROTO_TCP ); |
sock = socket( AF_INET, -999, IPPROTO_TCP ); |
if( m_socket != INVALID_SOCKET || WSAGetLastError() != WSAESOCKTNOSUPPORT ){ |
ok( sock == INVALID_SOCKET && WSAGetLastError() == WSAESOCKTNOSUPPORT, |
trace( "WSAEPROTONOSUPPORT test failed: %d %d\n", m_socket, WSAGetLastError() ); |
"socket: Got sock=%d WSAGetLastError()=%d Expected sock=%d WSAGetLastError()=%d\n", |
} |
sock, WSAGetLastError(), INVALID_SOCKET, WSAESOCKTNOSUPPORT ); |
|
|
|
// check the normal case |
|
sock = socket( af, type, protocol ); |
|
ok( sock != INVALID_SOCKET, "socket: Got sock=%d\n", sock ); |
|
return sock; |
| |
m_socket = socket( af, type, protocol ); |
|
if( m_socket == INVALID_SOCKET ){ |
|
trace( "Error at socket(): %d\n", WSAGetLastError() ); |
|
} | } |
| |
return m_socket; |
/* |
|
Test routine for listen(), a function that places a socket in a state in which it is listening for an |
|
incoming connection. |
|
Parameters: |
|
sock |
|
[in] Descriptor identifying a bound, unconnected socket. |
|
backlog |
|
[in] Maximum length of the queue of pending connections. If set to SOMAXCONN, the underlying service |
|
provider responsible for socket s will set the backlog to a maximum reasonable value. There is no standard |
|
provision to obtain the actual backlog value. |
|
|
|
Return Values: |
|
0 - success |
|
SOCKET_ERROR - failed |
|
|
|
Error code | Meaning |
|
WSANOTINITIALISED | A successful WSAStartup call must occur before using this function. |
|
WSAENETDOWN | The network subsystem has failed. |
|
WSAEADDRINUSE | The socket's local address is already in use and the socket was not marked to |
|
allow address reuse with SO_REUSEADDR. This error usually occurs during execution |
|
of the bind function, but could be delayed until this function if the bind was to a |
|
partially wildcard address (involving ADDR_ANY) and if a specific address needs to be |
|
committed at the time of this function. |
|
WSAEINPROGRESS | A blocking Windows Sockets 1.1 call is in progress, or the service provider is still |
|
processing a callback function. |
|
WSAEINVAL | The socket has not been bound with bind. |
|
WSAEISCONN | The socket is already connected. |
|
WSAEMFILE | No more socket descriptors are available. |
|
WSAENOBUFS | No buffer space is available. |
|
WSAENOTSOCK | The descriptor is not a socket. |
|
WSAEOPNOTSUPP | The referenced socket is not of a type that supports the listen operation. |
|
|
|
*/ |
|
static int listen_w( SOCKET sock, int backlog ) |
|
{ |
|
int retVal = 0; |
|
|
|
// WSAENOTSOCK test |
|
retVal = listen( -999, backlog ); |
|
ok( retVal == SOCKET_ERROR && WSAGetLastError() == WSAENOTSOCK, |
|
"listen: Got retVal=%d WSAGetLastError()=%d Expected retVal=%d WSAGetLastError()=%d\n", |
|
retVal, WSAGetLastError(), SOCKET_ERROR, WSAENOTSOCK ); |
|
retVal = 0; |
|
|
|
// normal case |
|
retVal = listen( sock, backlog ); |
|
eq( retVal, 0, "listen", "%d" ); |
|
|
|
return retVal; |
} | } |
| |
/* | /* |
|
|
eventCode = WSAGETSELECTEVENT( lParam ); | eventCode = WSAGETSELECTEVENT( lParam ); |
SOCKET socket = (SOCKET) wParam; | SOCKET socket = (SOCKET) wParam; |
SOCKET acceptSock; | SOCKET acceptSock; |
int bytesRecv = 0; |
int bytesRecv = 0, bytesTestCli = 0, bytesTestServ = 0; |
int i = 0, j = 0; |
|
char* buf; | char* buf; |
if( GetCurrentThreadId() == ServerThread.id ) | if( GetCurrentThreadId() == ServerThread.id ) |
buf = ServerThread.packet; | buf = ServerThread.packet; |
|
|
switch( eventCode ) | switch( eventCode ) |
{ | { |
case FD_ACCEPT: | case FD_ACCEPT: |
trace( "%ld accept called\n", GetCurrentThreadId() ); |
|
acceptSock = accept( socket, NULL, NULL ); | acceptSock = accept( socket, NULL, NULL ); |
if( acceptSock == INVALID_SOCKET ){ | if( acceptSock == INVALID_SOCKET ){ |
trace( "accept error %d\n", WSAGetLastError() ); | trace( "accept error %d\n", WSAGetLastError() ); |
closesocket( socket ); | closesocket( socket ); |
ExitThread( TH_ACCEPT_ERROR ); | ExitThread( TH_ACCEPT_ERROR ); |
} | } |
|
|
|
eq( GetCurrentThreadId(), ServerThread.id, "FD_ACCEPT", "%ld" ); |
trace( "%ld: connection accepted\n", GetCurrentThreadId() ); | trace( "%ld: connection accepted\n", GetCurrentThreadId() ); |
break; | break; |
| |
case FD_CONNECT: | case FD_CONNECT: |
| |
// a call to connect() has completed | // a call to connect() has completed |
trace("%ld CONNECTED!\n", GetCurrentThreadId() ); |
eq( GetCurrentThreadId(), ClientThread.id, "FD_CONNECT", "%ld" ); |
|
trace("%ld: connection established\n", GetCurrentThreadId() ); |
break; | break; |
| |
case FD_CLOSE: | case FD_CLOSE: |
| |
// a connection has been closed | // a connection has been closed |
closesocket( socket ); | closesocket( socket ); |
trace( "%ld: Closed socket\n", GetCurrentThreadId()); |
eq( GetCurrentThreadId(), ServerThread.id, "FD_CLOSE", "%ld" ); |
|
trace( "%ld: closed socket and exiting...\n", GetCurrentThreadId()); |
WSACleanup(); | WSACleanup(); |
ExitThread( 0 ); | ExitThread( 0 ); |
break; | break; |
|
|
// WinSock has data waiting to be read | // WinSock has data waiting to be read |
bytesRecv = recv( socket, buf, PACKET_SZ, 0 ); | bytesRecv = recv( socket, buf, PACKET_SZ, 0 ); |
| |
trace( "%ld received %d bytes: ", GetCurrentThreadId(), bytesRecv ); |
ok( bytesRecv == PACKET_SZ || bytesRecv == FILE_SZ%PACKET_SZ, |
while( i < PACKET_SZ && buf[i] != (char)EOF ){ |
"recv: Got bytesRecv=%d Expected bytesRecv=%d or %d\n", |
printf( "%c", buf[i] ); |
bytesRecv, PACKET_SZ, FILE_SZ%PACKET_SZ ); |
i++; |
trace( "Thread %ld received %d bytes\n", GetCurrentThreadId(), bytesRecv ); |
} |
|
printf( "\n" ); |
// print the bytes |
|
print_buf( FALSE, buf, PACKET_SZ ); |
|
|
| |
// Now send it back if you're the server | // Now send it back if you're the server |
if( GetCurrentThreadId() == ServerThread.id ){ | if( GetCurrentThreadId() == ServerThread.id ){ |
| |
if( send( socket, buf, bytesRecv, 0 ) == SOCKET_ERROR ){ |
// Test the bytes sent. Since we are sending small size packets |
|
// the entire packet shoudl be successfully sent. |
// check if the network buffer is full and can send no more |
// For this test, don't send any big file. |
// data. If so then break from the loop |
bytesTestServ = send( socket, buf, bytesRecv, 0 ); |
if( WSAGetLastError() == WSAEWOULDBLOCK ){ |
eq( bytesTestServ, bytesRecv, "send", "%d" ); |
// break from the loop buffer is full |
if( bytesTestServ == SOCKET_ERROR ) |
trace( "woud block send!\n" ); |
|
break; | break; |
} |
|
else{ // another error |
|
trace(" some send error\n" ); |
|
return 0; |
|
} |
|
} |
|
| |
trace( "%ld sent back %d bytes: ", GetCurrentThreadId(), bytesRecv ); |
trace( "Thread %ld sent %d bytes\n", GetCurrentThreadId(), bytesTestServ ); |
j = 0; |
|
while( j < bytesRecv ){ |
// print the bytes |
printf( "%c", buf[j] ); |
print_buf( TRUE, buf, PACKET_SZ ); |
j++; |
|
} |
|
printf( "\n" ); |
|
| |
} | } |
else{ | else{ |
|
|
// Client: close the connection when it received back FILE_SZ | // Client: close the connection when it received back FILE_SZ |
ClientThread.bytesRecvd += bytesRecv; | ClientThread.bytesRecvd += bytesRecv; |
if( ClientThread.bytesRecvd >= FILE_SZ ){ | if( ClientThread.bytesRecvd >= FILE_SZ ){ |
trace( "%ld closing socket\n", GetCurrentThreadId() ); |
|
closesocket( socket ); | closesocket( socket ); |
|
trace( "%ld closed socket and exiting..\n", GetCurrentThreadId() ); |
WSACleanup(); | WSACleanup(); |
ExitThread( 0 ); | ExitThread( 0 ); |
} | } |
|
|
if( GetCurrentThreadId() == ServerThread.id ) | if( GetCurrentThreadId() == ServerThread.id ) |
break; | break; |
| |
// enter an infinite loop |
// loop until the entire file is sent |
BOOL notDone = TRUE; | BOOL notDone = TRUE; |
while( notDone ){ | while( notDone ){ |
| |
|
|
ClientThread.bytesSent += bytesToSend; | ClientThread.bytesSent += bytesToSend; |
| |
// send the packet off to the Server if it is filled | // send the packet off to the Server if it is filled |
if( send( socket, buf, bytesToSend, 0 ) == SOCKET_ERROR ){ |
// Since it will always send a small packet, the entire packet |
|
// should be sent successfully. |
|
bytesTestCli = send( socket, buf, bytesToSend, 0 ); |
| |
// check if the network buffer is full and can send no more |
eq( bytesTestCli, bytesToSend, "send", "%d" ); |
// data. If so then break from the loop |
if( bytesTestCli == SOCKET_ERROR ) |
if( WSAGetLastError() == WSAEWOULDBLOCK ){ |
|
// break from the loop buffer is full |
|
break; | break; |
} |
|
else{ // another error |
|
return 0; |
|
} |
|
} |
|
| |
trace( "%ld sent: ", GetCurrentThreadId() ); |
trace( "Thread %ld sent %d bytes\n", GetCurrentThreadId(), bytesTestCli ); |
j = 0; |
|
while( j < PACKET_SZ && buf[j] != (char)EOF ){ |
// print the bytes |
printf( "%c", buf[j] ); |
print_buf( TRUE, buf, PACKET_SZ ); |
j++; |
|
} |
|
printf( "\n" ); |
|
} | } |
break; | break; |
} | } |
} | } |
|
|
return DefWindowProc( hWnd, msg, wParam, lParam ); | return DefWindowProc( hWnd, msg, wParam, lParam ); |
} | } |
| |
|
/* debugging function to print out the buffer */ |
|
static void print_buf( BOOL sent, char* buf, int max ) |
|
{ |
|
if( sent ) |
|
printf( "%s(%d): %ld sent: ", __FILE__, __LINE__, GetCurrentThreadId() ); |
|
else |
|
printf( "%s(%d): %ld recv: ", __FILE__, __LINE__, GetCurrentThreadId() ); |
|
|
|
int i = 0; |
|
while( i < max && buf[i] != (char)EOF ){ |
|
printf( "%c", buf[i] ); |
|
i++; |
|
} |
|
printf( "\n" ); |
|
|
|
} |
|
|
/* | /* |
Creates a hidden window object. | Creates a hidden window object. |
| |
|
|
} | } |
| |
// Create a socket. | // Create a socket. |
SOCKET sock = socket_w( AF_INET, SOCK_STREAM, IPPROTO_TCP ); if( sock == INVALID_SOCKET ){ |
SOCKET sock = socket_w( AF_INET, SOCK_STREAM, IPPROTO_TCP ); |
|
if( sock == INVALID_SOCKET ){ |
trace( "socket error: %d\n", WSAGetLastError() ); | trace( "socket error: %d\n", WSAGetLastError() ); |
ExitThread( TH_SOCKET_ERROR ); | ExitThread( TH_SOCKET_ERROR ); |
} | } |
|
|
sockAddr.sin_family = AF_INET; | sockAddr.sin_family = AF_INET; |
sockAddr.sin_addr.s_addr = inet_addr( "127.0.0.1" ); | sockAddr.sin_addr.s_addr = inet_addr( "127.0.0.1" ); |
sockAddr.sin_port = htons( PORT_NUM ); | sockAddr.sin_port = htons( PORT_NUM ); |
if( bind( sock, (SOCKADDR*) &sockAddr, sizeof( sockAddr ) ) == SOCKET_ERROR ){ |
if( bind_w( sock, (SOCKADDR*) &sockAddr, sizeof( sockAddr ) ) == SOCKET_ERROR ){ |
trace( "bind failed weith error code %d\n", WSAGetLastError() ); | trace( "bind failed weith error code %d\n", WSAGetLastError() ); |
closesocket( sock ); | closesocket( sock ); |
ExitThread( TH_BIND_ERROR ); | ExitThread( TH_BIND_ERROR ); |
|
|
} | } |
| |
// Listen on the socket. | // Listen on the socket. |
if ( listen( sock, 1 ) == SOCKET_ERROR ){ |
if( listen_w( sock, 1 ) == SOCKET_ERROR ){ |
trace( "listen error with error code %d.\n", WSAGetLastError() ); | trace( "listen error with error code %d.\n", WSAGetLastError() ); |
closesocket( sock ); ExitThread( TH_LISTEN_ERROR ); } |
closesocket( sock ); |
|
ExitThread( TH_LISTEN_ERROR ); |
|
} |
| |
// make the socket asynchronous and notify of read, write, connect and close events | // make the socket asynchronous and notify of read, write, connect and close events |
if( WSAAsyncSelect( sock, hwnd, ASYNC_EVENT, FD_WRITE | FD_ACCEPT |FD_READ | FD_CLOSE ) == SOCKET_ERROR ){ |
if( WSAAsyncSelect_w( sock, hwnd, ASYNC_EVENT, FD_WRITE | FD_ACCEPT |FD_READ | FD_CLOSE ) == SOCKET_ERROR ){ |
| |
trace( "WSAAsyncSelect Failed %d\n", WSAGetLastError() ); | trace( "WSAAsyncSelect Failed %d\n", WSAGetLastError() ); |
closesocket( sock ); | closesocket( sock ); |
|
|
| |
} | } |
| |
trace( "listening and going to loop %ld\n", GetCurrentThreadId() ); |
trace( "Server socket has disabled blocking\n" ); |
| |
BOOL retVal; | BOOL retVal; |
MSG msg; | MSG msg; |
|
|
| |
// make the socket asynchronous and notify of read, write, connect and close events | // make the socket asynchronous and notify of read, write, connect and close events |
// this is the client socket | // this is the client socket |
if( WSAAsyncSelect( sock, hwnd, ASYNC_EVENT, FD_WRITE | FD_CONNECT | FD_READ | FD_CLOSE ) == SOCKET_ERROR ){ |
if( WSAAsyncSelect_w( sock, hwnd, ASYNC_EVENT, FD_WRITE | FD_CONNECT | FD_READ | FD_CLOSE ) == SOCKET_ERROR ){ |
|
|
trace( "WSAAsyncSelect Failed %d\n", WSAGetLastError() ); | trace( "WSAAsyncSelect Failed %d\n", WSAGetLastError() ); |
closesocket( sock ); | closesocket( sock ); |
ExitThread( TH_SELECT_ERROR ); | ExitThread( TH_SELECT_ERROR ); |
|
|
} | } |
| |
|
trace( "Client socket has disabled blocking\n" ); |
|
|
/* connect */ | /* connect */ |
SOCKADDR_IN sockAddr; | SOCKADDR_IN sockAddr; |
sockAddr.sin_family = AF_INET; | sockAddr.sin_family = AF_INET; |
|
|
return; | return; |
} | } |
| |
trace( "%ld: server thread %ld created\n", GetCurrentThreadId(), ServerThread.id ); |
trace( "server thread %ld created. Wait %d seconds\n", ServerThread.id, SERVER_START_TIME ); |
| |
// Wait for three seconds to let server start | // Wait for three seconds to let server start |
DWORD waitRet = WaitForSingleObject( ServerThread.handle, SERVER_START_TIME ); | DWORD waitRet = WaitForSingleObject( ServerThread.handle, SERVER_START_TIME ); |
|
|
| |
} | } |
| |
trace( "%ld: server started successfully now creating client thread\n", GetCurrentThreadId() ); |
if( threadStatus != STILL_ACTIVE ){ |
|
|
|
trace( "Server thread in invalid status - %ld\n", threadStatus ); |
|
WineThreadCleanUp(); |
|
return; |
|
|
|
} |
| |
// Create client thread in suspended state | // Create client thread in suspended state |
ClientThread.handle = CreateThread( NULL, 0, (LPTHREAD_START_ROUTINE)WineRunClient, | ClientThread.handle = CreateThread( NULL, 0, (LPTHREAD_START_ROUTINE)WineRunClient, |
|
|
return; | return; |
} | } |
| |
trace( "%ld: client thread %ld created. Now wait.\n", GetCurrentThreadId(), ClientThread.id ); |
trace( "client thread %ld created.\n", ClientThread.id ); |
| |
// Wait for this server to suspsend itself | // Wait for this server to suspsend itself |
HANDLE threadHandles[NUM_THREADS_TO_WAIT] = { ServerThread.handle, ClientThread.handle }; | HANDLE threadHandles[NUM_THREADS_TO_WAIT] = { ServerThread.handle, ClientThread.handle }; |
|
|
// clean up the threads | // clean up the threads |
WineThreadCleanUp(); | WineThreadCleanUp(); |
| |
trace( "%ld: main thread exiting...\n", GetCurrentThreadId() ); |
trace( "Test completed. Main thread exiting\n" ); |
} | } |