82e5aca4e26377955a9b72f522c3a63f5832add1
[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 (function-prefix name functions)
65   ;; FIXME
66   ;;(member name (reverse functions) (lambda (a b) (equal? (car b) name)))
67   (let* ((x functions)
68          (x (if (and (pair? x) (equal? (caar x) "exit")) (reverse x) x)))
69     (member name x (lambda (a b) (equal? (car b) name)))))
70
71 (define function-offset
72   (let ((cache '()))
73     (lambda (name functions)
74       (or (assoc-ref cache name)
75           (let* ((prefix (function-prefix name functions))
76                  (offset (if prefix (length (functions->text (cdr prefix) '() 0 0 0))
77                              0)))
78             (if (and prefix (or (equal? name "exit") (> offset 0))) (set! cache (assoc-set! cache name offset)))
79             offset)))))
80
81 (define label-offset
82   (let ((cache '()))
83     (lambda (function label functions)
84       (or (assoc-ref cache (cons function label))
85           (let ((prefix (function-prefix function functions)))
86             (if (not prefix) 0
87                 (let* ((function-entry (car prefix))
88                        (offset (let loop ((text (cdr function-entry)))
89                                  (if (or (equal? (car text) label) (null? text)) 0
90                                      (let* ((l/l (car text))
91                                             (t ((lambda/label->list '() '() 0 0 0) l/l))
92                                             (n (length t)))
93                                        (+ (loop (cdr text)) n))))))
94                   (when (> offset 0)
95                     (set! cache (assoc-set! cache (cons function label) offset)))
96                   offset)))))))
97
98 (define (globals->data globals)
99   (append-map (compose global:value cdr) globals))
100
101 (define data-offset
102   (let ((cache '()))
103     (lambda (name globals)
104       (or (assoc-ref cache name)
105           (let* ((prefix (member name (reverse globals)
106                                  (lambda (a b)
107                                    (equal? (car b) name)))))
108             (if (not prefix) 0
109                 (let ((offset (length (globals->data (cdr prefix)))))
110                  (set! cache (assoc-set! cache name offset))
111                  offset)))))))