mescc: Compile all of mini-mes.
[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 (dec->hex o)
42   (cond ((number? o) (number->string o 16))
43         ((char? o) (number->string (char->integer o) 16))))
44
45 (define (functions->lambdas functions)
46   (append-map cdr functions))
47
48 (define (lambda/label->list f g ta t d)
49   (lambda (l/l)
50     (if (not (procedure? l/l)) '() (l/l f g ta t d))))
51
52 (define (text->list o)
53   (append-map (lambda/label->list '() '() 0 0 0) o))
54
55 (define (functions->text functions globals ta t d)
56   (let loop ((lambdas/labels (functions->lambdas functions)) (text '()))
57     (if (null? lambdas/labels) text
58         (loop (cdr lambdas/labels)
59               (append text ((lambda/label->list functions globals ta (- (length text)) d) (car lambdas/labels)))))))
60
61 ;; (define (functions->text functions globals ta t d)
62 ;;   (let loop ((functions functions) (text '()))
63 ;;     (if (null? functions) text
64 ;;         (loop (cdr functions)
65 ;;               (append '() ;;text
66 ;;                       (function->text functions globals ta t d text (car functions)))))))
67
68 ;; (define (function->text functions globals ta t d text function)
69 ;;   (format (current-error-port) "elf func=~a\n" (car function))
70 ;;   (let loop ((lambdas/labels (cdr function)) (text text))
71 ;;     (if (null? lambdas/labels) text
72 ;;         (loop (cdr lambdas/labels)
73 ;;               (append '() ;;text
74 ;;                       ((lambda/label->list functions globals ta (- (length text)) d) (car lambdas/labels)))))))
75
76 (define (function-prefix name functions)
77   ;; FIXME
78   ;;(member name (reverse functions) (lambda (a b) (equal? (car b) name)))
79   (let* ((x functions)
80          (x (if (and (pair? x) (equal? (caar x) "exit")) (reverse x) x)))
81     (member name x (lambda (a b) (equal? (car b) name)))))
82
83 (define function-offset
84   (let ((cache '()))
85     (lambda (name functions)
86       (or (assoc-ref cache name)
87           (let* ((prefix (function-prefix name functions))
88                  (offset (if prefix (length (functions->text (cdr prefix) '() 0 0 0))
89                              0)))
90             (if (and prefix (or (equal? name "exit") (> offset 0))) (set! cache (assoc-set! cache name offset)))
91             offset)))))
92
93 (define (label-offset function label functions)
94   (let ((prefix (function-prefix function functions)))
95     (if (not prefix) 0
96         (let ((function-entry (car prefix)))
97           (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
104 (define (globals->data globals)
105   (append-map (compose global:value cdr) globals))
106
107 (define (data-offset name globals)
108   (let* ((prefix (member name (reverse globals)
109                          (lambda (a b)
110                            (equal? (car b) name)))))
111     (if prefix (length (globals->data (cdr prefix)))
112         0)))