1 ! Copyright (C) 2004, 2010 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: accessors assocs byte-arrays byte-vectors continuations
4 continuations.private init kernel kernel.private math namespaces
8 BUILTIN: alien { underlying c-ptr read-only initial: f } expired ;
9 BUILTIN: dll { path byte-array read-only initial: B{ } } ;
11 PRIMITIVE: <callback> ( word return-rewind -- alien )
12 PRIMITIVE: <displaced-alien> ( displacement c-ptr -- alien )
13 PRIMITIVE: alien-address ( c-ptr -- addr )
14 PRIMITIVE: free-callback ( alien -- )
17 PRIMITIVE: current-callback ( -- n )
20 PREDICATE: pinned-alien < alien underlying>> not ;
22 UNION: pinned-c-ptr pinned-alien POSTPONE: f ;
24 GENERIC: element-size ( seq -- n ) flushable
26 M: byte-array element-size drop 1 ; inline
28 M: byte-vector element-size drop 1 ; inline
30 M: slice element-size seq>> element-size ; inline
32 M: f element-size drop 1 ; inline
34 GENERIC: byte-length ( obj -- n ) flushable
36 M: object byte-length [ length ] [ element-size ] bi * ; inline
38 GENERIC: >c-ptr ( obj -- c-ptr ) flushable
40 M: c-ptr >c-ptr ; inline
43 [ [ from>> ] [ element-size ] bi * ] [ seq>> >c-ptr ] bi
44 <displaced-alien> ; inline
48 M: object >c-ptr underlying>> ; inline
50 : binary-object ( obj -- c-ptr n )
51 [ >c-ptr ] [ byte-length ] bi ; inline
53 GENERIC: expired? ( c-ptr -- ? ) flushable
55 M: alien expired? expired>> ;
57 M: f expired? drop t ;
59 : <alien> ( address -- alien )
60 f <displaced-alien> { pinned-c-ptr } declare ; inline
62 : <bad-alien> ( -- alien )
63 -1 <alien> t >>expired ; inline
67 2dup [ expired? ] either? [
70 [ alien-address ] same?
76 M: pinned-alien hashcode*
77 nip dup expired>> [ drop 1234 ] [ alien-address ] if ;
79 SINGLETONS: stdcall thiscall fastcall cdecl mingw ;
81 UNION: abi stdcall thiscall fastcall cdecl mingw ;
83 : callee-cleanup? ( abi -- ? )
84 { stdcall fastcall thiscall } member? ;
86 ERROR: alien-callback-error ;
88 : alien-callback ( return parameters abi quot -- alien )
89 throw-alien-callback-error ;
91 ERROR: alien-indirect-error ;
93 : alien-indirect ( args... funcptr return parameters abi -- return... )
94 throw-alien-indirect-error ;
96 ERROR: alien-invoke-error library symbol ;
98 : alien-invoke ( args... return library function parameters -- return... )
99 2over throw-alien-invoke-error ;
101 ERROR: alien-assembly-error code ;
103 : alien-assembly ( args... return parameters abi quot -- return... )
104 dup throw-alien-assembly-error ;
108 ! Callbacks are registered in a global hashtable. Note that they
109 ! are also pinned in a special callback area, so clearing this
110 ! hashtable will not reclaim callbacks. It should only be
111 ! cleared on startup.
114 [ H{ } clone callbacks set-global ] "alien" add-startup-hook
116 ! Used by compiler.codegen to wrap callback bodies
117 : do-callback ( callback-quot wait-quot: ( callback -- ) -- )
118 t CONTEXT-OBJ-IN-CALLBACK-P set-context-object
122 [ 2drop call ] [ swap call( callback -- ) drop ] 3bi ; inline
124 ! A utility for defining global variables that are recompiled in
126 TUPLE: expiry-check object alien ;
128 : recompute-value? ( check -- ? )
129 [ alien>> expired? ] [ t ] if* ;
131 : delete-values ( value assoc -- )
132 [ rot drop = ] with assoc-reject! drop ;
136 : unregister-and-free-callback ( alien -- )
137 [ callbacks get delete-values ] [ free-callback ] bi ;
139 : with-callback ( alien quot -- )
140 over [ unregister-and-free-callback ] curry [ ] cleanup ; inline
142 : initialize-alien ( symbol quot -- )
143 swap dup get-global dup recompute-value?
144 [ drop [ call dup 31337 <alien> expiry-check boa ] dip set-global ]
145 [ 2nip object>> ] if ; inline