winbindd: use add_trusted_domain_from_auth
[bbaumbach/samba-autobuild/.git] / python / pyglue.c
index dbe7eb4ec690a9a0a92db7fbf53b45b8e9025ec6..68b14a072e45d9d31795dd4eb7db2a295f21a222 100644 (file)
@@ -18,6 +18,7 @@
 */
 
 #include <Python.h>
+#include "python/py3compat.h"
 #include "includes.h"
 #include "version.h"
 #include "param/pyparam.h"
@@ -38,7 +39,7 @@ static PyObject *py_generate_random_str(PyObject *self, PyObject *args)
                return NULL;
 
        retstr = generate_random_str(NULL, len);
-       ret = PyString_FromString(retstr);
+       ret = PyStr_FromString(retstr);
        talloc_free(retstr);
        return ret;
 }
@@ -55,11 +56,55 @@ static PyObject *py_generate_random_password(PyObject *self, PyObject *args)
        if (retstr == NULL) {
                return NULL;
        }
-       ret = PyString_FromString(retstr);
+       ret = PyStr_FromString(retstr);
        talloc_free(retstr);
        return ret;
 }
 
+static PyObject *py_generate_random_machine_password(PyObject *self, PyObject *args)
+{
+       int min, max;
+       PyObject *ret;
+       char *retstr;
+       if (!PyArg_ParseTuple(args, "ii", &min, &max))
+               return NULL;
+
+       retstr = generate_random_machine_password(NULL, min, max);
+       if (retstr == NULL) {
+               return NULL;
+       }
+       ret = PyUnicode_FromString(retstr);
+       talloc_free(retstr);
+       return ret;
+}
+
+static PyObject *py_check_password_quality(PyObject *self, PyObject *args)
+{
+       char *pass;
+
+       if (!PyArg_ParseTuple(args, "s", &pass)) {
+               return NULL;
+       }
+
+       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;
@@ -104,7 +149,7 @@ static PyObject *py_nttime2string(PyObject *self, PyObject *args)
        }
 
        string = nt_time_string(tmp_ctx, nt);
-       ret =  PyString_FromString(string);
+       ret =  PyStr_FromString(string);
 
        talloc_free(tmp_ctx);
 
@@ -134,6 +179,15 @@ static PyObject *py_is_ntvfs_fileserver_built(PyObject *self)
 #endif
 }
 
+static PyObject *py_is_heimdal_built(PyObject *self)
+{
+#ifdef SAMBA4_USES_HEIMDAL
+       Py_RETURN_TRUE;
+#else
+       Py_RETURN_FALSE;
+#endif
+}
+
 /*
   return the list of interface IPs we have configured
   takes an loadparm context, returns a list of IPs in string form
@@ -203,7 +257,7 @@ static PyObject *py_interface_ips(PyObject *self, PyObject *args)
                const char *ip = iface_list_n_ip(ifaces, i);
 
                if (all_interfaces) {
-                       PyList_SetItem(pylist, ifcount, PyString_FromString(ip));
+                       PyList_SetItem(pylist, ifcount, PyStr_FromString(ip));
                        ifcount++;
                        continue;
                }
@@ -224,7 +278,7 @@ static PyObject *py_interface_ips(PyObject *self, PyObject *args)
                        continue;
                }
 
-               PyList_SetItem(pylist, ifcount, PyString_FromString(ip));
+               PyList_SetItem(pylist, ifcount, PyStr_FromString(ip));
                ifcount++;
        }
        talloc_free(tmp_ctx);
@@ -252,7 +306,7 @@ static PyObject *py_strstr_m(PyObject *self, PyObject *args)
        if (!ret) {
                Py_RETURN_NONE;
        }
-       return PyString_FromString(ret);
+       return PyStr_FromString(ret);
 }
 
 static PyMethodDef py_misc_methods[] = {
@@ -261,7 +315,20 @@ static PyMethodDef py_misc_methods[] = {
                "Generate random string with specified length." },
        { "generate_random_password", (PyCFunction)py_generate_random_password,
                METH_VARARGS, "generate_random_password(min, max) -> string\n"
-               "Generate random password with a length >= min and <= max." },
+               "Generate random password (based on printable ascii characters) "
+               "with a length >= min and <= max." },
+       { "generate_random_machine_password", (PyCFunction)py_generate_random_machine_password,
+               METH_VARARGS, "generate_random_machine_password(min, max) -> string\n"
+               "Generate random password "
+               "(based on random utf16 characters converted to utf8 or "
+               "random ascii characters if 'unix charset' is not 'utf8')"
+               "with a length >= min (at least 14) and <= max (at most 255)." },
+       { "check_password_quality", (PyCFunction)py_check_password_quality,
+               METH_VARARGS, "check_password_quality(pass) -> bool\n"
+               "Check password quality against Samba's check_password_quality,"
+               "the implementation of Microsoft's rules:"
+               "http://msdn.microsoft.com/en-us/subscriptions/cc786468%28v=ws.10%29.aspx"
+       },
        { "unix2nttime", (PyCFunction)py_unix2nttime, METH_VARARGS,
                "unix2nttime(timestamp) -> nttime" },
        { "nttime2unix", (PyCFunction)py_nttime2unix, METH_VARARGS,
@@ -282,22 +349,36 @@ static PyMethodDef py_misc_methods[] = {
                "(for testing) find one string in another with Samba's strstr_m()"},
        { "is_ntvfs_fileserver_built", (PyCFunction)py_is_ntvfs_fileserver_built, METH_NOARGS,
                "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 }
 };
 
-void init_glue(void)
+static struct PyModuleDef moduledef = {
+    PyModuleDef_HEAD_INIT,
+    .m_name = "_glue",
+    .m_doc = "Python bindings for miscellaneous Samba functions.",
+    .m_size = -1,
+    .m_methods = py_misc_methods,
+};
+
+MODULE_INIT_FUNC(_glue)
 {
        PyObject *m;
 
        debug_setup_talloc_log();
 
-       m = Py_InitModule3("_glue", py_misc_methods, 
-                          "Python bindings for miscellaneous Samba functions.");
+       m = PyModule_Create(&moduledef);
        if (m == NULL)
-               return;
+               return NULL;
 
        PyModule_AddObject(m, "version",
-                                          PyString_FromString(SAMBA_VERSION_STRING));
+                                          PyStr_FromString(SAMBA_VERSION_STRING));
        PyExc_NTSTATUSError = PyErr_NewException(discard_const_p(char, "samba.NTSTATUSError"), PyExc_RuntimeError, NULL);
        if (PyExc_NTSTATUSError != NULL) {
                Py_INCREF(PyExc_NTSTATUSError);
@@ -322,5 +403,6 @@ void init_glue(void)
                PyModule_AddObject(m, "DsExtendedError", PyExc_DsExtendedError);
        }
 
+       return m;
 }