aboutsummaryrefslogtreecommitdiff
path: root/lauxlib.c
blob: d713cd2edeb7c8261484b112864a6ec39aa76ce1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
/*
** $Id: lauxlib.c,v 1.20 1999/10/05 18:33:43 roberto Exp roberto $
** Auxiliary functions for building Lua libraries
** See Copyright Notice in lua.h
*/


#include <stdarg.h>
#include <stdio.h>
#include <string.h>

/* Please Notice: This file uses only the official API of Lua
** Any function declared here could be written as an application function.
** With care, these functions can be used by other libraries.
*/

#define LUA_REENTRANT

#include "lauxlib.h"
#include "lua.h"
#include "luadebug.h"



int luaL_findstring (const char *name, const char *const list[]) {
  int i;
  for (i=0; list[i]; i++)
    if (strcmp(list[i], name) == 0)
      return i;
  return -1;  /* name not found */
}

void luaL_argerror (lua_State *L, int numarg, const char *extramsg) {
  lua_Function f = lua_stackedfunction(L, 0);
  const char *funcname;
  lua_getobjname(L, f, &funcname);
  numarg -= lua_nups(L, f);
  if (funcname == NULL)
    funcname = "?";
  luaL_verror(L, "bad argument #%d to function `%.50s' (%.100s)",
              numarg, funcname, extramsg);
}

static const char *checkstr (lua_State *L, lua_Object o, int n, long *len) {
  const char *s = lua_getstring(L, o);
  luaL_arg_check(L, s, n, "string expected");
  if (len) *len = lua_strlen(L, o);
  return s;
}

const char *luaL_check_lstr (lua_State *L, int n, long *len) {
  return checkstr(L, lua_getparam(L, n), n, len);
}

const char *luaL_opt_lstr (lua_State *L, int n, const char *def, long *len) {
  lua_Object o = lua_getparam(L, n);
  if (o == LUA_NOOBJECT) {
    if (len) *len = def ? strlen(def) : 0;
    return def;
  }
  else return checkstr(L, o, n, len);
}

double luaL_check_number (lua_State *L, int n) {
  lua_Object o = lua_getparam(L, n);
  luaL_arg_check(L, lua_isnumber(L, o), n, "number expected");
  return lua_getnumber(L, o);
}


double luaL_opt_number (lua_State *L, int n, double def) {
  lua_Object o = lua_getparam(L, n);
  if (o == LUA_NOOBJECT) return def;
  else {
    luaL_arg_check(L, lua_isnumber(L, o), n, "number expected");
    return lua_getnumber(L, o);
  }
}


lua_Object luaL_tablearg (lua_State *L, int arg) {
  lua_Object o = lua_getparam(L, arg);
  luaL_arg_check(L, lua_istable(L, o), arg, "table expected");
  return o;
}

lua_Object luaL_functionarg (lua_State *L, int arg) {
  lua_Object o = lua_getparam(L, arg);
  luaL_arg_check(L, lua_isfunction(L, o), arg, "function expected");
  return o;
}

lua_Object luaL_nonnullarg (lua_State *L, int n) {
  lua_Object o = lua_getparam(L, n);
  luaL_arg_check(L, o != LUA_NOOBJECT, n, "value expected");
  return o;
}

void luaL_openlib (lua_State *L, const struct luaL_reg *l, int n) {
  int i;
  for (i=0; i<n; i++)
    lua_register(L, l[i].name, l[i].func);
}


void luaL_verror (lua_State *L, const char *fmt, ...) {
  char buff[500];
  va_list argp;
  va_start(argp, fmt);
  vsprintf(buff, fmt, argp);
  va_end(argp);
  lua_error(L, buff);
}


void luaL_chunkid (char *out, const char *source, int len) {
  len -= 13;  /* 13 = strlen("string ''...\0") */
  if (*source == '@')
    sprintf(out, "file `%.*s'", len, source+1);
  else if (*source == '(')
    strcpy(out, "(C code)");
  else {
    const char *b = strchr(source , '\n');  /* stop string at first new line */
    int lim = (b && (b-source)<len) ? b-source : len;
    sprintf(out, "string `%.*s'", lim, source);
    strcpy(out+lim+(13-5), "...'");  /* 5 = strlen("...'\0") */
  }
}


void luaL_filesource (char *out, const char *filename, int len) {
  if (filename == NULL) filename = "(stdin)";
  sprintf(out, "@%.*s", len-2, filename);  /* -2 for '@' and '\0' */
}