Formatter.h

Go to the documentation of this file.

/***************************************************************************************************

  Zyan Disassembler Library (Zydis)

  Original Author : Florian Bernd

 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.

***************************************************************************************************/

#ifndef ZYDIS_FORMATTER_H
#define ZYDIS_FORMATTER_H

#include <Zycore/Defines.h>
#include <Zycore/String.h>
#include <Zycore/Types.h>
#include <Zydis/DecoderTypes.h>
#include <Zydis/FormatterBuffer.h>

#ifdef __cplusplus
extern "C" {
#endif

/* ============================================================================================== */
/* Constants                                                                                      */
/* ============================================================================================== */

#define ZYDIS_RUNTIME_ADDRESS_NONE (ZyanU64)(-1)

/* ============================================================================================== */
/* Enums and types                                                                                */
/* ============================================================================================== */

/* ---------------------------------------------------------------------------------------------- */
/* Formatter style                                                                                */
/* ---------------------------------------------------------------------------------------------- */

typedef enum ZydisFormatterStyle_
{
    ZYDIS_FORMATTER_STYLE_ATT,
    ZYDIS_FORMATTER_STYLE_INTEL,
    ZYDIS_FORMATTER_STYLE_INTEL_MASM,

    ZYDIS_FORMATTER_STYLE_MAX_VALUE = ZYDIS_FORMATTER_STYLE_INTEL_MASM,
    ZYDIS_FORMATTER_STYLE_REQUIRED_BITS = ZYAN_BITS_TO_REPRESENT(ZYDIS_FORMATTER_STYLE_MAX_VALUE)
} ZydisFormatterStyle;

/* ---------------------------------------------------------------------------------------------- */
/* Properties                                                                                     */
/* ---------------------------------------------------------------------------------------------- */

typedef enum ZydisFormatterProperty_
{
    /* ---------------------------------------------------------------------------------------- */
    /* General                                                                                  */
    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_PROP_FORCE_SIZE,
    ZYDIS_FORMATTER_PROP_FORCE_SEGMENT,
     ZYDIS_FORMATTER_PROP_FORCE_SCALE_ONE,
    ZYDIS_FORMATTER_PROP_FORCE_RELATIVE_BRANCHES,
    ZYDIS_FORMATTER_PROP_FORCE_RELATIVE_RIPREL,
    ZYDIS_FORMATTER_PROP_PRINT_BRANCH_SIZE,

    ZYDIS_FORMATTER_PROP_DETAILED_PREFIXES,

    /* ---------------------------------------------------------------------------------------- */
    /* Numeric values                                                                           */
    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_PROP_ADDR_BASE,
    ZYDIS_FORMATTER_PROP_ADDR_SIGNEDNESS,
    ZYDIS_FORMATTER_PROP_ADDR_PADDING_ABSOLUTE,
    ZYDIS_FORMATTER_PROP_ADDR_PADDING_RELATIVE,

    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_PROP_DISP_BASE,
    ZYDIS_FORMATTER_PROP_DISP_SIGNEDNESS,
    ZYDIS_FORMATTER_PROP_DISP_PADDING,

    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_PROP_IMM_BASE,
    ZYDIS_FORMATTER_PROP_IMM_SIGNEDNESS,
    ZYDIS_FORMATTER_PROP_IMM_PADDING,

    /* ---------------------------------------------------------------------------------------- */
    /* Text formatting                                                                          */
    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_PROP_UPPERCASE_PREFIXES,
    ZYDIS_FORMATTER_PROP_UPPERCASE_MNEMONIC,
    ZYDIS_FORMATTER_PROP_UPPERCASE_REGISTERS,
    ZYDIS_FORMATTER_PROP_UPPERCASE_TYPECASTS,
    ZYDIS_FORMATTER_PROP_UPPERCASE_DECORATORS,

    /* ---------------------------------------------------------------------------------------- */
    /* Number formatting                                                                        */
    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_PROP_DEC_PREFIX,
    ZYDIS_FORMATTER_PROP_DEC_SUFFIX,

    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_PROP_HEX_UPPERCASE,
    ZYDIS_FORMATTER_PROP_HEX_FORCE_LEADING_NUMBER,
    ZYDIS_FORMATTER_PROP_HEX_PREFIX,
    ZYDIS_FORMATTER_PROP_HEX_SUFFIX,

    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_PROP_MAX_VALUE = ZYDIS_FORMATTER_PROP_HEX_SUFFIX,
    ZYDIS_FORMATTER_PROP_REQUIRED_BITS = ZYAN_BITS_TO_REPRESENT(ZYDIS_FORMATTER_PROP_MAX_VALUE)
} ZydisFormatterProperty;

/* ---------------------------------------------------------------------------------------------- */

typedef enum ZydisNumericBase_
{
    ZYDIS_NUMERIC_BASE_DEC,
    ZYDIS_NUMERIC_BASE_HEX,

    ZYDIS_NUMERIC_BASE_MAX_VALUE = ZYDIS_NUMERIC_BASE_HEX,
    ZYDIS_NUMERIC_BASE_REQUIRED_BITS = ZYAN_BITS_TO_REPRESENT(ZYDIS_NUMERIC_BASE_MAX_VALUE)
} ZydisNumericBase;

/* ---------------------------------------------------------------------------------------------- */

typedef enum ZydisSignedness_
{
    ZYDIS_SIGNEDNESS_AUTO,
    ZYDIS_SIGNEDNESS_SIGNED,
    ZYDIS_SIGNEDNESS_UNSIGNED,

    ZYDIS_SIGNEDNESS_MAX_VALUE = ZYDIS_SIGNEDNESS_UNSIGNED,
    ZYDIS_SIGNEDNESS_REQUIRED_BITS = ZYAN_BITS_TO_REPRESENT(ZYDIS_SIGNEDNESS_MAX_VALUE)
} ZydisSignedness;

/* ---------------------------------------------------------------------------------------------- */

typedef enum ZydisPadding_
{
    ZYDIS_PADDING_DISABLED = 0,
    ZYDIS_PADDING_AUTO     = (-1),

    ZYDIS_PADDING_MAX_VALUE = ZYDIS_PADDING_AUTO,
    ZYDIS_PADDING_REQUIRED_BITS = ZYAN_BITS_TO_REPRESENT(ZYDIS_PADDING_MAX_VALUE)
} ZydisPadding;

/* ---------------------------------------------------------------------------------------------- */
/* Function types                                                                                 */
/* ---------------------------------------------------------------------------------------------- */

typedef enum ZydisFormatterFunction_
{
    /* ---------------------------------------------------------------------------------------- */
    /* Instruction                                                                              */
    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_FUNC_PRE_INSTRUCTION,
    ZYDIS_FORMATTER_FUNC_POST_INSTRUCTION,

    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_FUNC_FORMAT_INSTRUCTION,

    /* ---------------------------------------------------------------------------------------- */
    /* Operands                                                                                 */
    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_FUNC_PRE_OPERAND,
    ZYDIS_FORMATTER_FUNC_POST_OPERAND,

    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_FUNC_FORMAT_OPERAND_REG,
    ZYDIS_FORMATTER_FUNC_FORMAT_OPERAND_MEM,
    ZYDIS_FORMATTER_FUNC_FORMAT_OPERAND_PTR,
    ZYDIS_FORMATTER_FUNC_FORMAT_OPERAND_IMM,

    /* ---------------------------------------------------------------------------------------- */
    /* Elemental tokens                                                                         */
    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_FUNC_PRINT_MNEMONIC,

    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_FUNC_PRINT_REGISTER,
    ZYDIS_FORMATTER_FUNC_PRINT_ADDRESS_ABS,
    ZYDIS_FORMATTER_FUNC_PRINT_ADDRESS_REL,
    ZYDIS_FORMATTER_FUNC_PRINT_DISP,
    ZYDIS_FORMATTER_FUNC_PRINT_IMM,

    /* ---------------------------------------------------------------------------------------- */
    /* Optional tokens                                                                          */
    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_FUNC_PRINT_TYPECAST,
    ZYDIS_FORMATTER_FUNC_PRINT_SEGMENT,
    ZYDIS_FORMATTER_FUNC_PRINT_PREFIXES,
    ZYDIS_FORMATTER_FUNC_PRINT_DECORATOR,

    /* ---------------------------------------------------------------------------------------- */

    ZYDIS_FORMATTER_FUNC_MAX_VALUE = ZYDIS_FORMATTER_FUNC_PRINT_DECORATOR,
    ZYDIS_FORMATTER_FUNC_REQUIRED_BITS = ZYAN_BITS_TO_REPRESENT(ZYDIS_FORMATTER_FUNC_MAX_VALUE)
} ZydisFormatterFunction;

/* ---------------------------------------------------------------------------------------------- */
/* Decorator types                                                                                */
/* ---------------------------------------------------------------------------------------------- */

typedef enum ZydisDecorator_
{
    ZYDIS_DECORATOR_INVALID,
    ZYDIS_DECORATOR_MASK,
    ZYDIS_DECORATOR_BC,
    ZYDIS_DECORATOR_RC,
    ZYDIS_DECORATOR_SAE,
    ZYDIS_DECORATOR_SWIZZLE,
    ZYDIS_DECORATOR_CONVERSION,
    ZYDIS_DECORATOR_EH,

    ZYDIS_DECORATOR_MAX_VALUE = ZYDIS_DECORATOR_EH,
    ZYDIS_DECORATOR_REQUIRED_BITS = ZYAN_BITS_TO_REPRESENT(ZYDIS_DECORATOR_MAX_VALUE)
} ZydisDecorator;

/* ---------------------------------------------------------------------------------------------- */
/* Formatter context                                                                              */
/* ---------------------------------------------------------------------------------------------- */

typedef struct ZydisFormatter_ ZydisFormatter;

typedef struct ZydisFormatterContext_
{
    const ZydisDecodedInstruction* instruction;
    const ZydisDecodedOperand* operands;
    const ZydisDecodedOperand* operand;
    ZyanU64 runtime_address;
    void* user_data;
} ZydisFormatterContext;

/* ---------------------------------------------------------------------------------------------- */
/* Function prototypes                                                                            */
/* ---------------------------------------------------------------------------------------------- */

typedef ZyanStatus (*ZydisFormatterFunc)(const ZydisFormatter* formatter,
    ZydisFormatterBuffer* buffer, ZydisFormatterContext* context);

typedef ZyanStatus (*ZydisFormatterRegisterFunc)(const ZydisFormatter* formatter,
    ZydisFormatterBuffer* buffer, ZydisFormatterContext* context, ZydisRegister reg);

typedef ZyanStatus (*ZydisFormatterDecoratorFunc)(const ZydisFormatter* formatter,
    ZydisFormatterBuffer* buffer, ZydisFormatterContext* context, ZydisDecorator decorator);

/* ---------------------------------------------------------------------------------------------- */
/* Formatter struct                                                                               */
/* ---------------------------------------------------------------------------------------------- */

struct ZydisFormatter_
{
    ZydisFormatterStyle style;
    ZyanBool force_memory_size;
    ZyanBool force_memory_segment;
    ZyanBool force_memory_scale;
    ZyanBool force_relative_branches;
    ZyanBool force_relative_riprel;
    ZyanBool print_branch_size;
    ZyanBool detailed_prefixes;
    ZydisNumericBase addr_base;
    ZydisSignedness addr_signedness;
    ZydisPadding addr_padding_absolute;
    ZydisPadding addr_padding_relative;
    ZydisNumericBase disp_base;
    ZydisSignedness disp_signedness;
    ZydisPadding disp_padding;
    ZydisNumericBase imm_base;
    ZydisSignedness imm_signedness;
    ZydisPadding imm_padding;
    ZyanI32 case_prefixes;
    ZyanI32 case_mnemonic;
    ZyanI32 case_registers;
    ZyanI32 case_typecasts;
    ZyanI32 case_decorators;
    ZyanBool hex_uppercase;
    ZyanBool hex_force_leading_number;
    struct
    {
        const ZyanStringView* string;
        ZyanStringView string_data;
        char buffer[11];
    } number_format[ZYDIS_NUMERIC_BASE_MAX_VALUE + 1][2];
    ZydisFormatterFunc func_pre_instruction;
    ZydisFormatterFunc func_post_instruction;
    ZydisFormatterFunc func_format_instruction;
    ZydisFormatterFunc func_pre_operand;
    ZydisFormatterFunc func_post_operand;
    ZydisFormatterFunc func_format_operand_reg;
    ZydisFormatterFunc func_format_operand_mem;
    ZydisFormatterFunc func_format_operand_ptr;
    ZydisFormatterFunc func_format_operand_imm;
    ZydisFormatterFunc func_print_mnemonic;
    ZydisFormatterRegisterFunc func_print_register;
    ZydisFormatterFunc func_print_address_abs;
    ZydisFormatterFunc func_print_address_rel;
    ZydisFormatterFunc func_print_disp;
    ZydisFormatterFunc func_print_imm;
    ZydisFormatterFunc func_print_typecast;
    ZydisFormatterFunc func_print_segment;
    ZydisFormatterFunc func_print_prefixes;
    ZydisFormatterDecoratorFunc func_print_decorator;
};

/* ---------------------------------------------------------------------------------------------- */

/* ============================================================================================== */
/* Exported functions                                                                             */
/* ============================================================================================== */

/* ---------------------------------------------------------------------------------------------- */
/* Initialization                                                                                 */
/* ---------------------------------------------------------------------------------------------- */

ZYDIS_EXPORT ZyanStatus ZydisFormatterInit(ZydisFormatter* formatter, ZydisFormatterStyle style);

/* ---------------------------------------------------------------------------------------------- */
/* Setter                                                                                         */
/* ---------------------------------------------------------------------------------------------- */

ZYDIS_EXPORT ZyanStatus ZydisFormatterSetProperty(ZydisFormatter* formatter,
    ZydisFormatterProperty property, ZyanUPointer value);

ZYDIS_EXPORT ZyanStatus ZydisFormatterSetHook(ZydisFormatter* formatter,
    ZydisFormatterFunction type, const void** callback);

/* ---------------------------------------------------------------------------------------------- */
/* Formatting                                                                                     */
/* ---------------------------------------------------------------------------------------------- */

ZYDIS_EXPORT ZyanStatus ZydisFormatterFormatInstruction(const ZydisFormatter* formatter,
    const ZydisDecodedInstruction* instruction, const ZydisDecodedOperand* operands,
    ZyanU8 operand_count, char* buffer, ZyanUSize length, ZyanU64 runtime_address,
    void* user_data);

ZYDIS_EXPORT ZyanStatus ZydisFormatterFormatOperand(const ZydisFormatter* formatter,
    const ZydisDecodedInstruction* instruction, const ZydisDecodedOperand* operand,
    char* buffer, ZyanUSize length, ZyanU64 runtime_address, void* user_data);

/* ---------------------------------------------------------------------------------------------- */
/* Tokenizing                                                                                     */
/* ---------------------------------------------------------------------------------------------- */

ZYDIS_EXPORT ZyanStatus ZydisFormatterTokenizeInstruction(const ZydisFormatter* formatter,
    const ZydisDecodedInstruction* instruction, const ZydisDecodedOperand* operands,
    ZyanU8 operand_count, void* buffer, ZyanUSize length, ZyanU64 runtime_address,
    ZydisFormatterTokenConst** token, void* user_data);

ZYDIS_EXPORT ZyanStatus ZydisFormatterTokenizeOperand(const ZydisFormatter* formatter,
    const ZydisDecodedInstruction* instruction, const ZydisDecodedOperand* operand,
    void* buffer, ZyanUSize length, ZyanU64 runtime_address, ZydisFormatterTokenConst** token,
    void* user_data);

/* ---------------------------------------------------------------------------------------------- */

/* ============================================================================================== */

#ifdef __cplusplus
}
#endif

#endif /* ZYDIS_FORMATTER_H */