char *flatten_engine_names() {
char *names[NELEM(engines)];
int i;
for(i=0; i < NELEM(engines); i++)
names[i] = engines[i].name;
return flatten_argv(NELEM(engines), names, ", ");
}
static
std::auto_ptr< atf::check::check_result >
execute_with_shell(char* const* argv)
{
const std::string cmd = flatten_argv(argv);
const char* sh_argv[4];
sh_argv[0] = atf::config::get("atf_shell").c_str();
sh_argv[1] = "-c";
sh_argv[2] = cmd.c_str();
sh_argv[3] = NULL;
return execute(sh_argv);
}
errno_t uu_proc_setargs( int pid, const char **av, const char **env )
{
errno_t r = 0;
hal_mutex_lock(&proc_lock);
uuprocess_t * u = proc_by_pid(pid);
if( u )
{
if(u->argv) free_argv(u->argv);
if(u->envp) free_argv(u->envp);
u->argv = dup_argv( &u->argc, av, 0 );
u->envp = dup_argv( 0, env, 0 );
flatten_argv( u->cmd, MAX_UU_CMD, u->argv );
}
else
r = ESRCH;
hal_mutex_unlock(&proc_lock);
return r;
}
int main(int argc, char *argv[]) {
static struct option long_options[] = {
{"listen", required_argument, 0, 'l'},
{"port", required_argument, 0, 'p'},
{"help", no_argument, 0, 'h'},
{"verbose", no_argument, 0, 'v'},
{"ping-parent", no_argument, 0, 'x'},
{"gcc-command", required_argument, 0, 'g'},
{"vx32sdk", required_argument, 0, 's'},
{"tmpdir", required_argument, 0, 't'},
{"engine", required_argument, 0, 'e'},
{"no-vx32", no_argument, 0, 'n'},
{0, 0, 0, 0}
};
struct server *server = (struct server*)st_calloc(1, sizeof(struct server));
server->info_handler = &info_handler;
server->quit_handler = &quit_handler;
server->process_multi= &process_multi;
INIT_LIST_HEAD(&server->root);
server->host = "127.0.0.1";
server->port = 22122;
struct config *config = (struct config*)st_calloc(1, sizeof(struct config));
server->userdata = config;
config->tmpdir = "/tmp";
config->vx32sdk_path = "./untrusted/";
config->vx32sdk_gcc_command = strdup(flatten_argv(NELEM(default_vx32sdk_gcc_command), default_vx32sdk_gcc_command, " "));
config->syscall_limit = 4; /* 4 syscalls per request allowed */
int option_index;
int arg;
char *engine_name = NULL;
while((arg = getopt_long_only(argc, argv, "hxvnl:p:g:s:t:e:", long_options, &option_index)) != EOF) {
switch(arg) {
case 'h':
print_help(server, config);
exit(-1);
break;
case 'v':
server->trace = 1;
break;
case 'x':
server->ping_parent = 1;
break;
case 'l':
server->host = optarg;
break;
case 'p':
server->port = atoi(optarg);
if(server->port < 0 || server->port > 65536)
fatal("Port number broken: %i", server->port);
break;
case 'g':
free(config->vx32sdk_gcc_command);
config->vx32sdk_gcc_command = strdup(optarg);
break;
case 's':
config->vx32sdk_path = optarg;
break;
case 't':
config->tmpdir = optarg;
break;
case 'e':
engine_name = optarg;
break;
case 'n':
config->vx32_disabled = 1;
break;
case 0:
default:
fatal("\nUnknown option: \"%s\"\n", argv[optind-1]);
}
}
int i;
storage_engine_create *engine_create = NULL;
storage_engine_destroy *engine_destroy = NULL;
for(i=0; i<NELEM(engines); i++) {
if(engine_name && 0 == strcmp(engine_name, engines[i].name)) {
engine_create = engines[i].create;
engine_destroy = engines[i].destroy;
}
}
if(NULL == engine_create)
fatal("\nYou must specify a storage engine:"
" --engine=[%s]\n", flatten_engine_names() );
log_info("Process pid %i", getpid());
signal(SIGPIPE, SIG_IGN);
commands_initialize();
process_initialize(config);
char *params = flatten_argv(argc-optind, &argv[optind], ", ");
log_info("Loading database engine \"%s\" with parameters \"%s\"", engine_name, params);
config->api = engine_create(argc-optind, &argv[optind]);
do_event_loop(server);
log_info("Quit");
process_destroy();
commands_destroy();
pool_free();
engine_destroy(config->api);
free(config->vx32sdk_gcc_command);
free(config);
free(server);
exit(0);
return(0);
}
