mescc: Support switch.
[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 (functions->lambdas functions)
35   (append-map cdr functions))
36
37 (define (lambda/label->list f g t d)
38   (lambda (l/l)
39     (if (not (procedure? l/l)) '() (l/l f g t d))))
40
41 (define (text->list o)
42   (append-map (lambda/label->list '() '() 0 0) o))
43
44 (define (functions->text functions globals t d)
45   (let loop ((lambdas/labels (functions->lambdas functions)) (text '()))
46     (if (null? lambdas/labels) text
47         (loop (cdr lambdas/labels)
48               (append text ((lambda/label->list functions globals (- (length text)) d) (car lambdas/labels)))))))
49
50 (define (function-prefix name functions)
51   (member name (reverse functions) (lambda (a b) (equal? (car b) name))))
52
53 (define function-offset
54   (let ((cache '()))
55     (lambda (name functions)
56       (or (assoc-ref cache name)
57           (let* ((prefix (function-prefix name functions))
58                  (offset (if prefix (length (functions->text (cdr prefix) '() 0 0))
59                              0)))
60             (if (or (equal? name "exit") (> offset 0)) (set! cache (assoc-set! cache name offset)))
61             offset)))))
62
63 (define (label-offset function label functions)
64   (let ((prefix (function-prefix function functions)))
65     (if (not prefix) 0
66         (let ((function-entry (car prefix)))
67           (let loop ((text (cdr function-entry)))
68             (if (or (equal? (car text) label) (null? text)) 0
69                 (let* ((l/l (car text))
70                        (t ((lambda/label->list '() '() 0 0) l/l))
71                        (n (length t)))
72                   (+ (loop (cdr text)) n))))))))
73
74 (define (globals->data globals)
75   (append-map cdr globals))
76
77 (define (data-offset name globals)
78   (let* ((prefix (member name (reverse globals)
79                          (lambda (a b)
80                            (equal? (car b) name)))))
81     (if prefix (length (globals->data (cdr prefix)))
82         0)))