软件著作权源代码.doc

60 #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #define NS_MAIN 1#include #endif#ifdef DLZ#include #endifstatic tybs_boolean_twant_stats = TYBS_FALSE;static charprogram_nameTYBS_DIR_NAMEMAX = named;static charabsolute_conffileTYBS_DIR_PATHMAX;static charsaved_command_line512;static charversion512;static unsigned intmaxsocks = 0;voidns_main_earlywarning(const char *format, .) va_list args;va_start(args, format);if (ns_g_lctx != NULL) tybs_log_vwrite(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_WARNING, format, args); else fprintf(stderr, %s: , program_name);vfprintf(stderr, format, args);fprintf(stderr, n);fflush(stderr);va_end(args);Void ns_main_earlyfatal(const char *format, .) va_list args;va_start(args, format);if (ns_g_lctx != NULL) tybs_log_vwrite(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_CRITICAL, format, args);tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_CRITICAL, exiting (due to early fatal error); else fprintf(stderr, %s: , program_name);vfprintf(stderr, format, args);fprintf(stderr, n);fflush(stderr);va_end(args);exit(1);static voidassertion_failed(const char *file, int line, tybs_assertiontype_t type, const char *cond)if (ns_g_lctx != NULL) tybs_assertion_setcallback(NULL);tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_CRITICAL, %s:%d: %s(%s) failed, file, line, tybs_assertion_typetotext(type), cond);tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_CRITICAL, exiting (due to assertion failure); else fprintf(stderr, %s:%d: %s(%s) failedn,file, line, tybs_assertion_typetotext(type), cond);fflush(stderr);if (ns_g_coreok)abort();exit(1);static voidlibrary_fatal_error(const char *file, int line, const char *format, va_list args) TYBS_FORMAT_PRINTF(3, 0);static voidlibrary_fatal_error(const char *file, int line, const char *format, va_list args)if (ns_g_lctx != NULL) tybs_error_setfatal(NULL);tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_CRITICAL, %s:%d: fatal error:, file, line);tybs_log_vwrite(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_CRITICAL, format, args);tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_CRITICAL, exiting (due to fatal error in library); else fprintf(stderr, %s:%d: fatal error: , file, line);vfprintf(stderr, format, args);fprintf(stderr, n);fflush(stderr);if (ns_g_coreok)abort();exit(1);static voidlibrary_unexpected_error(const char *file, int line, const char *format, va_list args) TYBS_FORMAT_PRINTF(3, 0);static voidlibrary_unexpected_error(const char *file, int line, const char *format, va_list args)if (ns_g_lctx != NULL) tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_ERROR, %s:%d: unexpected error:, file, line);tybs_log_vwrite(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_ERROR, format, args); else fprintf(stderr, %s:%d: fatal error: , file, line);vfprintf(stderr, format, args);fprintf(stderr, n);fflush(stderr);static voidlwresd_usage(void) fprintf(stderr,usage: lwresd -4|-6 -c conffile | -C resolvconffile -d debugleveln -f|-g -n number_of_cpus -p port -P listen-port -sn -t chrootdir -u username -i pidfilen -m usage|trace|record|size|mctxn);static voidusage(void) if (ns_g_lwresdonly) lwresd_usage();return;fprintf(stderr,usage: named -4|-6 -c conffile -d debuglevel -f|-g -n number_of_cpusn -p port -s -t chrootdir -u usernamen -m usage|trace|record|size|mctxn);static voidsave_command_line(int argc, char *argv) int i;char *src;char *dst;char *eob;const char truncated = .;tybs_boolean_t quoted = TYBS_FALSE;dst = saved_command_line;eob = saved_command_line + sizeof(saved_command_line);for (i = 1; i argc & dst eob; i+) *dst+ = ;src = argvi;while (*src != 0 & dst = sizeof(truncated);if (dst = eob)strcpy(eob - sizeof(truncated), truncated);else*dst = 0;static intparse_int(char *arg, const char *desc) char *endp;int tmp;long int ltmp;ltmp = strtol(arg, &endp, 10);tmp = (int) ltmp;if (*endp != 0)ns_main_earlyfatal(%s %s must be numeric, desc, arg);if (tmp name != NULL; def+) if (arglen = (int)strlen(def-name) & memcmp(arg, def-name, arglen) = 0) *ret |= def-value;goto found;ns_main_earlyfatal(unrecognized flag %.*s, arglen, arg); found:if (*end = 0)break;arg = end + 1;static voidparse_command_line(int argc, char *argv) int ch;int port;tybs_boolean_t disable6 = TYBS_FALSE;tybs_boolean_t disable4 = TYBS_FALSE;save_command_line(argc, argv);tybs_commandline_errprint = TYBS_FALSE;while (ch = tybs_commandline_parse(argc, argv, 46c:C:d:fgi:lm:n:N:p:P: sS:t:T:u:vVx:) != -1) switch (ch) case 4:if (disable4)ns_main_earlyfatal(cannot specify -4 and -6);if (tybs_net_probeipv4() != TYBS_R_SUCCESS)ns_main_earlyfatal(IPv4 not supported by OS);tybs_net_disableipv6();disable6 = TYBS_TRUE;break;case 6:if (disable6)ns_main_earlyfatal(cannot specify -4 and -6);if (tybs_net_probeipv6() != TYBS_R_SUCCESS)ns_main_earlyfatal(IPv6 not supported by OS);tybs_net_disableipv4();disable4 = TYBS_TRUE;break;case c:ns_g_conffile = tybs_commandline_argument;lwresd_g_conffile = tybs_commandline_argument;if (lwresd_g_useresolvconf)ns_main_earlyfatal(cannot specify -c and -C);ns_g_conffileset = TYBS_TRUE;break;case C:lwresd_g_resolvconffile = tybs_commandline_argument;if (ns_g_conffileset)ns_main_earlyfatal(cannot specify -c and -C);lwresd_g_useresolvconf = TYBS_TRUE;break;case d:ns_g_debuglevel = parse_int(tybs_commandline_argument, debug level);break;case f:ns_g_foreground = TYBS_TRUE;break;case g:ns_g_foreground = TYBS_TRUE;ns_g_logstderr = TYBS_TRUE;break;/* XXXBEW -i should be removed */case i:lwresd_g_defaultpidfile = tybs_commandline_argument;break;case l:ns_g_lwresdonly = TYBS_TRUE;break;case m:set_flags(tybs_commandline_argument, mem_debug_flags, &tybs_mem_debugging);break;case N: /* Deprecated. */case n:ns_g_cpus = parse_int(tybs_commandline_argument, number of cpus);if (ns_g_cpus = 0)ns_g_cpus = 1;break;case p:port = parse_int(tybs_commandline_argument, port);if (port 65535)ns_main_earlyfatal(port %s out of range, tybs_commandline_argument);ns_g_port = port;break;/* XXXBEW Should -P be removed? */case P:port = parse_int(tybs_commandline_argument, port);if (port 65535)ns_main_earlyfatal(port %s out of range, tybs_commandline_argument);lwresd_g_listenport = port;break;case s:want_stats = TYBS_TRUE;break;case S:maxsocks = parse_int(tybs_commandline_argument, max number of sockets);break;case t:ns_g_chrootdir = tybs_commandline_argument;break;case T:if (strcmp(tybs_commandline_argument, clienttest) = 0)ns_g_clienttest = TYBS_TRUE;elsefprintf(stderr, unknown -T flag %sn,tybs_commandline_argument);break;case u:ns_g_username = tybs_commandline_argument;break;case v:printf(BIND %sn, ns_g_version);exit(0);case V:printf(BIND %s built with %sn, ns_g_version,ns_g_configargs);exit(0);case ?:usage();if (tybs_commandline_option = ?)exit(0);ns_main_earlyfatal(unknown option -%c, tybs_commandline_option);default:ns_main_earlyfatal(parsing options returned %d, ch);argc -= tybs_commandline_index;argv += tybs_commandline_index;if (argc 0) usage();ns_main_earlyfatal(extra command line arguments);static tybs_result_tcreate_managers(void) tybs_result_t result;unsigned int socks;#ifdef TYBS_PLATFORM_USETHREADSunsigned int cpus_detected;#endif#ifdef TYBS_PLATFORM_USETHREADScpus_detected = tybs_os_ncpus();if (ns_g_cpus = 0)ns_g_cpus = cpus_detected;tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_SERVER, TYBS_LOG_INFO, found %u CPU%s, using %u worker thread%s, cpus_detected, cpus_detected = 1 ? : s, ns_g_cpus, ns_g_cpus = 1 ? : s);#elsens_g_cpus = 1;#endifresult = tybs_taskmgr_create(ns_g_mctx, ns_g_cpus, 0, &ns_g_taskmgr);if (result != TYBS_R_SUCCESS) UNEXPECTED_ERROR(_FILE_, _LINE_, tybs_taskmgr_create() failed: %s, tybs_result_totext(result);return (TYBS_R_UNEXPECTED);result = tybs_timermgr_create(ns_g_mctx, &ns_g_timermgr);if (result != TYBS_R_SUCCESS) UNEXPECTED_ERROR(_FILE_, _LINE_, tybs_timermgr_create() failed: %s, tybs_result_totext(result);return (TYBS_R_UNEXPECTED);result = tybs_socketmgr_create2(ns_g_mctx, &ns_g_socketmgr, maxsocks);if (result != TYBS_R_SUCCESS) UNEXPECTED_ERROR(_FILE_, _LINE_, tybs_socketmgr_create() failed: %s, tybs_result_totext(result);return (TYBS_R_UNEXPECTED);result = tybs_socketmgr_getmaxsockets(ns_g_socketmgr, &socks);if (result = TYBS_R_SUCCESS) tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_SERVER, TYBS_LOG_INFO, using up to %u sockets, socks);result = tybs_entropy_create(ns_g_mctx, &ns_g_entropy);if (result != TYBS_R_SUCCESS) UNEXPECTED_ERROR(_FILE_, _LINE_, tybs_entropy_create() failed: %s, tybs_result_totext(result);return (TYBS_R_UNEXPECTED);result = tybs_hash_create(ns_g_mctx, ns_g_entropy, DNS_NAME_MAXWIRE);if (result != TYBS_R_SUCCESS) UNEXPECTED_ERROR(_FILE_, _LINE_, tybs_hash_create() failed: %s, tybs_result_totext(result);return (TYBS_R_UNEXPECTED);return (TYBS_R_SUCCESS);static voiddestroy_managers(void) ns_lwresd_shutdown();tybs_entropy_detach(&ns_g_entropy);if (ns_g_fallbackentropy != NULL)tybs_entropy_detach(&ns_g_fallbackentropy);tybs_taskmgr_destroy(&ns_g_taskmgr);tybs_timermgr_destroy(&ns_g_timermgr);tybs_socketmgr_destroy(&ns_g_socketmgr);tybs_hash_destroy();static voidsetup(void) tybs_result_t result;#ifdef HAVE_LIBSCFchar *instance = NULL;#endifns_os_inituserinfo(ns_g_username);ns_os_tzset();ns_os_opendevnull();#ifdef HAVE_LIBSCFresult = ns_smf_get_instance(&instance, 0, ns_g_mctx);if (result = TYBS_R_SUCCESS)ns_smf_got_instance = 1;elsens_smf_got_instance = 0;if (instance != NULL)tybs_mem_free(ns_g_mctx, instance);#endif /* HAVE_LIBSCF */#ifdef PATH_RANDOMDEVif (ns_g_chrootdir != NULL) result = tybs_entropy_create(ns_g_mctx, &ns_g_fallbackentropy);if (result != TYBS_R_SUCCESS)ns_main_earlyfatal(tybs_entropy_create() failed: %s, tybs_result_totext(result);result = tybs_entropy_createfilesource(ns_g_fallbackentropy, PATH_RANDOMDEV);if (result != TYBS_R_SUCCESS) ns_main_earlywarning(could not open pre-chroot entropy source %s: %s, PATH_RANDOMDEV, tybs_result_totext(result);tybs_entropy_detach(&ns_g_fallbackentropy);#endifns_os_chroot(ns_g_chrootdir);ns_os_minprivs();result = ns_log_init(TYBS_TF(ns_g_username != NULL);if (result != TYBS_R_SUCCESS)ns_main_earlyfatal(ns_log_init() failed: %s, tybs_result_totext(result);if (!ns_g_foreground)ns_os_daemonize();result = tybs_app_start();if (result != TYBS_R_SUCCESS)ns_main_earlyfatal(tybs_app_start() failed: %s, tybs_result_totext(result);tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_NOTICE, starting BIND %s%s, ns_g_version, saved_command_line);tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN, TYBS_LOG_NOTICE, built with %s, ns_g_configargs);(void)tybs_resource_getlimit(tybs_resource_stacksize, &ns_g_initstacksize);(void)tybs_resource_getlimit(tybs_resource_datasize, &ns_g_initdatasize);(void)tybs_resource_getlimit(tybs_resource_coresize, &ns_g_initcoresize);(void)tybs_resource_getlimit(tybs_resource_openfiles, &ns_g_initopenfiles);if (! tybs_file_isabsolute(ns_