pycades/PyEnrollContainerKeys.cpp at main · CryptoPro/pycades · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include "PyEnrollContainerKeys.h"
#include "PyEnrollContainerKey.h"
#include "PyEnrollContainer.h"

using namespace CryptoPro::PKI::Enroll;

static void ContainerKeys_dealloc(ContainerKeys *self)
{
    self->m_pCppEnrollImpl.reset();
    Py_TYPE(self)->tp_free((PyObject *)self);
}

static PyObject *ContainerKeys_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
    ContainerKeys *self;
    self = (ContainerKeys *)type->tp_alloc(type, 0);
    if (self != NULL)
    {
        self->m_pCppEnrollImpl = NS_SHARED_PTR::shared_ptr<CPPEnrollContainerKeys>(new CPPEnrollContainerKeys());
    }
    return (PyObject *)self;
}

static PyObject *ContainerKeys_getCount(ContainerKeys *self)
{
    unsigned int Count = 0;
    HR_METHOD_ERRORCHECK_RETURN(self->m_pCppEnrollImpl->get_Count(&Count));
    return Py_BuildValue("l", Count);
}

static PyObject *ContainerKeys_getItemByIndex(ContainerKeys *self, PyObject *args)
{
    long lIndex = 0;
    if (!PyArg_ParseTuple(args, "l", &lIndex))
    {
        return NULL;
    }
    NS_SHARED_PTR::shared_ptr<CPPEnrollContainerKey> pCPPEnrollContainerKey;
    HR_METHOD_ERRORCHECK_RETURN(self->m_pCppEnrollImpl->get_ItemByIndex(lIndex, pCPPEnrollContainerKey));
    PyObject *pPyContainerKey = PyObject_CallObject((PyObject *)&ContainerKeyType, NULL);
    ContainerKey *pContainerKey = (ContainerKey *)pPyContainerKey;
    pContainerKey->m_pCppEnrollImpl = pCPPEnrollContainerKey;
    return Py_BuildValue("N", pContainerKey);
}

static PyGetSetDef ContainerKeys_getset[] = {
    {"Count", (getter)ContainerKeys_getCount, NULL, "Count", NULL},
    {NULL} /* Sentinel */
};

static PyMethodDef ContainerKeys_methods[] = {
    {"ItemByIndex", (PyCFunction)ContainerKeys_getItemByIndex, METH_VARARGS, "ItemByIndex"},
    {NULL} /* Sentinel */
};

PyTypeObject ContainerKeysType = {
    PyVarObject_HEAD_INIT(NULL, 0) "pycades.ContainerKeys", /* tp_name */
    sizeof(ContainerKeys),                                  /* tp_basicsize */
    0,                                                      /* tp_itemsize */
    (destructor)ContainerKeys_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 | Py_TPFLAGS_BASETYPE,               /* tp_flags */
    "ContainerKeys object",                                 /* tp_doc */
    0,                                                      /* tp_traverse */
    0,                                                      /* tp_clear */
    0,                                                      /* tp_richcompare */
    0,                                                      /* tp_weaklistoffset */
    0,                                                      /* tp_iter */
    0,                                                      /* tp_iternext */
    ContainerKeys_methods,                                  /* tp_methods */
    0,                                                      /* tp_members */
    ContainerKeys_getset,                                   /* tp_getset */
    0,                                                      /* tp_base */
    0,                                                      /* tp_dict */
    0,                                                      /* tp_descr_get */
    0,                                                      /* tp_descr_set */
    0,                                                      /* tp_dictoffset */
    0,                                                      /* tp_init */
    0,                                                      /* tp_alloc */
    ContainerKeys_new,                                      /* tp_new */
};