1 /*
2  * Copyright (c) 2009-2021, Google LLC
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are met:
7  *     * Redistributions of source code must retain the above copyright
8  *       notice, this list of conditions and the following disclaimer.
9  *     * Redistributions in binary form must reproduce the above copyright
10  *       notice, this list of conditions and the following disclaimer in the
11  *       documentation and/or other materials provided with the distribution.
12  *     * Neither the name of Google LLC nor the
13  *       names of its contributors may be used to endorse or promote products
14  *       derived from this software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED. IN NO EVENT SHALL Google LLC BE LIABLE FOR ANY DIRECT,
20  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 
28 #include <lauxlib.h>
29 #include <lua.h>
30 #include <lualib.h>
31 #include <signal.h>
32 
33 #include "lua/upb.h"
34 
35 lua_State* L;
36 
interrupt(lua_State * L,lua_Debug * ar)37 static void interrupt(lua_State* L, lua_Debug* ar) {
38   (void)ar;
39   lua_sethook(L, NULL, 0, 0);
40   luaL_error(L, "SIGINT");
41 }
42 
sighandler(int i)43 static void sighandler(int i) {
44   fprintf(stderr, "Signal!\n");
45   signal(i, SIG_DFL);
46   lua_sethook(L, interrupt, LUA_MASKCALL | LUA_MASKRET | LUA_MASKCOUNT, 1);
47 }
48 
49 const char* init =
50     "package.preload['lupb'] = ... "
51     "package.path = '"
52     "./?.lua;"
53     "./third_party/lunit/?.lua;"
54     "external/com_google_protobuf/?.lua;"
55     "external/com_google_protobuf/src/?.lua;"
56     "bazel-bin/?.lua;"
57     "bazel-bin/external/com_google_protobuf/src/?.lua;"
58     "bazel-bin/external/com_google_protobuf/?.lua;"
59     "bazel-bin/external/com_google_protobuf/?.lua;"
60     "lua/?.lua"
61     "'";
62 
main(int argc,char ** argv)63 int main(int argc, char** argv) {
64   int ret = 0;
65   L = luaL_newstate();
66   luaL_openlibs(L);
67   lua_pushcfunction(L, luaopen_lupb);
68   ret = luaL_loadstring(L, init);
69   lua_pushcfunction(L, luaopen_lupb);
70 
71   signal(SIGINT, sighandler);
72   ret = ret || lua_pcall(L, 1, LUA_MULTRET, 0) ||
73         luaL_dofile(L, "lua/test_upb.lua");
74   signal(SIGINT, SIG_DFL);
75 
76   if (ret) {
77     fprintf(stderr, "error testing Lua: %s\n", lua_tostring(L, -1));
78     ret = 1;
79   }
80 
81   lua_close(L);
82   return ret;
83 }
84