source3/utils/net_lua.c

   1 /*
   2  *  Unix SMB/CIFS implementation.
   3  *  Lua experiments
   4  *  Copyright (C) Volker Lendecke 2006
   5  *
   6  *  This program is free software; you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation; either version 3 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20
  21 #include "includes.h"
  22 #include "utils/net.h"
  23
  24 #include "lua-5.1.4/src/lualib.h"
  25 #include "lua-5.1.4/src/lauxlib.h"
  26
  27 #define SOCK_METATABLE "cade1208-9029-4d76-8748-426dfc1436f7"
  28
  29 struct sock_userdata {
  30         int fd;
  31 };
  32
  33 static int sock_userdata_gc(lua_State *L)
  34 {
  35         struct sock_userdata *p = (struct sock_userdata *)
  36                 luaL_checkudata(L, 1, SOCK_METATABLE);
  37         close(p->fd);
  38         return 0;
  39 }
  40
  41 static int sock_userdata_tostring(lua_State *L)
  42 {
  43         struct sock_userdata *p = (struct sock_userdata *)
  44                 luaL_checkudata(L, 1, SOCK_METATABLE);
  45
  46         lua_pushfstring(L, "socket: %d", p->fd);
  47         return 1;
  48 }
  49
  50 static int sock_userdata_connect(lua_State *L)
  51 {
  52         struct sock_userdata *p = (struct sock_userdata *)
  53                 luaL_checkudata(L, 1, SOCK_METATABLE);
  54         const char *hostname;
  55         int port;
  56         struct sockaddr_in addr;
  57         int res;
  58
  59         if (!lua_isstring(L, 2)) {
  60                 luaL_error(L, _("connect: Expected IP-Address"));
  61         }
  62         hostname = lua_tostring(L, 2);
  63
  64         if (!lua_isnumber(L, 3)) {
  65                 luaL_error(L, _("connect: Expected port"));
  66         }
  67         port = lua_tointeger(L, 3);
  68
  69         if (lua_gettop(L) == 4) {
  70                 /*
  71                  * Here we expect an event context in the last argument to
  72                  * make connect() asynchronous.
  73                  */
  74         }
  75
  76         addr.sin_family = AF_INET;
  77         inet_aton(hostname, &addr.sin_addr);
  78         addr.sin_port = htons(port);
  79
  80         res = connect(p->fd, (struct sockaddr *)&addr, sizeof(addr));
  81         if (res == -1) {
  82                 int err = errno;
  83                 lua_pushnil(L);
  84                 lua_pushfstring(L, _("connect failed: %s"), strerror(err));
  85                 return 2;
  86         }
  87
  88         lua_pushboolean(L, 1);
  89         return 1;
  90 }
  91
  92 static const struct luaL_Reg sock_methods[] = {
  93         {"__gc",        sock_userdata_gc},
  94         {"__tostring",  sock_userdata_tostring},
  95         {"connect",     sock_userdata_connect},
  96         {NULL, NULL}
  97 };
  98
  99 static const struct {
 100         const char *name;
 101         int domain;
 102 } socket_domains[] = {
 103         {"PF_UNIX", PF_UNIX},
 104         {"PF_INET", PF_INET},
 105         {NULL, 0},
 106 };
 107
 108 static const struct {
 109         const char *name;
 110         int type;
 111 } socket_types[] = {
 112         {"SOCK_STREAM", SOCK_STREAM},
 113         {"SOCK_DGRAM", SOCK_DGRAM},
 114         {NULL, 0},
 115 };
 116
 117 static int sock_userdata_new(lua_State *L)
 118 {
 119         struct sock_userdata *result;
 120         const char *domain_str = luaL_checkstring(L, 1);
 121         const char *type_str = luaL_checkstring(L, 2);
 122         int i, domain, type;
 123
 124         i = 0;
 125         while (socket_domains[i].name != NULL) {
 126                 if (strcmp(domain_str, socket_domains[i].name) == 0) {
 127                         break;
 128                 }
 129                 i += 1;
 130         }
 131         if (socket_domains[i].name == NULL) {
 132                 return luaL_error(L, _("socket domain %s unknown"), domain_str);
 133         }
 134         domain = socket_domains[i].domain;
 135
 136         i = 0;
 137         while (socket_types[i].name != NULL) {
 138                 if (strcmp(type_str, socket_types[i].name) == 0) {
 139                         break;
 140                 }
 141                 i += 1;
 142         }
 143         if (socket_types[i].name == NULL) {
 144                 return luaL_error(L, _("socket type %s unknown"), type_str);
 145         }
 146         type = socket_types[i].type;
 147
 148         result = (struct sock_userdata *)lua_newuserdata(L, sizeof(*result));
 149         ZERO_STRUCTP(result);
 150
 151         result->fd = socket(domain, type, 0);
 152         if (result->fd == -1) {
 153                 int err = errno;
 154                 lua_pushnil(L);
 155                 lua_pushfstring(L, _("socket() failed: %s"), strerror(errno));
 156                 lua_pushinteger(L, err);
 157                 return 3;
 158         }
 159
 160         luaL_getmetatable(L, SOCK_METATABLE);
 161         lua_setmetatable(L, -2);
 162         return 1;
 163 }
 164
 165 static const struct luaL_Reg sock_funcs[] = {
 166         {"new",         sock_userdata_new},
 167         {NULL, NULL}
 168 };
 169
 170 static int sock_lua_init(lua_State *L, const char *libname) {
 171         luaL_newmetatable(L, SOCK_METATABLE);
 172
 173         lua_pushvalue(L, -1);
 174         lua_setfield(L, -2, "__index");
 175
 176         luaL_register(L, NULL, sock_methods);
 177         luaL_register(L, libname, sock_funcs);
 178         return 1;
 179 }
 180
 181 #define EVT_METATABLE "c42e0642-b24a-40f0-8483-d8eb4aee9ea3"
 182
 183 /*
 184  * The userdata we allocate from lua when a new event context is created
 185  */
 186 struct evt_userdata {
 187         struct event_context *ev;
 188 };
 189
 190 static bool evt_is_main_thread(lua_State *L) {
 191         int ret;
 192
 193         ret = lua_pushthread(L);
 194         lua_pop(L, 1);
 195         return (ret != 0);
 196 }
 197
 198 /*
 199  * Per event we allocate a struct thread_reference to keep the coroutine from
 200  * being garbage-collected. This is also the hook to find the right thread to
 201  * be resumed.
 202  */
 203
 204 struct thread_reference {
 205         struct lua_State *L;
 206         /*
 207          * Reference to the Thread (i.e. lua_State) this event is hanging on
 208          */
 209         int thread_ref;
 210 };
 211
 212 static int thread_reference_destructor(struct thread_reference *ref)
 213 {
 214         luaL_unref(ref->L, LUA_REGISTRYINDEX, ref->thread_ref);
 215         return 0;
 216 }
 217
 218 static struct thread_reference *evt_reference_thread(TALLOC_CTX *mem_ctx,
 219                                                      lua_State *L)
 220 {
 221         struct thread_reference *result;
 222
 223         result = talloc(mem_ctx, struct thread_reference);
 224         if (result == NULL) {
 225                 return NULL;
 226         }
 227
 228         lua_pushthread(L);
 229         result->thread_ref = luaL_ref(L, LUA_REGISTRYINDEX);
 230         result->L = L;
 231         talloc_set_destructor(result, thread_reference_destructor);
 232
 233         return result;
 234 }
 235
 236 static int evt_userdata_gc(lua_State *L)
 237 {
 238         struct evt_userdata *p = (struct evt_userdata *)
 239                 luaL_checkudata(L, 1, EVT_METATABLE);
 240         TALLOC_FREE(p->ev);
 241         return 0;
 242 }
 243
 244 static int evt_userdata_tostring(lua_State *L) {
 245         lua_pushstring(L, "event context");
 246         return 1;
 247 }
 248
 249 static void evt_userdata_sleep_done(struct event_context *event_ctx,
 250                                    struct timed_event *te,
 251                                    struct timeval now,
 252                                    void *priv)
 253 {
 254         struct thread_reference *ref = talloc_get_type_abort(
 255                 priv, struct thread_reference);
 256         lua_resume(ref->L, 0);
 257         TALLOC_FREE(ref);
 258 }
 259
 260 static int evt_userdata_sleep(lua_State *L)
 261 {
 262         struct evt_userdata *p = (struct evt_userdata *)
 263                 luaL_checkudata(L, 1, EVT_METATABLE);
 264         lua_Integer usecs = luaL_checkint(L, 2);
 265         struct thread_reference *ref;
 266         struct timed_event *te;
 267
 268         if (evt_is_main_thread(L)) {
 269                 /*
 270                  * Block in the main thread
 271                  */
 272                 smb_msleep(usecs/1000);
 273                 return 0;
 274         }
 275
 276         ref = evt_reference_thread(p->ev, L);
 277         if (ref == NULL) {
 278                 return luaL_error(L, _("evt_reference_thread failed\n"));
 279         }
 280
 281         te = event_add_timed(p->ev, ref, timeval_current_ofs(0, usecs),
 282                              evt_userdata_sleep_done,
 283                              ref);
 284
 285         if (te == NULL) {
 286                 TALLOC_FREE(ref);
 287                 return luaL_error(L, _("event_add_timed failed"));
 288         }
 289
 290         return lua_yield(L, 0);
 291 }
 292
 293 static int evt_userdata_once(lua_State *L)
 294 {
 295         struct evt_userdata *p = (struct evt_userdata *)
 296                 luaL_checkudata(L, 1, EVT_METATABLE);
 297
 298         if (!evt_is_main_thread(L)) {
 299                 return luaL_error(L,
 300                                   _("event_once called from non-base thread"));
 301         }
 302
 303         lua_pushinteger(L, event_loop_once(p->ev));
 304         return 1;
 305 }
 306
 307 static const struct luaL_Reg evt_methods[] = {
 308         {"__gc",        evt_userdata_gc},
 309         {"__tostring",  evt_userdata_tostring},
 310         {"sleep",       evt_userdata_sleep},
 311         {"once",        evt_userdata_once},
 312         {NULL, NULL}
 313 };
 314
 315 static int evt_userdata_new(lua_State *L) {
 316         struct evt_userdata *result;
 317
 318         result = (struct evt_userdata *)lua_newuserdata(L, sizeof(*result));
 319         ZERO_STRUCTP(result);
 320
 321         result->ev = event_context_init(NULL);
 322         if (result->ev == NULL) {
 323                 return luaL_error(L, _("event_context_init failed"));
 324         }
 325
 326         luaL_getmetatable(L, EVT_METATABLE);
 327         lua_setmetatable(L, -2);
 328         return 1;
 329 }
 330
 331 static const struct luaL_Reg evt_funcs[] = {
 332         {"new",         evt_userdata_new},
 333         {NULL, NULL}
 334 };
 335
 336 static int evt_lua_init(lua_State *L, const char *libname) {
 337         luaL_newmetatable(L, EVT_METATABLE);
 338
 339         lua_pushvalue(L, -1);
 340         lua_setfield(L, -2, "__index");
 341
 342         luaL_register(L, NULL, evt_methods);
 343         luaL_register(L, libname, evt_funcs);
 344         return 1;
 345 }
 346
 347 int net_lua(struct net_context *c, int argc, const char **argv)
 348 {
 349         lua_State *state;
 350
 351         state = lua_open();
 352         if (state == NULL) {
 353                 d_fprintf(stderr, _("lua_newstate failed\n"));
 354                 return -1;
 355         }
 356
 357         luaL_openlibs(state);
 358         evt_lua_init(state, "event");
 359         sock_lua_init(state, "socket");
 360
 361         while (1) {
 362                 char *line = NULL;
 363
 364                 line = smb_readline("lua> ", NULL, NULL);
 365                 if (line == NULL) {
 366                         break;
 367                 }
 368
 369                 if (line[0] == ':') {
 370                         if (luaL_dofile(state, &line[1])) {
 371                                 d_printf(_("luaL_dofile returned an error\n"));
 372                                 continue;
 373                         }
 374                 } else if (line[0] != '\n') {
 375                         if (luaL_dostring(state, line) != 0) {
 376                                 d_printf(_("luaL_dostring returned an "
 377                                            "error\n"));
 378                         }
 379                 }
 380
 381                 SAFE_FREE(line);
 382         }
 383
 384         lua_close(state);
 385         return -1;
 386 }