int
main(int argc, char *argv[])
{
int i, j = 0;
int create_flag = 0;
const unsigned char *user = 0, *group = 0, *workdir = 0;
int restart_flag = 0;
char **argv_restart = 0;
int pid;
time_t server_start_time = 0;
hr_set_symbolic_action_table(NEW_SRV_ACTION_LAST, ns_symbolic_action_table, ns_submit_button_labels, 0);
time(&server_start_time);
start_set_self_args(argc, argv);
/* certain options should be removed for restart */
XCALLOC(argv_restart, argc + 2);
argv_restart[j++] = argv[0];
params.program_name = argv[0];
for (i = 1; i < argc; ) {
if (!strcmp(argv[i], "-D")) {
params.daemon_mode_flag = 1;
i++;
} else if (!strcmp(argv[i], "-f")) {
params.force_socket_flag = 1;
argv_restart[j++] = argv[i];
i++;
} else if (!strcmp(argv[i], "--create")) {
create_flag = 1;
i++;
} else if (!strcmp(argv[i], "-u")) {
if (++i >= argc) startup_error("invalid usage");
user = argv[i++];
} else if (!strcmp(argv[i], "-g")) {
if (++i >= argc) startup_error("invalid usage");
group = argv[i++];
} else if (!strcmp(argv[i], "-C")) {
if (++i >= argc) startup_error("invalid usage");
workdir = argv[i++];
} else if (!strcmp(argv[i], "-R")) {
params.restart_mode_flag = 1;
++i;
} else if (!strcmp(argv[i], "--")) {
argv_restart[j++] = argv[i];
i++;
break;
} else if (argv[i][0] == '-') {
startup_error("invalid option `%s'", argv[i]);
} else
break;
}
argv_restart[j++] = "-R";
if (i < argc) {
argv_restart[j++] = argv[i];
ejudge_xml_path = argv[i++];
}
if (i != argc) startup_error("invalid number of parameters");
argv_restart[j] = 0;
start_set_args(argv_restart);
if (!(pid = start_find_process("ej-contests", 0))) {
params.force_socket_flag = 1;
} else if (pid > 0) {
fprintf(stderr, "%s: is already running as pid %d\n", argv[0], pid);
return 1;
}
if (start_prepare(user, group, workdir) < 0) return 1;
if (workdir && *workdir) {
if (chdir(workdir) < 0) {
err("cannot change directory to %s", workdir);
return 1;
}
}
#if defined EJUDGE_XML_PATH
if (!ejudge_xml_path) ejudge_xml_path = EJUDGE_XML_PATH;
#endif /* EJUDGE_XML_PATH */
if (!ejudge_xml_path) startup_error("configuration file is not specified");
ejudge_config = ejudge_cfg_parse(ejudge_xml_path, 0);
if (!ejudge_config) return 1;
if (contests_set_directory(ejudge_config->contests_dir) < 0) return 1;
l10n_prepare(ejudge_config->l10n, ejudge_config->l10n_dir);
if (!strcasecmp(EJUDGE_CHARSET, "UTF-8")) utf8_mode = 1;
#if defined EJUDGE_NEW_SERVER_SOCKET
if (!ejudge_config->new_server_socket)
ejudge_config->new_server_socket = xstrdup(EJUDGE_NEW_SERVER_SOCKET);
#endif
if (!ejudge_config->new_server_socket)
ejudge_config->new_server_socket=xstrdup(EJUDGE_NEW_SERVER_SOCKET_DEFAULT);
#if defined EJUDGE_CONTESTS_HOME_DIR
if (!ejudge_config->contests_home_dir)
ejudge_config->contests_home_dir = xstrdup(EJUDGE_CONTESTS_HOME_DIR);
#endif
setup_log_file();
info("ej-contests %s, compiled %s", compile_version, compile_date);
params.socket_path = ejudge_config->new_server_socket;
params.log_path = ejudge_config->new_server_log;
if (load_plugins() < 0) return 1;
// initialize the default plugin
if (nsdb_default->iface->open(nsdb_default->data) < 0) {
startup_error("default plugin failed to open its connection");
return 1;
}
if (create_flag) {
if (nsdb_default->iface->create(nsdb_default->data) < 0) {
startup_error("database creation failed");
return 1;
}
if (nsdb_default->iface->close(nsdb_default->data) < 0) {
startup_error("database closing failed");
return 1;
}
return 0;
}
if (nsdb_default->iface->check(nsdb_default->data) <= 0) {
startup_error("default plugin failed to check its data");
return 1;
}
if (!(state = nsf_init(¶ms, 0, server_start_time))) return 1;
if (nsf_prepare(state) < 0) return 1;
nsf_main_loop(state);
restart_flag = nsf_is_restart_requested(state);
ns_unload_contests();
nsf_cleanup(state);
nsdb_default->iface->close(nsdb_default->data);
if (restart_flag) start_restart();
return 0;
}
int
main(int argc, char *argv[])
{
int i = 1, r = 0;
const char *command = 0;
const char *ejudge_xml_path = 0;
struct ejudge_cfg *config = 0;
int cmd = 0;
int super_serve_fd = -1;
int signum = 0;
const unsigned char *signame = "";
int pid;
program_name = os_GetBasename(argv[0]);
if (argc < 2) startup_error("not enough parameters");
if (!strcmp(argv[i], "--help")) {
write_help();
} else if (!strcmp(argv[i], "--version")) {
write_version();
}
command = argv[i];
i++;
if (i < argc) {
ejudge_xml_path = argv[i];
i++;
}
if (i < argc) startup_error("too many parameters");
#if defined EJUDGE_XML_PATH
if (!ejudge_xml_path) ejudge_xml_path = EJUDGE_XML_PATH;
#endif /* EJUDGE_XML_PATH */
if (!ejudge_xml_path) startup_error("ejudge.xml path is not specified");
if (!(config = ejudge_cfg_parse(ejudge_xml_path))) return 1;
if (!strcmp(command, "stop")) {
cmd = SSERV_CMD_STOP;
signum = START_STOP;
signame = "TERM";
} else if (!strcmp(command, "restart")) {
cmd = SSERV_CMD_RESTART;
signum = START_RESTART;
signame = "HUP";
} else {
startup_error("invalid command");
}
(void) signame;
(void) signum;
if (!(pid = start_find_process("ej-super-server", 0))) {
op_error("ej-super-server is not running");
} else if (pid > 0) {
/*
fprintf(stderr, "%s: ej-super-server is running as pid %d\n",
program_name, pid);
fprintf(stderr, "%s: sending it the %s signal\n", program_name, signame);
if (start_kill(pid, signum) < 0) op_error("failed: %s", os_ErrorMsg());
return 0;
*/
}
if ((super_serve_fd = super_clnt_open(config->super_serve_socket)) < 0)
op_error("ej-super-server is not running");
r = super_clnt_control(super_serve_fd, cmd);
if (r < 0) op_error("%s", super_proto_strerror(-r));
return 0;
}
static int
load_plugins(void)
{
struct ejudge_plugin_iface *base_iface = 0;
struct nsdb_plugin_iface *nsdb_iface = 0;
struct xml_tree *p;
struct ejudge_plugin *plg, *files_plg;
void *plugin_data = 0;
plugin_set_directory(ejudge_config->plugin_dir);
//ejudge_cfg_unparse_plugins(ejudge_config, stdout);
// find config section for files plugin
for (p = ejudge_config->plugin_list; p; p = p->right) {
files_plg = (struct ejudge_plugin*) p;
if (!strcmp(files_plg->type, "nsdb") && !strcmp(files_plg->name, "files"))
break;
}
if (!p) files_plg = 0;
p = 0;
if (files_plg) p = files_plg->data;
// `files' plugin always loaded
nsdb_plugins_num = 0;
nsdb_plugins[nsdb_plugins_num].iface = &nsdb_plugin_files;
if (!(plugin_data = nsdb_plugin_files.init(ejudge_config))) {
startup_error("cannot initialize files database plugin");
return -1;
}
if (nsdb_plugin_files.parse(plugin_data, ejudge_config, p) < 0) {
startup_error("cannot initialize files database plugin");
return -1;
}
nsdb_plugins[nsdb_plugins_num].data = plugin_data;
nsdb_plugins_num++;
// load other userdb plugins
for (p = ejudge_config->plugin_list; p; p = p->right) {
plg = (struct ejudge_plugin*) p;
if (!plg->load_flag) continue;
if (strcmp(plg->type, "nsdb") != 0) continue;
// `files' plugin is already loaded
if (!strcmp(plg->name, "files")) continue;
if (nsdb_plugins_num == NSDB_PLUGIN_MAX_NUM) {
startup_error("too many userlist database plugins");
return -1;
}
if (!(base_iface = plugin_load(plg->path, plg->type, plg->name))) {
startup_error("cannot load plugin");
return -1;
}
nsdb_iface = (struct nsdb_plugin_iface*) base_iface;
if (nsdb_iface->b.size != sizeof(*nsdb_iface)) {
startup_error("plugin size mismatch");
return -1;
}
if (nsdb_iface->nsdb_version != NSDB_PLUGIN_IFACE_VERSION) {
startup_error("plugin version mismatch");
return -1;
}
if (!(plugin_data = nsdb_iface->init(ejudge_config))) {
startup_error("plugin initialization failed");
return -1;
}
if (nsdb_iface->parse(plugin_data, ejudge_config, plg->data) < 0) {
startup_error("plugin failed to parse its configuration");
return -1;
}
nsdb_plugins[nsdb_plugins_num].iface = nsdb_iface;
nsdb_plugins[nsdb_plugins_num].data = plugin_data;
if (plg->default_flag) {
if (nsdb_default) {
startup_error("more than one plugin is defined as default");
return -1;
}
nsdb_default = &nsdb_plugins[nsdb_plugins_num];
}
nsdb_plugins_num++;
}
if (!nsdb_default) {
info("using files as the new-server database");
nsdb_default = &nsdb_plugins[0];
}
return 0;
}
