source3/utils/net_serverid.c

   1 /*
   2    Samba Unix/Linux SMB client library
   3    net serverid commands
   4    Copyright (C) Volker Lendecke 2010
   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
  20 #include "includes.h"
  21 #include "utils/net.h"
  22 #include "dbwrap/dbwrap.h"
  23 #include "serverid.h"
  24 #include "session.h"
  25 #include "lib/conn_tdb.h"
  26
  27 static int net_serverid_list_fn(const struct server_id *id,
  28                                 uint32_t msg_flags, void *priv)
  29 {
  30         char *str = server_id_str(talloc_tos(), id);
  31         d_printf("%s %llu 0x%x\n", str, (unsigned long long)id->unique_id,
  32                  (unsigned int)msg_flags);
  33         TALLOC_FREE(str);
  34         return 0;
  35 }
  36
  37 static int net_serverid_list(struct net_context *c, int argc,
  38                              const char **argv)
  39 {
  40         d_printf("pid unique_id msg_flags\n");
  41         return serverid_traverse_read(net_serverid_list_fn, NULL) ? 0 : -1;
  42 }
  43
  44 static int net_serverid_wipe_fn(struct db_record *rec,
  45                                 const struct server_id *id,
  46                                 uint32_t msg_flags, void *private_data)
  47 {
  48         NTSTATUS status;
  49
  50         if (id->vnn != get_my_vnn()) {
  51                 return 0;
  52         }
  53         status = dbwrap_record_delete(rec);
  54         if (!NT_STATUS_IS_OK(status)) {
  55                 char *str = server_id_str(talloc_tos(), id);
  56                 DEBUG(1, ("Could not delete serverid.tdb record %s: %s\n",
  57                           str, nt_errstr(status)));
  58                 TALLOC_FREE(str);
  59         }
  60         return 0;
  61 }
  62
  63 static int net_serverid_wipe(struct net_context *c, int argc,
  64                              const char **argv)
  65 {
  66         return serverid_traverse(net_serverid_wipe_fn, NULL) ? 0 : -1;
  67 }
  68
  69 static int net_serverid_wipedbs_conn(
  70         struct db_record *rec,
  71         const struct connections_key *key,
  72         const struct connections_data *data,
  73         void *private_data)
  74 {
  75         if (!serverid_exists(&key->pid)) {
  76                 NTSTATUS status;
  77
  78                 DEBUG(10, ("Deleting connections.tdb record for pid %s\n",
  79                            server_id_str(talloc_tos(), &key->pid)));
  80
  81                 status = dbwrap_record_delete(rec);
  82                 if (!NT_STATUS_IS_OK(status)) {
  83                         DEBUG(1, ("Could not delete connections.tdb record "
  84                                   "for pid %s: %s\n",
  85                                   server_id_str(talloc_tos(), &key->pid),
  86                                   nt_errstr(status)));
  87                 }
  88         }
  89         return 0;
  90 }
  91
  92 static int net_serverid_wipedbs_sessionid(struct db_record *rec,
  93                                           const char *key,
  94                                           struct sessionid *session,
  95                                           void *private_data)
  96 {
  97         if (!serverid_exists(&session->pid)) {
  98                 NTSTATUS status;
  99
 100                 DEBUG(10, ("Deleting sessionid.tdb record for pid %s\n",
 101                            server_id_str(talloc_tos(), &session->pid)));
 102
 103                 status = dbwrap_record_delete(rec);
 104                 if (!NT_STATUS_IS_OK(status)) {
 105                         DEBUG(1, ("Could not delete session.tdb record "
 106                                   "for pid %s: %s\n",
 107                                   server_id_str(talloc_tos(), &session->pid),
 108                                   nt_errstr(status)));
 109                 }
 110         }
 111         return 0;
 112 }
 113
 114 static int net_serverid_wipedbs(struct net_context *c, int argc,
 115                                 const char **argv)
 116 {
 117         connections_forall(net_serverid_wipedbs_conn, NULL);
 118         sessionid_traverse(net_serverid_wipedbs_sessionid, NULL);
 119         return 0;
 120 }
 121
 122 int net_serverid(struct net_context *c, int argc, const char **argv)
 123 {
 124         struct functable func[] = {
 125                 {
 126                         "list",
 127                         net_serverid_list,
 128                         NET_TRANSPORT_LOCAL,
 129                         N_("List all entries from serverid.tdb"),
 130                         N_("net serverid list\n"
 131                            "    List all entries from serverid.tdb")
 132                 },
 133                 {
 134                         "wipe",
 135                         net_serverid_wipe,
 136                         NET_TRANSPORT_LOCAL,
 137                         N_("Wipe the serverid.tdb for the current node"),
 138                         N_("net serverid wipe\n"
 139                            "    Wipe the serverid.tdb for the current node")
 140                 },
 141                 {
 142                         "wipedbs",
 143                         net_serverid_wipedbs,
 144                         NET_TRANSPORT_LOCAL,
 145                         N_("Clean dead entries from connections.tdb and "
 146                            "sessionid.tdb"),
 147                         N_("net serverid wipedbs\n"
 148                            "    Clean dead entries from connections.tdb and "
 149                            "sessionid.tdb")
 150                 },
 151                 {NULL, NULL, 0, NULL, NULL}
 152         };
 153
 154         return net_run_function(c, argc, argv, "net serverid", func);
 155 }