Statistics
| Branch: | Revision:

root / dyngen.c @ 2a29ca73

History | View | Annotate | Download (34.9 kB)

1
/*
2
 *  Generic Dynamic compiler generator
3
 * 
4
 *  Copyright (c) 2003 Fabrice Bellard
5
 *
6
 *  This program is free software; you can redistribute it and/or modify
7
 *  it under the terms of the GNU General Public License as published by
8
 *  the Free Software Foundation; either version 2 of the License, or
9
 *  (at your option) any later version.
10
 *
11
 *  This program is distributed in the hope that it will be useful,
12
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
 *  GNU General Public License for more details.
15
 *
16
 *  You should have received a copy of the GNU General Public License
17
 *  along with this program; if not, write to the Free Software
18
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19
 */
20
#include <stdlib.h>
21
#include <stdio.h>
22
#include <string.h>
23
#include <stdarg.h>
24
#include <inttypes.h>
25
#include <unistd.h>
26
#include <fcntl.h>
27

    
28
#include "config.h"
29

    
30
/* elf format definitions. We use these macros to test the CPU to
31
   allow cross compilation (this tool must be ran on the build
32
   platform) */
33
#if defined(HOST_I386)
34

    
35
#define ELF_CLASS        ELFCLASS32
36
#define ELF_ARCH        EM_386
37
#define elf_check_arch(x) ( ((x) == EM_386) || ((x) == EM_486) )
38
#undef ELF_USES_RELOCA
39

    
40
#elif defined(HOST_PPC)
41

    
42
#define ELF_CLASS        ELFCLASS32
43
#define ELF_ARCH        EM_PPC
44
#define elf_check_arch(x) ((x) == EM_PPC)
45
#define ELF_USES_RELOCA
46

    
47
#elif defined(HOST_S390)
48

    
49
#define ELF_CLASS        ELFCLASS32
50
#define ELF_ARCH        EM_S390
51
#define elf_check_arch(x) ((x) == EM_S390)
52
#define ELF_USES_RELOCA
53

    
54
#elif defined(HOST_ALPHA)
55

    
56
#define ELF_CLASS        ELFCLASS64
57
#define ELF_ARCH        EM_ALPHA
58
#define elf_check_arch(x) ((x) == EM_ALPHA)
59
#define ELF_USES_RELOCA
60

    
61
#elif defined(HOST_IA64)
62

    
63
#define ELF_CLASS        ELFCLASS64
64
#define ELF_ARCH        EM_IA_64
65
#define elf_check_arch(x) ((x) == EM_IA_64)
66
#define ELF_USES_RELOCA
67

    
68
#elif defined(HOST_SPARC)
69

    
70
#define ELF_CLASS        ELFCLASS32
71
#define ELF_ARCH        EM_SPARC
72
#define elf_check_arch(x) ((x) == EM_SPARC || (x) == EM_SPARC32PLUS)
73
#define ELF_USES_RELOCA
74

    
75
#elif defined(HOST_SPARC64)
76

    
77
#define ELF_CLASS        ELFCLASS64
78
#define ELF_ARCH        EM_SPARCV9
79
#define elf_check_arch(x) ((x) == EM_SPARCV9)
80
#define ELF_USES_RELOCA
81

    
82
#else
83
#error unsupported CPU - please update the code
84
#endif
85

    
86
#include "elf.h"
87

    
88
#if ELF_CLASS == ELFCLASS32
89
typedef int32_t host_long;
90
typedef uint32_t host_ulong;
91
#define swabls(x) swab32s(x)
92
#else
93
typedef int64_t host_long;
94
typedef uint64_t host_ulong;
95
#define swabls(x) swab64s(x)
96
#endif
97

    
98
#include "thunk.h"
99

    
100
/* all dynamically generated functions begin with this code */
101
#define OP_PREFIX "op_"
102

    
103
int elf_must_swap(struct elfhdr *h)
104
{
105
  union {
106
      uint32_t i;
107
      uint8_t b[4];
108
  } swaptest;
109

    
110
  swaptest.i = 1;
111
  return (h->e_ident[EI_DATA] == ELFDATA2MSB) != 
112
      (swaptest.b[0] == 0);
113
}
114
  
115
void swab16s(uint16_t *p)
116
{
117
    *p = bswap16(*p);
118
}
119

    
120
void swab32s(uint32_t *p)
121
{
122
    *p = bswap32(*p);
123
}
124

    
125
void swab64s(uint64_t *p)
126
{
127
    *p = bswap64(*p);
128
}
129

    
130
void elf_swap_ehdr(struct elfhdr *h)
131
{
132
    swab16s(&h->e_type);                        /* Object file type */
133
    swab16s(&h->        e_machine);                /* Architecture */
134
    swab32s(&h->        e_version);                /* Object file version */
135
    swabls(&h->        e_entry);                /* Entry point virtual address */
136
    swabls(&h->        e_phoff);                /* Program header table file offset */
137
    swabls(&h->        e_shoff);                /* Section header table file offset */
138
    swab32s(&h->        e_flags);                /* Processor-specific flags */
139
    swab16s(&h->        e_ehsize);                /* ELF header size in bytes */
140
    swab16s(&h->        e_phentsize);                /* Program header table entry size */
141
    swab16s(&h->        e_phnum);                /* Program header table entry count */
142
    swab16s(&h->        e_shentsize);                /* Section header table entry size */
143
    swab16s(&h->        e_shnum);                /* Section header table entry count */
144
    swab16s(&h->        e_shstrndx);                /* Section header string table index */
145
}
146

    
147
void elf_swap_shdr(struct elf_shdr *h)
148
{
149
  swab32s(&h->        sh_name);                /* Section name (string tbl index) */
150
  swab32s(&h->        sh_type);                /* Section type */
151
  swabls(&h->        sh_flags);                /* Section flags */
152
  swabls(&h->        sh_addr);                /* Section virtual addr at execution */
153
  swabls(&h->        sh_offset);                /* Section file offset */
154
  swabls(&h->        sh_size);                /* Section size in bytes */
155
  swab32s(&h->        sh_link);                /* Link to another section */
156
  swab32s(&h->        sh_info);                /* Additional section information */
157
  swabls(&h->        sh_addralign);                /* Section alignment */
158
  swabls(&h->        sh_entsize);                /* Entry size if section holds table */
159
}
160

    
161
void elf_swap_phdr(struct elf_phdr *h)
162
{
163
    swab32s(&h->p_type);                        /* Segment type */
164
    swabls(&h->p_offset);                /* Segment file offset */
165
    swabls(&h->p_vaddr);                /* Segment virtual address */
166
    swabls(&h->p_paddr);                /* Segment physical address */
167
    swabls(&h->p_filesz);                /* Segment size in file */
168
    swabls(&h->p_memsz);                /* Segment size in memory */
169
    swab32s(&h->p_flags);                /* Segment flags */
170
    swabls(&h->p_align);                /* Segment alignment */
171
}
172

    
173
int do_swap;
174

    
175
uint16_t get16(uint16_t *p)
176
{
177
    uint16_t val;
178
    val = *p;
179
    if (do_swap)
180
        val = bswap16(val);
181
    return val;
182
}
183

    
184
uint32_t get32(uint32_t *p)
185
{
186
    uint32_t val;
187
    val = *p;
188
    if (do_swap)
189
        val = bswap32(val);
190
    return val;
191
}
192

    
193
void put16(uint16_t *p, uint16_t val)
194
{
195
    if (do_swap)
196
        val = bswap16(val);
197
    *p = val;
198
}
199

    
200
void put32(uint32_t *p, uint32_t val)
201
{
202
    if (do_swap)
203
        val = bswap32(val);
204
    *p = val;
205
}
206

    
207
void __attribute__((noreturn)) __attribute__((format (printf, 1, 2))) error(const char *fmt, ...)
208
{
209
    va_list ap;
210
    va_start(ap, fmt);
211
    fprintf(stderr, "dyngen: ");
212
    vfprintf(stderr, fmt, ap);
213
    fprintf(stderr, "\n");
214
    va_end(ap);
215
    exit(1);
216
}
217

    
218

    
219
struct elf_shdr *find_elf_section(struct elf_shdr *shdr, int shnum, const char *shstr, 
220
                                  const char *name)
221
{
222
    int i;
223
    const char *shname;
224
    struct elf_shdr *sec;
225

    
226
    for(i = 0; i < shnum; i++) {
227
        sec = &shdr[i];
228
        if (!sec->sh_name)
229
            continue;
230
        shname = shstr + sec->sh_name;
231
        if (!strcmp(shname, name))
232
            return sec;
233
    }
234
    return NULL;
235
}
236

    
237
void *load_data(int fd, long offset, unsigned int size)
238
{
239
    char *data;
240

    
241
    data = malloc(size);
242
    if (!data)
243
        return NULL;
244
    lseek(fd, offset, SEEK_SET);
245
    if (read(fd, data, size) != size) {
246
        free(data);
247
        return NULL;
248
    }
249
    return data;
250
}
251

    
252
int strstart(const char *str, const char *val, const char **ptr)
253
{
254
    const char *p, *q;
255
    p = str;
256
    q = val;
257
    while (*q != '\0') {
258
        if (*p != *q)
259
            return 0;
260
        p++;
261
        q++;
262
    }
263
    if (ptr)
264
        *ptr = p;
265
    return 1;
266
}
267

    
268
#define MAX_ARGS 3
269

    
270
/* generate op code */
271
void gen_code(const char *name, host_ulong offset, host_ulong size, 
272
              FILE *outfile, uint8_t *text, ELF_RELOC *relocs, int nb_relocs, int reloc_sh_type,
273
              ElfW(Sym) *symtab, char *strtab, int gen_switch)
274
{
275
    int copy_size = 0;
276
    uint8_t *p_start, *p_end;
277
    int nb_args, i, n;
278
    uint8_t args_present[MAX_ARGS];
279
    const char *sym_name, *p;
280
    ELF_RELOC *rel;
281

    
282
    /* compute exact size excluding return instruction */
283
    p_start = text + offset;
284
    p_end = p_start + size;
285
    switch(ELF_ARCH) {
286
    case EM_386:
287
        {
288
            uint8_t *p;
289
            p = p_end - 1;
290
            if (p == p_start)
291
                error("empty code for %s", name);
292
            if (p[0] != 0xc3)
293
                error("ret expected at the end of %s", name);
294
            copy_size = p - p_start;
295
        }
296
        break;
297
    case EM_PPC:
298
        {
299
            uint8_t *p;
300
            p = (void *)(p_end - 4);
301
            if (p == p_start)
302
                error("empty code for %s", name);
303
            if (get32((uint32_t *)p) != 0x4e800020)
304
                error("blr expected at the end of %s", name);
305
            copy_size = p - p_start;
306
        }
307
        break;
308
    case EM_S390:
309
        {
310
            uint8_t *p;
311
            p = (void *)(p_end - 2);
312
            if (p == p_start)
313
                error("empty code for %s", name);
314
            if (get16((uint16_t *)p) != 0x07fe && get16((uint16_t *)p) != 0x07f4)
315
                error("br %%r14 expected at the end of %s", name);
316
            copy_size = p - p_start;
317
        }
318
        break;
319
    case EM_ALPHA:
320
        {
321
            uint8_t *p;
322
            p = p_end - 4;
323
            if (p == p_start)
324
                error("empty code for %s", name);
325
            if (get32((uint32_t *)p) != 0x6bfa8001)
326
                error("ret expected at the end of %s", name);
327
            copy_size = p - p_start;            
328
        }
329
        break;
330
    case EM_IA_64:
331
        {
332
            uint8_t *p;
333
            p = (void *)(p_end - 4);
334
            if (p == p_start)
335
                error("empty code for %s", name);
336
            /* br.ret.sptk.many b0;; */
337
            /* 08 00 84 00 */
338
            if (get32((uint32_t *)p) != 0x00840008)
339
                error("br.ret.sptk.many b0;; expected at the end of %s", name);
340
            copy_size = p - p_start;
341
        }
342
        break;
343
    case EM_SPARC:
344
    case EM_SPARC32PLUS:
345
        {
346
            uint8_t *p;
347
            p = (void *)(p_end - 8);
348
            if (p <= p_start)
349
                error("empty code for %s", name);
350
            if (get32((uint32_t *)(p_start + 0x0)) != 0x9de3bf98)
351
                error("save %%sp,-104,%%sp expected at the start of %s "
352
                      "found [%08x]",
353
                      name, get32((uint32_t *)(p_start + 0x0)));
354
            if (get32((uint32_t *)(p + 0x0)) != 0x81c7e008 ||
355
                get32((uint32_t *)(p + 0x4)) != 0x81e80000)
356
                error("ret; restore; expected at the end of %s found [%08x:%08x]",
357
                      name,
358
                      get32((uint32_t *)(p + 0x0)),
359
                      get32((uint32_t *)(p + 0x4)));
360

    
361
            copy_size = p - p_start;
362
        }
363
        break;
364
    case EM_SPARCV9:
365
        {
366
            uint8_t *p;
367
            p = (void *)(p_end - 8);
368
            if (p <= p_start)
369
                error("empty code for %s", name);
370
            if (get32((uint32_t *)(p_start + 0x0)) != 0x9de3bf40)
371
                error("save %%sp,-192,%%sp expected at the start of %s "
372
                      "found [%08x]",
373
                      name, get32((uint32_t *)(p_start + 0x0)));
374
            if (get32((uint32_t *)(p + 0x0)) != 0x81cfe008 ||
375
                get32((uint32_t *)(p + 0x4)) != 0x01000000)
376
                error("rett %%i7+8; nop; expected at the end of %s "
377
                      "found [%08x:%08x]",
378
                      name,
379
                      get32((uint32_t *)(p + 0x0)),
380
                      get32((uint32_t *)(p + 0x4)));
381
            copy_size = p - p_start;
382
        }
383
        break;
384
    default:
385
        error("unknown ELF architecture");
386
    }
387

    
388
    /* compute the number of arguments by looking at the relocations */
389
    for(i = 0;i < MAX_ARGS; i++)
390
        args_present[i] = 0;
391

    
392
    for(i = 0, rel = relocs;i < nb_relocs; i++, rel++) {
393
        if (rel->r_offset >= offset && rel->r_offset < offset + copy_size) {
394
            sym_name = strtab + symtab[ELFW(R_SYM)(rel->r_info)].st_name;
395
            if (strstart(sym_name, "__op_param", &p)) {
396
                n = strtoul(p, NULL, 10);
397
                if (n >= MAX_ARGS)
398
                    error("too many arguments in %s", name);
399
                args_present[n - 1] = 1;
400
            }
401
        }
402
    }
403
    
404
    nb_args = 0;
405
    while (nb_args < MAX_ARGS && args_present[nb_args])
406
        nb_args++;
407
    for(i = nb_args; i < MAX_ARGS; i++) {
408
        if (args_present[i])
409
            error("inconsistent argument numbering in %s", name);
410
    }
411

    
412
    if (gen_switch == 2) {
413
        fprintf(outfile, "DEF(%s, %d)\n", name + 3, nb_args);
414
    } else if (gen_switch == 1) {
415

    
416
        /* output C code */
417
        fprintf(outfile, "case INDEX_%s: {\n", name);
418
        if (nb_args > 0) {
419
            fprintf(outfile, "    long ");
420
            for(i = 0; i < nb_args; i++) {
421
                if (i != 0)
422
                    fprintf(outfile, ", ");
423
                fprintf(outfile, "param%d", i + 1);
424
            }
425
            fprintf(outfile, ";\n");
426
        }
427
        fprintf(outfile, "    extern void %s();\n", name);
428

    
429
        for(i = 0, rel = relocs;i < nb_relocs; i++, rel++) {
430
            if (rel->r_offset >= offset && rel->r_offset < offset + copy_size) {
431
                sym_name = strtab + symtab[ELFW(R_SYM)(rel->r_info)].st_name;
432
                if (*sym_name && !strstart(sym_name, "__op_param", &p)) {
433
#if defined(HOST_SPARC)
434
                    if (sym_name[0] == '.') {
435
                        fprintf(outfile,
436
                                "extern char __dot_%s __asm__(\"%s\");\n",
437
                                sym_name+1, sym_name);
438
                        continue;
439
                    }
440
#endif
441
                    fprintf(outfile, "extern char %s;\n", sym_name);
442
                }
443
            }
444
        }
445

    
446
        fprintf(outfile, "    memcpy(gen_code_ptr, &%s, %d);\n", name, copy_size);
447
        for(i = 0; i < nb_args; i++) {
448
            fprintf(outfile, "    param%d = *opparam_ptr++;\n", i + 1);
449
        }
450

    
451
        /* patch relocations */
452
#if defined(HOST_I386)
453
            {
454
                char name[256];
455
                int type;
456
                int addend;
457
                for(i = 0, rel = relocs;i < nb_relocs; i++, rel++) {
458
                if (rel->r_offset >= offset && rel->r_offset < offset + copy_size) {
459
                    sym_name = strtab + symtab[ELFW(R_SYM)(rel->r_info)].st_name;
460
                    if (strstart(sym_name, "__op_param", &p)) {
461
                        snprintf(name, sizeof(name), "param%s", p);
462
                    } else {
463
                        snprintf(name, sizeof(name), "(long)(&%s)", sym_name);
464
                    }
465
                    type = ELF32_R_TYPE(rel->r_info);
466
                    addend = get32((uint32_t *)(text + rel->r_offset));
467
                    switch(type) {
468
                    case R_386_32:
469
                        fprintf(outfile, "    *(uint32_t *)(gen_code_ptr + %d) = %s + %d;\n", 
470
                                rel->r_offset - offset, name, addend);
471
                        break;
472
                    case R_386_PC32:
473
                        fprintf(outfile, "    *(uint32_t *)(gen_code_ptr + %d) = %s - (long)(gen_code_ptr + %d) + %d;\n", 
474
                                rel->r_offset - offset, name, rel->r_offset - offset, addend);
475
                        break;
476
                    default:
477
                        error("unsupported i386 relocation (%d)", type);
478
                    }
479
                }
480
                }
481
            }
482
#elif defined(HOST_PPC)
483
            {
484
                char name[256];
485
                int type;
486
                int addend;
487
                for(i = 0, rel = relocs;i < nb_relocs; i++, rel++) {
488
                    if (rel->r_offset >= offset && rel->r_offset < offset + copy_size) {
489
                        sym_name = strtab + symtab[ELFW(R_SYM)(rel->r_info)].st_name;
490
                        if (strstart(sym_name, "__op_param", &p)) {
491
                            snprintf(name, sizeof(name), "param%s", p);
492
                        } else {
493
                            snprintf(name, sizeof(name), "(long)(&%s)", sym_name);
494
                        }
495
                        type = ELF32_R_TYPE(rel->r_info);
496
                        addend = rel->r_addend;
497
                        switch(type) {
498
                        case R_PPC_ADDR32:
499
                            fprintf(outfile, "    *(uint32_t *)(gen_code_ptr + %d) = %s + %d;\n", 
500
                                    rel->r_offset - offset, name, addend);
501
                            break;
502
                        case R_PPC_ADDR16_LO:
503
                            fprintf(outfile, "    *(uint16_t *)(gen_code_ptr + %d) = (%s + %d);\n", 
504
                                    rel->r_offset - offset, name, addend);
505
                            break;
506
                        case R_PPC_ADDR16_HI:
507
                            fprintf(outfile, "    *(uint16_t *)(gen_code_ptr + %d) = (%s + %d) >> 16;\n", 
508
                                    rel->r_offset - offset, name, addend);
509
                            break;
510
                        case R_PPC_ADDR16_HA:
511
                            fprintf(outfile, "    *(uint16_t *)(gen_code_ptr + %d) = (%s + %d + 0x8000) >> 16;\n", 
512
                                    rel->r_offset - offset, name, addend);
513
                            break;
514
                        case R_PPC_REL24:
515
                            /* warning: must be at 32 MB distancy */
516
                            fprintf(outfile, "    *(uint32_t *)(gen_code_ptr + %d) = (*(uint32_t *)(gen_code_ptr + %d) & ~0x03fffffc) | ((%s - (long)(gen_code_ptr + %d) + %d) & 0x03fffffc);\n", 
517
                                    rel->r_offset - offset, rel->r_offset - offset, name, rel->r_offset - offset, addend);
518
                            break;
519
                        default:
520
                            error("unsupported powerpc relocation (%d)", type);
521
                        }
522
                    }
523
                }
524
            }
525
#elif defined(HOST_S390)
526
            {
527
                char name[256];
528
                int type;
529
                int addend;
530
                for(i = 0, rel = relocs;i < nb_relocs; i++, rel++) {
531
                    if (rel->r_offset >= offset && rel->r_offset < offset + copy_size) {
532
                        sym_name = strtab + symtab[ELFW(R_SYM)(rel->r_info)].st_name;
533
                        if (strstart(sym_name, "__op_param", &p)) {
534
                            snprintf(name, sizeof(name), "param%s", p);
535
                        } else {
536
                            snprintf(name, sizeof(name), "(long)(&%s)", sym_name);
537
                        }
538
                        type = ELF32_R_TYPE(rel->r_info);
539
                        addend = rel->r_addend;
540
                        switch(type) {
541
                        case R_390_32:
542
                            fprintf(outfile, "    *(uint32_t *)(gen_code_ptr + %d) = %s + %d;\n", 
543
                                    rel->r_offset - offset, name, addend);
544
                            break;
545
                        case R_390_16:
546
                            fprintf(outfile, "    *(uint16_t *)(gen_code_ptr + %d) = %s + %d;\n", 
547
                                    rel->r_offset - offset, name, addend);
548
                            break;
549
                        case R_390_8:
550
                            fprintf(outfile, "    *(uint8_t *)(gen_code_ptr + %d) = %s + %d;\n", 
551
                                    rel->r_offset - offset, name, addend);
552
                            break;
553
                        default:
554
                            error("unsupported s390 relocation (%d)", type);
555
                        }
556
                    }
557
                }
558
            }
559
#elif defined(HOST_ALPHA)
560
            {
561
                for (i = 0, rel = relocs; i < nb_relocs; i++, rel++) {
562
                    if (rel->r_offset >= offset && rel->r_offset < offset + copy_size) {
563
                        int type;
564

    
565
                        type = ELF64_R_TYPE(rel->r_info);
566
                        sym_name = strtab + symtab[ELF64_R_SYM(rel->r_info)].st_name;
567
                        switch (type) {
568
                        case R_ALPHA_GPDISP:
569
                            /* The gp is just 32 bit, and never changes, so it's easiest to emit it
570
                               as an immediate instead of constructing it from the pv or ra.  */
571
                            fprintf(outfile, "    immediate_ldah(gen_code_ptr + %ld, gp);\n",
572
                                    rel->r_offset - offset);
573
                            fprintf(outfile, "    immediate_lda(gen_code_ptr + %ld, gp);\n",
574
                                    rel->r_offset - offset + rel->r_addend);
575
                            break;
576
                        case R_ALPHA_LITUSE:
577
                            /* jsr to literal hint. Could be used to optimize to bsr. Ignore for
578
                               now, since some called functions (libc) need pv to be set up.  */
579
                            break;
580
                        case R_ALPHA_HINT:
581
                            /* Branch target prediction hint. Ignore for now.  Should be already
582
                               correct for in-function jumps.  */
583
                            break;
584
                        case R_ALPHA_LITERAL:
585
                            /* Load a literal from the GOT relative to the gp.  Since there's only a
586
                               single gp, nothing is to be done.  */
587
                            break;
588
                        case R_ALPHA_GPRELHIGH:
589
                            /* Handle fake relocations against __op_param symbol.  Need to emit the
590
                               high part of the immediate value instead.  Other symbols need no
591
                               special treatment.  */
592
                            if (strstart(sym_name, "__op_param", &p))
593
                                fprintf(outfile, "    immediate_ldah(gen_code_ptr + %ld, param%s);\n",
594
                                        rel->r_offset - offset, p);
595
                            break;
596
                        case R_ALPHA_GPRELLOW:
597
                            if (strstart(sym_name, "__op_param", &p))
598
                                fprintf(outfile, "    immediate_lda(gen_code_ptr + %ld, param%s);\n",
599
                                        rel->r_offset - offset, p);
600
                            break;
601
                        case R_ALPHA_BRSGP:
602
                            /* PC-relative jump. Tweak offset to skip the two instructions that try to
603
                               set up the gp from the pv.  */
604
                            fprintf(outfile, "    fix_bsr(gen_code_ptr + %ld, (uint8_t *) &%s - (gen_code_ptr + %ld) + 4);\n",
605
                                    rel->r_offset - offset, sym_name, rel->r_offset - offset);
606
                            break;
607
                        default:
608
                            error("unsupported Alpha relocation (%d)", type);
609
                        }
610
                    }
611
                }
612
            }
613
#elif defined(HOST_IA64)
614
            {
615
                char name[256];
616
                int type;
617
                int addend;
618
                for(i = 0, rel = relocs;i < nb_relocs; i++, rel++) {
619
                    if (rel->r_offset >= offset && rel->r_offset < offset + copy_size) {
620
                        sym_name = strtab + symtab[ELF64_R_SYM(rel->r_info)].st_name;
621
                        if (strstart(sym_name, "__op_param", &p)) {
622
                            snprintf(name, sizeof(name), "param%s", p);
623
                        } else {
624
                            snprintf(name, sizeof(name), "(long)(&%s)", sym_name);
625
                        }
626
                        type = ELF64_R_TYPE(rel->r_info);
627
                        addend = rel->r_addend;
628
                        switch(type) {
629
                        case R_IA64_LTOFF22:
630
                            error("must implemnt R_IA64_LTOFF22 relocation");
631
                        case R_IA64_PCREL21B:
632
                            error("must implemnt R_IA64_PCREL21B relocation");
633
                        default:
634
                            error("unsupported ia64 relocation (%d)", type);
635
                        }
636
                    }
637
                }
638
            }
639
#elif defined(HOST_SPARC)
640
            {
641
                char name[256];
642
                int type;
643
                int addend;
644
                for(i = 0, rel = relocs;i < nb_relocs; i++, rel++) {
645
                    if (rel->r_offset >= offset && rel->r_offset < offset + copy_size) {
646
                        sym_name = strtab + symtab[ELF32_R_SYM(rel->r_info)].st_name;
647
                        if (strstart(sym_name, "__op_param", &p)) {
648
                            snprintf(name, sizeof(name), "param%s", p);
649
                        } else {
650
                                if (sym_name[0] == '.')
651
                                        snprintf(name, sizeof(name),
652
                                                 "(long)(&__dot_%s)",
653
                                                 sym_name + 1);
654
                                else
655
                                        snprintf(name, sizeof(name),
656
                                                 "(long)(&%s)", sym_name);
657
                        }
658
                        type = ELF32_R_TYPE(rel->r_info);
659
                        addend = rel->r_addend;
660
                        switch(type) {
661
                        case R_SPARC_32:
662
                            fprintf(outfile, "    *(uint32_t *)(gen_code_ptr + %d) = %s + %d;\n", 
663
                                    rel->r_offset - offset, name, addend);
664
                            break;
665
                        case R_SPARC_HI22:
666
                            fprintf(outfile,
667
                                    "    *(uint32_t *)(gen_code_ptr + %d) = "
668
                                    "((*(uint32_t *)(gen_code_ptr + %d)) "
669
                                    " & ~0x3fffff) "
670
                                    " | ((%s + %d) & 0x3fffff);\n",
671
                                    rel->r_offset - offset,
672
                                    rel->r_offset - offset,
673
                                    name, addend);
674
                            break;
675
                        case R_SPARC_LO10:
676
                            fprintf(outfile,
677
                                    "    *(uint32_t *)(gen_code_ptr + %d) = "
678
                                    "((*(uint32_t *)(gen_code_ptr + %d)) "
679
                                    " & ~0x3ff) "
680
                                    " | ((%s + %d) & 0x3ff);\n",
681
                                    rel->r_offset - offset,
682
                                    rel->r_offset - offset,
683
                                    name, addend);
684
                            break;
685
                        case R_SPARC_WDISP30:
686
                            fprintf(outfile,
687
                                    "    *(uint32_t *)(gen_code_ptr + %d) = "
688
                                    "((*(uint32_t *)(gen_code_ptr + %d)) "
689
                                    " & ~0x3fffffff) "
690
                                    " | ((((%s + %d) - (long)gen_code_ptr)>>2) "
691
                                    "    & 0x3fffffff);\n",
692
                                    rel->r_offset - offset,
693
                                    rel->r_offset - offset,
694
                                    name, addend);
695
                            break;
696
                        default:
697
                            error("unsupported sparc relocation (%d)", type);
698
                        }
699
                    }
700
                }
701
            }
702
#elif defined(HOST_SPARC64)
703
            {
704
                char name[256];
705
                int type;
706
                int addend;
707
                for(i = 0, rel = relocs;i < nb_relocs; i++, rel++) {
708
                    if (rel->r_offset >= offset && rel->r_offset < offset + copy_size) {
709
                        sym_name = strtab + symtab[ELF64_R_SYM(rel->r_info)].st_name;
710
                        if (strstart(sym_name, "__op_param", &p)) {
711
                            snprintf(name, sizeof(name), "param%s", p);
712
                        } else {
713
                            snprintf(name, sizeof(name), "(long)(&%s)", sym_name);
714
                        }
715
                        type = ELF64_R_TYPE(rel->r_info);
716
                        addend = rel->r_addend;
717
                        switch(type) {
718
                        case R_SPARC_32:
719
                            fprintf(outfile, "    *(uint32_t *)(gen_code_ptr + %d) = %s + %d;\n",
720
                                    rel->r_offset - offset, name, addend);
721
                            break;
722
                        case R_SPARC_HI22:
723
                            fprintf(outfile,
724
                                    "    *(uint32_t *)(gen_code_ptr + %d) = "
725
                                    "((*(uint32_t *)(gen_code_ptr + %d)) "
726
                                    " & ~0x3fffff) "
727
                                    " | ((%s + %d) & 0x3fffff);\n",
728
                                    rel->r_offset - offset,
729
                                    rel->r_offset - offset,
730
                                    name, addend);
731
                            break;
732
                        case R_SPARC_LO10:
733
                            fprintf(outfile,
734
                                    "    *(uint32_t *)(gen_code_ptr + %d) = "
735
                                    "((*(uint32_t *)(gen_code_ptr + %d)) "
736
                                    " & ~0x3ff) "
737
                                    " | ((%s + %d) & 0x3ff);\n",
738
                                    rel->r_offset - offset,
739
                                    rel->r_offset - offset,
740
                                    name, addend);
741
                            break;
742
                        case R_SPARC_WDISP30:
743
                            fprintf(outfile,
744
                                    "    *(uint32_t *)(gen_code_ptr + %d) = "
745
                                    "((*(uint32_t *)(gen_code_ptr + %d)) "
746
                                    " & ~0x3fffffff) "
747
                                    " | ((((%s + %d) - (long)gen_code_ptr)>>2) "
748
                                    "    & 0x3fffffff);\n",
749
                                    rel->r_offset - offset,
750
                                    rel->r_offset - offset,
751
                                    name, addend);
752
                            break;
753
                        default:
754
                            error("unsupported sparc64 relocation (%d)", type);
755
                        }
756
                    }
757
                }
758
            }
759
#else
760
#error unsupported CPU
761
#endif
762
        fprintf(outfile, "    gen_code_ptr += %d;\n", copy_size);
763
        fprintf(outfile, "}\n");
764
        fprintf(outfile, "break;\n\n");
765
    } else {
766
        fprintf(outfile, "static inline void gen_%s(", name);
767
        if (nb_args == 0) {
768
            fprintf(outfile, "void");
769
        } else {
770
            for(i = 0; i < nb_args; i++) {
771
                if (i != 0)
772
                    fprintf(outfile, ", ");
773
                fprintf(outfile, "long param%d", i + 1);
774
            }
775
        }
776
        fprintf(outfile, ")\n");
777
        fprintf(outfile, "{\n");
778
        for(i = 0; i < nb_args; i++) {
779
            fprintf(outfile, "    *gen_opparam_ptr++ = param%d;\n", i + 1);
780
        }
781
        fprintf(outfile, "    *gen_opc_ptr++ = INDEX_%s;\n", name);
782
        fprintf(outfile, "}\n\n");
783
    }
784
}
785

    
786
/* load an elf object file */
787
int load_elf(const char *filename, FILE *outfile, int do_print_enum)
788
{
789
    int fd;
790
    struct elfhdr ehdr;
791
    struct elf_shdr *sec, *shdr, *symtab_sec, *strtab_sec, *text_sec;
792
    int i, j, nb_syms;
793
    ElfW(Sym) *symtab, *sym;
794
    char *shstr, *strtab;
795
    uint8_t *text;
796
    void *relocs;
797
    int nb_relocs, reloc_sh_type;
798
    
799
    fd = open(filename, O_RDONLY);
800
    if (fd < 0) 
801
        error("can't open file '%s'", filename);
802
    
803
    /* Read ELF header.  */
804
    if (read(fd, &ehdr, sizeof (ehdr)) != sizeof (ehdr))
805
        error("unable to read file header");
806

    
807
    /* Check ELF identification.  */
808
    if (ehdr.e_ident[EI_MAG0] != ELFMAG0
809
     || ehdr.e_ident[EI_MAG1] != ELFMAG1
810
     || ehdr.e_ident[EI_MAG2] != ELFMAG2
811
     || ehdr.e_ident[EI_MAG3] != ELFMAG3
812
     || ehdr.e_ident[EI_VERSION] != EV_CURRENT) {
813
        error("bad ELF header");
814
    }
815

    
816
    do_swap = elf_must_swap(&ehdr);
817
    if (do_swap)
818
        elf_swap_ehdr(&ehdr);
819
    if (ehdr.e_ident[EI_CLASS] != ELF_CLASS)
820
        error("Unsupported ELF class");
821
    if (ehdr.e_type != ET_REL)
822
        error("ELF object file expected");
823
    if (ehdr.e_version != EV_CURRENT)
824
        error("Invalid ELF version");
825
    if (!elf_check_arch(ehdr.e_machine))
826
        error("Unsupported CPU (e_machine=%d)", ehdr.e_machine);
827

    
828
    /* read section headers */
829
    shdr = load_data(fd, ehdr.e_shoff, ehdr.e_shnum * sizeof(struct elf_shdr));
830
    if (do_swap) {
831
        for(i = 0; i < ehdr.e_shnum; i++) {
832
            elf_swap_shdr(&shdr[i]);
833
        }
834
    }
835

    
836
    sec = &shdr[ehdr.e_shstrndx];
837
    shstr = load_data(fd, sec->sh_offset, sec->sh_size);
838

    
839
    /* text section */
840

    
841
    text_sec = find_elf_section(shdr, ehdr.e_shnum, shstr, ".text");
842
    if (!text_sec)
843
        error("could not find .text section");
844
    text = load_data(fd, text_sec->sh_offset, text_sec->sh_size);
845

    
846
    /* find text relocations, if any */
847
    nb_relocs = 0;
848
    relocs = NULL;
849
    reloc_sh_type = 0;
850
    for(i = 0; i < ehdr.e_shnum; i++) {
851
        sec = &shdr[i];
852
        if ((sec->sh_type == SHT_REL || sec->sh_type == SHT_RELA) &&
853
            sec->sh_info == (text_sec - shdr)) {
854
            reloc_sh_type = sec->sh_type;
855
            relocs = load_data(fd, sec->sh_offset, sec->sh_size);
856
            nb_relocs = sec->sh_size / sec->sh_entsize;
857
            if (do_swap) {
858
                if (sec->sh_type == SHT_REL) {
859
                    ElfW(Rel) *rel = relocs;
860
                    for(j = 0, rel = relocs; j < nb_relocs; j++, rel++) {
861
                        swabls(&rel->r_offset);
862
                        swabls(&rel->r_info);
863
                    }
864
                } else {
865
                    ElfW(Rela) *rel = relocs;
866
                    for(j = 0, rel = relocs; j < nb_relocs; j++, rel++) {
867
                        swabls(&rel->r_offset);
868
                        swabls(&rel->r_info);
869
                        swabls(&rel->r_addend);
870
                    }
871
                }
872
            }
873
            break;
874
        }
875
    }
876

    
877
    symtab_sec = find_elf_section(shdr, ehdr.e_shnum, shstr, ".symtab");
878
    if (!symtab_sec)
879
        error("could not find .symtab section");
880
    strtab_sec = &shdr[symtab_sec->sh_link];
881

    
882
    symtab = load_data(fd, symtab_sec->sh_offset, symtab_sec->sh_size);
883
    strtab = load_data(fd, strtab_sec->sh_offset, strtab_sec->sh_size);
884
    
885
    nb_syms = symtab_sec->sh_size / sizeof(ElfW(Sym));
886
    if (do_swap) {
887
        for(i = 0, sym = symtab; i < nb_syms; i++, sym++) {
888
            swab32s(&sym->st_name);
889
            swabls(&sym->st_value);
890
            swabls(&sym->st_size);
891
            swab16s(&sym->st_shndx);
892
        }
893
    }
894

    
895
    if (do_print_enum) {
896
        fprintf(outfile, "DEF(end, 0)\n");
897
        for(i = 0, sym = symtab; i < nb_syms; i++, sym++) {
898
            const char *name, *p;
899
            name = strtab + sym->st_name;
900
            if (strstart(name, OP_PREFIX, &p)) {
901
                gen_code(name, sym->st_value, sym->st_size, outfile, 
902
                         text, relocs, nb_relocs, reloc_sh_type, symtab, strtab, 2);
903
            }
904
        }
905
    } else {
906
        /* generate big code generation switch */
907
#ifdef HOST_ALPHA
908
fprintf(outfile,
909
"register int gp asm(\"$29\");\n"
910
"static inline void immediate_ldah(void *p, int val) {\n"
911
"    uint32_t *dest = p;\n"
912
"    long high = ((val >> 16) + ((val >> 15) & 1)) & 0xffff;\n"
913
"\n"
914
"    *dest &= ~0xffff;\n"
915
"    *dest |= high;\n"
916
"    *dest |= 31 << 16;\n"
917
"}\n"
918
"static inline void immediate_lda(void *dest, int val) {\n"
919
"    *(uint16_t *) dest = val;\n"
920
"}\n"
921
"void fix_bsr(void *p, int offset) {\n"
922
"    uint32_t *dest = p;\n"
923
"    *dest &= ~((1 << 21) - 1);\n"
924
"    *dest |= (offset >> 2) & ((1 << 21) - 1);\n"
925
"}\n");
926
#endif
927
fprintf(outfile,
928
"int dyngen_code(uint8_t *gen_code_buf,\n"
929
"                const uint16_t *opc_buf, const uint32_t *opparam_buf)\n"
930
"{\n"
931
"    uint8_t *gen_code_ptr;\n"
932
"    const uint16_t *opc_ptr;\n"
933
"    const uint32_t *opparam_ptr;\n"
934
"    gen_code_ptr = gen_code_buf;\n"
935
"    opc_ptr = opc_buf;\n"
936
"    opparam_ptr = opparam_buf;\n"
937
"    for(;;) {\n"
938
"        switch(*opc_ptr++) {\n"
939
);
940

    
941
        for(i = 0, sym = symtab; i < nb_syms; i++, sym++) {
942
            const char *name;
943
            name = strtab + sym->st_name;
944
            if (strstart(name, OP_PREFIX, NULL)) {
945
#if 0
946
                printf("%4d: %s pos=0x%08x len=%d\n", 
947
                       i, name, sym->st_value, sym->st_size);
948
#endif
949
                if (sym->st_shndx != (text_sec - shdr))
950
                    error("invalid section for opcode (0x%x)", sym->st_shndx);
951
                gen_code(name, sym->st_value, sym->st_size, outfile, 
952
                         text, relocs, nb_relocs, reloc_sh_type, symtab, strtab, 1);
953
            }
954
        }
955

    
956
fprintf(outfile,
957
"        default:\n"
958
"            goto the_end;\n"
959
"        }\n"
960
"    }\n"
961
" the_end:\n"
962
);
963

    
964
/* generate a return */ 
965
    switch(ELF_ARCH) {
966
    case EM_386:
967
        fprintf(outfile, "*gen_code_ptr++ = 0xc3; /* ret */\n");
968
        break;
969
    case EM_PPC:
970
        fprintf(outfile, "*((uint32_t *)gen_code_ptr)++ = 0x4e800020; /* blr */\n");
971
        break;
972
    case EM_S390:
973
        fprintf(outfile, "*((uint16_t *)gen_code_ptr)++ = 0x07fe; /* br %%r14 */\n");
974
        break;
975
    case EM_ALPHA:
976
        fprintf(outfile, "*((uint32_t *)gen_code_ptr)++ = 0x6bfa8001; /* ret */\n");
977
        break;
978
    case EM_IA_64:
979
        fprintf(outfile, "*((uint32_t *)gen_code_ptr)++ = 0x00840008; /* br.ret.sptk.many b0;; */\n");
980
        break;
981
    case EM_SPARC:
982
    case EM_SPARC32PLUS:
983
    case EM_SPARCV9:
984
        /* Fill the delay slot. */
985
        fprintf(outfile, "*((uint32_t *)gen_code_ptr) = *((uint32_t *)gen_code_ptr - 1); /* delay slot */\n");
986
        fprintf(outfile, "*((uint32_t *)gen_code_ptr - 1) = 0x81c3e008; /* retl */\n");
987
        fprintf(outfile, "gen_code_ptr++;\n");
988
        break;
989
    default:
990
        error("unknown ELF architecture");
991
    }
992
    
993
    fprintf(outfile, "return gen_code_ptr -  gen_code_buf;\n");
994
    fprintf(outfile, "}\n\n");
995

    
996
/* generate gen_xxx functions */
997
/* XXX: suppress the use of these functions to simplify code */
998
        for(i = 0, sym = symtab; i < nb_syms; i++, sym++) {
999
            const char *name;
1000
            name = strtab + sym->st_name;
1001
            if (strstart(name, OP_PREFIX, NULL)) {
1002
                if (sym->st_shndx != (text_sec - shdr))
1003
                    error("invalid section for opcode (0x%x)", sym->st_shndx);
1004
                gen_code(name, sym->st_value, sym->st_size, outfile, 
1005
                         text, relocs, nb_relocs, reloc_sh_type, symtab, strtab, 0);
1006
            }
1007
        }
1008
    }
1009

    
1010
    close(fd);
1011
    return 0;
1012
}
1013

    
1014
void usage(void)
1015
{
1016
    printf("dyngen (c) 2003 Fabrice Bellard\n"
1017
           "usage: dyngen [-o outfile] [-c] objfile\n"
1018
           "Generate a dynamic code generator from an object file\n"
1019
           "-c     output enum of operations\n"
1020
           );
1021
    exit(1);
1022
}
1023

    
1024
int main(int argc, char **argv)
1025
{
1026
    int c, do_print_enum;
1027
    const char *filename, *outfilename;
1028
    FILE *outfile;
1029

    
1030
    outfilename = "out.c";
1031
    do_print_enum = 0;
1032
    for(;;) {
1033
        c = getopt(argc, argv, "ho:c");
1034
        if (c == -1)
1035
            break;
1036
        switch(c) {
1037
        case 'h':
1038
            usage();
1039
            break;
1040
        case 'o':
1041
            outfilename = optarg;
1042
            break;
1043
        case 'c':
1044
            do_print_enum = 1;
1045
            break;
1046
        }
1047
    }
1048
    if (optind >= argc)
1049
        usage();
1050
    filename = argv[optind];
1051
    outfile = fopen(outfilename, "w");
1052
    if (!outfile)
1053
        error("could not open '%s'", outfilename);
1054
    load_elf(filename, outfile, do_print_enum);
1055
    fclose(outfile);
1056
    return 0;
1057
}