source3/utils/net_g_lock.c

   1 /*
   2  * Samba Unix/Linux SMB client library
   3  * Interface to the g_lock facility
   4  * Copyright (C) Volker Lendecke 2009
   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 "net.h"
  22 #include "lib/util/server_id.h"
  23 #include "g_lock.h"
  24 #include "messages.h"
  25
  26 static bool net_g_lock_init(TALLOC_CTX *mem_ctx,
  27                             struct tevent_context **pev,
  28                             struct messaging_context **pmsg,
  29                             struct g_lock_ctx **pg_ctx)
  30 {
  31         struct tevent_context *ev = NULL;
  32         struct messaging_context *msg = NULL;
  33         struct g_lock_ctx *g_ctx = NULL;
  34
  35         ev = samba_tevent_context_init(mem_ctx);
  36         if (ev == NULL) {
  37                 d_fprintf(stderr, "ERROR: could not init event context\n");
  38                 goto fail;
  39         }
  40         msg = messaging_init(mem_ctx, ev);
  41         if (msg == NULL) {
  42                 d_fprintf(stderr, "ERROR: could not init messaging context\n");
  43                 goto fail;
  44         }
  45         g_ctx = g_lock_ctx_init(mem_ctx, msg);
  46         if (g_ctx == NULL) {
  47                 d_fprintf(stderr, "ERROR: could not init g_lock context\n");
  48                 goto fail;
  49         }
  50
  51         *pev = ev;
  52         *pmsg = msg;
  53         *pg_ctx = g_ctx;
  54         return true;
  55 fail:
  56         TALLOC_FREE(g_ctx);
  57         TALLOC_FREE(msg);
  58         TALLOC_FREE(ev);
  59         return false;
  60 }
  61
  62 struct net_g_lock_do_state {
  63         const char *cmd;
  64         int result;
  65 };
  66
  67 static void net_g_lock_do_fn(void *private_data)
  68 {
  69         struct net_g_lock_do_state *state =
  70                 (struct net_g_lock_do_state *)private_data;
  71         state->result = system(state->cmd);
  72 }
  73
  74 static int net_g_lock_do(struct net_context *c, int argc, const char **argv)
  75 {
  76         struct net_g_lock_do_state state;
  77         const char *name, *cmd;
  78         int timeout;
  79         NTSTATUS status;
  80
  81         if (argc != 3) {
  82                 d_printf("Usage: net g_lock do <lockname> <timeout> "
  83                          "<command>\n");
  84                 return -1;
  85         }
  86         name = argv[0];
  87         timeout = atoi(argv[1]);
  88         cmd = argv[2];
  89
  90         state.cmd = cmd;
  91         state.result = -1;
  92
  93         status = g_lock_do(name, G_LOCK_WRITE,
  94                            timeval_set(timeout / 1000, timeout % 1000),
  95                            net_g_lock_do_fn, &state);
  96         if (!NT_STATUS_IS_OK(status)) {
  97                 d_fprintf(stderr, "ERROR: g_lock_do failed: %s\n",
  98                           nt_errstr(status));
  99                 goto done;
 100         }
 101         if (state.result == -1) {
 102                 d_fprintf(stderr, "ERROR: system() returned %s\n",
 103                           strerror(errno));
 104                 goto done;
 105         }
 106         d_fprintf(stderr, "command returned %d\n", state.result);
 107
 108 done:
 109         return state.result;
 110 }
 111
 112 static void net_g_lock_dump_fn(const struct g_lock_rec *locks,
 113                                size_t num_locks,
 114                                const uint8_t *data,
 115                                size_t datalen,
 116                                void *private_data)
 117 {
 118         size_t i;
 119
 120         for (i=0; i<num_locks; i++) {
 121                 const struct g_lock_rec *l = &locks[i];
 122                 struct server_id_buf idbuf;
 123                 d_printf("%s: %s\n", server_id_str_buf(l->pid, &idbuf),
 124                          (l->lock_type & 1) ? "WRITE" : "READ");
 125         }
 126         dump_data_file(data, datalen, true, stdout);
 127 }
 128
 129 static int net_g_lock_dump(struct net_context *c, int argc, const char **argv)
 130 {
 131         struct tevent_context *ev = NULL;
 132         struct messaging_context *msg = NULL;
 133         struct g_lock_ctx *g_ctx = NULL;
 134         int ret = -1;
 135
 136         if (argc != 1) {
 137                 d_printf("Usage: net g_lock dump <lockname>\n");
 138                 return -1;
 139         }
 140
 141         if (!net_g_lock_init(talloc_tos(), &ev, &msg, &g_ctx)) {
 142                 goto done;
 143         }
 144
 145         (void)g_lock_dump(g_ctx, argv[0], net_g_lock_dump_fn, NULL);
 146
 147         ret = 0;
 148 done:
 149         TALLOC_FREE(g_ctx);
 150         TALLOC_FREE(msg);
 151         TALLOC_FREE(ev);
 152         return ret;
 153 }
 154
 155 static int net_g_lock_locks_fn(const char *name, void *private_data)
 156 {
 157         d_printf("%s\n", name);
 158         return 0;
 159 }
 160
 161 static int net_g_lock_locks(struct net_context *c, int argc, const char **argv)
 162 {
 163         struct tevent_context *ev = NULL;
 164         struct messaging_context *msg = NULL;
 165         struct g_lock_ctx *g_ctx = NULL;
 166         int ret = -1;
 167
 168         if (argc != 0) {
 169                 d_printf("Usage: net g_lock locks\n");
 170                 return -1;
 171         }
 172
 173         if (!net_g_lock_init(talloc_tos(), &ev, &msg, &g_ctx)) {
 174                 goto done;
 175         }
 176
 177         ret = g_lock_locks(g_ctx, net_g_lock_locks_fn, NULL);
 178 done:
 179         TALLOC_FREE(g_ctx);
 180         TALLOC_FREE(msg);
 181         TALLOC_FREE(ev);
 182         return ret < 0 ? -1 : ret;
 183 }
 184
 185 int net_g_lock(struct net_context *c, int argc, const char **argv)
 186 {
 187         struct functable func[] = {
 188                 {
 189                         "do",
 190                         net_g_lock_do,
 191                         NET_TRANSPORT_LOCAL,
 192                         N_("Execute a shell command under a lock"),
 193                         N_("net g_lock do <lock name> <timeout> <command>\n")
 194                 },
 195                 {
 196                         "locks",
 197                         net_g_lock_locks,
 198                         NET_TRANSPORT_LOCAL,
 199                         N_("List all locknames"),
 200                         N_("net g_lock locks\n")
 201                 },
 202                 {
 203                         "dump",
 204                         net_g_lock_dump,
 205                         NET_TRANSPORT_LOCAL,
 206                         N_("Dump a g_lock locking table"),
 207                         N_("net g_lock dump <lock name>\n")
 208                 },
 209                 {NULL, NULL, 0, NULL, NULL}
 210         };
 211
 212         return net_run_function(c, argc, argv, "net g_lock", func);
 213 }