Skip to content

[mono][aot] Type load checks do not fail at compile time but produce a runtime exception #91261

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 21 commits into from
Sep 27, 2023
Merged

[mono][aot] Type load checks do not fail at compile time but produce a runtime exception #91261

Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
78ce32e
Enable tests.
jandupej Aug 28, 2023
3c32b67
When AOTing, type checks do not fail compilation but create a runtime…
jandupej Aug 29, 2023
7676e00
Cleaned up type load error cleaning. TypeLoadException icall now has …
jandupej Aug 30, 2023
611888b
Removed another instance of indiscriminate exception clearing.
jandupej Aug 30, 2023
15b7aa5
Fixed build warning.
jandupej Aug 30, 2023
90aa674
Using class const instead of string const. Reverted some compile to r…
jandupej Aug 31, 2023
d4b6c9f
White space.
jandupej Aug 31, 2023
bda12b3
Fixed build warning.
jandupej Aug 31, 2023
fbd3614
Trying to fix weird AOT errors, fixed type load throw function.
jandupej Aug 31, 2023
805d6c4
Fixed build error.
jandupej Aug 31, 2023
0b50974
Special handling for classes that are NULL.
jandupej Sep 1, 2023
19c5155
Providing for a null klass when generating exception.
jandupej Sep 6, 2023
83bcc30
Removed flow control directive from macro.
jandupej Sep 6, 2023
c6fe63b
Fixed stack corruption.
jandupej Sep 6, 2023
9cb3d1a
Attempt to push the correct type onto the stack.
jandupej Sep 8, 2023
025ed7a
Fixing uninitialized ins.
jandupej Sep 8, 2023
4633336
Fixing ro_type.
jandupej Sep 8, 2023
55db3c6
Initializing ins.
jandupej Sep 8, 2023
4b35af0
Complex cases with type load failures replace method body with a throw.
jandupej Sep 20, 2023
ca7530d
Cleaning up superfluous code changes.
jandupej Sep 21, 2023
9709a89
Restored sizeof cosntant on failed types.
jandupej Sep 21, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions src/mono/mono/metadata/jit-icall-reg.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -312,6 +312,7 @@ MONO_JIT_ICALL (mono_throw_bad_image) \
MONO_JIT_ICALL (mono_throw_not_supported) \
MONO_JIT_ICALL (mono_throw_platform_not_supported) \
MONO_JIT_ICALL (mono_throw_invalid_program) \
MONO_JIT_ICALL (mono_throw_type_load) \
MONO_JIT_ICALL (mono_trace_enter_method) \
MONO_JIT_ICALL (mono_trace_leave_method) \
MONO_JIT_ICALL (mono_trace_tail_method) \
Expand Down
11 changes: 11 additions & 0 deletions src/mono/mono/mini/jit-icalls.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1688,6 +1688,17 @@ mono_throw_invalid_program (const char *msg)
mono_error_set_pending_exception (error);
}

void
mono_throw_type_load (MonoClass* klass)
{
ERROR_DECL (error);
char* klass_name = mono_type_get_full_name (klass);
mono_error_set_type_load_class (error, klass, "Attempting to load invalid type '%s'.", klass_name);
mono_error_set_pending_exception (error);

g_free (klass_name);
}

void
mono_dummy_jit_icall (void)
{
Expand Down
2 changes: 2 additions & 0 deletions src/mono/mono/mini/jit-icalls.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -234,6 +234,8 @@ ICALL_EXPORT void mono_throw_platform_not_supported (void);

ICALL_EXPORT void mono_throw_invalid_program (const char *msg);

ICALL_EXPORT void mono_throw_type_load (MonoClass* klass);

ICALL_EXPORT void mono_dummy_jit_icall (void);

ICALL_EXPORT void mono_dummy_jit_icall_val (gpointer ptr);
Expand Down
50 changes: 44 additions & 6 deletions src/mono/mono/mini/method-to-ir.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ mono_tailcall_print (const char *format, ...)

#define TYPE_LOAD_ERROR(klass) do { \
cfg->exception_ptr = klass; \
LOAD_ERROR; \
LOAD_ERROR; \
} while (0)

#define CHECK_CFG_ERROR do {\
Expand Down Expand Up @@ -2303,6 +2303,18 @@ emit_not_supported_failure (MonoCompile *cfg)
mono_emit_jit_icall (cfg, mono_throw_not_supported, NULL);
}

static void
emit_type_load_failure (MonoCompile* cfg, MonoClass* klass)
{
MonoClass* failed_class = klass;
if (G_UNLIKELY (!failed_class))
failed_class = mono_defaults.object_class;

MonoInst* iargs[1];
EMIT_NEW_CLASSCONST (cfg, iargs [0], failed_class);
mono_emit_jit_icall (cfg, mono_throw_type_load, iargs);
}

static void
emit_invalid_program_with_msg (MonoCompile *cfg, MonoError *error_msg, MonoMethod *caller, MonoMethod *callee)
{
Expand Down Expand Up @@ -4963,6 +4975,15 @@ inline_method (MonoCompile *cfg, MonoMethod *cmethod, MonoMethodSignature *fsig,
#define CHECK_UNVERIFIABLE(cfg) if (cfg->unverifiable) UNVERIFIED
#define CHECK_TYPELOAD(klass) if (!(klass) || mono_class_has_failure (klass)) TYPE_LOAD_ERROR ((klass))

#define CLEAR_TYPELOAD_EXCEPTION(cfg) if (cfg->exception_type == MONO_EXCEPTION_TYPE_LOAD) { clear_cfg_error (cfg); cfg->exception_type = MONO_EXCEPTION_NONE; }
#define IF_TYPELOAD_ERROR(klass) if (!(klass) || mono_class_has_failure (klass))
#define HANDLE_TYPELOAD_ERROR(cfg,klass) do { \
if (!cfg->compile_aot) \
TYPE_LOAD_ERROR ((klass)); \
emit_type_load_failure (cfg, klass); \
CLEAR_TYPELOAD_EXCEPTION (cfg); \
} while (0)

/* offset from br.s -> br like opcodes */
#define BIG_BRANCH_OFFSET 13

Expand Down Expand Up @@ -5438,7 +5459,10 @@ emit_optimized_ldloca_ir (MonoCompile *cfg, guchar *ip, guchar *end, int local)
if ((ip = il_read_initobj (ip, end, &token)) && ip_in_bb (cfg, cfg->cbb, start + 1)) {
/* From the INITOBJ case */
klass = mini_get_class (cfg->current_method, token, cfg->generic_context);
CHECK_TYPELOAD (klass);
IF_TYPELOAD_ERROR (klass) {
HANDLE_TYPELOAD_ERROR (cfg, klass);
return ip;
}
type = mini_get_underlying_type (m_class_get_byval_arg (klass));
emit_init_local (cfg, local, type, TRUE);
return ip;
Expand Down Expand Up @@ -9898,8 +9922,12 @@ mono_method_to_ir (MonoCompile *cfg, MonoMethod *method, MonoBasicBlock *start_b
else {
klass = NULL;
field = mono_field_from_token_checked (image, token, &klass, generic_context, cfg->error);
if (!field)
CHECK_TYPELOAD (klass);
if (!field) {
IF_TYPELOAD_ERROR (klass) {
HANDLE_TYPELOAD_ERROR (cfg, klass);
break; // reached only in AOT
}
}
CHECK_CFG_ERROR;
}
if (!dont_verify && !cfg->skip_visibility && !mono_method_can_access_field (method, field))
Expand Down Expand Up @@ -11855,11 +11883,18 @@ mono_method_to_ir (MonoCompile *cfg, MonoMethod *method, MonoBasicBlock *start_b
case MONO_CEE_INITOBJ:
--sp;
klass = mini_get_class (method, token, generic_context);
CHECK_TYPELOAD (klass);

IF_TYPELOAD_ERROR (klass) {
HANDLE_TYPELOAD_ERROR (cfg, klass);
inline_costs += 10;
break; // reached only in AOT
}

if (mini_class_is_reference (klass))
MONO_EMIT_NEW_STORE_MEMBASE_IMM (cfg, OP_STORE_MEMBASE_IMM, sp [0]->dreg, 0, 0);
else
mini_emit_initobj (cfg, *sp, NULL, klass);

inline_costs += 1;
break;
case MONO_CEE_CONSTRAINED_:
Expand Down Expand Up @@ -11949,7 +11984,10 @@ mono_method_to_ir (MonoCompile *cfg, MonoMethod *method, MonoBasicBlock *start_b
EMIT_NEW_ICONST (cfg, ins, val);
} else {
klass = mini_get_class (method, token, generic_context);
CHECK_TYPELOAD (klass);
IF_TYPELOAD_ERROR (klass) {
HANDLE_TYPELOAD_ERROR (cfg, klass);
break;
}

if (mini_is_gsharedvt_klass (klass)) {
ins = mini_emit_get_gsharedvt_info_klass (cfg, klass, MONO_RGCTX_INFO_CLASS_SIZEOF);
Expand Down
1 change: 1 addition & 0 deletions src/mono/mono/mini/mini-runtime.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5139,6 +5139,7 @@ register_icalls (void)
register_icall (mono_throw_not_supported, mono_icall_sig_void, FALSE);
register_icall (mono_throw_platform_not_supported, mono_icall_sig_void, FALSE);
register_icall (mono_throw_invalid_program, mono_icall_sig_void_ptr, FALSE);
register_icall (mono_throw_type_load, mono_icall_sig_void_ptr, FALSE);
register_icall_no_wrapper (mono_dummy_jit_icall, mono_icall_sig_void);
//register_icall_no_wrapper (mono_dummy_jit_icall_val, mono_icall_sig_void_ptr);
register_icall_no_wrapper (mini_init_method_rgctx, mono_icall_sig_void_ptr_ptr);
Expand Down
9 changes: 7 additions & 2 deletions src/mono/mono/mini/mini.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -660,7 +660,9 @@ mono_compile_create_var_for_vreg (MonoCompile *cfg, MonoType *type, int opcode,
inst->backend.is_pinvoke = 0;
inst->dreg = vreg;

if (mono_class_has_failure (inst->klass))
// In AOT, we do not set the exception so that the compilation can succeed. To indicate
// the error, an exception is thrown in run-time.
if (!cfg->compile_aot && mono_class_has_failure (inst->klass))
mono_cfg_set_exception (cfg, MONO_EXCEPTION_TYPE_LOAD);

if (cfg->compute_gc_maps) {
Expand Down Expand Up @@ -1351,7 +1353,10 @@ mono_allocate_stack_slots (MonoCompile *cfg, gboolean backward, guint32 *stack_s
size = mini_type_stack_size (t, &ialign);
align = ialign;

if (mono_class_has_failure (mono_class_from_mono_type_internal (t)))
// In AOT, we do not set the exception but allow the compilation to succeed. The error will be
// indicated in runtime by throwing an exception when an operation with the invalid object is
// attempted.
if (!cfg->compile_aot && mono_class_has_failure (mono_class_from_mono_type_internal (t)))
mono_cfg_set_exception (cfg, MONO_EXCEPTION_TYPE_LOAD);

if (mini_class_is_simd (cfg, mono_class_from_mono_type_internal (t)))
Expand Down
9 changes: 0 additions & 9 deletions src/tests/issues.targets
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2405,15 +2405,6 @@
<ExcludeList Include="$(XunitTestBinBase)/Loader/classloader/explicitlayout/objrefandnonobjrefoverlap/case9/**">
<Issue>expected failure: overlapped structs fail at AOT compile time, not runtime</Issue>
</ExcludeList>
<ExcludeList Include = "$(XunitTestBinBase)/Loader/classloader/explicitlayout/NestedStructs/case03/**">
<Issue>expected failure: overlapped structs fail at AOT compile time, not runtime</Issue>
</ExcludeList>
<ExcludeList Include = "$(XunitTestBinBase)/Loader/classloader/explicitlayout/NestedStructs/case04/**">
<Issue>expected failure: overlapped structs fail at AOT compile time, not runtime</Issue>
</ExcludeList>
<ExcludeList Include = "$(XunitTestBinBase)/Loader/classloader/explicitlayout/NestedStructs/case05/**">
<Issue>expected failure: overlapped structs fail at AOT compile time, not runtime</Issue>
</ExcludeList>
<ExcludeList Include = "$(XunitTestBinBase)/Loader/classloader/RefFields/Validate/**">
<Issue>expected failure: unsupported type with ref field fails at AOT compile time, not runtime</Issue>
</ExcludeList>
Expand Down