C语言网络编程实战之线上五子棋（一）

C语言网络编程实战之线上五子棋（一）

要想用简洁的代码实现这个项目，首先要对构建网络通信模型的函数进行封装

下面是通信模型 “分配内存、创建socket、准备地址、绑定、监听、连接” 的封装

#include "network.h"

//	分配内存、创建socket、准备地址、绑定、监听、连接
NetWork* init_nw(int type,short port,const char* ip,bool is_svr)
{
	//	分配内存
	NetWork* nw = malloc(sizeof(NetWork));
	//	初始化成员
	nw->type = type;
	//	创建socket
	nw->sock_fd = socket(AF_INET,type,0);
	if(0 > nw->sock_fd)
	{
		perror("socket");
		free(nw);
		return NULL;
	}
	//	准备通信地址
	nw->addr.sin_family = AF_INET;
	nw->addr.sin_port = htons(port);
	nw->addr.sin_addr.s_addr = inet_addr(ip);
	//	计算地址结构体字节数
	nw->addrlen = sizeof(struct sockaddr_in);
	//	是否是服务端
	nw->is_svr = is_svr;

	//	是服务端
	if(is_svr)
	{
		//	绑定
		if(bind(nw->sock_fd,(SP)&nw->addr,nw->addrlen))
		{
			perror("bind");
			free(nw);
			return NULL;
		}
		//	TCP服务端 监听
		if(SOCK_STREAM == type && listen(nw->sock_fd,20))
		{
			perror("listen");
			free(nw);
			return NULL;
		}
	}
	else if(SOCK_STREAM == type)	//	TCP客户端
	{
		//	连接
		if(connect(nw->sock_fd,(SP)&nw->addr,nw->addrlen))
		{
			perror("connect");
			free(nw);
			return NULL;
		}
	}
	return nw;

}

//	等待连接，只有type是SOCK_STREAM并且是服务端才能调用此函数
NetWork* accept_nw(NetWork* nw)
{
	if(SOCK_STREAM != nw->type || !nw->is_svr)
	{
		printf("只有TCP且为服务端的NetWork对象才能调用accept_nw函数\n");
		return NULL;
	}

	//	为新的NetWork分配内存并初始化
	NetWork* cli_nw = malloc(sizeof(NetWork));
	cli_nw->type = nw->type;
	cli_nw->is_svr = true;
	cli_nw->addrlen = nw->addrlen;
	
	cli_nw->sock_fd = 
	accept(nw->sock_fd,(SP)&cli_nw->addr,&cli_nw->addrlen);
	if(0 > cli_nw->sock_fd)
	{
		perror("accept");
		free(cli_nw);
		return NULL;
	}
	return cli_nw;

}

//	具备send和sendto的发送函数
int send_nw(NetWork* nw,const void* buf,size_t len)
{
	if(SOCK_STREAM == nw->type)
		return send(nw->sock_fd,buf,len,0);
	else
		return sendto(nw->sock_fd,buf,len,0,(SP)&nw->addr,nw->addrlen);
}

//	具备recv和recvfrom的接收函数
int recv_nw(NetWork* nw,void* buf,size_t len)
{
	if(SOCK_STREAM == nw->type)
		return recv(nw->sock_fd,buf,len,0);
	else
		return recvfrom(nw->sock_fd,buf,len,0,(SP)&nw->addr,&nw->addrlen);
}

//	关闭socket并释放内存
void close_nw(NetWork* nw)
{
	close(nw->sock_fd);
	free(nw);
}