#ifndef DEX_H
#define DEX_H

#include <r_types.h>
#include <r_util.h>
#include <r_lib.h>
#include <r_bin.h>
#include <r_vec.h>

#define R_BIN_DEX_MAXSTR 256
#define DEX_CLASS_SIZE (32)
#define LEB_MAX_SIZE 6

/* method flags (ACC_ things) */
#define R_DEX_METH_PUBLIC 0x0001
#define R_DEX_METH_PRIVATE 0x0002
#define R_DEX_METH_PROTECTED 0x0004
#define R_DEX_METH_STATIC 0x0008
#define R_DEX_METH_FINAL 0x0010
#define R_DEX_METH_SYNCHRONIZED 0x0020
#define R_DEX_METH_VOLATILE 0x0040
#define R_DEX_METH_BRIDGE 0x0040
#define R_DEX_METH_TRANSIENT 0x0080
#define R_DEX_METH_VARARGS 0x0080
#define R_DEX_METH_NATIVE 0x0100
#define R_DEX_METH_INTERFACE 0x0200
#define R_DEX_METH_ABSTRACT 0x0400
#define R_DEX_METH_STRICT 0x0800
#define R_DEX_METH_SYNTHETIC 0x1000
#define R_DEX_METH_ANNOTATION 0x2000
#define R_DEX_METH_ENUM 0x4000
#define R_DEX_METH_MIRANDA 0x8000
#define R_DEX_METH_CONSTRUCTOR 0x10000
#define R_DEX_METH_DECLARED_SYNCHRONIZED 0x20000

// encoded value types
#define R_DEX_ENCVAL_BYTE 0x00
#define R_DEX_ENCVAL_SHORT 0x02
#define R_DEX_ENCVAL_CHAR 0x03
#define R_DEX_ENCVAL_INT 0x04
#define R_DEX_ENCVAL_LONG 0x06
#define R_DEX_ENCVAL_FLOAT 0x10
#define R_DEX_ENCVAL_DOUBLE 0x11
#define R_DEX_ENCVAL_STRING 0x17
#define R_DEX_ENCVAL_TYPE 0x18
#define R_DEX_ENCVAL_FIELD 0x19
#define R_DEX_ENCVAL_ENUM 0x1b
#define R_DEX_ENCVAL_METHOD 0x1a
#define R_DEX_ENCVAL_ARRAY 0x1c
#define R_DEX_ENCVAL_ANNOTATION 0x1d
#define R_DEX_ENCVAL_NULL 0x1e
#define R_DEX_ENCVAL_BOOLEAN 0x1f

// visibilities
#define R_DEX_VISIBILITY_BUILD 0
#define R_DEX_VISIBILITY_RUNTIME 1
#define R_DEX_VISIBILITY_SYSTEM 2

R_PACKED(
typedef struct dex_header_t {
	ut8 magic[8];
	ut32 checksum;
	ut8 signature[20];
	ut32 size;
	ut32 header_size;
	ut32 endian;
	ut32 linksection_size;
	ut32 linksection_offset;
	ut32 map_offset;
	ut32 strings_size;
	ut32 strings_offset;
	ut32 types_size;
	ut32 types_offset;
	ut32 prototypes_size;
	ut32 prototypes_offset;
	ut32 fields_size;
	ut32 fields_offset;
	ut32 method_size;
	ut32 method_offset;
	ut32 class_size;
	ut32 class_offset;
	ut32 data_size;
	ut32 data_offset;
}) DexHeader;

R_PACKED(
typedef struct dex_proto_t {
	ut32 shorty_id;
	ut32 return_type_id;
	ut32 parameters_off;
}) DexProto;

typedef struct dex_type_t {
	ut32 descriptor_id;
} DexType;

// #pragma pack(1)
typedef struct dex_field_t {
	ut16 class_id;
	ut16 type_id;
	ut32 name_id;
} DexField;

R_PACKED(
typedef struct dex_method_t {
	ut16 class_id;
	ut16 proto_id;
	ut32 name_id;
}) RBinDexMethod;

R_VEC_TYPE (RVecDexMethod, RBinDexMethod);

R_PACKED(
typedef struct dex_class_t {
	ut32 class_id; // index into typeids
	ut32 access_flags;
	ut32 super_class;
	ut32 interfaces_offset;
	ut32 source_file;
	ut32 anotations_offset;
	ut32 class_data_offset;
	ut32 static_values_offset;
	struct dex_class_data_item_t *class_data;
}) RBinDexClass;

R_VEC_TYPE_WITH_FINI (RVecRBinClass, RBinClass, r_bin_class_fini);

R_PACKED(
typedef struct dex_class_data_item_t {
	ut64 static_fields_size;
	ut64 instance_fields_size;
	ut64 direct_methods_size;
	ut64 virtual_methods_size;
}) RBinDexClassData;

typedef struct r_bin_dex_obj_t {
	ut32 size;
	const char *file;
	RBuffer *b;
	struct dex_header_t header;
	ut32 *strings;
	struct dex_type_t *types;
	struct dex_proto_t *protos;
	struct dex_field_t *fields;
	RVecDexMethod dex_methods;
	struct dex_class_t *classes;
	RVecRBinSymbol symbols_vec;
	RVecRBinImport imports_vec;
	RVecRBinClass classes_vec;
	RVecRBinTrycatch trycatch;
	bool trycatch_loaded;
	ut64 trycatch_budget; // try items, catch handlers and type name bytes dex_loadcode can still expand
	RList *lines_list;
	ut64 code_from;
	ut64 code_to;
	char *version;
	Sdb *kv;
	char **cal_strings;
	ut32 cal_strings_size;
	char **cal_protos;
	ut32 cal_protos_size;
	bool verbose;
	RStrBuf *sb; // used for dexdump output
	bool addrline_loaded;
	const char *dexSubsystem;
	bool simplifiedDemangling;
	Sdb *mdb;
} RBinDexObj;

typedef struct r_bin_dex_str_t {
	char str[R_BIN_DEX_MAXSTR];
	ut64 offset;
	ut64 ordinal;
	int size;
	int last;
} RBinDexStr;

struct dex_encoded_type_addr_pair_t {
	ut64 type_idx;
	ut64 addr;
};

struct dex_encoded_catch_handler_t {
	st64 size;
	struct dex_encoded_type_addr_pair_t *handlers;
	ut64 catch_all_addr;
};

struct dex_debug_position_t {
	ut32 source_file_idx;
	ut64 address;
	ut64 line;
};

struct dex_debug_local_t {
	const char *name;
	const char *descriptor;
	const char *signature;
	ut16 startAddress;
	bool live;
	int reg;
	ut16 endAddress;
};

R_IPI char* r_bin_dex_get_version(RBinDexObj * bin);
R_IPI void r_bin_dex_free(RBinDexObj *bin);
R_IPI RBinDexObj *r_bin_dex_new_buf(RBuffer *buf, bool verbose);
R_IPI RBinDexStr *r_bin_dex_get_strings(RBinDexObj *bin);

#endif