开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 代理服务器 > 查看源码
main.c
资源名称：squid-2.2.devel3-src.tar.gz　[点击查看]
上传用户：liugui
上传日期：2007-01-04
资源大小：822k
文件大小：20k
源码类别：

代理服务器

开发平台：

Unix_Linux

main.c：源码内容
  2. /*
  3.  * $Id: main.c,v 1.290 1999/01/24 04:03:52 wessels Exp $
  4.  *
  5.  * DEBUG: section 1     Startup and Main Loop
  6.  * AUTHOR: Harvest Derived
  7.  *
  8.  * SQUID Internet Object Cache  http://squid.nlanr.net/Squid/
  9.  * ----------------------------------------------------------
  10.  *
  11.  *  Squid is the result of efforts by numerous individuals from the
  12.  *  Internet community.  Development is led by Duane Wessels of the
  13.  *  National Laboratory for Applied Network Research and funded by the
  14.  *  National Science Foundation.  Squid is Copyrighted (C) 1998 by
  15.  *  Duane Wessels and the University of California San Diego.  Please
  16.  *  see the COPYRIGHT file for full details.  Squid incorporates
  17.  *  software developed and/or copyrighted by other sources.  Please see
  18.  *  the CREDITS file for full details.
  19.  *
  20.  *  This program is free software; you can redistribute it and/or modify
  21.  *  it under the terms of the GNU General Public License as published by
  22.  *  the Free Software Foundation; either version 2 of the License, or
  23.  *  (at your option) any later version.
  24.  *  
  25.  *  This program is distributed in the hope that it will be useful,
  26.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  27.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  28.  *  GNU General Public License for more details.
  29.  *  
  30.  *  You should have received a copy of the GNU General Public License
  31.  *  along with this program; if not, write to the Free Software
  32.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
  33.  *
  34.  */
  36. #include "squid.h"
  38. /* for error reporting from xmalloc and friends */
  39. extern void (*failure_notify) (const char *);
  41. static int opt_send_signal = -1;
  42. static int opt_no_daemon = 0;
  43. static int opt_parse_cfg_only = 0;
  44. static int httpPortNumOverride = 1;
  45. static int icpPortNumOverride = 1; /* Want to detect "-u 0" */
  46. static int configured_once = 0;
  47. #if MALLOC_DBG
  48. static int malloc_debug_level = 0;
  49. #endif
  50. static volatile int do_reconfigure = 0;
  51. static volatile int do_rotate = 0;
  52. static volatile int do_shutdown = 0;
  54. static void mainRotate(void);
  55. static void mainReconfigure(void);
  56. static SIGHDLR rotate_logs;
  57. static SIGHDLR reconfigure;
  58. #if ALARM_UPDATES_TIME
  59. static SIGHDLR time_tick;
  60. #endif
  61. static void mainInitialize(void);
  62. static void usage(void);
  63. static void mainParseOptions(int, char **);
  64. static void sendSignal(void);
  65. static void serverConnectionsOpen(void);
  66. static void watch_child(char **);
  67. static void setEffectiveUser(void);
  68. #if MEM_GEN_TRACE
  69. extern void log_trace_done();
  70. extern void log_trace_init(char *);
  71. #endif
  72. static EVH SquidShutdown;
  74. static void
  75. usage(void)
  76. {
  77.     fprintf(stderr,
  78. "Usage: %s [-dhsvzCDFNRVYX] [-f config-file] [-[au] port] [-k signal]n"
  79. "       -a port   Specify HTTP port number (default: %d).n"
  80. "       -d level  Write debugging to stderr also.n"
  81. "       -f file   Use given config-file instead ofn"
  82. "                 %sn"
  83. "       -h        Print help message.n"
  84. "       -k reconfigure|rotate|shutdown|interrupt|kill|debug|check|parsen"
  85. "                 Parse configuration file, then send signal to n"
  86. "                 running copy (except -k parse) and exit.n"
  87. "       -s        Enable logging to syslog.n"
  88. "       -u port   Specify ICP port number (default: %d), disable with 0.n"
  89. "       -v        Print version.n"
  90. "       -z        Create swap directoriesn"
  91. "       -C        Do not catch fatal signals.n"
  92. "       -D        Disable initial DNS tests.n"
  93. "       -F        Foreground fast store rebuild.n"
  94. "       -N        No daemon mode.n"
  95. "       -R        Do not set REUSEADDR on port.n"
  96. "       -V        Virtual host httpd-accelerator.n"
  97. "       -X        Force full debugging.n"
  98. "       -Y        Only return UDP_HIT or UDP_MISS_NOFETCH during fast reload.n",
  99. appname, CACHE_HTTP_PORT, DefaultConfigFile, CACHE_ICP_PORT);
  100.     exit(1);
  101. }
  103. static void
  104. mainParseOptions(int argc, char *argv[])
  105. {
  106.     extern char *optarg;
  107.     int c;
  109.     while ((c = getopt(argc, argv, "CDFNRSVYXa:d:f:hk:m::su:vz?")) != -1) {
  110. switch (c) {
  111. case 'C':
  112.     opt_catch_signals = 0;
  113.     break;
  114. case 'D':
  115.     opt_dns_tests = 0;
  116.     break;
  117. case 'F':
  118.     opt_foreground_rebuild = 1;
  119.     break;
  120. case 'N':
  121.     opt_no_daemon = 1;
  122.     break;
  123. case 'R':
  124.     opt_reuseaddr = 0;
  125.     break;
  126. case 'S':
  127.     opt_store_doublecheck = 1;
  128.     break;
  129. case 'V':
  130.     vhost_mode = 1;
  131.     break;
  132. case 'X':
  133.     /* force full debugging */
  134.     sigusr2_handle(SIGUSR2);
  135.     break;
  136. case 'Y':
  137.     opt_reload_hit_only = 1;
  138.     break;
  139. case 'a':
  140.     httpPortNumOverride = atoi(optarg);
  141.     break;
  142. case 'd':
  143.     opt_debug_stderr = atoi(optarg);
  144.     break;
  145. case 'f':
  146.     xfree(ConfigFile);
  147.     ConfigFile = xstrdup(optarg);
  148.     break;
  149. case 'h':
  150.     usage();
  151.     break;
  152. case 'k':
  153.     if ((int) strlen(optarg) < 1)
  154. usage();
  155.     if (!strncmp(optarg, "reconfigure", strlen(optarg)))
  156. opt_send_signal = SIGHUP;
  157.     else if (!strncmp(optarg, "rotate", strlen(optarg)))
  158. #ifdef _SQUID_LINUX_THREADS_
  159. opt_send_signal = SIGQUIT;
  160. #else
  161. opt_send_signal = SIGUSR1;
  162. #endif
  163.     else if (!strncmp(optarg, "debug", strlen(optarg)))
  164. #ifdef _SQUID_LINUX_THREADS_
  165. opt_send_signal = SIGTRAP;
  166. #else
  167. opt_send_signal = SIGUSR2;
  168. #endif
  169.     else if (!strncmp(optarg, "shutdown", strlen(optarg)))
  170. opt_send_signal = SIGTERM;
  171.     else if (!strncmp(optarg, "interrupt", strlen(optarg)))
  172. opt_send_signal = SIGINT;
  173.     else if (!strncmp(optarg, "kill", strlen(optarg)))
  174. opt_send_signal = SIGKILL;
  175.     else if (!strncmp(optarg, "check", strlen(optarg)))
  176. opt_send_signal = 0; /* SIGNULL */
  177.     else if (!strncmp(optarg, "parse", strlen(optarg)))
  178. opt_parse_cfg_only = 1; /* parse cfg file only */
  179.     else
  180. usage();
  181.     break;
  182. case 'm':
  183.     if (optarg) {
  184. #if MALLOC_DBG
  185. malloc_debug_level = atoi(optarg);
  186. /* NOTREACHED */
  187. break;
  188. #else
  189. fatal("Need to add -DMALLOC_DBG when compiling to use -mX option");
  190. /* NOTREACHED */
  191. #endif
  192.     } else {
  193. #if XMALLOC_TRACE
  194. xmalloc_trace = !xmalloc_trace;
  195. #else
  196. fatal("Need to configure --enable-xmalloc-debug-trace to use -m option");
  197. #endif
  198.     }
  199. case 's':
  200.     opt_syslog_enable = 1;
  201.     break;
  202. case 'u':
  203.     icpPortNumOverride = atoi(optarg);
  204.     if (icpPortNumOverride < 0)
  205. icpPortNumOverride = 0;
  206.     break;
  207. case 'v':
  208.     printf("Squid Cache: Version %sn", version_string);
  209.     exit(0);
  210.     /* NOTREACHED */
  211. case 'z':
  212.     opt_create_swap_dirs = 1;
  213.     break;
  214. case '?':
  215. default:
  216.     usage();
  217.     break;
  218. }
  219.     }
  220. }
  222. /* ARGSUSED */
  223. static void
  224. rotate_logs(int sig)
  225. {
  226.     do_rotate = 1;
  227. #if !HAVE_SIGACTION
  228.     signal(sig, rotate_logs);
  229. #endif
  230. }
  232. #if ALARM_UPDATES_TIME
  233. static void
  234. time_tick(int sig)
  235. {
  236.     getCurrentTime();
  237.     alarm(1);
  238. #if !HAVE_SIGACTION
  239.     signal(sig, time_tick);
  240. #endif
  241. }
  243. #endif
  245. /* ARGSUSED */
  246. static void
  247. reconfigure(int sig)
  248. {
  249.     do_reconfigure = 1;
  250. #if !HAVE_SIGACTION
  251.     signal(sig, reconfigure);
  252. #endif
  253. }
  255. void
  256. shut_down(int sig)
  257. {
  258.     do_shutdown = sig == SIGINT ? -1 : 1;
  259. #ifdef KILL_PARENT_OPT
  260.     if (getppid() > 1) {
  261. debug(1, 1) ("Killing RunCache, pid %dn", getppid());
  262. kill(getppid(), sig);
  263.     }
  264. #endif
  265. #if SA_RESETHAND == 0
  266.     signal(SIGTERM, SIG_DFL);
  267.     signal(SIGINT, SIG_DFL);
  268. #endif
  269. }
  271. static void
  272. serverConnectionsOpen(void)
  273. {
  274.     clientHttpConnectionsOpen();
  275.     icpConnectionsOpen();
  276. #if USE_HTCP
  277.     htcpInit();
  278. #endif
  279. #ifdef SQUID_SNMP
  280.     snmpConnectionOpen();
  281. #endif
  282.     clientdbInit();
  283.     icmpOpen();
  284.     netdbInit();
  285.     asnInit();
  286.     peerSelectInit();
  287. #if USE_CARP
  288.     carpInit();
  289. #endif
  290. }
  292. void
  293. serverConnectionsClose(void)
  294. {
  295.     assert(shutting_down || reconfiguring);
  296.     clientHttpConnectionsClose();
  297.     icpConnectionShutdown();
  298. #if USE_HTCP
  299.     htcpSocketShutdown();
  300. #endif
  301.     icmpClose();
  302. #ifdef SQUID_SNMP
  303.     snmpConnectionShutdown();
  304. #endif
  305.     asnFreeMemory();
  306. }
  308. static void
  309. mainReconfigure(void)
  310. {
  311.     debug(1, 1) ("Restarting Squid Cache (version %s)...n", version_string);
  312.     reconfiguring = 1;
  313.     /* Already called serverConnectionsClose and ipcacheShutdownServers() */
  314.     serverConnectionsClose();
  315.     icpConnectionClose();
  316. #if USE_HTCP
  317.     htcpSocketClose();
  318. #endif
  319. #ifdef SQUID_SNMP
  320.     snmpConnectionClose();
  321. #endif
  322.     dnsShutdown();
  323.     redirectShutdown();
  324.     authenticateShutdown();
  325.     storeDirCloseSwapLogs();
  326.     errorClean();
  327.     parseConfigFile(ConfigFile);
  328.     _db_init(Config.Log.log, Config.debugOptions);
  329.     ipcache_restart(); /* clear stuck entries */
  330.     fqdncache_restart(); /* sigh, fqdncache too */
  331.     errorInitialize(); /* reload error pages */
  332.     dnsInit();
  333.     redirectInit();
  334.     authenticateInit();
  335.     serverConnectionsOpen();
  336.     if (theOutIcpConnection >= 0) {
  337. if (!Config2.Accel.on || Config.onoff.accel_with_proxy)
  338.     neighbors_open(theOutIcpConnection);
  339. else
  340.     debug(1, 1) ("ICP port disabled in httpd_accelerator moden");
  341.     }
  342.     storeDirOpenSwapLogs();
  343.     writePidFile(); /* write PID file */
  344.     debug(1, 1) ("Ready to serve requests.n");
  345.     reconfiguring = 0;
  346. }
  348. static void
  349. mainRotate(void)
  350. {
  351.     icmpClose();
  352.     _db_rotate_log(); /* cache.log */
  353.     storeDirWriteCleanLogs(1);
  354.     storeLogRotate(); /* store.log */
  355.     accessLogRotate(); /* access.log */
  356.     useragentRotateLog(); /* useragent.log */
  357.     icmpOpen();
  358. }
  360. static void
  361. setEffectiveUser(void)
  362. {
  363.     leave_suid(); /* Run as non privilegied user */
  364. #ifdef _SQUID_OS2_
  365.     return;
  366. #endif
  367.     if (geteuid() == 0) {
  368. debug(0, 0) ("Squid is not safe to run as root!  If you mustn");
  369. debug(0, 0) ("start Squid as root, then you must configuren");
  370. debug(0, 0) ("it to run as a non-priveledged user with then");
  371. debug(0, 0) ("'cache_effective_user' option in the config file.n");
  372. fatal("Don't run Squid as root, set 'cache_effective_user'!");
  373.     }
  374. }
  376. static void
  377. mainInitialize(void)
  378. {
  379.     if (opt_catch_signals) {
  380. squid_signal(SIGSEGV, death, SA_NODEFER | SA_RESETHAND);
  381. squid_signal(SIGBUS, death, SA_NODEFER | SA_RESETHAND);
  382.     }
  383.     squid_signal(SIGPIPE, SIG_IGN, SA_RESTART);
  384.     squid_signal(SIGCHLD, sig_child, SA_NODEFER | SA_RESTART);
  386.     setEffectiveUser();
  387.     assert(Config.Port.http);
  388.     if (httpPortNumOverride != 1)
  389. Config.Port.http->i = (u_short) httpPortNumOverride;
  390.     if (icpPortNumOverride != 1)
  391. Config.Port.icp = (u_short) icpPortNumOverride;
  393.     _db_init(Config.Log.log, Config.debugOptions);
  394.     fd_open(fileno(debug_log), FD_LOG, Config.Log.log);
  395. #if MEM_GEN_TRACE
  396.     log_trace_init("/tmp/squid.alloc");
  397. #endif
  398.     debug(1, 0) ("Starting Squid Cache version %s for %s...n",
  399. version_string,
  400. CONFIG_HOST_TYPE);
  401.     debug(1, 1) ("Process ID %dn", (int) getpid());
  402.     debug(1, 1) ("With %d file descriptors availablen", Squid_MaxFD);
  404.     if (!configured_once)
  405. disk_init(); /* disk_init must go before ipcache_init() */
  406.     ipcache_init();
  407.     fqdncache_init();
  408.     dnsInit();
  409.     redirectInit();
  410.     authenticateInit();
  411.     useragentOpenLog();
  412.     httpHeaderInitModule(); /* must go before any header processing (e.g. the one in errorInitialize) */
  413.     httpReplyInitModule(); /* must go before accepting replies */
  414.     errorInitialize();
  415.     accessLogInit();
  416. #if USE_IDENT
  417.     identInit();
  418. #endif
  419. #ifdef SQUID_SNMP
  420.     snmpInit();
  421. #endif
  423. #if MALLOC_DBG
  424.     malloc_debug(0, malloc_debug_level);
  425. #endif
  427.     if (!configured_once) {
  428. unlinkdInit();
  429. urlInitialize();
  430. cachemgrInit();
  431. statInit();
  432. storeInit();
  433. if (Config.effectiveUser) {
  434.     /* we were probably started as root, so cd to a swap
  435.      * directory in case we dump core */
  436.     if (chdir(storeSwapDir(0)) < 0) {
  437. debug(50, 0) ("%s: %sn", storeSwapDir(0), xstrerror());
  438. fatal_dump("Cannot cd to swap directory?");
  439.     }
  440. }
  441. /* after this point we want to see the mallinfo() output */
  442. do_mallinfo = 1;
  443. mimeInit(Config.mimeTablePathname);
  444. pconnInit();
  445. eventInit();
  446. refreshInit();
  447. #if DELAY_POOLS
  448. delayPoolsInit();
  449. #endif
  450. fwdInit();
  451.     }
  452.     serverConnectionsOpen();
  453.     if (theOutIcpConnection >= 0) {
  454. if (!Config2.Accel.on || Config.onoff.accel_with_proxy)
  455.     neighbors_open(theOutIcpConnection);
  456. else
  457.     debug(1, 1) ("ICP port disabled in httpd_accelerator moden");
  458.     }
  459.     if (!configured_once)
  460. writePidFile(); /* write PID file */
  462. #ifdef _SQUID_LINUX_THREADS_
  463.     squid_signal(SIGQUIT, rotate_logs, SA_RESTART);
  464.     squid_signal(SIGTRAP, sigusr2_handle, SA_RESTART);
  465. #else
  466.     squid_signal(SIGUSR1, rotate_logs, SA_RESTART);
  467.     squid_signal(SIGUSR2, sigusr2_handle, SA_RESTART);
  468. #endif
  469.     squid_signal(SIGHUP, reconfigure, SA_RESTART);
  470.     squid_signal(SIGTERM, shut_down, SA_NODEFER | SA_RESETHAND | SA_RESTART);
  471.     squid_signal(SIGINT, shut_down, SA_NODEFER | SA_RESETHAND | SA_RESTART);
  472. #if ALARM_UPDATES_TIME
  473.     squid_signal(SIGALRM, time_tick, SA_RESTART);
  474.     alarm(1);
  475. #endif
  476.     debug(1, 1) ("Ready to serve requests.n");
  478.     if (!configured_once) {
  479. eventAdd("storeMaintain", storeMaintainSwapSpace, NULL, 1.0, 1);
  480. eventAdd("storeDirClean", storeDirClean, NULL, 15.0, 1);
  481. if (Config.onoff.announce)
  482.     eventAdd("start_announce", start_announce, NULL, 3600.0, 1);
  483. eventAdd("ipcache_purgelru", ipcache_purgelru, NULL, 10.0, 1);
  484. eventAdd("fqdncache_purgelru", fqdncache_purgelru, NULL, 15.0, 1);
  485.     }
  486.     configured_once = 1;
  487. }
  489. int
  490. main(int argc, char **argv)
  491. {
  492.     int errcount = 0;
  493.     int n; /* # of GC'd objects */
  494.     time_t loop_delay;
  496.     debug_log = stderr;
  497.     if (FD_SETSIZE < Squid_MaxFD)
  498. Squid_MaxFD = FD_SETSIZE;
  500.     /* call mallopt() before anything else */
  501. #if HAVE_MALLOPT
  502. #ifdef M_GRAIN
  503.     /* Round up all sizes to a multiple of this */
  504.     mallopt(M_GRAIN, 16);
  505. #endif
  506. #ifdef M_MXFAST
  507.     /* biggest size that is considered a small block */
  508.     mallopt(M_MXFAST, 256);
  509. #endif
  510. #ifdef M_NBLKS
  511.     /* allocate this many small blocks at once */
  512.     mallopt(M_NLBLKS, 32);
  513. #endif
  514. #endif /* HAVE_MALLOPT */
  516.     memset(&local_addr, '', sizeof(struct in_addr));
  517.     safe_inet_addr(localhost, &local_addr);
  518.     memset(&any_addr, '', sizeof(struct in_addr));
  519.     safe_inet_addr("0.0.0.0", &any_addr);
  520.     memset(&no_addr, '', sizeof(struct in_addr));
  521.     safe_inet_addr("255.255.255.255", &no_addr);
  522.     squid_srandom(time(NULL));
  524.     getCurrentTime();
  525.     squid_start = current_time;
  526.     failure_notify = fatal_dump;
  528.     mainParseOptions(argc, argv);
  530.     /* parse configuration file
  531.      * note: in "normal" case this used to be called from mainInitialize() */
  532.     {
  533. int parse_err;
  534. if (!ConfigFile)
  535.     ConfigFile = xstrdup(DefaultConfigFile);
  536. assert(!configured_once);
  537. cbdataInit();
  538. #if USE_LEAKFINDER
  539. leakInit();
  540. #endif
  541. memInit(); /* memInit is required for config parsing */
  542. parse_err = parseConfigFile(ConfigFile);
  544. if (opt_parse_cfg_only)
  545.     return parse_err;
  546.     }
  548.     /* send signal to running copy and exit */
  549.     if (opt_send_signal != -1) {
  550. sendSignal();
  551. /* NOTREACHED */
  552.     }
  553.     if (opt_create_swap_dirs) {
  554. setEffectiveUser();
  555. debug(0, 0) ("Creating Swap Directoriesn");
  556. storeCreateSwapDirectories();
  557. return 0;
  558.     }
  559.     if (!opt_no_daemon)
  560. watch_child(argv);
  561.     setMaxFD();
  563.     if (opt_catch_signals)
  564. for (n = Squid_MaxFD; n > 2; n--)
  565.     close(n);
  567.     /* init comm module */
  568.     comm_init();
  569.     comm_select_init();
  571.     if (opt_no_daemon) {
  572. /* we have to init fdstat here. */
  573. fd_open(0, FD_LOG, "stdin");
  574. fd_open(1, FD_LOG, "stdout");
  575. fd_open(2, FD_LOG, "stderr");
  576.     }
  577.     mainInitialize();
  579.     /* main loop */
  580.     for (;;) {
  581. if (do_reconfigure) {
  582.     mainReconfigure();
  583.     do_reconfigure = 0;
  584. } else if (do_rotate) {
  585.     mainRotate();
  586.     do_rotate = 0;
  587. } else if (do_shutdown) {
  588.     time_t wait = do_shutdown > 0 ? (int) Config.shutdownLifetime : 0;
  589.     debug(1, 1) ("Preparing for shutdown after %d requestsn",
  590. Counter.client_http.requests);
  591.     debug(1, 1) ("Waiting %d seconds for active connections to finishn",
  592. wait);
  593.     do_shutdown = 0;
  594.     shutting_down = 1;
  595.     serverConnectionsClose();
  596.     dnsShutdown();
  597.     redirectShutdown();
  598.     authenticateShutdown();
  599.     eventAdd("SquidShutdown", SquidShutdown, NULL, (double) (wait + 1), 1);
  600. }
  601. eventRun();
  602. if ((loop_delay = eventNextTime()) < 0)
  603.     loop_delay = 0;
  604. #if HAVE_POLL
  605. switch (comm_poll(loop_delay)) {
  606. #else
  607. switch (comm_select(loop_delay)) {
  608. #endif
  609. case COMM_OK:
  610.     errcount = 0; /* reset if successful */
  611.     break;
  612. case COMM_ERROR:
  613.     errcount++;
  614.     debug(1, 0) ("Select loop Error. Retry %dn", errcount);
  615.     if (errcount == 10)
  616. fatal_dump("Select Loop failed!");
  617.     break;
  618. case COMM_TIMEOUT:
  619.     break;
  620. case COMM_SHUTDOWN:
  621.     SquidShutdown(NULL);
  622.     break;
  623. default:
  624.     fatal_dump("MAIN: Internal error -- this should never happen.");
  625.     break;
  626. }
  627.     }
  628.     /* NOTREACHED */
  629.     return 0;
  630. }
  632. static void
  633. sendSignal(void)
  634. {
  635.     pid_t pid;
  636.     debug_log = stderr;
  637.     pid = readPidFile();
  638.     if (pid > 1) {
  639. if (kill(pid, opt_send_signal) &&
  640. /* ignore permissions if just running check */
  641.     !(opt_send_signal == 0 && errno == EPERM)) {
  642.     fprintf(stderr, "%s: ERROR: Could not send ", appname);
  643.     fprintf(stderr, "signal %d to process %d: %sn",
  644. opt_send_signal, (int) pid, xstrerror());
  645.     exit(1);
  646. }
  647.     } else {
  648. fprintf(stderr, "%s: ERROR: No running copyn", appname);
  649. exit(1);
  650.     }
  651.     /* signal successfully sent */
  652.     exit(0);
  653. }
  655. static void
  656. watch_child(char *argv[])
  657. {
  658.     char *prog;
  659.     int failcount = 0;
  660.     time_t start;
  661.     time_t stop;
  662. #ifdef _SQUID_NEXT_
  663.     union wait status;
  664. #else
  665.     int status;
  666. #endif
  667.     pid_t pid;
  668.     int i;
  669.     if (*(argv[0]) == '(')
  670. return;
  671.     openlog(appname, LOG_PID | LOG_NDELAY | LOG_CONS, LOG_LOCAL4);
  672.     if ((pid = fork()) < 0)
  673. syslog(LOG_ALERT, "fork failed: %s", xstrerror());
  674.     else if (pid > 0)
  675. exit(0);
  676.     if (setsid() < 0)
  677. syslog(LOG_ALERT, "setsid failed: %s", xstrerror());
  678. #ifdef TIOCNOTTY
  679.     if ((i = open("/dev/tty", O_RDWR)) >= 0) {
  680. ioctl(i, TIOCNOTTY, NULL);
  681. close(i);
  682.     }
  683. #endif
  684.     for (i = 0; i < Squid_MaxFD; i++)
  685. close(i);
  686.     umask(0);
  687.     for (;;) {
  688. if ((pid = fork()) == 0) {
  689.     /* child */
  690.     prog = xstrdup(argv[0]);
  691.     argv[0] = xstrdup("(squid)");
  692.     execvp(prog, argv);
  693.     syslog(LOG_ALERT, "execvp failed: %s", xstrerror());
  694. }
  695. /* parent */
  696. syslog(LOG_NOTICE, "Squid Parent: child process %d started", pid);
  697. time(&start);
  698. squid_signal(SIGINT, SIG_IGN, SA_RESTART);
  699. #ifdef _SQUID_NEXT_
  700. pid = wait3(&status, 0, NULL);
  701. #else
  702. pid = waitpid(-1, &status, 0);
  703. #endif
  704. time(&stop);
  705. if (WIFEXITED(status)) {
  706.     syslog(LOG_NOTICE,
  707. "Squid Parent: child process %d exited with status %d",
  708. pid, WEXITSTATUS(status));
  709. } else if (WIFSIGNALED(status)) {
  710.     syslog(LOG_NOTICE,
  711. "Squid Parent: child process %d exited due to signal %d",
  712. pid, WTERMSIG(status));
  713. } else {
  714.     syslog(LOG_NOTICE, "Squid Parent: child process %d exited", pid);
  715. }
  716. if (stop - start < 10)
  717.     failcount++;
  718. else
  719.     failcount = 0;
  720. if (failcount == 5) {
  721.     syslog(LOG_ALERT, "Exiting due to repeated, frequent failures");
  722.     exit(1);
  723. }
  724. if (WIFEXITED(status))
  725.     if (WEXITSTATUS(status) == 0)
  726. exit(0);
  727. squid_signal(SIGINT, SIG_DFL, SA_RESTART);
  728. sleep(3);
  729.     }
  730.     /* NOTREACHED */
  731. }
  733. static void
  734. SquidShutdown(void *unused)
  735. {
  736.     debug(1, 1) ("Shutting down...n");
  737.     if (Config.pidFilename && strcmp(Config.pidFilename, "none")) {
  738. enter_suid();
  739. safeunlink(Config.pidFilename, 0);
  740. leave_suid();
  741.     }
  742.     icpConnectionClose();
  743. #if USE_HTCP
  744.     htcpSocketClose();
  745. #endif
  746. #ifdef SQUID_SNMP
  747.     snmpConnectionClose();
  748. #endif
  749.     releaseServerSockets();
  750.     commCloseAllSockets();
  751.     unlinkdClose();
  752. #if USE_ASYNC_IO
  753.     aioSync(); /* flush pending object writes / unlinks */
  754. #endif
  755.     storeDirWriteCleanLogs(0);
  756.     PrintRusage();
  757.     dumpMallocStats();
  758. #if USE_ASYNC_IO
  759.     aioSync(); /* flush log writes */
  760. #endif
  761.     storeLogClose();
  762.     accessLogClose();
  763. #if USE_ASYNC_IO
  764.     aioSync(); /* flush log close */
  765. #endif
  766. #if PURIFY || XMALLOC_TRACE
  767.     configFreeMemory();
  768.     storeFreeMemory();
  769.     /*stmemFreeMemory(); */
  770.     netdbFreeMemory();
  771.     ipcacheFreeMemory();
  772.     fqdncacheFreeMemory();
  773.     asnFreeMemory();
  774.     clientdbFreeMemory();
  775.     httpHeaderCleanModule();
  776.     statFreeMemory();
  777.     eventFreeMemory();
  778.     mimeFreeMemory();
  779.     errorClean();
  780. #endif
  781. #if !XMALLOC_TRACE
  782.     if (opt_no_daemon) {
  783. file_close(0);
  784. file_close(1);
  785. file_close(2);
  786.     }
  787. #endif
  788.     fdDumpOpen();
  789.     fdFreeMemory();
  790.     memClean();
  791. #if XMALLOC_TRACE
  792.     xmalloc_find_leaks();
  793.     debug(1, 0) ("Memory used after shutdown: %dn", xmalloc_total);
  794. #endif
  795. #if MEM_GEN_TRACE
  796.     log_trace_done();
  797. #endif
  798.     debug(1, 1) ("Squid Cache (Version %s): Exiting normally.n",
  799. version_string);
  800.     fclose(debug_log);
  801.     exit(0);
  802. }