talloc: avoid a function call in TALLOC_FREE() if possible.
[Samba.git] / source3 / utils / status.c
blob4e823da76e04f0849b066b8f0d7b6679a144e272
1 /*
2 Unix SMB/CIFS implementation.
3 status reporting
4 Copyright (C) Andrew Tridgell 1994-1998
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.
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.
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/>.
19 Revision History:
21 12 aug 96: Erik.Devriendt@te6.siemens.be
22 added support for shared memory implementation of share mode locking
24 21-Jul-1998: rsharpe@ns.aus.com (Richard Sharpe)
25 Added -L (locks only) -S (shares only) flags and code
30 * This program reports current SMB connections
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"
47 #define SMB_MAXPIDS 2048
48 static uid_t Ucrit_uid = 0; /* added by OH */
49 static struct server_id Ucrit_pid[SMB_MAXPIDS]; /* Ugly !!! */ /* added by OH */
50 static int Ucrit_MaxPid=0; /* added by OH */
51 static unsigned int Ucrit_IsActive = 0; /* added by OH */
53 static bool verbose, brief;
54 static bool shares_only; /* Added by RJS */
55 static bool locks_only; /* Added by RJS */
56 static bool processes_only;
57 static bool show_brl;
58 static bool numeric_only;
59 static bool do_checks = true;
61 const char *username = NULL;
63 extern bool status_profile_dump(bool be_verbose);
64 extern bool status_profile_rates(bool be_verbose);
66 /* added by OH */
67 static void Ucrit_addUid(uid_t uid)
69 Ucrit_uid = uid;
70 Ucrit_IsActive = 1;
73 static unsigned int Ucrit_checkUid(uid_t uid)
75 if ( !Ucrit_IsActive )
76 return 1;
78 if ( uid == Ucrit_uid )
79 return 1;
81 return 0;
84 static unsigned int Ucrit_checkPid(struct server_id pid)
86 int i;
88 if ( !Ucrit_IsActive )
89 return 1;
91 for (i=0;i<Ucrit_MaxPid;i++) {
92 if (serverid_equal(&pid, &Ucrit_pid[i])) {
93 return 1;
97 return 0;
100 static bool Ucrit_addPid( struct server_id pid )
102 if ( !Ucrit_IsActive )
103 return True;
105 if ( Ucrit_MaxPid >= SMB_MAXPIDS ) {
106 d_printf("ERROR: More than %d pids for user %s!\n",
107 SMB_MAXPIDS, uidtoname(Ucrit_uid));
109 return False;
112 Ucrit_pid[Ucrit_MaxPid++] = pid;
114 return True;
117 static void print_share_mode(const struct share_mode_entry *e,
118 const char *sharepath,
119 const char *fname,
120 void *dummy)
122 static int count;
124 if (do_checks && !is_valid_share_mode_entry(e)) {
125 return;
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");
133 count++;
135 if (Ucrit_checkPid(e->pid)) {
136 d_printf("%-11s ",procid_str_static(&e->pid));
137 d_printf("%-9u ", (unsigned int)e->uid);
138 switch (map_share_mode_to_deny_mode(e->share_access,
139 e->private_options)) {
140 case DENY_NONE: d_printf("DENY_NONE "); break;
141 case DENY_ALL: d_printf("DENY_ALL "); break;
142 case DENY_DOS: d_printf("DENY_DOS "); break;
143 case DENY_READ: d_printf("DENY_READ "); break;
144 case DENY_WRITE:printf("DENY_WRITE "); break;
145 case DENY_FCB: d_printf("DENY_FCB "); break;
146 default: {
147 d_printf("unknown-please report ! "
148 "e->share_access = 0x%x, "
149 "e->private_options = 0x%x\n",
150 (unsigned int)e->share_access,
151 (unsigned int)e->private_options );
152 break;
155 d_printf("0x%-8x ",(unsigned int)e->access_mask);
156 if ((e->access_mask & (FILE_READ_DATA|FILE_WRITE_DATA))==
157 (FILE_READ_DATA|FILE_WRITE_DATA)) {
158 d_printf("RDWR ");
159 } else if (e->access_mask & FILE_WRITE_DATA) {
160 d_printf("WRONLY ");
161 } else {
162 d_printf("RDONLY ");
165 if((e->op_type & (EXCLUSIVE_OPLOCK|BATCH_OPLOCK)) ==
166 (EXCLUSIVE_OPLOCK|BATCH_OPLOCK)) {
167 d_printf("EXCLUSIVE+BATCH ");
168 } else if (e->op_type & EXCLUSIVE_OPLOCK) {
169 d_printf("EXCLUSIVE ");
170 } else if (e->op_type & BATCH_OPLOCK) {
171 d_printf("BATCH ");
172 } else if (e->op_type & LEVEL_II_OPLOCK) {
173 d_printf("LEVEL_II ");
174 } else {
175 d_printf("NONE ");
178 d_printf(" %s %s %s",sharepath, fname, time_to_asc((time_t)e->time.tv_sec));
182 static void print_brl(struct file_id id,
183 struct server_id pid,
184 enum brl_type lock_type,
185 enum brl_flavour lock_flav,
186 br_off start,
187 br_off size,
188 void *private_data)
190 static int count;
191 unsigned int i;
192 static const struct {
193 enum brl_type lock_type;
194 const char *desc;
195 } lock_types[] = {
196 { READ_LOCK, "R" },
197 { WRITE_LOCK, "W" },
198 { PENDING_READ_LOCK, "PR" },
199 { PENDING_WRITE_LOCK, "PW" },
200 { UNLOCK_LOCK, "U" }
202 const char *desc="X";
203 const char *sharepath = "";
204 char *fname = NULL;
205 struct share_mode_lock *share_mode;
207 if (count==0) {
208 d_printf("Byte range locks:\n");
209 d_printf("Pid dev:inode R/W start size SharePath Name\n");
210 d_printf("--------------------------------------------------------------------------------\n");
212 count++;
214 share_mode = fetch_share_mode_unlocked(NULL, id);
215 if (share_mode) {
216 bool has_stream = share_mode->data->stream_name != NULL;
218 fname = talloc_asprintf(NULL, "%s%s%s",
219 share_mode->data->base_name,
220 has_stream ? ":" : "",
221 has_stream ?
222 share_mode->data->stream_name :
223 "");
224 } else {
225 fname = talloc_strdup(NULL, "");
226 if (fname == NULL) {
227 return;
231 for (i=0;i<ARRAY_SIZE(lock_types);i++) {
232 if (lock_type == lock_types[i].lock_type) {
233 desc = lock_types[i].desc;
237 d_printf("%-10s %-15s %-4s %-9.0f %-9.0f %-24s %-24s\n",
238 procid_str_static(&pid), file_id_string_tos(&id),
239 desc,
240 (double)start, (double)size,
241 sharepath, fname);
243 TALLOC_FREE(fname);
244 TALLOC_FREE(share_mode);
247 static int traverse_connections(const struct connections_key *key,
248 const struct connections_data *crec,
249 void *state)
251 if (crec->cnum == TID_FIELD_INVALID)
252 return 0;
254 if (do_checks &&
255 (!process_exists(crec->pid) || !Ucrit_checkUid(crec->uid))) {
256 return 0;
259 d_printf("%-10s %s %-12s %s",
260 crec->servicename,procid_str_static(&crec->pid),
261 crec->machine,
262 time_to_asc(crec->start));
264 return 0;
267 static int traverse_sessionid(const char *key, struct sessionid *session,
268 void *private_data)
270 fstring uid_str, gid_str;
272 if (do_checks &&
273 (!process_exists(session->pid) ||
274 !Ucrit_checkUid(session->uid))) {
275 return 0;
278 Ucrit_addPid(session->pid);
280 fstrcpy(uid_str, "-1");
282 if (session->uid != -1) {
283 if (numeric_only) {
284 fstr_sprintf(uid_str, "%u", (unsigned int)session->uid);
285 } else {
286 fstrcpy(uid_str, uidtoname(session->uid));
290 fstrcpy(gid_str, "-1");
292 if (session->gid != -1) {
293 if (numeric_only) {
294 fstr_sprintf(gid_str, "%u", (unsigned int)session->gid);
295 } else {
296 fstrcpy(gid_str, gidtoname(session->gid));
300 d_printf("%-7s %-12s %-12s %-12s (%s)\n",
301 procid_str_static(&session->pid),
302 uid_str, gid_str,
303 session->remote_machine, session->hostname);
305 return 0;
309 static void print_notify_recs(const char *path,
310 struct notify_db_entry *entries,
311 size_t num_entries,
312 time_t deleted_time, void *private_data)
314 size_t i;
315 d_printf("%s\n", path);
317 if (num_entries == 0) {
318 d_printf("deleted %s\n", time_to_asc(deleted_time));
321 for (i=0; i<num_entries; i++) {
322 struct notify_db_entry *e = &entries[i];
323 char *str;
325 str = server_id_str(talloc_tos(), &e->server);
326 printf("%s %x %x\n", str, (unsigned)e->filter,
327 (unsigned)e->subdir_filter);
328 TALLOC_FREE(str);
330 printf("\n");
333 int main(int argc, char *argv[])
335 int c;
336 int profile_only = 0;
337 bool show_processes, show_locks, show_shares;
338 bool show_notify = false;
339 poptContext pc;
340 struct poptOption long_options[] = {
341 POPT_AUTOHELP
342 {"processes", 'p', POPT_ARG_NONE, NULL, 'p', "Show processes only" },
343 {"verbose", 'v', POPT_ARG_NONE, NULL, 'v', "Be verbose" },
344 {"locks", 'L', POPT_ARG_NONE, NULL, 'L', "Show locks only" },
345 {"shares", 'S', POPT_ARG_NONE, NULL, 'S', "Show shares only" },
346 {"notify", 'N', POPT_ARG_NONE, NULL, 'N', "Show notifies" },
347 {"user", 'u', POPT_ARG_STRING, &username, 'u', "Switch to user" },
348 {"brief", 'b', POPT_ARG_NONE, NULL, 'b', "Be brief" },
349 {"profile", 'P', POPT_ARG_NONE, NULL, 'P', "Do profiling" },
350 {"profile-rates", 'R', POPT_ARG_NONE, NULL, 'R', "Show call rates" },
351 {"byterange", 'B', POPT_ARG_NONE, NULL, 'B', "Include byte range locks"},
352 {"numeric", 'n', POPT_ARG_NONE, NULL, 'n', "Numeric uid/gid"},
353 {"fast", 'f', POPT_ARG_NONE, NULL, 'f', "Skip checks if processes still exist"},
354 POPT_COMMON_SAMBA
355 POPT_TABLEEND
357 TALLOC_CTX *frame = talloc_stackframe();
358 int ret = 0;
359 struct messaging_context *msg_ctx;
361 sec_init();
362 load_case_tables();
364 setup_logging(argv[0], DEBUG_STDERR);
366 if (getuid() != geteuid()) {
367 d_printf("smbstatus should not be run setuid\n");
368 ret = 1;
369 goto done;
372 pc = poptGetContext(NULL, argc, (const char **) argv, long_options,
373 POPT_CONTEXT_KEEP_FIRST);
375 while ((c = poptGetNextOpt(pc)) != -1) {
376 switch (c) {
377 case 'p':
378 processes_only = true;
379 break;
380 case 'v':
381 verbose = true;
382 break;
383 case 'L':
384 locks_only = true;
385 break;
386 case 'S':
387 shares_only = true;
388 break;
389 case 'N':
390 show_notify = true;
391 break;
392 case 'b':
393 brief = true;
394 break;
395 case 'u':
396 Ucrit_addUid(nametouid(poptGetOptArg(pc)));
397 break;
398 case 'P':
399 case 'R':
400 profile_only = c;
401 break;
402 case 'B':
403 show_brl = true;
404 break;
405 case 'n':
406 numeric_only = true;
407 break;
408 case 'f':
409 do_checks = false;
410 break;
414 /* setup the flags based on the possible combincations */
416 show_processes = !(shares_only || locks_only || profile_only) || processes_only;
417 show_locks = !(shares_only || processes_only || profile_only) || locks_only;
418 show_shares = !(processes_only || locks_only || profile_only) || shares_only;
420 if ( username )
421 Ucrit_addUid( nametouid(username) );
423 if (verbose) {
424 d_printf("using configfile = %s\n", get_dyn_CONFIGFILE());
427 if (!lp_load_initial_only(get_dyn_CONFIGFILE())) {
428 fprintf(stderr, "Can't load %s - run testparm to debug it\n",
429 get_dyn_CONFIGFILE());
430 ret = -1;
431 goto done;
435 if (lp_clustering()) {
437 * This implicitly initializes the global ctdbd
438 * connection, usable by the db_open() calls further
439 * down.
441 msg_ctx = messaging_init(NULL, event_context_init(NULL));
442 if (msg_ctx == NULL) {
443 fprintf(stderr, "messaging_init failed\n");
444 ret = -1;
445 goto done;
449 if (!lp_load_global(get_dyn_CONFIGFILE())) {
450 fprintf(stderr, "Can't load %s - run testparm to debug it\n",
451 get_dyn_CONFIGFILE());
452 ret = -1;
453 goto done;
456 switch (profile_only) {
457 case 'P':
458 /* Dump profile data */
459 return status_profile_dump(verbose);
460 case 'R':
461 /* Continuously display rate-converted data */
462 return status_profile_rates(verbose);
463 default:
464 break;
467 if ( show_processes ) {
468 d_printf("\nSamba version %s\n",samba_version_string());
469 d_printf("PID Username Group Machine \n");
470 d_printf("-------------------------------------------------------------------\n");
472 sessionid_traverse_read(traverse_sessionid, NULL);
474 if (processes_only) {
475 goto done;
479 if ( show_shares ) {
480 if (verbose) {
481 d_printf("Opened %s\n", lock_path("connections.tdb"));
484 if (brief) {
485 goto done;
488 d_printf("\nService pid machine Connected at\n");
489 d_printf("-------------------------------------------------------\n");
491 connections_forall_read(traverse_connections, NULL);
493 d_printf("\n");
495 if ( shares_only ) {
496 goto done;
500 if ( show_locks ) {
501 int result;
502 struct db_context *db;
503 db = db_open(NULL, lock_path("locking.tdb"), 0,
504 TDB_CLEAR_IF_FIRST|TDB_INCOMPATIBLE_HASH, O_RDONLY, 0,
505 DBWRAP_LOCK_ORDER_1);
507 if (!db) {
508 d_printf("%s not initialised\n",
509 lock_path("locking.tdb"));
510 d_printf("This is normal if an SMB client has never "
511 "connected to your server.\n");
512 exit(0);
513 } else {
514 TALLOC_FREE(db);
517 if (!locking_init_readonly()) {
518 d_printf("Can't initialise locking module - exiting\n");
519 ret = 1;
520 goto done;
523 result = share_mode_forall(print_share_mode, NULL);
525 if (result == 0) {
526 d_printf("No locked files\n");
527 } else if (result < 0) {
528 d_printf("locked file list truncated\n");
531 d_printf("\n");
533 if (show_brl) {
534 brl_forall(print_brl, NULL);
537 locking_end();
540 if (show_notify) {
541 struct notify_context *n;
543 n = notify_init(talloc_tos(), NULL, NULL);
544 if (n == NULL) {
545 goto done;
547 notify_walk(n, print_notify_recs, NULL);
548 TALLOC_FREE(n);
551 done:
552 TALLOC_FREE(frame);
553 return ret;