2 Unix SMB/CIFS implementation.
4 Send messages to other Samba daemons
6 Copyright (C) Tim Potter 2003
7 Copyright (C) Andrew Tridgell 1994-1998
8 Copyright (C) Martin Pool 2001-2002
9 Copyright (C) Simo Sorce 2002
11 This program is free software; you can redistribute it and/or modify
12 it under the terms of the GNU General Public License as published by
13 the Free Software Foundation; either version 2 of the License, or
14 (at your option) any later version.
16 This program is distributed in the hope that it will be useful,
17 but WITHOUT ANY WARRANTY; without even the implied warranty of
18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 GNU General Public License for more details.
21 You should have received a copy of the GNU General Public License
22 along with this program; if not, write to the Free Software
23 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
28 /* Default timeout value when waiting for replies (in seconds) */
30 #define DEFAULT_TIMEOUT 10
32 static int timeout = DEFAULT_TIMEOUT;
33 static int num_replies; /* Used by message callback fns */
35 /* Send a message to a destination pid. Zero means broadcast smbd. */
37 static BOOL send_message(pid_t pid, int msg_type, const void *buf, int len,
48 return message_send_pid(pid, msg_type, buf, len, duplicates);
50 tdb = tdb_open_log(lock_path("connections.tdb"), 0,
51 TDB_DEFAULT, O_RDWR, 0);
53 fprintf(stderr,"Failed to open connections database"
54 ": %s\n", strerror(errno));
58 ret = message_send_all(tdb,msg_type, buf, len, duplicates,
60 DEBUG(10,("smbcontrol/send_message: broadcast message to "
61 "%d processes\n", n_sent));
68 /* Wait for one or more reply messages */
70 static void wait_replies(BOOL multiple_replies)
72 time_t start_time = time(NULL);
74 /* Wait around a bit. This is pretty disgusting - we have to
75 busy-wait here as there is no nicer way to do it. */
79 if (num_replies > 0 && !multiple_replies)
82 } while (timeout - (time(NULL) - start_time) > 0);
85 /* Message handler callback that displays a string on stdout */
87 static void print_string_cb(int msg_type, pid_t pid, void *buf, size_t len)
89 printf("%.*s", (int)len, (const char *)buf);
93 /* Send no message. Useful for testing. */
95 static BOOL do_noop(const pid_t pid, const int argc, const char **argv)
98 fprintf(stderr, "Usage: smbcontrol <dest> noop\n");
102 /* Move along, nothing to see here */
107 /* Send a debug string */
109 static BOOL do_debug(const pid_t pid, const int argc, const char **argv)
112 fprintf(stderr, "Usage: smbcontrol <dest> debug "
118 pid, MSG_DEBUG, argv[1], strlen(argv[1]) + 1, False);
121 /* Force a browser election */
123 static BOOL do_election(const pid_t pid, const int argc, const char **argv)
126 fprintf(stderr, "Usage: smbcontrol <dest> force-election\n");
131 pid, MSG_FORCE_ELECTION, NULL, 0, False);
134 /* Ping a samba daemon process */
136 static void pong_cb(int msg_type, pid_t pid, void *buf, size_t len)
138 printf("PONG from pid %u\n", (unsigned int)pid);
142 static BOOL do_ping(const pid_t pid, const int argc, const char **argv)
145 fprintf(stderr, "Usage: smbcontrol <dest> ping\n");
149 /* Send a message and register our interest in a reply */
151 if (!send_message(pid, MSG_PING, NULL, 0, False))
154 message_register(MSG_PONG, pong_cb);
156 wait_replies(pid == 0);
158 /* No replies were received within the timeout period */
160 if (num_replies == 0)
161 printf("No replies received\n");
163 message_deregister(MSG_PONG);
168 /* Set profiling options */
170 static BOOL do_profile(const pid_t pid, const int argc, const char **argv)
175 fprintf(stderr, "Usage: smbcontrol <dest> profile "
176 "<off|count|on|flush>\n");
180 if (strcmp(argv[1], "off") == 0) {
182 } else if (strcmp(argv[1], "count") == 0) {
184 } else if (strcmp(argv[1], "on") == 0) {
186 } else if (strcmp(argv[1], "flush") == 0) {
189 fprintf(stderr, "Unknown profile command '%s'\n", argv[1]);
193 return send_message(pid, MSG_PROFILE, &v, sizeof(int), False);
196 /* Return the profiling level */
198 static void profilelevel_cb(int msg_type, pid_t pid, void *buf, size_t len)
205 if (len != sizeof(int)) {
206 fprintf(stderr, "invalid message length %d returned\n", len);
210 memcpy(&level, buf, sizeof(int));
223 s = "count and time";
230 printf("Profiling %s on pid %u\n",s,(unsigned int)pid);
233 static void profilelevel_rqst(int msg_type, pid_t pid, void *buf, size_t len)
237 /* Send back a dummy reply */
239 send_message(pid, MSG_PROFILELEVEL, &v, sizeof(int), False);
242 static BOOL do_profilelevel(const pid_t pid, const int argc, const char **argv)
245 fprintf(stderr, "Usage: smbcontrol <dest> profilelevel\n");
249 /* Send a message and register our interest in a reply */
251 if (!send_message(pid, MSG_REQ_PROFILELEVEL, NULL, 0, False))
254 message_register(MSG_PROFILELEVEL, profilelevel_cb);
255 message_register(MSG_REQ_PROFILELEVEL, profilelevel_rqst);
257 wait_replies(pid == 0);
259 /* No replies were received within the timeout period */
261 if (num_replies == 0)
262 printf("No replies received\n");
264 message_deregister(MSG_PROFILE);
269 /* Display debug level settings */
271 static BOOL do_debuglevel(const pid_t pid, const int argc, const char **argv)
274 fprintf(stderr, "Usage: smbcontrol <dest> debuglevel\n");
278 /* Send a message and register our interest in a reply */
280 if (!send_message(pid, MSG_REQ_DEBUGLEVEL, NULL, 0, False))
283 message_register(MSG_DEBUGLEVEL, print_string_cb);
285 wait_replies(pid == 0);
287 /* No replies were received within the timeout period */
289 if (num_replies == 0)
290 printf("No replies received\n");
292 message_deregister(MSG_DEBUGLEVEL);
297 /* Send a print notify message */
299 static BOOL do_printnotify(const pid_t pid, const int argc, const char **argv)
303 /* Check for subcommand */
306 fprintf(stderr, "Must specify subcommand:\n");
307 fprintf(stderr, "\tqueuepause <printername>\n");
308 fprintf(stderr, "\tqueueresume <printername>\n");
309 fprintf(stderr, "\tjobpause <printername> <unix jobid>\n");
310 fprintf(stderr, "\tjobresume <printername> <unix jobid>\n");
311 fprintf(stderr, "\tjobdelete <printername> <unix jobid>\n");
312 fprintf(stderr, "\tprinter <printername> <comment|port|"
313 "driver> <value>\n");
320 if (strcmp(cmd, "queuepause") == 0) {
323 fprintf(stderr, "Usage: smbcontrol <dest> printnotify"
324 " queuepause <printername>\n");
328 notify_printer_status_byname(argv[2], PRINTER_STATUS_PAUSED);
332 } else if (strcmp(cmd, "queueresume") == 0) {
335 fprintf(stderr, "Usage: smbcontrol <dest> printnotify"
336 " queuereume <printername>\n");
340 notify_printer_status_byname(argv[2], PRINTER_STATUS_OK);
344 } else if (strcmp(cmd, "jobpause") == 0) {
348 fprintf(stderr, "Usage: smbcontrol <dest> printnotify"
349 " jobpause <printername> <unix-jobid>\n");
353 jobid = atoi(argv[3]);
355 notify_job_status_byname(
356 argv[2], jobid, JOB_STATUS_PAUSED,
357 SPOOLSS_NOTIFY_MSG_UNIX_JOBID);
361 } else if (strcmp(cmd, "jobresume") == 0) {
365 fprintf(stderr, "Usage: smbcontrol <dest> printnotify"
366 " jobpause <printername> <unix-jobid>\n");
370 jobid = atoi(argv[3]);
372 notify_job_status_byname(
373 argv[2], jobid, JOB_STATUS_QUEUED,
374 SPOOLSS_NOTIFY_MSG_UNIX_JOBID);
378 } else if (strcmp(cmd, "jobdelete") == 0) {
382 fprintf(stderr, "Usage: smbcontrol <dest> printnotify"
383 " jobpause <printername> <unix-jobid>\n");
387 jobid = atoi(argv[3]);
389 notify_job_status_byname(
390 argv[2], jobid, JOB_STATUS_DELETING,
391 SPOOLSS_NOTIFY_MSG_UNIX_JOBID);
393 notify_job_status_byname(
394 argv[2], jobid, JOB_STATUS_DELETING|
396 SPOOLSS_NOTIFY_MSG_UNIX_JOBID);
400 } else if (strcmp(cmd, "printer") == 0) {
404 fprintf(stderr, "Usage: smbcontrol <dest> printnotify "
405 "printer <printername> <comment|port|driver> "
410 if (strcmp(argv[3], "comment") == 0) {
411 attribute = PRINTER_NOTIFY_COMMENT;
412 } else if (strcmp(argv[3], "port") == 0) {
413 attribute = PRINTER_NOTIFY_PORT_NAME;
414 } else if (strcmp(argv[3], "driver") == 0) {
415 attribute = PRINTER_NOTIFY_DRIVER_NAME;
417 fprintf(stderr, "Invalid printer command '%s'\n",
422 notify_printer_byname(argv[2], attribute, argv[4]);
427 fprintf(stderr, "Invalid subcommand '%s'\n", cmd);
431 print_notify_send_messages(0);
437 static BOOL do_closeshare(const pid_t pid, const int argc, const char **argv)
440 fprintf(stderr, "Usage: smbcontrol <dest> close-share "
446 pid, MSG_SMB_FORCE_TDIS, argv[1], strlen(argv[1]) + 1, False);
449 /* Force a SAM synchronisation */
451 static BOOL do_samsync(const pid_t pid, const int argc, const char **argv)
454 fprintf(stderr, "Usage: smbcontrol <dest> samsync\n");
459 pid, MSG_SMB_SAM_SYNC, NULL, 0, False);
462 /* Force a SAM replication */
464 static BOOL do_samrepl(const pid_t pid, const int argc, const char **argv)
467 fprintf(stderr, "Usage: smbcontrol <dest> samrepl\n");
472 pid, MSG_SMB_SAM_REPL, NULL, 0, False);
475 /* Display talloc pool usage */
477 static BOOL do_poolusage(const pid_t pid, const int argc, const char **argv)
480 fprintf(stderr, "Usage: smbcontrol <dest> pool-usage\n");
484 /* Send a message and register our interest in a reply */
486 if (!send_message(pid, MSG_REQ_POOL_USAGE, NULL, 0, False))
489 message_register(MSG_POOL_USAGE, print_string_cb);
491 wait_replies(pid == 0);
493 /* No replies were received within the timeout period */
495 if (num_replies == 0)
496 printf("No replies received\n");
498 message_deregister(MSG_POOL_USAGE);
503 /* Perform a dmalloc mark */
505 static BOOL do_dmalloc_mark(const pid_t pid, const int argc, const char **argv)
508 fprintf(stderr, "Usage: smbcontrol <dest> dmalloc-mark\n");
513 pid, MSG_REQ_DMALLOC_MARK, NULL, 0, False);
516 /* Perform a dmalloc changed */
518 static BOOL do_dmalloc_changed(const pid_t pid, const int argc, const char **argv)
521 fprintf(stderr, "Usage: smbcontrol <dest> "
522 "dmalloc-log-changed\n");
527 pid, MSG_REQ_DMALLOC_LOG_CHANGED, NULL, 0, False);
530 /* Shutdown a server process */
532 static BOOL do_shutdown(const pid_t pid, const int argc, const char **argv)
535 fprintf(stderr, "Usage: smbcontrol <dest> shutdown\n");
539 return send_message(pid, MSG_SHUTDOWN, NULL, 0, False);
542 /* Notify a driver upgrade */
544 static BOOL do_drvupgrade(const pid_t pid, const int argc, const char **argv)
547 fprintf(stderr, "Usage: smbcontrol <dest> drvupgrade "
553 pid, MSG_DEBUG, argv[1], strlen(argv[1]) + 1, False);
556 static BOOL do_reload_config(const pid_t pid, const int argc, const char **argv)
559 fprintf(stderr, "Usage: smbcontrol <dest> reload-config\n");
563 return send_message(pid, MSG_SMB_CONF_UPDATED, NULL, 0, False);
566 /* A list of message type supported */
568 static const struct {
569 const char *name; /* Option name */
570 BOOL (*fn)(const pid_t pid, const int argc, const char **argv);
571 const char *help; /* Short help text */
573 { "debug", do_debug, "Set debuglevel" },
574 { "force-election", do_election,
575 "Force a browse election" },
576 { "ping", do_ping, "Elicit a response" },
577 { "profile", do_profile, "" },
578 { "profilelevel", do_profilelevel, "" },
579 { "debuglevel", do_debuglevel, "Display current debuglevels" },
580 { "printnotify", do_printnotify, "Send a print notify message" },
581 { "close-share", do_closeshare, "Forcibly disconnect a share" },
582 { "samsync", do_samsync, "Initiate SAM synchronisation" },
583 { "samrepl", do_samrepl, "Initiate SAM replication" },
584 { "pool-usage", do_poolusage, "Display talloc memory usage" },
585 { "dmalloc-mark", do_dmalloc_mark, "" },
586 { "dmalloc-log-changed", do_dmalloc_changed, "" },
587 { "shutdown", do_shutdown, "Shut down daemon" },
588 { "drvupgrade", do_drvupgrade, "Notify a printer driver has changed" },
589 { "reload-config", do_reload_config, "Force smbd or winbindd to reload config file"},
590 { "noop", do_noop, "Do nothing" },
594 /* Display usage information */
596 static void usage(poptContext *pc)
600 poptPrintHelp(*pc, stderr, 0);
602 fprintf(stderr, "\n");
603 fprintf(stderr, "<destination> is one of \"nmbd\", \"smbd\" or a "
606 fprintf(stderr, "\n");
607 fprintf(stderr, "<message-type> is one of:\n");
609 for (i = 0; msg_types[i].name; i++)
610 fprintf(stderr, "\t%-30s%s\n", msg_types[i].name,
613 fprintf(stderr, "\n");
618 /* Return the pid number for a string destination */
620 static pid_t parse_dest(const char *dest)
624 /* Zero is a special return value for broadcast smbd */
626 if (strequal(dest, "smbd"))
629 /* Try self - useful for testing */
631 if (strequal(dest, "self"))
634 /* Check for numeric pid number */
636 if ((pid = atoi(dest)) != 0)
639 /* Look up other destinations in pidfile directory */
641 if ((pid = pidfile_pid(dest)) != 0)
644 fprintf(stderr,"Can't find pid for destination '%s'\n", dest);
649 /* Execute smbcontrol command */
651 static BOOL do_command(int argc, const char **argv)
653 const char *dest = argv[0], *command = argv[1];
657 /* Check destination */
659 if ((pid = parse_dest(dest)) == -1)
664 for (i = 0; msg_types[i].name; i++) {
665 if (strequal(command, msg_types[i].name))
666 return msg_types[i].fn(pid, argc - 1, argv + 1);
669 fprintf(stderr, "smbcontrol: unknown command '%s'\n", command);
676 int main(int argc, const char **argv)
681 static struct poptOption wbinfo_options[] = {
682 { "timeout", 't', POPT_ARG_INT, &timeout, 't',
683 "Set timeout value in seconds", "TIMEOUT" },
685 { "configfile", 's', POPT_ARG_STRING, NULL, 's',
686 "Use alternative configuration file", "CONFIGFILE" },
691 struct poptOption options[] = {
692 { NULL, 0, POPT_ARG_INCLUDE_TABLE, wbinfo_options, 0,
700 setup_logging(argv[0],True);
702 /* Parse command line arguments using popt */
705 "smbcontrol", argc, (const char **)argv, options, 0);
707 poptSetOtherOptionHelp(pc, "[OPTION...] <destination> <message-type> "
713 while ((opt = poptGetNextOpt(pc)) != -1) {
715 case 't': /* --timeout */
718 case 's': /* --configfile */
719 pstrcpy(dyn_CONFIGFILE, optarg);
723 fprintf(stderr, "Invalid option\n");
724 poptPrintHelp(pc, stderr, 0);
729 /* We should now have the remaining command line arguments in
730 argv. The argc parameter should have been decremented to the
731 correct value in the above switch statement. */
733 argv = (const char **)poptGetArgs(pc);
734 argc--; /* Don't forget about argv[0] */
739 lp_load(dyn_CONFIGFILE,False,False,False);
741 /* Need to invert sense of return code -- samba
742 * routines mostly return True==1 for success, but
745 return !do_command(argc, argv);