third_party/upb/tests/bindings/lua/main.c

   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" AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  19  * DISCLAIMED. IN NO EVENT SHALL Google LLC BE LIABLE FOR ANY
  20  * DIRECT, 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 "upb/bindings/lua/upb.h"
  34
  35 lua_State *L;
  36
  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
  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     "upb/bindings/lua/?.lua"
  61   "'";
  62
  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 ||
  73         lua_pcall(L, 1, LUA_MULTRET, 0) ||
  74         luaL_dofile(L, "tests/bindings/lua/test_upb.lua");
  75   signal(SIGINT, SIG_DFL);
  76
  77   if (ret) {
  78     fprintf(stderr, "error testing Lua: %s\n", lua_tostring(L, -1));
  79     ret = 1;
  80   }
  81
  82   lua_close(L);
  83   return ret;
  84 }