400 lines
17 KiB
C
400 lines
17 KiB
C
#include <stdint.h>
|
|
|
|
#include "assert.h"
|
|
#include "class_file.h"
|
|
#include "hash_table.h"
|
|
#include "malloc.h"
|
|
#include "class_resolver.h"
|
|
#include "string.h"
|
|
#include "debug_class_file.h"
|
|
#include "memory_allocator.h"
|
|
#include "printf.h"
|
|
|
|
static void class_resolver_create_interfaces_hash_table(struct class_entry * class_entry)
|
|
{
|
|
/*
|
|
struct class_file * class_file = class_entry->class_file;
|
|
|
|
uint32_t interfaces_hash_table_size = (sizeof (struct hash_table_entry)) * class_file->interfaces_count * 3 / 2;
|
|
struct hash_table_entry * interfaces_hash_table = malloc_class_arena(interfaces_hash_table_size);
|
|
for (int i = 0; i < class_file->interfaces_count; i++) {
|
|
|
|
}
|
|
*/
|
|
}
|
|
|
|
static int field_size(struct class_file * class_file, struct field_info * field_info)
|
|
{
|
|
struct constant * field_descriptor_constant = &class_file->constant_pool[field_info->descriptor_index - 1];
|
|
#ifdef DEBUG
|
|
assert(field_descriptor_constant->tag == CONSTANT_Utf8);
|
|
#endif
|
|
|
|
switch (field_descriptor_constant->utf8.bytes[0]) {
|
|
case 'B': [[fallthrough]];
|
|
case 'C': [[fallthrough]];
|
|
case 'F': [[fallthrough]];
|
|
case 'I': [[fallthrough]];
|
|
case 'L': [[fallthrough]];
|
|
case 'S': [[fallthrough]];
|
|
case 'Z': [[fallthrough]];
|
|
case '[':
|
|
return 1;
|
|
case 'D': [[fallthrough]];
|
|
case 'J':
|
|
return 2;
|
|
default:
|
|
assert(false);
|
|
}
|
|
}
|
|
|
|
static int32_t class_resolver_create_fields_hash_table(struct class_entry * class_entry)
|
|
{
|
|
struct class_file * class_file = class_entry->class_file;
|
|
int fields_hash_table_length = hash_table_next_power_of_two(class_file->fields_count * 2);
|
|
uint32_t fields_hash_table_size = (sizeof (struct hash_table_entry)) * fields_hash_table_length;
|
|
struct hash_table_entry * fields_hash_table = malloc_class_arena(fields_hash_table_size);
|
|
uint32_t field_entry_size = (sizeof (struct field_entry)) * class_file->fields_count;
|
|
struct field_entry * field_entry = malloc_class_arena(field_entry_size);
|
|
|
|
int32_t static_index = 0;
|
|
int32_t instance_index = 0;
|
|
|
|
for (int i = 0; i < class_file->fields_count; i++) {
|
|
u2 name_index = class_file->fields[i].name_index;
|
|
struct constant * name_constant = &class_file->constant_pool[name_index - 1];
|
|
assert(name_constant->tag == CONSTANT_Utf8);
|
|
debugf("hash table entry for field: ");
|
|
print_utf8_string(name_constant);
|
|
debugf("\n");
|
|
|
|
struct field_info * field_info = &class_file->fields[i];
|
|
|
|
if (field_info->access_flags & FIELD_ACC_STATIC) {
|
|
field_entry[i].static_index = static_index;
|
|
static_index += field_size(class_file, field_info);
|
|
} else {
|
|
field_entry[i].instance_index = instance_index;
|
|
instance_index += field_size(class_file, field_info);
|
|
}
|
|
field_entry[i].field_info = field_info;
|
|
|
|
hash_table_add(fields_hash_table_length,
|
|
fields_hash_table,
|
|
name_constant->utf8.bytes,
|
|
name_constant->utf8.length,
|
|
(void *)&field_entry[i]);
|
|
}
|
|
|
|
class_entry->fields.length = fields_hash_table_length;
|
|
class_entry->fields.entry = fields_hash_table;
|
|
|
|
return static_index;
|
|
}
|
|
|
|
static void class_resolver_create_methods_hash_table(struct class_entry * class_entry)
|
|
{
|
|
struct class_file * class_file = class_entry->class_file;
|
|
int methods_hash_table_length = hash_table_next_power_of_two(class_file->methods_count * 2);
|
|
uint32_t methods_hash_table_size = (sizeof (struct hash_table_entry)) * methods_hash_table_length;
|
|
struct hash_table_entry * methods_hash_table = malloc_class_arena(methods_hash_table_size);
|
|
for (int i = 0; i < class_file->methods_count; i++) {
|
|
u2 name_index = class_file->methods[i].name_index;
|
|
struct constant * name_constant = &class_file->constant_pool[name_index - 1];
|
|
assert(name_constant->tag == CONSTANT_Utf8);
|
|
u2 descriptor_index = class_file->methods[i].descriptor_index;
|
|
struct constant * descriptor_constant = &class_file->constant_pool[descriptor_index - 1];
|
|
assert(descriptor_constant->tag == CONSTANT_Utf8);
|
|
debugf("hash table entry for method: ");
|
|
print_utf8_string(name_constant);
|
|
debugf("\n");
|
|
|
|
hash_table_add2(methods_hash_table_length,
|
|
methods_hash_table,
|
|
name_constant->utf8.bytes,
|
|
name_constant->utf8.length,
|
|
descriptor_constant->utf8.bytes,
|
|
descriptor_constant->utf8.length,
|
|
(void *)&class_file->methods[i]);
|
|
}
|
|
|
|
class_entry->methods.length = methods_hash_table_length;
|
|
class_entry->methods.entry = methods_hash_table;
|
|
}
|
|
|
|
static void class_resolver_allocate_static_fields(struct class_entry * class_entry, int32_t static_field_count)
|
|
{
|
|
uint32_t static_fields_size = (sizeof (int32_t)) * static_field_count;
|
|
int32_t * static_fields = malloc_class_arena(static_fields_size);
|
|
|
|
for (int i = 0; i < static_field_count; i++) {
|
|
static_fields[i] = 0;
|
|
}
|
|
|
|
class_entry->static_fields = static_fields;
|
|
}
|
|
|
|
static void class_resolver_allocate_attribute_entry(struct class_entry * class_entry)
|
|
{
|
|
struct class_file * class_file = class_entry->class_file;
|
|
uint32_t attribute_entry_size = (sizeof (union attribute_entry)) * class_file->constant_pool_count;
|
|
union attribute_entry * attribute_entry = malloc_class_arena(attribute_entry_size);
|
|
|
|
for (int i = 0; i < class_file->constant_pool_count; i++) {
|
|
attribute_entry[i].class_entry = nullptr;
|
|
}
|
|
|
|
class_entry->attribute_entry = attribute_entry;
|
|
}
|
|
|
|
struct hash_table_entry * class_resolver_load_from_buffers(const uint8_t * class_names[],
|
|
const int class_names_length[],
|
|
uint8_t * buffers[],
|
|
int length,
|
|
int * hash_table_length)
|
|
{
|
|
int class_hash_table_length = hash_table_next_power_of_two(hash_table_next_power_of_two(length * 2));
|
|
uint32_t class_hash_table_size = (sizeof (struct hash_table_entry)) * class_hash_table_length;
|
|
struct hash_table_entry * class_hash_table = malloc_class_arena(class_hash_table_size);
|
|
uint32_t class_entry_size = (sizeof (struct class_entry)) * length;
|
|
struct class_entry * class_entry = malloc_class_arena(class_entry_size);
|
|
|
|
for (int i = 0; i < length; i++) {
|
|
struct class_file * class_file = class_file_parse(buffers[i]);
|
|
|
|
assert(class_file->magic == 0xcafebabe);
|
|
|
|
class_entry[i].class_file = class_file;
|
|
class_entry[i].initialization_state = CLASS_UNINITIALIZED;
|
|
|
|
hash_table_add(class_hash_table_length,
|
|
class_hash_table,
|
|
class_names[i],
|
|
class_names_length[i],
|
|
&class_entry[i]);
|
|
|
|
// make hash table for interfaces
|
|
class_resolver_create_interfaces_hash_table(&class_entry[i]);
|
|
|
|
// make hash table for fields
|
|
int32_t static_field_count = class_resolver_create_fields_hash_table(&class_entry[i]);
|
|
|
|
// make hash table for methods
|
|
class_resolver_create_methods_hash_table(&class_entry[i]);
|
|
|
|
// allocate static fields
|
|
class_resolver_allocate_static_fields(&class_entry[i], static_field_count);
|
|
|
|
// allocate attribute_entry
|
|
class_resolver_allocate_attribute_entry(&class_entry[i]);
|
|
};
|
|
|
|
*hash_table_length = class_hash_table_length;
|
|
|
|
return class_hash_table;
|
|
}
|
|
|
|
struct class_entry * class_resolver_lookup_class(int class_hash_table_length,
|
|
struct hash_table_entry * class_hash_table,
|
|
const uint8_t * class_name,
|
|
int class_name_length)
|
|
{
|
|
debugf("class_resolver_lookup_class: ");
|
|
for (int i = 0; i < class_name_length; i++) { debugc(class_name[i]); }
|
|
debugc('\n');
|
|
|
|
struct hash_table_entry * e = hash_table_find(class_hash_table_length,
|
|
class_hash_table,
|
|
class_name,
|
|
class_name_length);
|
|
assert(e != nullptr);
|
|
|
|
return (struct class_entry *)e->value;
|
|
}
|
|
|
|
struct class_entry * class_resolver_lookup_class_from_class_index(int class_hash_table_length,
|
|
struct hash_table_entry * class_hash_table,
|
|
struct class_entry * class_entry,
|
|
int32_t class_index)
|
|
{
|
|
if (class_entry->attribute_entry[class_index - 1].class_entry != nullptr) {
|
|
debugf("class_resolver_lookup_class_from_class_index %d: [cached]\n", class_index);
|
|
return class_entry->attribute_entry[class_index - 1].class_entry;
|
|
}
|
|
|
|
struct constant * class_constant = &class_entry->class_file->constant_pool[class_index - 1];
|
|
#ifdef DEBUG
|
|
assert(class_constant->tag == CONSTANT_Class);
|
|
#endif
|
|
struct constant * class_name_constant = &class_entry->class_file->constant_pool[class_constant->class.name_index - 1];
|
|
#ifdef DEBUG
|
|
assert(class_name_constant->tag == CONSTANT_Utf8);
|
|
#endif
|
|
|
|
struct class_entry * _class_entry = class_resolver_lookup_class(class_hash_table_length,
|
|
class_hash_table,
|
|
class_name_constant->utf8.bytes,
|
|
class_name_constant->utf8.length);
|
|
|
|
// cache the result
|
|
class_entry->attribute_entry[class_index - 1].class_entry = _class_entry;
|
|
|
|
return _class_entry;
|
|
}
|
|
|
|
struct field_entry * class_resolver_lookup_field(int fields_hash_table_length,
|
|
struct hash_table_entry * fields_hash_table,
|
|
const uint8_t * field_name,
|
|
int field_name_length)
|
|
{
|
|
debugf("class_resolver_lookup_field: ");
|
|
for (int i = 0; i < field_name_length; i++) { debugc(field_name[i]); }
|
|
debugc('\n');
|
|
|
|
struct hash_table_entry * e = hash_table_find(fields_hash_table_length,
|
|
fields_hash_table,
|
|
field_name,
|
|
field_name_length);
|
|
assert(e != nullptr);
|
|
|
|
return (struct field_entry *)e->value;
|
|
}
|
|
|
|
struct field_entry * class_resolver_lookup_field_from_fieldref_index(int fields_hash_table_length,
|
|
struct hash_table_entry * fields_hash_table,
|
|
struct class_entry * class_entry,
|
|
int fieldref_index)
|
|
{
|
|
if (class_entry->attribute_entry[fieldref_index - 1].field_entry != nullptr) {
|
|
debugf("class_resolver_lookup_method_from_fieldref_index %d: [cached]\n", fieldref_index);
|
|
return class_entry->attribute_entry[fieldref_index - 1].field_entry;
|
|
}
|
|
|
|
struct constant * fieldref_constant = &class_entry->class_file->constant_pool[fieldref_index - 1];
|
|
#ifdef DEBUG
|
|
assert(fieldref_constant->tag == CONSTANT_Fieldref);
|
|
#endif
|
|
struct constant * nameandtype_constant = &class_entry->class_file->constant_pool[fieldref_constant->fieldref.name_and_type_index - 1];
|
|
#ifdef DEBUG
|
|
assert(nameandtype_constant->tag == CONSTANT_NameAndType);
|
|
#endif
|
|
struct constant * field_name_constant = &class_entry->class_file->constant_pool[nameandtype_constant->nameandtype.name_index - 1];
|
|
#ifdef DEBUG
|
|
assert(field_name_constant->tag == CONSTANT_Utf8);
|
|
#endif
|
|
|
|
struct field_entry * field_entry = class_resolver_lookup_field(fields_hash_table_length,
|
|
fields_hash_table,
|
|
field_name_constant->utf8.bytes,
|
|
field_name_constant->utf8.length);
|
|
|
|
// cache the result
|
|
class_entry->attribute_entry[fieldref_index - 1].field_entry = field_entry;
|
|
|
|
return field_entry;
|
|
}
|
|
|
|
struct method_info * class_resolver_lookup_method(int methods_hash_table_length,
|
|
struct hash_table_entry * methods_hash_table,
|
|
const uint8_t * method_name,
|
|
int method_name_length,
|
|
const uint8_t * method_descriptor,
|
|
int method_descriptor_length)
|
|
{
|
|
debugf("class_resolver_lookup_method: ");
|
|
for (int i = 0; i < method_name_length; i++) { debugc(method_name[i]); }
|
|
debugc(' ');
|
|
for (int i = 0; i < method_descriptor_length; i++) { debugc(method_descriptor[i]); }
|
|
debugc('\n');
|
|
|
|
struct hash_table_entry * e = hash_table_find2(methods_hash_table_length,
|
|
methods_hash_table,
|
|
method_name,
|
|
method_name_length,
|
|
method_descriptor,
|
|
method_descriptor_length);
|
|
if (e == nullptr)
|
|
return nullptr;
|
|
|
|
return (struct method_info *)e->value;
|
|
}
|
|
|
|
struct method_info * class_resolver_lookup_method_from_methodref_index(int methods_hash_table_length,
|
|
struct hash_table_entry * methods_hash_table,
|
|
struct class_entry * class_entry,
|
|
int methodref_index)
|
|
{
|
|
if (class_entry->attribute_entry[methodref_index - 1].method_info != nullptr) {
|
|
debugf("class_resolver_lookup_method_from_methodref_index %d: [cached]\n", methodref_index);
|
|
return class_entry->attribute_entry[methodref_index - 1].method_info;
|
|
}
|
|
|
|
struct constant * methodref_constant = &class_entry->class_file->constant_pool[methodref_index - 1];
|
|
#ifdef DEBUG
|
|
assert(methodref_constant->tag == CONSTANT_Methodref);
|
|
#endif
|
|
struct constant * nameandtype_constant = &class_entry->class_file->constant_pool[methodref_constant->methodref.name_and_type_index - 1];
|
|
#ifdef DEBUG
|
|
assert(nameandtype_constant->tag == CONSTANT_NameAndType);
|
|
#endif
|
|
struct constant * method_name_constant = &class_entry->class_file->constant_pool[nameandtype_constant->nameandtype.name_index - 1];
|
|
#ifdef DEBUG
|
|
assert(method_name_constant->tag == CONSTANT_Utf8);
|
|
#endif
|
|
struct constant * method_descriptor_constant = &class_entry->class_file->constant_pool[nameandtype_constant->nameandtype.descriptor_index - 1];
|
|
#ifdef DEBUG
|
|
assert(method_descriptor_constant->tag == CONSTANT_Utf8);
|
|
#endif
|
|
|
|
struct method_info * method_info = class_resolver_lookup_method(methods_hash_table_length,
|
|
methods_hash_table,
|
|
method_name_constant->utf8.bytes,
|
|
method_name_constant->utf8.length,
|
|
method_descriptor_constant->utf8.bytes,
|
|
method_descriptor_constant->utf8.length);
|
|
|
|
// cache the result
|
|
class_entry->attribute_entry[methodref_index - 1].method_info = method_info;
|
|
|
|
return method_info;
|
|
}
|
|
|
|
int32_t * class_resolver_lookup_string(int class_hash_table_length,
|
|
struct hash_table_entry * class_hash_table,
|
|
struct class_entry * class_entry,
|
|
const int string_index)
|
|
{
|
|
debugf("class_resolver_lookup_string: %d\n", string_index);
|
|
if (class_entry->attribute_entry[string_index - 1].string_objectref != nullptr) {
|
|
debugf("class_resolver_lookup_string: [cached]\n");
|
|
return class_entry->attribute_entry[string_index - 1].string_objectref;
|
|
}
|
|
|
|
struct constant * utf8_constant = &class_entry->class_file->constant_pool[string_index - 1];
|
|
assert(utf8_constant->tag == CONSTANT_Utf8);
|
|
|
|
struct class_entry * string_class_entry = class_resolver_lookup_class(class_hash_table_length,
|
|
class_hash_table,
|
|
(const uint8_t *)"java/lang/String",
|
|
16);
|
|
|
|
int32_t size = utf8_constant->utf8.length + 4;
|
|
int32_t * arrayref = memory_allocate(size);
|
|
assert(arrayref != nullptr);
|
|
arrayref[0] = utf8_constant->utf8.length;
|
|
uint8_t * bytearray = (uint8_t *)&arrayref[1];
|
|
for (int i = 0; i < utf8_constant->utf8.length; i++)
|
|
bytearray[i] = utf8_constant->utf8.bytes[i];
|
|
|
|
assert(string_class_entry != nullptr);
|
|
int32_t * objectref = memory_allocate(4 + 4);
|
|
assert(objectref != nullptr);
|
|
objectref[0] = (int32_t)string_class_entry;
|
|
objectref[1] = (int32_t)arrayref;
|
|
|
|
// cache the result
|
|
class_entry->attribute_entry[string_index - 1].string_objectref = objectref;
|
|
|
|
return objectref;
|
|
}
|