smbstatus: pass talloc context to traverse_connections
[Samba.git] / source3 / utils / status.c
blob3764e887ad7224637ba2a78862596be2507f122d
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"
46 #include "serverid.h"
47 #include "status_profile.h"
48 #include "smbd/notifyd/notifyd.h"
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 */
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;
64 const char *username = NULL;
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 int print_share_mode(const struct share_mode_entry *e,
118 const char *sharepath,
119 const char *fname,
120 const char *sname,
121 void *dummy)
123 static int count;
125 if (do_checks && !is_valid_share_mode_entry(e)) {
126 return 0;
129 if (count==0) {
130 d_printf("Locked files:\n");
131 d_printf("Pid Uid DenyMode Access R/W Oplock SharePath Name Time\n");
132 d_printf("--------------------------------------------------------------------------------------------------\n");
134 count++;
136 if (do_checks && !serverid_exists(&e->pid)) {
137 /* the process for this entry does not exist any more */
138 return 0;
141 if (Ucrit_checkPid(e->pid)) {
142 struct server_id_buf tmp;
143 d_printf("%-11s ", server_id_str_buf(e->pid, &tmp));
144 d_printf("%-9u ", (unsigned int)e->uid);
145 switch (map_share_mode_to_deny_mode(e->share_access,
146 e->private_options)) {
147 case DENY_NONE: d_printf("DENY_NONE "); break;
148 case DENY_ALL: d_printf("DENY_ALL "); break;
149 case DENY_DOS: d_printf("DENY_DOS "); break;
150 case DENY_READ: d_printf("DENY_READ "); break;
151 case DENY_WRITE:printf("DENY_WRITE "); break;
152 case DENY_FCB: d_printf("DENY_FCB "); break;
153 default: {
154 d_printf("unknown-please report ! "
155 "e->share_access = 0x%x, "
156 "e->private_options = 0x%x\n",
157 (unsigned int)e->share_access,
158 (unsigned int)e->private_options );
159 break;
162 d_printf("0x%-8x ",(unsigned int)e->access_mask);
163 if ((e->access_mask & (FILE_READ_DATA|FILE_WRITE_DATA))==
164 (FILE_READ_DATA|FILE_WRITE_DATA)) {
165 d_printf("RDWR ");
166 } else if (e->access_mask & FILE_WRITE_DATA) {
167 d_printf("WRONLY ");
168 } else {
169 d_printf("RDONLY ");
172 if((e->op_type & (EXCLUSIVE_OPLOCK|BATCH_OPLOCK)) ==
173 (EXCLUSIVE_OPLOCK|BATCH_OPLOCK)) {
174 d_printf("EXCLUSIVE+BATCH ");
175 } else if (e->op_type & EXCLUSIVE_OPLOCK) {
176 d_printf("EXCLUSIVE ");
177 } else if (e->op_type & BATCH_OPLOCK) {
178 d_printf("BATCH ");
179 } else if (e->op_type & LEVEL_II_OPLOCK) {
180 d_printf("LEVEL_II ");
181 } else if (e->op_type == LEASE_OPLOCK) {
182 uint32_t lstate = e->lease->current_state;
183 d_printf("LEASE(%s%s%s)%s%s%s ",
184 (lstate & SMB2_LEASE_READ)?"R":"",
185 (lstate & SMB2_LEASE_WRITE)?"W":"",
186 (lstate & SMB2_LEASE_HANDLE)?"H":"",
187 (lstate & SMB2_LEASE_READ)?"":" ",
188 (lstate & SMB2_LEASE_WRITE)?"":" ",
189 (lstate & SMB2_LEASE_HANDLE)?"":" ");
190 } else {
191 d_printf("NONE ");
194 d_printf(" %s %s%s %s",
195 sharepath, fname,
196 sname ? sname : "",
197 time_to_asc((time_t)e->time.tv_sec));
200 return 0;
203 static void print_brl(struct file_id id,
204 struct server_id pid,
205 enum brl_type lock_type,
206 enum brl_flavour lock_flav,
207 br_off start,
208 br_off size,
209 void *private_data)
211 static int count;
212 unsigned int i;
213 static const struct {
214 enum brl_type lock_type;
215 const char *desc;
216 } lock_types[] = {
217 { READ_LOCK, "R" },
218 { WRITE_LOCK, "W" },
219 { PENDING_READ_LOCK, "PR" },
220 { PENDING_WRITE_LOCK, "PW" },
221 { UNLOCK_LOCK, "U" }
223 const char *desc="X";
224 const char *sharepath = "";
225 char *fname = NULL;
226 struct share_mode_lock *share_mode;
227 struct server_id_buf tmp;
229 if (count==0) {
230 d_printf("Byte range locks:\n");
231 d_printf("Pid dev:inode R/W start size SharePath Name\n");
232 d_printf("--------------------------------------------------------------------------------\n");
234 count++;
236 share_mode = fetch_share_mode_unlocked(NULL, id);
237 if (share_mode) {
238 bool has_stream = share_mode->data->stream_name != NULL;
240 fname = talloc_asprintf(NULL, "%s%s%s",
241 share_mode->data->base_name,
242 has_stream ? ":" : "",
243 has_stream ?
244 share_mode->data->stream_name :
245 "");
246 } else {
247 fname = talloc_strdup(NULL, "");
248 if (fname == NULL) {
249 return;
253 for (i=0;i<ARRAY_SIZE(lock_types);i++) {
254 if (lock_type == lock_types[i].lock_type) {
255 desc = lock_types[i].desc;
259 d_printf("%-10s %-15s %-4s %-9jd %-9jd %-24s %-24s\n",
260 server_id_str_buf(pid, &tmp), file_id_string_tos(&id),
261 desc,
262 (intmax_t)start, (intmax_t)size,
263 sharepath, fname);
265 TALLOC_FREE(fname);
266 TALLOC_FREE(share_mode);
269 static const char *session_dialect_str(uint16_t dialect)
271 static fstring unkown_dialect;
273 switch(dialect){
274 case SMB2_DIALECT_REVISION_000:
275 return "NT1";
276 case SMB2_DIALECT_REVISION_202:
277 return "SMB2_02";
278 case SMB2_DIALECT_REVISION_210:
279 return "SMB2_10";
280 case SMB2_DIALECT_REVISION_222:
281 return "SMB2_22";
282 case SMB2_DIALECT_REVISION_224:
283 return "SMB2_24";
284 case SMB3_DIALECT_REVISION_300:
285 return "SMB3_00";
286 case SMB3_DIALECT_REVISION_302:
287 return "SMB3_02";
288 case SMB3_DIALECT_REVISION_310:
289 return "SMB3_10";
290 case SMB3_DIALECT_REVISION_311:
291 return "SMB3_11";
294 fstr_sprintf(unkown_dialect, "Unknown (0x%04x)", dialect);
295 return unkown_dialect;
298 static int traverse_connections(const struct connections_key *key,
299 const struct connections_data *crec,
300 void *private_data)
302 TALLOC_CTX *mem_ctx = (TALLOC_CTX *)private_data;
303 struct server_id_buf tmp;
305 if (crec->cnum == TID_FIELD_INVALID)
306 return 0;
308 if (do_checks &&
309 (!process_exists(crec->pid) || !Ucrit_checkUid(crec->uid))) {
310 return 0;
313 d_printf("%-10s %s %-12s %s",
314 crec->servicename, server_id_str_buf(crec->pid, &tmp),
315 crec->machine,
316 time_to_asc(crec->start));
318 return 0;
321 static int traverse_sessionid(const char *key, struct sessionid *session,
322 void *private_data)
324 TALLOC_CTX *mem_ctx = (TALLOC_CTX *)private_data;
325 fstring uid_str, gid_str;
326 struct server_id_buf tmp;
328 if (do_checks &&
329 (!process_exists(session->pid) ||
330 !Ucrit_checkUid(session->uid))) {
331 return 0;
334 Ucrit_addPid(session->pid);
336 fstrcpy(uid_str, "-1");
338 if (session->uid != -1) {
339 if (numeric_only) {
340 fstr_sprintf(uid_str, "%u", (unsigned int)session->uid);
341 } else {
342 fstrcpy(uid_str, uidtoname(session->uid));
346 fstrcpy(gid_str, "-1");
348 if (session->gid != -1) {
349 if (numeric_only) {
350 fstr_sprintf(gid_str, "%u", (unsigned int)session->gid);
351 } else {
352 fstrcpy(gid_str, gidtoname(session->gid));
356 d_printf("%-7s %-12s %-12s %-12s (%s) %-12s\n",
357 server_id_str_buf(session->pid, &tmp),
358 uid_str, gid_str,
359 session->remote_machine, session->hostname,
360 session_dialect_str(session->connection_dialect));
362 return 0;
366 static bool print_notify_rec(const char *path, struct server_id server,
367 const struct notify_instance *instance,
368 void *private_data)
370 struct server_id_buf idbuf;
372 d_printf("%s\\%s\\%x\\%x\n", path, server_id_str_buf(server, &idbuf),
373 (unsigned)instance->filter,
374 (unsigned)instance->subdir_filter);
376 return true;
379 int main(int argc, const char *argv[])
381 int c;
382 int profile_only = 0;
383 bool show_processes, show_locks, show_shares;
384 bool show_notify = false;
385 poptContext pc;
386 struct poptOption long_options[] = {
387 POPT_AUTOHELP
388 {"processes", 'p', POPT_ARG_NONE, NULL, 'p', "Show processes only" },
389 {"verbose", 'v', POPT_ARG_NONE, NULL, 'v', "Be verbose" },
390 {"locks", 'L', POPT_ARG_NONE, NULL, 'L', "Show locks only" },
391 {"shares", 'S', POPT_ARG_NONE, NULL, 'S', "Show shares only" },
392 {"notify", 'N', POPT_ARG_NONE, NULL, 'N', "Show notifies" },
393 {"user", 'u', POPT_ARG_STRING, &username, 'u', "Switch to user" },
394 {"brief", 'b', POPT_ARG_NONE, NULL, 'b', "Be brief" },
395 {"profile", 'P', POPT_ARG_NONE, NULL, 'P', "Do profiling" },
396 {"profile-rates", 'R', POPT_ARG_NONE, NULL, 'R', "Show call rates" },
397 {"byterange", 'B', POPT_ARG_NONE, NULL, 'B', "Include byte range locks"},
398 {"numeric", 'n', POPT_ARG_NONE, NULL, 'n', "Numeric uid/gid"},
399 {"fast", 'f', POPT_ARG_NONE, NULL, 'f', "Skip checks if processes still exist"},
400 POPT_COMMON_SAMBA
401 POPT_TABLEEND
403 TALLOC_CTX *frame = talloc_stackframe();
404 int ret = 0;
405 struct messaging_context *msg_ctx = NULL;
406 char *db_path;
407 bool ok;
409 sec_init();
410 smb_init_locale();
412 setup_logging(argv[0], DEBUG_STDERR);
413 lp_set_cmdline("log level", "0");
415 if (getuid() != geteuid()) {
416 d_printf("smbstatus should not be run setuid\n");
417 ret = 1;
418 goto done;
421 if (getuid() != 0) {
422 d_printf("smbstatus only works as root!\n");
423 ret = 1;
424 goto done;
428 pc = poptGetContext(NULL, argc, argv, long_options,
429 POPT_CONTEXT_KEEP_FIRST);
431 while ((c = poptGetNextOpt(pc)) != -1) {
432 switch (c) {
433 case 'p':
434 processes_only = true;
435 break;
436 case 'v':
437 verbose = true;
438 break;
439 case 'L':
440 locks_only = true;
441 break;
442 case 'S':
443 shares_only = true;
444 break;
445 case 'N':
446 show_notify = true;
447 break;
448 case 'b':
449 brief = true;
450 break;
451 case 'u':
452 Ucrit_addUid(nametouid(poptGetOptArg(pc)));
453 break;
454 case 'P':
455 case 'R':
456 profile_only = c;
457 break;
458 case 'B':
459 show_brl = true;
460 break;
461 case 'n':
462 numeric_only = true;
463 break;
464 case 'f':
465 do_checks = false;
466 break;
470 /* setup the flags based on the possible combincations */
472 show_processes = !(shares_only || locks_only || profile_only) || processes_only;
473 show_locks = !(shares_only || processes_only || profile_only) || locks_only;
474 show_shares = !(processes_only || locks_only || profile_only) || shares_only;
476 if ( username )
477 Ucrit_addUid( nametouid(username) );
479 if (verbose) {
480 d_printf("using configfile = %s\n", get_dyn_CONFIGFILE());
483 if (!lp_load_initial_only(get_dyn_CONFIGFILE())) {
484 fprintf(stderr, "Can't load %s - run testparm to debug it\n",
485 get_dyn_CONFIGFILE());
486 ret = -1;
487 goto done;
492 * This implicitly initializes the global ctdbd connection,
493 * usable by the db_open() calls further down.
495 msg_ctx = messaging_init(NULL, samba_tevent_context_init(NULL));
496 if (msg_ctx == NULL) {
497 fprintf(stderr, "messaging_init failed\n");
498 ret = -1;
499 goto done;
502 if (!lp_load_global(get_dyn_CONFIGFILE())) {
503 fprintf(stderr, "Can't load %s - run testparm to debug it\n",
504 get_dyn_CONFIGFILE());
505 ret = -1;
506 goto done;
509 switch (profile_only) {
510 case 'P':
511 /* Dump profile data */
512 ok = status_profile_dump(verbose);
513 return ok ? 0 : 1;
514 case 'R':
515 /* Continuously display rate-converted data */
516 ok = status_profile_rates(verbose);
517 return ok ? 0 : 1;
518 default:
519 break;
522 if ( show_processes ) {
523 d_printf("\nSamba version %s\n",samba_version_string());
524 d_printf("PID Username Group Machine Protocol Version \n");
525 d_printf("------------------------------------------------------------------------------\n");
527 sessionid_traverse_read(traverse_sessionid, frame);
529 if (processes_only) {
530 goto done;
534 if ( show_shares ) {
535 if (brief) {
536 goto done;
539 d_printf("\nService pid machine Connected at\n");
540 d_printf("-------------------------------------------------------\n");
542 connections_forall_read(traverse_connections, frame);
544 d_printf("\n");
546 if ( shares_only ) {
547 goto done;
551 if ( show_locks ) {
552 int result;
553 struct db_context *db;
555 db_path = lock_path("locking.tdb");
556 if (db_path == NULL) {
557 d_printf("Out of memory - exiting\n");
558 ret = -1;
559 goto done;
562 db = db_open(NULL, db_path, 0,
563 TDB_CLEAR_IF_FIRST|TDB_INCOMPATIBLE_HASH, O_RDONLY, 0,
564 DBWRAP_LOCK_ORDER_1, DBWRAP_FLAG_NONE);
566 if (!db) {
567 d_printf("%s not initialised\n", db_path);
568 d_printf("This is normal if an SMB client has never "
569 "connected to your server.\n");
570 TALLOC_FREE(db_path);
571 exit(0);
572 } else {
573 TALLOC_FREE(db);
574 TALLOC_FREE(db_path);
577 if (!locking_init_readonly()) {
578 d_printf("Can't initialise locking module - exiting\n");
579 ret = 1;
580 goto done;
583 result = share_entry_forall(print_share_mode, NULL);
585 if (result == 0) {
586 d_printf("No locked files\n");
587 } else if (result < 0) {
588 d_printf("locked file list truncated\n");
591 d_printf("\n");
593 if (show_brl) {
594 brl_forall(print_brl, NULL);
597 locking_end();
600 if (show_notify) {
601 struct notify_context *n;
603 n = notify_init(talloc_tos(), msg_ctx,
604 messaging_tevent_context(msg_ctx));
605 if (n == NULL) {
606 goto done;
608 notify_walk(n, print_notify_rec, NULL);
609 TALLOC_FREE(n);
612 done:
613 TALLOC_FREE(frame);
614 return ret;