pyglue: Add function to generate a random byte string
authorGary Lockyer <gary@catalyst.net.nz>
Wed, 1 Nov 2017 21:15:29 +0000 (10:15 +1300)
committerAndrew Bartlett <abartlet@samba.org>
Sun, 17 Dec 2017 23:10:16 +0000 (00:10 +0100)
Adds a function to generate a random byte string using the samba random
routines.

Signed-off-by: Gary Lockyer <gary@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
python/pyglue.c
python/samba/__init__.py

index d928b4cda41f3f80bf390e63c4a42eb47495117f..68b14a072e45d9d31795dd4eb7db2a295f21a222 100644 (file)
@@ -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 }
 };
 
index 6ba7c9980747216d9df46b95292affda160956f6..f62f5e3920263a9eb88e791d6fcb9c685c66c4b2 100644 (file)
@@ -389,6 +389,7 @@ unix2nttime = _glue.unix2nttime
 generate_random_password = _glue.generate_random_password
 generate_random_machine_password = _glue.generate_random_machine_password
 check_password_quality = _glue.check_password_quality
+generate_random_bytes = _glue.generate_random_bytes
 strcasecmp_m = _glue.strcasecmp_m
 strstr_m = _glue.strstr_m
 is_ntvfs_fileserver_built = _glue.is_ntvfs_fileserver_built