Skip to content
Snippets Groups Projects
Commit c60589c0 authored by Jeff Epler's avatar Jeff Epler Committed by Damien George
Browse files

py/objtype: Fix assertion failures in super_attr by checking type.

Fixes assertion failures and segmentation faults when making calls like:

    super(1, 1).x
parent 05b13fd2
No related branches found
No related tags found
No related merge requests found
......@@ -1112,6 +1112,9 @@ STATIC mp_obj_t super_make_new(const mp_obj_type_t *type_in, size_t n_args, size
// 0 arguments are turned into 2 in the compiler
// 1 argument is not yet implemented
mp_arg_check_num(n_args, n_kw, 2, 2, false);
if (!MP_OBJ_IS_TYPE(args[0], &mp_type_type)) {
mp_raise_TypeError(NULL);
}
mp_obj_super_t *o = m_new_obj(mp_obj_super_t);
*o = (mp_obj_super_t){{type_in}, args[0], args[1]};
return MP_OBJ_FROM_PTR(o);
......
......@@ -35,6 +35,12 @@ class B(A):
return super().foo().count(2) # calling a subsequent method
print(B().foo())
# first arg to super must be a type
try:
super(1, 1)
except TypeError:
print('TypeError')
# store/delete of super attribute not allowed
assert hasattr(super(B, B()), 'foo')
try:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment