core+mini-mes: Replace manual snippets by snarfed includes.
[mes.git] / module / mes / elf-util.mes
1 ;;; -*-scheme-*-
2
3 ;;; Mes --- Maxwell Equations of Software
4 ;;; Copyright © 2016,2017 Jan Nieuwenhuizen <janneke@gnu.org>
5 ;;;
6 ;;; This file is part of Mes.
7 ;;;
8 ;;; Mes is free software; you can redistribute it and/or modify it
9 ;;; under the terms of the GNU General Public License as published by
10 ;;; the Free Software Foundation; either version 3 of the License, or (at
11 ;;; your option) any later version.
12 ;;;
13 ;;; Mes is distributed in the hope that it will be useful, but
14 ;;; WITHOUT ANY WARRANTY; without even the implied warranty of
15 ;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 ;;; GNU General Public License for more details.
17 ;;;
18 ;;; You should have received a copy of the GNU General Public License
19 ;;; along with Mes.  If not, see <http://www.gnu.org/licenses/>.
20
21 ;;; Commentary:
22
23 ;;; compiler.mes produces an i386 binary from the C produced by
24 ;;; Nyacc c99.
25
26 ;;; Code:
27
28 (cond-expand
29  (guile)
30  (guile-2)
31  (mes
32   (mes-use-module (srfi srfi-1))))
33
34 (define (make-global name type pointer value)
35   (cons name (list type pointer value)))
36
37 (define global:type car)
38 (define global:pointer cadr)
39 (define global:value caddr)
40
41 (define (drop-s:-prefix o) (substring o 2))
42 (define (add-s:-prefix o) (string-append "s:" o))
43
44 (define (dec->hex o)
45   (cond ((number? o) (number->string o 16))
46         ((char? o) (number->string (char->integer o) 16))))
47
48 (define (functions->lambdas functions)
49   (append-map cdr functions))
50
51 (define (lambda/label->list f g ta t d)
52   (lambda (l/l)
53     (if (not (procedure? l/l)) '() (l/l f g ta t d))))
54
55 (define (text->list o)
56   (append-map (lambda/label->list '() '() 0 0 0) o))
57
58 (define (functions->text functions globals ta t d)
59   (let loop ((lambdas/labels (functions->lambdas functions)) (text '()))
60     (if (null? lambdas/labels) text
61         (loop (cdr lambdas/labels)
62               (append text ((lambda/label->list functions globals ta (- (length text)) d) (car lambdas/labels)))))))
63
64 ;; (define (functions->text functions globals ta t d)
65 ;;   (let loop ((functions functions) (text '()))
66 ;;     (if (null? functions) text
67 ;;         (loop (cdr functions)
68 ;;               (append '() ;;text
69 ;;                       (function->text functions globals ta t d text (car functions)))))))
70
71 ;; (define (function->text functions globals ta t d text function)
72 ;;   (format (current-error-port) "elf func=~a\n" (car function))
73 ;;   (let loop ((lambdas/labels (cdr function)) (text text))
74 ;;     (if (null? lambdas/labels) text
75 ;;         (loop (cdr lambdas/labels)
76 ;;               (append '() ;;text
77 ;;                       ((lambda/label->list functions globals ta (- (length text)) d) (car lambdas/labels)))))))
78
79 (define (function-prefix name functions)
80   ;; FIXME
81   ;;(member name (reverse functions) (lambda (a b) (equal? (car b) name)))
82   (let* ((x functions)
83          (x (if (and (pair? x) (equal? (caar x) "exit")) (reverse x) x)))
84     (member name x (lambda (a b) (equal? (car b) name)))))
85
86 (define function-offset
87   (let ((cache '()))
88     (lambda (name functions)
89       (or (assoc-ref cache name)
90           (let* ((prefix (function-prefix name functions))
91                  (offset (if prefix (length (functions->text (cdr prefix) '() 0 0 0))
92                              0)))
93             (if (and prefix (or (equal? name "exit") (> offset 0))) (set! cache (assoc-set! cache name offset)))
94             offset)))))
95
96 (define label-offset
97   (let ((cache '()))
98     (lambda (function label functions)
99       (or (assoc-ref cache (cons function label))
100           (let ((prefix (function-prefix function functions)))
101             (if (not prefix) 0
102                 (let* ((function-entry (car prefix))
103                        (offset (let loop ((text (cdr function-entry)))
104                                  (if (or (equal? (car text) label) (null? text)) 0
105                                      (let* ((l/l (car text))
106                                             (t ((lambda/label->list '() '() 0 0 0) l/l))
107                                             (n (length t)))
108                                        (+ (loop (cdr text)) n))))))
109                   (when (> offset 0)
110                       (set! cache (assoc-set! cache (cons function label) offset)))
111                   offset)))))))
112
113 (define (globals->data globals)
114   (append-map (compose global:value cdr) globals))
115
116 (define (data-offset name globals)
117   (let* ((prefix (member name (reverse globals)
118                          (lambda (a b)
119                            (equal? (car b) name)))))
120     (if prefix (length (globals->data (cdr prefix)))
121         0)))