多线程

  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <string.h>
  4. #include <unistd.h>
  5. #include <pthread.h>
  6. #include <netinet/ip.h>
  7. #include <arpa/inet.h>
  8. #include <sys/un.h>
  9. #include <ctype.h>
  10. #include "wrap.h"
  11. #include <signal.h>
  12. #include <sys/wait.h>
  13. #define LISTEN_BACKLOG 50
  14. #define EXIT_FAILURE 1
  15. #define SERVICE_PORT 2222
  17. void* thread_func(void* arg){
  18.     /*处理请求*/
  19.     int cfd=(int)arg;
  20.     char buffer[BUFSIZ];
  21.     int ret=read(cfd,buffer,BUFSIZ);
  22.     write(STDOUT_FILENO,buffer,ret);
  23.     for(int i=0;i<ret;i++){
  24.         buffer[i]=toupper(buffer[i]);
  25.     }
  26.     write(cfd,buffer,ret);
  27.     close(cfd);        //关闭连接套接字
  28.     return NULL;
  29. }
  30. /* 用于其他线程回收线程 */
  31. void* garbage_collection(void* arg){
  32.     pthread_t *pid=(pthread_t*)arg;
  33.     pthread_join(*pid,NULL);
  34.     pthread_detach(pthread_self());
  35.     return NULL;
  36. }
  37. int main(int argc,char *argv[]){
  38.     int lfd,cfd;
  39.     pthread_t tid,gid;
  40.     struct sockaddr_in srv_addr,client_addr;
  41.     char client_ip_addr[20];
  42.     socklen_t client_addr_size;
  45.     /* 创建套接字 */
  46.     lfd=Socket(AF_INET,SOCK_STREAM,0);
  47.     /* 创建服务器地址 */
  48.     memset(&srv_addr,0,sizeof(struct sockaddr_in));
  49.     srv_addr.sin_family=AF_INET;
  50.     srv_addr.sin_port=htons(SERVICE_PORT);
  51.     srv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
  52.     /*端口复用*/
  53.     int optval=1;
  54.     setsockopt(lfd,SOL_SOCKET,SO_REUSEADDR,(void*)&optval,sizeof(optval));
  55.     /* 绑定 */
  56.     Bind(lfd,(struct sockaddr*)&srv_addr,sizeof(srv_addr));
  57.     /* 设置监听上限 */
  58.     Listen(lfd,LISTEN_BACKLOG);
  60.     client_addr_size=sizeof(struct sockaddr_in);
  61.     while(1){
  62.     /* 等待连接    */
  63.         cfd=Accept(lfd,(struct sockaddr*)&client_addr,&client_addr_size);
  64.         //将IP地址转换为点分十进制
  65.         inet_ntop(AF_INET,&client_addr.sin_addr.s_addr,client_ip_addr,sizeof(client_ip_addr));
  66.         printf("new client coming! ip: %s port %d \n",client_ip_addr,ntohs(client_addr.sin_port));
  67.         /*创建线程*/
  68.         int create_ret=pthread_create(&tid,NULL,thread_func,(void*)cfd);
  69.         if(create_ret!=0) perr_exit("join error");
  70.         pthread_create(&gid,NULL,garbage_collection,(void*)&tid);
  71.         continue;
  72.     }
  73. #ifdef DEBUGLOG
  74.     printf("client info: ip: %s port %d \n",
  75.         inet_ntop(AF_INET,&client_addr.sin_addr.s_addr,client_ip_addr,sizeof(client_ip_addr))
  76.         ,client_addr.sin_port);
  77. #endif
  79.     return 0;
  81. }

多进程

  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <string.h>
  4. #include <unistd.h>
  5. #include <pthread.h>
  6. #include <netinet/ip.h>
  7. #include <arpa/inet.h>
  8. #include <sys/un.h>
  9. #include <ctype.h>
  10. #include "wrap.h"
  11. #include <signal.h>
  12. #include <sys/wait.h>
  13. #define LISTEN_BACKLOG 50
  14. #define EXIT_FAILURE 1
  15. #define SERVICE_PORT 2222
  16. void Sig_handle(int signo){
  17.     if(signo==SIGCHLD){
  18.         pid_t ret;
  19.         int state;
  20.         while((ret=waitpid(-1,&state,WNOHANG))!=-1){
  21.             if(ret!=0 && WIFEXITED(state)){
  22.                 printf("child %d finish the task with exit state %d\n",ret,WEXITSTATUS(state));
  23.             }
  24.         }
  25.     }
  26. }
  28. int main(int argc,char *argv[]){
  30.     int lfd,cfd;
  31.     pid_t pid;
  32.     struct sockaddr_in srv_addr,client_addr;
  33.     char client_ip_addr[20];
  34.     socklen_t client_addr_size;
  36.     /* 信号相关 */
  37.     sigset_t set,oldset;
  38.     sigemptyset(&set);
  39.     sigaddset(&set,SIGCHLD);
  40.     /* 创建套接字 */
  41.     lfd=Socket(AF_INET,SOCK_STREAM,0);
  42.     /* 创建服务器地址 */
  43.     memset(&srv_addr,0,sizeof(struct sockaddr_in));
  44.     srv_addr.sin_family=AF_INET;
  45.     srv_addr.sin_port=htons(SERVICE_PORT);
  46.     srv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
  47.     /* 绑定 */
  48.     Bind(lfd,(struct sockaddr*)&srv_addr,sizeof(srv_addr));
  49.     /* 设置监听上限 */
  50.     Listen(lfd,LISTEN_BACKLOG);
  52.     client_addr_size=sizeof(struct sockaddr_in);
  53.     while(1){
  54.     /* 等待连接    */
  55.         cfd=Accept(lfd,(struct sockaddr*)&client_addr,&client_addr_size);
  56.         printf("new client coming! ip: %s port %d \n",
  57.         inet_ntop(AF_INET,&client_addr.sin_addr.s_addr,client_ip_addr,sizeof(client_ip_addr))
  58.         ,client_addr.sin_port);
  60.         // sigprocmask(SIG_BLOCK,&set,&oldset);
  61.         pid=Fork();
  62.         if(pid==0){//child
  63.             close(lfd);     //关闭监听套接字
  64.             /*处理请求*/
  65.             char buffer[BUFSIZ];
  66.             int ret=read(cfd,buffer,BUFSIZ);
  67.             write(STDOUT_FILENO,buffer,ret);
  68.             for(int i=0;i<ret;i++){
  69.                 buffer[i]=toupper(buffer[i]);
  70.             }
  71.             write(cfd,buffer,ret);
  72.             close(cfd);        //关闭连接套接字
  73.         }else if(pid>0){
  74.             close(cfd);
  75.             struct sigaction act;
  76.             /*信号回收子进程*/
  77.             sigemptyset(&act.sa_mask);
  78.             act.sa_flags=0;
  79.             act.sa_handler=Sig_handle;
  80.             int sig_ret=sigaction(SIGCHLD,&act,NULL);
  81.             if(sig_ret!=0) 
  82.                 perr_exit("sigaction");
  83.             // sigprocmask(SIG_SETMASK,&oldset,NULL);
  84.             continue;
  85.         }
  86.     }
  87. #ifdef DEBUGLOG
  88.     printf("client info: ip: %s port %d \n",
  89.         inet_ntop(AF_INET,&client_addr.sin_addr.s_addr,client_ip_addr,sizeof(client_ip_addr))
  90.         ,client_addr.sin_port);
  91. #endif
  93.     return 0;
  95. }