control-logic.lsp

   1 # Apply timing control to this host
   2 # This is run as a cron job to either "close" or "open" the blocking of
   3 # the host via a configured control implementation.
   4
   5 # File control.dat defines the limits, and control mechanism.
   6 # File activity-$date.dat is the local activity.
   7
   8 (define (die)
   9   (write-line 2 (join (map string args)))
  10   (exit 1))
  11
  12 (constant
  13  'NOW (date-value)
  14  'ACTDIR SITE:listener.activity.dir
  15  'CONTROL.DAT SITE:control.dat
  16  'EXTRA.DAT SITE:control.extra.dat
  17  'USAGE.DAT SITE:usage.dat
  18  'USAGE.TMP SITE:usage.tmp
  19  )
  20
  21 # Set current time variables in local timezone
  22 (map set '(YEAR MONTH DATE HOUR MINUTE SECOND DOY DOW)
  23      (date-list (+ NOW (* 60 (now 0 -2)))))
  24 ;(println (list YEAR MONTH DATE HOUR MINUTE SECOND DOY DOW))
  25
  26 (setf
  27  DAY (list YEAR MONTH DATE)
  28  HM (list HOUR MINUTE)
  29  TOTAL '()
  30  GAP SITE:control.activity.gap
  31  CLIP SITE:control.activity.clip
  32  )
  33
  34 # Load CONTROL.DAT
  35 # ( (control "file") (gap minutes) ( weekday start limit stop ) ... )
  36 (setf CONTROL (read-expr (or (read-file CONTROL.DAT)
  37                              (die "** Missing " CONTROL.DAT " ** Exiting."))
  38                          ))
  39 (map set '(dow MODE START LIMIT END)
  40      (or (assoc DOW CONTROL)
  41          (assoc 'policy CONTROL)
  42          (list DOW (6 30) 120 (20 0)))
  43      )
  44
  45 # Load the configured control mechanism
  46 (if (lookup 'control CONTROL) (load $it)
  47   (die "** Unknown control mechanism. Exiting!!"))
  48 (unless control
  49     (die "** Control function (control cmd reason) not defined. Exiting!!"))
  50
  51 ;; Apply control command with reason, then exit
  52 (define (do-control x r) (control x r) (exit 0))
  53
  54 ;; Utility: Combine an (hour minutes) pair into total minutes
  55 (define (minutes x) (+ (* (x 0) 60) (x 1)))
  56
  57 ;; Utility: Combine hours and minutes into total seconds
  58 (define (seconds H M) (+ (* 3600 H) (* 60 M)))
  59
  60 # Apply EXTRA.DAT. This is a pair of hours and minutes to force open,
  61 # relative to the modification time of the file.
  62 (setf OVERRIDE
  63       (when (regex "([0-9]+) ([0-9]+)" (or (read-file EXTRA.DAT) "") 0)
  64         (<= NOW (+ (file-info EXTRA.DAT 6) (seconds (int $1) (int $2))))))
  65
  66 ;;==== Utilities for activity data
  67 # Activity is lines of timestamps. Collect TOTAL as list of unique
  68 # time values (H M) within the start-end time span.
  69
  70 (define (log-name-fmt t)
  71   (format "%d%02d%02d-.*\\.dat" (0 3 (date-list t))))
  72
  73 (define (log-lines f)
  74   (find-all "([0-9]+( \\S+)?).*" (read-file (format "%s/%s" ACTDIR f)) $1 0))
  75
  76 # Collect all timestamps of the UTC date of the given time stamp
  77 (define (logs t)
  78   (flat (map log-lines (directory ACTDIR (log-name-fmt t)))))
  79
  80 # Translate timestamp into its local time (hour minute), if it's
  81 # within the applicable day, null otherwise.
  82 (define (period-minute x)
  83   (when x
  84     (letn ((d (date-list (+ (int x 0 10) (* 60 (now 0 -2)))))
  85            (tm (3 2 d))
  86            (on (if (regex "^[0-9]+ ([0-9]+)$" x 0) (> (int $1 0 10) CLIP) 1))
  87            )
  88       (and on (= (0 3 d) DAY) tm)))) ; (>= tm START) (< tm END) tm))))
  89 ;; ======
  90
  91 # Collect all mentioned minutes from the activity logs
  92 (setf TOTAL
  93       (unique
  94        (clean null?
  95               (map period-minute
  96                    (sort (extend (logs (- NOW 86400)) (logs NOW)))))))
  97
  98 # Add all mentioned minutes, and fill in any time periods of less than
  99 # the configured GAP minutes between them.
 100 (setf SUM 0)
 101 (when TOTAL
 102   (setf  LAST (minutes (pop TOTAL) SUM 1))
 103   (dolist (x TOTAL)
 104     (letn ((M (minutes x)) (V (- M LAST)))
 105       (inc SUM (if (< V GAP) V 1))
 106       (setf LAST M)))
 107   )
 108
 109 # Rework SUM into (h m) format
 110 (setf SUM (letn ((h (/ SUM 60)) (m (- SUM (* 60 h)))) (list h m)))
 111
 112 # Write out current usage with atomic update (jic)
 113 (write-file USAGE.TMP (string SUM))
 114 (rename-file USAGE.TMP USAGE.DAT)
 115
 116 # Apply current policy setting
 117 ; ** Note that do-control exits
 118 (when OVERRIDE (do-control "open" "override"))
 119 (case MODE
 120   (closed (do-control "close" "closed"))
 121   (opened (do-control "open" "open"))
 122   (timed ;; Close host outside start-end times
 123    (when (< HM START) (do-control "close" "early"))
 124    (when (>= HM END) (do-control "close" "late"))
 125    (when (> SUM LIMIT) (do-control "close" "usage"))
 126    (do-control "open" "usage")
 127    )
 128   (true (die "Unknown control mode " MODE))
 129   )