Revert PySet_Add() changes.

1 files changed, 5 insertions, 0 deletions

diff --git a/Objects/setobject.c b/Objects/setobject.c
index 8a5f7cfe82..c8db7cef47 100644
--- a/Objects/setobject.c
+++ b/Objects/setobject.c
@@ -2198,6 +2198,10 @@ PySet_Discard(PyObject *set, PyObject *key)
 int
 PySet_Add(PyObject *set, PyObject *key)
 {
+	if (!PyType_IsSubtype(Py_TYPE(set), &PySet_Type)) {
+		PyErr_BadInternalCall();
+		return -1;
+	}
 	return set_add_key((PySetObject *)set, key);
 }
 
@@ -2341,6 +2345,7 @@ test_c_api(PySetObject *so)
 	f = PyFrozenSet_New(dup);
 	assert(PySet_Size(f) == 3);
 	assert(PyFrozenSet_CheckExact(f));
+	assertRaises(PySet_Add(f, elem) == -1, PyExc_SystemError);
 	assertRaises(PySet_Discard(f, elem) == -1, PyExc_SystemError);
 	assertRaises(PySet_Pop(f) == NULL, PyExc_SystemError);
 	Py_DECREF(f);