diff options
Diffstat (limited to 'util/m5/lua_gem5Op.c')
| -rw-r--r-- | util/m5/lua_gem5Op.c | 280 |
1 files changed, 280 insertions, 0 deletions
diff --git a/util/m5/lua_gem5Op.c b/util/m5/lua_gem5Op.c new file mode 100644 index 000000000..598187747 --- /dev/null +++ b/util/m5/lua_gem5Op.c @@ -0,0 +1,280 @@ +/* Copyright (c) 2017 Hanhwi Jang + * All rights reserved. + + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are + * met: redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer; + * redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution; + * neither the name of the copyright holders nor the names of its + * contributors may be used to endorse or promote products derived from + * this software without specific prior written permission. + + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + * + * Authors: Hanhwi Jang + */ + + +#include <assert.h> +#include <lauxlib.h> +#include <lua.h> +#include <lualib.h> +#include <stdlib.h> + +#include <gem5/m5ops.h> + +#include "m5_mmap.h" + +static int +do_arm(lua_State *L) +{ + uint64_t address = lua_tointeger(L, 1); + m5_arm(address); + return 0; +} + +static int +do_quiesce(lua_State *L) +{ + m5_quiesce(); + return 0; +} + +static int +do_quiesce_ns(lua_State *L) +{ + uint64_t ns = lua_tointeger(L, 1); + m5_quiesce_ns(ns); + return 0; +} + +static int +do_quiesce_cycle(lua_State *L) +{ + uint64_t cycles = lua_tointeger(L, 1); + m5_quiesce_cycle(cycles); + return 0; +} + +static int +do_quiesce_time(lua_State *L) +{ + uint64_t ns = m5_quiesce_time(); + lua_pushinteger(L, ns); + return 1; +} + +static int +do_rpns(lua_State *L) +{ + uint64_t ns = m5_rpns(); + lua_pushinteger(L, ns); + return 1; +} + +static int +do_wake_cpu(lua_State *L) +{ + uint64_t cpuid = lua_tointeger(L, 1); + m5_wake_cpu(cpuid); + return 0; +} + +static int +do_exit(lua_State *L) +{ + uint64_t ns_delay = lua_tointeger(L, 1); + m5_exit(ns_delay); + return 0; +} + +static int +do_fail(lua_State *L) +{ + uint64_t ns_delay = lua_tointeger(L, 1); + uint64_t code = lua_tointeger(L, 2); + m5_fail(ns_delay, code); + return 0; +} + +static int +do_init_param(lua_State *L) +{ + uint64_t key_str1 = lua_tointeger(L, 1); + uint64_t key_str2 = lua_tointeger(L, 2); + lua_pushinteger(L, m5_init_param(key_str1, key_str2)); + return 1; +} + +static int +do_checkpoint(lua_State *L) +{ + uint64_t delay = lua_tointeger(L, 1); + uint64_t period = lua_tointeger(L, 2); + m5_checkpoint(delay, period); + return 0; +} + +static int +do_reset_stats(lua_State *L) +{ + uint64_t ns_delay = lua_tointeger(L, 1); + uint64_t ns_period = lua_tointeger(L, 2); + m5_reset_stats(ns_delay, ns_period); + return 0; +} + +static int +do_dump_stats(lua_State *L) +{ + uint64_t delay = lua_tointeger(L, 1); + uint64_t period = lua_tointeger(L, 2); + m5_dump_stats(delay, period); + return 0; +} + +static int +do_dump_reset_stats(lua_State *L) +{ + uint64_t delay = lua_tointeger(L, 1); + uint64_t period = lua_tointeger(L, 2); + m5_dump_reset_stats(delay, period); + return 0; +} + +static int +do_read_file(lua_State *L) +{ + uint64_t len = lua_tointeger(L, 1); + uint64_t offset = lua_tointeger(L, 2); + char *buf = malloc(len); + uint64_t readlen = m5_read_file(buf, len, offset); + lua_pushlstring(L, buf, readlen); + return 1; +} + +static int +do_write_file(lua_State *L) +{ + const char* buf = lua_tostring(L, 1); + uint64_t len = lua_tointeger(L, 2); + assert(len <= lua_strlen(L, 1)); + uint64_t offset = lua_tointeger(L, 3); + const char *filename = lua_tostring(L, 4); + uint64_t w_len = m5_write_file((void *)buf, len, offset, filename); + lua_pushinteger(L, w_len); + return 1; +} + +static int +do_debug_break(lua_State *L) +{ + m5_debug_break(); + return 0; +} + +static int +do_switch_cpu(lua_State *L) +{ + m5_switch_cpu(); + return 0; +} + +static int +do_dist_toggle_sync(lua_State *L) +{ + m5_dist_toggle_sync(); + return 0; +} + +static int +do_add_symbol(lua_State *L) +{ + uint64_t addr = lua_tointeger(L, 1); + char *string = (char*) lua_tostring(L, 2); + m5_add_symbol(addr, string); + return 0; +} + +static int +do_loadsymbol(lua_State *L) +{ + m5_load_symbol(); + return 0; +} + +static int +do_panic(lua_State *L) +{ + m5_panic(); + return 0; +} + +static int +do_work_begin(lua_State *L) +{ + uint64_t workid = lua_tointeger(L, 1); + uint64_t threadid = lua_tointeger(L, 2); + m5_work_begin(workid, threadid); + return 0; +} + +static int +do_work_end(lua_State *L) +{ + uint64_t workid = lua_tointeger(L, 1); + uint64_t threadid = lua_tointeger(L, 2); + m5_work_end(workid, threadid); + return 0; +} + +int +luaopen_gem5OpLua(lua_State *L) +{ + map_m5_mem(); +#define ADD_FUNC(fname) do{ \ + lua_pushcfunction(L, fname); \ + lua_setfield(L, -2, #fname); \ + }while (0) + + lua_newtable(L); + ADD_FUNC(do_arm); + ADD_FUNC(do_quiesce); + ADD_FUNC(do_quiesce_ns); + ADD_FUNC(do_quiesce_cycle); + ADD_FUNC(do_quiesce_time); + ADD_FUNC(do_rpns); + ADD_FUNC(do_wake_cpu); + ADD_FUNC(do_exit); + ADD_FUNC(do_fail); + ADD_FUNC(do_init_param); + ADD_FUNC(do_checkpoint); + ADD_FUNC(do_reset_stats); + ADD_FUNC(do_dump_stats); + ADD_FUNC(do_dump_reset_stats); + ADD_FUNC(do_read_file); + ADD_FUNC(do_write_file); + ADD_FUNC(do_debug_break); + ADD_FUNC(do_switch_cpu); + ADD_FUNC(do_dist_toggle_sync); + ADD_FUNC(do_add_symbol); + ADD_FUNC(do_loadsymbol); + ADD_FUNC(do_panic); + ADD_FUNC(do_work_begin); + ADD_FUNC(do_work_end); +#undef ADD_FUNC + return 1; +} |