linux(一)------多程序併發伺服器實現(fork)
阿新 • • 發佈:2019-01-04
本文實現一個多程序併發伺服器
實現思路:
利用fork()函式實現每有一個client連線時就會建立一個子程序程序與client通訊,父程序負責回收子程序的PCB
直接上程式碼!
#include <stdio.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>
#include <sys/wait.h>
#include <ctype.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#define SERVER_PORT 8888
#define MAXLINE 8192
//程序回收函式
void do_sigchlid(int num)
{
pid_t pid;
while((pid = waitpid(-1, NULL, WNOHANG)) > 0);//回收任何子程序
{
printf("child died, pid = %d\n", pid);
}
}
int main()
{
int listenfd, connfd;
pid_t pid;
int len;
char buf[MAXLINE];
char str[256];
int i;
//建立監聽描述符
listenfd = socket(AF_INET, SOCK_STREAM, 0);
if(listenfd == -1)
{
perror("socket error");
exit(1);
}
//she值埠複用
int opt = 1;
setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof (opt));
//繫結ip和埠
struct sockaddr_in serveraddr;
bzero(&serveraddr, sizeof(serveraddr));
serveraddr.sin_port = htons(SERVER_PORT);
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
int ret = bind(listenfd, (struct sockaddr*)&serveraddr, sizeof(serveraddr));
if(ret == -1)
{
perror("bind error");
exit(1);
}
//設定監聽
ret = listen(listenfd, 20);
if(ret == -1)
{
perror("listen error");
exit(1);
}
printf("Accepting connections ...\n");
struct sockaddr_in clientaddr;
socklen_t clientaddr_len = sizeof(clientaddr);
//通訊
while(1)
{
//等待接收客戶端的連線
connfd = accept(listenfd, (struct sockaddr*)&clientaddr, &clientaddr_len);
printf("connected client ip:%s, port: %d\n",
inet_ntop(AF_INET, &clientaddr.sin_addr, str, sizeof(str)), ntohs(clientaddr.sin_port));
//有一個新的連線就建立一個程序
pid = fork();
if(pid == 0)
{
//子程序
//關閉監聽檔案描述符
close(listenfd);
while(1)//與客戶端通訊,機型資料處理
{
//讀資料
len = read(connfd, buf, MAXLINE);
if(len == 0)
{
printf("客戶端斷開了連線...\n");
break;
}
for(i = 0; i < len; ++i)
{
buf[i] = toupper(buf[i]);
}
//寫回客戶端
write(STDOUT_FILENO, buf, len);//把buf寫到標準輸出中去
write(connfd, buf, len);//把buf寫回客戶端
}
close(connfd);
return 0;
}
else if(pid > 0)
{
//父程序,負責對子程序pcb資源的回收
//使用訊號來回收,子程序結束會給父程序傳送SIGCHLD訊號
//預設父程序會遮蔽此訊號,若父程序想回收子程序資源,
//可以捕捉此訊號
//訊號捕捉
//註冊捕捉函式
struct sigaction act;
act.sa_flags = 0;
act.sa_handler = do_sigchlid;//回撥函式,通過sa_handler指標呼叫
sigemptyset(&act.sa_mask);//將訊號集初始化為空
sigaction(SIGCHLD, &act, NULL);
}
}
return 0;
}