C语言TCP服务端程序
项目介绍
本项目的简易TCP服务端程序能够接受客户端发来的文字消息，并将其显示在终端上
关键技术
接受并发请求的服务器一般有两种实现方式：
- 一请求一线程
- IO多路复用，epoll技术
代码实现
一请求一线程方式
首先，在服务端需要创建一个用于监听的文件描述符（sockfd），这就相当于酒店迎宾的门童，进了酒店之后顾客由其他服务员进行服务，相当于创建通信文件描述符（clientfd）：
int main(int argc, char *argv[])
{
if(argc < 2)
{
printf("Param Error\n");
return -1;
}
int port = atoi(argv[1]);
//创建监听的文件描述符，相当于酒店门口迎宾的人员
//为客户提供服务由其他服务员来做
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in addr;
memset(&addr, 0, sizeof(struct sockaddr_in));//清空结构体
addr.sin_family = AF_INET;
addr.sin_port = htons(port); //主机字节序转换为网络字节序
addr.sin_addr.s_addr = INADDR_ANY;
//设置端口复用，在bind之前调用
int opt = 1;
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
if(bind(sockfd, (struct sockaddr *)&addr, sizeof(struct sockaddr_in)) < 0)
{
perror("bind");
return -2;
}
//利用sockfd进行监听
//第二个参数指定能处理的最大连接请求
if(listen(sockfd, 5) < 0)
{
perror("listen");
return -3;
}
//创建实际通信用的fd
while(1)
{
//这个用户地址结构体变量是用来接受数据的
struct sockaddr_in client_addr;
memset(&client_addr, 0x00, sizeof(struct sockaddr_in));
socklen_t client_len = sizeof(client_addr);
//通过监听文件描述符创建通信文件描述符,默认创建的是阻塞的
//客户端信息会存放在client_addr中
int clientfd = accept(sockfd, (struct sockaddr*)&client_addr, &client_len);
//一请求，一线程
pthread_t thread_t;
//第一个参数是输出参数，下面函数会往该参数里写数据
pthread_create(&thread_t, NULL, client_routine, &clientfd);
}
}
用于接受数据的回调函数
void *client_routine(void *arg)
{
int clientfd = *(int *)arg;
while(1)
{
char buffer[BUFFER_LENGTH];
memset(buffer, 0x00, BUFFER_LENGTH);
//阻塞模式recv,没数据读到就挂起等待
//默认 socket 是阻塞的 解阻塞与非阻塞recv返回值没有区分，都是 <0 出错 ；=0 连接关闭 ；>0 接收到数据大小，
int len = recv(clientfd, buffer, BUFFER_LENGTH, 0);
if(len < 0)
{
close(clientfd);
break;
}
else if(len == 0) //断开连接
{
close(clientfd);
break;
}
else
{
printf("Recv: %s, %d byte(s)\n", buffer, len);
}
}
}
epoll方式实现多个客户端连接的TCP服务端程序
epoll是一种IO多路复用技术，可以简单的理解为epoll帮我们监视我们指定的文件描述符（socket）是否可读或者是否可写。
关键字：epoll_ctl,epoll_create,epoll_wait
epoll的水平触发和边缘触发
这个边缘触发和水平触发都是为了检验IO是否有数据到来：
水平触发（EPOLLLT）
有数据就一直触发
边缘触发（EPOLLET）
数据从无到有才会触发
通常来讲边缘触发在一些场景的时候性能更高（触发次数少）。
代码如下：
int main(int argc, char *argv[])
{
if(argc < 2)
{
printf("Param Error\n");
return -1;
}
int port = atoi(argv[1]);
//创建监听的文件描述符，相当于酒店门口迎宾的人员
//为客户提供服务由其他服务员来做
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in addr;
memset(&addr, 0, sizeof(struct sockaddr_in));//清空结构体
addr.sin_family = AF_INET;
addr.sin_port = htons(port); //主机字节序转换为网络字节序
addr.sin_addr.s_addr = INADDR_ANY;
//设置端口复用，在bind之前调用
int opt = 1;
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
if(bind(sockfd, (struct sockaddr *)&addr, sizeof(struct sockaddr_in)) < 0)
{
perror("bind");
return -2;
}
//利用sockfd进行监听
//第二个参数指定能处理的最大连接请求
if(listen(sockfd, 5) < 0)
{
perror("listen");
return -3;
}
//使用epoll复用IO
//参数只要填一个大于0的数就可以了，这个参数以前是指定最大监听事件的数量
//从Linux 2.6.8开始，max_size参数将被忽略，但必须大于零,
//因为现在可动态调整所需数据结构的大小
int epfd = epoll_create(1); // 1.创建一个监视事件的管理员fd
//用于接受结果的事件数组
struct epoll_event events[EPOLL_SIZE] = {0};
struct epoll_event ev;
//有数据到来socketfd，表示可读了，则会触发EPOLLIN
//对端（客户端）读取了一些数据走，则表示可以往这个socketfd写了，则会触发EPOLLOUT
ev.events = EPOLLIN;
ev.data.fd = sockfd;
//EPOLL_CTL_ADD:在epoll的监视列表中添加一个文件描述符（即参数fd），指定监视的事件类型（参数event）
epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &ev);
while(1)
{
//返回触发事件的数量，触发的事件存放到events中，
//第四参数：表示在没有检测到事件发生时最多等待的时间
// -1表示没有事件到来就阻塞，0表示非阻塞，没事件到来也会返回并跳过
int nready = epoll_wait(epfd, events, EPOLL_SIZE, 5);
if(nready == -1) continue;
int i = 0;
for(i = 0;i < nready;++i)
{
// 产生变化的事件是监听文件描述符
//表示有连接到来
if(events[i].data.fd == sockfd)
{
struct sockaddr_in client_addr;
memset(&client_addr, 0x00, sizeof(struct sockaddr_in));
socklen_t client_len = sizeof(client_addr);
//获得通信文件描述符,并获得客户端的地址
int clientfd = accept(sockfd, (struct sockaddr*)&client_addr, &client_len);
//让epoll的管理员监视这个clientfd
ev.events = EPOLLIN | EPOLLET;
ev.data.fd = clientfd;
epoll_ctl(epfd, EPOLL_CTL_ADD, clientfd, &ev);
}
else //触发的事件对应的文件描述符是通信文件描述符，则接受数据
{
int clientfd = events[i].data.fd;
char buffer[BUFFER_LENGTH] = {0};
int len = recv(clientfd, buffer, BUFFER_LENGTH, 0);
if(len < 0)
{
close(clientfd);
//将client对应事件解除监视
ev.events = EPOLLIN;
ev.data.fd = clientfd;
epoll_ctl(epfd, EPOLL_CTL_DEL, clientfd, &ev);
}
else if(len == 0) // 连接断开
{
close(clientfd);
ev.events = EPOLLIN;
ev.data.fd = clientfd;
epoll_ctl(epfd, EPOLL_CTL_DEL, clientfd, &ev);
}
else
{
printf("Recv: %s, %d byte(s)\n", buffer, len);
}
}
}
}
return 0;
}
完整代码
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <errno.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/epoll.h>
#include <fcntl.h>
#define BUFFER_LENGTH 1024
#define EPOLL_SIZE 1024
void *client_routine(void *arg)
{
int clientfd = *(int *)arg;
while(1)
{
char buffer[BUFFER_LENGTH];
memset(buffer, 0x00, BUFFER_LENGTH);
//阻塞模式recv,没数据读到就挂起等待
int len = recv(clientfd, buffer, BUFFER_LENGTH, 0);
if(len < 0)
{
close(clientfd);
break;
}
else if(len == 0) //断开连接
{
close(clientfd);
break;
}
else
{
printf("Recv: %s, %d byte(s)\n", buffer, len);
}
}
}
int main(int argc, char *argv[])
{
if(argc < 2)
{
printf("Param Error\n");
return -1;
}
int port = atoi(argv[1]);
//创建监听的文件描述符，相当于酒店门口迎宾的人员
//为客户提供服务由其他服务员来做
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in addr;
memset(&addr, 0, sizeof(struct sockaddr_in));//清空结构体
addr.sin_family = AF_INET;
addr.sin_port = htons(port); //主机字节序转换为网络字节序
addr.sin_addr.s_addr = INADDR_ANY;
//设置端口复用，在bind之前调用
int opt = 1;
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
if(bind(sockfd, (struct sockaddr *)&addr, sizeof(struct sockaddr_in)) < 0)
{
perror("bind");
return -2;
}
//利用sockfd进行监听
//第二个参数指定能处理的最大连接请求
if(listen(sockfd, 5) < 0)
{
perror("listen");
return -3;
}
#if 0
//创建实际通信用的fd
while(1)
{
//这个用户地址结构体变量是用来接受数据的
struct sockaddr_in client_addr;
memset(&client_addr, 0x00, sizeof(struct sockaddr_in));
socklen_t client_len = sizeof(client_addr);
//创建通信文件描述符
int clientfd = accept(sockfd, (struct sockaddr*)&client_addr, &client_len);
//一请求，一线程
pthread_t thread_t;
//第一个参数是输出参数，下面函数会往该参数里写数据
pthread_create(&thread_t, NULL, client_routine, &clientfd);
}
#else
//使用epoll复用IO
//参数只要填一个大于0的数就可以了，这个参数以前是指定最大监听事件的数量
//从Linux 2.6.8开始，max_size参数将被忽略，但必须大于零,
//因为现在可动态调整所需数据结构的大小
int epfd = epoll_create(1); // 1.创建一个监视事件的管理员fd
//用于接受结果的事件数组
struct epoll_event events[EPOLL_SIZE] = {0};
struct epoll_event ev;
//有数据到来socketfd，表示可读了，则会触发EPOLLIN
//对端（客户端）读取了一些数据走，则表示可以往这个socketfd写了，则会触发EPOLLOUT
ev.events = EPOLLIN;
ev.data.fd = sockfd;
//EPOLL_CTL_ADD:在epoll的监视列表中添加一个文件描述符（即参数fd），指定监视的事件类型（参数event）
epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &ev);
while(1)
{
//返回触发事件的数量，触发的事件存放到events中，
//第四参数：表示在没有检测到事件发生时最多等待的时间
// -1表示没有事件到来就阻塞，0表示非阻塞，没事件到来也会返回并跳过
int nready = epoll_wait(epfd, events, EPOLL_SIZE, 5);
if(nready == -1) continue;
int i = 0;
for(i = 0;i < nready;++i)
{
// 产生变化的事件是监听文件描述符
//表示有连接到来
if(events[i].data.fd == sockfd)
{
struct sockaddr_in client_addr;
memset(&client_addr, 0x00, sizeof(struct sockaddr_in));
socklen_t client_len = sizeof(client_addr);
//获得通信文件描述符,并获得客户端的地址
int clientfd = accept(sockfd, (struct sockaddr*)&client_addr, &client_len);
//让epoll的管理员监视这个clientfd
ev.events = EPOLLIN | EPOLLET;
ev.data.fd = clientfd;
epoll_ctl(epfd, EPOLL_CTL_ADD, clientfd, &ev);
}
else //触发的事件对应的文件描述符是通信文件描述符，则接受数据
{
int clientfd = events[i].data.fd;
char buffer[BUFFER_LENGTH] = {0};
int len = recv(clientfd, buffer, BUFFER_LENGTH, 0);
if(len < 0)
{
close(clientfd);
//将client对应事件解除监视
ev.events = EPOLLIN;
ev.data.fd = clientfd;
epoll_ctl(epfd, EPOLL_CTL_DEL, clientfd, &ev);
}
else if(len == 0) // 连接断开
{
close(clientfd);
ev.events = EPOLLIN;
ev.data.fd = clientfd;
epoll_ctl(epfd, EPOLL_CTL_DEL, clientfd, &ev);
}
else
{
printf("Recv: %s, %d byte(s)\n", buffer, len);
}
}
}
}
#endif
return 0;
}
编译和启动
使用NetAssist测试
服务端接受数据
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
