162 lines
3.8 KiB
C
162 lines
3.8 KiB
C
/*
|
|
* Copyright (C) 2022-2023 Jordan Bancino <@jordan:bancino.net>
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person
|
|
* obtaining a copy of this software and associated documentation files
|
|
* (the "Software"), to deal in the Software without restriction,
|
|
* including without limitation the rights to use, copy, modify, merge,
|
|
* publish, distribute, sublicense, and/or sell copies of the Software,
|
|
* and to permit persons to whom the Software is furnished to do so,
|
|
* subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be
|
|
* included in all copies or portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
|
|
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
|
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*/
|
|
#include <Runtime.h>
|
|
#include <Array.h>
|
|
#include <HashMap.h>
|
|
#include <Stream.h>
|
|
#include <Log.h>
|
|
#include <Memory.h>
|
|
#include <Str.h>
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <pthread.h>
|
|
|
|
/* Specified by POSIX to contain environment variables */
|
|
extern char **environ;
|
|
|
|
/* The linking program is expected to provide Main */
|
|
extern int Main(Array *, HashMap *);
|
|
|
|
typedef struct MainArgs
|
|
{
|
|
Array *args;
|
|
HashMap *env;
|
|
} MainArgs;
|
|
|
|
static void *
|
|
MainThread(void *argp)
|
|
{
|
|
long ret;
|
|
MainArgs *args = argp;
|
|
|
|
args = argp;
|
|
ret = Main(args->args, args->env);
|
|
|
|
return (void *) ret;
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
pthread_t mainThread;
|
|
size_t i;
|
|
int ret;
|
|
|
|
char *key;
|
|
char *val;
|
|
char **envp;
|
|
|
|
MainArgs args;
|
|
|
|
args.args = NULL;
|
|
args.env = NULL;
|
|
|
|
args.args = ArrayCreate();
|
|
|
|
if (!args.args)
|
|
{
|
|
Log(LOG_ERR, "Bootstrap error: Unable to allocate memory for arguments.");
|
|
ret = EXIT_FAILURE;
|
|
goto finish;
|
|
}
|
|
|
|
args.env = HashMapCreate();
|
|
if (!args.env)
|
|
{
|
|
Log(LOG_ERR, "Bootstrap error: Unable to allocate memory for environment.");
|
|
ret = EXIT_FAILURE;
|
|
goto finish;
|
|
}
|
|
|
|
for (i = 0; i < (size_t) argc; i++)
|
|
{
|
|
ArrayAdd(args.args, StrDuplicate(argv[i]));
|
|
}
|
|
|
|
envp = environ;
|
|
while (*envp)
|
|
{
|
|
size_t valInd;
|
|
|
|
/* It is unclear whether or not envp strings are writable, so
|
|
* we make our own copy to manipulate it */
|
|
key = StrDuplicate(*envp);
|
|
valInd = strcspn(key, "=");
|
|
|
|
key[valInd] = '\0';
|
|
val = key + valInd + 1;
|
|
HashMapSet(args.env, key, StrDuplicate(val));
|
|
Free(key);
|
|
envp++;
|
|
}
|
|
|
|
if (pthread_create(&mainThread, NULL, MainThread, &args) != 0)
|
|
{
|
|
Log(LOG_ERR, "Bootstrap error: Unable to create main thread.");
|
|
ret = EXIT_FAILURE;
|
|
goto finish;
|
|
}
|
|
|
|
if (pthread_join(mainThread, (void **) &ret) != 0)
|
|
{
|
|
/* Should never happen */
|
|
Log(LOG_ERR, "Unable to join main thread.");
|
|
ret = EXIT_FAILURE;
|
|
goto finish;
|
|
}
|
|
|
|
finish:
|
|
if (args.args)
|
|
{
|
|
for (i = 0; i < ArraySize(args.args); i++)
|
|
{
|
|
Free(ArrayGet(args.args, i));
|
|
}
|
|
ArrayFree(args.args);
|
|
}
|
|
|
|
if (args.env)
|
|
{
|
|
while (HashMapIterate(args.env, &key, (void **) &val))
|
|
{
|
|
Free(val);
|
|
}
|
|
HashMapFree(args.env);
|
|
}
|
|
|
|
Log(LOG_DEBUG, "Exitting with code: %d", ret);
|
|
|
|
LogConfigFree(LogConfigGlobal());
|
|
|
|
StreamClose(StreamStdout());
|
|
StreamClose(StreamStdin());
|
|
StreamClose(StreamStderr());
|
|
|
|
GenerateMemoryReport(argv[0]);
|
|
|
|
MemoryFreeAll();
|
|
|
|
return ret;
|
|
}
|