s3-popt: Only include popt-common.h when needed.
[nivanova/samba-autobuild/.git] / source3 / utils / smbcontrol.c
index 6ea200bfec5dcfb3d2771f101c5e851bc9a8d196..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,7 +65,7 @@ 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;
 }
 
@@ -109,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++;
 }
 
@@ -122,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++;
 }
 
@@ -259,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)
 {
@@ -291,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;
@@ -476,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));
 }
 
@@ -572,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;
        }
 
@@ -585,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;
@@ -597,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;
@@ -650,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,
@@ -660,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> "
@@ -669,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]);
@@ -934,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"));
@@ -954,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");
@@ -986,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");
@@ -1006,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 "
@@ -1069,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");
@@ -1239,7 +1246,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. */
@@ -1247,15 +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 */
@@ -1346,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(
@@ -1389,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 = tevent_context_init(NULL)) ||
-           !(msg_ctx = messaging_init(NULL, server_id_self(), evt_ctx))) {
+           !(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;