teach IR1-TRANSFORM-TYPE-PREDICATE about alien types
[sbcl.git] / src / code / target-alieneval.lisp
1 ;;;; This file contains parts of the ALIEN implementation that
2 ;;;; are not part of the compiler.
3
4 ;;;; This software is part of the SBCL system. See the README file for
5 ;;;; more information.
6 ;;;;
7 ;;;; This software is derived from the CMU CL system, which was
8 ;;;; written at Carnegie Mellon University and released into the
9 ;;;; public domain. The software is in the public domain and is
10 ;;;; provided with absolutely no warranty. See the COPYING and CREDITS
11 ;;;; files for more information.
12
13 (in-package "SB!ALIEN")
14
15 (/show0 "target-alieneval.lisp 15")
16 \f
17 ;;;; alien variables
18
19 ;;; Make a string out of the symbol, converting all uppercase letters to
20 ;;; lower case and hyphens into underscores.
21 (eval-when (:compile-toplevel :load-toplevel :execute)
22   (defun guess-alien-name-from-lisp-name (lisp-name)
23     (declare (type symbol lisp-name))
24     (nsubstitute #\_ #\- (string-downcase (symbol-name lisp-name)))))
25
26 ;;; The opposite of GUESS-ALIEN-NAME-FROM-LISP-NAME. Make a symbol out
27 ;;; of the string, converting all lowercase letters to uppercase and
28 ;;; underscores into hyphens.
29 (eval-when (:compile-toplevel :load-toplevel :execute)
30   (defun guess-lisp-name-from-alien-name (alien-name)
31     (declare (type simple-string alien-name))
32     (intern (nsubstitute #\- #\_ (string-upcase alien-name)))))
33
34 ;;; Extract the Lisp and alien names from NAME. If only one is given,
35 ;;; guess the other.
36 (eval-when (:compile-toplevel :load-toplevel :execute)
37   (defun pick-lisp-and-alien-names (name)
38     (etypecase name
39       (string
40        (values (guess-lisp-name-from-alien-name name) name))
41       (symbol
42        (values name (guess-alien-name-from-lisp-name name)))
43       (list
44        (unless (proper-list-of-length-p name 2)
45          (error "badly formed alien name"))
46        (values (cadr name) (car name))))))
47
48 (defmacro define-alien-variable (name type &environment env)
49   #!+sb-doc
50   "Define NAME as an external alien variable of type TYPE. NAME should be
51    a list of a string holding the alien name and a symbol to use as the Lisp
52    name. If NAME is just a symbol or string, then the other name is guessed
53    from the one supplied."
54   (multiple-value-bind (lisp-name alien-name) (pick-lisp-and-alien-names name)
55     (with-auxiliary-alien-types env
56       (let ((alien-type (parse-alien-type type env)))
57         `(eval-when (:compile-toplevel :load-toplevel :execute)
58            ,@(when *new-auxiliary-types*
59                `((%def-auxiliary-alien-types ',*new-auxiliary-types*)))
60            (%define-alien-variable ',lisp-name
61                                    ',alien-name
62                                    ',alien-type))))))
63
64 ;;; Do the actual work of DEFINE-ALIEN-VARIABLE.
65 (eval-when (:compile-toplevel :load-toplevel :execute)
66   (defun %define-alien-variable (lisp-name alien-name type)
67     (setf (info :variable :kind lisp-name) :alien)
68     (setf (info :variable :where-from lisp-name) :defined)
69     (setf (info :variable :alien-info lisp-name)
70           (make-heap-alien-info :type type
71                                 :sap-form `(foreign-symbol-sap ',alien-name t)))))
72
73 (defmacro extern-alien (name type &environment env)
74   #!+sb-doc
75   "Access the alien variable named NAME, assuming it is of type TYPE. This
76    is SETFable."
77   (let* ((alien-name (etypecase name
78                        (symbol (guess-alien-name-from-lisp-name name))
79                        (string name)))
80          (alien-type (parse-alien-type type env))
81          (datap (not (alien-fun-type-p alien-type))))
82     `(%heap-alien ',(make-heap-alien-info
83                      :type alien-type
84                      :sap-form `(foreign-symbol-sap ',alien-name ,datap)))))
85
86 (defmacro with-alien (bindings &body body &environment env)
87   #!+sb-doc
88   "Establish some local alien variables. Each BINDING is of the form:
89      VAR TYPE [ ALLOCATION ] [ INITIAL-VALUE | EXTERNAL-NAME ]
90    ALLOCATION should be one of:
91      :LOCAL (the default)
92        The alien is allocated on the stack, and has dynamic extent.
93      :EXTERN
94        No alien is allocated, but VAR is established as a local name for
95        the external alien given by EXTERNAL-NAME."
96   ;; FIXME:
97   ;;      :STATIC
98   ;;        The alien is allocated on the heap, and has infinite extent. The alien
99   ;;        is allocated at load time, so the same piece of memory is used each time
100   ;;        this form executes.
101   (/show "entering WITH-ALIEN" bindings)
102   (with-auxiliary-alien-types env
103     (dolist (binding (reverse bindings))
104       (/show binding)
105       (destructuring-bind
106             (symbol type &optional (opt1 nil opt1p) (opt2 nil opt2p))
107           binding
108         (/show symbol type opt1 opt2)
109         (let* ((alien-type (parse-alien-type type env))
110                (datap (not (alien-fun-type-p alien-type))))
111           (/show alien-type)
112           (multiple-value-bind (allocation initial-value)
113               (if opt2p
114                   (values opt1 opt2)
115                   (case opt1
116                     (:extern
117                      (values opt1 (guess-alien-name-from-lisp-name symbol)))
118                     (:static
119                      (values opt1 nil))
120                     (t
121                      (values :local opt1))))
122             (/show allocation initial-value)
123             (setf body
124                   (ecase allocation
125                     #+nil
126                     (:static
127                      (let ((sap
128                             (make-symbol (concatenate 'string "SAP-FOR-"
129                                                       (symbol-name symbol)))))
130                        `((let ((,sap (load-time-value (%make-alien ...))))
131                            (declare (type system-area-pointer ,sap))
132                            (symbol-macrolet
133                                ((,symbol (sap-alien ,sap ,type)))
134                              ,@(when initial-value
135                                  `((setq ,symbol ,initial-value)))
136                              ,@body)))))
137                     (:extern
138                      (/show0 ":EXTERN case")
139                      (let ((info (make-heap-alien-info
140                                   :type alien-type
141                                   :sap-form `(foreign-symbol-sap ',initial-value
142                                                                  ,datap))))
143                        `((symbol-macrolet
144                              ((,symbol (%heap-alien ',info)))
145                            ,@body))))
146                     (:local
147                      (/show0 ":LOCAL case")
148                      (let* ((var (sb!xc:gensym "VAR"))
149                             (initval (if initial-value (sb!xc:gensym "INITVAL")))
150                             (info (make-local-alien-info :type alien-type))
151                             (inner-body
152                              `((note-local-alien-type ',info ,var)
153                                (symbol-macrolet ((,symbol (local-alien ',info ,var)))
154                                  ,@(when initial-value
155                                      `((setq ,symbol ,initval)))
156                                  ,@body)))
157                             (body-forms
158                              (if initial-value
159                                  `((let ((,initval ,initial-value))
160                                      ,@inner-body))
161                                  inner-body)))
162                        (/show var initval info)
163                        #!+(or x86 x86-64)
164                        `((let ((,var (make-local-alien ',info)))
165                            ,@body-forms))
166                        ;; FIXME: This version is less efficient then it needs to be, since
167                        ;; it could just save and restore the number-stack pointer once,
168                        ;; instead of doing multiple decrements if there are multiple bindings.
169                        #!-(or x86 x86-64)
170                        `((let (,var)
171                            (unwind-protect
172                                (progn
173                                  (setf ,var (make-local-alien ',info))
174                                  (let ((,var ,var))
175                                    ,@body-forms))
176                              (dispose-local-alien ',info ,var))))))))))))
177     (/show "revised" body)
178     (verify-local-auxiliaries-okay)
179     (/show0 "back from VERIFY-LOCAL-AUXILIARIES-OK, returning")
180     `(symbol-macrolet ((&auxiliary-type-definitions&
181                         ,(append *new-auxiliary-types*
182                                  (auxiliary-type-definitions env))))
183        #!+(or x86 x86-64)
184        (let ((sb!vm::*alien-stack* sb!vm::*alien-stack*))
185          ,@body)
186        #!-(or x86 x86-64)
187        ,@body)))
188 \f
189 ;;;; runtime C values that don't correspond directly to Lisp types
190
191 ;;; Note: The DEFSTRUCT for ALIEN-VALUE lives in a separate file
192 ;;; 'cause it has to be real early in the cold-load order.
193 #!-sb-fluid (declaim (freeze-type alien-value))
194 (def!method print-object ((value alien-value) stream)
195   (print-unreadable-object (value stream)
196     (format stream
197             "~S ~S #X~8,'0X ~S ~S"
198             'alien-value
199             :sap (sap-int (alien-value-sap value))
200             :type (unparse-alien-type (alien-value-type value)))))
201
202 #!-sb-fluid (declaim (inline null-alien))
203 (defun null-alien (x)
204   #!+sb-doc
205   "Return true if X (which must be an ALIEN pointer) is null, false otherwise."
206   (zerop (sap-int (alien-sap x))))
207
208 (defmacro sap-alien (sap type &environment env)
209   #!+sb-doc
210   "Convert the system area pointer SAP to an ALIEN of the specified TYPE (not
211    evaluated.) TYPE must be pointer-like."
212   (let ((alien-type (parse-alien-type type env)))
213     (if (eq (compute-alien-rep-type alien-type) 'system-area-pointer)
214         `(%sap-alien ,sap ',alien-type)
215         (error "cannot make an alien of type ~S out of a SAP" type))))
216
217 (defun %sap-alien (sap type)
218   (declare (type system-area-pointer sap)
219            (type alien-type type))
220   (make-alien-value :sap sap :type type))
221
222 (defun alien-sap (alien)
223   #!+sb-doc
224   "Return a System-Area-Pointer pointing to Alien's data."
225   (declare (type alien-value alien))
226   (alien-value-sap alien))
227 \f
228 ;;;; allocation/deallocation of heap aliens
229
230 (defmacro make-alien (type &optional size &environment env)
231   #!+sb-doc
232   "Allocate an alien of type TYPE in foreign heap, and return an alien
233 pointer to it. The allocated memory is not initialized, and may
234 contain garbage. The memory is allocated using malloc(3), so it can be
235 passed to foreign functions which use free(3), or released using
236 FREE-ALIEN.
237
238 For alien stack allocation, see macro WITH-ALIEN.
239
240 The TYPE argument is not evaluated. If SIZE is supplied, how it is
241 interpreted depends on TYPE:
242
243   * When TYPE is a foreign array type, an array of that type is
244     allocated, and a pointer to it is returned. Note that you
245     must use DEREF to first access the arrey through the pointer.
246
247     If supplied, SIZE is used as the first dimension for the array.
248
249   * When TYPE is any other foreign type, then an object for that
250     type is allocated, and a pointer to it is returned. So
251     (make-alien int) returns a (* int).
252
253     If SIZE is specified, then a block of that many objects is
254     allocated, with the result pointing to the first one.
255
256 Examples:
257
258   (defvar *foo* (make-alien (array char 10)))
259   (type-of *foo*)                 ; => (alien (* (array (signed 8) 10)))
260   (setf (deref (deref foo) 0) 10) ; => 10
261
262   (make-alien char 12)            ; => (alien (* (signed 8)))
263 "
264   (let ((alien-type (if (alien-type-p type)
265                         type
266                         (parse-alien-type type env))))
267     (multiple-value-bind (size-expr element-type)
268         (if (alien-array-type-p alien-type)
269             (let ((dims (alien-array-type-dimensions alien-type)))
270               (cond
271                 (size
272                  (unless dims
273                    (error
274                     "cannot override the size of zero-dimensional arrays"))
275                  (when (constantp size)
276                    (setf alien-type (copy-alien-array-type alien-type))
277                    (setf (alien-array-type-dimensions alien-type)
278                          (cons (constant-form-value size) (cdr dims)))))
279                 (dims
280                  (setf size (car dims)))
281                 (t
282                  (setf size 1)))
283               (values `(* ,size ,@(cdr dims))
284                       (alien-array-type-element-type alien-type)))
285             (values (or size 1) alien-type))
286       (let ((bits (alien-type-bits element-type))
287             (alignment (alien-type-alignment element-type)))
288         (unless bits
289           (error "The size of ~S is unknown."
290                  (unparse-alien-type element-type)))
291         (unless alignment
292           (error "The alignment of ~S is unknown."
293                  (unparse-alien-type element-type)))
294         ;; This is the one place where the %SAP-ALIEN note is quite
295         ;; undesirable, in most uses of MAKE-ALIEN the %SAP-ALIEN
296         ;; cannot be optimized away.
297         `(locally (declare (muffle-conditions compiler-note))
298            ;; FIXME: Do we really need the ASH/+7 here after ALIGN-OFFSET?
299            (%sap-alien (%make-alien (* ,(ash (+ 7 (align-offset bits alignment)) -3)
300                                        (the index ,size-expr)))
301                        ',(make-alien-pointer-type :to alien-type)))))))
302
303 (defun malloc-error (bytes errno)
304   (error 'simple-storage-condition
305          :format-control "~A: malloc() of ~S bytes failed."
306          :format-arguments (list (strerror errno) bytes)))
307
308 ;;; Allocate a block of memory at least BITS bits long and return a
309 ;;; system area pointer to it.
310 #!-sb-fluid (declaim (inline %make-alien))
311 (defun %make-alien (bytes)
312   (declare (type index bytes)
313            (optimize (sb!c:alien-funcall-saves-fp-and-pc 0)))
314   (let ((sap (alien-funcall (extern-alien "malloc"
315                                           (function system-area-pointer size-t))
316                             bytes)))
317     (if (and (not (eql 0 bytes)) (eql 0 (sap-int sap)))
318         (malloc-error bytes (get-errno))
319         sap)))
320
321 #!-sb-fluid (declaim (inline free-alien))
322 (defun free-alien (alien)
323   #!+sb-doc
324   "Dispose of the storage pointed to by ALIEN. The ALIEN must have been
325 allocated by MAKE-ALIEN, MAKE-ALIEN-STRING or malloc(3)."
326   (alien-funcall (extern-alien "free" (function (values) system-area-pointer))
327                  (alien-sap alien))
328   nil)
329
330 (declaim (type (sfunction * system-area-pointer) %make-alien-string))
331 (defun %make-alien-string (string &key (start 0) end
332                                        (external-format :default)
333                                        (null-terminate t))
334   ;; FIXME: This is slow. We want a function to get the length of the
335   ;; encoded string so we can allocate the foreign memory first and
336   ;; encode directly there.
337   (let* ((octets (string-to-octets string
338                                    :start start :end end
339                                    :external-format external-format
340                                    :null-terminate null-terminate))
341          (count (length octets))
342          (buf (%make-alien count)))
343     (sb!kernel:copy-ub8-to-system-area octets 0 buf 0 count)
344     buf))
345
346 (defun make-alien-string (string &rest rest
347                                  &key (start 0) end
348                                       (external-format :default)
349                                       (null-terminate t))
350   "Copy part of STRING delimited by START and END into freshly
351 allocated foreign memory, freeable using free(3) or FREE-ALIEN.
352 Returns the allocated string as a (* CHAR) alien, and the number of
353 bytes allocated as secondary value.
354
355 The string is encoded using EXTERNAL-FORMAT. If NULL-TERMINATE is
356 true (the default), the alien string is terminated by an additional
357 null byte.
358 "
359   (declare (ignore start end external-format null-terminate))
360   (multiple-value-bind (sap bytes)
361       (apply #'%make-alien-string string rest)
362     (values (%sap-alien sap (parse-alien-type '(* char) nil))
363             bytes)))
364
365 (define-compiler-macro make-alien-string (&rest args)
366   `(multiple-value-bind (sap bytes) (%make-alien-string ,@args)
367      (values (%sap-alien sap ',(parse-alien-type '(* char) nil))
368              bytes)))
369 \f
370 ;;;; the SLOT operator
371
372 ;;; Find the field named SLOT, or die trying.
373 (defun slot-or-lose (type slot)
374   (declare (type alien-record-type type)
375            (type symbol slot))
376   (or (find slot (alien-record-type-fields type)
377             :key #'alien-record-field-name)
378       (error "There is no slot named ~S in ~S." slot type)))
379
380 ;;; Extract the value from the named slot from the record ALIEN. If
381 ;;; ALIEN is actually a pointer, then DEREF it first.
382 (defun slot (alien slot)
383   #!+sb-doc
384   "Extract SLOT from the Alien STRUCT or UNION ALIEN. May be set with SETF."
385   (declare (type alien-value alien)
386            (type symbol slot)
387            (optimize (inhibit-warnings 3)))
388   (let ((type (alien-value-type alien)))
389     (etypecase type
390       (alien-pointer-type
391        (slot (deref alien) slot))
392       (alien-record-type
393        (let ((field (slot-or-lose type slot)))
394          (extract-alien-value (alien-value-sap alien)
395                               (alien-record-field-offset field)
396                               (alien-record-field-type field)))))))
397
398 ;;; Deposit the value in the specified slot of the record ALIEN. If
399 ;;; the ALIEN is really a pointer, DEREF it first. The compiler uses
400 ;;; this when it can't figure out anything better.
401 (defun %set-slot (alien slot value)
402   (declare (type alien-value alien)
403            (type symbol slot)
404            (optimize (inhibit-warnings 3)))
405   (let ((type (alien-value-type alien)))
406     (etypecase type
407       (alien-pointer-type
408        (%set-slot (deref alien) slot value))
409       (alien-record-type
410        (let ((field (slot-or-lose type slot)))
411          (deposit-alien-value (alien-value-sap alien)
412                               (alien-record-field-offset field)
413                               (alien-record-field-type field)
414                               value))))))
415
416 ;;; Compute the address of the specified slot and return a pointer to it.
417 (defun %slot-addr (alien slot)
418   (declare (type alien-value alien)
419            (type symbol slot)
420            (optimize (inhibit-warnings 3)))
421   (let ((type (alien-value-type alien)))
422     (etypecase type
423       (alien-pointer-type
424        (%slot-addr (deref alien) slot))
425       (alien-record-type
426        (let* ((field (slot-or-lose type slot))
427               (offset (alien-record-field-offset field))
428               (field-type (alien-record-field-type field)))
429          (%sap-alien (sap+ (alien-sap alien) (/ offset sb!vm:n-byte-bits))
430                      (make-alien-pointer-type :to field-type)))))))
431 \f
432 ;;;; the DEREF operator
433
434 ;;; This function does most of the work of the different DEREF
435 ;;; methods. It returns two values: the type and the offset (in bits)
436 ;;; of the referred-to alien.
437 (defun deref-guts (alien indices)
438   (declare (type alien-value alien)
439            (type list indices)
440            (values alien-type integer))
441   (let ((type (alien-value-type alien)))
442     (etypecase type
443       (alien-pointer-type
444        (when (cdr indices)
445          (error "too many indices when DEREF'ing ~S: ~W"
446                 type
447                 (length indices)))
448        (let ((element-type (alien-pointer-type-to type)))
449          (values element-type
450                  (if indices
451                      (* (align-offset (alien-type-bits element-type)
452                                       (alien-type-alignment element-type))
453                         (car indices))
454                      0))))
455       (alien-array-type
456        (unless (= (length indices) (length (alien-array-type-dimensions type)))
457          (error "incorrect number of indices when DEREF'ing ~S: ~W"
458                 type (length indices)))
459        (labels ((frob (dims indices offset)
460                   (if (null dims)
461                       offset
462                       (frob (cdr dims) (cdr indices)
463                         (+ (if (zerop offset)
464                                0
465                                (* offset (car dims)))
466                            (car indices))))))
467          (let ((element-type (alien-array-type-element-type type)))
468            (values element-type
469                    (* (align-offset (alien-type-bits element-type)
470                                     (alien-type-alignment element-type))
471                       (frob (alien-array-type-dimensions type)
472                         indices 0)))))))))
473
474 ;;; Dereference the alien and return the results.
475 (defun deref (alien &rest indices)
476   #!+sb-doc
477   "De-reference an Alien pointer or array. If an array, the indices are used
478    as the indices of the array element to access. If a pointer, one index can
479    optionally be specified, giving the equivalent of C pointer arithmetic."
480   (declare (type alien-value alien)
481            (type list indices)
482            (optimize (inhibit-warnings 3)))
483   (multiple-value-bind (target-type offset) (deref-guts alien indices)
484     (extract-alien-value (alien-value-sap alien)
485                          offset
486                          target-type)))
487
488 (defun %set-deref (alien value &rest indices)
489   (declare (type alien-value alien)
490            (type list indices)
491            (optimize (inhibit-warnings 3)))
492   (multiple-value-bind (target-type offset) (deref-guts alien indices)
493     (deposit-alien-value (alien-value-sap alien)
494                          offset
495                          target-type
496                          value)))
497
498 (defun %deref-addr (alien &rest indices)
499   (declare (type alien-value alien)
500            (type list indices)
501            (optimize (inhibit-warnings 3)))
502   (multiple-value-bind (target-type offset) (deref-guts alien indices)
503     (%sap-alien (sap+ (alien-value-sap alien) (/ offset sb!vm:n-byte-bits))
504                 (make-alien-pointer-type :to target-type))))
505 \f
506 ;;;; accessing heap alien variables
507
508 (defun %heap-alien (info)
509   (declare (type heap-alien-info info)
510            (optimize (inhibit-warnings 3)))
511   (extract-alien-value (eval (heap-alien-info-sap-form info))
512                        0
513                        (heap-alien-info-type info)))
514
515 (defun %set-heap-alien (info value)
516   (declare (type heap-alien-info info)
517            (optimize (inhibit-warnings 3)))
518   (deposit-alien-value (eval (heap-alien-info-sap-form info))
519                        0
520                        (heap-alien-info-type info)
521                        value))
522
523 (defun %heap-alien-addr (info)
524   (declare (type heap-alien-info info)
525            (optimize (inhibit-warnings 3)))
526   (%sap-alien (eval (heap-alien-info-sap-form info))
527               (make-alien-pointer-type :to (heap-alien-info-type info))))
528 \f
529 ;;;; accessing local aliens
530
531 (defun make-local-alien (info)
532   (let* ((alien (eval `(make-alien ,(local-alien-info-type info))))
533          (alien-sap (alien-sap alien)))
534     (finalize
535      alien
536      (lambda ()
537        (alien-funcall
538         (extern-alien "free" (function (values) system-area-pointer))
539         alien-sap))
540      :dont-save t)
541     alien))
542
543 (defun note-local-alien-type (info alien)
544   (declare (ignore info alien))
545   nil)
546
547 (defun local-alien (info alien)
548   (declare (ignore info))
549   (deref alien))
550
551 (defun %set-local-alien (info alien value)
552   (declare (ignore info))
553   (setf (deref alien) value))
554
555 (define-setf-expander local-alien (&whole whole info alien)
556   (let ((value (gensym))
557         (info-var (gensym))
558         (alloc-tmp (gensym))
559         (info (if (and (consp info)
560                        (eq (car info) 'quote))
561                   (second info)
562                   (error "Something is wrong; local-alien-info not found: ~S"
563                          whole))))
564     (values nil
565             nil
566             (list value)
567             `(if (%local-alien-forced-to-memory-p ',info)
568                  (%set-local-alien ',info ,alien ,value)
569                    (let* ((,info-var ',(local-alien-info-type info))
570                           (,alloc-tmp (deport-alloc ,value ,info-var)))
571                      (maybe-with-pinned-objects (,alloc-tmp) (,(local-alien-info-type info))
572                        (setf ,alien (deport ,alloc-tmp ,info-var)))))
573             whole)))
574
575 (defun %local-alien-forced-to-memory-p (info)
576   (local-alien-info-force-to-memory-p info))
577
578 (defun %local-alien-addr (info alien)
579   (declare (type local-alien-info info))
580   (unless (local-alien-info-force-to-memory-p info)
581     (error "~S isn't forced to memory. Something went wrong." alien))
582   alien)
583
584 (defun dispose-local-alien (info alien)
585   (declare (ignore info))
586   (cancel-finalization alien)
587   (free-alien alien))
588 \f
589 ;;;; the CAST macro
590
591 (defmacro cast (alien type &environment env)
592   #!+sb-doc
593   "Convert ALIEN to an Alien of the specified TYPE (not evaluated.)  Both types
594    must be Alien array, pointer or function types."
595   `(%cast ,alien ',(parse-alien-type type env)))
596
597 (defun %cast (alien target-type)
598   (declare (type alien-value alien)
599            (type alien-type target-type)
600            (optimize (safety 2))
601            (optimize (inhibit-warnings 3)))
602   (if (or (alien-pointer-type-p target-type)
603           (alien-array-type-p target-type)
604           (alien-fun-type-p target-type))
605       (let ((alien-type (alien-value-type alien)))
606         (if (or (alien-pointer-type-p alien-type)
607                 (alien-array-type-p alien-type)
608                 (alien-fun-type-p alien-type))
609             (naturalize (alien-value-sap alien) target-type)
610             (error "~S cannot be casted." alien)))
611       (error "cannot cast to alien type ~S" (unparse-alien-type target-type))))
612 \f
613 ;;;; the ALIEN-SIZE macro
614
615 (defmacro alien-size (type &optional (units :bits) &environment env)
616   #!+sb-doc
617   "Return the size of the alien type TYPE. UNITS specifies the units to
618    use and can be either :BITS, :BYTES, or :WORDS."
619   (let* ((alien-type (parse-alien-type type env))
620          (bits (alien-type-bits alien-type)))
621     (if bits
622         (values (ceiling bits
623                          (ecase units
624                            (:bits 1)
625                            (:bytes sb!vm:n-byte-bits)
626                            (:words sb!vm:n-word-bits))))
627         (error "unknown size for alien type ~S"
628                (unparse-alien-type alien-type)))))
629 \f
630 ;;;; NATURALIZE, DEPORT, EXTRACT-ALIEN-VALUE, DEPOSIT-ALIEN-VALUE
631
632 (defun coerce-to-interpreted-function (lambda-form)
633   (let (#!+sb-eval
634         (*evaluator-mode* :interpret))
635     (coerce lambda-form 'function)))
636
637 (defun naturalize (alien type)
638   (declare (type alien-type type))
639   (funcall (coerce-to-interpreted-function (compute-naturalize-lambda type))
640            alien type))
641
642 (defun deport (value type)
643   (declare (type alien-type type))
644   (funcall (coerce-to-interpreted-function (compute-deport-lambda type))
645            value type))
646
647 (defun deport-alloc (value type)
648   (declare (type alien-type type))
649   (funcall (coerce-to-interpreted-function (compute-deport-alloc-lambda type))
650            value type))
651
652 (defun extract-alien-value (sap offset type)
653   (declare (type system-area-pointer sap)
654            (type unsigned-byte offset)
655            (type alien-type type))
656   (funcall (coerce-to-interpreted-function (compute-extract-lambda type))
657            sap offset type))
658
659 (defun deposit-alien-value (sap offset type value)
660   (declare (type system-area-pointer sap)
661            (type unsigned-byte offset)
662            (type alien-type type))
663   (funcall (coerce-to-interpreted-function (compute-deposit-lambda type))
664            sap offset type value))
665 \f
666 ;;;; ALIEN-FUNCALL, DEFINE-ALIEN-ROUTINE
667
668 (defun alien-funcall (alien &rest args)
669   #!+sb-doc
670   "Call the foreign function ALIEN with the specified arguments. ALIEN's
671    type specifies the argument and result types."
672   (declare (type alien-value alien))
673   (let ((type (alien-value-type alien)))
674     (typecase type
675       (alien-pointer-type
676        (apply #'alien-funcall (deref alien) args))
677       (alien-fun-type
678        (unless (= (length (alien-fun-type-arg-types type))
679                   (length args))
680          (error "wrong number of arguments for ~S~%expected ~W, got ~W"
681                 type
682                 (length (alien-fun-type-arg-types type))
683                 (length args)))
684        (let ((stub (alien-fun-type-stub type)))
685          (unless stub
686            (setf stub
687                  (let ((fun (sb!xc:gensym "FUN"))
688                        (parms (make-gensym-list (length args))))
689                    (compile nil
690                             `(lambda (,fun ,@parms)
691                                (declare (optimize (sb!c::insert-step-conditions 0)))
692                                (declare (type (alien ,type) ,fun))
693                                (alien-funcall ,fun ,@parms)))))
694            (setf (alien-fun-type-stub type) stub))
695          (apply stub alien args)))
696       (t
697        (error "~S is not an alien function." alien)))))
698
699 (defmacro define-alien-routine (name result-type
700                                      &rest args
701                                      &environment lexenv)
702   #!+sb-doc
703   "DEFINE-ALIEN-ROUTINE Name Result-Type {(Arg-Name Arg-Type [Style])}*
704
705   Define a foreign interface function for the routine with the specified NAME.
706   Also automatically DECLAIM the FTYPE of the defined function.
707
708   NAME may be either a string, a symbol, or a list of the form (string symbol).
709
710   RETURN-TYPE is the alien type for the function return value. VOID may be
711   used to specify a function with no result.
712
713   The remaining forms specify individual arguments that are passed to the
714   routine. ARG-NAME is a symbol that names the argument, primarily for
715   documentation. ARG-TYPE is the C type of the argument. STYLE specifies the
716   way that the argument is passed.
717
718   :IN
719         An :IN argument is simply passed by value. The value to be passed is
720         obtained from argument(s) to the interface function. No values are
721         returned for :In arguments. This is the default mode.
722
723   :OUT
724         The specified argument type must be a pointer to a fixed sized object.
725         A pointer to a preallocated object is passed to the routine, and the
726         the object is accessed on return, with the value being returned from
727         the interface function. :OUT and :IN-OUT cannot be used with pointers
728         to arrays, records or functions.
729
730   :COPY
731         This is similar to :IN, except that the argument values are stored
732         on the stack, and a pointer to the object is passed instead of
733         the value itself.
734
735   :IN-OUT
736         This is a combination of :OUT and :COPY. A pointer to the argument is
737         passed, with the object being initialized from the supplied argument
738         and the return value being determined by accessing the object on
739         return."
740   (multiple-value-bind (lisp-name alien-name)
741       (pick-lisp-and-alien-names name)
742     (collect ((docs) (lisp-args) (lisp-arg-types)
743               (lisp-result-types
744                (cond ((eql result-type 'void)
745                       ;; What values does a function return, if it
746                       ;; returns no values? Exactly one - NIL. -- APD,
747                       ;; 2003-03-02
748                       (list 'null))
749                      (t
750                       ;; FIXME: Check for VALUES.
751                       (list `(alien ,result-type)))))
752               (arg-types) (alien-vars)
753               (alien-args) (results))
754       (dolist (arg args)
755         (if (stringp arg)
756             (docs arg)
757             (destructuring-bind (name type &optional (style :in)) arg
758               (unless (member style '(:in :copy :out :in-out))
759                 (error "bogus argument style ~S in ~S" style arg))
760               (when (and (member style '(:out :in-out))
761                          (typep (parse-alien-type type lexenv)
762                                 'alien-pointer-type))
763                 (error "can't use :OUT or :IN-OUT on pointer-like type:~%  ~S"
764                        type))
765               (let (arg-type)
766                 (cond ((eq style :in)
767                        (setq arg-type type)
768                        (alien-args name))
769                       (t
770                        (setq arg-type `(* ,type))
771                        (if (eq style :out)
772                            (alien-vars `(,name ,type))
773                            (alien-vars `(,name ,type ,name)))
774                        (alien-args `(addr ,name))))
775                 (arg-types arg-type)
776                 (unless (eq style :out)
777                   (lisp-args name)
778                   (lisp-arg-types t
779                                   ;; FIXME: It should be something
780                                   ;; like `(ALIEN ,ARG-TYPE), except
781                                   ;; for we also accept SAPs where
782                                   ;; pointers are required.
783                                   )))
784               (when (or (eq style :out) (eq style :in-out))
785                 (results name)
786                 (lisp-result-types `(alien ,type))))))
787       `(progn
788          ;; The theory behind this automatic DECLAIM is that (1) if
789          ;; you're calling C, static typing is what you're doing
790          ;; anyway, and (2) such a declamation can be (especially for
791          ;; alien values) both messy to do by hand and very important
792          ;; for performance of later code which uses the return value.
793          (declaim (ftype (function ,(lisp-arg-types)
794                                    (values ,@(lisp-result-types) &optional))
795                          ,lisp-name))
796          (defun ,lisp-name ,(lisp-args)
797            ,@(docs)
798            (with-alien
799             ((,lisp-name (function ,result-type ,@(arg-types))
800                          :extern ,alien-name)
801              ,@(alien-vars))
802              ,@(if (eq 'void result-type)
803                    `((alien-funcall ,lisp-name ,@(alien-args))
804                      (values nil ,@(results)))
805                    `((values (alien-funcall ,lisp-name ,@(alien-args))
806                              ,@(results))))))))))
807 \f
808 (defun alien-typep (object type)
809   #!+sb-doc
810   "Return T iff OBJECT is an alien of type TYPE."
811   (let ((lisp-rep-type (compute-lisp-rep-type type)))
812     (if lisp-rep-type
813         (typep object lisp-rep-type)
814         (and (alien-value-p object)
815              (alien-subtype-p (alien-value-type object) type)))))
816
817 (defun alien-value-typep (object type)
818   (when (alien-value-p object)
819     (alien-subtype-p (alien-value-type object) type)))
820
821 ;;;; ALIEN CALLBACKS
822 ;;;;
823 ;;;; See "Foreign Linkage / Callbacks" in the SBCL Internals manual.
824
825 (defvar *alien-callback-info* nil
826   "Maps SAPs to corresponding CALLBACK-INFO structures: contains all the
827 information we need to manipulate callbacks after their creation. Used for
828 changing the lisp-side function they point to, invalidation, etc.")
829
830 (defstruct callback-info
831   specifier
832   function ; NULL if invalid
833   wrapper
834   index)
835
836 (defun callback-info-key (info)
837   (cons (callback-info-specifier info) (callback-info-function info)))
838
839 (defun alien-callback-info (alien)
840   (cdr (assoc (alien-sap alien) *alien-callback-info* :test #'sap=)))
841
842 (defvar *alien-callbacks* (make-hash-table :test #'equal)
843   "Cache of existing callback SAPs, indexed with (SPECIFER . FUNCTION). Used for
844 memoization: we don't create new callbacks if one pointing to the correct
845 function with the same specifier already exists.")
846
847 (defvar *alien-callback-wrappers* (make-hash-table :test #'equal)
848   "Cache of existing lisp weappers, indexed with SPECIFER. Used for memoization:
849 we don't create new wrappers if one for the same specifier already exists.")
850
851 (defvar *alien-callback-trampolines* (make-array 32 :fill-pointer 0 :adjustable t)
852   "Lisp trampoline store: assembler wrappers contain indexes to this, and
853 ENTER-ALIEN-CALLBACK pulls the corresponsing trampoline out and calls it.")
854
855 (defun %alien-callback-sap (specifier result-type argument-types function wrapper)
856   (let ((key (cons specifier function)))
857     (or (gethash key *alien-callbacks*)
858         (setf (gethash key *alien-callbacks*)
859               (let* ((index (fill-pointer *alien-callback-trampolines*))
860                      ;; Aside from the INDEX this is known at
861                      ;; compile-time, which could be utilized by
862                      ;; having the two-stage assembler tramp &
863                      ;; wrapper mentioned in [1] above: only the
864                      ;; per-function tramp would need assembler at
865                      ;; runtime. Possibly we could even pregenerate
866                      ;; the code and just patch the index in later.
867                      (assembler-wrapper (alien-callback-assembler-wrapper
868                                          index result-type argument-types)))
869                 (vector-push-extend
870                  (alien-callback-lisp-trampoline wrapper function)
871                  *alien-callback-trampolines*)
872                 ;; Assembler-wrapper is static, so sap-taking is safe.
873                 (let ((sap (vector-sap assembler-wrapper)))
874                   (push (cons sap (make-callback-info :specifier specifier
875                                                       :function function
876                                                       :wrapper wrapper
877                                                       :index index))
878                         *alien-callback-info*)
879                   sap))))))
880
881 (defun alien-callback-lisp-trampoline (wrapper function)
882   (declare (function wrapper) (optimize speed))
883   (lambda (args-pointer result-pointer)
884     (funcall wrapper args-pointer result-pointer function)))
885
886 (defun alien-callback-lisp-wrapper-lambda (specifier result-type argument-types env)
887   (let* ((arguments (make-gensym-list (length argument-types)))
888          (argument-names arguments)
889          (argument-specs (cddr specifier)))
890     `(lambda (args-pointer result-pointer function)
891        ;; FIXME: the saps are not gc safe
892        (let ((args-sap (int-sap
893                         (sb!kernel:get-lisp-obj-address args-pointer)))
894              (res-sap (int-sap
895                        (sb!kernel:get-lisp-obj-address result-pointer))))
896          (declare (ignorable args-sap res-sap))
897          (with-alien
898              ,(loop
899                  with offset = 0
900                  for spec in argument-specs
901                  collect `(,(pop argument-names) ,spec
902                             :local ,(alien-callback-accessor-form
903                                      spec 'args-sap offset))
904                  do (incf offset (alien-callback-argument-bytes spec env)))
905            ,(flet ((store (spec real-type)
906                           (if spec
907                               `(setf (deref (sap-alien res-sap (* ,spec)))
908                                      ,(if real-type
909                                           `(the ,real-type
910                                              (funcall function ,@arguments))
911                                           `(funcall function ,@arguments)))
912                               `(funcall function ,@arguments))))
913                   (cond ((alien-void-type-p result-type)
914                          (store nil nil))
915                         ((alien-integer-type-p result-type)
916                          ;; Integer types should be padded out to a full
917                          ;; register width, to comply with most ABI calling
918                          ;; conventions, but should be typechecked on the
919                          ;; declared type width, hence the following:
920                          (if (alien-integer-type-signed result-type)
921                              (store `(signed
922                                       ,(alien-type-word-aligned-bits result-type))
923                                     `(signed-byte ,(alien-type-bits result-type)))
924                              (store
925                               `(unsigned
926                                 ,(alien-type-word-aligned-bits result-type))
927                               `(unsigned-byte ,(alien-type-bits result-type)))))
928                         (t
929                          (store (unparse-alien-type result-type) nil))))))
930        (values))))
931
932 (defun invalid-alien-callback (&rest arguments)
933   (declare (ignore arguments))
934   (error "Invalid alien callback called."))
935
936
937 (defun parse-callback-specification (result-type lambda-list)
938   (values
939    `(function ,result-type ,@(mapcar #'second lambda-list))
940    (mapcar #'first lambda-list)))
941
942
943 (defun parse-alien-ftype (specifier env)
944   (destructuring-bind (function result-type &rest argument-types)
945       specifier
946     (aver (eq 'function function))
947     (values (let ((*values-type-okay* t))
948               (parse-alien-type result-type env))
949             (mapcar (lambda (spec)
950                       (parse-alien-type spec env))
951                     argument-types))))
952
953 (defun alien-void-type-p (type)
954   (and (alien-values-type-p type) (not (alien-values-type-values type))))
955
956 (defun alien-type-word-aligned-bits (type)
957   (align-offset (alien-type-bits type) sb!vm:n-word-bits))
958
959 (defun alien-callback-argument-bytes (spec env)
960   (let ((type (parse-alien-type spec env)))
961     (if (or (alien-integer-type-p type)
962             (alien-float-type-p type)
963             (alien-pointer-type-p type)
964             (alien-system-area-pointer-type-p type))
965         (ceiling (alien-type-word-aligned-bits type) sb!vm:n-byte-bits)
966         (error "Unsupported callback argument type: ~A" type))))
967
968 (defun enter-alien-callback (index return arguments)
969   (funcall (aref *alien-callback-trampolines* index)
970            return
971            arguments))
972
973 ;;; To ensure that callback wrapper functions continue working even
974 ;;; if #'ENTER-ALIEN-CALLBACK moves in memory, access to it is indirected
975 ;;; through the *ENTER-ALIEN-CALLBACK* static symbol. -- JES, 2006-01-01
976 (defvar *enter-alien-callback* #'enter-alien-callback)
977
978 ;;;; interface (not public, yet) for alien callbacks
979
980 (defmacro alien-callback (specifier function &environment env)
981   "Returns an alien-value with of alien ftype SPECIFIER, that can be passed to
982 an alien function as a pointer to the FUNCTION. If a callback for the given
983 SPECIFIER and FUNCTION already exists, it is returned instead of consing a new
984 one."
985   ;; Pull out as much work as is convenient to macro-expansion time, specifically
986   ;; everything that can be done given just the SPECIFIER and ENV.
987   (multiple-value-bind (result-type argument-types) (parse-alien-ftype specifier env)
988     `(%sap-alien
989       (%alien-callback-sap ',specifier ',result-type ',argument-types
990                            ,function
991                            (or (gethash ',specifier *alien-callback-wrappers*)
992                                (setf (gethash ',specifier *alien-callback-wrappers*)
993                                      (compile nil
994                                               ',(alien-callback-lisp-wrapper-lambda
995                                                  specifier result-type argument-types env)))))
996       ',(parse-alien-type specifier env))))
997
998 (defun alien-callback-p (alien)
999   "Returns true if the alien is associated with a lisp-side callback,
1000 and a secondary return value of true if the callback is still valid."
1001   (let ((info (alien-callback-info alien)))
1002     (when info
1003       (values t (and (callback-info-function info) t)))))
1004
1005 (defun alien-callback-function (alien)
1006   "Returns the lisp function designator associated with the callback."
1007   (let ((info (alien-callback-info alien)))
1008     (when info
1009       (callback-info-function info))))
1010
1011 (defun (setf alien-callback-function) (function alien)
1012   "Changes the lisp function designated by the callback."
1013   (let ((info (alien-callback-info alien)))
1014     (unless info
1015       (error "Not an alien callback: ~S" alien))
1016     ;; sap cache
1017     (let ((key (callback-info-key info)))
1018       (remhash key *alien-callbacks*)
1019       (setf (gethash key *alien-callbacks*) (alien-sap alien)))
1020     ;; trampoline
1021     (setf (aref *alien-callback-trampolines* (callback-info-index info))
1022           (alien-callback-lisp-trampoline (callback-info-wrapper info) function))
1023     ;; metadata
1024     (setf (callback-info-function info) function)
1025     function))
1026
1027 (defun invalidate-alien-callback (alien)
1028   "Invalidates the callback designated by the alien, if any, allowing the
1029 associated lisp function to be GC'd, and causing further calls to the same
1030 callback signal an error."
1031   (let ((info (alien-callback-info alien)))
1032     (when (and info (callback-info-function info))
1033       ;; sap cache
1034       (remhash (callback-info-key info) *alien-callbacks*)
1035       ;; trampoline
1036       (setf (aref *alien-callback-trampolines* (callback-info-index info))
1037             #'invalid-alien-callback)
1038       ;; metadata
1039       (setf (callback-info-function info) nil)
1040       t)))
1041
1042 ;;; FIXME: This call assembles a new callback for every closure,
1043 ;;; which sucks hugely. ...not that I can think of an obvious
1044 ;;; solution. Possibly maybe we could write a generalized closure
1045 ;;; callback analogous to closure_tramp, and share the actual wrapper?
1046 ;;;
1047 ;;; For lambdas that result in simple-funs we get the callback from
1048 ;;; the cache on subsequent calls.
1049 (defmacro alien-lambda (result-type typed-lambda-list &body forms)
1050   (multiple-value-bind (specifier lambda-list)
1051       (parse-callback-specification result-type typed-lambda-list)
1052     `(alien-callback ,specifier (lambda ,lambda-list ,@forms))))
1053
1054 ;;; FIXME: Should subsequent (SETF FDEFINITION) affect the callback or not?
1055 ;;; What about subsequent DEFINE-ALIEN-CALLBACKs? My guess is that changing
1056 ;;; the FDEFINITION should invalidate the callback, and redefining the
1057 ;;; callback should change existing callbacks to point to the new defintion.
1058 (defmacro define-alien-callback (name result-type typed-lambda-list &body forms)
1059   "Defines #'NAME as a function with the given body and lambda-list, and NAME as
1060 the alien callback for that function with the given alien type."
1061   (declare (symbol name))
1062   (multiple-value-bind (specifier lambda-list)
1063       (parse-callback-specification result-type typed-lambda-list)
1064     `(progn
1065        (defun ,name ,lambda-list ,@forms)
1066        (defparameter ,name (alien-callback ,specifier #',name)))))