;;; -*- Package: x86 -*-
;;;
;;; **********************************************************************
;;; This code was written as part of the CMU Common Lisp project at
;;; Carnegie Mellon University, and has been placed in the public domain.
;;; If you want to use this code or any part of CMU Common Lisp, please contact
;;; Scott Fahlman or slisp-group@cs.cmu.edu.
;;;
(ext:file-comment
  "$Header$")
;;;
;;; **********************************************************************
;;; 
;;; This file contains the machine specific support routines needed by
;;; the file assembler.
;;;
;;; Written by William Lott
;;;
(in-package :x86)


;;;; Return-multiple

;;; For return-multiple, we have to move the results from the end of the
;;; frame for the function that is returning to the end of the frame for
;;; the function being returned to.

#+assembler ;; we don't want a vop for this one.
(define-assembly-routine
    (return-multiple (:return-style :none))
    (;; These four are really arguments.
     (:temp eax dword-reg eax-offset)
     (:temp ebx dword-reg ebx-offset)
     (:temp ecx dword-reg ecx-offset)
     (:temp esi dword-reg esi-offset)

     ;; These we need as temporaries.
     (:temp edx dword-reg edx-offset)
     (:temp edi dword-reg edi-offset))
     
  ;; Pick off the cases where everything fits in regiser args.
  (inst jecxz zero-values)
  (inst cmp ecx (fixnum 1))
  (inst jmp :e one-value)
  (inst cmp ecx (fixnum 2))
  (inst jmp :e two-values)
  (inst cmp ecx (fixnum 3))
  (inst jmp :e three-values)

  ;; Save the count, because the loop is going to destroy it.
  (inst mov edx ecx)

  ;; Blit the values down the stack.  Note: there might be overlap, so we have
  ;; to be careful not to clobber values before we've read them.  Because the
  ;; stack builds down, we are coping to a larger address.  Therefore, we need
  ;; to iterate from larger addresses to smaller addresses.
  (inst std)
  (inst sub esi 4)
  (inst lea edi (make-ea :dword :base ebx :disp (- word-btyes)))
  (inst rep)
  (inst movsd)

  ;; Set the stack top to after the last result.
  (inst mov esp-tn edi)

  ;; Restore the count.
  (inst mov ecx edx)

  ;; Load the register args.
  (loadw edx ebx -1)
  (loadw edi ebx -2)
  (loadw esi ebx -3)

  ;; And back we go.
  (inst jmp-near eax)

  ;; Handle the register arg cases.
  ZERO-VALUES
  (move esp-tn ebx)
  (inst mov edx nil-value)
  (inst mov edi edx)
  (inst mov esi edx)
  (inst jmp-near eax)

  ONE-VALUE ; Note: we can get this, because the return-multiple vop
	    ; doesn't check for this case when size > speed.
  (loadw edx esi -1)
  (inst mov esp-tn ebx)
  (inst add eax 2)
  (inst jmp-near eax)

  TWO-VALUES
  (loadw edx esi -1)
  (loadw edi esi -2)
  (inst mov esi nil-value)
  (inst lea esp-tn (make-ea :dword :base ebx :disp (* -2 word-bytes)))
  (inst jmp-near eax)

  THREE-VALUES
  (loadw edx esi -1)
  (loadw edi esi -2)
  (loadw esi esi -3)
  (inst lea esp-tn (make-ea :dword :base ebx :disp (* -2 word-bytes)))
  (inst jmp-near eax))



;;;; tail-call-variable.

;;; For tail-call-variable, we have to copy the arguments from the end of our
;;; stack frame (were args are produced) to the start of our stack frame
;;; (were args are expected).
;;;
;;; We take the function to call in EAX and a pointer to the arguments in
;;; ESI.  EBP says the same over the jump, and the old frame pointer is
;;; still saved in the first stack slot.  The return-pc is saved in
;;; the second stack slot, so we have to push it to make it look like
;;; we actually called.  We also have to compute ECX from the difference
;;; between ESI and the stack top.
;;; 
#+assembler ;; no vop for this one either.
(define-assembly-routine
    (tail-call-variable
     (:return-style :none))

    ((:temp eax dword-reg eax-offset)
     (:temp ebx dword-reg ebx-offset)
     (:temp ecx dword-reg ecx-offset)
     (:temp edx dword-reg edx-offset)
     (:temp edi dword-reg edi-offset)
     (:temp esi dword-reg esi-offset))

  ;; Calculate NARGS (as a fixnum)
  (move ecx esi)
  (inst sub exc esp-tn)

  ;; Check for all the args fitting the the registers.
  (inst cmp ecx (fixnum 3))
  (inst jmp :le REGISTER-ARGS)

  ;; Save the OLD-FP and RETURN-PC because the blit it going to trash
  ;; those stack locations.  Save the ECX, because the loop is going
  ;; to trash it.
  (pushw ebp-tn -1)
  (loadw ebx ebp-tn -2)
  (inst push ecx)

  ;; Do the blit.  Because we are coping from smaller addresses to larger
  ;; addresses, we have to start at the largest pair and work our way down.
  (inst std)
  (inst lea edi (make-ea :dword :base ebp-tn :disp (- word-bytes)))
  (inst sub esi 4)
  (inst rep)
  (inst movsd)

  ;; Restore OLD-FP and ECX.
  (inst pop ecx)
  (popw ebp-tn -1)

  ;; Blow off the stack above the arguments.
  (inst lea esp-tn (make-ea :dword :base edi :disp word-bytes))

  ;; Push the (saved) return-pc so it looks like we just called.
  (inst push ebx)

  ;; Load the register arguments.
  (loadw edx ebp-tn -1)
  (loadw edi ebp-tn -2)
  (loadw esi ebp-tn -3)

  ;; And jump into the function.
  (inst jmp-indirect
	(make-ea :byte :base eax :disp (- (* closure-function-slot word-bytes)
					  function-pointer-type)))

  ;; All the arguments fit in registers, so load them.
  REGISTER-ARGS
  (loadw edx esi -1)
  (loadw edi esi -2)
  (loadw esi esi -2)

  ;; Clear most of the stack.
  (inst lea esp-tn
	(make-ea :dword :base ebp-tn :disp (* -3 word-bytes)))

  ;; Push the return-pc so it looks like we just called.
  (pushw ebp-tn -2)

  ;; And away we go.
  (inst jmp-indirect
	(make-ea :byte :base eax :disp (- (* closure-function-slot word-bytes)
					  function-pointer-type))))


;;;; Non-local exit noise.

(define-assembly-routine (unwind
			  (:return-style :none)
			  (:translate %continue-unwind)
			  (:policy :fast-safe))
			 ((:arg block (any-reg descriptor-reg) eax-offset)
			  (:arg start (any-reg descriptor-reg) ebx-offset)
			  (:arg count (any-reg descriptor-reg) ecx-offset)
			  (:temp uwp dword-reg esi-offset))
  (declare (ignore start count))

  (let ((error (generate-error-code nil invalid-unwind-error)))
    (inst or block block)
    (inst jmp :z error))
  
  (load-symbol-value uwp lisp::*current-unwind-protect-block*)
  (inst cmp uwp
	(make-ea :dword :base block
		 :disp (* unwind-block-current-uwp-slot word-bytes)))
  (inst jmp :e do-exit)
  (move block uwp)
  (loadw uwp uwp unwind-block-current-uwp-slot)
  (store-symbol-value uwp lisp::*current-unwind-protect-block*)
  DO-EXIT
  (loadw ebp-tn block vm:unwind-block-current-cont-slot)
  (inst jmp-indirect
	(make-ea :byte :base block
		 :disp (* unwind-block-entry-pc-slot word-bytes))))

(define-assembly-routine (throw
			  (:return-style :none))
			 ((:arg target (descriptor-reg any-reg) edx-offset)
			  (:arg start any-reg ebx-offset)
			  (:arg count any-reg ecx-offset)
			  (:temp catch any-reg eax-offset))
  
  (declare (ignore start count))

  (load-symbol-value catch lisp::*current-catch-block*)
  
  LOOP
  
  (let ((error (generate-error-code nil unseen-throw-tag-error target)))
    (inst or catch catch)
    (inst jmp :z error))
  
  (inst cmp target
	(make-ea :dword :base catch
		 :disp (* catch-block-tag-slot word-bytes)))
  (inst jmp :e exit)
  (loadw catch catch vm:catch-block-previous-catch-slot)
  (inst jmp loop)
  
  EXIT
  
  (inst jmp-near (make-fixup 'unwind :assembly-routine)))