summaryrefslogtreecommitdiff
path: root/src/format/dex
diff options
context:
space:
mode:
Diffstat (limited to 'src/format/dex')
-rw-r--r--src/format/dex/pool.c19
-rw-r--r--src/format/dex/pool.h5
2 files changed, 19 insertions, 5 deletions
diff --git a/src/format/dex/pool.c b/src/format/dex/pool.c
index dbcee52..b2ca105 100644
--- a/src/format/dex/pool.c
+++ b/src/format/dex/pool.c
@@ -48,6 +48,9 @@ const char *get_string_from_dex_pool(const GDexFormat *format, uint32_t index)
string_id_item str_id; /* Identifiant de chaîne */
string_data_item str_data; /* Description de chaîne */
+ if (index >= format->header.string_ids_size)
+ return NULL;
+
pos = format->header.string_ids_off + index * sizeof(string_id_item);
if (!read_dex_string_id_item(format, &pos, &str_id))
@@ -92,6 +95,9 @@ GOpenidaType *get_type_from_dex_pool(const GDexFormat *format, uint16_t index)
//printf("Index :: 0x%04hx\n", index);
+ if (index >= format->header.type_ids_size)
+ return NULL;
+
pos = format->header.type_ids_off + index * sizeof(type_id_item);
if (!read_dex_type_id_item(format, &pos, &type_id))
@@ -135,12 +141,15 @@ GOpenidaType *get_type_from_dex_pool(const GDexFormat *format, uint16_t index)
* *
******************************************************************************/
-GOpenidaType *get_class_from_dex_pool(const GDexFormat *format, uint16_t index)
+GOpenidaType *get_class_from_dex_pool(const GDexFormat *format, uint32_t index)
{
GOpenidaType *result; /* Instance à retourner */
off_t pos; /* Tête de lecture */
class_def_item class_def; /* Définition de la classe */
+ if (index >= format->header.class_defs_size)
+ return NULL;
+
pos = format->header.class_defs_off + index * sizeof(class_def_item);
if (!read_dex_class_def_item(format, &pos, &class_def))
@@ -180,6 +189,9 @@ GBinVariable *get_field_from_dex_pool(const GDexFormat *format, uint32_t index)
const char *name; /* Désignation humaine */
GOpenidaType *owner; /* Propriétaire du champ */
+ if (index >= format->header.field_ids_size)
+ return NULL;
+
pos = format->header.field_ids_off + index * sizeof(field_id_item);
if (!read_dex_field_id_item(format, &pos, &field_id))
@@ -231,7 +243,7 @@ GBinVariable *get_field_from_dex_pool(const GDexFormat *format, uint32_t index)
* *
******************************************************************************/
-GBinRoutine *get_routine_from_dex_pool(const GDexFormat *format, uleb128_t index)
+GBinRoutine *get_routine_from_dex_pool(const GDexFormat *format, uint32_t index)
{
GBinRoutine *result; /* Instance à retourner */
off_t pos; /* Tête de lecture */
@@ -243,6 +255,9 @@ GBinRoutine *get_routine_from_dex_pool(const GDexFormat *format, uleb128_t index
string_id_item str_id; /* Identifiant de chaîne */
string_data_item str_data; /* Description de chaîne */
+ if (index >= format->header.method_ids_size)
+ return NULL;
+
pos = format->header.method_ids_off + index * sizeof(method_id_item);
if (!read_dex_method_id_item(format, &pos, &meth_id))
diff --git a/src/format/dex/pool.h b/src/format/dex/pool.h
index 263a807..56fd99c 100644
--- a/src/format/dex/pool.h
+++ b/src/format/dex/pool.h
@@ -27,7 +27,6 @@
#include "dex.h"
#include "../../analysis/routine.h"
-#include "../../common/leb128.h"
@@ -39,7 +38,7 @@ GOpenidaType *get_type_from_dex_pool(const GDexFormat *, uint16_t);
/* Extrait une représentation de classe d'une table DEX. */
-GOpenidaType *get_class_from_dex_pool(const GDexFormat *, uint16_t);
+GOpenidaType *get_class_from_dex_pool(const GDexFormat *, uint32_t);
@@ -49,7 +48,7 @@ GOpenidaType *get_class_from_dex_pool(const GDexFormat *, uint16_t);
GBinVariable *get_field_from_dex_pool(const GDexFormat *, uint32_t);
/* Extrait une représentation de routine d'une table DEX. */
-GBinRoutine *get_routine_from_dex_pool(const GDexFormat *, uleb128_t);
+GBinRoutine *get_routine_from_dex_pool(const GDexFormat *, uint32_t);