2016-06-06 23:54:23 +01:00
|
|
|
(define-module (sh pipe)
|
|
|
|
:use-module (ice-9 popen)
|
2016-10-15 23:03:45 +01:00
|
|
|
:use-module (srfi srfi-1)
|
2016-06-06 23:54:23 +01:00
|
|
|
:use-module (srfi srfi-8)
|
2016-10-15 23:03:45 +01:00
|
|
|
:use-module (srfi srfi-9)
|
|
|
|
:use-module (srfi srfi-26)
|
2016-10-18 08:15:53 +01:00
|
|
|
:export (pipeline job-control-init))
|
|
|
|
|
|
|
|
(define (stdout . o)
|
|
|
|
(map display o)
|
|
|
|
(newline))
|
|
|
|
|
|
|
|
(define (stderr . o)
|
|
|
|
(map (cut display <> (current-error-port)) o)
|
|
|
|
(newline))
|
|
|
|
|
|
|
|
(define-record-type <process>
|
|
|
|
(make-process pid command status)
|
|
|
|
process?
|
|
|
|
(pid process-pid)
|
|
|
|
(command process-command)
|
|
|
|
(status process-status set-process-status!)) ;; '(running stopped completed terminated)
|
|
|
|
|
|
|
|
(define-record-type <job>
|
|
|
|
(make-job id pgid processes)
|
|
|
|
job?
|
|
|
|
(id job-id)
|
|
|
|
(pgid job-pgid set-job-pgid!)
|
|
|
|
(processes job-processes set-job-processes!))
|
|
|
|
|
|
|
|
(define job-table '()) ;; list of <job>
|
|
|
|
|
|
|
|
;; (define (job-at index)
|
|
|
|
;; (let ((len (length job-table)))
|
|
|
|
;; (if (or (> index len) (< index 0)) #f
|
|
|
|
;; (list-ref job-table (- len index)))))
|
|
|
|
|
|
|
|
(define (add-to-process-group job pid)
|
|
|
|
(let* ((pgid (job-pgid job))
|
2016-10-18 22:11:31 +01:00
|
|
|
(pgid (or pgid pid)))
|
2016-10-18 08:15:53 +01:00
|
|
|
(setpgid pid pgid)
|
|
|
|
pgid))
|
|
|
|
|
|
|
|
(define (job-add-process job pid command)
|
|
|
|
(let ((pgid (add-to-process-group job pid)))
|
|
|
|
(set-job-pgid! job pgid)
|
2016-10-18 22:11:31 +01:00
|
|
|
(tcsetpgrp (current-error-port) pgid)
|
|
|
|
(set-job-processes! job (cons (make-process pid command 'running) (job-processes job)))))
|
2016-10-18 08:15:53 +01:00
|
|
|
|
|
|
|
(define (job-control-init)
|
2016-10-18 22:11:31 +01:00
|
|
|
(let* ((interactive? (isatty? (current-error-port)))
|
2016-10-18 08:15:53 +01:00
|
|
|
(pgid (getpgrp))
|
|
|
|
(pid (getpid)))
|
|
|
|
(when interactive?
|
2016-10-18 22:11:31 +01:00
|
|
|
(while (not (eqv? (tcgetpgrp (current-error-port)) pgid))
|
2016-10-18 08:15:53 +01:00
|
|
|
(kill (- pgid) SIGTTIN)) ;; we are not in the foreground
|
|
|
|
(map (cut sigaction <> SIG_IGN)
|
|
|
|
(list SIGINT SIGQUIT SIGTSTP SIGTTIN SIGTTOU))
|
|
|
|
(sigaction SIGCHLD SIG_DFL)
|
|
|
|
(setpgid pid pid) ;; create new process group for ourself
|
2016-10-18 22:11:31 +01:00
|
|
|
(tcsetpgrp (current-error-port) pid))))
|
2016-05-30 23:13:04 +01:00
|
|
|
|
|
|
|
(define (pipe*)
|
|
|
|
(let ((p (pipe)))
|
|
|
|
(values (car p) (cdr p))))
|
|
|
|
|
2016-10-15 23:03:45 +01:00
|
|
|
;; lhs rhs
|
2016-05-30 23:13:04 +01:00
|
|
|
;; [source] w -> r [filter] w -> r [sink]
|
|
|
|
|
2016-10-15 23:03:45 +01:00
|
|
|
(define (exec* command) ;; list of strings
|
2016-05-30 23:13:04 +01:00
|
|
|
(apply execlp (cons (car command) command)))
|
|
|
|
|
2016-10-18 08:15:53 +01:00
|
|
|
(define (spawn-source job interactive? command)
|
2016-05-30 23:13:04 +01:00
|
|
|
(receive (r w) (pipe*)
|
|
|
|
(let ((pid (primitive-fork)))
|
|
|
|
(cond ((= 0 pid) (close r)
|
2016-10-18 22:11:31 +01:00
|
|
|
(tcsetpgrp (current-error-port) (add-to-process-group job (getpid)))
|
2016-05-30 23:13:04 +01:00
|
|
|
(move->fdes w 1)
|
|
|
|
(exec* command))
|
|
|
|
(#t
|
2016-10-18 08:15:53 +01:00
|
|
|
(job-add-process job pid command)
|
2016-05-30 23:13:04 +01:00
|
|
|
(close w)
|
|
|
|
r)))))
|
|
|
|
|
2016-10-18 08:15:53 +01:00
|
|
|
(define (spawn-filter job interactive? src command)
|
2016-05-30 23:13:04 +01:00
|
|
|
(receive (r w) (pipe*)
|
|
|
|
(let ((pid (primitive-fork)))
|
|
|
|
(cond ((= 0 pid)
|
2016-10-18 22:11:31 +01:00
|
|
|
(tcsetpgrp (current-error-port) (add-to-process-group job (getpid)))
|
2016-05-30 23:13:04 +01:00
|
|
|
(move->fdes src 0)
|
|
|
|
(close r)
|
|
|
|
(move->fdes w 1)
|
|
|
|
(exec* command))
|
|
|
|
(#t
|
2016-10-18 08:15:53 +01:00
|
|
|
(job-add-process job pid command)
|
2016-05-30 23:13:04 +01:00
|
|
|
(close w)
|
|
|
|
r)))))
|
|
|
|
|
2016-10-18 08:15:53 +01:00
|
|
|
(define (spawn-sink job interactive? src command)
|
2016-05-30 23:13:04 +01:00
|
|
|
(let ((pid (primitive-fork)))
|
|
|
|
(cond ((= 0 pid)
|
2016-10-18 22:11:31 +01:00
|
|
|
(tcsetpgrp (current-error-port) (add-to-process-group job (getpid)))
|
|
|
|
|
2016-10-18 08:15:53 +01:00
|
|
|
(map (cut sigaction <> SIG_DFL)
|
|
|
|
(list SIGINT SIGQUIT SIGTSTP SIGTTIN SIGTTOU SIGCHLD))
|
|
|
|
|
|
|
|
(and src (move->fdes src 0))
|
2016-05-30 23:13:04 +01:00
|
|
|
(exec* command))
|
|
|
|
(#t
|
2016-10-18 08:15:53 +01:00
|
|
|
(job-add-process job pid command)
|
|
|
|
(and src (close src))))))
|
2016-05-30 23:13:04 +01:00
|
|
|
|
2016-10-18 22:11:31 +01:00
|
|
|
(define (job-launch command fg?) ;; todo: integrate into pipeline
|
|
|
|
(let* ((interactive? (isatty? (current-error-port)))
|
|
|
|
(foo (stdout "ISATTY: " interactive?))
|
|
|
|
(pgid (getpid))
|
|
|
|
(pid (primitive-fork)))
|
|
|
|
(if (= 0 pid)
|
|
|
|
(when interactive? ;; the child i.e. command
|
|
|
|
(setpgid (getpid) (getpid)) ;; put job in own process group
|
|
|
|
(if fg? (tcsetpgrp (current-error-port) (getpid))) ;; put job in foreground
|
|
|
|
|
|
|
|
(map (cut sigaction <> SIG_DFL)
|
|
|
|
(list SIGINT SIGQUIT SIGTSTP SIGTTIN SIGTTOU SIGCHLD))
|
|
|
|
(fdes->inport 0)
|
|
|
|
(map fdes->outport '(1 2))
|
|
|
|
|
|
|
|
(exec* command)
|
|
|
|
(exit 1))
|
|
|
|
(when interactive? ;; the parent i.e. shell
|
|
|
|
(setpgid pid pid) ;; put job in own process group
|
|
|
|
(tcsetpgrp (current-error-port) pid) ;; put job in foreground
|
|
|
|
(when fg?
|
|
|
|
(waitpid pid WUNTRACED)
|
|
|
|
(tcsetpgrp (current-error-port) (getpid))))))) ;; put shell in foreground
|
|
|
|
|
|
|
|
(define (pipelinex . commands)
|
|
|
|
(stdout "pipeline: " commands)
|
|
|
|
(job-launch (car commands) #t))
|
|
|
|
|
2016-10-10 08:54:18 +01:00
|
|
|
(define (pipeline . commands)
|
2016-10-18 22:11:31 +01:00
|
|
|
(let ((interactive? (isatty? (current-error-port)))
|
|
|
|
(job (make-job (length job-table) #f '())))
|
2016-10-18 08:15:53 +01:00
|
|
|
(set! job-table (cons job job-table))
|
2016-10-18 22:11:31 +01:00
|
|
|
(if (> (length commands) 1)
|
2016-10-18 08:15:53 +01:00
|
|
|
(let loop ((src (spawn-source job interactive? (car commands)))
|
|
|
|
(commands (cdr commands)))
|
|
|
|
(if (null? (cdr commands))
|
|
|
|
(spawn-sink job interactive? src (car commands))
|
|
|
|
(loop (spawn-filter job interactive? src (car commands))
|
|
|
|
(cdr commands))))
|
|
|
|
(spawn-sink job interactive? #f (car commands))))
|
|
|
|
(waitpid WAIT_ANY WUNTRACED)
|
2016-10-18 22:11:31 +01:00
|
|
|
(tcsetpgrp (current-error-port) (getpgrp))
|
2016-10-18 08:15:53 +01:00
|
|
|
(stdout "job-table: " job-table))
|
2016-05-30 23:13:04 +01:00
|
|
|
|
2016-10-15 23:03:45 +01:00
|
|
|
;;(pipeline (list "ls" "/")
|
2016-10-10 08:54:18 +01:00
|
|
|
;;(pipeline (list "ls" "/") (list "grep" "o") (list "tr" "o" "e"))
|
2016-10-15 23:03:45 +01:00
|
|
|
|
|
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
|
|
|
2016-10-18 08:15:53 +01:00
|
|
|
;; (define (mark-job-status pid status)
|
|
|
|
;; (if (not (= 0 pid))
|
|
|
|
;; (cond ((status:stop-sig status)
|
|
|
|
;; (set-job-stopped! pid job-table))
|
|
|
|
;; ((status:term-sig status)
|
|
|
|
;; (set-job-terminated! pid job-table))
|
|
|
|
;; ((status:exit-val status)
|
|
|
|
;; (set-job-completed! pid job-table)))))
|
|
|
|
|
|
|
|
;; (define (job-wait job)
|
|
|
|
;; (let loop ()
|
|
|
|
;; (let* ((status (job-status job))
|
|
|
|
;; (pid-status (waitpid WAIT_ANY WUNTRACED))
|
|
|
|
;; (pid (car pid-status))
|
|
|
|
;; (status (cdr pid-status)))
|
|
|
|
;; (mark-job-status pid status)
|
|
|
|
;; (if (eq? status (job-status job)) (loop)))))
|
|
|
|
|
|
|
|
;; (define (set-job-stopped! job-table pid)
|
|
|
|
;; (set-job-status! (assoc-ref job-table pid) 'stopped))
|
|
|
|
|
|
|
|
;; (define (set-job-terminated! job-table pid) ;; signal
|
|
|
|
;; (set-job-status! (assoc-ref job-table pid) 'terminated))
|
|
|
|
|
|
|
|
;; (define (set-job-completed! job-table pid) ;; exit value
|
|
|
|
;; (set-job-status! (assoc-ref job-table pid) 'completed))
|
|
|
|
|
|
|
|
;; (define (notify-job-status job-table) ;; call when prompting, from SIGCHLD handler or
|
|
|
|
;; (let* ((pid-status (waitpid WAIT_ANY (logior WUNTRACED WNOHANG)))
|
|
|
|
;; (pid (car pid-status))
|
|
|
|
;; (status (cdr pid-status)))
|
|
|
|
;; (mark-job-status pid status)))
|
|
|
|
|
|
|
|
;; (define (job-by-id job-table id)
|
|
|
|
;; (if (< job-id (length job-table))
|
|
|
|
;; (list-ref job-table job-id)
|
|
|
|
;; #f))
|
|
|
|
|
|
|
|
;; (define (background job-id)
|
|
|
|
;; (let ((job (job-by-id job-table job-id)))
|
|
|
|
;; (if (and job (eq? 'stopped (job-status job)))
|
|
|
|
;; (kill (- (job-pgid job)) SIGCONT))))
|
|
|
|
|
|
|
|
;; (define (foreground job-id)
|
|
|
|
;; (let ((job (job-by-id job-table job-id)))
|
2016-10-18 22:11:31 +01:00
|
|
|
;; (tcsetpgrp (current-error-port) (job-pgid job))
|
2016-10-18 08:15:53 +01:00
|
|
|
;; (if (and job (eq? 'stopped (job-status job)))
|
|
|
|
;; (kill (- (job-pgid job)) SIGCONT))
|
|
|
|
;; (job-wait job)
|
2016-10-18 22:11:31 +01:00
|
|
|
;; (tcsetpgrp (current-error-port) (getpid))))
|