资源描述
#include <tybs/mandline、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>
#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_smf_globals、h>
#endif
#ifdef DLZ
#include <dlz/dlz_drivers、h>
#endif
static tybs_boolean_tﻩwant_stats = TYBS_FALSE;
static charﻩ program_name[TYBS_DIR_NAMEMAX] = "named";
static char absolute_conf];
static charﻩ saved_mand_line[512];
static charﻩﻩversion[512];
static unsigned intﻩmaxsocks = 0;
void
ns_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 void
assertion_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) failed\n",
ﻩ file, line, tybs_assertion_typetotext(type), cond);
ﻩﻩfflush(stderr);
}
ﻩif (ns_g_coreok)
ﻩ abort();
ﻩexit(1);
}
static void
library_fatal_error(const char *file, int line, const char *format,
ﻩ va_list args) TYBS_FORMAT_PRINTF(3, 0);
static void
library_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 void
library_unexpected_error(const char *file, int line, const char *format,
ﻩ ﻩ va_list args) TYBS_FORMAT_PRINTF(3, 0);
static void
library_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 void
lwresd_usage(void) {
fprintf(stderr,
ﻩﻩ"usage: lwresd [-4|-6] [-c conffile | -C resolvconffile] "
"[-d debuglevel]\n"
ﻩ " [-f|-g] [-n number_of_cpus] [-p port] "
ﻩ"[-P listen-port] [-s]\n"
" [-t chrootdir] [-u username] [-i pidfile]\n"
ﻩ" [-m {usage|trace|record|size|mctx}]\n");
}
static void
usage(void) {
if (ns_g_lwresdonly) {
ﻩﻩlwresd_usage();
ﻩ return;
}
fprintf(stderr,
ﻩﻩ"usage: named [-4|-6] [-c conffile] [-d debuglevel] "
ﻩ "[-f|-g] [-n number_of_cpus]\n"
" [-p port] [-s] [-t chrootdir] [-u username]\n"
ﻩﻩ" [-m {usage|trace|record|size|mctx}]\n");
}
static void
save_mand_line(int argc, char *argv[]) {
ﻩint i;
ﻩchar *src;
char *dst;
char *eob;
const char truncated[] = "、、、";
tybs_boolean_t quoted = TYBS_FALSE;
ﻩdst = saved_mand_line;
ﻩeob = saved_mand_line + sizeof(saved_mand_line);
for (i = 1; i < argc && dst < eob; i++) {
ﻩ *dst++ = ' ';
ﻩﻩsrc = argv[i];
ﻩﻩwhile (*src != '\0' && dst < eob) {
ﻩﻩ if (quoted || isalnum(*src & 0xff) ||
ﻩﻩﻩ *src == '-' || *src == '_' ||
ﻩﻩﻩ *src == '、' || *src == '/') {
ﻩ*dst++ = *src++;
ﻩ ﻩ quoted = TYBS_FALSE;
ﻩﻩﻩ} else {
ﻩ*dst++ = '\\';
ﻩ ﻩquoted = TYBS_TRUE;
ﻩﻩ }
ﻩ }
}
INSIST(sizeof(saved_mand_line) >= sizeof(truncated));
ﻩif (dst == eob)
ﻩstrcpy(eob - sizeof(truncated), truncated);
ﻩelse
ﻩ*dst = '\0';
}
static int
parse_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' out 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", TYBS_MEM_DEBUGCTX },
{ NULL, 0 }
};
static void
set_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; 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 void
parse_mand_line(int argc, char *argv[]) {
int ch;
ﻩint port;
tybs_boolean_t disable6 = TYBS_FALSE;
ﻩtybs_boolean_t disable4 = TYBS_FALSE;
save_mand_line(argc, argv);
ﻩtybs_mandline_errprint = TYBS_FALSE;
while ((ch = tybs_mandline_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_mandline_argument;
lwresd_g_conffile = tybs_mandline_argument;
ﻩ ﻩif (lwresd_g_useresolvconf)
ﻩﻩ ns_main_earlyfatal("cannot specify -c and -C");
ﻩﻩns_g_conf = TYBS_TRUE;
ﻩﻩ break;
case 'C':
ﻩ ﻩlwresd_g_resolvconffile = tybs_mandline_argument;
ﻩif (ns_g_conf)
ﻩﻩﻩns_main_earlyfatal("cannot specify -c and -C");
ﻩlwresd_g_useresolvconf = TYBS_TRUE;
ﻩﻩ break;
ﻩﻩcase 'd':
ﻩ ns_g_debuglevel = parse_int(tybs_mandline_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_mandline_argument;
ﻩ ﻩbreak;
ﻩﻩcase 'l':
ﻩﻩﻩns_g_lwresdonly = TYBS_TRUE;
ﻩﻩbreak;
ﻩﻩcase 'm':
set_flags(tybs_mandline_argument, mem_debug_flags,
ﻩﻩ ﻩ &tybs_mem_debugging);
ﻩ break;
case 'N': /* Deprecated、 */
ﻩcase 'n':
ﻩﻩﻩns_g_cpus = parse_int(tybs_mandline_argument,
ﻩ "number of cpus");
ﻩif (ns_g_cpus == 0)
ﻩ ﻩns_g_cpus = 1;
ﻩ break;
case 'p':
port = parse_int(tybs_mandline_argument, "port");
ﻩ if (port < 1 || port > 65535)
ﻩ ns_main_earlyfatal("port '%s' out of range",
ﻩ ﻩﻩ tybs_mandline_argument);
ﻩ ns_g_port = port;
ﻩ break;
ﻩ/* XXXBEW Should -P be removed? */
ﻩcase 'P':
ﻩ port = parse_int(tybs_mandline_argument, "port");
ﻩif (port < 1 || port > 65535)
ﻩﻩ ns_main_earlyfatal("port '%s' out of range",
ﻩ tybs_mandline_argument);
ﻩﻩﻩlwresd_g_listenport = port;
ﻩ ﻩbreak;
ﻩ case 's':
ﻩﻩ want_stats = TYBS_TRUE;
ﻩﻩbreak;
case 'S':
ﻩ maxsocks = parse_int(tybs_mandline_argument,
ﻩ ﻩﻩ "max number of sockets");
ﻩﻩ break;
ﻩ case 't':
ﻩ ns_g_chrootdir = tybs_mandline_argument;
ﻩ ﻩbreak;
ﻩ case 'T':
ﻩﻩif (strcmp(tybs_mandline_argument, "clienttest") == 0)
ﻩﻩ ﻩns_g_clienttest = TYBS_TRUE;
ﻩ else
ﻩ ﻩ fprintf(stderr, "unknown -T flag '%s\n",
ﻩ ﻩﻩtybs_mandline_argument);
ﻩbreak;
ﻩ case 'u':
ﻩﻩns_g_username = tybs_mandline_argument;
ﻩﻩ break;
ﻩcase 'v':
ﻩﻩ printf("BIND %s\n", ns_g_version);
ﻩﻩexit(0);
ﻩcase 'V':
ﻩﻩprintf("BIND %s built with %s\n", ns_g_version,
ﻩ ns_g_configargs);
ﻩ ﻩexit(0);
ﻩﻩcase '?':
ﻩ usage();
ﻩﻩif (tybs_mandline_option == '?')
ﻩ ﻩﻩexit(0);
ﻩ ns_main_earlyfatal("unknown option '-%c'",
ﻩ tybs_mandline_option);
ﻩdefault:
ns_main_earlyfatal("parsing options returned %d", ch);
}
}
argc -= tybs_mandline_index;
ﻩargv += tybs_mandline_index;
if (argc > 0) {
ﻩﻩusage();
ﻩns_main_earlyfatal("extra mand line arguments");
}
}
static tybs_result_t
create_managers(void) {
ﻩtybs_result_t result;
unsigned int socks;
#ifdef TYBS_PLATFORM_USETHREADS
unsigned int cpus_detected;
#endif
#ifdef TYBS_PLATFORM_USETHREADS
ﻩcpus_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");
#else
ﻩns_g_cpus = 1;
#endif
result = 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 void
destroy_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 void
setup(void) {
tybs_result_t result;
#ifdef HAVE_LIBSCF
char *instance = NULL;
#endif
ns_os_inituserinfo(ns_g_username);
ns_os_tzset();
ﻩns_os_opendevnull();
#ifdef HAVE_LIBSCF
result = ns_smf_get_instance(&instance, 0, ns_g_mctx);
if (result == TYBS_R_SUCCESS)
ﻩ ns_smf_got_instance = 1;
ﻩelse
ﻩns_smf_got_instance = 0;
ﻩif (instance != NULL)
tybs_mem_free(ns_g_mctx, instance);
#endif /* HAVE_LIBSCF */
#ifdef PATH_RANDOMDEV
if (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_create(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);
ﻩ }
}
#endif
ﻩns_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_mand_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_(ns_g_conffile)) {
ﻩﻩresult = tybs_(ns_g_conffile,
ﻩ ﻩﻩ absolute_conffile,
ﻩ ﻩ sizeof(absolute_conffile));
ﻩﻩif (result != TYBS_R_SUCCESS)
ﻩ ns_main_ear
展开阅读全文