~dzejrou/helenos/lua

« back to all changes in this revision

Viewing changes to uspace/lib/lua/src/lstate.c

  • Committer: Jaroslav Jindrak
  • Date: 2017-07-08 19:46:20 UTC
  • Revision ID: dzejrou@gmail.com-20170708194620-a2qasrlljcqdbkm7
Added source files, adjusted some parts of the code to integrate into HelenOS - names of functions, semantics. Commented out all the inclusions of math.h and included the temporary compatibility header lmath.h.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/*
 
2
** $Id: lstate.c,v 2.133 2015/11/13 12:16:51 roberto Exp $
 
3
** Global State
 
4
** See Copyright Notice in lua.h
 
5
*/
 
6
 
 
7
#define lstate_c
 
8
#define LUA_CORE
 
9
 
 
10
#include "lprefix.h"
 
11
 
 
12
 
 
13
#include <stddef.h>
 
14
#include <string.h>
 
15
 
 
16
#include "lua.h"
 
17
 
 
18
#include "lapi.h"
 
19
#include "ldebug.h"
 
20
#include "ldo.h"
 
21
#include "lfunc.h"
 
22
#include "lgc.h"
 
23
#include "llex.h"
 
24
#include "lmem.h"
 
25
#include "lstate.h"
 
26
#include "lstring.h"
 
27
#include "ltable.h"
 
28
#include "ltm.h"
 
29
 
 
30
 
 
31
#if !defined(LUAI_GCPAUSE)
 
32
#define LUAI_GCPAUSE    200  /* 200% */
 
33
#endif
 
34
 
 
35
#if !defined(LUAI_GCMUL)
 
36
#define LUAI_GCMUL      200 /* GC runs 'twice the speed' of memory allocation */
 
37
#endif
 
38
 
 
39
 
 
40
/*
 
41
** a macro to help the creation of a unique random seed when a state is
 
42
** created; the seed is used to randomize hashes.
 
43
*/
 
44
#if !defined(luai_makeseed)
 
45
#include <time.h>
 
46
#define luai_makeseed()         cast(unsigned int, time(NULL))
 
47
#endif
 
48
 
 
49
 
 
50
 
 
51
/*
 
52
** thread state + extra space
 
53
*/
 
54
typedef struct LX {
 
55
  lu_byte extra_[LUA_EXTRASPACE];
 
56
  lua_State l;
 
57
} LX;
 
58
 
 
59
 
 
60
/*
 
61
** Main thread combines a thread state and the global state
 
62
*/
 
63
typedef struct LG {
 
64
  LX l;
 
65
  global_State g;
 
66
} LG;
 
67
 
 
68
 
 
69
 
 
70
#define fromstate(L)    (cast(LX *, cast(lu_byte *, (L)) - offsetof(LX, l)))
 
71
 
 
72
 
 
73
/*
 
74
** Compute an initial seed as random as possible. Rely on Address Space
 
75
** Layout Randomization (if present) to increase randomness..
 
76
*/
 
77
#define addbuff(b,p,e) \
 
78
  { size_t t = cast(size_t, e); \
 
79
    memcpy(b + p, &t, sizeof(t)); p += sizeof(t); }
 
80
 
 
81
static unsigned int makeseed (lua_State *L) {
 
82
  char buff[4 * sizeof(size_t)];
 
83
  unsigned int h = luai_makeseed();
 
84
  int p = 0;
 
85
  addbuff(buff, p, L);  /* heap variable */
 
86
  addbuff(buff, p, &h);  /* local variable */
 
87
  addbuff(buff, p, luaO_nilobject);  /* global variable */
 
88
  addbuff(buff, p, &lua_newstate);  /* public function */
 
89
  lua_assert(p == sizeof(buff));
 
90
  return luaS_hash(buff, p, h);
 
91
}
 
92
 
 
93
 
 
94
/*
 
95
** set GCdebt to a new value keeping the value (totalbytes + GCdebt)
 
96
** invariant (and avoiding underflows in 'totalbytes')
 
97
*/
 
98
void luaE_setdebt (global_State *g, l_mem debt) {
 
99
  l_mem tb = gettotalbytes(g);
 
100
  lua_assert(tb > 0);
 
101
  if (debt < tb - MAX_LMEM)
 
102
    debt = tb - MAX_LMEM;  /* will make 'totalbytes == MAX_LMEM' */
 
103
  g->totalbytes = tb - debt;
 
104
  g->GCdebt = debt;
 
105
}
 
106
 
 
107
 
 
108
CallInfo *luaE_extendCI (lua_State *L) {
 
109
  CallInfo *ci = luaM_new(L, CallInfo);
 
110
  lua_assert(L->ci->next == NULL);
 
111
  L->ci->next = ci;
 
112
  ci->previous = L->ci;
 
113
  ci->next = NULL;
 
114
  L->nci++;
 
115
  return ci;
 
116
}
 
117
 
 
118
 
 
119
/*
 
120
** free all CallInfo structures not in use by a thread
 
121
*/
 
122
void luaE_freeCI (lua_State *L) {
 
123
  CallInfo *ci = L->ci;
 
124
  CallInfo *next = ci->next;
 
125
  ci->next = NULL;
 
126
  while ((ci = next) != NULL) {
 
127
    next = ci->next;
 
128
    luaM_free(L, ci);
 
129
    L->nci--;
 
130
  }
 
131
}
 
132
 
 
133
 
 
134
/*
 
135
** free half of the CallInfo structures not in use by a thread
 
136
*/
 
137
void luaE_shrinkCI (lua_State *L) {
 
138
  CallInfo *ci = L->ci;
 
139
  CallInfo *next2;  /* next's next */
 
140
  /* while there are two nexts */
 
141
  while (ci->next != NULL && (next2 = ci->next->next) != NULL) {
 
142
    luaM_free(L, ci->next);  /* free next */
 
143
    L->nci--;
 
144
    ci->next = next2;  /* remove 'next' from the list */
 
145
    next2->previous = ci;
 
146
    ci = next2;  /* keep next's next */
 
147
  }
 
148
}
 
149
 
 
150
 
 
151
static void stack_init (lua_State *L1, lua_State *L) {
 
152
  int i; CallInfo *ci;
 
153
  /* initialize stack array */
 
154
  L1->stack = luaM_newvector(L, BASIC_STACK_SIZE, TValue);
 
155
  L1->stacksize = BASIC_STACK_SIZE;
 
156
  for (i = 0; i < BASIC_STACK_SIZE; i++)
 
157
    setnilvalue(L1->stack + i);  /* erase new stack */
 
158
  L1->top = L1->stack;
 
159
  L1->stack_last = L1->stack + L1->stacksize - EXTRA_STACK;
 
160
  /* initialize first ci */
 
161
  ci = &L1->base_ci;
 
162
  ci->next = ci->previous = NULL;
 
163
  ci->callstatus = 0;
 
164
  ci->func = L1->top;
 
165
  setnilvalue(L1->top++);  /* 'function' entry for this 'ci' */
 
166
  ci->top = L1->top + LUA_MINSTACK;
 
167
  L1->ci = ci;
 
168
}
 
169
 
 
170
 
 
171
static void freestack (lua_State *L) {
 
172
  if (L->stack == NULL)
 
173
    return;  /* stack not completely built yet */
 
174
  L->ci = &L->base_ci;  /* free the entire 'ci' list */
 
175
  luaE_freeCI(L);
 
176
  lua_assert(L->nci == 0);
 
177
  luaM_freearray(L, L->stack, L->stacksize);  /* free stack array */
 
178
}
 
179
 
 
180
 
 
181
/*
 
182
** Create registry table and its predefined values
 
183
*/
 
184
static void init_registry (lua_State *L, global_State *g) {
 
185
  TValue temp;
 
186
  /* create registry */
 
187
  Table *registry = luaH_new(L);
 
188
  sethvalue(L, &g->l_registry, registry);
 
189
  luaH_resize(L, registry, LUA_RIDX_LAST, 0);
 
190
  /* registry[LUA_RIDX_MAINTHREAD] = L */
 
191
  setthvalue(L, &temp, L);  /* temp = L */
 
192
  luaH_setint(L, registry, LUA_RIDX_MAINTHREAD, &temp);
 
193
  /* registry[LUA_RIDX_GLOBALS] = table of globals */
 
194
  sethvalue(L, &temp, luaH_new(L));  /* temp = new table (global table) */
 
195
  luaH_setint(L, registry, LUA_RIDX_GLOBALS, &temp);
 
196
}
 
197
 
 
198
 
 
199
/*
 
200
** open parts of the state that may cause memory-allocation errors.
 
201
** ('g->version' != NULL flags that the state was completely build)
 
202
*/
 
203
static void f_luaopen (lua_State *L, void *ud) {
 
204
  global_State *g = G(L);
 
205
  UNUSED(ud);
 
206
  stack_init(L, L);  /* init stack */
 
207
  init_registry(L, g);
 
208
  luaS_init(L);
 
209
  luaT_init(L);
 
210
  luaX_init(L);
 
211
  g->gcrunning = 1;  /* allow gc */
 
212
  g->version = lua_version(NULL);
 
213
  luai_userstateopen(L);
 
214
}
 
215
 
 
216
 
 
217
/*
 
218
** preinitialize a thread with consistent values without allocating
 
219
** any memory (to avoid errors)
 
220
*/
 
221
static void preinit_thread (lua_State *L, global_State *g) {
 
222
  G(L) = g;
 
223
  L->stack = NULL;
 
224
  L->ci = NULL;
 
225
  L->nci = 0;
 
226
  L->stacksize = 0;
 
227
  L->twups = L;  /* thread has no upvalues */
 
228
  L->errorJmp = NULL;
 
229
  L->nCcalls = 0;
 
230
  L->hook = NULL;
 
231
  L->hookmask = 0;
 
232
  L->basehookcount = 0;
 
233
  L->allowhook = 1;
 
234
  resethookcount(L);
 
235
  L->openupval = NULL;
 
236
  L->nny = 1;
 
237
  L->status = LUA_OK;
 
238
  L->errfunc = 0;
 
239
}
 
240
 
 
241
 
 
242
static void close_state (lua_State *L) {
 
243
  global_State *g = G(L);
 
244
  luaF_close(L, L->stack);  /* close all upvalues for this thread */
 
245
  luaC_freeallobjects(L);  /* collect all objects */
 
246
  if (g->version)  /* closing a fully built state? */
 
247
    luai_userstateclose(L);
 
248
  luaM_freearray(L, G(L)->strt.hash, G(L)->strt.size);
 
249
  freestack(L);
 
250
  lua_assert(gettotalbytes(g) == sizeof(LG));
 
251
  (*g->frealloc)(g->ud, fromstate(L), sizeof(LG), 0);  /* free main block */
 
252
}
 
253
 
 
254
 
 
255
LUA_API lua_State *lua_newthread (lua_State *L) {
 
256
  global_State *g = G(L);
 
257
  lua_State *L1;
 
258
  lua_lock(L);
 
259
  luaC_checkGC(L);
 
260
  /* create new thread */
 
261
  L1 = &cast(LX *, luaM_newobject(L, LUA_TTHREAD, sizeof(LX)))->l;
 
262
  L1->marked = luaC_white(g);
 
263
  L1->tt = LUA_TTHREAD;
 
264
  /* link it on list 'allgc' */
 
265
  L1->next = g->allgc;
 
266
  g->allgc = obj2gco(L1);
 
267
  /* anchor it on L stack */
 
268
  setthvalue(L, L->top, L1);
 
269
  api_incr_top(L);
 
270
  preinit_thread(L1, g);
 
271
  L1->hookmask = L->hookmask;
 
272
  L1->basehookcount = L->basehookcount;
 
273
  L1->hook = L->hook;
 
274
  resethookcount(L1);
 
275
  /* initialize L1 extra space */
 
276
  memcpy(lua_getextraspace(L1), lua_getextraspace(g->mainthread),
 
277
         LUA_EXTRASPACE);
 
278
  luai_userstatethread(L, L1);
 
279
  stack_init(L1, L);  /* init stack */
 
280
  lua_unlock(L);
 
281
  return L1;
 
282
}
 
283
 
 
284
 
 
285
void luaE_freethread (lua_State *L, lua_State *L1) {
 
286
  LX *l = fromstate(L1);
 
287
  luaF_close(L1, L1->stack);  /* close all upvalues for this thread */
 
288
  lua_assert(L1->openupval == NULL);
 
289
  luai_userstatefree(L, L1);
 
290
  freestack(L1);
 
291
  luaM_free(L, l);
 
292
}
 
293
 
 
294
 
 
295
LUA_API lua_State *lua_newstate (lua_Alloc f, void *ud) {
 
296
  int i;
 
297
  lua_State *L;
 
298
  global_State *g;
 
299
  LG *l = cast(LG *, (*f)(ud, NULL, LUA_TTHREAD, sizeof(LG)));
 
300
  if (l == NULL) return NULL;
 
301
  L = &l->l.l;
 
302
  g = &l->g;
 
303
  L->next = NULL;
 
304
  L->tt = LUA_TTHREAD;
 
305
  g->currentwhite = bitmask(WHITE0BIT);
 
306
  L->marked = luaC_white(g);
 
307
  preinit_thread(L, g);
 
308
  g->frealloc = f;
 
309
  g->ud = ud;
 
310
  g->mainthread = L;
 
311
  g->seed = makeseed(L);
 
312
  g->gcrunning = 0;  /* no GC while building state */
 
313
  g->GCestimate = 0;
 
314
  g->strt.size = g->strt.nuse = 0;
 
315
  g->strt.hash = NULL;
 
316
  setnilvalue(&g->l_registry);
 
317
  g->panic = NULL;
 
318
  g->version = NULL;
 
319
  g->gcstate = GCSpause;
 
320
  g->gckind = KGC_NORMAL;
 
321
  g->allgc = g->finobj = g->tobefnz = g->fixedgc = NULL;
 
322
  g->sweepgc = NULL;
 
323
  g->gray = g->grayagain = NULL;
 
324
  g->weak = g->ephemeron = g->allweak = NULL;
 
325
  g->twups = NULL;
 
326
  g->totalbytes = sizeof(LG);
 
327
  g->GCdebt = 0;
 
328
  g->gcfinnum = 0;
 
329
  g->gcpause = LUAI_GCPAUSE;
 
330
  g->gcstepmul = LUAI_GCMUL;
 
331
  for (i=0; i < LUA_NUMTAGS; i++) g->mt[i] = NULL;
 
332
  if (luaD_rawrunprotected(L, f_luaopen, NULL) != LUA_OK) {
 
333
    /* memory allocation error: free partial state */
 
334
    close_state(L);
 
335
    L = NULL;
 
336
  }
 
337
  return L;
 
338
}
 
339
 
 
340
 
 
341
LUA_API void lua_close (lua_State *L) {
 
342
  L = G(L)->mainthread;  /* only the main thread can be closed */
 
343
  lua_lock(L);
 
344
  close_state(L);
 
345
}
 
346
 
 
347