source3/utils/status.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    status reporting
   4    Copyright (C) Andrew Tridgell 1994-1998
   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    Revision History:
  20
  21    12 aug 96: Erik.Devriendt@te6.siemens.be
  22    added support for shared memory implementation of share mode locking
  23
  24    21-Jul-1998: rsharpe@ns.aus.com (Richard Sharpe)
  25    Added -L (locks only) -S (shares only) flags and code
  26
  27 */
  28
  29 /*
  30  * This program reports current SMB connections
  31  */
  32
  33 #include "includes.h"
  34 #include "system/filesys.h"
  35 #include "popt_common.h"
  36 #include "dbwrap/dbwrap.h"
  37 #include "dbwrap/dbwrap_open.h"
  38 #include "../libcli/security/security.h"
  39 #include "session.h"
  40 #include "locking/proto.h"
  41 #include "messages.h"
  42 #include "librpc/gen_ndr/open_files.h"
  43 #include "smbd/smbd.h"
  44 #include "librpc/gen_ndr/notify.h"
  45 #include "lib/conn_tdb.h"
  46 #include "serverid.h"
  47 #include "status_profile.h"
  48 #include "smbd/notifyd/notifyd.h"
  49
  50 #define SMB_MAXPIDS             2048
  51 static uid_t            Ucrit_uid = 0;               /* added by OH */
  52 static struct server_id Ucrit_pid[SMB_MAXPIDS];  /* Ugly !!! */   /* added by OH */
  53 static int              Ucrit_MaxPid=0;                    /* added by OH */
  54 static unsigned int     Ucrit_IsActive = 0;                /* added by OH */
  55
  56 static bool verbose, brief;
  57 static bool shares_only;            /* Added by RJS */
  58 static bool locks_only;            /* Added by RJS */
  59 static bool processes_only;
  60 static bool show_brl;
  61 static bool numeric_only;
  62 static bool do_checks = true;
  63
  64 const char *username = NULL;
  65
  66 /* added by OH */
  67 static void Ucrit_addUid(uid_t uid)
  68 {
  69         Ucrit_uid = uid;
  70         Ucrit_IsActive = 1;
  71 }
  72
  73 static unsigned int Ucrit_checkUid(uid_t uid)
  74 {
  75         if ( !Ucrit_IsActive )
  76                 return 1;
  77
  78         if ( uid == Ucrit_uid )
  79                 return 1;
  80
  81         return 0;
  82 }
  83
  84 static unsigned int Ucrit_checkPid(struct server_id pid)
  85 {
  86         int i;
  87
  88         if ( !Ucrit_IsActive )
  89                 return 1;
  90
  91         for (i=0;i<Ucrit_MaxPid;i++) {
  92                 if (serverid_equal(&pid, &Ucrit_pid[i])) {
  93                         return 1;
  94                 }
  95         }
  96
  97         return 0;
  98 }
  99
 100 static bool Ucrit_addPid( struct server_id pid )
 101 {
 102         if ( !Ucrit_IsActive )
 103                 return True;
 104
 105         if ( Ucrit_MaxPid >= SMB_MAXPIDS ) {
 106                 d_printf("ERROR: More than %d pids for user %s!\n",
 107                          SMB_MAXPIDS, uidtoname(Ucrit_uid));
 108
 109                 return False;
 110         }
 111
 112         Ucrit_pid[Ucrit_MaxPid++] = pid;
 113
 114         return True;
 115 }
 116
 117 static int print_share_mode(const struct share_mode_entry *e,
 118                             const char *sharepath,
 119                             const char *fname,
 120                             void *dummy)
 121 {
 122         static int count;
 123
 124         if (do_checks && !is_valid_share_mode_entry(e)) {
 125                 return 0;
 126         }
 127
 128         if (count==0) {
 129                 d_printf("Locked files:\n");
 130                 d_printf("Pid          Uid        DenyMode   Access      R/W        Oplock           SharePath   Name   Time\n");
 131                 d_printf("--------------------------------------------------------------------------------------------------\n");
 132         }
 133         count++;
 134
 135         if (do_checks && !serverid_exists(&e->pid)) {
 136                 /* the process for this entry does not exist any more */
 137                 return 0;
 138         }
 139
 140         if (Ucrit_checkPid(e->pid)) {
 141                 struct server_id_buf tmp;
 142                 d_printf("%-11s  ", server_id_str_buf(e->pid, &tmp));
 143                 d_printf("%-9u  ", (unsigned int)e->uid);
 144                 switch (map_share_mode_to_deny_mode(e->share_access,
 145                                                     e->private_options)) {
 146                         case DENY_NONE: d_printf("DENY_NONE  "); break;
 147                         case DENY_ALL:  d_printf("DENY_ALL   "); break;
 148                         case DENY_DOS:  d_printf("DENY_DOS   "); break;
 149                         case DENY_READ: d_printf("DENY_READ  "); break;
 150                         case DENY_WRITE:printf("DENY_WRITE "); break;
 151                         case DENY_FCB:  d_printf("DENY_FCB "); break;
 152                         default: {
 153                                 d_printf("unknown-please report ! "
 154                                          "e->share_access = 0x%x, "
 155                                          "e->private_options = 0x%x\n",
 156                                          (unsigned int)e->share_access,
 157                                          (unsigned int)e->private_options );
 158                                 break;
 159                         }
 160                 }
 161                 d_printf("0x%-8x  ",(unsigned int)e->access_mask);
 162                 if ((e->access_mask & (FILE_READ_DATA|FILE_WRITE_DATA))==
 163                                 (FILE_READ_DATA|FILE_WRITE_DATA)) {
 164                         d_printf("RDWR       ");
 165                 } else if (e->access_mask & FILE_WRITE_DATA) {
 166                         d_printf("WRONLY     ");
 167                 } else {
 168                         d_printf("RDONLY     ");
 169                 }
 170
 171                 if((e->op_type & (EXCLUSIVE_OPLOCK|BATCH_OPLOCK)) ==
 172                                         (EXCLUSIVE_OPLOCK|BATCH_OPLOCK)) {
 173                         d_printf("EXCLUSIVE+BATCH ");
 174                 } else if (e->op_type & EXCLUSIVE_OPLOCK) {
 175                         d_printf("EXCLUSIVE       ");
 176                 } else if (e->op_type & BATCH_OPLOCK) {
 177                         d_printf("BATCH           ");
 178                 } else if (e->op_type & LEVEL_II_OPLOCK) {
 179                         d_printf("LEVEL_II        ");
 180                 } else if (e->op_type == LEASE_OPLOCK) {
 181                         uint32_t lstate = e->lease->current_state;
 182                         d_printf("LEASE(%s%s%s)%s%s%s      ",
 183                                  (lstate & SMB2_LEASE_READ)?"R":"",
 184                                  (lstate & SMB2_LEASE_WRITE)?"W":"",
 185                                  (lstate & SMB2_LEASE_HANDLE)?"H":"",
 186                                  (lstate & SMB2_LEASE_READ)?"":" ",
 187                                  (lstate & SMB2_LEASE_WRITE)?"":" ",
 188                                  (lstate & SMB2_LEASE_HANDLE)?"":" ");
 189                 } else {
 190                         d_printf("NONE            ");
 191                 }
 192
 193                 d_printf(" %s   %s   %s",sharepath, fname, time_to_asc((time_t)e->time.tv_sec));
 194         }
 195
 196         return 0;
 197 }
 198
 199 static void print_brl(struct file_id id,
 200                         struct server_id pid,
 201                         enum brl_type lock_type,
 202                         enum brl_flavour lock_flav,
 203                         br_off start,
 204                         br_off size,
 205                         void *private_data)
 206 {
 207         static int count;
 208         unsigned int i;
 209         static const struct {
 210                 enum brl_type lock_type;
 211                 const char *desc;
 212         } lock_types[] = {
 213                 { READ_LOCK, "R" },
 214                 { WRITE_LOCK, "W" },
 215                 { PENDING_READ_LOCK, "PR" },
 216                 { PENDING_WRITE_LOCK, "PW" },
 217                 { UNLOCK_LOCK, "U" }
 218         };
 219         const char *desc="X";
 220         const char *sharepath = "";
 221         char *fname = NULL;
 222         struct share_mode_lock *share_mode;
 223         struct server_id_buf tmp;
 224
 225         if (count==0) {
 226                 d_printf("Byte range locks:\n");
 227                 d_printf("Pid        dev:inode       R/W  start     size      SharePath               Name\n");
 228                 d_printf("--------------------------------------------------------------------------------\n");
 229         }
 230         count++;
 231
 232         share_mode = fetch_share_mode_unlocked(NULL, id);
 233         if (share_mode) {
 234                 bool has_stream = share_mode->data->stream_name != NULL;
 235
 236                 fname = talloc_asprintf(NULL, "%s%s%s",
 237                                         share_mode->data->base_name,
 238                                         has_stream ? ":" : "",
 239                                         has_stream ?
 240                                         share_mode->data->stream_name :
 241                                         "");
 242         } else {
 243                 fname = talloc_strdup(NULL, "");
 244                 if (fname == NULL) {
 245                         return;
 246                 }
 247         }
 248
 249         for (i=0;i<ARRAY_SIZE(lock_types);i++) {
 250                 if (lock_type == lock_types[i].lock_type) {
 251                         desc = lock_types[i].desc;
 252                 }
 253         }
 254
 255         d_printf("%-10s %-15s %-4s %-9jd %-9jd %-24s %-24s\n",
 256                  server_id_str_buf(pid, &tmp), file_id_string_tos(&id),
 257                  desc,
 258                  (intmax_t)start, (intmax_t)size,
 259                  sharepath, fname);
 260
 261         TALLOC_FREE(fname);
 262         TALLOC_FREE(share_mode);
 263 }
 264
 265 static int traverse_connections(const struct connections_key *key,
 266                                 const struct connections_data *crec,
 267                                 void *state)
 268 {
 269         struct server_id_buf tmp;
 270
 271         if (crec->cnum == TID_FIELD_INVALID)
 272                 return 0;
 273
 274         if (do_checks &&
 275             (!process_exists(crec->pid) || !Ucrit_checkUid(crec->uid))) {
 276                 return 0;
 277         }
 278
 279         d_printf("%-10s   %s   %-12s  %s",
 280                  crec->servicename, server_id_str_buf(crec->pid, &tmp),
 281                  crec->machine,
 282                  time_to_asc(crec->start));
 283
 284         return 0;
 285 }
 286
 287 static int traverse_sessionid(const char *key, struct sessionid *session,
 288                               void *private_data)
 289 {
 290         fstring uid_str, gid_str;
 291         struct server_id_buf tmp;
 292
 293         if (do_checks &&
 294             (!process_exists(session->pid) ||
 295              !Ucrit_checkUid(session->uid))) {
 296                 return 0;
 297         }
 298
 299         Ucrit_addPid(session->pid);
 300
 301         fstrcpy(uid_str, "-1");
 302
 303         if (session->uid != -1) {
 304                 if (numeric_only) {
 305                         fstr_sprintf(uid_str, "%u", (unsigned int)session->uid);
 306                 } else {
 307                         fstrcpy(uid_str, uidtoname(session->uid));
 308                 }
 309         }
 310
 311         fstrcpy(gid_str, "-1");
 312
 313         if (session->gid != -1) {
 314                 if (numeric_only) {
 315                         fstr_sprintf(gid_str, "%u", (unsigned int)session->gid);
 316                 } else {
 317                         fstrcpy(gid_str, gidtoname(session->gid));
 318                 }
 319         }
 320
 321         d_printf("%-7s   %-12s  %-12s  %-12s (%s) %-12s\n",
 322                  server_id_str_buf(session->pid, &tmp),
 323                  uid_str, gid_str,
 324                  session->remote_machine, session->hostname, session->protocol_ver);
 325
 326         return 0;
 327 }
 328
 329
 330 static bool print_notify_rec(const char *path, struct server_id server,
 331                              const struct notify_instance *instance,
 332                              void *private_data)
 333 {
 334         struct server_id_buf idbuf;
 335
 336         d_printf("%s\\%s\\%x\\%x\n", path, server_id_str_buf(server, &idbuf),
 337                  (unsigned)instance->filter,
 338                  (unsigned)instance->subdir_filter);
 339
 340         return true;
 341 }
 342
 343 int main(int argc, const char *argv[])
 344 {
 345         int c;
 346         int profile_only = 0;
 347         bool show_processes, show_locks, show_shares;
 348         bool show_notify = false;
 349         poptContext pc;
 350         struct poptOption long_options[] = {
 351                 POPT_AUTOHELP
 352                 {"processes",   'p', POPT_ARG_NONE,     NULL, 'p', "Show processes only" },
 353                 {"verbose",     'v', POPT_ARG_NONE,     NULL, 'v', "Be verbose" },
 354                 {"locks",       'L', POPT_ARG_NONE,     NULL, 'L', "Show locks only" },
 355                 {"shares",      'S', POPT_ARG_NONE,     NULL, 'S', "Show shares only" },
 356                 {"notify",      'N', POPT_ARG_NONE,     NULL, 'N', "Show notifies" },
 357                 {"user",        'u', POPT_ARG_STRING,   &username, 'u', "Switch to user" },
 358                 {"brief",       'b', POPT_ARG_NONE,     NULL, 'b', "Be brief" },
 359                 {"profile",     'P', POPT_ARG_NONE, NULL, 'P', "Do profiling" },
 360                 {"profile-rates", 'R', POPT_ARG_NONE, NULL, 'R', "Show call rates" },
 361                 {"byterange",   'B', POPT_ARG_NONE,     NULL, 'B', "Include byte range locks"},
 362                 {"numeric",     'n', POPT_ARG_NONE,     NULL, 'n', "Numeric uid/gid"},
 363                 {"fast",        'f', POPT_ARG_NONE,     NULL, 'f', "Skip checks if processes still exist"},
 364                 POPT_COMMON_SAMBA
 365                 POPT_TABLEEND
 366         };
 367         TALLOC_CTX *frame = talloc_stackframe();
 368         int ret = 0;
 369         struct messaging_context *msg_ctx = NULL;
 370         char *db_path;
 371         bool ok;
 372
 373         sec_init();
 374         smb_init_locale();
 375
 376         setup_logging(argv[0], DEBUG_STDERR);
 377         lp_set_cmdline("log level", "0");
 378
 379         if (getuid() != geteuid()) {
 380                 d_printf("smbstatus should not be run setuid\n");
 381                 ret = 1;
 382                 goto done;
 383         }
 384
 385         if (getuid() != 0) {
 386                 d_printf("smbstatus only works as root!\n");
 387                 ret = 1;
 388                 goto done;
 389         }
 390
 391
 392         pc = poptGetContext(NULL, argc, argv, long_options,
 393                             POPT_CONTEXT_KEEP_FIRST);
 394
 395         while ((c = poptGetNextOpt(pc)) != -1) {
 396                 switch (c) {
 397                 case 'p':
 398                         processes_only = true;
 399                         break;
 400                 case 'v':
 401                         verbose = true;
 402                         break;
 403                 case 'L':
 404                         locks_only = true;
 405                         break;
 406                 case 'S':
 407                         shares_only = true;
 408                         break;
 409                 case 'N':
 410                         show_notify = true;
 411                         break;
 412                 case 'b':
 413                         brief = true;
 414                         break;
 415                 case 'u':
 416                         Ucrit_addUid(nametouid(poptGetOptArg(pc)));
 417                         break;
 418                 case 'P':
 419                 case 'R':
 420                         profile_only = c;
 421                         break;
 422                 case 'B':
 423                         show_brl = true;
 424                         break;
 425                 case 'n':
 426                         numeric_only = true;
 427                         break;
 428                 case 'f':
 429                         do_checks = false;
 430                         break;
 431                 }
 432         }
 433
 434         /* setup the flags based on the possible combincations */
 435
 436         show_processes = !(shares_only || locks_only || profile_only) || processes_only;
 437         show_locks     = !(shares_only || processes_only || profile_only) || locks_only;
 438         show_shares    = !(processes_only || locks_only || profile_only) || shares_only;
 439
 440         if ( username )
 441                 Ucrit_addUid( nametouid(username) );
 442
 443         if (verbose) {
 444                 d_printf("using configfile = %s\n", get_dyn_CONFIGFILE());
 445         }
 446
 447         if (!lp_load_initial_only(get_dyn_CONFIGFILE())) {
 448                 fprintf(stderr, "Can't load %s - run testparm to debug it\n",
 449                         get_dyn_CONFIGFILE());
 450                 ret = -1;
 451                 goto done;
 452         }
 453
 454
 455         if (lp_clustering()) {
 456                 /*
 457                  * This implicitly initializes the global ctdbd
 458                  * connection, usable by the db_open() calls further
 459                  * down.
 460                  */
 461                 msg_ctx = messaging_init(NULL, samba_tevent_context_init(NULL));
 462                 if (msg_ctx == NULL) {
 463                         fprintf(stderr, "messaging_init failed\n");
 464                         ret = -1;
 465                         goto done;
 466                 }
 467         }
 468
 469         if (!lp_load_global(get_dyn_CONFIGFILE())) {
 470                 fprintf(stderr, "Can't load %s - run testparm to debug it\n",
 471                         get_dyn_CONFIGFILE());
 472                 ret = -1;
 473                 goto done;
 474         }
 475
 476         switch (profile_only) {
 477                 case 'P':
 478                         /* Dump profile data */
 479                         ok = status_profile_dump(verbose);
 480                         return ok ? 0 : 1;
 481                 case 'R':
 482                         /* Continuously display rate-converted data */
 483                         ok = status_profile_rates(verbose);
 484                         return ok ? 0 : 1;
 485                 default:
 486                         break;
 487         }
 488
 489         if ( show_processes ) {
 490                 d_printf("\nSamba version %s\n",samba_version_string());
 491                 d_printf("PID     Username      Group         Machine            Protocol Version       \n");
 492                 d_printf("------------------------------------------------------------------------------\n");
 493
 494                 sessionid_traverse_read(traverse_sessionid, NULL);
 495
 496                 if (processes_only) {
 497                         goto done;
 498                 }
 499         }
 500
 501         if ( show_shares ) {
 502                 if (verbose) {
 503                         db_path = lock_path("connections.tdb");
 504                         if (db_path == NULL) {
 505                                 d_printf("Out of memory - exiting\n");
 506                                 ret = -1;
 507                                 goto done;
 508                         }
 509                         d_printf("Opened %s\n", db_path);
 510                         TALLOC_FREE(db_path);
 511                 }
 512
 513                 if (brief) {
 514                         goto done;
 515                 }
 516
 517                 d_printf("\nService      pid     machine       Connected at\n");
 518                 d_printf("-------------------------------------------------------\n");
 519
 520                 connections_forall_read(traverse_connections, NULL);
 521
 522                 d_printf("\n");
 523
 524                 if ( shares_only ) {
 525                         goto done;
 526                 }
 527         }
 528
 529         if ( show_locks ) {
 530                 int result;
 531                 struct db_context *db;
 532
 533                 db_path = lock_path("locking.tdb");
 534                 if (db_path == NULL) {
 535                         d_printf("Out of memory - exiting\n");
 536                         ret = -1;
 537                         goto done;
 538                 }
 539
 540                 db = db_open(NULL, db_path, 0,
 541                              TDB_CLEAR_IF_FIRST|TDB_INCOMPATIBLE_HASH, O_RDONLY, 0,
 542                              DBWRAP_LOCK_ORDER_1, DBWRAP_FLAG_NONE);
 543
 544                 if (!db) {
 545                         d_printf("%s not initialised\n", db_path);
 546                         d_printf("This is normal if an SMB client has never "
 547                                  "connected to your server.\n");
 548                         TALLOC_FREE(db_path);
 549                         exit(0);
 550                 } else {
 551                         TALLOC_FREE(db);
 552                         TALLOC_FREE(db_path);
 553                 }
 554
 555                 if (!locking_init_readonly()) {
 556                         d_printf("Can't initialise locking module - exiting\n");
 557                         ret = 1;
 558                         goto done;
 559                 }
 560
 561                 result = share_entry_forall(print_share_mode, NULL);
 562
 563                 if (result == 0) {
 564                         d_printf("No locked files\n");
 565                 } else if (result < 0) {
 566                         d_printf("locked file list truncated\n");
 567                 }
 568
 569                 d_printf("\n");
 570
 571                 if (show_brl) {
 572                         brl_forall(print_brl, NULL);
 573                 }
 574
 575                 locking_end();
 576         }
 577
 578         if (show_notify) {
 579                 struct notify_context *n;
 580
 581                 if (msg_ctx == NULL) {
 582                         msg_ctx = messaging_init(
 583                                 NULL, samba_tevent_context_init(NULL));
 584                         if (msg_ctx == NULL) {
 585                                 fprintf(stderr, "messaging_init failed\n");
 586                                 ret = -1;
 587                                 goto done;
 588                         }
 589                 }
 590
 591                 n = notify_init(talloc_tos(), msg_ctx,
 592                                 messaging_tevent_context(msg_ctx));
 593                 if (n == NULL) {
 594                         goto done;
 595                 }
 596                 notify_walk(n, print_notify_rec, NULL);
 597                 TALLOC_FREE(n);
 598         }
 599
 600 done:
 601         TALLOC_FREE(frame);
 602         return ret;
 603 }