From d8b4e87af669c2b260686a5db67f7f02b4c164d9 Mon Sep 17 00:00:00 2001
From: lemon <lsof@mailbox.org>
Date: Sun, 14 Dec 2025 12:15:59 +0100
Subject: various relocation related optimization

With 59ca5a8db, querying if a symbol is defined is cheap. If we're
compiling code that calls foo() and we defined foo() in this compilation
unit, we already know its offset within the .text section, so use it
instead of emitting a relocation for the linker to handle.  Also, put
small literal data in the .text section instead of .rodata.  This seems
to improve performance (cache locality?), and as a bonus, it will be
good for aarch64's instr encoding with smallish PC-relative offsets.
---
 obj/elf.c | 29 +++++++++++++++--------------
 1 file changed, 15 insertions(+), 14 deletions(-)

(limited to 'obj/elf.c')

diff --git a/obj/elf.c b/obj/elf.c
index e20da99..521a159 100644
--- a/obj/elf.c
+++ b/obj/elf.c
@@ -91,16 +91,23 @@ enum {
    BSS_SHNDX    = 4,
 };
 
+static const char sect2ndx[] = {
+   [Snone] = SHN_UND,
+   [Stext] = TEXT_SHNDX, [Srodata] = RODATA_SHNDX,
+   [Sdata] = DATA_SHNDX, [Sbss] = BSS_SHNDX,
+}, shndx2sect[] = {
+   [SHN_UND] = Snone,
+   [TEXT_SHNDX] = Stext, [RODATA_SHNDX] = Srodata,
+   [DATA_SHNDX] = Sdata, [BSS_SHNDX] = Sbss,
+};
+
 enum section
-elfhassym(const char *nam)
+elfhassym(const char *nam, uint *value)
 {
    struct sym *sym = findsym(nam);
-   if (sym) switch (sym->shndx) {
-   case SHN_UND: return Snone;
-   case TEXT_SHNDX: return Stext;
-   case RODATA_SHNDX: return Srodata;
-   case DATA_SHNDX: return Sdata;
-   case BSS_SHNDX: return Sbss;
+   if (sym) {
+      if (value) *value = sym->value;
+      return shndx2sect[sym->shndx];
    }
    return Snone;
 }
@@ -115,13 +122,7 @@ elfaddsym(const char *nam, int info, enum section sect, uvlong value, uvlong siz
    }
    sym->bind = info >> 4;
    sym->type = info & 0xF;
-   switch (sect) {
-   case Snone:   sym->shndx = SHN_UND; break;
-   case Stext:   sym->shndx = TEXT_SHNDX; break;
-   case Srodata: sym->shndx = RODATA_SHNDX; break;
-   case Sdata:   sym->shndx = DATA_SHNDX; break;
-   case Sbss:    sym->shndx = BSS_SHNDX; break;
-   }
+   sym->shndx = sect2ndx[sect];
    sym->value = value;
    sym->size = size;
    if (sym == &sym0) {
-- 
cgit v1.2.3