source: trunk/server/source4/librpc/ndr/py_misc.c

/* 
   Unix SMB/CIFS implementation.
   Samba utility functions
   Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2008
   
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 3 of the License, or
   (at your option) any later version.
   
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
   
   You should have received a copy of the GNU General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/
#include <Python.h>
#include "librpc/gen_ndr/misc.h"

#ifndef Py_RETURN_NONE
#define Py_RETURN_NONE return Py_INCREF(Py_None), Py_None
#endif

static int py_GUID_cmp(PyObject *py_self, PyObject *py_other)
{
        int ret;
        struct GUID *self = py_talloc_get_ptr(py_self), *other;
        other = py_talloc_get_ptr(py_other);
        if (other == NULL)
                return -1;

        ret = GUID_compare(self, other);
        if (ret < 0) {
                return -1;
        } else if (ret > 0) {
                return 1;
        } else {
                return 0;
        }
}

static PyObject *py_GUID_str(PyObject *py_self)
{
        struct GUID *self = py_talloc_get_ptr(py_self);
        char *str = GUID_string(NULL, self);
        PyObject *ret = PyString_FromString(str);
        talloc_free(str);
        return ret;
}

static PyObject *py_GUID_repr(PyObject *py_self)
{
        struct GUID *self = py_talloc_get_ptr(py_self);
        char *str = GUID_string(NULL, self);
        PyObject *ret = PyString_FromFormat("GUID('%s')", str);
        talloc_free(str);
        return ret;
}

static int py_GUID_init(PyObject *self, PyObject *args, PyObject *kwargs)
{
        PyObject *str = NULL;
        NTSTATUS status;
        struct GUID *guid = py_talloc_get_ptr(self);
        const char *kwnames[] = { "str", NULL };

        if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|O", discard_const_p(char *, kwnames), &str))
                return -1;

        if (str != NULL) {
                DATA_BLOB guid_val;

                if (!PyString_Check(str)) {
                        PyErr_SetString(PyExc_TypeError, "Expected a string argument to GUID()");
                        return -1;
                }
                guid_val.data = (uint8_t *)PyString_AsString(str);
                guid_val.length = PyString_Size(str);
                status = GUID_from_data_blob(&guid_val, guid);
                if (!NT_STATUS_IS_OK(status)) {
                        PyErr_SetNTSTATUS(status);
                        return -1;
                }
        }

        return 0;
}

static void py_GUID_patch(PyTypeObject *type)
{
        type->tp_init = py_GUID_init;
        type->tp_str = py_GUID_str;
        type->tp_repr = py_GUID_repr;
        type->tp_compare = py_GUID_cmp;
}

#define PY_GUID_PATCH py_GUID_patch

static int py_policy_handle_init(PyObject *self, PyObject *args, PyObject *kwargs)
{
        char *str = NULL;
        NTSTATUS status;
        struct policy_handle *handle = py_talloc_get_ptr(self);
        const char *kwnames[] = { "uuid", "type", NULL };

        if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|si", discard_const_p(char *, kwnames), &str, &handle->handle_type))
                return -1;

        if (str != NULL) {
                status = GUID_from_string(str, &handle->uuid);
                if (!NT_STATUS_IS_OK(status)) {
                        PyErr_SetNTSTATUS(status);
                        return -1;
                }
        }

        return 0;
}

static PyObject *py_policy_handle_repr(PyObject *py_self)
{
        struct policy_handle *self = py_talloc_get_ptr(py_self);
        char *uuid_str = GUID_string(NULL, &self->uuid);
        PyObject *ret = PyString_FromFormat("policy_handle(%d, '%s')", self->handle_type, uuid_str);
        talloc_free(uuid_str);
        return ret;
}

static PyObject *py_policy_handle_str(PyObject *py_self)
{
        struct policy_handle *self = py_talloc_get_ptr(py_self);
        char *uuid_str = GUID_string(NULL, &self->uuid);
        PyObject *ret = PyString_FromFormat("%d, %s", self->handle_type, uuid_str);
        talloc_free(uuid_str);
        return ret;
}

static void py_policy_handle_patch(PyTypeObject *type)
{
        type->tp_init = py_policy_handle_init;
        type->tp_repr = py_policy_handle_repr;
        type->tp_str = py_policy_handle_str;
}

#define PY_POLICY_HANDLE_PATCH py_policy_handle_patch