factor/library/alien/compiler.factor

131 lines
4.0 KiB
Factor

! Copyright (C) 2004, 2005 Slava Pestov.
! See http://factor.sf.net/license.txt for BSD license.
IN: alien
USING: assembler compiler compiler-backend errors generic
inference kernel lists math namespaces sequences stdio strings
unparser words ;
! ! ! WARNING ! ! !
! Reloading this file into a running Factor instance on Win32
! or Unix with FFI I/O will bomb the runtime, since I/O words
! would become uncompiled, and FFI calls can only be made from
! compiled code.
! USAGE:
!
! Command line parameters given to the runtime specify libraries
! to load.
!
! -libraries:<foo>:name=<soname> -- define a library <foo>, to be
! loaded from the <soname> DLL.
!
! -libraries:<foo>:abi=stdcall -- define a library using the
! stdcall ABI. This ABI is usually used on Win32. Any other abi
! parameter, or a missing abi parameter indicates the cdecl ABI
! should be used, which is common on Unix.
! FFI code does not run in the interpreter.
TUPLE: alien-error symbol library ;
C: alien-error ( lib sym -- )
[ set-alien-error-symbol ] keep
[ set-alien-error-library ] keep ;
M: alien-error error. ( error -- )
[
"C library interface words cannot be interpreted. " %
"Either the compiler is disabled, " %
"or the " % dup alien-error-library unparse %
" library does not define the " %
alien-error-symbol unparse %
" symbol." %
] make-string print ;
: alien-invoke ( ... returns library function parameters -- ... )
#! Call a C library function.
#! 'returns' is a type spec, and 'parameters' is a list of
#! type specs. 'library' is an entry in the "libraries"
#! namespace.
drop <alien-error> throw ;
! These are set in the alien-invoke dataflow IR node.
SYMBOL: alien-returns
SYMBOL: alien-parameters
: set-alien-returns ( returns node -- )
[ dup alien-returns set ] bind
"void" = [
[ object ] produce-d 1 0 node-outputs
] unless ;
: set-alien-parameters ( parameters node -- )
[ dup alien-parameters set ] bind
[ drop object ] map dup dup ensure-d
length 0 node-inputs consume-d ;
: ensure-dlsym ( symbol library -- ) load-library dlsym drop ;
: alien-invoke-node ( returns params function library -- )
#! We should fail if the library does not exist, so that
#! compilation does not keep trying to compile FFI words
#! over and over again if the library is not loaded.
2dup ensure-dlsym
cons \ alien-invoke dataflow,
[ set-alien-parameters ] keep
set-alien-returns ;
: infer-alien-invoke ( -- )
\ alien-invoke "infer-effect" word-prop car ensure-d
pop-literal
pop-literal >r
pop-literal
pop-literal -rot
r> swap alien-invoke-node ;
: parameters [ alien-parameters get reverse ] bind ;
: stack-space ( parameters -- n )
0 swap [ c-size cell align + ] each ;
: unbox-parameter ( n parameter -- )
c-type [ "unboxer" get cons "unbox-op" get ] bind execute , ;
: linearize-parameters ( node -- count )
#! Generate code for boxing a list of C types, then generate
#! code for moving these parameters to register on
#! architectures where parameters are passed in registers
#! (PowerPC).
#!
#! Return amount stack must be unwound by.
parameters
dup stack-space
dup %parameters , >r
dup dup length swap [ >r 1 - dup r> unbox-parameter ] each drop
length [ %parameter ] project % r> ;
: linearize-returns ( returns -- )
[ alien-returns get ] bind dup "void" = [
drop
] [
c-type [ "boxer" get "box-op" get ] bind execute ,
] ifte ;
: linearize-alien-invoke ( node -- )
dup linearize-parameters >r
dup [ node-param get ] bind %alien-invoke ,
dup [ node-param get cdr library-abi "stdcall" = ] bind
r> swap [ drop ] [ %cleanup , ] ifte
linearize-returns ;
\ alien-invoke [ linearize-alien-invoke ] "linearizer" set-word-prop
\ alien-invoke [ [ string string string general-list ] [ ] ]
"infer-effect" set-word-prop
\ alien-invoke [ infer-alien-invoke ] "infer" set-word-prop
global [
"libraries" get [ <namespace> "libraries" set ] unless
] bind