在linux平台用C语言编写一个socket服务端示例代码,要求用到select()函数
时间: 2024-02-09 16:09:43 浏览: 199
好的,以下是一个基于 select() 函数的简单的 C 语言代码示例,用于在 Linux 平台上编写一个 socket 服务端:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/select.h>
#define MAX_CLIENTS 10
int main(){
int server_fd, client_fd, max_fd, activity, i, valread;
int client_sockets[MAX_CLIENTS];
struct sockaddr_in server_addr, client_addr;
socklen_t client_len;
char buffer[1024];
// 创建 socket 对象
server_fd = socket(AF_INET, SOCK_STREAM, 0);
if(server_fd == -1){
perror("socket");
exit(EXIT_FAILURE);
}
// 初始化 server 地址结构体
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
server_addr.sin_port = htons(8888);
// 绑定端口号
if(bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1){
perror("bind");
exit(EXIT_FAILURE);
}
// 监听端口号
if(listen(server_fd, 5) == -1){
perror("listen");
exit(EXIT_FAILURE);
}
printf("服务器启动,监听端口号:8888\n");
// 初始化客户端 socket 数组
for(i = 0; i < MAX_CLIENTS; i++){
client_sockets[i] = 0;
}
while(1){
fd_set read_fds;
FD_ZERO(&read_fds);
FD_SET(server_fd, &read_fds);
max_fd = server_fd;
// 添加所有客户端 socket 到 read_fds 集合中
for(i = 0; i < MAX_CLIENTS; i++){
client_fd = client_sockets[i];
if(client_fd > 0){
FD_SET(client_fd, &read_fds);
}
if(client_fd > max_fd){
max_fd = client_fd;
}
}
// 使用 select() 函数等待可读事件
activity = select(max_fd + 1, &read_fds, NULL, NULL, NULL);
if(activity == -1){
perror("select");
exit(EXIT_FAILURE);
}
// 如果是服务器 socket 可读,则有新的客户端连接进来
if(FD_ISSET(server_fd, &read_fds)){
client_len = sizeof(client_addr);
client_fd = accept(server_fd, (struct sockaddr *)&client_addr, &client_len);
if(client_fd == -1){
perror("accept");
continue;
}
printf("客户端 %s:%d 已连接\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
// 将新的客户端 socket 添加到数组中
for(i = 0; i < MAX_CLIENTS; i++){
if(client_sockets[i] == 0){
client_sockets[i] = client_fd;
break;
}
}
}
// 遍历所有客户端 socket,查看是否有数据可读,如果有,就读取数据并发送回去
for(i = 0; i < MAX_CLIENTS; i++){
client_fd = client_sockets[i];
if(FD_ISSET(client_fd, &read_fds)){
memset(buffer, 0, sizeof(buffer));
if((valread = read(client_fd, buffer, sizeof(buffer))) == 0){
// 如果客户端断开了连接,就关闭 socket 并从数组中移除
getpeername(client_fd, (struct sockaddr *)&client_addr, &client_len);
printf("客户端 %s:%d 已断开连接\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
close(client_fd);
client_sockets[i] = 0;
}
else{
// 向客户端发送收到的数据
send(client_fd, buffer, valread, 0);
}
}
}
}
close(server_fd);
return 0;
}
```
这个程序会在本机上的 8888 端口上监听,并且每次接收到一个客户端连接请求,就会向客户端发送一条欢迎信息,并且关闭连接。使用 select() 函数可以同时处理多个客户端连接,提高了服务器的并发能力。
阅读全文