LXR uboot/tools/prelink-riscv.c

   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * Copyright (C) 2017 Andes Technology
   4 * Chih-Mao Chen <cmchen@andestech.com>
   5 *
   6 * Statically process runtime relocations on RISC-V ELF images
   7 * so that it can be directly executed when loaded at LMA
   8 * without fixup. Both RV32 and RV64 are supported.
   9 */
  10
  11#if __BYTE_ORDER__ != __ORDER_LITTLE_ENDIAN__
  12#error "Only little-endian host is supported"
  13#endif
  14
  15#include <errno.h>
  16#include <stdbool.h>
  17#include <stdint.h>
  18#include <stdio.h>
  19#include <stdlib.h>
  20#include <string.h>
  21
  22#include <elf.h>
  23#include <fcntl.h>
  24#include <sys/mman.h>
  25#include <sys/stat.h>
  26#include <sys/types.h>
  27#include <unistd.h>
  28
  29#ifndef EM_RISCV
  30#define EM_RISCV 243
  31#endif
  32
  33#ifndef R_RISCV_32
  34#define R_RISCV_32 1
  35#endif
  36
  37#ifndef R_RISCV_64
  38#define R_RISCV_64 2
  39#endif
  40
  41#ifndef R_RISCV_RELATIVE
  42#define R_RISCV_RELATIVE 3
  43#endif
  44
  45const char *argv0;
  46
  47#define die(fmt, ...) \
  48        do { \
  49                fprintf(stderr, "%s: " fmt "\n", argv0, ## __VA_ARGS__); \
  50                exit(EXIT_FAILURE); \
  51        } while (0)
  52
  53#define PRELINK_INC_BITS 32
  54#include "prelink-riscv.inc"
  55#undef PRELINK_INC_BITS
  56
  57#define PRELINK_INC_BITS 64
  58#include "prelink-riscv.inc"
  59#undef PRELINK_INC_BITS
  60
  61int main(int argc, const char *const *argv)
  62{
  63        argv0 = argv[0];
  64
  65        if (argc < 2) {
  66                fprintf(stderr, "Usage: %s <u-boot>\n", argv0);
  67                exit(EXIT_FAILURE);
  68        }
  69
  70        int fd = open(argv[1], O_RDWR, 0);
  71
  72        if (fd < 0)
  73                die("Cannot open %s: %s", argv[1], strerror(errno));
  74
  75        struct stat st;
  76
  77        if (fstat(fd, &st) < 0)
  78                die("Cannot stat %s: %s", argv[1], strerror(errno));
  79
  80        void *data =
  81                mmap(0, st.st_size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
  82
  83        if (data == MAP_FAILED)
  84                die("Cannot mmap %s: %s", argv[1], strerror(errno));
  85
  86        close(fd);
  87
  88        unsigned char *e_ident = (unsigned char *)data;
  89
  90        if (memcmp(e_ident, ELFMAG, SELFMAG) != 0)
  91                die("Invalid ELF file %s", argv[1]);
  92
  93        bool is64 = e_ident[EI_CLASS] == ELFCLASS64;
  94
  95        if (is64)
  96                prelink64(data);
  97        else
  98                prelink32(data);
  99
 100        return 0;
 101}
 102