Index: third_party/protobuf/python/google/protobuf/pyext/descriptor_pool.cc |
diff --git a/third_party/protobuf/python/google/protobuf/pyext/descriptor_pool.cc b/third_party/protobuf/python/google/protobuf/pyext/descriptor_pool.cc |
index 0bc76bc90a6d47d9976456029d4159bd8dc43d9b..1faff96bc2e44c40a3aa8da0550f947cd67b40df 100644 |
--- a/third_party/protobuf/python/google/protobuf/pyext/descriptor_pool.cc |
+++ b/third_party/protobuf/python/google/protobuf/pyext/descriptor_pool.cc |
@@ -190,8 +190,8 @@ PyObject* FindMessageByName(PyDescriptorPool* self, PyObject* arg) { |
// Add a message class to our database. |
int RegisterMessageClass(PyDescriptorPool* self, |
- const Descriptor *message_descriptor, |
- PyObject *message_class) { |
+ const Descriptor* message_descriptor, |
+ CMessageClass* message_class) { |
Py_INCREF(message_class); |
typedef PyDescriptorPool::ClassesByMessageMap::iterator iterator; |
std::pair<iterator, bool> ret = self->classes_by_descriptor->insert( |
@@ -205,8 +205,8 @@ int RegisterMessageClass(PyDescriptorPool* self, |
} |
// Retrieve the message class added to our database. |
-PyObject *GetMessageClass(PyDescriptorPool* self, |
- const Descriptor *message_descriptor) { |
+CMessageClass* GetMessageClass(PyDescriptorPool* self, |
+ const Descriptor* message_descriptor) { |
typedef PyDescriptorPool::ClassesByMessageMap::iterator iterator; |
iterator ret = self->classes_by_descriptor->find(message_descriptor); |
if (ret == self->classes_by_descriptor->end()) { |