source3/param/pyparam_util.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) David Mulder <dmulder@samba.org> 2021
   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 <Python.h>
  21 #include "includes.h"
  22 #include "param/param.h"
  23 #include "param/pyparam.h"
  24 #include "param/loadparm.h"
  25 #include "param/s3_param.h"
  26 #include <pytalloc.h>
  27
  28 #define PyErr_FromString(str) Py_BuildValue("(s)", discard_const_p(char, str))
  29 #define PyLoadparmContext_AsLoadparmContext(obj) pytalloc_get_type(obj, struct loadparm_context)
  30
  31 _PUBLIC_ struct loadparm_context *lpcfg_from_py_object(TALLOC_CTX *mem_ctx, PyObject *py_obj)
  32 {
  33         PyObject *param_mod;
  34         PyTypeObject *lp_type;
  35         bool is_lpobj;
  36         const struct loadparm_s3_helpers *s3_context;
  37         struct loadparm_context *s4_context;
  38
  39         if (py_obj == Py_None) {
  40                 s3_context = loadparm_s3_helpers();
  41
  42                 s4_context = loadparm_init_s3(mem_ctx, s3_context);
  43                 if (s4_context == NULL) {
  44                         PyErr_NoMemory();
  45                         return NULL;
  46                 }
  47
  48                 if (!lpcfg_load_default(s4_context)) {
  49                         PyErr_FromString("Failed to load defaults\n");
  50                         return NULL;
  51                 }
  52
  53                 return s4_context;
  54         }
  55
  56         param_mod = PyImport_ImportModule("samba.param");
  57         if (param_mod == NULL) {
  58                 return NULL;
  59         }
  60
  61         lp_type = (PyTypeObject *)PyObject_GetAttrString(param_mod, "LoadParm");
  62         Py_DECREF(param_mod);
  63         if (lp_type == NULL) {
  64                 PyErr_SetString(PyExc_RuntimeError, "Unable to import LoadParm");
  65                 return NULL;
  66         }
  67
  68         is_lpobj = PyObject_TypeCheck(py_obj, lp_type);
  69         Py_DECREF(lp_type);
  70         if (is_lpobj) {
  71                 return talloc_reference(mem_ctx, PyLoadparmContext_AsLoadparmContext(py_obj));
  72         }
  73
  74         PyErr_SetNone(PyExc_TypeError);
  75         return NULL;
  76 }