33 lines
962 B
C
33 lines
962 B
C
// This file is part of the CircuitPython project: https://circuitpython.org
|
|
//
|
|
// SPDX-FileCopyrightText: Copyright (c) 2019 Jeff Epler for Adafruit Industries
|
|
//
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
#include "py/obj.h"
|
|
#include "py/proto.h"
|
|
#include "py/runtime.h"
|
|
|
|
#ifndef MICROPY_UNSAFE_PROTO
|
|
const void *mp_proto_get(uint16_t name, mp_const_obj_t obj) {
|
|
const mp_obj_type_t *type = mp_obj_get_type(obj);
|
|
const void *protocol = MP_OBJ_TYPE_GET_SLOT_OR_NULL(type, protocol);
|
|
if (!protocol) {
|
|
return NULL;
|
|
}
|
|
uint16_t proto_name = *(const uint16_t *)protocol;
|
|
if (proto_name == name) {
|
|
return protocol;
|
|
}
|
|
return NULL;
|
|
}
|
|
#endif
|
|
|
|
const void *mp_proto_get_or_throw(uint16_t name, mp_const_obj_t obj) {
|
|
const void *proto = mp_proto_get(name, obj);
|
|
if (proto) {
|
|
return proto;
|
|
}
|
|
mp_raise_TypeError_varg(MP_ERROR_TEXT("'%q' object does not support '%q'"),
|
|
mp_obj_get_type_qstr(obj), name);
|
|
}
|