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
pull/3829/head
Jeff Epler 2018-03-25 16:13:49 -05:00 zatwierdzone przez Damien George
rodzic 05b13fd292
commit c60589c02b
2 zmienionych plików z 9 dodań i 0 usunięć

Wyświetl plik

@ -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);

Wyświetl plik

@ -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: