summaryrefslogtreecommitdiff
path: root/tools/libdol2asm/data/function/store.py
blob: 794a26bddea5092144f61f8a9e9ed5f6ac1c7ef2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54


from dataclasses import dataclass, field
from capstone.ppc import PPC_REG_R3

from ...builder import AsyncBuilder
from ...types import *
from .base import *


@dataclass(eq=False)
class StoreFunction(Function):
    async def export_store(self, exporter, builder):
        assert False

    async def export_function_body(self, exporter, builder: AsyncBuilder):
        await builder.write(f" {{")
        await self.export_store(exporter, builder)
        await builder.write(f"}}")


@dataclass(eq=False)
class Store_R3_OffsetRX_Function(StoreFunction):
    dst: int = 0
    dst_offset: int = 0
    src: int = 0
    store_type: Type = None

    def calculate_params(self):
        params = {}

        gr = PPC_REG_R3
        if self.has_class:
            if self.demangled_name and self.demangled_name.require_specialization:
                params[gr] = "_this"
            else:
                params[gr] = "this"
            gr += 1

        for i, arg in enumerate(self.argument_types):
            params[gr] = f"param_{i}"
            gr += 1

        return params

    async def export_store(self, exporter, builder):
        params = self.calculate_params()
        dst = params[self.dst]
        src = params[self.src]
        if self.dst_offset > 0:
            dst = f"(((u8*){params[self.dst]})+{self.dst_offset}) /* {params[self.dst]}->field_0x{self.dst_offset:x} */"

        pointer_type = PointerType(self.store_type)
        await builder.write(f"\t*({pointer_type.type()}){dst} = ({self.store_type.type()})({src});")