多进程服务端
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <sys/time.h>
#include <time.h>
#include <pthread.h> // 第三方库,编译需要加上-l pthread参数
#include <semaphore.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <errno.h>
void recyleChild(int arg) { // 父进程回收子进程资源
while (1) {
int ret = waitpid(-1, NULL, WNOHANG); // 设置为非阻塞
if (ret == -1) break;
else if (ret > 0) printf("子进程%d被回收了\n", ret);
}
}
int main() {
// 注册SIGCHLD信号捕捉
struct sigaction act;
act.sa_handler = recyleChild;
act.sa_flags = 0;
sigemptyset(&act.sa_mask);
sigaction(SIGCHLD, &act, NULL);
// 创建socket
int lfd = socket(AF_INET, SOCK_STREAM, 0);
// 绑定服务器地址
struct sockaddr_in Saddr;
Saddr.sin_family = AF_INET;
// inet_pton(AF_INET, "10.16.90.125", Saddr.sin_addr.s_addr);
Saddr.sin_addr.s_addr = INADDR_ANY;
Saddr.sin_port = htons(9999);
bind(lfd, (struct sockaddr*)&Saddr, sizeof Saddr);
// 监听
int ret = listen(lfd, 128);
if (ret == -1) {
perror("listen");
exit(-1);
}
while (1) { // 每收到一个连接, 就fork一个子进程去通信
struct sockaddr_in cliaddr;
int len = sizeof cliaddr;
int cfd = accept(lfd, (struct sockaddr*)&cliaddr, &len); // 正常返回connect socket
if (cfd == -1) {
// 当accept阻塞时, 若有子进程终止, 则主进程会去调用recyleChild(相当于一个软中断), 调用完成后, 这里的accept会变成非阻塞, 但此时并没有客户连接, 因此需要重新accept
if (errno == EINTR) continue; // 当accept被系统调用打断时, error会被设置为EINTR, 并且accept会变成非阻塞
else { // 这里是正常的调用失败分支
perror("accept");
exit(-1);
}
}
pid_t pid = fork();
if (pid == 0) { // 子进程
char cliIp[16]; // 客户端IP
inet_ntop(AF_INET, &cliaddr.sin_addr.s_addr, cliIp, sizeof cliIp);
unsigned short cliPort = ntohs(cliaddr.sin_port); // 客户端port
printf("client ip is : %s, port is %d\n", cliIp, cliPort);
char recvBuf[1024];
while (1) { // 不断与客户端通信
int len = read(cfd, recvBuf, sizeof recvBuf);
if (len == -1) {
perror("read");
exit(-1);
} else if (len > 0) {
printf("recv client data: %s\n", recvBuf);
} else if (len == 0) { // 客户关闭了, 需要break避免执行下面的write
printf("client closed...\n");
break;
}
write(cfd, recvBuf, sizeof recvBuf);
}
close(cfd); // 关闭connect socket
}
}
close(lfd); // 关闭监听socket
return 0;
}
多线程服务端
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <sys/time.h>
#include <time.h>
#include <pthread.h> // 第三方库,编译需要加上-l pthread参数
#include <semaphore.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <errno.h>
struct sockInfo {
int fd; // connect socket
pthread_t tid; // 子线程的tid
struct sockaddr_in addr; // 与之通信的客户端信息(主要是IP,port)
} sockInfos[128]; // 可以支持128个线程同时通信
void* working(void* args) { // 子线程与客户端通信
struct sockInfo* pinfo = (struct sockInfo*)args;
pinfo->tid = pthread_self();
// 与之通信的客户端信息(IP, port)
char cliIp[16];
inet_ntop(AF_INET, &pinfo->addr.sin_addr.s_addr, cliIp, sizeof cliIp);
unsigned short cliPort = ntohs(pinfo->addr.sin_port);
printf("client ip is : %s, port is %d\n", cliIp, cliPort);
char recvBuf[1024];
while (1) { // 与客户端通信
int len = read(pinfo->fd, recvBuf, sizeof recvBuf);
if (len == -1) {
perror("read");
exit(-1);
}
else if (len > 0) {
printf("recv client data: %s\n", recvBuf);
}
else {
printf("client closed...\n");
break;
}
write(pinfo->fd, recvBuf, sizeof recvBuf);
}
close(pinfo->fd); // 通信结束, 关闭connect socket
}
int main() {
// 创建socket
int lfd = socket(AF_INET, SOCK_STREAM, 0);
// 绑定服务器地址
struct sockaddr_in Saddr;
Saddr.sin_family = AF_INET;
// inet_pton(AF_INET, "10.16.90.125", Saddr.sin_addr.s_addr);
Saddr.sin_addr.s_addr = INADDR_ANY;
Saddr.sin_port = htons(9999);
bind(lfd, (struct sockaddr*)&Saddr, sizeof Saddr);
// 监听
int ret = listen(lfd, 128);
if (ret == -1) {
perror("listen");
exit(-1);
}
// 初始化数据
int max = sizeof(sockInfos) / sizeof(sockInfos[0]) ;
for (int i = 0; i < max; ++i) {
bzero(&sockInfos[i], sizeof(sockInfos[i]));
sockInfos[i].fd = -1; // -1表示还没使用
sockInfos[i].tid = -1;
}
while (1) {// 每收到连接就创建一个新的子线程去通信
// 等待连接
struct sockaddr_in cliaddr;
int len = sizeof cliaddr;
int cfd = accept(lfd, (struct sockaddr*)&cliaddr, &len);
// 创建子线程的准备工作
struct sockInfo* pinfo; // 作用: 将主线程中的一些信息传入working, 给子线程用
for (int i = 0; i < max; ++i) { // 找到一个可用的sockInfos[i]
if(sockInfos[i].fd == -1) {
pinfo = &sockInfos[i];
break;
}
if (i == max - 1) { // 如果128个都不能用了, 就等1s, 再去尝试
sleep(1);
i = -1;
}
}
pinfo->fd = cfd;
memcpy(&pinfo->addr, &cliaddr, len);
// 这里pinfo->tid可以等进入子线程执行的代码段, 即working中通过pthread_self()知道
// 创建子线程
pthread_t tid;
pthread_create(&pinfo->tid, NULL, working, pinfo);
// 回收子线程资源
pthread_detach(pinfo->tid);
}
close(lfd); // 关闭监听socket
return 0;
}
客户端
#include <stdio.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
int main() {
// 创建socket
int fd = socket(AF_INET, SOCK_STREAM, 0);
if(fd == -1) {
perror("socket");
exit(-1);
}
// 连接服务器
struct sockaddr_in serveraddr;
serveraddr.sin_family = AF_INET;
inet_pton(AF_INET, "10.16.90.125", &serveraddr.sin_addr.s_addr);
serveraddr.sin_port = htons(9999);
int ret = connect(fd, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
if(ret == -1) {
perror("connect");
exit(-1);
}
// 通信
char recvBuf[1024];
int i = 0;
while(1) {
sprintf(recvBuf, "data : %d\n", i++);
// 给服务器端发送数据
write(fd, recvBuf, strlen(recvBuf)+1);
int len = read(fd, recvBuf, sizeof(recvBuf));
if(len == -1) {
perror("read");
exit(-1);
} else if(len > 0) {
printf("recv server : %s\n", recvBuf);
} else if(len == 0) { // 表示服务器端断开连接
printf("server closed...");
break;
}
sleep(1); // 这里不太懂? https://blog.csdn.net/xc_zhou/article/details/80950753
}
// 关闭连接
close(fd);
return 0;
}