Merge 2610c05b5b95cc7036b3d6dfb894c6cfbdb68483 as Samba-4.0alpha16
[amitay/samba.git] / source3 / utils / smbcontrol.c
index 09f4f3146fb8480c3b9cf1916f128030c2b80571..5520dcf608a3017539e2e723240b161749f3bd01 100644 (file)
 */
 
 #include "includes.h"
+#include "system/filesys.h"
 #include "popt_common.h"
-#include "librpc/gen_ndr/messaging.h"
 #include "librpc/gen_ndr/spoolss.h"
 #include "nt_printing.h"
+#include "printing/notify.h"
+#include "libsmb/nmblib.h"
+#include "messages.h"
+#include "util_tdb.h"
 
 #if HAVE_LIBUNWIND_H
 #include <libunwind.h>
@@ -60,7 +64,7 @@ static bool send_message(struct messaging_context *msg_ctx,
        if (procid_to_pid(&pid) != 0)
                return NT_STATUS_IS_OK(
                        messaging_send_buf(msg_ctx, pid, msg_type,
-                                          (uint8 *)buf, len));
+                                          (const uint8 *)buf, len));
 
        ret = message_send_all(msg_ctx, msg_type, buf, len, &n_sent);
        DEBUG(10,("smbcontrol/send_message: broadcast message to "
@@ -115,7 +119,7 @@ static void print_pid_string_cb(struct messaging_context *msg,
 {
        char *pidstr;
 
-       pidstr = procid_str(talloc_tos(), &pid);
+       pidstr = server_id_str(talloc_tos(), &pid);
        printf("PID %s: %.*s", pidstr, (int)data->length,
               (const char *)data->data);
        TALLOC_FREE(pidstr);
@@ -166,6 +170,54 @@ static bool do_debug(struct messaging_context *msg_ctx,
                            strlen(argv[1]) + 1);
 }
 
+
+static bool do_idmap(struct messaging_context *msg_ctx,
+                    const struct server_id pid,
+                    const int argc, const char **argv)
+{
+       static const char* usage = "Usage: "
+               "smbcontrol <dest> idmap <cmd> [arg]\n"
+               "\tcmd:\tflush [gid|uid]\n"
+               "\t\tdelete \"UID <uid>\"|\"GID <gid>\"|<sid>\n"
+               "\t\tkill \"UID <uid>\"|\"GID <gid>\"|<sid>\n";
+       const char* arg = NULL;
+       int arglen = 0;
+       int msg_type;
+
+       switch (argc) {
+       case 2:
+               break;
+       case 3:
+               arg = argv[2];
+               arglen = strlen(arg) + 1;
+               break;
+       default:
+               fprintf(stderr, "%s", usage);
+               return false;
+       }
+
+       if (strcmp(argv[1], "flush") == 0) {
+               msg_type = MSG_IDMAP_FLUSH;
+       }
+       else if (strcmp(argv[1], "delete") == 0) {
+               msg_type = MSG_IDMAP_DELETE;
+       }
+       else if (strcmp(argv[1], "kill") == 0) {
+               msg_type = MSG_IDMAP_KILL;
+       }
+       else if (strcmp(argv[1], "help") == 0) {
+               fprintf(stdout, "%s", usage);
+               return true;
+       }
+       else {
+               fprintf(stderr, "%s", usage);
+               return false;
+       }
+
+       return send_message(msg_ctx, pid, msg_type, arg, arglen);
+}
+
+
 #if defined(HAVE_LIBUNWIND_PTRACE) && defined(HAVE_LINUX_PTRACE)
 
 /* Return the name of a process given it's PID. This will only work on Linux,
@@ -381,7 +433,7 @@ static void pong_cb(struct messaging_context *msg,
                    struct server_id pid,
                    DATA_BLOB *data)
 {
-       char *src_string = procid_str(NULL, &pid);
+       char *src_string = server_id_str(NULL, &pid);
        printf("PONG from pid %s\n", src_string);
        TALLOC_FREE(src_string);
        num_replies++;
@@ -593,7 +645,9 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                        return False;
                }
 
-               notify_printer_status_byname(argv[2], PRINTER_STATUS_PAUSED);
+               notify_printer_status_byname(messaging_event_context(msg_ctx),
+                                            msg_ctx, argv[2],
+                                            PRINTER_STATUS_PAUSED);
 
                goto send;
 
@@ -605,7 +659,9 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                        return False;
                }
 
-               notify_printer_status_byname(argv[2], PRINTER_STATUS_OK);
+               notify_printer_status_byname(messaging_event_context(msg_ctx),
+                                            msg_ctx, argv[2],
+                                            PRINTER_STATUS_OK);
 
                goto send;
 
@@ -621,7 +677,8 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                jobid = atoi(argv[3]);
 
                notify_job_status_byname(
-                       argv[2], jobid, JOB_STATUS_PAUSED, 
+                       messaging_event_context(msg_ctx), msg_ctx,
+                       argv[2], jobid, JOB_STATUS_PAUSED,
                        SPOOLSS_NOTIFY_MSG_UNIX_JOBID);
 
                goto send;
@@ -638,6 +695,7 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                jobid = atoi(argv[3]);
 
                notify_job_status_byname(
+                       messaging_event_context(msg_ctx), msg_ctx,
                        argv[2], jobid, JOB_STATUS_QUEUED, 
                        SPOOLSS_NOTIFY_MSG_UNIX_JOBID);
 
@@ -655,10 +713,12 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                jobid = atoi(argv[3]);
 
                notify_job_status_byname(
+                       messaging_event_context(msg_ctx), msg_ctx,
                        argv[2], jobid, JOB_STATUS_DELETING,
                        SPOOLSS_NOTIFY_MSG_UNIX_JOBID);
 
                notify_job_status_byname(
+                       messaging_event_context(msg_ctx), msg_ctx,
                        argv[2], jobid, JOB_STATUS_DELETING|
                        JOB_STATUS_DELETED,
                        SPOOLSS_NOTIFY_MSG_UNIX_JOBID);
@@ -687,8 +747,9 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                        return False;
                }
 
-               notify_printer_byname(argv[2], attribute,
-                                     CONST_DISCARD(char *, argv[4]));
+               notify_printer_byname(messaging_event_context(msg_ctx),
+                                     msg_ctx, argv[2], attribute,
+                                     discard_const_p(char, argv[4]));
 
                goto send;
        }
@@ -717,6 +778,22 @@ static bool do_closeshare(struct messaging_context *msg_ctx,
                            strlen(argv[1]) + 1);
 }
 
+/* Tell winbindd an IP got dropped */
+
+static bool do_ip_dropped(struct messaging_context *msg_ctx,
+                         const struct server_id pid,
+                         const int argc, const char **argv)
+{
+       if (argc != 2) {
+               fprintf(stderr, "Usage: smbcontrol <dest> ip-dropped "
+                       "<ip-address>\n");
+               return False;
+       }
+
+       return send_message(msg_ctx, pid, MSG_WINBIND_IP_DROPPED, argv[1],
+                           strlen(argv[1]) + 1);
+}
+
 /* force a blocking lock retry */
 
 static bool do_lockretry(struct messaging_context *msg_ctx,
@@ -909,7 +986,8 @@ static bool do_winbind_offline(struct messaging_context *msg_ctx,
 
        tdb = tdb_open_log(cache_path("winbindd_cache.tdb"),
                                WINBINDD_CACHE_TDB_DEFAULT_HASH_SIZE,
-                               TDB_DEFAULT /* TDB_CLEAR_IF_FIRST */, O_RDWR|O_CREAT, 0600);
+                               TDB_DEFAULT|TDB_INCOMPATIBLE_HASH /* TDB_CLEAR_IF_FIRST */,
+                               O_RDWR|O_CREAT, 0600);
 
        if (!tdb) {
                fprintf(stderr, "Cannot open the tdb %s for writing.\n",
@@ -961,7 +1039,7 @@ static bool do_winbind_onlinestatus(struct messaging_context *msg_ctx,
 {
        struct server_id myid;
 
-       myid = procid_self();
+       myid = messaging_server_id(msg_ctx);
 
        if (argc != 1) {
                fprintf(stderr, "Usage: smbcontrol winbindd onlinestatus\n");
@@ -993,7 +1071,7 @@ static bool do_dump_event_list(struct messaging_context *msg_ctx,
 {
        struct server_id myid;
 
-       myid = procid_self();
+       myid = messaging_server_id(msg_ctx);
 
        if (argc != 1) {
                fprintf(stderr, "Usage: smbcontrol <dest> dump-event-list\n");
@@ -1013,10 +1091,10 @@ static bool do_winbind_dump_domain_list(struct messaging_context *msg_ctx,
        uint8_t *buf = NULL;
        int buf_len = 0;
 
-       myid = procid_self();
+       myid = messaging_server_id(msg_ctx);
 
        if (argc < 1 || argc > 2) {
-               fprintf(stderr, "Usage: smbcontrol <dest> dump_domain_list "
+               fprintf(stderr, "Usage: smbcontrol <dest> dump-domain-list "
                        "<domain>\n");
                return false;
        }
@@ -1065,7 +1143,7 @@ static void winbind_validate_cache_cb(struct messaging_context *msg,
                                      struct server_id pid,
                                      DATA_BLOB *data)
 {
-       char *src_string = procid_str(NULL, &pid);
+       char *src_string = server_id_str(NULL, &pid);
        printf("Winbindd cache is %svalid. (answer from pid %s)\n",
               (*(data->data) == 0 ? "" : "NOT "), src_string);
        TALLOC_FREE(src_string);
@@ -1076,7 +1154,9 @@ static bool do_winbind_validate_cache(struct messaging_context *msg_ctx,
                                      const struct server_id pid,
                                      const int argc, const char **argv)
 {
-       struct server_id myid = procid_self();
+       struct server_id myid;
+
+       myid = messaging_server_id(msg_ctx);
 
        if (argc != 1) {
                fprintf(stderr, "Usage: smbcontrol winbindd validate-cache\n");
@@ -1122,7 +1202,7 @@ static void my_make_nmb_name( struct nmb_name *n, const char *name, int type)
        strupper_m(unix_name);
        push_ascii(n->name, unix_name, sizeof(n->name), STR_TERMINATE);
        n->name_type = (unsigned int)type & 0xFF;
-       push_ascii(n->scope,  global_scope(), 64, STR_TERMINATE);
+       push_ascii(n->scope,  lp_netbios_scope(), 64, STR_TERMINATE);
 }
 
 static bool do_nodestatus(struct messaging_context *msg_ctx,
@@ -1172,6 +1252,7 @@ static const struct {
        const char *help;       /* Short help text */
 } msg_types[] = {
        { "debug", do_debug, "Set debuglevel"  },
+       { "idmap", do_idmap, "Manipulate idmap cache" },
        { "force-election", do_election,
          "Force a browse election" },
        { "ping", do_ping, "Elicit a response" },
@@ -1184,6 +1265,7 @@ static const struct {
        { "debuglevel", do_debuglevel, "Display current debuglevels" },
        { "printnotify", do_printnotify, "Send a print notify message" },
        { "close-share", do_closeshare, "Forcibly disconnect a share" },
+       { "ip-dropped", do_ip_dropped, "Tell winbind that an IP got dropped" },
        { "lockretry", do_lockretry, "Force a blocking lock retry" },
        { "brl-revalidate", do_brl_revalidate, "Revalidate all brl entries" },
         { "samsync", do_samsync, "Initiate SAM synchronisation" },
@@ -1232,7 +1314,8 @@ static void usage(poptContext pc)
 
 /* Return the pid number for a string destination */
 
-static struct server_id parse_dest(const char *dest)
+static struct server_id parse_dest(struct messaging_context *msg,
+                                  const char *dest)
 {
        struct server_id result = {-1};
        pid_t pid;
@@ -1246,7 +1329,7 @@ static struct server_id parse_dest(const char *dest)
        /* Try self - useful for testing */
 
        if (strequal(dest, "self")) {
-               return procid_self();
+               return messaging_server_id(msg);
        }
 
        /* Fix winbind typo. */
@@ -1284,7 +1367,7 @@ static bool do_command(struct messaging_context *msg_ctx,
 
        /* Check destination */
 
-       pid = parse_dest(dest);
+       pid = parse_dest(msg_ctx, dest);
        if (!procid_valid(&pid)) {
                return False;
        }
@@ -1349,7 +1432,7 @@ int main(int argc, const char **argv)
 
        load_case_tables();
 
-       setup_logging(argv[0],True);
+       setup_logging(argv[0], DEBUG_STDOUT);
 
        /* Parse command line arguments using popt */