smbstatus: always initialize a messaing context
[samba.git] / source3 / utils / status.c
index cbf0dcb2b5a283270256ef021fec5435f31d6f9e..8d84537c4955c483b87a26e7525e14735f11528d 100644 (file)
@@ -2,17 +2,17 @@
    Unix SMB/CIFS implementation.
    status reporting
    Copyright (C) Andrew Tridgell 1994-1998
-   
+
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
-   
+
    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
  */
 
 #include "includes.h"
+#include "system/filesys.h"
+#include "popt_common.h"
+#include "dbwrap/dbwrap.h"
+#include "dbwrap/dbwrap_open.h"
+#include "../libcli/security/security.h"
+#include "session.h"
+#include "locking/proto.h"
+#include "messages.h"
+#include "librpc/gen_ndr/open_files.h"
+#include "smbd/smbd.h"
+#include "librpc/gen_ndr/notify.h"
+#include "lib/conn_tdb.h"
+#include "serverid.h"
+#include "status_profile.h"
+#include "smbd/notifyd/notifyd.h"
 
 #define SMB_MAXPIDS            2048
 static uid_t           Ucrit_uid = 0;               /* added by OH */
@@ -44,12 +59,10 @@ static bool locks_only;            /* Added by RJS */
 static bool processes_only;
 static bool show_brl;
 static bool numeric_only;
+static bool do_checks = true;
 
 const char *username = NULL;
 
-extern bool status_profile_dump(bool be_verbose);
-extern bool status_profile_rates(bool be_verbose);
-
 /* added by OH */
 static void Ucrit_addUid(uid_t uid)
 {
@@ -61,25 +74,26 @@ static unsigned int Ucrit_checkUid(uid_t uid)
 {
        if ( !Ucrit_IsActive ) 
                return 1;
-       
+
        if ( uid == Ucrit_uid ) 
                return 1;
-       
+
        return 0;
 }
 
 static unsigned int Ucrit_checkPid(struct server_id pid)
 {
        int i;
-       
+
        if ( !Ucrit_IsActive ) 
                return 1;
-       
+
        for (i=0;i<Ucrit_MaxPid;i++) {
-               if (cluster_id_equal(&pid, &Ucrit_pid[i])) 
+               if (serverid_equal(&pid, &Ucrit_pid[i])) {
                        return 1;
+               }
        }
-       
+
        return 0;
 }
 
@@ -96,23 +110,20 @@ static bool Ucrit_addPid( struct server_id pid )
        }
 
        Ucrit_pid[Ucrit_MaxPid++] = pid;
-       
+
        return True;
 }
 
-static void print_share_mode(const struct share_mode_entry *e,
-                            const char *sharepath,
-                            const char *fname,
-                            void *dummy)
+static int print_share_mode(const struct share_mode_entry *e,
+                           const char *sharepath,
+                           const char *fname,
+                           const char *sname,
+                           void *dummy)
 {
        static int count;
 
-       if (!is_valid_share_mode_entry(e)) {
-               return;
-       }
-
-       if (!process_exists(e->pid)) {
-               return;
+       if (do_checks && !is_valid_share_mode_entry(e)) {
+               return 0;
        }
 
        if (count==0) {
@@ -122,8 +133,14 @@ static void print_share_mode(const struct share_mode_entry *e,
        }
        count++;
 
+       if (do_checks && !serverid_exists(&e->pid)) {
+               /* the process for this entry does not exist any more */
+               return 0;
+       }
+
        if (Ucrit_checkPid(e->pid)) {
-               d_printf("%-11s  ",procid_str_static(&e->pid));
+               struct server_id_buf tmp;
+               d_printf("%-11s  ", server_id_str_buf(e->pid, &tmp));
                d_printf("%-9u  ", (unsigned int)e->uid);
                switch (map_share_mode_to_deny_mode(e->share_access,
                                                    e->private_options)) {
@@ -161,12 +178,26 @@ static void print_share_mode(const struct share_mode_entry *e,
                        d_printf("BATCH           ");
                } else if (e->op_type & LEVEL_II_OPLOCK) {
                        d_printf("LEVEL_II        ");
+               } else if (e->op_type == LEASE_OPLOCK) {
+                       uint32_t lstate = e->lease->current_state;
+                       d_printf("LEASE(%s%s%s)%s%s%s      ",
+                                (lstate & SMB2_LEASE_READ)?"R":"",
+                                (lstate & SMB2_LEASE_WRITE)?"W":"",
+                                (lstate & SMB2_LEASE_HANDLE)?"H":"",
+                                (lstate & SMB2_LEASE_READ)?"":" ",
+                                (lstate & SMB2_LEASE_WRITE)?"":" ",
+                                (lstate & SMB2_LEASE_HANDLE)?"":" ");
                } else {
                        d_printf("NONE            ");
                }
 
-               d_printf(" %s   %s   %s",sharepath, fname, time_to_asc((time_t)e->time.tv_sec));
+               d_printf(" %s   %s%s   %s",
+                        sharepath, fname,
+                        sname ? sname : "",
+                        time_to_asc((time_t)e->time.tv_sec));
        }
+
+       return 0;
 }
 
 static void print_brl(struct file_id id,
@@ -178,7 +209,7 @@ static void print_brl(struct file_id id,
                        void *private_data)
 {
        static int count;
-       int i;
+       unsigned int i;
        static const struct {
                enum brl_type lock_type;
                const char *desc;
@@ -191,8 +222,9 @@ static void print_brl(struct file_id id,
        };
        const char *desc="X";
        const char *sharepath = "";
-       const char *fname = "";
+       char *fname = NULL;
        struct share_mode_lock *share_mode;
+       struct server_id_buf tmp;
 
        if (count==0) {
                d_printf("Byte range locks:\n");
@@ -201,10 +233,21 @@ static void print_brl(struct file_id id,
        }
        count++;
 
-       share_mode = fetch_share_mode_unlocked(NULL, id, "__unspecified__", "__unspecified__");
+       share_mode = fetch_share_mode_unlocked(NULL, id);
        if (share_mode) {
-               sharepath = share_mode->servicepath;
-               fname = share_mode->filename;
+               bool has_stream = share_mode->data->stream_name != NULL;
+
+               fname = talloc_asprintf(NULL, "%s%s%s",
+                                       share_mode->data->base_name,
+                                       has_stream ? ":" : "",
+                                       has_stream ?
+                                       share_mode->data->stream_name :
+                                       "");
+       } else {
+               fname = talloc_strdup(NULL, "");
+               if (fname == NULL) {
+                       return;
+               }
        }
 
        for (i=0;i<ARRAY_SIZE(lock_types);i++) {
@@ -213,71 +256,100 @@ static void print_brl(struct file_id id,
                }
        }
 
-       d_printf("%-10s %-15s %-4s %-9.0f %-9.0f %-24s %-24s\n", 
-                procid_str_static(&pid), file_id_string_tos(&id),
+       d_printf("%-10s %-15s %-4s %-9jd %-9jd %-24s %-24s\n",
+                server_id_str_buf(pid, &tmp), file_id_string_tos(&id),
                 desc,
-                (double)start, (double)size,
+                (intmax_t)start, (intmax_t)size,
                 sharepath, fname);
 
+       TALLOC_FREE(fname);
        TALLOC_FREE(share_mode);
 }
 
-static int traverse_fn1(struct db_record *rec,
-                       const struct connections_key *key,
-                       const struct connections_data *crec,
-                       void *state)
+static int traverse_connections(const struct connections_key *key,
+                               const struct connections_data *crec,
+                               void *state)
 {
-       if (crec->cnum == -1)
+       struct server_id_buf tmp;
+
+       if (crec->cnum == TID_FIELD_INVALID)
                return 0;
 
-       if (!process_exists(crec->pid) || !Ucrit_checkUid(crec->uid)) {
+       if (do_checks &&
+           (!process_exists(crec->pid) || !Ucrit_checkUid(crec->uid))) {
                return 0;
        }
 
        d_printf("%-10s   %s   %-12s  %s",
-                crec->servicename,procid_str_static(&crec->pid),
+                crec->servicename, server_id_str_buf(crec->pid, &tmp),
                 crec->machine,
                 time_to_asc(crec->start));
 
        return 0;
 }
 
-static int traverse_sessionid(struct db_record *db, void *state)
+static int traverse_sessionid(const char *key, struct sessionid *session,
+                             void *private_data)
 {
-       struct sessionid sessionid;
        fstring uid_str, gid_str;
+       struct server_id_buf tmp;
 
-       if (db->value.dsize != sizeof(sessionid))
+       if (do_checks &&
+           (!process_exists(session->pid) ||
+            !Ucrit_checkUid(session->uid))) {
                return 0;
+       }
 
-       memcpy(&sessionid, db->value.dptr, sizeof(sessionid));
+       Ucrit_addPid(session->pid);
 
-       if (!process_exists(sessionid.pid) || !Ucrit_checkUid(sessionid.uid)) {
-               return 0;
+       fstrcpy(uid_str, "-1");
+
+       if (session->uid != -1) {
+               if (numeric_only) {
+                       fstr_sprintf(uid_str, "%u", (unsigned int)session->uid);
+               } else {
+                       fstrcpy(uid_str, uidtoname(session->uid));
+               }
        }
 
-       Ucrit_addPid( sessionid.pid );
+       fstrcpy(gid_str, "-1");
+
+       if (session->gid != -1) {
+               if (numeric_only) {
+                       fstr_sprintf(gid_str, "%u", (unsigned int)session->gid);
+               } else {
+                       fstrcpy(gid_str, gidtoname(session->gid));
+               }
+       }
 
-       fstr_sprintf(uid_str, "%u", (unsigned int)sessionid.uid);
-       fstr_sprintf(gid_str, "%u", (unsigned int)sessionid.gid);
+       d_printf("%-7s   %-12s  %-12s  %-12s (%s) %-12s\n",
+                server_id_str_buf(session->pid, &tmp),
+                uid_str, gid_str,
+                session->remote_machine, session->hostname, session->protocol_ver);
 
-       d_printf("%-7s   %-12s  %-12s  %-12s (%s)\n",
-                procid_str_static(&sessionid.pid),
-                numeric_only ? uid_str : uidtoname(sessionid.uid),
-                numeric_only ? gid_str : gidtoname(sessionid.gid), 
-                sessionid.remote_machine, sessionid.hostname);
-       
        return 0;
 }
 
 
+static bool print_notify_rec(const char *path, struct server_id server,
+                            const struct notify_instance *instance,
+                            void *private_data)
+{
+       struct server_id_buf idbuf;
+
+       d_printf("%s\\%s\\%x\\%x\n", path, server_id_str_buf(server, &idbuf),
+                (unsigned)instance->filter,
+                (unsigned)instance->subdir_filter);
 
+       return true;
+}
 
- int main(int argc, char *argv[])
+int main(int argc, const char *argv[])
 {
        int c;
        int profile_only = 0;
        bool show_processes, show_locks, show_shares;
+       bool show_notify = false;
        poptContext pc;
        struct poptOption long_options[] = {
                POPT_AUTOHELP
@@ -285,35 +357,45 @@ static int traverse_sessionid(struct db_record *db, void *state)
                {"verbose",     'v', POPT_ARG_NONE,     NULL, 'v', "Be verbose" },
                {"locks",       'L', POPT_ARG_NONE,     NULL, 'L', "Show locks only" },
                {"shares",      'S', POPT_ARG_NONE,     NULL, 'S', "Show shares only" },
+               {"notify",      'N', POPT_ARG_NONE,     NULL, 'N', "Show notifies" },
                {"user",        'u', POPT_ARG_STRING,   &username, 'u', "Switch to user" },
                {"brief",       'b', POPT_ARG_NONE,     NULL, 'b', "Be brief" },
                {"profile",     'P', POPT_ARG_NONE, NULL, 'P', "Do profiling" },
                {"profile-rates", 'R', POPT_ARG_NONE, NULL, 'R', "Show call rates" },
                {"byterange",   'B', POPT_ARG_NONE,     NULL, 'B', "Include byte range locks"},
                {"numeric",     'n', POPT_ARG_NONE,     NULL, 'n', "Numeric uid/gid"},
+               {"fast",        'f', POPT_ARG_NONE,     NULL, 'f', "Skip checks if processes still exist"},
                POPT_COMMON_SAMBA
                POPT_TABLEEND
        };
        TALLOC_CTX *frame = talloc_stackframe();
        int ret = 0;
-       struct messaging_context *msg_ctx;
+       struct messaging_context *msg_ctx = NULL;
+       char *db_path;
+       bool ok;
 
        sec_init();
-       load_case_tables();
+       smb_init_locale();
+
+       setup_logging(argv[0], DEBUG_STDERR);
+       lp_set_cmdline("log level", "0");
 
-       setup_logging(argv[0],True);
-       
-       dbf = x_stderr;
-       
        if (getuid() != geteuid()) {
                d_printf("smbstatus should not be run setuid\n");
                ret = 1;
                goto done;
        }
 
-       pc = poptGetContext(NULL, argc, (const char **) argv, long_options, 
+       if (getuid() != 0) {
+               d_printf("smbstatus only works as root!\n");
+               ret = 1;
+               goto done;
+       }
+
+
+       pc = poptGetContext(NULL, argc, argv, long_options,
                            POPT_CONTEXT_KEEP_FIRST);
-       
+
        while ((c = poptGetNextOpt(pc)) != -1) {
                switch (c) {
                case 'p':
@@ -328,6 +410,9 @@ static int traverse_sessionid(struct db_record *db, void *state)
                case 'S':
                        shares_only = true;
                        break;
+               case 'N':
+                       show_notify = true;
+                       break;
                case 'b':
                        brief = true;
                        break;
@@ -344,6 +429,9 @@ static int traverse_sessionid(struct db_record *db, void *state)
                case 'n':
                        numeric_only = true;
                        break;
+               case 'f':
+                       do_checks = false;
+                       break;
                }
        }
 
@@ -367,21 +455,19 @@ static int traverse_sessionid(struct db_record *db, void *state)
                goto done;
        }
 
+
        /*
-        * This implicitly initializes the global ctdbd connection, usable by
-        * the db_open() calls further down.
+        * This implicitly initializes the global ctdbd connection,
+        * usable by the db_open() calls further down.
         */
-
-       msg_ctx = messaging_init(NULL, procid_self(),
-                                event_context_init(NULL));
-
+       msg_ctx = messaging_init(NULL, samba_tevent_context_init(NULL));
        if (msg_ctx == NULL) {
                fprintf(stderr, "messaging_init failed\n");
                ret = -1;
                goto done;
        }
 
-       if (!lp_load(get_dyn_CONFIGFILE(),False,False,False,True)) {
+       if (!lp_load_global(get_dyn_CONFIGFILE())) {
                fprintf(stderr, "Can't load %s - run testparm to debug it\n",
                        get_dyn_CONFIGFILE());
                ret = -1;
@@ -391,51 +477,48 @@ static int traverse_sessionid(struct db_record *db, void *state)
        switch (profile_only) {
                case 'P':
                        /* Dump profile data */
-                       return status_profile_dump(verbose);
+                       ok = status_profile_dump(verbose);
+                       return ok ? 0 : 1;
                case 'R':
                        /* Continuously display rate-converted data */
-                       return status_profile_rates(verbose);
+                       ok = status_profile_rates(verbose);
+                       return ok ? 0 : 1;
                default:
                        break;
        }
 
        if ( show_processes ) {
-               struct db_context *db;
-               db = db_open(NULL, lock_path("sessionid.tdb"), 0,
-                            TDB_CLEAR_IF_FIRST, O_RDONLY, 0644);
-               if (!db) {
-                       d_printf("sessionid.tdb not initialised\n");
-               } else {
-                       d_printf("\nSamba version %s\n",samba_version_string());
-                       d_printf("PID     Username      Group         Machine                        \n");
-                       d_printf("-------------------------------------------------------------------\n");
-                       if (lp_security() == SEC_SHARE) {
-                               d_printf(" <processes do not show up in "
-                                   "anonymous mode>\n");
-                       }
+               d_printf("\nSamba version %s\n",samba_version_string());
+               d_printf("PID     Username      Group         Machine            Protocol Version       \n");
+               d_printf("------------------------------------------------------------------------------\n");
 
-                       db->traverse_read(db, traverse_sessionid, NULL);
-                       TALLOC_FREE(db);
-               }
+               sessionid_traverse_read(traverse_sessionid, NULL);
 
                if (processes_only) {
                        goto done;
                }
        }
-  
+
        if ( show_shares ) {
                if (verbose) {
-                       d_printf("Opened %s\n", lock_path("connections.tdb"));
+                       db_path = lock_path("connections.tdb");
+                       if (db_path == NULL) {
+                               d_printf("Out of memory - exiting\n");
+                               ret = -1;
+                               goto done;
+                       }
+                       d_printf("Opened %s\n", db_path);
+                       TALLOC_FREE(db_path);
                }
 
                if (brief) {
                        goto done;
                }
-               
+
                d_printf("\nService      pid     machine       Connected at\n");
                d_printf("-------------------------------------------------------\n");
-       
-               connections_forall(traverse_fn1, NULL);
+
+               connections_forall_read(traverse_connections, NULL);
 
                d_printf("\n");
 
@@ -447,17 +530,27 @@ static int traverse_sessionid(struct db_record *db, void *state)
        if ( show_locks ) {
                int result;
                struct db_context *db;
-               db = db_open(NULL, lock_path("locking.tdb"), 0,
-                            TDB_CLEAR_IF_FIRST, O_RDONLY, 0);
+
+               db_path = lock_path("locking.tdb");
+               if (db_path == NULL) {
+                       d_printf("Out of memory - exiting\n");
+                       ret = -1;
+                       goto done;
+               }
+
+               db = db_open(NULL, db_path, 0,
+                            TDB_CLEAR_IF_FIRST|TDB_INCOMPATIBLE_HASH, O_RDONLY, 0,
+                            DBWRAP_LOCK_ORDER_1, DBWRAP_FLAG_NONE);
 
                if (!db) {
-                       d_printf("%s not initialised\n",
-                                lock_path("locking.tdb"));
+                       d_printf("%s not initialised\n", db_path);
                        d_printf("This is normal if an SMB client has never "
                                 "connected to your server.\n");
+                       TALLOC_FREE(db_path);
                        exit(0);
                } else {
                        TALLOC_FREE(db);
+                       TALLOC_FREE(db_path);
                }
 
                if (!locking_init_readonly()) {
@@ -465,24 +558,36 @@ static int traverse_sessionid(struct db_record *db, void *state)
                        ret = 1;
                        goto done;
                }
-               
-               result = share_mode_forall(print_share_mode, NULL);
+
+               result = share_entry_forall(print_share_mode, NULL);
 
                if (result == 0) {
                        d_printf("No locked files\n");
-               } else if (result == -1) {
+               } else if (result < 0) {
                        d_printf("locked file list truncated\n");
                }
-               
+
                d_printf("\n");
 
                if (show_brl) {
                        brl_forall(print_brl, NULL);
                }
-               
+
                locking_end();
        }
 
+       if (show_notify) {
+               struct notify_context *n;
+
+               n = notify_init(talloc_tos(), msg_ctx,
+                               messaging_tevent_context(msg_ctx));
+               if (n == NULL) {
+                       goto done;
+               }
+               notify_walk(n, print_notify_rec, NULL);
+               TALLOC_FREE(n);
+       }
+
 done:
        TALLOC_FREE(frame);
        return ret;