Linux socket多线程实现回声服务器
程序员文章站
2024-03-23 10:18:52
...
服务端
#include <stdio.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <pthread.h>
#include <errno.h>
#define CONNNUM 10
int fd_connect[CONNNUM] = {0}; // 连接描述符数组
int fd_sock; // 套接字描述符
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER; // 互斥锁
// 服务器初始化
void server_init(short port);
// 服务器运行
void server_run();
// 服务器结束
void server_end();
// 数据处理
void *data_handle(int fd_conn);
// 线程入口函数
void *thread_func(void *arg);
int main(int argc, char const *argv[])
{
if (argc != 2)
{
fprintf(stderr, "Usage: %s <server-port>\n", argv[0]);
return 1;
}
server_init(atoi(argv[1]));
server_run();
server_end();
return 0;
}
// 服务器端初始化
void server_init(short port)
{
// 创建套接字
fd_sock = socket(AF_INET, SOCK_STREAM, 0);
if (fd_sock == -1)
{
perror("socket");
exit(1);
}
// 准备通信地址
struct sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr.s_addr = htonl(INADDR_ANY);
// 绑定
int res = bind(fd_sock, (const struct sockaddr *)&addr, sizeof(addr));
if (res == -1)
{
perror("bind");
exit(1);
}
printf("bind success!\n");
// 监听
res = listen(fd_sock, 10);
if (res == -1)
{
perror("listen");
exit(1);
}
printf("server init success!\n");
}
// 服务器端运行
void server_run()
{
while (1)
{
// 接收连接
int fd_conn = accept(fd_sock, NULL, NULL);
if (fd_conn == -1)
{
perror("accept");
exit(1);
}
printf("new connection!\n");
pthread_t tid; // 创建新线程
int res = pthread_create(&tid, NULL, thread_func, (void *)(long)fd_conn);
if (res != 0)
{
errno = res;
perror("pthread_creat");
exit(1);
}
// 设置线程属性为分离
pthread_detach(tid);
// pid_t pid = fork(); // 创建子进程
// if (pid == -1)
// {
// perror("fork");
// exit(1);
// }
// else if (pid == 0) // 子进程
// {
// // 数据处理
// data_handle(fd_conn);
// exit(0);
// }
}
}
// 数据处理
void *data_handle(int fd_conn)
{
// 修改连接描述符数组
int index;
pthread_mutex_lock(&mutex);
for (int i = 0; i < 10; i++)
{
if (fd_connect[i] == 0)
{
fd_connect[i] = fd_conn;
index = i;
break;
}
}
pthread_mutex_unlock(&mutex);
// 接收数据
char str[512] = {0};
struct sockaddr_in client_addr;
socklen_t addrlen = sizeof(client_addr);
getpeername(fd_conn, (struct sockaddr *)&client_addr, &addrlen);
printf("[%s : %d] online!\n", inet_ntoa(client_addr.sin_addr), client_addr.sin_port);
while (1)
{
int r = read(fd_conn, str, sizeof(str));
if (r == -1)
{
perror("recv cilent");
exit(1);
}
else if (r == 0) // 对端关闭连接
{
break;
}
printf("recv data: %s\n", str);
if (!strcmp(str, "end"))
break;
pthread_mutex_lock(&mutex);
for (int i = 0; i < 10; i++) // 群发消息
{
if (fd_connect[i] != 0)
{
r = write(fd_connect[i], str, r);
if (r == -1)
{
perror("send cilent");
exit(1);
}
}
}
pthread_mutex_unlock(&mutex);
}
printf("[%s : %d] offline!\n", inet_ntoa(client_addr.sin_addr), client_addr.sin_port);
pthread_mutex_lock(&mutex);
fd_connect[index] = 0;
pthread_mutex_unlock(&mutex);
close(fd_conn);
}
void server_end()
{
close(fd_sock);
}
// 线程函数
void *thread_func(void *arg)
{
int fd_conn = (long)arg;
data_handle(fd_conn);
}
客户端
/*
* @Descripttion:
* @version:
* @Author: Chilk
* @Date: 2020-07-29 17:01:23
* @LastEditors: Chilk
* @LastEditTime: 2020-08-01 09:54:54
*/
#include <stdio.h>
#include <assert.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
int main(int argc, char const *argv[])
{
//创建一个IPv4的流式套接字
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd == -1)
{
perror("socket error");
exit(1);
}
struct sockaddr_in info;
info.sin_family = AF_INET;
// info.sin_port = htons(54321);
info.sin_port = htons(atol(argv[2]));
info.sin_addr.s_addr = inet_addr(argv[1]);
// info.sin_addr.s_addr = htonl(INADDR_ANY);
if (connect(sockfd, (const struct sockaddr *)(&info), sizeof(struct sockaddr)) == -1)
{
perror("connect error");
exit(1);
}
printf("连接成功\n");
while (1)
{
char buf[256];
gets(buf);
write(sockfd, buf, strlen(buf)+1);
char rev_buf[256];
read(sockfd, rev_buf, 254);
printf("%s\n", rev_buf);
}
close(sockfd);
return 0;
}
上一篇: maven依赖性传递
下一篇: makefile自动生成依赖性