2 Samba Unix/Linux SMB client library
4 Copyright (C) Volker Lendecke 2010
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/>.
21 #include "utils/net.h"
22 #include "dbwrap/dbwrap.h"
23 #include "dbwrap/dbwrap_rbt.h"
26 #include "lib/conn_tdb.h"
27 #include "smbd/globals.h"
30 static int net_serverid_list_fn(const struct server_id
*id
,
31 uint32_t msg_flags
, void *priv
)
33 char *str
= server_id_str(talloc_tos(), id
);
34 d_printf("%s %llu 0x%x\n", str
, (unsigned long long)id
->unique_id
,
35 (unsigned int)msg_flags
);
40 static int net_serverid_list(struct net_context
*c
, int argc
,
43 d_printf("pid unique_id msg_flags\n");
44 return serverid_traverse_read(net_serverid_list_fn
, NULL
) ? 0 : -1;
47 static int net_serverid_wipe_fn(struct db_record
*rec
,
48 const struct server_id
*id
,
49 uint32_t msg_flags
, void *private_data
)
53 if (id
->vnn
!= get_my_vnn()) {
56 status
= dbwrap_record_delete(rec
);
57 if (!NT_STATUS_IS_OK(status
)) {
58 char *str
= server_id_str(talloc_tos(), id
);
59 DEBUG(1, ("Could not delete serverid.tdb record %s: %s\n",
60 str
, nt_errstr(status
)));
66 static int net_serverid_wipe(struct net_context
*c
, int argc
,
69 return serverid_traverse(net_serverid_wipe_fn
, NULL
) ? 0 : -1;
73 struct wipedbs_record_marker
{
74 struct wipedbs_record_marker
*prev
, *next
;
79 struct wipedbs_server_data
{
80 struct server_id server_id
;
81 const char *server_id_str
;
83 struct wipedbs_record_marker
*session_records
;
84 struct wipedbs_record_marker
*tcon_records
;
85 struct wipedbs_record_marker
*open_records
;
88 struct wipedbs_state
{
89 struct db_context
*id2server_data
;
101 } session
, tcon
, open
;
104 struct server_id
*server_ids
;
107 struct db_context
*session_db
;
108 struct db_context
*tcon_db
;
109 struct db_context
*open_db
;
115 static struct wipedbs_server_data
*get_server_data(struct wipedbs_state
*state
,
116 const struct server_id
*id
)
118 struct wipedbs_server_data
*ret
= NULL
;
119 TDB_DATA key
, val
= tdb_null
;
122 key
= make_tdb_data((const void*)&id
->unique_id
, sizeof(id
->unique_id
));
123 status
= dbwrap_fetch(state
->id2server_data
, talloc_tos(), key
, &val
);
124 if (NT_STATUS_IS_OK(status
)) {
125 ret
= *(struct wipedbs_server_data
**) val
.dptr
;
126 TALLOC_FREE(val
.dptr
);
127 } else if (NT_STATUS_EQUAL(status
, NT_STATUS_NOT_FOUND
)) {
128 ret
= talloc_zero(state
->id2server_data
,
129 struct wipedbs_server_data
);
131 DEBUG(0, ("Failed to allocate server entry for %s\n",
132 server_id_str(talloc_tos(), id
)));
135 ret
->server_id
= *id
;
136 ret
->server_id_str
= server_id_str(ret
, id
);
138 val
= make_tdb_data((const void*)&ret
, sizeof(ret
));
139 status
= dbwrap_store(state
->id2server_data
,
140 key
, val
, TDB_INSERT
);
141 if (!NT_STATUS_IS_OK(status
)) {
142 DEBUG(0, ("Failed to store server entry for %s: %s\n",
143 server_id_str(talloc_tos(), id
),
148 DEBUG(0, ("Failed to fetch server entry for %s: %s\n",
149 server_id_str(talloc_tos(), id
), nt_errstr(status
)));
152 if (!server_id_equal(id
, &ret
->server_id
)) {
153 DEBUG(0, ("uniq id collision for %s and %s\n",
154 server_id_str(talloc_tos(), id
),
155 server_id_str(talloc_tos(), &ret
->server_id
)));
156 smb_panic("server_id->unique_id not unique!");
162 static int wipedbs_traverse_sessions(struct smbXsrv_session_global0
*session
,
165 struct wipedbs_state
*state
=
166 talloc_get_type_abort(wipedbs_state
,
167 struct wipedbs_state
);
168 struct wipedbs_server_data
*sd
;
169 struct wipedbs_record_marker
*rec
;
173 assert(session
->num_channels
== 1);
175 state
->stat
.session
.total
++;
177 sd
= get_server_data(state
, &session
->channels
[0].server_id
);
182 if (server_id_is_disconnected(&sd
->server_id
)) {
183 state
->stat
.session
.disconnected
++;
186 rec
= talloc_zero(sd
, struct wipedbs_record_marker
);
188 DEBUG(0, ("Out of memory!\n"));
192 tmp
= dbwrap_record_get_key(session
->db_rec
);
193 rec
->key
= tdb_data_talloc_copy(rec
, tmp
);
194 tmp
= dbwrap_record_get_value(session
->db_rec
);
195 rec
->val
= tdb_data_talloc_copy(rec
, tmp
);
197 rec
->desc
= talloc_asprintf(
198 rec
, "session[global: %u wire: %llu]",
199 session
->session_global_id
,
200 (long long unsigned)session
->session_wire_id
);
202 if ((rec
->key
.dptr
== NULL
) || (rec
->val
.dptr
== NULL
) ||
205 DEBUG(0, ("Out of memory!\n"));
209 state
->session_db
= dbwrap_record_get_db(session
->db_rec
);
211 DLIST_ADD(sd
->session_records
, rec
);
217 static int wipedbs_traverse_tcon(struct smbXsrv_tcon_global0
*tcon
,
220 struct wipedbs_state
*state
=
221 talloc_get_type_abort(wipedbs_state
,
222 struct wipedbs_state
);
223 struct wipedbs_server_data
*sd
;
224 struct wipedbs_record_marker
*rec
;
228 state
->stat
.tcon
.total
++;
230 sd
= get_server_data(state
, &tcon
->server_id
);
235 if (server_id_is_disconnected(&sd
->server_id
)) {
236 state
->stat
.tcon
.disconnected
++;
239 rec
= talloc_zero(sd
, struct wipedbs_record_marker
);
241 DEBUG(0, ("Out of memory!\n"));
245 tmp
= dbwrap_record_get_key(tcon
->db_rec
);
246 rec
->key
= tdb_data_talloc_copy(rec
, tmp
);
247 tmp
= dbwrap_record_get_value(tcon
->db_rec
);
248 rec
->val
= tdb_data_talloc_copy(rec
, tmp
);
250 rec
->desc
= talloc_asprintf(
251 rec
, "tcon[global: %u wire: %u session: %u share: %s]",
252 tcon
->tcon_global_id
, tcon
->tcon_wire_id
,
253 tcon
->session_global_id
, tcon
->share_name
);
255 if ((rec
->key
.dptr
== NULL
) || (rec
->val
.dptr
== NULL
) ||
258 DEBUG(0, ("Out of memory!\n"));
262 state
->tcon_db
= dbwrap_record_get_db(tcon
->db_rec
);
264 DLIST_ADD(sd
->tcon_records
, rec
);
271 static int wipedbs_traverse_open(struct smbXsrv_open_global0
*open
,
274 struct wipedbs_state
*state
=
275 talloc_get_type_abort(wipedbs_state
,
276 struct wipedbs_state
);
277 struct wipedbs_server_data
*sd
;
278 struct wipedbs_record_marker
*rec
;
282 state
->stat
.open
.total
++;
284 sd
= get_server_data(state
, &open
->server_id
);
289 if (server_id_is_disconnected(&sd
->server_id
)) {
290 struct timeval disconnect_time
;
294 state
->stat
.open
.disconnected
++;
296 nttime_to_timeval(&disconnect_time
, open
->disconnect_time
);
297 tdiff
= usec_time_diff(&state
->now
, &disconnect_time
);
298 reached
= (tdiff
>= 1000*open
->durable_timeout_msec
);
300 if (state
->verbose
) {
301 TALLOC_CTX
*mem_ctx
= talloc_new(talloc_tos());
302 d_printf("open[global: %u] disconnected at "
303 "[%s] %us ago with timeout of %us "
305 open
->open_global_id
,
306 nt_time_string(mem_ctx
, open
->disconnect_time
),
307 (unsigned)(tdiff
/1000000),
308 open
->durable_timeout_msec
/ 1000,
309 reached
? "" : " not");
310 talloc_free(mem_ctx
);
317 state
->stat
.open_timed_out
++;
320 rec
= talloc_zero(sd
, struct wipedbs_record_marker
);
322 DEBUG(0, ("Out of memory!\n"));
326 tmp
= dbwrap_record_get_key(open
->db_rec
);
327 rec
->key
= tdb_data_talloc_copy(rec
, tmp
);
328 tmp
= dbwrap_record_get_value(open
->db_rec
);
329 rec
->val
= tdb_data_talloc_copy(rec
, tmp
);
331 rec
->desc
= talloc_asprintf(
332 rec
, "open[global: %u persistent: %llu volatile: %llu]",
333 open
->open_global_id
,
334 (long long unsigned)open
->open_persistent_id
,
335 (long long unsigned)open
->open_volatile_id
);
337 if ((rec
->key
.dptr
== NULL
) || (rec
->val
.dptr
== NULL
) ||
340 DEBUG(0, ("Out of memory!\n"));
344 state
->open_db
= dbwrap_record_get_db(open
->db_rec
);
346 DLIST_ADD(sd
->open_records
, rec
);
353 static int wipedbs_traverse_nop(struct db_record
*rec
, void *private_data
)
358 static int wipedbs_traverse_fill_ids(struct db_record
*rec
, void *wipedbs_state
)
360 struct wipedbs_state
*state
= talloc_get_type_abort(
361 wipedbs_state
, struct wipedbs_state
);
363 TDB_DATA val
= dbwrap_record_get_value(rec
);
365 struct wipedbs_server_data
*sd
= talloc_get_type_abort(
366 *(void**)val
.dptr
, struct wipedbs_server_data
);
368 state
->server_ids
[state
->idx
] = sd
->server_id
;
373 static int wipedbs_traverse_set_exists(struct db_record
*rec
,
376 struct wipedbs_state
*state
= talloc_get_type_abort(
377 wipedbs_state
, struct wipedbs_state
);
379 TDB_DATA val
= dbwrap_record_get_value(rec
);
381 struct wipedbs_server_data
*sd
= talloc_get_type_abort(
382 *(void**)val
.dptr
, struct wipedbs_server_data
);
384 /* assume a stable traverse order for rbt */
385 SMB_ASSERT(server_id_equal(&state
->server_ids
[state
->idx
],
387 sd
->exists
= state
->server_exists
[state
->idx
];
390 state
->stat
.server
.existing
++;
392 if (server_id_is_disconnected(&sd
->server_id
)) {
393 state
->stat
.server
.disconnected
++;
400 static NTSTATUS
wipedbs_check_server_exists(struct wipedbs_state
*state
)
406 status
= dbwrap_traverse_read(state
->id2server_data
,
407 wipedbs_traverse_nop
, NULL
, &num_servers
);
408 if (!NT_STATUS_IS_OK(status
)) {
409 DEBUG(0, ("Failed to traverse temporary database\n"));
412 state
->stat
.server
.total
= num_servers
;
414 state
->server_ids
= talloc_array(state
, struct server_id
, num_servers
);
415 state
->server_exists
= talloc_array(state
, bool, num_servers
);
416 if (state
->server_ids
== NULL
|| state
->server_exists
== NULL
) {
417 DEBUG(0, ("Out of memory\n"));
422 status
= dbwrap_traverse_read(state
->id2server_data
,
423 wipedbs_traverse_fill_ids
,
425 if (!NT_STATUS_IS_OK(status
)) {
426 DEBUG(0, ("Failed to traverse temporary database\n"));
430 ok
= serverids_exist(state
->server_ids
, num_servers
, state
->server_exists
);
432 DEBUG(0, ("Calling serverids_exist failed\n"));
433 status
= NT_STATUS_UNSUCCESSFUL
;
438 status
= dbwrap_traverse_read(state
->id2server_data
,
439 wipedbs_traverse_set_exists
, state
, NULL
);
440 if (!NT_STATUS_IS_OK(status
)) {
441 DEBUG(0, ("Failed to traverse temporary database\n"));
445 TALLOC_FREE(state
->server_ids
);
446 TALLOC_FREE(state
->server_exists
);
450 static int wipedbs_delete_records(struct db_context
*db
,
451 struct wipedbs_record_marker
*records
,
452 bool dry_run
, bool verbose
, int *count
)
454 struct wipedbs_record_marker
*cur
;
455 struct db_record
*rec
;
458 unsigned num
=0, total
=0;
464 for (cur
= records
; cur
!= NULL
; cur
= cur
->next
) {
466 rec
= dbwrap_fetch_locked(db
, talloc_tos(), cur
->key
);
468 DEBUG(0, ("Failed to fetch record <%s> from %s",
469 cur
->desc
, dbwrap_name(db
)));
472 val
= dbwrap_record_get_value(rec
);
473 if (tdb_data_equal(val
, cur
->val
)) {
475 status
= NT_STATUS_OK
;
477 status
= dbwrap_record_delete(rec
);
479 if (NT_STATUS_IS_OK(status
)) {
482 DEBUG(0, ("Failed to delete record <%s> from %s"
483 ": %s\n", cur
->desc
, dbwrap_name(db
),
487 DEBUG(0, ("Warning: record <%s> from %s changed"
489 cur
->desc
, dbwrap_name(db
)));
492 d_printf("deleting %s\n", cur
->desc
);
498 d_printf("Deleted %u of %u records from %s\n",
499 num
, total
, dbwrap_name(db
));
509 static int wipedbs_traverse_server_data(struct db_record
*rec
,
512 struct wipedbs_state
*state
= talloc_get_type_abort(
513 wipedbs_state
, struct wipedbs_state
);
514 bool dry_run
= state
->testmode
;
515 TDB_DATA val
= dbwrap_record_get_value(rec
);
517 struct wipedbs_server_data
*sd
= talloc_get_type_abort(
518 *(void**)val
.dptr
, struct wipedbs_server_data
);
520 if (state
->verbose
) {
521 d_printf("Server: '%s' %s\n", sd
->server_id_str
,
524 "does not exist, cleaning up...");
531 ret
= wipedbs_delete_records(state
->session_db
, sd
->session_records
,
532 dry_run
, state
->verbose
,
533 &state
->stat
.session
.todelete
);
534 state
->stat
.session
.failure
+= ret
;
536 ret
= wipedbs_delete_records(state
->tcon_db
, sd
->tcon_records
,
537 dry_run
, state
->verbose
,
538 &state
->stat
.tcon
.todelete
);
539 state
->stat
.tcon
.failure
+= ret
;
541 ret
= wipedbs_delete_records(state
->open_db
, sd
->open_records
,
542 dry_run
, state
->verbose
,
543 &state
->stat
.open
.todelete
);
544 state
->stat
.open
.failure
+= ret
;
549 static int net_serverid_wipedbs(struct net_context
*c
, int argc
,
554 struct wipedbs_state
*state
= talloc_zero(talloc_tos(),
555 struct wipedbs_state
);
557 if (c
->display_usage
) {
560 _("net serverid wipedbs [--test] [--verbose]\n"));
563 _("net serverid wipedbs -v\n"));
567 state
->now
= timeval_current();
568 state
->testmode
= c
->opt_testmode
;
569 state
->verbose
= c
->opt_verbose
;
571 state
->id2server_data
= db_open_rbt(state
);
572 if (state
->id2server_data
== NULL
) {
573 DEBUG(0, ("Failed to open temporary database\n"));
577 status
= smbXsrv_session_global_traverse(wipedbs_traverse_sessions
,
579 if (!NT_STATUS_IS_OK(status
)) {
583 status
= smbXsrv_tcon_global_traverse(wipedbs_traverse_tcon
, state
);
584 if (!NT_STATUS_IS_OK(status
)) {
588 status
= smbXsrv_open_global_traverse(wipedbs_traverse_open
, state
);
589 if (!NT_STATUS_IS_OK(status
)) {
593 status
= wipedbs_check_server_exists(state
);
594 if (!NT_STATUS_IS_OK(status
)) {
598 status
= dbwrap_traverse_read(state
->id2server_data
,
599 wipedbs_traverse_server_data
,
601 if (!NT_STATUS_IS_OK(status
)) {
602 DEBUG(0, ("Failed to traverse db: %s\n", nt_errstr(status
)));
606 d_printf("Found %d serverids, %d alive and %d disconnected\n",
607 state
->stat
.server
.total
,
608 state
->stat
.server
.existing
,
609 state
->stat
.server
.disconnected
);
610 d_printf("Found %d sessions, %d alive and %d disconnected"
611 ", cleaned up %d of %d entries\n",
612 state
->stat
.session
.total
,
613 state
->stat
.session
.total
- state
->stat
.session
.todelete
,
614 state
->stat
.session
.disconnected
,
615 state
->stat
.session
.todelete
- state
->stat
.session
.failure
,
616 state
->stat
.session
.todelete
);
617 d_printf("Found %d tcons, %d alive and %d disconnected"
618 ", cleaned up %d of %d entries\n",
619 state
->stat
.tcon
.total
,
620 state
->stat
.tcon
.total
- state
->stat
.tcon
.todelete
,
621 state
->stat
.tcon
.disconnected
,
622 state
->stat
.tcon
.todelete
- state
->stat
.tcon
.failure
,
623 state
->stat
.tcon
.todelete
);
624 d_printf("Found %d opens, %d alive, %d disconnected and %d timed out"
625 ", cleaned up %d of %d entries\n",
626 state
->stat
.open
.total
,
627 state
->stat
.open
.total
- state
->stat
.open
.todelete
628 - (state
->stat
.open
.disconnected
- state
->stat
.open_timed_out
),
629 state
->stat
.open
.disconnected
,
630 state
->stat
.open_timed_out
,
631 state
->stat
.open
.todelete
- state
->stat
.open
.failure
,
632 state
->stat
.open
.todelete
);
640 int net_serverid(struct net_context
*c
, int argc
, const char **argv
)
642 struct functable func
[] = {
647 N_("List all entries from serverid.tdb"),
648 N_("net serverid list\n"
649 " List all entries from serverid.tdb")
655 N_("Wipe the serverid.tdb for the current node"),
656 N_("net serverid wipe\n"
657 " Wipe the serverid.tdb for the current node")
661 net_serverid_wipedbs
,
663 N_("Clean dead entries from temporary databases"),
664 N_("net serverid wipedbs\n"
665 " Clean dead entries from temporary databases")
667 {NULL
, NULL
, 0, NULL
, NULL
}
670 return net_run_function(c
, argc
, argv
, "net serverid", func
);