source4/librpc/ndr/py_misc.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4
   5    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2008
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20 #include "lib/replace/system/python.h"
  21 #include "python/py3compat.h"
  22 #include "librpc/gen_ndr/misc.h"
  23
  24 static PyObject *py_GUID_richcmp(PyObject *py_self, PyObject *py_other, int op)
  25 {
  26         int ret;
  27         struct GUID *self = pytalloc_get_ptr(py_self), *other;
  28         other = pytalloc_get_ptr(py_other);
  29         if (other == NULL) {
  30                 Py_INCREF(Py_NotImplemented);
  31                 return Py_NotImplemented;
  32         }
  33
  34         ret = GUID_compare(self, other);
  35
  36         switch (op) {
  37                 case Py_EQ: if (ret == 0) Py_RETURN_TRUE; else Py_RETURN_FALSE;
  38                 case Py_NE: if (ret != 0) Py_RETURN_TRUE; else Py_RETURN_FALSE;
  39                 case Py_LT: if (ret <  0) Py_RETURN_TRUE; else Py_RETURN_FALSE;
  40                 case Py_GT: if (ret >  0) Py_RETURN_TRUE; else Py_RETURN_FALSE;
  41                 case Py_LE: if (ret <= 0) Py_RETURN_TRUE; else Py_RETURN_FALSE;
  42                 case Py_GE: if (ret >= 0) Py_RETURN_TRUE; else Py_RETURN_FALSE;
  43         }
  44         Py_INCREF(Py_NotImplemented);
  45         return Py_NotImplemented;
  46 }
  47
  48 static PyObject *py_GUID_str(PyObject *py_self)
  49 {
  50         struct GUID *self = pytalloc_get_ptr(py_self);
  51         struct GUID_txt_buf buf;
  52         PyObject *ret = PyUnicode_FromString(GUID_buf_string(self, &buf));
  53         return ret;
  54 }
  55
  56 static PyObject *py_GUID_repr(PyObject *py_self)
  57 {
  58         struct GUID *self = pytalloc_get_ptr(py_self);
  59         struct GUID_txt_buf buf;
  60         PyObject *ret = PyUnicode_FromFormat(
  61                 "GUID('%s')", GUID_buf_string(self, &buf));
  62         return ret;
  63 }
  64
  65 static int py_GUID_init(PyObject *self, PyObject *args, PyObject *kwargs)
  66 {
  67         PyObject *str = NULL;
  68         NTSTATUS status;
  69         struct GUID *guid = pytalloc_get_ptr(self);
  70         const char *kwnames[] = { "str", NULL };
  71
  72         if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|O", discard_const_p(char *, kwnames), &str))
  73                 return -1;
  74
  75         if (str != NULL) {
  76                 DATA_BLOB guid_val;
  77                 Py_ssize_t _size;
  78
  79                 if (PyUnicode_Check(str)) {
  80                         guid_val.data =
  81                                 discard_const_p(uint8_t,
  82                                                 PyUnicode_AsUTF8AndSize(str, &_size));
  83                 } else if (PyBytes_Check(str)) {
  84                         guid_val.data =
  85                                 discard_const_p(uint8_t,
  86                                                 PyBytes_AsString(str));
  87                         _size = PyBytes_Size(str);
  88                 } else {
  89                         PyErr_SetString(PyExc_TypeError,
  90                                         "Expected a string or bytes argument to GUID()");
  91                         return -1;
  92                 }
  93                 guid_val.length = _size;
  94                 status = GUID_from_data_blob(&guid_val, guid);
  95                 if (!NT_STATUS_IS_OK(status)) {
  96                         PyErr_SetNTSTATUS(status);
  97                         return -1;
  98                 }
  99         }
 100
 101         return 0;
 102 }
 103
 104 static void py_GUID_patch(PyTypeObject *type)
 105 {
 106         type->tp_init = py_GUID_init;
 107         type->tp_str = py_GUID_str;
 108         type->tp_repr = py_GUID_repr;
 109         type->tp_richcompare = py_GUID_richcmp;
 110 }
 111
 112 #define PY_GUID_PATCH py_GUID_patch
 113
 114 static int py_policy_handle_init(PyObject *self, PyObject *args, PyObject *kwargs)
 115 {
 116         char *str = NULL;
 117         NTSTATUS status;
 118         struct policy_handle *handle = pytalloc_get_ptr(self);
 119         const char *kwnames[] = { "uuid", "type", NULL };
 120
 121         if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|sI", discard_const_p(char *, kwnames), &str, &handle->handle_type))
 122                 return -1;
 123
 124         if (str != NULL) {
 125                 status = GUID_from_string(str, &handle->uuid);
 126                 if (!NT_STATUS_IS_OK(status)) {
 127                         PyErr_SetNTSTATUS(status);
 128                         return -1;
 129                 }
 130         }
 131
 132         return 0;
 133 }
 134
 135 static PyObject *py_policy_handle_repr(PyObject *py_self)
 136 {
 137         struct policy_handle *self = pytalloc_get_ptr(py_self);
 138         struct GUID_txt_buf buf;
 139         PyObject *ret = PyUnicode_FromFormat(
 140                 "policy_handle(%d, '%s')",
 141                 self->handle_type,
 142                 GUID_buf_string(&self->uuid, &buf));
 143         return ret;
 144 }
 145
 146 static PyObject *py_policy_handle_str(PyObject *py_self)
 147 {
 148         struct policy_handle *self = pytalloc_get_ptr(py_self);
 149         struct GUID_txt_buf buf;
 150         PyObject *ret = PyUnicode_FromFormat(
 151                 "%d, %s",
 152                 self->handle_type,
 153                 GUID_buf_string(&self->uuid, &buf));
 154         return ret;
 155 }
 156
 157 static void py_policy_handle_patch(PyTypeObject *type)
 158 {
 159         type->tp_init = py_policy_handle_init;
 160         type->tp_repr = py_policy_handle_repr;
 161         type->tp_str = py_policy_handle_str;
 162 }
 163
 164 #define PY_POLICY_HANDLE_PATCH py_policy_handle_patch
 165