Add o segment type

Author: mkst

docs/Segments.md

@@ -290,6 +290,24 @@ It looks for libraries in the [`lib_path`](https://github.com/ethteck/splat/wiki
 - { type: lib, name: a_lib, object: b_obj, section: .text }
 ```
 
+## `o`
+
+Similar to the `lib` segment but used to reference an object file (for example extracted from a shared library file). Does not extract anything from the input binary.
+
+```yaml
+# link to .text of myobject.o
+- [auto, o, myobject]
+```
+
+```yaml
+# link to .rodata of myobject.o
+- [auto, o, myobject, .rodata]
+```
+
+```yaml
+# link to .text of myobject.o (dict representation)
+- { type: o, name: myobject, section: .text }
+```
 
 ## `pad`
 

src/splat/segtypes/common/o.py

@@ -0,0 +1,50 @@
+from typing import Optional
+
+from pathlib import Path
+
+from ...util import options
+from ..common.lib import CommonSegment
+from ..segment import parse_segment_vram
+
+
+class CommonSegO(CommonSegment):
+    def __init__(
+        self,
+        rom_start: Optional[int],
+        rom_end: Optional[int],
+        type: str,
+        name: str,
+        vram_start: Optional[int],
+        args: list,
+        yaml,
+    ):
+        super().__init__(
+            rom_start,
+            rom_end,
+            type,
+            name,
+            vram_start,
+            args=args,
+            yaml=yaml,
+        )
+
+        vram = parse_segment_vram(self.yaml)
+        if vram is not None:
+            self.vram_start = vram
+
+        if isinstance(yaml, dict):
+            self.section = yaml.get("section", ".text")
+        else:
+            if len(args) > 0:
+                self.section = args[0]
+            else:
+                self.section = ".text"
+
+        self.extract = False
+
+    def get_linker_section(self) -> str:
+        return self.section
+
+    def out_path(self) -> Optional[Path]:
+        out_path = options.opts.build_path / f"{self.name}"
+        return out_path

test.py

@@ -16,6 +16,7 @@
 from src.splat.segtypes.common.code import CommonSegCode
 from src.splat.segtypes.common.c import CommonSegC
 from src.splat.segtypes.common.bss import CommonSegBss
+from src.splat.segtypes.common.o import CommonSegO
 from src.splat.segtypes.segment import Segment
 
 
@@ -368,6 +369,44 @@ def test_disassemble_data(self):
         assert bss.spim_section.get_section().bssVramEnd == 0x300
 
 
+class O(unittest.TestCase):
+    def test_o_text(self):
+        # Segment __init__ requires opts to be initialized
+        test_init()
+
+        o = CommonSegO(
+            rom_start=0x100,
+            rom_end=0x200,
+            vram_start=0x0,
+            type="o",
+            name="myobject",
+            args=[],
+            yaml=None,
+        )
+        expected = Path("test/basic_app/build/myobject")
+        self.assertEqual(expected, o.out_path())
+        self.assertEqual(".text", o.get_linker_section())
+
+    def test_o_rodata(self):
+        # Segment __init__ requires opts to be initialized
+        test_init()
+
+        section = ".rodata"
+        o = CommonSegO(
+            rom_start=0x100,
+            rom_end=0x200,
+            vram_start=0x0,
+            type="o",
+            name="myobject",
+            args=[section],
+            yaml=None,
+        )
+        out_path = Path("test/basic_app/build/myobject")
+
+        self.assertEqual(out_path, o.out_path())
+        self.assertEqual(section, o.get_linker_section())
+
+
 class SymbolsInitialize(unittest.TestCase):
     def test_attrs(self):
         symbols.reset_symbols()