X-Git-Url: http://repo.macrolet.net/gitweb/?a=blobdiff_plain;f=src%2Fcode%2Frun-program.lisp;h=cb6c489033f8f1a754eaecc863895dd0ccb21799;hb=23c0c48f562d7dc5d1615bf13cb831b46c91d106;hp=537b367fc288e558adbda8b2cd9ea4ea01e740fc;hpb=3eb0a28fe6a7912d6ff2b97221325c0e3bfc5703;p=sbcl.git diff --git a/src/code/run-program.lisp b/src/code/run-program.lisp index 537b367..cb6c489 100644 --- a/src/code/run-program.lisp +++ b/src/code/run-program.lisp @@ -141,7 +141,6 @@ (not (zerop (ldb (byte 1 7) status))))))))) ;;;; process control stuff -#-win32 (defvar *active-processes* nil #+sb-doc "List of process structures for all active processes.") @@ -153,11 +152,13 @@ ;;; *ACTIVE-PROCESSES* can be accessed from multiple threads so a ;;; mutex is needed. More importantly the sigchld signal handler also ;;; accesses it, that's why we need without-interrupts. -#-win32 (defmacro with-active-processes-lock (() &body body) + #-win32 `(without-interrupts (sb-thread:with-mutex (*active-processes-lock*) - ,@body))) + ,@body)) + #+win32 + `(progn ,@body)) (defstruct (process (:copier nil)) pid ; PID of child process @@ -187,12 +188,16 @@ #+sb-doc (setf (documentation 'process-pid 'function) "The pid of the child process.") +#+win32 +(define-alien-routine ("GetExitCodeProcess@8" get-exit-code-process) + int + (handle unsigned) (exit-code unsigned :out)) + (defun process-status (process) #+sb-doc "Return the current status of PROCESS. The result is one of :RUNNING, :STOPPED, :EXITED, or :SIGNALED." - #-win32 - (get-processes-status-changes) + (get-processes-status-changes) (process-%status process)) #+sb-doc @@ -228,12 +233,11 @@ The function is called with PROCESS as its only argument.") (setf (documentation 'process-plist 'function) "A place for clients to stash things.") -#-win32 (defun process-wait (process &optional check-for-stopped) #+sb-doc - "Wait for PROCESS to quit running for some reason. - When CHECK-FOR-STOPPED is T, also returns when PROCESS is - stopped. Returns PROCESS." + "Wait for PROCESS to quit running for some reason. When +CHECK-FOR-STOPPED is T, also returns when PROCESS is stopped. Returns +PROCESS." (loop (case (process-status process) (:running) @@ -298,7 +302,6 @@ The function is called with PROCESS as its only argument.") (t t))))) -#-win32 (defun process-alive-p (process) #+sb-doc "Return T if PROCESS is still alive, NIL otherwise." @@ -308,16 +311,19 @@ The function is called with PROCESS as its only argument.") t nil))) -#-win32 (defun process-close (process) #+sb-doc - "Close all streams connected to PROCESS and stop maintaining the status slot." + "Close all streams connected to PROCESS and stop maintaining the +status slot." (macrolet ((frob (stream abort) `(when ,stream (close ,stream :abort ,abort)))) - (frob (process-pty process) t) ; Don't FLUSH-OUTPUT to dead process, .. - (frob (process-input process) t) ; .. 'cause it will generate SIGPIPE. + #-win32 + (frob (process-pty process) t) ; Don't FLUSH-OUTPUT to dead process, + (frob (process-input process) t) ; .. 'cause it will generate SIGPIPE. (frob (process-output process) nil) - (frob (process-error process) nil)) + (frob (process-error process) nil)) + ;; FIXME: Given that the status-slot is no longer updated, + ;; maybe it should be set to :CLOSED, or similar? (with-active-processes-lock () (setf *active-processes* (delete process *active-processes*))) process) @@ -328,25 +334,47 @@ The function is called with PROCESS as its only argument.") (declare (ignore ignore1 ignore2 ignore3)) (get-processes-status-changes)) -#-win32 (defun get-processes-status-changes () + #-win32 (loop - (multiple-value-bind (pid what code core) - (wait3 t t) - (unless pid - (return)) - (let ((proc (with-active-processes-lock () - (find pid *active-processes* :key #'process-pid)))) - (when proc - (setf (process-%status proc) what) - (setf (process-exit-code proc) code) - (setf (process-core-dumped proc) core) - (when (process-status-hook proc) - (funcall (process-status-hook proc) proc)) - (when (position what #(:exited :signaled)) - (with-active-processes-lock () - (setf *active-processes* - (delete proc *active-processes*))))))))) + (multiple-value-bind (pid what code core) + (wait3 t t) + (unless pid + (return)) + (let ((proc (with-active-processes-lock () + (find pid *active-processes* :key #'process-pid)))) + (when proc + (setf (process-%status proc) what) + (setf (process-exit-code proc) code) + (setf (process-core-dumped proc) core) + (when (process-status-hook proc) + (funcall (process-status-hook proc) proc)) + (when (position what #(:exited :signaled)) + (with-active-processes-lock () + (setf *active-processes* + (delete proc *active-processes*)))))))) + #+win32 + (let (exited) + (with-active-processes-lock () + (setf *active-processes* + (delete-if (lambda (proc) + (multiple-value-bind (ok code) + (get-exit-code-process (process-pid proc)) + (when (and (plusp ok) (/= code 259)) + (setf (process-%status proc) :exited + (process-exit-code proc) code) + (when (process-status-hook proc) + (push proc exited)) + t))) + *active-processes*))) + ;; Can't call the hooks before all the processes have been deal + ;; with, as calling a hook may cause re-entry to + ;; GET-PROCESS-STATUS-CHANGES. That may be OK when using wait3, + ;; but in the Windows implementation is would be deeply bad. + (dolist (proc exited) + (let ((hook (process-status-hook proc))) + (when hook + (funcall hook proc)))))) ;;;; RUN-PROGRAM and close friends @@ -731,7 +759,7 @@ argument. ARGS are the standard arguments that can be passed to a program. For no arguments, use NIL (which means that just the name of the program is passed as arg 0). -RUN-PROGRAM will either return a PROCESS structure. See the CMU +RUN-PROGRAM will return a PROCESS structure. See the CMU Common Lisp Users Manual for details about the PROCESS structure. The &KEY arguments have the following meanings: @@ -744,7 +772,7 @@ Common Lisp Users Manual for details about the PROCESS structure. NIL, continue running Lisp until the program finishes. :INPUT Either T, NIL, a pathname, a stream, or :STREAM. If T, the standard - input for the current process is inherited. If NIL, /dev/null + input for the current process is inherited. If NIL, nul is used. If a pathname, the file so specified is used. If a stream, all the input is read from that stream and send to the subprocess. If :STREAM, the PROCESS-INPUT slot is filled in with a stream that sends @@ -756,7 +784,7 @@ Common Lisp Users Manual for details about the PROCESS structure. NIL (the default) to return NIL from RUN-PROGRAM :OUTPUT Either T, NIL, a pathname, a stream, or :STREAM. If T, the standard - output for the current process is inherited. If NIL, /dev/null + output for the current process is inherited. If NIL, nul is used. If a pathname, the file so specified is used. If a stream, all the output from the process is written to this stream. If :STREAM, the PROCESS-OUTPUT slot is filled in with a stream that can @@ -818,7 +846,13 @@ Common Lisp Users Manual for details about the PROCESS structure. (error "Couldn't spawn program: ~A" (strerror))) (setf proc (if wait - (make-process :%status :exited + (make-process :pid handle + :%status :exited + :input input-stream + :output output-stream + :error error-stream + :status-hook status-hook + :cookie cookie :exit-code handle) (make-process :pid handle :%status :running @@ -826,7 +860,14 @@ Common Lisp Users Manual for details about the PROCESS structure. :output output-stream :error error-stream :status-hook status-hook - :cookie cookie)))))))))) + :cookie cookie))) + (push proc *active-processes*))))))) + (dolist (fd *close-in-parent*) + (sb-unix:unix-close fd))) + (unless proc + (dolist (fd *close-on-error*) + (sb-unix:unix-close fd))) + proc)) ;;; Install a handler for any input that shows up on the file @@ -888,6 +929,19 @@ Common Lisp Users Manual for details about the PROCESS structure. (write-string string stream :end count))))))))))) +(defun get-stream-fd (stream direction) + (typecase stream + (sb-sys:fd-stream + (values (sb-sys:fd-stream-fd stream) nil)) + (synonym-stream + (get-stream-fd (symbol-value (synonym-stream-symbol stream)) direction)) + (two-way-stream + (ecase direction + (:input + (get-stream-fd (two-way-stream-input-stream stream) direction)) + (:output + (get-stream-fd (two-way-stream-output-stream stream) direction)))))) + ;;; Find a file descriptor to use for object given the direction. ;;; Returns the descriptor. If object is :STREAM, returns the created ;;; stream as the second value. @@ -911,7 +965,8 @@ Common Lisp Users Manual for details about the PROCESS structure. (t sb-unix:o_rdwr)) #o666) (unless fd - (error "~@" + (error #-win32 "~@" + #+win32 "~@" (strerror errno))) (push fd *close-in-parent*) (values fd nil))) @@ -948,56 +1003,56 @@ Common Lisp Users Manual for details about the PROCESS structure. (t (error "couldn't duplicate file descriptor: ~A" (strerror errno))))))) - ((sb-sys:fd-stream-p object) - (values (sb-sys:fd-stream-fd object) nil)) ((streamp object) (ecase direction (:input - ;; FIXME: We could use a better way of setting up - ;; temporary files, both here and in LOAD-FOREIGN. - (dotimes (count - 256 - (error "could not open a temporary file in /tmp")) - (let* ((name (coerce (format nil "/tmp/.run-program-~D" count) - 'base-string)) - (fd (sb-unix:unix-open name - (logior sb-unix:o_rdwr - sb-unix:o_creat - sb-unix:o_excl) - #o666))) - (sb-unix:unix-unlink name) - (when fd - (let ((newline (string #\Newline))) - (loop - (multiple-value-bind - (line no-cr) - (read-line object nil nil) - (unless line - (return)) - (sb-unix:unix-write - fd - ;; FIXME: this really should be - ;; (STRING-TO-OCTETS :EXTERNAL-FORMAT ...). - ;; RUN-PROGRAM should take an - ;; external-format argument, which should - ;; be passed down to here. Something - ;; similar should happen on :OUTPUT, too. - (map '(vector (unsigned-byte 8)) #'char-code line) - 0 (length line)) - (if no-cr - (return) - (sb-unix:unix-write fd newline 0 1))))) - (sb-unix:unix-lseek fd 0 sb-unix:l_set) - (push fd *close-in-parent*) - (return (values fd nil)))))) + (or (get-stream-fd object :input) + ;; FIXME: We could use a better way of setting up + ;; temporary files + (dotimes (count + 256 + (error "could not open a temporary file in /tmp")) + (let* ((name (coerce (format nil "/tmp/.run-program-~D" count) + 'base-string)) + (fd (sb-unix:unix-open name + (logior sb-unix:o_rdwr + sb-unix:o_creat + sb-unix:o_excl) + #o666))) + (sb-unix:unix-unlink name) + (when fd + (let ((newline (string #\Newline))) + (loop + (multiple-value-bind + (line no-cr) + (read-line object nil nil) + (unless line + (return)) + (sb-unix:unix-write + fd + ;; FIXME: this really should be + ;; (STRING-TO-OCTETS :EXTERNAL-FORMAT ...). + ;; RUN-PROGRAM should take an + ;; external-format argument, which should + ;; be passed down to here. Something + ;; similar should happen on :OUTPUT, too. + (map '(vector (unsigned-byte 8)) #'char-code line) + 0 (length line)) + (if no-cr + (return) + (sb-unix:unix-write fd newline 0 1))))) + (sb-unix:unix-lseek fd 0 sb-unix:l_set) + (push fd *close-in-parent*) + (return (values fd nil))))))) (:output - (multiple-value-bind (read-fd write-fd) - (sb-unix:unix-pipe) - (unless read-fd - (error "couldn't create pipe: ~S" (strerror write-fd))) - (copy-descriptor-to-stream read-fd object cookie) - (push read-fd *close-on-error*) - (push write-fd *close-in-parent*) - (values write-fd nil))))) + (or (get-stream-fd object :output) + (multiple-value-bind (read-fd write-fd) + (sb-unix:unix-pipe) + (unless read-fd + (error "couldn't create pipe: ~S" (strerror write-fd))) + (copy-descriptor-to-stream read-fd object cookie) + (push read-fd *close-on-error*) + (push write-fd *close-in-parent*) + (values write-fd nil)))))) (t (error "invalid option to RUN-PROGRAM: ~S" object))))