<|BOS|> consumers of a resource."""
return _get_cached_tracker().get_resource_versions(resource_type)
def update_versions(consumer, resource_versions):
"""Update the resources' versions for a consumer id."""
_get_cached_tracker().update_versions(consumer, resource_versions)
def report():
"""Report resource versions in debug logs."""
_get_cached_tracker().report()
<|EOS|>from collections import namedtuple, defaultdict
import operator
import idaapi
import idc
from . import exceptions
from .code import lines
FF_TYPES = [idc.FF_BYTE, idc.FF_WORD, idc.FF_DWRD, idc.FF_QWRD, idc.FF_OWRD, ]
FF_SIZES = [1, 2, 4, 8, 16, ]
SIZE_TO_TYPE = dict(zip(FF_SIZES, FF_TYPES))
STRUCT_ERROR_MAP = {
idc.STRUC_ERROR_MEMBER_NAME:
(exceptions.SarkErrorStructMemberName, "already has member with this name (bad name)"),
idc.STRUC_ERROR_MEMBER_OFFSET:
(exceptions.SarkErrorStructMemberOffset, "already has member at this offset"),
idc.STRUC_ERROR_MEMBER_SIZE:
(exceptions.SarkErrorStructMemberSize, "bad number of bytes or bad sizeof(type)"),
idc.STRUC_ERROR_MEMBER_TINFO:
(exceptions.SarkErrorStructMemberTinfo, "bad typeid parameter"),
idc.STRUC_ERROR_MEMBER_STRUCT:
(exceptions.SarkErrorStructMemberStruct, "bad struct id (the 1st argument)"),
idc.STRUC_ERROR_MEMBER_UNIVAR:
(exceptions.SarkErrorStructMemberUnivar, "unions can't have variable sized members"),
idc.STRUC_ERROR_MEMBER_VARLAST:
(exceptions.SarkErrorStructMemberVarlast, "variable sized member should be the last member in the structure"),
}
def struct_member_error(err, sid, name, offset, size):
"""Create and format a struct member exception.
Args:
err: The error value returned from struct member creation
sid: The struct id
name: The member name
offset: Memeber offset
size: Member size
Returns:
A ``SarkErrorAddStructMemeberFailed`` derivative exception, with an
informative message.
"""
exception, msg = STRUCT_ERROR_MAP[err]
struct_name = idc.GetStrucName(sid)
return exception(('AddStructMember(struct="{}", member="{}", offset={}, size={}) '
'failed: {}').format(
struct_name,
name,
offset,
size,
msg
))
def create_struct(name):
"""Create a structure.
Args:
name: The structure's name
Returns:
The sturct ID
Raises:
exceptions.SarkStructAlreadyExists: A struct with the same name already exists
exceptions.SarkCreationFailed: Struct creation failed
"""
sid = idc.GetStrucIdByName(name)
if sid != idaapi.BADADDR:
# The struct already exists.
raise exceptions.SarkStructAlreadyExists("A struct names {!r} already exists.".format(name))
sid = idc.AddStrucEx(-1, name, 0)
if sid == idaapi.BADADDR:
raise exceptions.SarkStructCreationFailed("Struct creation failed.")
return sid
def get_struct(name):
"""Get a struct by it's name.
Args:
name: The name of the struct
Returns:
The struct's id
Raises:
exceptions.SarkStructNotFound: is the struct does not exist.
"""
sid = idc.GetStrucIdByName(name)
if sid == idaapi.BADADDR:
raise exceptions.SarkStructNotFound()
return sid
def size_to_flags(size):
return SIZE_TO_TYPE[size] | idc.FF_DATA
def add_struct_member(sid, name, offset, size):
failure = idc.AddStrucMember(sid, name, offset, size_to_flags(size), -1, size