| // Protocol Buffers - Google's data interchange format |
| // Copyright 2008 Google Inc. All rights reserved. |
| // https://developers.google.com/protocol-buffers/ |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions are |
| // met: |
| // |
| // * Redistributions of source code must retain the above copyright |
| // notice, this list of conditions and the following disclaimer. |
| // * Redistributions in binary form must reproduce the above |
| // copyright notice, this list of conditions and the following disclaimer |
| // in the documentation and/or other materials provided with the |
| // distribution. |
| // * Neither the name of Google Inc. nor the names of its |
| // contributors may be used to endorse or promote products derived from |
| // this software without specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| #include <Python.h> |
| |
| #include <google/protobuf/pyext/message.h> |
| |
| #include <google/protobuf/message_lite.h> |
| |
| static PyObject* GetPythonProto3PreserveUnknownsDefault( |
| PyObject* /*m*/, PyObject* /*args*/) { |
| if (google::protobuf::internal::GetProto3PreserveUnknownsDefault()) { |
| Py_RETURN_TRUE; |
| } else { |
| Py_RETURN_FALSE; |
| } |
| } |
| |
| static PyObject* SetPythonProto3PreserveUnknownsDefault( |
| PyObject* /*m*/, PyObject* arg) { |
| if (!arg || !PyBool_Check(arg)) { |
| PyErr_SetString( |
| PyExc_TypeError, |
| "Argument to SetPythonProto3PreserveUnknownsDefault must be boolean"); |
| return NULL; |
| } |
| google::protobuf::internal::SetProto3PreserveUnknownsDefault(PyObject_IsTrue(arg)); |
| Py_RETURN_NONE; |
| } |
| |
| static const char module_docstring[] = |
| "python-proto2 is a module that can be used to enhance proto2 Python API\n" |
| "performance.\n" |
| "\n" |
| "It provides access to the protocol buffers C++ reflection API that\n" |
| "implements the basic protocol buffer functions."; |
| |
| static PyMethodDef ModuleMethods[] = { |
| {"SetAllowOversizeProtos", |
| (PyCFunction)google::protobuf::python::cmessage::SetAllowOversizeProtos, |
| METH_O, "Enable/disable oversize proto parsing."}, |
| // DO NOT USE: For migration and testing only. |
| {"GetPythonProto3PreserveUnknownsDefault", |
| (PyCFunction)GetPythonProto3PreserveUnknownsDefault, |
| METH_NOARGS, "Get Proto3 preserve unknowns default."}, |
| // DO NOT USE: For migration and testing only. |
| {"SetPythonProto3PreserveUnknownsDefault", |
| (PyCFunction)SetPythonProto3PreserveUnknownsDefault, |
| METH_O, "Enable/disable proto3 unknowns preservation."}, |
| { NULL, NULL} |
| }; |
| |
| #if PY_MAJOR_VERSION >= 3 |
| static struct PyModuleDef _module = { |
| PyModuleDef_HEAD_INIT, |
| "_message", |
| module_docstring, |
| -1, |
| ModuleMethods, /* m_methods */ |
| NULL, |
| NULL, |
| NULL, |
| NULL |
| }; |
| #define INITFUNC PyInit__message |
| #define INITFUNC_ERRORVAL NULL |
| #else // Python 2 |
| #define INITFUNC init_message |
| #define INITFUNC_ERRORVAL |
| #endif |
| |
| extern "C" { |
| PyMODINIT_FUNC INITFUNC(void) { |
| PyObject* m; |
| #if PY_MAJOR_VERSION >= 3 |
| m = PyModule_Create(&_module); |
| #else |
| m = Py_InitModule3("_message", ModuleMethods, |
| module_docstring); |
| #endif |
| if (m == NULL) { |
| return INITFUNC_ERRORVAL; |
| } |
| |
| if (!google::protobuf::python::InitProto2MessageModule(m)) { |
| Py_DECREF(m); |
| return INITFUNC_ERRORVAL; |
| } |
| |
| #if PY_MAJOR_VERSION >= 3 |
| return m; |
| #endif |
| } |
| } |