网络通信socket套接字函数封装

最新推荐文章于 2024-06-26 23:13:50 发布

原创最新推荐文章于 2024-06-26 23:13:50 发布 · 488 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#网络 #socket #网络通信

学习笔记专栏收录该内容

29 篇文章

订阅专栏

本文介绍了在网络通信中，无论是UDP还是TCP，都存在相似的流程。为了提高开发效率，作者分享了如何将这些通用流程封装到socket函数中，以供后续使用。详细封装代码分别在`network.h`和`network.c`文件中。

写网络通信的时候发现这个流程是差不多的不管是udp还是tcp通信都有很多地方是相同的，那么能不能把这些流程封装一下，下次再写的时候就可以节约很多的时间。所以代码如下：
network.h

#ifndef NETWORK_H
#define NETWORK_H

#include <stdio.h>
#include <stdbool.h>
#include <netinet/in.h>

// 网络通信结构
typedef struct NetWork
{
	int sock;					// socket描述符 				
	int type;					// 协议类型
	struct sockaddr_in addr; 	// 通信地址
	socklen_t len; 				// 地址字节数
	bool svr; 					// 服务端标志
}NetWork;

// 打开网络通信对象
NetWork* nw_open(int type,short port,const char* ip,bool svr);

// 等待客户端连接，成功网络对象
NetWork* nw_accept(NetWork* svr_nw);

// 发送数据(阻塞状态)
size_t nw_send(NetWork* nw,const char* buf,size_t len);

// 专门发送字符串
size_t nw_sends(NetWork* nw,const char* buf);

// 接收数据(阻塞状态)
size_t nw_recv(NetWork* nw,char* buf,size_t len);

// 关闭网络通信对象
void nw_close(NetWork* nw);

#endif//NETWORK_H

network.c

#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "network.h"

typedef struct sockaddr* SP;

// 打开网络通信对象
NetWork* nw_open(int type,short port,const char* ip,bool svr)
{
	NetWork* nw = malloc(sizeof(NetWork));
	nw->sock = socket(AF_INET,type,0);
	if(0 > nw->sock)
	{
		free(nw);
		perror("socket");
		return NULL;
	}

	nw->type = type;
	nw->len = sizeof(nw->addr);
	nw->addr.sin_family = AF_INET;
	nw->addr.sin_port = htons(port);
	nw->addr.sin_addr.s_addr = inet_addr(ip);

	if(svr)
	{
		nw->svr = true;

		if(bind(nw->sock,(SP)&nw->addr,nw->len))
		{
			free(nw);
			perror("bind");
			return NULL;
		}

		if(SOCK_STREAM == type)
		{
			if(listen(nw->sock,50))
			{
				free(nw);
				perror("listen");
				return NULL;
			}
		}
	}
	else if(SOCK_STREAM == type)
	{
		if(connect(nw->sock,(SP)&nw->addr,nw->len))
		{
			free(nw);
			perror("listen");
			return NULL;
		}
	}

	return nw;
}

// 等待客户端连接，成功网络对象
NetWork* nw_accept(NetWork* svr_nw)
{
	if(SOCK_STREAM != svr_nw->type || !svr_nw->svr)
	{
		printf("svr_nw error!\n");
		return NULL;
	}

	NetWork* nw = malloc(sizeof(NetWork));
	nw->len = sizeof(nw->addr);

	nw->sock = accept(svr_nw->sock,(SP)&nw->addr,&nw->len);
	if(0 > nw->sock)
	{
		free(nw);
		perror("accept");
		return NULL;
	}

	nw->type = SOCK_STREAM;
	nw->svr = false;

	return nw;
}

// 发送数据(阻塞状态)
size_t nw_send(NetWork* nw,const char* buf,size_t len)
{
	if(SOCK_STREAM == nw->type)
	{
		return send(nw->sock,buf,len,0);
	}
	else
	{
		return sendto(nw->sock,buf,len,0,(SP)&nw->addr,nw->len);
	}
}

// 专门发送字符串
size_t nw_sends(NetWork* nw,const char* buf)
{
	if(SOCK_STREAM == nw->type)
	{
		return send(nw->sock,buf,strlen(buf)+1,0);
	}
	else
	{
		return sendto(nw->sock,buf,strlen(buf)+1,0,(SP)&nw->addr,nw->len);
	}
}

// 接收数据(阻塞状态)
size_t nw_recv(NetWork* nw,char* buf,size_t len)
{
	if(SOCK_STREAM == nw->type)
	{
		return recv(nw->sock,buf,len,0);
	}
	else
	{
		return recvfrom(nw->sock,buf,len,0,(SP)&nw->addr,&nw->len);
	}
}

// 关闭网络通信对象
void nw_close(NetWork* nw)
{
	close(nw->sock);
	free(nw);
}