source4/scripting/python/sidmodule.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include "includes.h"
  21 #include "scripting/python/talloc.h"
  22 #include "Python.h"
  23 #include "libcli/security/security.h"
  24
  25 static PyObject *sid_random(PyObject *self, PyObject *args)
  26 {
  27         char *str;
  28
  29         if (!PyArg_ParseTuple(args, ""))
  30                 return NULL;
  31
  32         str = talloc_asprintf(NULL, "S-1-5-21-%u-%u-%u",
  33                                   (unsigned)generate_random(),
  34                                   (unsigned)generate_random(),
  35                                   (unsigned)generate_random());
  36
  37         if (str == NULL) {
  38                 PyErr_SetString(PyExc_TypeError, "can't generate random sid");
  39                 return NULL;
  40         }
  41
  42         return PyString_FromString(str);
  43 }
  44
  45 static PyMethodDef methods[] = {
  46         { "random", (PyCFunction)sid_random, METH_VARARGS, NULL},
  47         { NULL, NULL }
  48 };
  49
  50 PyDoc_STRVAR(param_doc, "SID helper routines");
  51
  52 PyMODINIT_FUNC initsid(void)
  53 {
  54         PyObject *mod = Py_InitModule3("sid", methods, param_doc);
  55         if (mod == NULL)
  56                 return;
  57 }