3 ** Copyright (C) 2005-2014 Mike Pall. See Copyright Notice in luajit.h
25 #include "lj_cparse.h"
28 #include "lj_carith.h"
30 #include "lj_ccallback.h"
35 /* -- C type checks ------------------------------------------------------- */
37 /* Check first argument for a C type and returns its ID. */
38 static CTypeID
ffi_checkctype(lua_State
*L
, CTState
*cts
, TValue
*param
)
43 lj_err_argtype(L
, 1, "C type");
45 if (tvisstr(o
)) { /* Parse an abstract C type declaration. */
51 cp
.srcname
= strdata(s
);
54 cp
.mode
= CPARSE_MODE_ABSTRACT
|CPARSE_MODE_NOIMPLICIT
;
55 errcode
= lj_cparse(&cp
);
56 if (errcode
) lj_err_throw(L
, errcode
); /* Propagate errors. */
60 if (!tviscdata(o
)) goto err_argtype
;
61 if (param
&& param
< L
->top
) lj_err_arg(L
, 1, LJ_ERR_FFI_NUMPARAM
);
63 return cd
->ctypeid
== CTID_CTYPEID
? *(CTypeID
*)cdataptr(cd
) : cd
->ctypeid
;
67 /* Check argument for C data and return it. */
68 static GCcdata
*ffi_checkcdata(lua_State
*L
, int narg
)
70 TValue
*o
= L
->base
+ narg
-1;
71 if (!(o
< L
->top
&& tviscdata(o
)))
72 lj_err_argt(L
, narg
, LUA_TCDATA
);
76 /* Convert argument to C pointer. */
77 static void *ffi_checkptr(lua_State
*L
, int narg
, CTypeID id
)
79 CTState
*cts
= ctype_cts(L
);
80 TValue
*o
= L
->base
+ narg
-1;
83 lj_err_arg(L
, narg
, LJ_ERR_NOVAL
);
84 lj_cconv_ct_tv(cts
, ctype_get(cts
, id
), (uint8_t *)&p
, o
, CCF_ARG(narg
));
88 /* Convert argument to int32_t. */
89 static int32_t ffi_checkint(lua_State
*L
, int narg
)
91 CTState
*cts
= ctype_cts(L
);
92 TValue
*o
= L
->base
+ narg
-1;
95 lj_err_arg(L
, narg
, LJ_ERR_NOVAL
);
96 lj_cconv_ct_tv(cts
, ctype_get(cts
, CTID_INT32
), (uint8_t *)&i
, o
,
101 /* -- C type metamethods -------------------------------------------------- */
103 #define LJLIB_MODULE_ffi_meta
105 /* Handle ctype __index/__newindex metamethods. */
106 static int ffi_index_meta(lua_State
*L
, CTState
*cts
, CType
*ct
, MMS mm
)
108 CTypeID id
= ctype_typeid(cts
, ct
);
109 cTValue
*tv
= lj_ctype_meta(cts
, id
, mm
);
110 TValue
*base
= L
->base
;
114 s
= strdata(lj_ctype_repr(L
, id
, NULL
));
115 if (tvisstr(L
->base
+1)) {
116 lj_err_callerv(L
, LJ_ERR_FFI_BADMEMBER
, s
, strVdata(L
->base
+1));
118 const char *key
= tviscdata(L
->base
+1) ?
119 strdata(lj_ctype_repr(L
, cdataV(L
->base
+1)->ctypeid
, NULL
)) :
120 lj_typename(L
->base
+1);
121 lj_err_callerv(L
, LJ_ERR_FFI_BADIDXW
, s
, key
);
125 if (mm
== MM_index
) {
126 cTValue
*o
= lj_meta_tget(L
, tv
, base
+1);
128 if (tvisnil(o
)) goto err_index
;
129 copyTV(L
, L
->top
-1, o
);
133 TValue
*o
= lj_meta_tset(L
, tv
, base
+1);
135 copyTV(L
, o
, base
+2);
139 copyTV(L
, base
, L
->top
);
142 return lj_meta_tailcall(L
, tv
);
145 LJLIB_CF(ffi_meta___index
) LJLIB_REC(cdata_index
0)
147 CTState
*cts
= ctype_cts(L
);
152 if (!(o
+1 < L
->top
&& tviscdata(o
))) /* Also checks for presence of key. */
153 lj_err_argt(L
, 1, LUA_TCDATA
);
154 ct
= lj_cdata_index(cts
, cdataV(o
), o
+1, &p
, &qual
);
156 return ffi_index_meta(L
, cts
, ct
, MM_index
);
157 if (lj_cdata_get(cts
, ct
, L
->top
-1, p
))
162 LJLIB_CF(ffi_meta___newindex
) LJLIB_REC(cdata_index
1)
164 CTState
*cts
= ctype_cts(L
);
169 if (!(o
+2 < L
->top
&& tviscdata(o
))) /* Also checks for key and value. */
170 lj_err_argt(L
, 1, LUA_TCDATA
);
171 ct
= lj_cdata_index(cts
, cdataV(o
), o
+1, &p
, &qual
);
173 if ((qual
& CTF_CONST
))
174 lj_err_caller(L
, LJ_ERR_FFI_WRCONST
);
175 return ffi_index_meta(L
, cts
, ct
, MM_newindex
);
177 lj_cdata_set(cts
, ct
, p
, o
+2, qual
);
181 /* Common handler for cdata arithmetic. */
182 static int ffi_arith(lua_State
*L
)
184 MMS mm
= (MMS
)(curr_func(L
)->c
.ffid
- (int)FF_ffi_meta___eq
+ (int)MM_eq
);
185 return lj_carith_op(L
, mm
);
188 /* The following functions must be in contiguous ORDER MM. */
189 LJLIB_CF(ffi_meta___eq
) LJLIB_REC(cdata_arith MM_eq
)
194 LJLIB_CF(ffi_meta___len
) LJLIB_REC(cdata_arith MM_len
)
199 LJLIB_CF(ffi_meta___lt
) LJLIB_REC(cdata_arith MM_lt
)
204 LJLIB_CF(ffi_meta___le
) LJLIB_REC(cdata_arith MM_le
)
209 LJLIB_CF(ffi_meta___concat
) LJLIB_REC(cdata_arith MM_concat
)
214 /* Forward declaration. */
215 static int lj_cf_ffi_new(lua_State
*L
);
217 LJLIB_CF(ffi_meta___call
) LJLIB_REC(cdata_call
)
219 CTState
*cts
= ctype_cts(L
);
220 GCcdata
*cd
= ffi_checkcdata(L
, 1);
221 CTypeID id
= cd
->ctypeid
;
225 if (cd
->ctypeid
== CTID_CTYPEID
) {
226 id
= *(CTypeID
*)cdataptr(cd
);
229 int ret
= lj_ccall_func(L
, cd
);
233 /* Handle ctype __call/__new metamethod. */
234 ct
= ctype_raw(cts
, id
);
235 if (ctype_isptr(ct
->info
)) id
= ctype_cid(ct
->info
);
236 tv
= lj_ctype_meta(cts
, id
, mm
);
238 return lj_meta_tailcall(L
, tv
);
239 else if (mm
== MM_call
)
240 lj_err_callerv(L
, LJ_ERR_FFI_BADCALL
, strdata(lj_ctype_repr(L
, id
, NULL
)));
241 return lj_cf_ffi_new(L
);
244 LJLIB_CF(ffi_meta___add
) LJLIB_REC(cdata_arith MM_add
)
249 LJLIB_CF(ffi_meta___sub
) LJLIB_REC(cdata_arith MM_sub
)
254 LJLIB_CF(ffi_meta___mul
) LJLIB_REC(cdata_arith MM_mul
)
259 LJLIB_CF(ffi_meta___div
) LJLIB_REC(cdata_arith MM_div
)
264 LJLIB_CF(ffi_meta___mod
) LJLIB_REC(cdata_arith MM_mod
)
269 LJLIB_CF(ffi_meta___pow
) LJLIB_REC(cdata_arith MM_pow
)
274 LJLIB_CF(ffi_meta___unm
) LJLIB_REC(cdata_arith MM_unm
)
278 /* End of contiguous ORDER MM. */
280 LJLIB_CF(ffi_meta___tostring
)
282 GCcdata
*cd
= ffi_checkcdata(L
, 1);
283 const char *msg
= "cdata<%s>: %p";
284 CTypeID id
= cd
->ctypeid
;
285 void *p
= cdataptr(cd
);
286 if (id
== CTID_CTYPEID
) {
290 CTState
*cts
= ctype_cts(L
);
291 CType
*ct
= ctype_raw(cts
, id
);
292 if (ctype_isref(ct
->info
)) {
294 ct
= ctype_rawchild(cts
, ct
);
296 if (ctype_iscomplex(ct
->info
)) {
297 setstrV(L
, L
->top
-1, lj_ctype_repr_complex(L
, cdataptr(cd
), ct
->size
));
299 } else if (ct
->size
== 8 && ctype_isinteger(ct
->info
)) {
300 setstrV(L
, L
->top
-1, lj_ctype_repr_int64(L
, *(uint64_t *)cdataptr(cd
),
301 (ct
->info
& CTF_UNSIGNED
)));
303 } else if (ctype_isfunc(ct
->info
)) {
305 } else if (ctype_isenum(ct
->info
)) {
306 msg
= "cdata<%s>: %d";
307 p
= (void *)(uintptr_t)*(uint32_t **)p
;
309 if (ctype_isptr(ct
->info
)) {
310 p
= cdata_getptr(p
, ct
->size
);
311 ct
= ctype_rawchild(cts
, ct
);
313 if (ctype_isstruct(ct
->info
) || ctype_isvector(ct
->info
)) {
314 /* Handle ctype __tostring metamethod. */
315 cTValue
*tv
= lj_ctype_meta(cts
, ctype_typeid(cts
, ct
), MM_tostring
);
317 return lj_meta_tailcall(L
, tv
);
321 lj_str_pushf(L
, msg
, strdata(lj_ctype_repr(L
, id
, NULL
)), p
);
327 static int ffi_pairs(lua_State
*L
, MMS mm
)
329 CTState
*cts
= ctype_cts(L
);
330 CTypeID id
= ffi_checkcdata(L
, 1)->ctypeid
;
331 CType
*ct
= ctype_raw(cts
, id
);
333 if (ctype_isptr(ct
->info
)) id
= ctype_cid(ct
->info
);
334 tv
= lj_ctype_meta(cts
, id
, mm
);
336 lj_err_callerv(L
, LJ_ERR_FFI_BADMM
, strdata(lj_ctype_repr(L
, id
, NULL
)),
337 strdata(mmname_str(G(L
), mm
)));
338 return lj_meta_tailcall(L
, tv
);
341 LJLIB_CF(ffi_meta___pairs
)
343 return ffi_pairs(L
, MM_pairs
);
346 LJLIB_CF(ffi_meta___ipairs
)
348 return ffi_pairs(L
, MM_ipairs
);
351 LJLIB_PUSH("ffi") LJLIB_SET(__metatable
)
353 #include "lj_libdef.h"
355 /* -- C library metamethods ----------------------------------------------- */
357 #define LJLIB_MODULE_ffi_clib
359 /* Index C library by a name. */
360 static TValue
*ffi_clib_index(lua_State
*L
)
364 if (!(o
< L
->top
&& tvisudata(o
) && udataV(o
)->udtype
== UDTYPE_FFI_CLIB
))
365 lj_err_argt(L
, 1, LUA_TUSERDATA
);
366 cl
= (CLibrary
*)uddata(udataV(o
));
367 if (!(o
+1 < L
->top
&& tvisstr(o
+1)))
368 lj_err_argt(L
, 2, LUA_TSTRING
);
369 return lj_clib_index(L
, cl
, strV(o
+1));
372 LJLIB_CF(ffi_clib___index
) LJLIB_REC(clib_index
1)
374 TValue
*tv
= ffi_clib_index(L
);
376 CTState
*cts
= ctype_cts(L
);
377 GCcdata
*cd
= cdataV(tv
);
378 CType
*s
= ctype_get(cts
, cd
->ctypeid
);
379 if (ctype_isextern(s
->info
)) {
380 CTypeID sid
= ctype_cid(s
->info
);
381 void *sp
= *(void **)cdataptr(cd
);
382 CType
*ct
= ctype_raw(cts
, sid
);
383 if (lj_cconv_tv_ct(cts
, ct
, sid
, L
->top
-1, sp
))
388 copyTV(L
, L
->top
-1, tv
);
392 LJLIB_CF(ffi_clib___newindex
) LJLIB_REC(clib_index
0)
394 TValue
*tv
= ffi_clib_index(L
);
395 TValue
*o
= L
->base
+2;
396 if (o
< L
->top
&& tviscdata(tv
)) {
397 CTState
*cts
= ctype_cts(L
);
398 GCcdata
*cd
= cdataV(tv
);
399 CType
*d
= ctype_get(cts
, cd
->ctypeid
);
400 if (ctype_isextern(d
->info
)) {
402 for (;;) { /* Skip attributes and collect qualifiers. */
403 d
= ctype_child(cts
, d
);
404 if (!ctype_isattrib(d
->info
)) break;
405 if (ctype_attrib(d
->info
) == CTA_QUAL
) qual
|= d
->size
;
407 if (!((d
->info
|qual
) & CTF_CONST
)) {
408 lj_cconv_ct_tv(cts
, d
, *(void **)cdataptr(cd
), o
, 0);
413 lj_err_caller(L
, LJ_ERR_FFI_WRCONST
);
414 return 0; /* unreachable */
417 LJLIB_CF(ffi_clib___gc
)
420 if (o
< L
->top
&& tvisudata(o
) && udataV(o
)->udtype
== UDTYPE_FFI_CLIB
)
421 lj_clib_unload((CLibrary
*)uddata(udataV(o
)));
425 #include "lj_libdef.h"
427 /* -- Callback function metamethods --------------------------------------- */
429 #define LJLIB_MODULE_ffi_callback
431 static int ffi_callback_set(lua_State
*L
, GCfunc
*fn
)
433 GCcdata
*cd
= ffi_checkcdata(L
, 1);
434 CTState
*cts
= ctype_cts(L
);
435 CType
*ct
= ctype_raw(cts
, cd
->ctypeid
);
436 if (ctype_isptr(ct
->info
) && (LJ_32
|| ct
->size
== 8)) {
437 MSize slot
= lj_ccallback_ptr2slot(cts
, *(void **)cdataptr(cd
));
438 if (slot
< cts
->cb
.sizeid
&& cts
->cb
.cbid
[slot
] != 0) {
439 GCtab
*t
= cts
->miscmap
;
440 TValue
*tv
= lj_tab_setint(L
, t
, (int32_t)slot
);
443 lj_gc_anybarriert(L
, t
);
446 cts
->cb
.cbid
[slot
] = 0;
447 cts
->cb
.topid
= slot
< cts
->cb
.topid
? slot
: cts
->cb
.topid
;
452 lj_err_caller(L
, LJ_ERR_FFI_BADCBACK
);
456 LJLIB_CF(ffi_callback_free
)
458 return ffi_callback_set(L
, NULL
);
461 LJLIB_CF(ffi_callback_set
)
463 GCfunc
*fn
= lj_lib_checkfunc(L
, 2);
464 return ffi_callback_set(L
, fn
);
467 LJLIB_PUSH(top
-1) LJLIB_SET(__index
)
469 #include "lj_libdef.h"
471 /* -- FFI library functions ----------------------------------------------- */
473 #define LJLIB_MODULE_ffi
477 GCstr
*s
= lj_lib_checkstr(L
, 1);
481 cp
.cts
= ctype_cts(L
);
482 cp
.srcname
= strdata(s
);
484 cp
.param
= L
->base
+1;
485 cp
.mode
= CPARSE_MODE_MULTI
|CPARSE_MODE_DIRECT
;
486 errcode
= lj_cparse(&cp
);
487 if (errcode
) lj_err_throw(L
, errcode
); /* Propagate errors. */
492 LJLIB_CF(ffi_new
) LJLIB_REC(.)
494 CTState
*cts
= ctype_cts(L
);
495 CTypeID id
= ffi_checkctype(L
, cts
, NULL
);
496 CType
*ct
= ctype_raw(cts
, id
);
498 CTInfo info
= lj_ctype_info(cts
, id
, &sz
);
499 TValue
*o
= L
->base
+1;
501 if ((info
& CTF_VLA
)) {
503 sz
= lj_ctype_vlsize(cts
, ct
, (CTSize
)ffi_checkint(L
, 2));
505 if (sz
== CTSIZE_INVALID
)
506 lj_err_arg(L
, 1, LJ_ERR_FFI_INVSIZE
);
507 if (!(info
& CTF_VLA
) && ctype_align(info
) <= CT_MEMALIGN
)
508 cd
= lj_cdata_new(cts
, id
, sz
);
510 cd
= lj_cdata_newv(cts
, id
, sz
, ctype_align(info
));
511 setcdataV(L
, o
-1, cd
); /* Anchor the uninitialized cdata. */
512 lj_cconv_ct_init(cts
, ct
, sz
, cdataptr(cd
),
513 o
, (MSize
)(L
->top
- o
)); /* Initialize cdata. */
514 if (ctype_isstruct(ct
->info
)) {
515 /* Handle ctype __gc metamethod. Use the fast lookup here. */
516 cTValue
*tv
= lj_tab_getinth(cts
->miscmap
, -(int32_t)id
);
517 if (tv
&& tvistab(tv
) && (tv
= lj_meta_fast(L
, tabV(tv
), MM_gc
))) {
518 GCtab
*t
= cts
->finalizer
;
519 if (gcref(t
->metatable
)) {
520 /* Add to finalizer table, if still enabled. */
521 copyTV(L
, lj_tab_set(L
, t
, o
-1), tv
);
522 lj_gc_anybarriert(L
, t
);
523 cd
->marked
|= LJ_GC_CDATA_FIN
;
527 L
->top
= o
; /* Only return the cdata itself. */
532 LJLIB_CF(ffi_cast
) LJLIB_REC(ffi_new
)
534 CTState
*cts
= ctype_cts(L
);
535 CTypeID id
= ffi_checkctype(L
, cts
, NULL
);
536 CType
*d
= ctype_raw(cts
, id
);
537 TValue
*o
= lj_lib_checkany(L
, 2);
538 L
->top
= o
+1; /* Make sure this is the last item on the stack. */
539 if (!(ctype_isnum(d
->info
) || ctype_isptr(d
->info
) || ctype_isenum(d
->info
)))
540 lj_err_arg(L
, 1, LJ_ERR_FFI_INVTYPE
);
541 if (!(tviscdata(o
) && cdataV(o
)->ctypeid
== id
)) {
542 GCcdata
*cd
= lj_cdata_new(cts
, id
, d
->size
);
543 lj_cconv_ct_tv(cts
, d
, cdataptr(cd
), o
, CCF_CAST
);
550 LJLIB_CF(ffi_typeof
) LJLIB_REC(.)
552 CTState
*cts
= ctype_cts(L
);
553 CTypeID id
= ffi_checkctype(L
, cts
, L
->base
+1);
554 GCcdata
*cd
= lj_cdata_new(cts
, CTID_CTYPEID
, 4);
555 *(CTypeID
*)cdataptr(cd
) = id
;
556 setcdataV(L
, L
->top
-1, cd
);
561 LJLIB_CF(ffi_istype
) LJLIB_REC(.)
563 CTState
*cts
= ctype_cts(L
);
564 CTypeID id1
= ffi_checkctype(L
, cts
, NULL
);
565 TValue
*o
= lj_lib_checkany(L
, 2);
568 GCcdata
*cd
= cdataV(o
);
569 CTypeID id2
= cd
->ctypeid
== CTID_CTYPEID
? *(CTypeID
*)cdataptr(cd
) :
571 CType
*ct1
= lj_ctype_rawref(cts
, id1
);
572 CType
*ct2
= lj_ctype_rawref(cts
, id2
);
575 } else if (ctype_type(ct1
->info
) == ctype_type(ct2
->info
) &&
576 ct1
->size
== ct2
->size
) {
577 if (ctype_ispointer(ct1
->info
))
578 b
= lj_cconv_compatptr(cts
, ct1
, ct2
, CCF_IGNQUAL
);
579 else if (ctype_isnum(ct1
->info
) || ctype_isvoid(ct1
->info
))
580 b
= (((ct1
->info
^ ct2
->info
) & ~(CTF_QUAL
|CTF_LONG
)) == 0);
581 } else if (ctype_isstruct(ct1
->info
) && ctype_isptr(ct2
->info
) &&
582 ct1
== ctype_rawchild(cts
, ct2
)) {
586 setboolV(L
->top
-1, b
);
587 setboolV(&G(L
)->tmptv2
, b
); /* Remember for trace recorder. */
591 LJLIB_CF(ffi_sizeof
) LJLIB_REC(ffi_xof FF_ffi_sizeof
)
593 CTState
*cts
= ctype_cts(L
);
594 CTypeID id
= ffi_checkctype(L
, cts
, NULL
);
596 if (LJ_UNLIKELY(tviscdata(L
->base
) && cdataisv(cdataV(L
->base
)))) {
597 sz
= cdatavlen(cdataV(L
->base
));
599 CType
*ct
= lj_ctype_rawref(cts
, id
);
600 if (ctype_isvltype(ct
->info
))
601 sz
= lj_ctype_vlsize(cts
, ct
, (CTSize
)ffi_checkint(L
, 2));
603 sz
= ctype_hassize(ct
->info
) ? ct
->size
: CTSIZE_INVALID
;
604 if (LJ_UNLIKELY(sz
== CTSIZE_INVALID
)) {
609 setintV(L
->top
-1, (int32_t)sz
);
613 LJLIB_CF(ffi_alignof
) LJLIB_REC(ffi_xof FF_ffi_alignof
)
615 CTState
*cts
= ctype_cts(L
);
616 CTypeID id
= ffi_checkctype(L
, cts
, NULL
);
618 CTInfo info
= lj_ctype_info(cts
, id
, &sz
);
619 setintV(L
->top
-1, 1 << ctype_align(info
));
623 LJLIB_CF(ffi_offsetof
) LJLIB_REC(ffi_xof FF_ffi_offsetof
)
625 CTState
*cts
= ctype_cts(L
);
626 CTypeID id
= ffi_checkctype(L
, cts
, NULL
);
627 GCstr
*name
= lj_lib_checkstr(L
, 2);
628 CType
*ct
= lj_ctype_rawref(cts
, id
);
630 if (ctype_isstruct(ct
->info
) && ct
->size
!= CTSIZE_INVALID
) {
631 CType
*fct
= lj_ctype_getfield(cts
, ct
, name
, &ofs
);
633 setintV(L
->top
-1, ofs
);
634 if (ctype_isfield(fct
->info
)) {
636 } else if (ctype_isbitfield(fct
->info
)) {
637 setintV(L
->top
++, ctype_bitpos(fct
->info
));
638 setintV(L
->top
++, ctype_bitbsz(fct
->info
));
646 LJLIB_CF(ffi_errno
) LJLIB_REC(.)
649 if (L
->top
> L
->base
)
650 errno
= ffi_checkint(L
, 1);
651 setintV(L
->top
++, err
);
655 LJLIB_CF(ffi_string
) LJLIB_REC(.)
657 CTState
*cts
= ctype_cts(L
);
658 TValue
*o
= lj_lib_checkany(L
, 1);
661 if (o
+1 < L
->top
&& !tvisnil(o
+1)) {
662 len
= (size_t)ffi_checkint(L
, 2);
663 lj_cconv_ct_tv(cts
, ctype_get(cts
, CTID_P_CVOID
), (uint8_t *)&p
, o
,
666 lj_cconv_ct_tv(cts
, ctype_get(cts
, CTID_P_CCHAR
), (uint8_t *)&p
, o
,
670 L
->top
= o
+1; /* Make sure this is the last item on the stack. */
671 setstrV(L
, o
, lj_str_new(L
, p
, len
));
676 LJLIB_CF(ffi_copy
) LJLIB_REC(.)
678 void *dp
= ffi_checkptr(L
, 1, CTID_P_VOID
);
679 void *sp
= ffi_checkptr(L
, 2, CTID_P_CVOID
);
680 TValue
*o
= L
->base
+1;
682 if (tvisstr(o
) && o
+1 >= L
->top
)
683 len
= strV(o
)->len
+1; /* Copy Lua string including trailing '\0'. */
685 len
= (CTSize
)ffi_checkint(L
, 3);
690 LJLIB_CF(ffi_fill
) LJLIB_REC(.)
692 void *dp
= ffi_checkptr(L
, 1, CTID_P_VOID
);
693 CTSize len
= (CTSize
)ffi_checkint(L
, 2);
695 if (L
->base
+2 < L
->top
&& !tvisnil(L
->base
+2)) fill
= ffi_checkint(L
, 3);
696 memset(dp
, fill
, len
);
700 #define H_(le, be) LJ_ENDIAN_SELECT(0x##le, 0x##be)
702 /* Test ABI string. */
703 LJLIB_CF(ffi_abi
) LJLIB_REC(.)
705 GCstr
*s
= lj_lib_checkstr(L
, 1);
709 case H_(849858eb
,ad35fd06
): b
= 1; break; /* 64bit */
711 case H_(662d3c79
,d0e22477
): b
= 1; break; /* 32bit */
714 case H_(e33ee463
,e33ee463
): b
= 1; break; /* fpu */
717 case H_(61211a23
,c2e8c81c
): b
= 1; break; /* softfp */
719 case H_(539417a8
,8ce0812f
): b
= 1; break; /* hardfp */
722 case H_(2182df8f
,f2ed1152
): b
= 1; break; /* eabi */
725 case H_(4ab624a8
,4ab624a8
): b
= 1; break; /* win */
727 case H_(3af93066
,1f001464
): b
= 1; break; /* le/be */
731 setboolV(L
->top
-1, b
);
732 setboolV(&G(L
)->tmptv2
, b
); /* Remember for trace recorder. */
738 LJLIB_PUSH(top
-8) LJLIB_SET(!) /* Store reference to miscmap table. */
740 LJLIB_CF(ffi_metatype
)
742 CTState
*cts
= ctype_cts(L
);
743 CTypeID id
= ffi_checkctype(L
, cts
, NULL
);
744 GCtab
*mt
= lj_lib_checktab(L
, 2);
745 GCtab
*t
= cts
->miscmap
;
746 CType
*ct
= ctype_get(cts
, id
); /* Only allow raw types. */
749 if (!(ctype_isstruct(ct
->info
) || ctype_iscomplex(ct
->info
) ||
750 ctype_isvector(ct
->info
)))
751 lj_err_arg(L
, 1, LJ_ERR_FFI_INVTYPE
);
752 tv
= lj_tab_setinth(L
, t
, -(int32_t)id
);
754 lj_err_caller(L
, LJ_ERR_PROTMT
);
756 lj_gc_anybarriert(L
, t
);
757 cd
= lj_cdata_new(cts
, CTID_CTYPEID
, 4);
758 *(CTypeID
*)cdataptr(cd
) = id
;
759 setcdataV(L
, L
->top
-1, cd
);
764 LJLIB_PUSH(top
-7) LJLIB_SET(!) /* Store reference to finalizer table. */
766 LJLIB_CF(ffi_gc
) LJLIB_REC(.)
768 GCcdata
*cd
= ffi_checkcdata(L
, 1);
769 TValue
*fin
= lj_lib_checkany(L
, 2);
770 CTState
*cts
= ctype_cts(L
);
771 GCtab
*t
= cts
->finalizer
;
772 CType
*ct
= ctype_raw(cts
, cd
->ctypeid
);
773 if (!(ctype_isptr(ct
->info
) || ctype_isstruct(ct
->info
) ||
774 ctype_isrefarray(ct
->info
)))
775 lj_err_arg(L
, 1, LJ_ERR_FFI_INVTYPE
);
776 if (gcref(t
->metatable
)) { /* Update finalizer table, if still enabled. */
777 copyTV(L
, lj_tab_set(L
, t
, L
->base
), fin
);
778 lj_gc_anybarriert(L
, t
);
780 cd
->marked
|= LJ_GC_CDATA_FIN
;
782 cd
->marked
&= ~LJ_GC_CDATA_FIN
;
784 L
->top
= L
->base
+1; /* Pass through the cdata object. */
788 LJLIB_PUSH(top
-5) LJLIB_SET(!) /* Store clib metatable in func environment. */
792 GCstr
*name
= lj_lib_checkstr(L
, 1);
793 int global
= (L
->base
+1 < L
->top
&& tvistruecond(L
->base
+1));
794 lj_clib_load(L
, tabref(curr_func(L
)->c
.env
), name
, global
);
798 LJLIB_PUSH(top
-4) LJLIB_SET(C
)
799 LJLIB_PUSH(top
-3) LJLIB_SET(os
)
800 LJLIB_PUSH(top
-2) LJLIB_SET(arch
)
802 #include "lj_libdef.h"
804 /* ------------------------------------------------------------------------ */
806 /* Create special weak-keyed finalizer table. */
807 static GCtab
*ffi_finalizer(lua_State
*L
)
809 /* NOBARRIER: The table is new (marked white). */
810 GCtab
*t
= lj_tab_new(L
, 0, 1);
811 settabV(L
, L
->top
++, t
);
812 setgcref(t
->metatable
, obj2gco(t
));
813 setstrV(L
, lj_tab_setstr(L
, t
, lj_str_newlit(L
, "__mode")),
814 lj_str_newlit(L
, "K"));
815 t
->nomm
= (uint8_t)(~(1u<<MM_mode
));
819 /* Register FFI module as loaded. */
820 static void ffi_register_module(lua_State
*L
)
822 cTValue
*tmp
= lj_tab_getstr(tabV(registry(L
)), lj_str_newlit(L
, "_LOADED"));
823 if (tmp
&& tvistab(tmp
)) {
824 GCtab
*t
= tabV(tmp
);
825 copyTV(L
, lj_tab_setstr(L
, t
, lj_str_newlit(L
, LUA_FFILIBNAME
)), L
->top
-1);
826 lj_gc_anybarriert(L
, t
);
830 LUALIB_API
int luaopen_ffi(lua_State
*L
)
832 CTState
*cts
= lj_ctype_init(L
);
833 settabV(L
, L
->top
++, (cts
->miscmap
= lj_tab_new(L
, 0, 1)));
834 cts
->finalizer
= ffi_finalizer(L
);
835 LJ_LIB_REG(L
, NULL
, ffi_meta
);
836 /* NOBARRIER: basemt is a GC root. */
837 setgcref(basemt_it(G(L
), LJ_TCDATA
), obj2gco(tabV(L
->top
-1)));
838 LJ_LIB_REG(L
, NULL
, ffi_clib
);
839 LJ_LIB_REG(L
, NULL
, ffi_callback
);
840 /* NOBARRIER: the key is new and lj_tab_newkey() handles the barrier. */
841 settabV(L
, lj_tab_setstr(L
, cts
->miscmap
, &cts
->g
->strempty
), tabV(L
->top
-1));
843 lj_clib_default(L
, tabV(L
->top
-1)); /* Create ffi.C default namespace. */
844 lua_pushliteral(L
, LJ_OS_NAME
);
845 lua_pushliteral(L
, LJ_ARCH_NAME
);
846 LJ_LIB_REG(L
, NULL
, ffi
); /* Note: no global "ffi" created! */
847 ffi_register_module(L
);