用Winsock编写服务端和客户端 (C++)

/*****************************************************************************************************************************
*    服务端流程如下：
*	1、声明套接字库，创建套接字（WSAStartup()/socket()）;
*	2、绑定套接字到一个IP地址和一个端口上（bind()）;
*	3、将套接字设置为监听模式等待连接请求；
*	4、请求到来之后，接受连接请求，返回一个新的对应于此次连接的套接字(accept());
*	5、用返回的套接字和客户端进行通信（send()/recv()）;
*	6、返回，等待另一个连接请求
*	7、关闭套接字，关闭加载的套接字库(closesocket()/WSACleanup());
*****************************************************************************************************************************/

#include <stdio.h>
#include <stdlib.h>
#include <winsock2.h> //一般基于套接字的编程引用这个库
void ErrorHandling(char* message);

int main(int argc, char* argv[])
{
	WSADATA wsaData;
	SOCKET hSocket; //SOCKET作为数据类型 声明一个套接字对象
	SOCKADDR_IN servAddr; //用于记录服务端地址

	char message[30]; //发送消息
	int strLen;

	if(argc!=3)
	{
		printf("Usage : %s <IP> <port>\n", argv[0]);
		exit(1);
	}

	if(WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
		ErrorHandling("WSAStartup() error!");  
	
	hSocket=socket(PF_INET, SOCK_STREAM, 0);
	if(hSocket==INVALID_SOCKET)
		ErrorHandling("socket() error");
	
	memset(&servAddr, 0, sizeof(servAddr));
	servAddr.sin_family=AF_INET;
	servAddr.sin_addr.s_addr=inet_addr(argv[1]);
	servAddr.sin_port=htons(atoi(argv[2]));
	
	if(connect(hSocket, (SOCKADDR*)&servAddr, sizeof(servAddr))==SOCKET_ERROR)
		ErrorHandling("connect() error!");
 
	strLen=recv(hSocket, message, sizeof(message)-1, 0);
	if(strLen==-1)
		ErrorHandling("read() error!");
	printf("Message from server: %s \n", message);  

	closesocket(hSocket); //用完后务必记住要关闭套接字
	WSACleanup(); //做一次清理
	return 0;
}

void ErrorHandling(char* message)
{
	fputs(message, stderr);
	fputc('\n', stderr);
	exit(1);
}

/*****************************************************************************************************************************
*    客户端流程如下：
*    1：加载套接字库，创建套接字(WSAStartup()/socket())；
*    2：向服务器发出连接请求(connect())；
*    3：和服务器端进行通信(send()/recv())；
*    4：关闭套接字，关闭加载的套接字库(closesocket()/WSACleanup())
*****************************************************************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <winsock2.h>
void ErrorHandling(char* message);

int main(int argc, char* argv[])
{
    //客户端与服务端有相似之处
	WSADATA wsaData;
	SOCKET hSocket;
	SOCKADDR_IN servAddr;

	char message[30];
	int strLen;

	if(argc!=3)
	{
		printf("Usage : %s <IP> <port>\n", argv[0]);
		exit(1);
	}

	if(WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
		ErrorHandling("WSAStartup() error!");  
	
	hSocket=socket(PF_INET, SOCK_STREAM, 0);
	if(hSocket==INVALID_SOCKET)
		ErrorHandling("socket() error");
	
	memset(&servAddr, 0, sizeof(servAddr));
	servAddr.sin_family=AF_INET;
	servAddr.sin_addr.s_addr=inet_addr(argv[1]); //配置地址
	servAddr.sin_port=htons(atoi(argv[2])); //配置端口号
	
	if(connect(hSocket, (SOCKADDR*)&servAddr, sizeof(servAddr))==SOCKET_ERROR)
		ErrorHandling("connect() error!");
 
	strLen=recv(hSocket, message, sizeof(message)-1, 0);
	if(strLen==-1)
		ErrorHandling("read() error!");
	printf("Message from server: %s \n", message);  

	closesocket(hSocket);
	WSACleanup();
	return 0;
}

void ErrorHandling(char* message)
{
	fputs(message, stderr);
	fputc('\n', stderr);
	exit(1);
}