pyglue: Add function to generate a random byte string

[amitay/samba.git] / python / pyglue.c

diff --git a/python/pyglue.c b/python/pyglue.c
index d928b4cda41f3f80bf390e63c4a42eb47495117f..68b14a072e45d9d31795dd4eb7db2a295f21a222 100644 (file)
--- a/python/pyglue.c
+++ b/python/pyglue.c
@@ -89,6 +89,22 @@ static PyObject *py_check_password_quality(PyObject *self, PyObject *args)
        return PyBool_FromLong(check_password_quality(pass));
 }
 
+static PyObject *py_generate_random_bytes(PyObject *self, PyObject *args)
+{
+       int len;
+       PyObject *ret;
+       uint8_t *bytes = NULL;
+
+       if (!PyArg_ParseTuple(args, "i", &len))
+               return NULL;
+
+       bytes = talloc_zero_size(NULL, len);
+       generate_random_buffer(bytes, len);
+       ret = PyBytes_FromStringAndSize((const char *)bytes, len);
+       talloc_free(bytes);
+       return ret;
+}
+
 static PyObject *py_unix2nttime(PyObject *self, PyObject *args)
 {
        time_t t;
@@ -335,6 +351,11 @@ static PyMethodDef py_misc_methods[] = {
                "is the NTVFS file server built in this installation?" },
        { "is_heimdal_built", (PyCFunction)py_is_heimdal_built, METH_NOARGS,
                "is Samba built with Heimdal Kerberbos?" },
+       { "generate_random_bytes",
+               (PyCFunction)py_generate_random_bytes,
+               METH_VARARGS,
+               "generate_random_bytes(len) -> bytes\n"
+               "Generate random bytes with specified length." },
        { NULL }
 };