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