"""
Created on 10 Apr 2013

@author: mike
"""
import copy

from volatility.framework import interfaces, objects


class NativeTable(interfaces.symbols.NativeTableInterface):
    """Symbol List that handles Native types"""

    def __init__(self, name, native_dictionary):
        super().__init__(name, self)
        self._native_dictionary = copy.deepcopy(native_dictionary)
        self._overrides = {}
        for native_type in self._native_dictionary:
            native_class, _native_struct = self._native_dictionary[native_type]
            self._overrides[native_type] = native_class
        # Create this once early, because it may get used a lot
        self._types = set(self._native_dictionary).union(
            {'Enumeration', 'array', 'BitField', 'void', 'pointer', 'String', 'Bytes'})

    def get_type_class(self, name):
        ntype, fmt = native_types.get(name, (objects.Integer, ''))
        return ntype

    @property
    def types(self):
        """Returns an iterator of the symbol type names"""
        return self._types

    def get_type(self, type_name):
        """Resolves a symbol name into an object template

           symbol_space is used to resolve any target symbols if they don't exist in this list
        """
        # NOTE: These need updating whenever the object init signatures change
        additional = {}
        obj = None
        if type_name == 'void':
            obj = objects.Void
        elif type_name == 'array':
            obj = objects.Array
            additional = {"count": 0, "target": self.get_type('void')}
        elif type_name == 'Enumeration':
            obj = objects.Enumeration
            additional = {"target": self.get_type('void'), "choices": {}}
        elif type_name == 'BitField':
            obj = objects.BitField
            additional = {"start_bit": 0, "end_bit": 0}
        elif type_name == 'String':
            obj = objects.String
            additional = {"max_length": 0}
        elif type_name == 'Bytes':
            obj = objects.Bytes
            additional = {"length": 0}
        if obj is not None:
            return objects.templates.ObjectTemplate(obj, type_name = type_name, **additional)

        _native_type, native_format = self._native_dictionary[type_name]
        if type_name == 'pointer':
            additional = {'target': self.get_type('void')}
        return objects.templates.ObjectTemplate(self.get_type_class(type_name),  # pylint: disable=W0142
                                                type_name = type_name,
                                                struct_format = native_format,
                                                **additional)


native_types = {'int': (objects.Integer, '<i'),
                'long': (objects.Integer, '<i'),
                'unsigned long': (objects.Integer, '<I'),
                'unsigned int': (objects.Integer, '<I'),
                'pointer': (objects.Pointer, '<I'),
                'char': (objects.Integer, '<b'),
                'byte': (objects.Bytes, '<c'),
                'unsigned char': (objects.Integer, '<B'),
                'unsigned short int': (objects.Integer, '<H'),
                'unsigned short': (objects.Integer, '<H'),
                'unsigned be short': (objects.Integer, '>H'),
                'short': (objects.Integer, '<h'),
                'long long': (objects.Integer, '<q'),
                'unsigned long long': (objects.Integer, '<Q'),
                'float': (objects.Float, "<d"),
                'double': (objects.Float, "<d")}
x86NativeTable = NativeTable("native", native_types)
native_types['pointer'] = (objects.Pointer, '<Q')
x64NativeTable = NativeTable("native", native_types)