1
0
mirror of https://github.com/bitdefender/bddisasm.git synced 2024-12-22 14:08:08 +00:00

Improved REG_ID macros - make sure we include block addressing and High8 designator in the reg ID. Alsom, make sure the register size fits in, since the new tile register can be 1K in size, which previously overflowed...

This commit is contained in:
BITDEFENDER\vlutas 2021-12-03 12:44:57 +02:00
parent 4ff620cb76
commit 2f50ce9b4e
4 changed files with 31 additions and 8 deletions

View File

@ -12,7 +12,7 @@ from setuptools import find_packages, setup, Command, Extension, Distribution
from codecs import open
VERSION = (0, 1, 3)
LIBRARY_VERSION = (1, 34, 7)
LIBRARY_VERSION = (1, 34, 8)
LIBRARY_INSTRUX_SIZE = 856
packages = ['pybddisasm']

View File

@ -455,16 +455,39 @@ typedef uint32_t ND_REG_SIZE;
#define ND_SAE_SUPPORT(ix) (!!((ix)->ValidDecorators.Sae))
#define ND_BROADCAST_SUPPORT(ix) (!!((ix)->ValidDecorators.Broadcast))
// Generates a unique ID per register type, size and reg.
#define ND_OP_REG_ID(op) (((op)->Type << 24) | ((op)->Info.Register.Type << 16) | \
((op)->Info.Register.Size << 8) | ((op)->Info.Register.Reg))
// Generates a unique ID per register type, size and reg. The layout is the following:
// - bits [63, 60] (4 bits) - the operand type (ND_OP_REG)
// - bits [59, 52] (8 bits) - the register type
// - bits [51, 36] (16 bits) - the register size, in bytes
// - bits [35, 30] (6 bits) - the number of registers accessed starting with this reg (for block addressing)
// - bits [29, 9] (21 bits) - reserved
// - bit 8 - High8 indicator: indicates whether the reg is AH/CH/DH/BH
// - bits [7, 0] (8 bits) - the register ID
#define ND_OP_REG_ID(op) (((uint64_t)((op)->Type & 0xF) << 60) | \
((uint64_t)((op)->Info.Register.Type & 0xFF) << 52) | \
((uint64_t)((op)->Info.Register.Size & 0xFFFF) << 36) | \
((uint64_t)((op)->Info.Register.Count & 0x3F) << 30) | \
((uint64_t)((op)->Info.Register.IsHigh8 & 0x1) << 8) | \
((uint64_t)((op)->Info.Register.Reg)))
// Example: ND_IS_OP_REG(op, ND_REG_GPR, 4, REG_ESP)
// Example: ND_IS_OP_REG(op, ND_REG_CR, 8, REG_CR3)
// Example: ND_IS_OP_REG(op, ND_REG_RIP, 8, 0)
// Checks if the indicated operand op is a register of type t, with size s and index r.
#define ND_IS_OP_REG(op, t, s, r) ((uint32_t)(ND_OP_REG_ID(op)) == (uint32_t)((ND_OP_REG << 24)|(t << 16)|(s << 8)|(r)))
#define ND_IS_OP_REG(op, t, s, r) (ND_OP_REG_ID(op) == (((uint64_t)(ND_OP_REG) << 60) | \
((uint64_t)((t) & 0xFF) << 52) | \
((uint64_t)((s) & 0xFFFF) << 36) | \
((uint64_t)(1) << 30) | \
((uint64_t)(r))))
// Checks if the indicated operand op is a register of type t, with size s and index r.
#define ND_IS_OP_REG_EX(op, t, s, r, b, h) (ND_OP_REG_ID(op) == (((uint64_t)(ND_OP_REG) << 60) | \
((uint64_t)((t) & 0xFF) << 52) | \
((uint64_t)((s) & 0xFFFF) << 36) | \
((uint64_t)((b) & 0x3F) << 30) | \
((uint64_t)((h) & 0x1) << 8) | \
((uint64_t)(r))))
// Checjs if the indicated operand is the stack.
#define ND_IS_OP_STACK(op) ((op)->Type == ND_OP_MEM && (op)->Info.Memory.IsStack)
@ -1619,7 +1642,7 @@ NdGetFullAccessMap(
//
NDSTATUS
NdGetOperandRlut(
INSTRUX *Instrux,
const INSTRUX *Instrux,
ND_OPERAND_RLUT *Rlut
);

View File

@ -5,7 +5,7 @@
#ifndef DISASM_TYPES_H
#define DISASM_TYPES_H
#if defined(KERNEL_MODE) && defined(_MSC_VER)
#if defined(_KERNEL_MODE) && defined(_MSC_VER)
# include <ntddk.h>
# include <Ntstrsafe.h>
# include <basetsd.h>

View File

@ -7,6 +7,6 @@
#define DISASM_VERSION_MAJOR 1
#define DISASM_VERSION_MINOR 34
#define DISASM_VERSION_REVISION 7
#define DISASM_VERSION_REVISION 8
#endif // DISASM_VER_H