Winsock学习2:用winsock实现socket通讯的client端。

编译环境：VC++6.0
/*
*File:client.cpp
*
*/

#include <winsock2.h>
#include <stdio.h>
#include <stdlib.h>
#define DEFAULT_PORT 5150
#define DEFAULT_BUFFER 4096
#define DEFAULT_MESSAGE "LOGIN"
#define DEFAULT_COUNT 20
int iPort = DEFAULT_PORT;
char szServer[128];
char szMessage[1024];
DWORD dwCount = DEFAULT_COUNT;
BOOL bSendOnly = FALSE;
//Function:usage
//Description:
// Print usage information and exit
void usage()
{
printf("usage:client [-p:x] [-i:ip] [-n:x] [-o] /n/n");
printf(" -p:x Remote port to send to /n");
printf(" -i:str Server's address or host name /n");
printf(" -n:x Number of times to send message/n");
printf(" -o Send messages only;don't recevie/n");
printf("camel 2004-8-29");
ExitProcess(1);
}
// Function:
void ValidateArgs(int argc,char **argv)
{
int i;
for (i = 1; i < argc ;i++)
{
if ((argv[i][0]=='-') || (argv[1][0]=='/') )
{
switch (tolower(argv[i][1]))
{
case 'p':
iPort = atoi(&argv[i][3]);
break;
case 'i':

if(strlen(argv[i]) > 3)
strcpy(szServer,&argv[i][3]);
break;
case 'n':
if(strlen(argv[i])>3)
dwCount = atoi(&argv[i][3]);
case 'o':
bSendOnly = TRUE;
break;
default:
usage();
break;
}
}
}

}

int main(int argc,char ** argv)
{
WSADATA wsd;
SOCKET sClient;
char szBuffer[DEFAULT_BUFFER];
int ret,i;
struct sockaddr_in server;
struct hostent *host = NULL;
printf("======client====/n");
ValidateArgs(argc,argv);
if (WSAStartup(MAKEWORD(2,2),&wsd)!=0)
{
printf("Failed to load Winsock!/n");
return 1;
}
strcpy(szMessage,DEFAULT_MESSAGE);
//
//Create the socket,and attempt to connect to the server
//
sClient = socket(AF_INET,SOCK_STREAM,IPPROTO_IP);
if (sClient == SOCKET_ERROR)
{
printf("socket() failed: %d/n",WSAGetLastError());
return 1;
}
//Select the local interface,and bind to it
server.sin_family = AF_INET;
server.sin_port = htons(iPort);
server.sin_addr.s_addr = inet_addr(szServer);
//
//If the supplied server address wasn't in the form
//"aaa.bbb.ccc.ddd" It's a host name, so try to resolver it
//
if(server.sin_addr.s_addr == INADDR_NONE)
{
host = gethostbyname(szServer);
if(host == NULL)
printf("Unable to resolve server:%s/n",szServer);
return 1;
}
printf("debug: CopyMemory");
//CopyMemory(&server.sin_addr,host->h_addr_list[0],host->h_length);
printf("debug: CopyMemory over");

if(connect(sClient,(struct sockaddr *)&server,sizeof(server)) == SOCKET_ERROR)
{
printf("connect() failed:%d]n",WSAGetLastError());
return 1;
}
//
//Send and receive data
//
for(i = 0; i<dwCount;i++)
{
ret = send(sClient,szMessage,strlen(szMessage),0);
if (ret == 0 )
break;
else if(ret == SOCKET_ERROR)
{
printf("send() failed:%d/n",WSAGetLastError);
break;
}
printf("send %d bytes/n",ret);
if( !bSendOnly)
{
ret = recv(sClient,szBuffer,DEFAULT_BUFFER,0);
if(ret == 0)
break;
else if(ret == SOCKET_ERROR)
{
printf("recv() failed:%d/n",WSAGetLastError());
break;
}
szBuffer[ret] = '/0';
printf("RECV [%d bytes]:'%s'/n",ret,szBuffer);
}
}
closesocket(sClient);
WSACleanup();
return 0;
}