Fix crash when creating thread

This commit is contained in:
Haoyu Qiu 2021-09-25 23:58:58 +08:00
parent 90a7342114
commit 7be67ee1b7
2 changed files with 17 additions and 9 deletions

View file

@ -2617,6 +2617,12 @@ void _Thread::_start_func(void *ud) {
Ref<_Thread> *tud = (Ref<_Thread> *)ud; Ref<_Thread> *tud = (Ref<_Thread> *)ud;
Ref<_Thread> t = *tud; Ref<_Thread> t = *tud;
memdelete(tud); memdelete(tud);
Object *target_instance = ObjectDB::get_instance(t->target_instance_id);
if (!target_instance) {
ERR_FAIL_MSG(vformat("Could not call function '%s' on previously freed instance to start thread %s.", t->target_method, t->get_id()));
}
Variant::CallError ce; Variant::CallError ce;
const Variant *arg[1] = { &t->userdata }; const Variant *arg[1] = { &t->userdata };
int argc = 0; int argc = 0;
@ -2635,16 +2641,18 @@ void _Thread::_start_func(void *ud) {
// We must check if we are in case b). // We must check if we are in case b).
int target_param_count = 0; int target_param_count = 0;
int target_default_arg_count = 0; int target_default_arg_count = 0;
Ref<Script> script = t->target_instance->get_script(); Ref<Script> script = target_instance->get_script();
if (script.is_valid()) { if (script.is_valid()) {
MethodInfo mi = script->get_method_info(t->target_method); MethodInfo mi = script->get_method_info(t->target_method);
target_param_count = mi.arguments.size(); target_param_count = mi.arguments.size();
target_default_arg_count = mi.default_arguments.size(); target_default_arg_count = mi.default_arguments.size();
} else { } else {
MethodBind *method = ClassDB::get_method(t->target_instance->get_class_name(), t->target_method); MethodBind *method = ClassDB::get_method(target_instance->get_class_name(), t->target_method);
if (method) {
target_param_count = method->get_argument_count(); target_param_count = method->get_argument_count();
target_default_arg_count = method->get_default_argument_count(); target_default_arg_count = method->get_default_argument_count();
} }
}
if (target_param_count >= 1 && target_default_arg_count < target_param_count) { if (target_param_count >= 1 && target_default_arg_count < target_param_count) {
argc = 1; argc = 1;
} }
@ -2652,7 +2660,7 @@ void _Thread::_start_func(void *ud) {
Thread::set_name(t->target_method); Thread::set_name(t->target_method);
t->ret = t->target_instance->call(t->target_method, arg, argc, ce); t->ret = target_instance->call(t->target_method, arg, argc, ce);
if (ce.error != Variant::CallError::CALL_OK) { if (ce.error != Variant::CallError::CALL_OK) {
String reason; String reason;
switch (ce.error) { switch (ce.error) {
@ -2684,7 +2692,7 @@ Error _Thread::start(Object *p_instance, const StringName &p_method, const Varia
ret = Variant(); ret = Variant();
target_method = p_method; target_method = p_method;
target_instance = p_instance; target_instance_id = p_instance->get_instance_id();
userdata = p_userdata; userdata = p_userdata;
active.set(); active.set();
@ -2709,7 +2717,7 @@ Variant _Thread::wait_to_finish() {
thread.wait_to_finish(); thread.wait_to_finish();
Variant r = ret; Variant r = ret;
target_method = StringName(); target_method = StringName();
target_instance = nullptr; target_instance_id = ObjectID();
userdata = Variant(); userdata = Variant();
active.clear(); active.clear();
@ -2727,7 +2735,7 @@ void _Thread::_bind_methods() {
BIND_ENUM_CONSTANT(PRIORITY_HIGH); BIND_ENUM_CONSTANT(PRIORITY_HIGH);
} }
_Thread::_Thread() { _Thread::_Thread() {
target_instance = nullptr; target_instance_id = ObjectID();
} }
_Thread::~_Thread() { _Thread::~_Thread() {

View file

@ -684,7 +684,7 @@ protected:
Variant ret; Variant ret;
Variant userdata; Variant userdata;
SafeFlag active; SafeFlag active;
Object *target_instance; ObjectID target_instance_id;
StringName target_method; StringName target_method;
Thread thread; Thread thread;
static void _bind_methods(); static void _bind_methods();