Squid的main函数源码分析

字体大小: 中小 标准 ->行高大小: 标准

要分析一款开源的软件除了要弄清楚一些基本的使用和配置之外,其次最重要的就是对源代码进行分析。对源代码进行分析首先应从其main函数分析入手,了解他在启动时的涉及的哪方面的功能,并勾勒出他的运行流程图,现在squid已经支持windows平台了,并且支持以服务的方式启动。其代码如下:

  1. #if USE_WIN32_SERVICE    /* When USE_WIN32_SERVICE is defined, the main function is placed in win32.c */  
  2. void WINAPI   SquidWinSvcMain(int argc, char **argv)  
  3. {       SquidMain(argc, argv);  
  4. }     
  5. int   SquidMain(int argc, char **argv)  
  6. #else    int  
  7. main(int argc, char **argv)   #endif   
  8. {       int errcount = 0;  
  9.     int loop_delay;   #ifdef _SQUID_WIN32_   
  10.     int WIN32_init_err;   #endif   
  11.    #if HAVE_SBRK   
  12.     /*      * HAVE_SBRK - 这个宏的产生我说明一下,这个是用configure产生的,在autoconf.h中能看到,这个是configure做系统 
  13.     * 函数fun功能性检查的时候,如果该操作系统支持其系统调用,则在autoconf.h中定义宏HAVE_FUN,FUN是其函数名字的大写      * sbrk  - 是从堆中分配空间,本质是移动一个位置,向后移就是分配空间,向前移就是释放空间,返回以页为单位 
  14.     * 的虚拟内存使用情况,squid用它来计算整个进程的内存使用情况。      */  
  15.     sbrk_start = sbrk(0);   #endif   
  16.        debug_log = stderr;  
  17.    #ifdef _SQUID_WIN32_   
  18.     if ((WIN32_init_err = WIN32_Subsystem_Init(&argc, &argv)))       return WIN32_init_err;  
  19. #endif      
  20.     /* call mallopt() before anything else */   #if HAVE_MALLOPT   
  21. #ifdef M_GRAIN        /* Round up all sizes to a multiple of this */  
  22.     mallopt(M_GRAIN, 16);   #endif   
  23. #ifdef M_MXFAST        /* biggest size that is considered a small block */  
  24.     mallopt(M_MXFAST, 256);   #endif   
  25. #ifdef M_NBLKS        /* allocate this many small blocks at once */  
  26.     mallopt(M_NLBLKS, 32);   #endif   
  27. #endif /* HAVE_MALLOPT */      
  28.     /* 初始化本地地址local_addr,默认地址any_addr和广播地址no_addr */       memset(&local_addr, '\0'sizeof(struct in_addr));  
  29.     safe_inet_addr(localhost, &local_addr);       memset(&any_addr, '\0'sizeof(struct in_addr));  
  30.     safe_inet_addr("0.0.0.0", &any_addr);       memset(&no_addr, '\0'sizeof(struct in_addr));  
  31.     safe_inet_addr("255.255.255.255", &no_addr);     
  32.     /* 用当前时间生成随机种子 */       squid_srandom(time(NULL));  
  33.        /* 初始化当前时间 */  
  34.     getCurrentTime();       squid_start = current_time;  
  35.     failure_notify = fatal_dump;    /* 设置失败或者出现重大错误时候退出的回调函数指针 */     
  36. #if defined(USE_WIN32_SERVICE) && defined(_SQUID_WIN32_)        WIN32_svcstatusupdate(SERVICE_START_PENDING, 10000);  
  37. #endif      
  38.     /* 解析命令行参数,如./squid -D -N -d3 之类的 */       mainParseOptions(argc, argv);  
  39.    #if HAVE_SYSLOG && defined(LOG_LOCAL4)   
  40.     /* 打开系统日志,将日志写入系统日志吧 */       openlog(appname, LOG_PID | LOG_NDELAY | LOG_CONS, syslog_facility);  
  41. #endif      
  42. #if defined(USE_WIN32_SERVICE) && defined(_SQUID_WIN32_)        if (opt_install_service) {  
  43.     WIN32_InstallService();       return 0;  
  44.     }       if (opt_remove_service) {  
  45.     WIN32_RemoveService();       return 0;  
  46.     }       if (opt_command_line) {  
  47.     WIN32_SetServiceCommandLine();       return 0;  
  48.     }   #endif   
  49.        /* parse configuration file 
  50.      * note: in "normal" case this used to be called from mainInitialize() */       {  
  51.     /* 配置文件名称初始化,这个可也通过命令行的方式指定,默认是squid.conf */        int parse_err;  
  52.     if (!ConfigFile)           ConfigFile = xstrdup(DefaultConfigFile);  
  53.     assert(!configured_once);   #if USE_LEAKFINDER   
  54.     /*       * 内存泄漏检测功能初始化,通过--enable-leakfinder打开。创建一个哈希表htable,缓存动态内存分配 
  55.     * 情况,cachemgrRegister注册获取内存泄漏跟踪统计数据的action 。      * 但是奇怪的事,我就没看到哪些地方在调用leakAdd leakTouch leakFree之类的接口。 
  56.     */        leakInit();  
  57. #endif        /*  
  58.     * squid提供的内存池功能,这里主要是对预定义的对象内存分配器初始化,以后这些对象就用这些定义好了的内存分配器来      * 分配内存了!cachemgrRegister注册获取内存使用跟踪统计数据的action 。squid实现的内存池功能比apache简单多了! 
  59.     */       memInit();  
  60.        /*  
  61.     * squid提供的专用于回调函数参数对象内存分配器初始化,用这个种分配器分配的内存保存了一些其他的信息用于识别和验证。      * cachemgrRegister注册获取回调参数对象跟踪统计数据的action 。 
  62.     */       cbdataInit();  
  63.        /*  
  64.     * Squid的事件机制的作用是,提供一种触发机制,可以定时地执行某些操作. Squid任何一个功能模块中的代码可以灵活地指定      * “在x秒钟之后,我要做某操作”,而到了x秒钟之后,该操作就可以自动地执行。 
  65.     * 这里主要是对event的对象内存分配器初始化, cachemgrRegister注册获取event对象跟踪统计数据的action 。      */  
  66.     eventInit();        /* eventInit() is required for config parsing */     
  67.     /*      * storeReplSetup()和storeFsSetup(),替换策略模块和存储策略模块选择的初始化,这些模块的选择是通过 
  68.     * ./configure --enable-storeio=afus,diskd,ufs和 ./configure --enable-removal-policies=heap,lru来确定的!      * 替换和存储模块都提供了一个兼容层,这些兼容层只是一个管理功能接口的函数指针对象,初始化就是用用户 
  69.     * configure配置的接口来初始化函数指针管理对象。      */  
  70.     storeFsInit();      /* required for config parsing */     
  71.     /*      * authSchemeSetup(),squid配置的验证机制初始化,这个是squid提供给用户的代理认证,可以通过 
  72.     * ./configure --enable-auth=basic,digest,ntlm来配置,默认支持basic认证方式。并且需要配置一种      * 对应的外部认证程序,通过--enable-basic-auth-helpers=ncsa完成,这里只是配置了认证机制basic 
  73.     * 使用ncsa外部程序来完成认证,这些外部程序都是通过pipe方式来进行通信完成认证的!      */  
  74.     authenticateSchemeInit();   /* required for config parsing */     
  75.     /*      * 这个没什么的,看起来就像是解析配置文件,默认的配置文件是squid.conf,cachemgrRegister注册获取配置数据的action  
  76.     * 处理过程先设置部分默认值,然后解析配置文件覆盖或者初始化一些配置值,然后看哪些没值的就设置默认值,最后配置信息      * 合法化检查,如果配置信息不合理就退出吧!! 
  77.     */       parse_err = parseConfigFile(ConfigFile);  
  78.        if (opt_parse_cfg_only)  
  79.         return parse_err;       }  
  80.     setUmask(Config.umask);     
  81.     /*      * 重复启动运行监测,如果不是像squid发送进程信号的话,就提示不能重复启动运行的错误。squid进程启动的时候会向squid.pid 
  82.     * 写入当前运行进程的pid,checkRunningPid()就是检查的这个文件来确认squid是不是已经启动了!当squid退出的时候会删除      * squid.pid的这个文件。 
  83.     */       if (-1 == opt_send_signal)  
  84.     if (checkRunningPid())           exit(1);  
  85.        /* Make sure the OS allows core dumps if enabled in squid.conf */  
  86.     /*      * 这个只是设置os系统参数,允许squid在crash的时候生成coredump文件。 
  87.     */       enableCoredumps();  
  88.    /* 
  89. * 这个看样子是测试access方面宏,如果定义了该宏,会#include "test_access.c"这个文件,  * 不过哥哥没找到这个文件。我想其他版本有吧,先跳过这个不管算了! 
  90. */   #if TEST_ACCESS   
  91.     comm_init();       comm_select_init();  
  92.     mainInitialize();       test_access();  
  93.     return 0;   #endif   
  94.     /*      * squid进程除了可以作为启动进程外,还可以通过一些命令参数作为管理进程,来完成一些基本的管理工作, 
  95.     * 如opt_send_signal选项,如果该选项设置了就可以向正在运行的squid进程发送信号,我看了就发送了kill信号!      * 如opt_create_swap_dirs选项,如果该选项设置了就可以创建cache目录,根据cache_dir配置指令来完成创建。 
  96.     */       /* send signal to running copy and exit */  
  97.     if (opt_send_signal != -1) {       /* chroot if configured to run inside chroot */  
  98.     if (Config.chroot_dir) {           if (chroot(Config.chroot_dir))  
  99.         fatal("failed to chroot");           no_suid();  
  100.     } else {           leave_suid();  
  101.     }       sendSignal();  
  102.     /* NOTREACHED */       }  
  103.     if (opt_create_swap_dirs) {       /* chroot if configured to run inside chroot */  
  104.     if (Config.chroot_dir && chroot(Config.chroot_dir)) {           fatal("failed to chroot");  
  105.     }       setEffectiveUser();  
  106.     debug(0, 0) ("Creating Swap Directories\n");       storeCreateSwapDirectories();  
  107.     return 0;       }  
  108.        /* 
  109.     * 这里squid以后台服务的方式来启动运行。      */  
  110.     if (!opt_no_daemon)       watch_child(argv);  
  111.     setMaxFD();     
  112.     /* init comm module */       /* 
  113.     * 现在squid网络模块支持devpoll,poll,epoll,kqueue,select和select_win32模式。      * 网络初始化,这里包括client-side和server-side套接字fd,pipe,以及文件fd的管理结构fde,内存分配器和套接字状态管理结构初始化。 
  114.     */       comm_init();  
  115.     comm_select_init();     
  116.     if (opt_no_daemon) {       /* we have to init fdstat here. */  
  117.     if (!opt_stdin_overrides_http_port)           fd_open(0, FD_LOG, "stdin");  
  118.     fd_open(1, FD_LOG, "stdout");       fd_open(2, FD_LOG, "stderr");  
  119.     }   #if defined(USE_WIN32_SERVICE) && defined(_SQUID_WIN32_)   
  120.     WIN32_svcstatusupdate(SERVICE_START_PENDING, 10000);   #endif   
  121.     /*      * squid主要信息初始化,这个会将在mainInitialize()功能做详细的分析。 
  122.     */       mainInitialize();  
  123.    #if defined(USE_WIN32_SERVICE) && defined(_SQUID_WIN32_)   
  124.     WIN32_svcstatusupdate(SERVICE_RUNNING, 0);   #endif   
  125.        /* main loop */  
  126.     /*      * 这个应该就是squid的心跳系统,这里除了处理squid管理发过来的信号之类外,就只处理event事件和读写处理。 
  127.     * 这里应该就只包括套接字和文件的读写吧,整个框架就3点功能:      *   - 处理squid管理发过来的信号,do_reconfigure表示是重新加载配置信息,这个处理起来还不简单啊,要关闭所 
  128.     * 有端口,所有连接,所有文件读写等,然后切换到root用户,解析配置文件,在然后切换回来运行用户,然后启动      * 之前关闭的端口,连接等等信息,太多了! 
  129.     *   - eventRun(),运行event,遍历event管理列表tasks,如果event->when中满足当前运行条件,就执行event->func      * 中的函数功能,运行完后从tasks中删除之,并释放资源吧。灵活地指定“在x秒钟之后,我要做某操作”,这个可以做 
  130.     * 任何事情,不错。。。      *   - 接下来就是网络套接字和文件io的读写!loop_delay = eventNextTime()这个是取得下一个最快将要执行的event 
  131.     * 时间,loop_delay不能超过1000就是1秒,通过这个时间来在一些端口上设置等待超时时间,comm_select(loop_delay)      * 就是在网络套接字和文件FD上等待预先设置好了的操作,等成功返回后就用预先设置好了的函数作回调执行处理得到 
  132.     * 的数据。比如listen FD监听到来的请求,receive FD接收到来的request和response data,预先设置好了数据发出去      * 去的FD,写入disk的FD,等等!!! 
  133.     */       for (;;) {  
  134.     if (do_reconfigure) {           mainReconfigure();  
  135.         do_reconfigure = 0;       } else if (do_rotate) {  
  136.         mainRotate();           do_rotate = 0;  
  137.     } else if (do_shutdown) {           time_t wait = do_shutdown > 0 ? (int) Config.shutdownLifetime : 0;  
  138.         debug(1, 1) ("Preparing for shutdown after %d requests\n",           statCounter.client_http.requests);  
  139.         debug(1, 1) ("Waiting %d seconds for active connections to finish\n",           (int) wait);  
  140.         do_shutdown = 0;           shutting_down = 1;  
  141. #if defined(USE_WIN32_SERVICE) && defined(_SQUID_WIN32_)            WIN32_svcstatusupdate(SERVICE_STOP_PENDING, (wait + 1) * 1000);  
  142. #endif            serverConnectionsClose();  
  143.         eventAdd("SquidShutdown", SquidShutdown, NULL, (double) (wait + 1), 1);       }  
  144.     eventRun();       if ((loop_delay = eventNextTime()) < 0)  
  145.         loop_delay = 0;       if (debug_log_flush() && loop_delay > 1000)  
  146.         loop_delay = 1000;       switch (comm_select(loop_delay)) {  
  147.     case COMM_OK:           errcount = 0;   /* reset if successful */  
  148.         break;       case COMM_ERROR:  
  149.         errcount++;           debug(1, 0) ("Select loop Error. Retry %d\n", errcount);  
  150.         if (errcount == 10)           fatal_dump("Select Loop failed!");  
  151.         break;       case COMM_TIMEOUT:  
  152.         break;       case COMM_SHUTDOWN:  
  153.         SquidShutdown(NULL);           break;  
  154.     default:           fatal_dump("MAIN: Internal error -- this should never happen.");  
  155.         break;       }  
  156.     }       /* NOTREACHED */  
  157.     return 0;   }  

此文章由 http://www.ositren.com 收集整理 ,地址为: http://www.ositren.com/htmls/57776.html