156 lines
4.9 KiB
C++
156 lines
4.9 KiB
C++
#include "rust_internal.h"
|
|
|
|
struct
|
|
command_line_args : public kernel_owned<command_line_args>
|
|
{
|
|
rust_kernel *kernel;
|
|
rust_task *task;
|
|
int argc;
|
|
char **argv;
|
|
rust_str **strs;
|
|
|
|
// [str] passed to rust_task::start.
|
|
rust_ivec *args;
|
|
|
|
command_line_args(rust_task *task,
|
|
int sys_argc,
|
|
char **sys_argv)
|
|
: kernel(task->kernel),
|
|
task(task),
|
|
argc(sys_argc),
|
|
argv(sys_argv)
|
|
{
|
|
#if defined(__WIN32__)
|
|
LPCWSTR cmdline = GetCommandLineW();
|
|
LPWSTR *wargv = CommandLineToArgvW(cmdline, &argc);
|
|
kernel->win32_require("CommandLineToArgvW", wargv != NULL);
|
|
argv = (char **) kernel->malloc(sizeof(char*) * argc,
|
|
"win32 command line");
|
|
for (int i = 0; i < argc; ++i) {
|
|
int n_chars = WideCharToMultiByte(CP_UTF8, 0, wargv[i], -1,
|
|
NULL, 0, NULL, NULL);
|
|
kernel->win32_require("WideCharToMultiByte(0)", n_chars != 0);
|
|
argv[i] = (char *) kernel->malloc(n_chars,
|
|
"win32 command line arg");
|
|
n_chars = WideCharToMultiByte(CP_UTF8, 0, wargv[i], -1,
|
|
argv[i], n_chars, NULL, NULL);
|
|
kernel->win32_require("WideCharToMultiByte(1)", n_chars != 0);
|
|
}
|
|
LocalFree(wargv);
|
|
#endif
|
|
size_t vec_fill = sizeof(rust_str *) * argc;
|
|
size_t vec_alloc = next_power_of_two(vec_fill);
|
|
void *mem = kernel->malloc(vec_alloc, "command line");
|
|
strs = (rust_str**) mem;
|
|
for (int i = 0; i < argc; ++i) {
|
|
size_t str_fill = strlen(argv[i]) + 1;
|
|
size_t str_alloc = next_power_of_two(sizeof(rust_str) + str_fill);
|
|
mem = kernel->malloc(str_alloc, "command line arg");
|
|
strs[i] = new (mem) rust_str(str_alloc, str_fill,
|
|
(uint8_t const *)argv[i]);
|
|
strs[i]->ref_count++;
|
|
}
|
|
|
|
size_t ivec_interior_sz =
|
|
sizeof(size_t) * 2 + sizeof(rust_str *) * 4;
|
|
args = (rust_ivec *)
|
|
kernel->malloc(ivec_interior_sz,
|
|
"command line arg interior");
|
|
args->fill = 0;
|
|
size_t ivec_exterior_sz = sizeof(rust_str *) * argc;
|
|
args->alloc = ivec_exterior_sz;
|
|
// NB: _rust_main owns the ivec payload and will be responsible for
|
|
// freeing it
|
|
args->payload.ptr = (rust_ivec_heap *)
|
|
kernel->malloc(ivec_exterior_sz + sizeof(size_t),
|
|
"command line arg exterior");
|
|
args->payload.ptr->fill = ivec_exterior_sz;
|
|
memcpy(&args->payload.ptr->data, strs, ivec_exterior_sz);
|
|
}
|
|
|
|
~command_line_args() {
|
|
kernel->free(args);
|
|
for (int i = 0; i < argc; ++i)
|
|
kernel->free(strs[i]);
|
|
kernel->free(strs);
|
|
|
|
#ifdef __WIN32__
|
|
for (int i = 0; i < argc; ++i) {
|
|
kernel->free(argv[i]);
|
|
}
|
|
kernel->free(argv);
|
|
#endif
|
|
}
|
|
};
|
|
|
|
|
|
/**
|
|
* Main entry point into the Rust runtime. Here we create a Rust service,
|
|
* initialize the kernel, create the root domain and run it.
|
|
*/
|
|
|
|
int check_claims = 0;
|
|
|
|
extern "C" CDECL int
|
|
rust_start(uintptr_t main_fn, int argc, char **argv,
|
|
void* crate_map) {
|
|
|
|
rust_env *env = load_env();
|
|
|
|
update_log_settings(crate_map, env->logspec);
|
|
check_claims = env->check_claims;
|
|
|
|
rust_srv *srv = new rust_srv(env);
|
|
rust_kernel *kernel = new rust_kernel(srv, env->num_sched_threads);
|
|
rust_task_id root_id = kernel->create_task(NULL, "main");
|
|
rust_task *root_task = kernel->get_task_by_id(root_id);
|
|
I(kernel, root_task != NULL);
|
|
rust_scheduler *sched = root_task->sched;
|
|
command_line_args *args
|
|
= new (kernel, "main command line args")
|
|
command_line_args(root_task, argc, argv);
|
|
|
|
DLOG(sched, dom, "startup: %d args in 0x%" PRIxPTR,
|
|
args->argc, (uintptr_t)args->args);
|
|
for (int i = 0; i < args->argc; i++) {
|
|
DLOG(sched, dom, "startup: arg[%d] = '%s'", i, args->argv[i]);
|
|
}
|
|
|
|
root_task->start(main_fn, (uintptr_t)args->args);
|
|
root_task->deref();
|
|
root_task = NULL;
|
|
|
|
int ret = kernel->start_task_threads();
|
|
delete args;
|
|
delete kernel;
|
|
delete srv;
|
|
|
|
free_env(env);
|
|
|
|
#if !defined(__WIN32__)
|
|
// Don't take down the process if the main thread exits without an
|
|
// error.
|
|
if (!ret)
|
|
pthread_exit(NULL);
|
|
#endif
|
|
return ret;
|
|
}
|
|
|
|
// FIXME: Transitional. Please remove.
|
|
extern "C" CDECL int
|
|
rust_start_ivec(uintptr_t main_fn, int argc, char **argv,
|
|
void* crate_map, int main_takes_ivec) {
|
|
return rust_start(main_fn, argc, argv, crate_map);
|
|
}
|
|
|
|
|
|
//
|
|
// Local Variables:
|
|
// mode: C++
|
|
// fill-column: 78;
|
|
// indent-tabs-mode: nil
|
|
// c-basic-offset: 4
|
|
// buffer-file-coding-system: utf-8-unix
|
|
// compile-command: "make -k -C $RBUILD 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
|
|
// End:
|
|
//
|