Clamav杀毒软件源码分析笔记[五]

 

  Clamav杀毒软件源码分析笔记[五]

刺猬@http://blog.csdn.net/littlehedgehog

[Socket通信]

套间字socket常用于服务/客户模型的应用程序之间的通信和数据连接，需要重点关注的是这个所谓的服务端/客户端完全可以是一台计算机的两个应用程序。前面我们已经提到了，Clamd就是这个服务端，也就是说它是整个程序的顶梁柱，所有的关键逻辑处理都是它一人为之，不可不谓之强悍。所有的这些功劳都应该归功于Clamd中的线程处理，不过我们先还不是来关注线程的，先看看socket的建立。

1. int localserver(const struct optstruct *opt, const struct cfgstruct *copt, struct cl_node *root)
2. {
3.     struct sockaddr_un server;
4.     int sockfd, backlog;
5.     struct cfgstruct *cpt;
6.     struct stat foo;
7.     char *estr;
8. 
   
9.     /* 下面是初始化套间字 顺便贴上了配置文件中有关localsocket的信息
10. 
    
11.     # Path to a local socket file the daemon will listen on.
12.     # Default: disabled
13.     LocalSocket /tmp/clamd
14.     */
15.     memset((char *) &server, 0, sizeof(server));
16.     server.sun_family = AF_UNIX;
17.     strncpy(server.sun_path, cfgopt(copt, "LocalSocket")->strarg, sizeof(server.sun_path));
18. 
    
19.     if ((sockfd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
20.     {
21.         estr = strerror(errno);
22.         /*
23.         fprintf(stderr, "ERROR: socket() error: %s/n", estr);
24.         */
25.         logg("!Socket allocation error: %s/n", estr);
26.         exit(1);
27.     }
28. 
    
29.     if (bind(sockfd, (struct sockaddr *) &server, sizeof(struct sockaddr_un)) == -1)
30.     {
31.         if (errno == EADDRINUSE)        //端口号或者socket被其他进程使用
32.         {
33.             if (connect(sockfd, (struct sockaddr *) &server, sizeof(struct sockaddr_un)) >= 0)  //这里连接成功表明socket确实是被别的进程使用了
34.             {
35.                 close(sockfd);
36.                 logg("!Socket file %s is in use by another process./n", server.sun_path);
37.                 exit(1);
38.             }
39.             if (cfgopt(copt, "FixStaleSocket"))         //该socket不是被别的进程占据了，下面我们要做修复工作，配置文件中有FixStaleSocket 意思是要修理失效的socket
40.             {
41.                 logg("^Socket file %s exists. Unclean shutdown? Removing.../n", server.sun_path);
42.                 if (unlink(server.sun_path) == -1)      //从文件系统中删除一个名称。如果名称是文件的最后一个连接，并且没有其它进程将文件打开，名称对应的文件会实际被删除。
43.                 {
44.                     estr = strerror(errno);
45.                     logg("!Socket file %s could not be removed: %s/n", server.sun_path, estr);
46.                     exit(1);
47.                 }
48.                 if (bind(sockfd, (struct sockaddr *) &server, sizeof(struct sockaddr_un)) == -1)        //再次bind如果失败
49.                 {
50.                     estr = strerror(errno);
51.                     logg("!Socket file %s could not be bound: %s (unlink tried)/n", server.sun_path, estr);
52.                     exit(1);
53.                 }
54.             }
55.             else if (stat(server.sun_path, &foo) != -1)
56.             {
57.                 logg("!Socket file %s exists. Either remove it, or configure a different one./n", server.sun_path);
58.                 exit(1);
59.             }
60.         }
61.         else        //这里出现了其他错误
62.         {
63.             estr = strerror(errno);
64.             logg("!Socket file %s could not be bound: %s/n", server.sun_path, estr);
65.             exit(1);
66.         }
67.     }
68. 
    
69.     logg("Unix socket file %s/n", server.sun_path);
70. 
    
71.     if ((cpt = cfgopt(copt, "MaxConnectionQueueLength")))   //Maximum length the queue of pending connections may grow to.  backlog参数为提出连接请求后，在服务器接收该连接请求时的等待队列中的连接数
72.         backlog = cpt->numarg;
73.     else
74.         backlog = CL_DEFAULT_BACKLOG;
75. 
    
76.     logg("Setting connection queue length to %d/n", backlog);
77. 
    
78.     if (listen(sockfd, backlog) == -1)
79.     {
80.         estr = strerror(errno);
81.         /*
82.         fprintf(stderr, "ERROR: listen() error: %s/n", estr);
83.         */
84.         logg("!listen() error: %s/n", estr);
85.         exit(1);
86.     }
87. 
    
88.     acceptloop_th(sockfd, root, copt);
89. 
    
90.     return 0;
91. }

注意这里建立的是本地socket的，区别比较大，但是在我这里唯一的区别就是加粗了，形成了很黑很粗壮的效果...

这里我们主要就是建立个socket的，其实也没啥好说的，主要是整个函数的调用，将改变整个历史进程方向──acceptloop_th(sockfd, root, copt);  因为我们要被一直束缚在里面，貌似永世不得超生了。

acceptloop_th 很长很强大，所以我也就不把源码贴出来了，否则有凑字数的嫌疑，重点关注下下面一块：

1.  /* set up signal handling */
2.     sigfillset(&sigset);
3.     sigdelset(&sigset, SIGINT);
4.     sigdelset(&sigset, SIGTERM);
5.     sigdelset(&sigset, SIGSEGV);
6.     sigdelset(&sigset, SIGHUP);
7.     sigdelset(&sigset, SIGPIPE);
8.     sigdelset(&sigset, SIGUSR2);
9.     sigprocmask(SIG_SETMASK, &sigset, NULL);        //通过这条语句 进程有了新的信号屏蔽集　这里是屏蔽一些我们不关心的信号　
10. 
    
11.     /* SIGINT, SIGTERM, SIGSEGV */
12.     sigact.sa_handler = sighandler_th;
13.     sigemptyset(&sigact.sa_mask);
14.     sigaddset(&sigact.sa_mask, SIGINT);
15.     sigaddset(&sigact.sa_mask, SIGTERM);
16.     sigaddset(&sigact.sa_mask, SIGHUP);
17.     sigaddset(&sigact.sa_mask, SIGPIPE);
18.     sigaddset(&sigact.sa_mask, SIGUSR2);
19.     sigaction(SIGINT, &sigact, NULL);               //下面都是安装新的信号处理程序　
20.     sigaction(SIGTERM, &sigact, NULL);
21.     sigaction(SIGHUP, &sigact, NULL);
22.     sigaction(SIGPIPE, &sigact, NULL);
23.     sigaction(SIGUSR2, &sigact, NULL);
24. 
    

为什么这里需要重新设定这么多的信号处理呢？原因就是我们要屏蔽或者拦截其中有些信号，做出相应的反应，这貌似是废话...???? 比如SIGINT,这信号会导致进程终止,我们需要在终止之前把记录写入log文件,ok,必须拦截信号函数.