|
|
@@ -0,0 +1,157 @@
|
|
|
+/*
|
|
|
+ *
|
|
|
+ * Copyright 2015, Google Inc.
|
|
|
+ * All rights reserved.
|
|
|
+ *
|
|
|
+ * 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 "_adapter/_server_credentials.h"
|
|
|
+
|
|
|
+#include <Python.h>
|
|
|
+#include <grpc/grpc_security.h>
|
|
|
+#include <grpc/support/alloc.h>
|
|
|
+
|
|
|
+static int pygrpc_server_credentials_init(ServerCredentials *self,
|
|
|
+ PyObject *args, PyObject *kwds) {
|
|
|
+ char *root_certificates;
|
|
|
+ PyObject *pair_sequence;
|
|
|
+ Py_ssize_t pair_count;
|
|
|
+ grpc_ssl_pem_key_cert_pair *pairs;
|
|
|
+ int error;
|
|
|
+ PyObject *iterator;
|
|
|
+ int i;
|
|
|
+ PyObject *pair;
|
|
|
+
|
|
|
+ if (!(PyArg_ParseTuple(args, "zO", &root_certificates, &pair_sequence))) {
|
|
|
+ self->c_server_credentials = NULL;
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ pair_count = PySequence_Length(pair_sequence);
|
|
|
+ if (pair_count == -1) {
|
|
|
+ self->c_server_credentials = NULL;
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ iterator = PyObject_GetIter(pair_sequence);
|
|
|
+ if (iterator == NULL) {
|
|
|
+ self->c_server_credentials = NULL;
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+ pairs = gpr_malloc(pair_count * sizeof(grpc_ssl_pem_key_cert_pair));
|
|
|
+ error = 0;
|
|
|
+ for (i = 0; i < pair_count; i++) {
|
|
|
+ pair = PyIter_Next(iterator);
|
|
|
+ if (pair == NULL) {
|
|
|
+ error = 1;
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ if (!(PyArg_ParseTuple(pair, "ss", &pairs[i].private_key,
|
|
|
+ &pairs[i].cert_chain))) {
|
|
|
+ error = 1;
|
|
|
+ Py_DECREF(pair);
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ Py_DECREF(pair);
|
|
|
+ }
|
|
|
+ Py_DECREF(iterator);
|
|
|
+
|
|
|
+ if (error) {
|
|
|
+ self->c_server_credentials = NULL;
|
|
|
+ gpr_free(pairs);
|
|
|
+ return -1;
|
|
|
+ } else {
|
|
|
+ self->c_server_credentials = grpc_ssl_server_credentials_create(
|
|
|
+ root_certificates, pairs, pair_count);
|
|
|
+ gpr_free(pairs);
|
|
|
+ return 0;
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static void pygrpc_server_credentials_dealloc(ServerCredentials *self) {
|
|
|
+ if (self->c_server_credentials != NULL) {
|
|
|
+ grpc_server_credentials_release(self->c_server_credentials);
|
|
|
+ }
|
|
|
+ self->ob_type->tp_free((PyObject *)self);
|
|
|
+}
|
|
|
+
|
|
|
+PyTypeObject pygrpc_ServerCredentialsType = {
|
|
|
+ PyObject_HEAD_INIT(NULL)0, /*ob_size*/
|
|
|
+ "_grpc.ServerCredencials", /*tp_name*/
|
|
|
+ sizeof(ServerCredentials), /*tp_basicsize*/
|
|
|
+ 0, /*tp_itemsize*/
|
|
|
+ (destructor)pygrpc_server_credentials_dealloc, /*tp_dealloc*/
|
|
|
+ 0, /*tp_print*/
|
|
|
+ 0, /*tp_getattr*/
|
|
|
+ 0, /*tp_setattr*/
|
|
|
+ 0, /*tp_compare*/
|
|
|
+ 0, /*tp_repr*/
|
|
|
+ 0, /*tp_as_number*/
|
|
|
+ 0, /*tp_as_sequence*/
|
|
|
+ 0, /*tp_as_mapping*/
|
|
|
+ 0, /*tp_hash */
|
|
|
+ 0, /*tp_call*/
|
|
|
+ 0, /*tp_str*/
|
|
|
+ 0, /*tp_getattro*/
|
|
|
+ 0, /*tp_setattro*/
|
|
|
+ 0, /*tp_as_buffer*/
|
|
|
+ Py_TPFLAGS_DEFAULT, /*tp_flags*/
|
|
|
+ "Wrapping of grpc_server_credentials.", /* tp_doc */
|
|
|
+ 0, /* tp_traverse */
|
|
|
+ 0, /* tp_clear */
|
|
|
+ 0, /* tp_richcompare */
|
|
|
+ 0, /* tp_weaklistoffset */
|
|
|
+ 0, /* tp_iter */
|
|
|
+ 0, /* tp_iternext */
|
|
|
+ 0, /* tp_methods */
|
|
|
+ 0, /* tp_members */
|
|
|
+ 0, /* tp_getset */
|
|
|
+ 0, /* tp_base */
|
|
|
+ 0, /* tp_dict */
|
|
|
+ 0, /* tp_descr_get */
|
|
|
+ 0, /* tp_descr_set */
|
|
|
+ 0, /* tp_dictoffset */
|
|
|
+ (initproc)pygrpc_server_credentials_init, /* tp_init */
|
|
|
+};
|
|
|
+
|
|
|
+int pygrpc_add_server_credentials(PyObject *module) {
|
|
|
+ pygrpc_ServerCredentialsType.tp_new = PyType_GenericNew;
|
|
|
+ if (PyType_Ready(&pygrpc_ServerCredentialsType) < 0) {
|
|
|
+ PyErr_SetString(PyExc_RuntimeError,
|
|
|
+ "Error defining pygrpc_ServerCredentialsType!");
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+ if (PyModule_AddObject(module, "ServerCredentials",
|
|
|
+ (PyObject *)&pygrpc_ServerCredentialsType) == -1) {
|
|
|
+ PyErr_SetString(PyExc_ImportError,
|
|
|
+ "Couldn't add ServerCredentials type to module!");
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+ return 0;
|
|
|
+}
|
Nathaniel Manista