Line data    Source code

       1             : /* 
       2             :    Unix SMB/CIFS implementation.
       3             :    Samba utility functions
       4             :    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007-2008
       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 "lib/replace/system/python.h"
      21             : #include "py3compat.h"
      22             : #include "includes.h"
      23             : #include "param/param.h"
      24             : #include "param/pyparam.h"
      25             : #include "param/loadparm.h"
      26             : #include <pytalloc.h>
      27             : 
      28             : #define PyLoadparmContext_AsLoadparmContext(obj) pytalloc_get_type(obj, struct loadparm_context)
      29             : 
      30      101956 : _PUBLIC_ struct loadparm_context *lpcfg_from_py_object(TALLOC_CTX *mem_ctx, PyObject *py_obj)
      31             : {
      32        1263 :         struct loadparm_context *lp_ctx;
      33        1263 :         PyObject *param_mod;
      34        1263 :         PyTypeObject *lp_type;
      35        1263 :         bool is_lpobj;
      36             : 
      37      101956 :         if (PyUnicode_Check(py_obj)) {
      38           0 :                 lp_ctx = loadparm_init_global(false);
      39           0 :                 if (lp_ctx == NULL) {
      40           0 :                         return NULL;
      41             :                 }
      42           0 :                 if (!lpcfg_load(lp_ctx, PyUnicode_AsUTF8(py_obj))) {
      43           0 :                         PyErr_Format(PyExc_RuntimeError, "Unable to load %s", 
      44             :                                      PyUnicode_AsUTF8(py_obj));
      45           0 :                         return NULL;
      46             :                 }
      47           0 :                 return lp_ctx;
      48             :         }
      49             : 
      50      101956 :         if (py_obj == Py_None) {
      51       24483 :                 return loadparm_init_global(true);
      52             :         }
      53             : 
      54       77473 :         param_mod = PyImport_ImportModule("samba.param");
      55       77473 :         if (param_mod == NULL) {
      56           0 :                 return NULL;
      57             :         }
      58             : 
      59       77473 :         lp_type = (PyTypeObject *)PyObject_GetAttrString(param_mod, "LoadParm");
      60       57005 :         Py_DECREF(param_mod);
      61       77473 :         if (lp_type == NULL) {
      62           0 :                 PyErr_SetString(PyExc_RuntimeError, "Unable to import LoadParm");
      63           0 :                 return NULL;
      64             :         }
      65             : 
      66       77473 :         is_lpobj = PyObject_TypeCheck(py_obj, lp_type);
      67       57005 :         Py_DECREF(lp_type);
      68       77473 :         if (is_lpobj) {
      69       77473 :                 return talloc_reference(mem_ctx, PyLoadparmContext_AsLoadparmContext(py_obj));
      70             :         }
      71             : 
      72           0 :         PyErr_SetNone(PyExc_TypeError);
      73           0 :         return NULL;
      74             : }
      75             : 
      76        1425 : struct loadparm_context *py_default_loadparm_context(TALLOC_CTX *mem_ctx)
      77             : {
      78        1425 :         return loadparm_init_global(true);
      79             : }
      80             : 
      81             :