path: root/server/src/luautil.cc

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            luautil.cc
 *
 *  Fri Apr 13 14:38:53 CEST 2007
 *  Copyright 2007 Bent Bisballe Nyeng
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *  This file is part of Pracro.
 *
 *  Pracro is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  Pracro is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with Pracro; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
 */
#include "luautil.h"

#include <hugin.hpp>

#include <string>

#include <vector>

std::vector< void * > pointers;

// for ntohl and htonl
#include <arpa/inet.h>

#define GLOBAL_PREFIX "magic_global_"

//
// Get the (somewhat hacky) global pointer to the parser object.
// It is set in the preload method.
//
void *Pracro::getGlobal(lua_State *L, std::string name)
{
  unsigned int top;
  unsigned int index;

  std::string var = std::string(GLOBAL_PREFIX) + name;

  lua_getfield(L, LUA_GLOBALSINDEX, var.c_str()); 
  top = lua_gettop(L);
  index = lua_tointeger(L, top);

  return pointers.at(index);
}

void Pracro::setGlobal(lua_State *L, std::string name, void *p)
{
  // Put the value of this in the globals
  char value_of_this[256];

  std::string val = std::string(GLOBAL_PREFIX) + name;

  /*
  unsigned int parser = (unsigned int)p;
  parser = htonl(parser);
  */

  pointers.push_back(p);
  unsigned int index = pointers.size() - 1;

  sprintf(value_of_this, "%s = %u\n", val.c_str(), index);
  int s = luaL_loadstring(L, value_of_this);
  switch(s) {
  case 0: //no errors;
    break;
  case LUA_ERRSYNTAX: //syntax error during pre-compilation;
  case LUA_ERRMEM: //memory allocation error.
    fprintf(stderr, "Error: %s\n", lua_tostring(L, lua_gettop(L)));
  default:
    fprintf(stderr, "Unknown return value of luaL_loadstring.\n");
  }

  // Run program (init)
  s = lua_pcall(L, 0, LUA_MULTRET, 0);
  // Check for errors
  switch(s) {
  case 0: // Success
    break;
  case LUA_ERRRUN:// a runtime error.
  case LUA_ERRMEM:// memory allocation error.
    // For such errors, Lua does not call the error handler function.
  case LUA_ERRERR:// error while running the error handler function.
    fprintf(stderr, "Error: %s\n", lua_tostring(L, lua_gettop(L)));
    break;
  default:
    fprintf(stderr, "Error: Unknown return value of lua_pcall.\n");
    break;
  }
}

void Pracro::call(lua_State *L, std::string function, int numargs)
{
  // Get function
  lua_getglobal(L, function.c_str());

  // Call it
  int s = lua_pcall(L, numargs, LUA_MULTRET, 0);

  // Check for errors
  switch(s) {
  case 0: // Success
    break;
  case LUA_ERRRUN:// a runtime error.
  case LUA_ERRMEM:// memory allocation error.
    // For such errors, Lua does not call the error handler function.
  case LUA_ERRERR:// error while running the error handler function.
    fprintf(stderr, "Error: %s\n", lua_tostring(L, lua_gettop(L)));
    break;
  default:
    fprintf(stderr, "Error: Unknown return value of lua_pcall.\n");
    break;
  }
}


/*
lua_isboolean
lua_iscfunction
lua_isfunction
lua_islightuserdata
lua_isnil
lua_isnone
lua_isnoneornil
lua_isnumber
lua_isstring
lua_istable
lua_isthread
lua_isuserdata
*/
int Pracro::checkParameters(lua_State *L, ...)
{
  va_list ap;
  va_start(ap, L);

  size_t nargs = lua_gettop(L); // number of arguments

  size_t size = 0;
  int t;
  while(1) { 
    t = va_arg(ap, int);
    if(t == T_END) break;

    switch(t) {
    case T_STRING:
    case T_NUMBER:
    case T_BOOLEAN:
      break;

    default:
      return luaL_error(L,"Unknown type specifier [%d] at position %d. "
                        "Missing TYPE_END?", t, size+1);
    }

    size++;
  }

  va_end(ap);

  if(nargs != size) {
    return luaL_error(L, "Number of args expected %d, got %d", size, nargs);
  }

  va_start(ap, L);

  size_t idx = 0;
  while(1) {
    t = va_arg(ap, int);
    if(t == T_END) break;

    switch(t) {
    case T_STRING:
      if(lua_isstring(L, lua_gettop(L)-(size-idx-1)) == 0) {
        va_end(ap);
        return luaL_error(L, "Parameter %d should be of type string.", idx+1);
      }
      break;

    case T_NUMBER:
      if(lua_isnumber(L, lua_gettop(L)-(size-idx-1)) == 0) {
        va_end(ap);
        return luaL_error(L, "Parameter %d should be of type number.", idx+1);
      }
      break;

    case T_BOOLEAN:
      if(lua_isboolean(L, lua_gettop(L)-(size-idx-1)) == 0) {
        va_end(ap);
        return luaL_error(L, "Parameter %d should be of type boolean.", idx+1);
      }
      break;

    default:
      va_end(ap);
      return luaL_error(L,"Unknown type specifier [%d] at position %d. "
                        "Missing TYPE_END?", t, idx+1);
    }

    idx++;
  }

  va_end(ap);

  return 0;
}

void Pracro::stack(lua_State* l, const char *fmt, ...)
{
  int i;
  int top = lua_gettop(l);

  va_list va;
  va_start(va, fmt);

  printf("---Stack: ");
  vprintf(fmt, va);
  printf("---\nSize: %d\n", top);
  
  for (i = 1; i <= top; i++) {
    int t = lua_type(l, i);
    switch (t) {
    case LUA_TSTRING:
      printf("string: '%s'\n", lua_tostring(l, i));
      break;
    case LUA_TBOOLEAN:
      printf("boolean %s\n",lua_toboolean(l, i) ? "true" : "false");
      break;
    case LUA_TNUMBER:
      printf("number: %g\n", lua_tonumber(l, i));
      break;
    default:
      printf("%s\n", lua_typename(l, t));
      break;
    }
    printf("  ");
  }
  printf("\n");
  fflush(stdout);
}

bool Pracro::hasField(lua_State *L, int i, std::string name)
{
                                             // stack []
  lua_getfield(L, i, name.c_str());          // stack [field|nil]
  bool ret = !lua_isnil(L, lua_gettop(L));   // stack [field|nil]
  lua_pop(L, 1);                             // stack []

  return ret;
}

bool Pracro::testField(lua_State *L, std::vector<std::string> groups,
                       std::string field)
{
  int top = lua_gettop(L);

  groups.push_back(field);

  int grp = LUA_GLOBALSINDEX;
  std::vector<std::string>::iterator gi = groups.begin();
  while(gi != groups.end()) {
    const char *name = gi->c_str();

    if(!hasField(L, grp, name)) {
      while(lua_gettop(L) > top) lua_pop(L, 1);
      return false;
    }
    lua_getfield(L, grp, name);

    grp = lua_gettop(L);
    gi++;
  }

  while(lua_gettop(L) > top) lua_pop(L, 1);

  return true;
}

bool Pracro::getField(lua_State *L, std::vector<std::string> groups,
                      std::string field)
{
  int top = lua_gettop(L);
  groups.push_back(field);

  int grp = LUA_GLOBALSINDEX;
  std::vector<std::string>::iterator gi = groups.begin();
  while(gi != groups.end()) {
    const char *name = gi->c_str();

    if(!hasField(L, grp, name)) {
      while(lua_gettop(L) > top) lua_pop(L, 1);
      return false;
    }
    lua_getfield(L, grp, name);

    grp = lua_gettop(L);
    gi++;
  }

  return true;
}

bool Pracro::createField(lua_State *L, std::vector<std::string> groups,
                         std::string field)
{
  int top = lua_gettop(L);

  int grp = LUA_GLOBALSINDEX;
  std::vector<std::string>::iterator gi = groups.begin();
  while(gi != groups.end()) {
    const char *name = gi->c_str();

    if(!hasField(L, grp, name)) {
      while(lua_gettop(L) > top) lua_pop(L, 1);
      return false;
    }
    lua_getfield(L, grp, name);

    grp = lua_gettop(L);
    gi++;
  }

  lua_newtable(L);
  lua_setfield(L, grp, field.c_str());

  while(lua_gettop(L) > top) lua_pop(L, 1);

  return true;
}

#ifdef TEST_LUAUTIL
//deps:
//cflags: $(LUA_CFLAGS) -I..
//libs: $(LUA_LIBS)
#include "test.h"

#define LUAPROG \
  "testfunc('a string', 42, true)"

#define LUAPROG_BAD1 \
  "testfunc('a string', 42)"

#define LUAPROG_BAD2 \
  "testfunc('a string', 42, true, 'another one')"

#define LUAPROG_BAD3 \
  "testfunc(false, 42, 'string')"

#define LUAPROG_MISSING_END \
  "testfunc_bad('a string', 42, true, 1)"

static int testfunc(lua_State *L)
{
  Pracro::checkParameters(L,
                          Pracro::T_STRING,
                          Pracro::T_NUMBER,
                          Pracro::T_BOOLEAN,
                          Pracro::T_END);
  return 0;
}

static int testfunc_bad(lua_State *L)
{
  Pracro::checkParameters(L,
                          Pracro::T_STRING,
                          Pracro::T_NUMBER,
                          Pracro::T_BOOLEAN,
                          Pracro::T_NUMBER);
  return 0;
}


TEST_BEGIN;

int a, b, c;

lua_State *L;
L = luaL_newstate();
if(L == NULL) TEST_FATAL("Could not allocate lua state.");
luaL_openlibs(L);

Pracro::setGlobal(L, "a", &a);
Pracro::setGlobal(L, "b", &b);
Pracro::setGlobal(L, "c", &c);

TEST_EQUAL(Pracro::getGlobal(L, "b"), &b, "Test get global");
TEST_EQUAL(Pracro::getGlobal(L, "c"), &c, "Test get global");
TEST_EQUAL(Pracro::getGlobal(L, "a"), &a, "Test get global");

lua_register(L, "testfunc", testfunc);

if(luaL_loadstring(L, LUAPROG)) {
  TEST_FATAL(lua_tostring(L, lua_gettop(L)))
}
TEST_EQUAL_INT(lua_pcall(L, 0, LUA_MULTRET, 0), 0, "Good one");

if(luaL_loadstring(L, LUAPROG_BAD1)) {
  TEST_FATAL(lua_tostring(L, lua_gettop(L)))
}
TEST_NOTEQUAL_INT(lua_pcall(L, 0, LUA_MULTRET, 0), 0, "Too short one");

if(luaL_loadstring(L, LUAPROG_BAD2)) {
  TEST_FATAL(lua_tostring(L, lua_gettop(L)))
}
TEST_NOTEQUAL_INT(lua_pcall(L, 0, LUA_MULTRET, 0), 0, "Too long one");

if(luaL_loadstring(L, LUAPROG_BAD3)) {
  TEST_FATAL(lua_tostring(L, lua_gettop(L)))
}
TEST_NOTEQUAL_INT(lua_pcall(L, 0, LUA_MULTRET, 0), 0, "Plain wrong");


lua_register(L, "testfunc_bad", testfunc_bad);
if(luaL_loadstring(L, LUAPROG_MISSING_END)) {
  TEST_FATAL(lua_tostring(L, lua_gettop(L)))
}
TEST_NOTEQUAL_INT(lua_pcall(L, 0, LUA_MULTRET, 0), 0, "Missing T_END");

std::string src = 
  "foo = {}"
  "foo.bar = {}"
  "foo.bar.bas = {}"
  "foo.bar.bas.value = 'hello'"
  "print('done')";

luaL_loadbuffer(L, src.c_str(), src.size(), "src");
lua_pcall(L, 0, LUA_MULTRET, 0);

std::vector<std::string> g;
g.push_back("foo");
g.push_back("bar");
g.push_back("bas");
TEST_TRUE(Pracro::testField(L, g, "value"), "?");

TEST_TRUE(Pracro::createField(L, g, "abe"), "?");

TEST_TRUE(Pracro::testField(L, g, "abe"), "?");

lua_close(L);

TEST_END;

#endif/*TEST_LUAUTIL*/