s3-build: avoid to globally include printing and spoolss headers.
[vlendec/samba-autobuild/.git] / source3 / utils / smbcontrol.c
index 750030d91664a12f2df164b7fcf44e6526310c08..ee96a02a6f89b502eb28a6baadb5f02b8d855881 100644 (file)
@@ -24,6 +24,9 @@
 */
 
 #include "includes.h"
+#include "librpc/gen_ndr/messaging.h"
+#include "librpc/gen_ndr/spoolss.h"
+#include "nt_printing.h"
 
 #if HAVE_LIBUNWIND_H
 #include <libunwind.h>
@@ -65,10 +68,10 @@ static bool send_message(struct messaging_context *msg_ctx,
        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 +83,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 +112,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 +129,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 +266,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 +297,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;
@@ -667,11 +675,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 +876,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 +906,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;
        }
 
@@ -952,7 +960,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 +992,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 +1012,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 +1075,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 +1137,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;
 
@@ -1237,7 +1245,7 @@ static struct server_id parse_dest(const char *dest)
        /* Try self - useful for testing */
 
        if (strequal(dest, "self")) {
-               return pid_to_procid(sys_getpid());
+               return procid_self();
        }
 
        /* Fix winbind typo. */
@@ -1245,15 +1253,12 @@ static struct server_id parse_dest(const char *dest)
                dest = "winbindd";
        }
 
-       if (!(strequal(dest, "winbindd") || strequal(dest, "nmbd"))) {
-               /* Check for numeric pid number */
+       /* Check for numeric pid number */
+       result = interpret_pid(dest);
 
-               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 */
@@ -1325,7 +1330,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[] = {
@@ -1388,8 +1393,8 @@ int main(int argc, const char **argv)
          * 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);