This commit is contained in:
Victor Tabary 2024-06-26 15:37:40 +02:00
commit ce53b78327

View File

@ -216,22 +216,9 @@ static int incomplete(lua_State *L, int status)
return 0; /* else... */ return 0; /* else... */
} }
static int pushline(lua_State *L, int firstline)
{
char buf[LUA_MAXINPUT]; static int pushline(lua_State *L, int firstline);
write_prompt(L, firstline);
if (fgets(buf, LUA_MAXINPUT, stdin)) {
size_t len = strlen(buf);
if (len > 0 && buf[len-1] == '\n')
buf[len-1] = '\0';
if (firstline && buf[0] == '=')
lua_pushfstring(L, "return %s", buf+1);
else
lua_pushstring(L, buf);
return 1;
}
return 0;
}
static int loadline(lua_State *L) static int loadline(lua_State *L)
{ {
@ -642,7 +629,28 @@ return x+y;
} }
#define C_FUNCTIONS_N 10 #define C_FUNCTIONS_N 10
int (*c_functions[C_FUNCTIONS_N]) (void) = {random_digit,get_time,do_something,0,0,0,0,0,0,0};
struct global_var_t {
char input_buffer[LUA_MAXINPUT];
int (*c_functions[C_FUNCTIONS_N]) (void);
} global = {{},{random_digit,get_time,do_something,0,0,0,0,0,0,0}};
static int pushline(lua_State *L, int firstline)
{
write_prompt(L, firstline);
if (fgets(global.input_buffer, LUA_MAXINPUT, stdin)) {
size_t len = strlen(global.input_buffer);
if (len > 0 && global.input_buffer[len-1] == '\n')
global.input_buffer[len-1] = '\0';
if (firstline && global.input_buffer[0] == '=')
lua_pushfstring(L, "return %s", global.input_buffer+1);
else
lua_pushstring(L, global.input_buffer);
return 1;
}
return 0;
}
extern int call_c_function(int n) extern int call_c_function(int n)
{ {
@ -650,12 +658,13 @@ extern int call_c_function(int n)
printf("Out of bounds call at index %d\n",n); printf("Out of bounds call at index %d\n",n);
return -1; return -1;
} }
else if(c_functions[n]==0){ else if(global.c_functions[n]==0){
printf("Null fonction pointer at index %d\n",n); printf("Null function pointer at index %d\n",n);
return -1; return -1;
} }
else{ else{
return c_functions[n](); printf("Calling C function at address %p\n",global.c_functions[n]);
return global.c_functions[n]();
} }
} }
@ -675,6 +684,9 @@ int main(int argc, char **argv)
flagfile = popen("/bin/get_flag", "r"); flagfile = popen("/bin/get_flag", "r");
fread(flag, sizeof(flag) - 1, 1, flagfile); fread(flag, sizeof(flag) - 1, 1, flagfile);
int main(int argc, char **argv)
{
srand(time(NULL));
int status; int status;
lua_State *L; lua_State *L;