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#include <errno.h>
  12#include <stdbool.h>
  13#include <stdint.h>
  14#include <stdio.h>
  15#include <stdlib.h>
  16#include <string.h>
  17
  18#include <elf.h>
  19#include <fcntl.h>
  20#include <sys/mman.h>
  21#include <sys/stat.h>
  22#include <sys/types.h>
  23#include <unistd.h>
  24#include <compiler.h>
  25
  26#ifndef EM_RISCV
  27#define EM_RISCV 243
  28#endif
  29
  30#ifndef R_RISCV_32
  31#define R_RISCV_32 1
  32#endif
  33
  34#ifndef R_RISCV_64
  35#define R_RISCV_64 2
  36#endif
  37
  38#ifndef R_RISCV_RELATIVE
  39#define R_RISCV_RELATIVE 3
  40#endif
  41
  42const char *argv0;
  43
  44#define die(fmt, ...) \
  45        do { \
  46                fprintf(stderr, "%s: " fmt "\n", argv0, ## __VA_ARGS__); \
  47                exit(EXIT_FAILURE); \
  48        } while (0)
  49
  50#define PRELINK_BYTEORDER le
  51#define PRELINK_INC_BITS 32
  52#include "prelink-riscv.inc"
  53#undef PRELINK_BYTEORDER
  54#undef PRELINK_INC_BITS
  55
  56#define PRELINK_BYTEORDER le
  57#define PRELINK_INC_BITS 64
  58#include "prelink-riscv.inc"
  59#undef PRELINK_BYTEORDER
  60#undef PRELINK_INC_BITS
  61
  62#define PRELINK_BYTEORDER be
  63#define PRELINK_INC_BITS 32
  64#include "prelink-riscv.inc"
  65#undef PRELINK_BYTEORDER
  66#undef PRELINK_INC_BITS
  67
  68#define PRELINK_BYTEORDER be
  69#define PRELINK_INC_BITS 64
  70#include "prelink-riscv.inc"
  71#undef PRELINK_BYTEORDER
  72#undef PRELINK_INC_BITS
  73
  74int main(int argc, const char *const *argv)
  75{
  76        argv0 = argv[0];
  77
  78        if (argc < 2) {
  79                fprintf(stderr, "Usage: %s <u-boot>\n", argv0);
  80                exit(EXIT_FAILURE);
  81        }
  82
  83        int fd = open(argv[1], O_RDWR, 0);
  84
  85        if (fd < 0)
  86                die("Cannot open %s: %s", argv[1], strerror(errno));
  87
  88        struct stat st;
  89
  90        if (fstat(fd, &st) < 0)
  91                die("Cannot stat %s: %s", argv[1], strerror(errno));
  92
  93        void *data =
  94                mmap(0, st.st_size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
  95
  96        if (data == MAP_FAILED)
  97                die("Cannot mmap %s: %s", argv[1], strerror(errno));
  98
  99        close(fd);
 100
 101        unsigned char *e_ident = (unsigned char *)data;
 102
 103        if (memcmp(e_ident, ELFMAG, SELFMAG) != 0)
 104                die("Invalid ELF file %s", argv[1]);
 105
 106        bool is64 = e_ident[EI_CLASS] == ELFCLASS64;
 107        bool isbe = e_ident[EI_DATA] == ELFDATA2MSB;
 108
 109        if (is64) {
 110                if (isbe)
 111                        prelink_be64(data);
 112                else
 113                        prelink_le64(data);
 114        } else {
 115                if (isbe)
 116                        prelink_be32(data);
 117                else
 118                        prelink_le32(data);
 119        }
 120
 121        return 0;
 122}
 123