软件著作权源代码

1、.*include <tybs/commandline.h>*include <tybs/dir.h>*include <tybs/entropy.h>*include <tybs/file.h>*include <tybs/hash.h>*include <tybs/os.h>*include <tybs/platform.h>*include <tybs/resource.h>*include <tybs/stdio.h>*include <tybs/string.h>*

2、include <tybs/task.h>*include <tybs/timer.h>*include <tybs/util.h>*include <tybscc/result.h>*include <dns/dispatch.h>*include <dns/name.h>*include <dns/result.h>*include <dns/view.h>*include <dst/result.h>*define NS_MAIN 1*include <named/ns_sm

3、f_globals.h>*endif*ifdef DLZ*include <dlz/dlz_drivers.h>*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 intmaxsoc

4、ks = 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)

5、;fprintf(stderr, "n");fflush(stderr);va_end(args);Voidns_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_LOG

6、CATEGORY_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

7、, 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(n

8、s_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_e

9、rror(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

10、, 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 librar

11、y)"); 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_PRI

12、NTF(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_LOGCA

13、TEGORY_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 re

14、solvconffile ""-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,&qu

15、ot;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 = &quo

16、t;."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 < eob) if (quoted | isalnum(*src & 0xff) | *

17、src = '-' | *src = '_' | *src = '.' | *src = '/') *dst+ = *src+;quoted = TYBS_FALSE; else *dst+ = ''quoted = TYBS_TRUE;INSIST(sizeof(saved_command_line) >= sizeof(truncated);if (dst = eob)strcpy(eob - sizeof(truncated), truncated);else*dst = '0'stat

18、ic 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 < 0 | tmp != ltmp)ns_main_earlyfatal("%s '%s' o

19、ut of range", desc, arg);return (tmp);static struct flag_def const char *name;unsigned int value; mem_debug_flags = "trace", TYBS_MEM_DEBUGTRACE , "record", TYBS_MEM_DEBUGRECORD , "usage", TYBS_MEM_DEBUGUSAGE , "size", TYBS_MEM_DEBUGSIZE , "mctx"

20、;, TYBS_MEM_DEBUGCTX , NULL, 0 ;static voidset_flags(const char *arg, struct flag_def *defs, unsigned int *ret) for (;) const struct flag_def *def;const char *end = strchr(arg, ',');int arglen;if (end = NULL)end = arg + strlen(arg);arglen = end - arg;for (def = defs; def->name != NULL; de

21、f+) 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 *

22、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 &

23、#39;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

24、 -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)

25、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_

26、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_command

27、line_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"

28、);if (ns_g_cpus = 0)ns_g_cpus = 1;break;case 'p':port = parse_int(tybs_commandline_argument, "port");if (port < 1 | port > 65535)ns_main_earlyfatal("port '%s' out of range",tybs_commandline_argument);ns_g_port = port;break;/* XXXBEW Should -P be removed"

29、; */case 'P':port = parse_int(tybs_commandline_argument, "port");if (port < 1 | 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'

30、;: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, "unknow

31、n -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 '&

32、quot;':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 (arg

33、c > 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_cpu

34、s = 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

35、_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_

36、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(_FIL

37、E_, _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, "u

38、sing 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_e

39、ntropy, 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_entrop

40、y);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 *ins

41、tance = 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);*

42、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

43、_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(

44、);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(

45、) 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 %

46、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_resourc

47、e_openfiles, &ns_g_initopenfiles);if (! tybs_file_isabsolute(ns_g_conffile) result = tybs_file_absolutepath(ns_g_conffile, absolute_conffile, sizeof(absolute_conffile);if (result != TYBS_R_SUCCESS)ns_main_earlyfatal("could not construct absolute path of " "configuration file: %s&q

48、uot;,tybs_result_totext(result);ns_g_conffile = absolute_conffile;result = tybs_time_now(&ns_g_boottime);if (result != TYBS_R_SUCCESS)ns_main_earlyfatal("tybs_time_now() failed: %s",tybs_result_totext(result);result = create_managers();if (result != TYBS_R_SUCCESS)ns_main_earlyfatal(&q

49、uot;create_managers() failed: %s",tybs_result_totext(result);ns_builtin_init();*ifdef DLZresult = dlz_drivers_init();if (result != TYBS_R_SUCCESS)ns_main_earlyfatal("dlz_drivers_init() failed: %s",tybs_result_totext(result);*endifns_server_create(ns_g_mctx, &ns_g_server);static vo

50、idcleanup(void) destroy_managers();ns_server_destroy(&ns_g_server);ns_builtin_deinit();*ifdef DLZdlz_drivers_clear();*endifdns_name_destroy();tybs_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL, NS_LOGMODULE_MAIN,TYBS_LOG_NOTICE, "exiting");ns_log_shutdown();static char *memstats = NULL;v

51、oidns_main_setmemstats(const char *filename) if (memstats != NULL) free(memstats);memstats = NULL;if (filename = NULL)return;memstats = malloc(strlen(filename) + 1);if (memstats)strcpy(memstats, filename);*ifdef HAVE_LIBSCFtybs_result_tns_smf_get_instance(char *ins_name, int debug, tybs_mem_t *mctx)

52、 scf_handle_t *h = NULL;int namelen;char *instance;REQUIRE(ins_name != NULL && *ins_name = NULL);if (h = scf_handle_create(SCF_VERSION) = NULL) if (debug)UNEXPECTED_ERROR(_FILE_, _LINE_, "scf_handle_create() failed: %s", scf_strerror(scf_error();return (TYBS_R_FAILURE);if (scf_hand

53、le_bind(h) = -1) if (debug)UNEXPECTED_ERROR(_FILE_, _LINE_, "scf_handle_bind() failed: %s", scf_strerror(scf_error();scf_handle_destroy(h);return (TYBS_R_FAILURE);if (namelen = scf_myname(h, NULL, 0) = -1) if (debug)UNEXPECTED_ERROR(_FILE_, _LINE_, "scf_myname() failed: %s", scf_

54、strerror(scf_error();scf_handle_destroy(h);return (TYBS_R_FAILURE);if (instance = tybs_mem_allocate(mctx, namelen + 1) = NULL) UNEXPECTED_ERROR(_FILE_, _LINE_, "ns_smf_get_instance memory " "allocation failed: %s",tybs_result_totext(TYBS_R_NOMEMORY);scf_handle_destroy(h);return (

55、TYBS_R_FAILURE);if (scf_myname(h, instance, namelen + 1) = -1) if (debug)UNEXPECTED_ERROR(_FILE_, _LINE_, "scf_myname() failed: %s", scf_strerror(scf_error();scf_handle_destroy(h);tybs_mem_free(mctx, instance);return (TYBS_R_FAILURE);scf_handle_destroy(h);*ins_name = instance;return (TYBS_R_SUCCESS);*endif intmain(int argc, char *argv) tybs_result_t result;*ifdef HAVE_LIBSCFchar *instance = NULL;*endifstrlcat(version,*ifdef _DATE_"named version: BIND " VERSION " (" _DATE_ ")",*else"named versio