legacy-libs/grpc/deps/grpc/third_party/upb/upb/bindings/lua/upb/pb.c

   1 /*
   2 ** require("upb.pb") -- A Lua extension for upb.pb.
   3 **
   4 ** Exposes all the types defined in upb/pb/{*}.h
   5 ** Also defines a few convenience functions on top.
   6 */
   7
   8 #include "upb/bindings/lua/upb.h"
   9 #include "upb/decode.h"
  10 #include "upb/encode.h"
  11
  12 #define LUPB_PBDECODERMETHOD "lupb.pb.decodermethod"
  13
  14 static int lupb_pb_decode(lua_State *L) {
  15   size_t len;
  16   const upb_msglayout *layout;
  17   upb_msg *msg = lupb_msg_checkmsg2(L, 1, &layout);
  18   const char *pb = lua_tolstring(L, 2, &len);
  19
  20   upb_decode(pb, len, msg, layout, lupb_arena_get(L));
  21   /* TODO(haberman): check for error. */
  22
  23   return 0;
  24 }
  25
  26 static int lupb_pb_encode(lua_State *L) {
  27   const upb_msglayout *layout;
  28   const upb_msg *msg = lupb_msg_checkmsg2(L, 1, &layout);
  29   upb_arena *arena = upb_arena_new();
  30   size_t size;
  31   char *result;
  32
  33   result = upb_encode(msg, (const void*)layout, arena, &size);
  34
  35   /* Free resources before we potentially bail on error. */
  36   lua_pushlstring(L, result, size);
  37   upb_arena_free(arena);
  38   /* TODO(haberman): check for error. */
  39
  40   return 1;
  41 }
  42
  43 static const struct luaL_Reg toplevel_m[] = {
  44   {"decode", lupb_pb_decode},
  45   {"encode", lupb_pb_encode},
  46   {NULL, NULL}
  47 };
  48
  49 int luaopen_upb_pb_c(lua_State *L) {
  50   static char module_key;
  51   if (lupb_openlib(L, &module_key, "upb.pb_c", toplevel_m)) {
  52     return 1;
  53   }
  54
  55   return 1;
  56 }