Finish smbstatus in Python.

(This used to be commit 988508c2d3269cc88ed38df2fc207a1c0aaccc6b)

diff --git a/source4/lib/messaging/pymessaging.c b/source4/lib/messaging/pymessaging.c
index e05d5eb1f1ac54fae7c2d2a3a39e6253739a561b..1c22fb431a3a531207e840d8b807576d4bb39526 100644 (file)
--- a/source4/lib/messaging/pymessaging.c
+++ b/source4/lib/messaging/pymessaging.c
@@ -530,6 +530,7 @@ static bool irpc_AddNdrRpcMethods(PyTypeObject *ifacetype, const struct PyNdrRpc
 
 void initmessaging(void)
 {
+       extern void initirpc(void);
        PyObject *mod;
 
        if (PyType_Ready(&irpc_ClientConnectionType) < 0)
@@ -548,6 +549,8 @@ void initmessaging(void)
        if (mod == NULL)
                return;
 
+       initirpc();
+
        Py_INCREF((PyObject *)&irpc_ClientConnectionType);
        PyModule_AddObject(mod, "ClientConnection", (PyObject *)&irpc_ClientConnectionType);
 

diff --git a/source4/pidl/lib/Parse/Pidl/Samba4/Python.pm b/source4/pidl/lib/Parse/Pidl/Samba4/Python.pm
index b5ae801ff8276ac7e136792db93014b5210628d3..60d0dafc8aba21a29b10380db99ffc7b99050cb1 100644 (file)
--- a/source4/pidl/lib/Parse/Pidl/Samba4/Python.pm
+++ b/source4/pidl/lib/Parse/Pidl/Samba4/Python.pm
@@ -21,7 +21,7 @@ $VERSION = '0.01';
 sub new($) {
        my ($class) = @_;
        my $self = { res => "", res_hdr => "", tabs => "", constants => {},
-                    module_methods => [], module_objects => [], module_types => [],
+                    module_methods => [], module_objects => [], ready_types => [],
                                 readycode => [] };
        bless($self, $class);
 }
@@ -108,7 +108,7 @@ sub FromUnionToPythonFunction($$$$)
                $self->indent;
 
                if ($e->{NAME}) {
-                       $self->ConvertObjectToPython($mem_ctx, {}, $e, "$name->$e->{NAME}", "ret");
+                       $self->ConvertObjectToPython($mem_ctx, {}, $e, "$name->$e->{NAME}", "ret", "return NULL;");
                } else {
                        $self->pidl("ret = Py_None;");
                }
@@ -182,7 +182,7 @@ sub PythonStruct($$$$$$)
                        $self->indent;
                        $self->pidl("$cname *object = py_talloc_get_ptr(obj);");
                        $self->pidl("PyObject *py_$e->{NAME};");
-                       $self->ConvertObjectToPython("py_talloc_get_mem_ctx(obj)", $env, $e, $varname, "py_$e->{NAME}");
+                       $self->ConvertObjectToPython("py_talloc_get_mem_ctx(obj)", $env, $e, $varname, "py_$e->{NAME}", "return NULL;");
                        $self->pidl("return py_$e->{NAME};");
                        $self->deindent;
                        $self->pidl("}");
@@ -383,7 +383,7 @@ sub PythonFunctionUnpackOut($$$)
                next if ($metadata_args->{out}->{$e->{NAME}});
                my $py_name = "py_$e->{NAME}";
                if (grep(/out/,@{$e->{DIRECTION}})) {
-                       $self->ConvertObjectToPython("r", $env, $e, "r->out.$e->{NAME}", $py_name);
+                       $self->ConvertObjectToPython("r", $env, $e, "r->out.$e->{NAME}", $py_name, "return NULL;");
                        if ($result_size > 1) {
                                $self->pidl("PyTuple_SetItem(result, $i, $py_name);");
                                $i++;
@@ -792,7 +792,13 @@ sub register_module_typeobject($$$)
 
        $self->register_module_object($name, "(PyObject *)$py_name");
 
-       push (@{$self->{module_types}}, [$name, $py_name])
+       $self->check_ready_type($py_name);
+}
+
+sub check_ready_type($$)
+{
+       my ($self, $py_name) = @_;
+       push (@{$self->{ready_types}}, $py_name) unless (grep(/^$py_name$/,@{$self->{ready_types}}));
 }
 
 sub register_module_readycode($$)
@@ -1026,14 +1032,24 @@ sub ConvertObjectToPythonData($$$$$)
        die("unknown type ".mapTypeName($ctype) . ": $cvar");
 }
 
-sub ConvertObjectToPythonLevel($$$$$)
+sub fail_on_null($$$)
 {
-       my ($self, $mem_ctx, $env, $e, $l, $var_name, $py_var) = @_;
+       my ($self, $var, $fail) = @_;
+       $self->pidl("if ($var == NULL) {");
+       $self->indent;
+       $self->pidl($fail);
+       $self->deindent;
+       $self->pidl("}");
+}
+
+sub ConvertObjectToPythonLevel($$$$$$)
+{
+       my ($self, $mem_ctx, $env, $e, $l, $var_name, $py_var, $fail) = @_;
        my $nl = GetNextLevel($e, $l);
 
        if ($l->{TYPE} eq "POINTER") {
                if ($nl->{TYPE} eq "DATA" and Parse::Pidl::Typelist::scalar_is_reference($nl->{DATA_TYPE})) {
-                       $self->ConvertObjectToPythonLevel($var_name, $env, $e, $nl, $var_name, $py_var);
+                       $self->ConvertObjectToPythonLevel($var_name, $env, $e, $nl, $var_name, $py_var, $fail);
                        return;
                }
                if ($l->{POINTER_TYPE} ne "ref") {
@@ -1044,7 +1060,7 @@ sub ConvertObjectToPythonLevel($$$$$)
                        $self->pidl("} else {");
                        $self->indent;
                }
-               $self->ConvertObjectToPythonLevel($var_name, $env, $e, $nl, get_value_of($var_name), $py_var);
+               $self->ConvertObjectToPythonLevel($var_name, $env, $e, $nl, get_value_of($var_name), $py_var, $fail);
                if ($l->{POINTER_TYPE} ne "ref") {
                        $self->deindent;
                        $self->pidl("}");
@@ -1067,6 +1083,7 @@ sub ConvertObjectToPythonLevel($$$$$)
 
                        $length = ParseExpr($length, $env, $e);
                        $self->pidl("$py_var = PyList_New($length);");
+                       $self->fail_on_null($py_var, $fail);
                        $self->pidl("{");
                        $self->indent;
                        my $counter = "$e->{NAME}_cntr_$l->{LEVEL_INDEX}";
@@ -1075,7 +1092,7 @@ sub ConvertObjectToPythonLevel($$$$$)
                        $self->indent;
                        my $member_var = "py_$e->{NAME}_$l->{LEVEL_INDEX}";
                        $self->pidl("PyObject *$member_var;");
-                       $self->ConvertObjectToPythonLevel($var_name, $env, $e, GetNextLevel($e, $l), $var_name."[$counter]", $member_var);
+                       $self->ConvertObjectToPythonLevel($var_name, $env, $e, GetNextLevel($e, $l), $var_name."[$counter]", $member_var, $fail);
                        $self->pidl("PyList_SetItem($py_var, $counter, $member_var);");
                        $self->deindent;
                        $self->pidl("}");
@@ -1086,6 +1103,8 @@ sub ConvertObjectToPythonLevel($$$$$)
                $var_name = get_pointer_to($var_name);
                my $switch = ParseExpr($l->{SWITCH_IS}, $env, $e);
                $self->pidl("$py_var = py_import_" . GetNextLevel($e, $l)->{DATA_TYPE} . "($mem_ctx, $switch, $var_name);");
+               $self->fail_on_null($py_var, $fail);
+
        } elsif ($l->{TYPE} eq "DATA") {
                if (not Parse::Pidl::Typelist::is_scalar($l->{DATA_TYPE})) {
                        $var_name = get_pointer_to($var_name);
@@ -1093,7 +1112,7 @@ sub ConvertObjectToPythonLevel($$$$$)
                my $conv = $self->ConvertObjectToPythonData($mem_ctx, $l->{DATA_TYPE}, $var_name);
                $self->pidl("$py_var = $conv;");
        } elsif ($l->{TYPE} eq "SUBCONTEXT") {
-               $self->ConvertObjectToPythonLevel($mem_ctx, $env, $e, GetNextLevel($e, $l), $var_name, $py_var);
+               $self->ConvertObjectToPythonLevel($mem_ctx, $env, $e, GetNextLevel($e, $l), $var_name, $py_var, $fail);
        } else {
                die("Unknown level type $l->{TYPE} $var_name");
        }
@@ -1101,9 +1120,9 @@ sub ConvertObjectToPythonLevel($$$$$)
 
 sub ConvertObjectToPython($$$$$$)
 {
-       my ($self, $mem_ctx, $env, $ctype, $cvar, $py_var) = @_;
+       my ($self, $mem_ctx, $env, $ctype, $cvar, $py_var, $fail) = @_;
 
-       $self->ConvertObjectToPythonLevel($mem_ctx, $env, $ctype, $ctype->{LEVELS}[0], $cvar, $py_var);
+       $self->ConvertObjectToPythonLevel($mem_ctx, $env, $ctype, $ctype->{LEVELS}[0], $cvar, $py_var, $fail);
 }
 
 sub Parse($$$$$)
@@ -1153,9 +1172,8 @@ sub Parse($$$$$)
        $self->pidl("PyObject *m;");
        $self->pidl("");
 
-       foreach (@{$self->{module_types}}) {
-               my ($object_name, $c_name) = @$_;
-               $self->pidl("if (PyType_Ready($c_name) < 0)");
+       foreach (@{$self->{ready_types}}) {
+               $self->pidl("if (PyType_Ready($_) < 0)");
                $self->pidl("\treturn;");
        }
 

diff --git a/source4/scripting/bin/smbstatus b/source4/scripting/bin/smbstatus
index 6d852c279cd63f2156292310995f08827170f323..7e58ee1269b7cda4e5a83e5a75c3f5b321307eca 100755 (executable)
--- a/source4/scripting/bin/smbstatus
+++ b/source4/scripting/bin/smbstatus
@@ -9,13 +9,13 @@
 #  Released under the GNU GPL version 3 or later
 #
 
-import sys
+import os, sys
 
 sys.path.insert(0, "bin/python")
 
 import optparse
 import samba.getopt as options
-from samba import messaging, irpc
+from samba import irpc, messaging
 
 def show_sessions(conn):
        """show open sessions"""
@@ -35,7 +35,7 @@ def show_tcons(open_connection):
        print "Share                                 Client      Connected at"
        print "-------------------------------------------------------------------------------"
        for tcon in tcons:
-               print "%-30s %16s   %s\n" % (tcon.share_name, tcon.client_ip, sys.httptime(tcon.connect_time))
+               print "%-30s %16s   %s" % (tcon.share_name, tcon.client_ip, sys.httptime(tcon.connect_time))
 
 
 def show_nbt(open_connection):
@@ -43,9 +43,14 @@ def show_nbt(open_connection):
        conn = open_connection("nbt_server")
        stats = conn.nbtd_information(irpc.NBTD_INFO_STATISTICS).next()
        print "NBT server statistics:"
-       for r in stats:
-               print "\t" + r + ":\t" + getattr(stats, r) + "\n"
-       print ""
+       fields = [("total_received", "Total received"),
+                         ("total_sent", "Total sent"),
+                         ("query_count", "Query count"),
+                         ("register_count", "Register count"),
+                         ("release_count", "Release count")]
+       for (field, description) in fields:
+               print "\t%s:\t%s" % (description, getattr(stats, field))
+       print
 
 parser = optparse.OptionParser("%s [options]" % sys.argv[0])
 sambaopts = options.SambaOptions(parser)
@@ -58,10 +63,12 @@ opts, args = parser.parse_args()
 
 lp = sambaopts.get_loadparm()
 
-print "%s\n\n" % lp.get("server string")
+print "%s" % lp.get("server string")
+
+messaging_path = (opts.messaging_path or os.path.join(lp.get("private dir"), "smbd.tmp", "messaging"))
 
 def open_connection(name):
-       return messaging.ClientConnection(name, messaging_path=opts.messaging_path)
+       return messaging.ClientConnection(name, messaging_path=messaging_path)
 
 if opts.nbt:
        show_nbt(open_connection)