s3-popt: Only include popt-common.h when needed.
[nivanova/samba-autobuild/.git] / source3 / utils / smbcontrol.c
index 6699763cd2d983624089f41d1eb567af9fa5b4cd..09f4f3146fb8480c3b9cf1916f128030c2b80571 100644 (file)
@@ -8,22 +8,26 @@
    Copyright (C) Martin Pool 2001-2002
    Copyright (C) Simo Sorce 2002
    Copyright (C) James Peach 2006
-   
+
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
-   
+
    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
+#include "popt_common.h"
+#include "librpc/gen_ndr/messaging.h"
+#include "librpc/gen_ndr/spoolss.h"
+#include "nt_printing.h"
 
 #if HAVE_LIBUNWIND_H
 #include <libunwind.h>
@@ -61,14 +65,14 @@ static bool send_message(struct messaging_context *msg_ctx,
        ret = message_send_all(msg_ctx, msg_type, buf, len, &n_sent);
        DEBUG(10,("smbcontrol/send_message: broadcast message to "
                  "%d processes\n", n_sent));
-       
+
        return ret;
 }
 
-static void timeout_handler(struct event_context *event_ctx,
-                           struct timed_event *te,
-                           const struct timeval *now,
-                           void *private_data)
+static void smbcontrol_timeout(struct tevent_context *event_ctx,
+                              struct tevent_timer *te,
+                              struct timeval now,
+                              void *private_data)
 {
        bool *timed_out = (bool *)private_data;
        TALLOC_FREE(te);
@@ -80,22 +84,24 @@ static void timeout_handler(struct event_context *event_ctx,
 static void wait_replies(struct messaging_context *msg_ctx,
                         bool multiple_replies)
 {
-       struct timed_event *te;
+       struct tevent_timer *te;
        bool timed_out = False;
 
-       if (!(te = event_add_timed(messaging_event_context(msg_ctx), NULL,
-                                  timeval_current_ofs(timeout, 0),
-                                  "smbcontrol_timeout",
-                                  timeout_handler, (void *)&timed_out))) {
-               DEBUG(0, ("event_add_timed failed\n"));
+       if (!(te = tevent_add_timer(messaging_event_context(msg_ctx), NULL,
+                                   timeval_current_ofs(timeout, 0),
+                                   smbcontrol_timeout, (void *)&timed_out))) {
+               DEBUG(0, ("tevent_add_timer failed\n"));
                return;
        }
 
        while (!timed_out) {
-               message_dispatch(msg_ctx);
+               int ret;
                if (num_replies > 0 && !multiple_replies)
                        break;
-               event_loop_once(messaging_event_context(msg_ctx));
+               ret = tevent_loop_once(messaging_event_context(msg_ctx));
+               if (ret != 0) {
+                       break;
+               }
        }
 }
 
@@ -107,8 +113,12 @@ static void print_pid_string_cb(struct messaging_context *msg,
                                struct server_id pid,
                                DATA_BLOB *data)
 {
-       printf("PID %u: %.*s", (unsigned int)procid_to_pid(&pid),
-              (int)data->length, (const char *)data->data);
+       char *pidstr;
+
+       pidstr = procid_str(talloc_tos(), &pid);
+       printf("PID %s: %.*s", pidstr, (int)data->length,
+              (const char *)data->data);
+       TALLOC_FREE(pidstr);
        num_replies++;
 }
 
@@ -120,7 +130,7 @@ static void print_string_cb(struct messaging_context *msg,
                            struct server_id pid,
                            DATA_BLOB *data)
 {
-       printf("%.*s", (int)data->length, (const char *)data->data);
+       printf("%*s", (int)data->length, (const char *)data->data);
        num_replies++;
 }
 
@@ -257,8 +267,7 @@ cleanup:
        ptrace(PTRACE_DETACH, pid, NULL, NULL);
 }
 
-static int stack_trace_connection(struct db_record *rec,
-                                 const struct connections_key *key,
+static int stack_trace_connection(const struct connections_key *key,
                                  const struct connections_data *crec,
                                  void *priv)
 {
@@ -289,7 +298,7 @@ static bool do_daemon_stack_trace(struct messaging_context *msg_ctx,
                 */
                print_stack_trace(dest, &count);
        } else {
-               connections_forall(stack_trace_connection, &count);
+               connections_forall_read(stack_trace_connection, &count);
        }
 
        return True;
@@ -474,7 +483,7 @@ static void profilelevel_cb(struct messaging_context *msg_ctx,
                s = "BOGUS";
                break;
        }
-       
+
        printf("Profiling %s on pid %u\n",s,(unsigned int)procid_to_pid(&pid));
 }
 
@@ -570,7 +579,7 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                fprintf(stderr, "\tjobdelete <printername> <unix jobid>\n");
                fprintf(stderr, "\tprinter <printername> <comment|port|"
                        "driver> <value>\n");
-               
+
                return False;
        }
 
@@ -583,7 +592,7 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                                " queuepause <printername>\n");
                        return False;
                }
-               
+
                notify_printer_status_byname(argv[2], PRINTER_STATUS_PAUSED);
 
                goto send;
@@ -595,7 +604,7 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                                " queuereume <printername>\n");
                        return False;
                }
-               
+
                notify_printer_status_byname(argv[2], PRINTER_STATUS_OK);
 
                goto send;
@@ -648,7 +657,7 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                notify_job_status_byname(
                        argv[2], jobid, JOB_STATUS_DELETING,
                        SPOOLSS_NOTIFY_MSG_UNIX_JOBID);
-               
+
                notify_job_status_byname(
                        argv[2], jobid, JOB_STATUS_DELETING|
                        JOB_STATUS_DELETED,
@@ -658,7 +667,7 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
 
        } else if (strcmp(cmd, "printer") == 0) {
                uint32 attribute;
-               
+
                if (argc != 5) {
                        fprintf(stderr, "Usage: smbcontrol <dest> printnotify "
                                "printer <printername> <comment|port|driver> "
@@ -667,11 +676,11 @@ static bool do_printnotify(struct messaging_context *msg_ctx,
                }
 
                if (strcmp(argv[3], "comment") == 0) {
-                       attribute = PRINTER_NOTIFY_COMMENT;
+                       attribute = PRINTER_NOTIFY_FIELD_COMMENT;
                } else if (strcmp(argv[3], "port") == 0) {
-                       attribute = PRINTER_NOTIFY_PORT_NAME;
+                       attribute = PRINTER_NOTIFY_FIELD_PORT_NAME;
                } else if (strcmp(argv[3], "driver") == 0) {
-                       attribute = PRINTER_NOTIFY_DRIVER_NAME;
+                       attribute = PRINTER_NOTIFY_FIELD_DRIVER_NAME;
                } else {
                        fprintf(stderr, "Invalid printer command '%s'\n",
                                argv[3]);
@@ -868,10 +877,10 @@ static bool do_winbind_online(struct messaging_context *msg_ctx,
        /* Remove the entry in the winbindd_cache tdb to tell a later
           starting winbindd that we're online. */
 
-       tdb = tdb_open_log(lock_path("winbindd_cache.tdb"), 0, TDB_DEFAULT, O_RDWR, 0600);
+       tdb = tdb_open_log(cache_path("winbindd_cache.tdb"), 0, TDB_DEFAULT, O_RDWR, 0600);
        if (!tdb) {
                fprintf(stderr, "Cannot open the tdb %s for writing.\n",
-                       lock_path("winbindd_cache.tdb"));
+                       cache_path("winbindd_cache.tdb"));
                return False;
        }
 
@@ -898,13 +907,13 @@ static bool do_winbind_offline(struct messaging_context *msg_ctx,
           starting winbindd that we're offline. We may actually create
           it here... */
 
-       tdb = tdb_open_log(lock_path("winbindd_cache.tdb"),
+       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);
 
        if (!tdb) {
                fprintf(stderr, "Cannot open the tdb %s for writing.\n",
-                       lock_path("winbindd_cache.tdb"));
+                       cache_path("winbindd_cache.tdb"));
                return False;
        }
 
@@ -932,7 +941,7 @@ static bool do_winbind_offline(struct messaging_context *msg_ctx,
 
                /* Check that the entry "WINBINDD_OFFLINE" still exists. */
                d = tdb_fetch_bystring( tdb, "WINBINDD_OFFLINE" );
-       
+
                if (!d.dptr || d.dsize != 4) {
                        SAFE_FREE(d.dptr);
                        DEBUG(10,("do_winbind_offline: offline state not set - retrying.\n"));
@@ -952,7 +961,7 @@ static bool do_winbind_onlinestatus(struct messaging_context *msg_ctx,
 {
        struct server_id myid;
 
-       myid = pid_to_procid(sys_getpid());
+       myid = procid_self();
 
        if (argc != 1) {
                fprintf(stderr, "Usage: smbcontrol winbindd onlinestatus\n");
@@ -984,7 +993,7 @@ static bool do_dump_event_list(struct messaging_context *msg_ctx,
 {
        struct server_id myid;
 
-       myid = pid_to_procid(sys_getpid());
+       myid = procid_self();
 
        if (argc != 1) {
                fprintf(stderr, "Usage: smbcontrol <dest> dump-event-list\n");
@@ -1004,7 +1013,7 @@ static bool do_winbind_dump_domain_list(struct messaging_context *msg_ctx,
        uint8_t *buf = NULL;
        int buf_len = 0;
 
-       myid = pid_to_procid(sys_getpid());
+       myid = procid_self();
 
        if (argc < 1 || argc > 2) {
                fprintf(stderr, "Usage: smbcontrol <dest> dump_domain_list "
@@ -1067,7 +1076,7 @@ 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 = pid_to_procid(sys_getpid());
+       struct server_id myid = procid_self();
 
        if (argc != 1) {
                fprintf(stderr, "Usage: smbcontrol winbindd validate-cache\n");
@@ -1129,7 +1138,7 @@ static bool do_nodestatus(struct messaging_context *msg_ctx,
 
        ZERO_STRUCT(p);
 
-       (void)interpret_addr2(&p.ip, argv[1]);
+       p.ip = interpret_addr2(argv[1]);
        p.port = 137;
        p.packet_type = NMB_PACKET;
 
@@ -1228,16 +1237,16 @@ static struct server_id parse_dest(const char *dest)
        struct server_id result = {-1};
        pid_t pid;
 
-       /* Zero is a special return value for broadcast smbd */
+       /* Zero is a special return value for broadcast to all processes */
 
-       if (strequal(dest, "smbd")) {
+       if (strequal(dest, "all")) {
                return interpret_pid(MSG_BROADCAST_PID_STR);
        }
 
        /* Try self - useful for testing */
 
        if (strequal(dest, "self")) {
-               return pid_to_procid(sys_getpid());
+               return procid_self();
        }
 
        /* Fix winbind typo. */
@@ -1245,16 +1254,12 @@ static struct server_id parse_dest(const char *dest)
                dest = "winbindd";
        }
 
-       
-       if (!(strequal(dest, "winbindd") || strequal(dest, "nmbd"))) {
-               /* Check for numeric pid number */
-
-               result = interpret_pid(dest);
+       /* Check for numeric pid number */
+       result = interpret_pid(dest);
 
-               /* Zero isn't valid if not smbd. */
-               if (result.pid && procid_valid(&result)) {
-                       return result;
-               }
+       /* Zero isn't valid if not "all". */
+       if (result.pid && procid_valid(&result)) {
+               return result;
        }
 
        /* Look up other destinations in pidfile directory */
@@ -1266,7 +1271,7 @@ static struct server_id parse_dest(const char *dest)
        fprintf(stderr,"Can't find pid for destination '%s'\n", dest);
 
        return result;
-}      
+}
 
 /* Execute smbcontrol command */
 
@@ -1326,7 +1331,7 @@ int main(int argc, const char **argv)
 {
        poptContext pc;
        int opt;
-       struct event_context *evt_ctx;
+       struct tevent_context *evt_ctx;
        struct messaging_context *msg_ctx;
 
        static struct poptOption long_options[] = {
@@ -1345,7 +1350,7 @@ int main(int argc, const char **argv)
        load_case_tables();
 
        setup_logging(argv[0],True);
-       
+
        /* Parse command line arguments using popt */
 
        pc = poptGetContext(
@@ -1388,14 +1393,14 @@ int main(int argc, const char **argv)
        /* Need to invert sense of return code -- samba
          * routines mostly return True==1 for success, but
          * shell needs 0. */ 
-       
-       if (!(evt_ctx = event_context_init(NULL)) ||
-           !(msg_ctx = messaging_init(NULL, server_id_self(), evt_ctx))) {
+
+       if (!(evt_ctx = tevent_context_init(NULL)) ||
+           !(msg_ctx = messaging_init(NULL, procid_self(), evt_ctx))) {
                fprintf(stderr, "could not init messaging context\n");
                TALLOC_FREE(frame);
                exit(1);
        }
-       
+
        ret = !do_command(msg_ctx, argc, argv);
        TALLOC_FREE(frame);
        return ret;