143 lines
3.5 KiB
Go
143 lines
3.5 KiB
Go
// Copyright (c) 2026 Petr Balvín <opensource@petrbalvin.org> (https://petrbalvin.org)
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package asm
|
|
|
|
import (
|
|
"bytes"
|
|
"debug/elf"
|
|
"testing"
|
|
|
|
"sourcedock.dev/petrbalvin/gasm-devkit/parser"
|
|
)
|
|
|
|
// TestELFAARCH64Object checks the structure of the emitted AArch64 ELF64
|
|
// relocatable object: sections, the symbol table (bindings, types, values,
|
|
// sizes) and the .rela.text relocation pair for the static-symbol load,
|
|
// parsed back with debug/elf.
|
|
func TestELFAARCH64Object(t *testing.T) {
|
|
f, errs := parser.Parse("k_arm64.s", `
|
|
#include "textflag.h"
|
|
|
|
TEXT ·add(SB), NOSPLIT, $0-24
|
|
MOVD a+0(FP), R4
|
|
MOVD b+8(FP), R5
|
|
ADD R5, R4, R4
|
|
MOVD R4, ret+16(FP)
|
|
RET
|
|
|
|
TEXT ·getanswer(SB), NOSPLIT, $0-8
|
|
MOVD answer<>(SB), R4
|
|
MOVD R4, ret+0(FP)
|
|
RET
|
|
|
|
GLOBL answer<>(SB), RODATA, $8
|
|
DATA answer<>+0(SB)/8, $42
|
|
`)
|
|
if len(errs) > 0 {
|
|
t.Fatalf("parse: %v", errs)
|
|
}
|
|
img, err := AssembleFileARM64(f)
|
|
if err != nil {
|
|
t.Fatalf("AssembleFileARM64: %v", err)
|
|
}
|
|
obj, err := img.ELFAARCH64Object()
|
|
if err != nil {
|
|
t.Fatalf("ELFAARCH64Object: %v", err)
|
|
}
|
|
ef, err := elf.NewFile(bytes.NewReader(obj))
|
|
if err != nil {
|
|
t.Fatalf("parse emitted object: %v", err)
|
|
}
|
|
defer ef.Close()
|
|
|
|
if ef.Type != elf.ET_REL || ef.Machine != elf.EM_AARCH64 {
|
|
t.Errorf("type/machine = %v/%v, want ET_REL/EM_AARCH64", ef.Type, ef.Machine)
|
|
}
|
|
|
|
text := ef.Section(".text")
|
|
data := ef.Section(".data")
|
|
if text == nil || data == nil {
|
|
t.Fatal("missing .text or .data section")
|
|
}
|
|
if text.Size == 0 {
|
|
t.Error(".text section is empty")
|
|
}
|
|
|
|
syms, err := ef.Symbols()
|
|
if err != nil {
|
|
t.Fatalf("symbols: %v", err)
|
|
}
|
|
|
|
foundAdd, foundGetanswer, foundAnswer := false, false, false
|
|
for _, s := range syms {
|
|
switch s.Name {
|
|
case "add":
|
|
foundAdd = true
|
|
if elf.SymType(s.Info&0xf) != elf.STT_FUNC || elf.SymBind(s.Info>>4) != elf.STB_GLOBAL {
|
|
t.Errorf("add: info=0x%02x, want STT_FUNC|STB_GLOBAL", s.Info)
|
|
}
|
|
case "getanswer":
|
|
foundGetanswer = true
|
|
if elf.SymType(s.Info&0xf) != elf.STT_FUNC || elf.SymBind(s.Info>>4) != elf.STB_GLOBAL {
|
|
t.Errorf("getanswer: info=0x%02x, want STT_FUNC|STB_GLOBAL", s.Info)
|
|
}
|
|
case "answer":
|
|
foundAnswer = true
|
|
if elf.SymType(s.Info&0xf) != elf.STT_OBJECT || elf.SymBind(s.Info>>4) != elf.STB_LOCAL {
|
|
t.Errorf("answer: info=0x%02x, want STT_OBJECT|STB_LOCAL", s.Info)
|
|
}
|
|
}
|
|
}
|
|
if !foundAdd {
|
|
t.Error("symbol 'add' not found")
|
|
}
|
|
if !foundGetanswer {
|
|
t.Error("symbol 'getanswer' not found")
|
|
}
|
|
if !foundAnswer {
|
|
t.Error("symbol 'answer' not found")
|
|
}
|
|
|
|
// Check that .rela.text exists (getanswer has SB reference).
|
|
relaText := ef.Section(".rela.text")
|
|
if relaText == nil {
|
|
t.Error("missing .rela.text section")
|
|
}
|
|
}
|
|
|
|
// TestELFAARCH64ObjectNoRelocations checks the ELF output when there are no
|
|
// static-symbol references (no .rela.text section).
|
|
func TestELFAARCH64ObjectNoRelocations(t *testing.T) {
|
|
f, errs := parser.Parse("k_arm64.s", `
|
|
#include "textflag.h"
|
|
|
|
TEXT ·add(SB), NOSPLIT, $0-24
|
|
MOVD a+0(FP), R4
|
|
MOVD b+8(FP), R5
|
|
ADD R5, R4, R4
|
|
MOVD R4, ret+16(FP)
|
|
RET
|
|
`)
|
|
if len(errs) > 0 {
|
|
t.Fatalf("parse: %v", errs)
|
|
}
|
|
img, err := AssembleFileARM64(f)
|
|
if err != nil {
|
|
t.Fatalf("AssembleFileARM64: %v", err)
|
|
}
|
|
obj, err := img.ELFAARCH64Object()
|
|
if err != nil {
|
|
t.Fatalf("ELFAARCH64Object: %v", err)
|
|
}
|
|
ef, err := elf.NewFile(bytes.NewReader(obj))
|
|
if err != nil {
|
|
t.Fatalf("parse emitted object: %v", err)
|
|
}
|
|
defer ef.Close()
|
|
|
|
if ef.Section(".rela.text") != nil {
|
|
t.Error("unexpected .rela.text section when there are no relocations")
|
|
}
|
|
}
|