andrius
/
M2-Mesoplanet
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Spawn functions

This commit is contained in:
Jeremiah Orians 2021-11-23 05:58:45 -05:00
parent 9de58d6d9c
commit 668c5b7c4c
No known key found for this signature in database
GPG Key ID: 6B3A3F198708F894
5 changed files with 150 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cc.c
View File

@ -42,6 +42,7 @@ int main(int argc, char** argv, char** envp)
FUZZING = FALSE;
MAX_STRING = 4096;
PREPROCESSOR_MODE = FALSE;
STDIO_USED = FALSE;
int debug_flag = TRUE;
FILE* in = stdin;
FILE* tempfile;
@ -92,7 +93,8 @@ int main(int argc, char** argv, char** envp)
}
else if(match(argv[i], "-o") || match(argv[i], "--output"))
{
destination_file = fopen(argv[i + 1], "w");
destination_name = argv[i + 1];
destination_file = fopen(destination_name, "w");
if(NULL == destination_file)
{
fputs("Unable to open for writing file: ", stderr);

22
cc_env.c
View File

@ -22,9 +22,6 @@
void init_macro_env(char* sym, char* value, char* source, int num);
char* env_lookup(char* variable);
char* Architecture;
int WORDSIZE;
void setup_env()
{
char* ARCH = NULL;
@ -45,32 +42,47 @@ void setup_env()
/* Set desired architecture */
WORDSIZE = 32;
if(match("knight-native", ARCH)) Architecture = "knight-native";
else if(match("knight-posix", ARCH)) Architecture = "knight-posix";
ENDIAN = FALSE;
BASEADDRESS = "0x0";
if(match("knight-native", ARCH))
{
ENDIAN = TRUE;
Architecture = "knight-native";
}
else if(match("knight-posix", ARCH))
{
ENDIAN = TRUE;
Architecture = "knight-posix";
}
else if(match("x86", ARCH))
{
BASEADDRESS = "0x8048000";
Architecture = "x86";
init_macro_env("__i386__", "1", "--architecture", 0);
}
else if(match("amd64", ARCH))
{
BASEADDRESS = "0x00600000";
Architecture = "amd64";
WORDSIZE = 64;
init_macro_env("__x86_64__", "1", "--architecture", 0);
}
else if(match("armv7l", ARCH))
{
BASEADDRESS = "0x10000";
Architecture = "armv7l";
init_macro_env("__arm__", "1", "--architecture", 0);
}
else if(match("aarch64", ARCH))
{
BASEADDRESS = "0x400000";
Architecture = "aarch64";
WORDSIZE = 64;
init_macro_env("__aarch64__", "1", "--architecture", 0);
}
else if(match("riscv64", ARCH))
{
BASEADDRESS = "0x600000";
Architecture = "riscv64";
WORDSIZE = 64;
init_macro_env("__riscv", "1", "--architecture", 0);

5
cc_globals.c
View File

@ -39,6 +39,11 @@ int PREPROCESSOR_MODE;
/* enable spawn behavior to be effective */
char* M2LIBC_PATH;
char* Architecture;
int WORDSIZE;
int ENDIAN;
char* BASEADDRESS;
int STDIO_USED;
/* So we don't shoot ourself in the face */
int FUZZING;

3
cc_globals.h
View File

@ -42,6 +42,9 @@ extern int PREPROCESSOR_MODE;
extern char* M2LIBC_PATH;
extern char* Architecture;
extern int WORDSIZE;
extern int ENDIAN;
extern char* BASEADDRESS;
extern int STDIO_USED;
/* So we don't shoot ourself in the face */
extern int FUZZING;

153
cc_spawn.c
View File

@ -120,6 +120,20 @@ char* find_executable(char* name)
return NULL;
}
void sanity_command_check(char** array)
{
int i = 0;
char* s = array[0];
while(NULL != s)
{
fputs(s, stderr);
fputc(' ', stderr);
i = i + 1;
s = array[i];
}
fputc('\n', stderr);
}
int _execute(char* name, char** array, char** envp)
{
int status; /* i.e. return code */
@ -135,6 +149,8 @@ int _execute(char* name, char** array, char** envp)
exit(EXIT_FAILURE);
}
sanity_command_check(array);
int f = fork();
/* Ensure fork succeeded */
@ -186,15 +202,42 @@ void insert_array(char** array, int index, char* string)
int spawn_hex2(char* input, char* output, char* architecture, char** envp, int debug)
{
/* TODO FINISH */
char* elf_header = calloc(MAX_STRING, sizeof(char));
elf_header = strcat(elf_header, M2LIBC_PATH);
elf_header = strcat(elf_header, "/");
elf_header = strcat(elf_header, architecture);
elf_header = strcat(elf_header, "/ELF-");
elf_header = strcat(elf_header, architecture);
if(debug)
{
elf_header = strcat(elf_header, "-debug.hex2");
}
else
{
elf_header = strcat(elf_header, ".hex2");
}
fputs("# starting hex2 linking\n", stdout);
char** array = calloc(MAX_ARRAY, sizeof(char*));
insert_array(array, 0, "hex2");
insert_array(array, 1, "--file");
insert_array(array, 2, input);
insert_array(array, 3, "--output");
insert_array(array, 4, output);
insert_array(array, 5, "--architecture");
insert_array(array, 6, architecture);
insert_array(array, 2, elf_header);
insert_array(array, 3, "--file");
insert_array(array, 4, input);
insert_array(array, 5, "--output");
insert_array(array, 6, output);
insert_array(array, 7, "--architecture");
insert_array(array, 8, architecture);
insert_array(array, 9, "--base-address");
insert_array(array, 10, BASEADDRESS);
if(ENDIAN)
{
insert_array(array, 11, "--big-endian");
}
else
{
insert_array(array, 11, "--little-endian");
}
int r = _execute("hex2", array, envp);
return r;
}
@ -202,48 +245,89 @@ int spawn_hex2(char* input, char* output, char* architecture, char** envp, int d
int spawn_M1(char* input, char* debug_file, char* output, char* architecture, char** envp, int debug_flag)
{
char** array = calloc(MAX_ARRAY, sizeof(char*));
insert_array(array, 0, "M1");
insert_array(array, 1, "--file");
insert_array(array, 2, input);
if(debug_flag)
fputs("# starting M1 assembly\n", stdout);
char* definitions = calloc(MAX_STRING, sizeof(char));
definitions = strcat(definitions, M2LIBC_PATH);
definitions = strcat(definitions, "/");
definitions = strcat(definitions, architecture);
definitions = strcat(definitions, "/");
definitions = strcat(definitions, architecture);
definitions = strcat(definitions, "_defs.M1");
char* libc = calloc(MAX_STRING, sizeof(char));
libc = strcat(libc, M2LIBC_PATH);
libc = strcat(libc, "/");
libc = strcat(libc, architecture);
if(STDIO_USED)
{
insert_array(array, 3, "--file");
insert_array(array, 4, debug_file);
insert_array(array, 5, "--output");
insert_array(array, 6, output);
insert_array(array, 7, "--architecture");
insert_array(array, 8, architecture);
libc = strcat(libc, "/libc-full.M1");
}
else
{
insert_array(array, 3, "--output");
insert_array(array, 4, output);
insert_array(array, 5, "--architecture");
insert_array(array, 6, architecture);
libc = strcat(libc, "/libc-core.M1");
}
char** array = calloc(MAX_ARRAY, sizeof(char*));
insert_array(array, 0, "M1");
insert_array(array, 1, "--file");
insert_array(array, 2, definitions);
insert_array(array, 3, "--file");
insert_array(array, 4, libc);
insert_array(array, 5, "--file");
insert_array(array, 6, input);
if(ENDIAN)
{
insert_array(array, 7, "--big-endian");
}
else
{
insert_array(array, 7, "--little-endian");
}
insert_array(array, 8, "--architecture");
insert_array(array, 9, architecture);
if(debug_flag)
{
insert_array(array, 10, "--file");
insert_array(array, 11, debug_file);
insert_array(array, 12, "--output");
insert_array(array, 13, output);
}
else
{
insert_array(array, 10, "--output");
insert_array(array, 11, output);
}
int r = _execute("M1", array, envp);
return r;
}
int spawn_blood_elf(char* input, char* output, char* architecture, char** envp, int large_flag)
int spawn_blood_elf(char* input, char* output, char** envp, int large_flag)
{
fputs("# starting Blood-elf stub generation\n", stdout);
char** array = calloc(MAX_ARRAY, sizeof(char*));
insert_array(array, 0, "blood-elf");
insert_array(array, 1, "--file");
insert_array(array, 2, input);
insert_array(array, 3, "--output");
insert_array(array, 4, output);
insert_array(array, 5, "--architecture");
insert_array(array, 6, architecture);
if(large_flag) insert_array(array, 7, "--64");
if(ENDIAN)
{
insert_array(array, 3, "--big-endian");
}
else
{
insert_array(array, 3, "--little-endian");
}
insert_array(array, 4, "--output");
insert_array(array, 5, output);
if(large_flag) insert_array(array, 6, "--64");
int r = _execute("blood-elf", array, envp);
return r;
}
int spawn_M2(char* input, char* output, char* architecture, char** envp, int debug_flag)
{
fputs("# starting M2-Planet build\n", stdout);
char** array = calloc(MAX_ARRAY, sizeof(char*));
insert_array(array, 0, "M2-Planet");
insert_array(array, 1, "--file");
@ -267,7 +351,9 @@ void spawn_processes(int debug_flag, char* preprocessed_file, char* destination,
int i = mkstemp(M2_output);
if(-1 != i)
{
spawn_M2(preprocessed_file, M2_output, Architecture, envp, debug_flag);
close(i);
i = spawn_M2(preprocessed_file, M2_output, Architecture, envp, debug_flag);
if(0 != i) exit(EXIT_FAILURE);
}
else
{
@ -283,7 +369,9 @@ void spawn_processes(int debug_flag, char* preprocessed_file, char* destination,
i = mkstemp(blood_output);
if(-1 != i)
{
spawn_blood_elf(M2_output, blood_output, Architecture, envp, large_flag);
close(i);
i = spawn_blood_elf(M2_output, blood_output, envp, large_flag);
if(0 != i) exit(EXIT_FAILURE);
}
else
{
@ -297,7 +385,9 @@ void spawn_processes(int debug_flag, char* preprocessed_file, char* destination,
i = mkstemp(M1_output);
if(-1 != i)
{
spawn_M1(M2_output, blood_output, M1_output, Architecture, envp, debug_flag);
close(i);
i = spawn_M1(M2_output, blood_output, M1_output, Architecture, envp, debug_flag);
if(0 != i) exit(EXIT_FAILURE);
}
else
{
@ -311,7 +401,8 @@ void spawn_processes(int debug_flag, char* preprocessed_file, char* destination,
if(!match("", blood_output)) remove(blood_output);
/* Build the final binary */
spawn_hex2(M1_output, destination, Architecture, envp, debug_flag);
i = spawn_hex2(M1_output, destination, Architecture, envp, debug_flag);
if(0 != i) exit(EXIT_FAILURE);
/* clean up after ourselves*/
remove(M1_output);