2 Unix SMB/Netbios implementation.
4 program to send control messages to Samba processes
5 Copyright (C) Andrew Tridgell 1994-1998
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, write to the Free Software
19 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 extern BOOL AllowDebugChange
;
31 {"force-election", MSG_FORCE_ELECTION
},
33 {"profile", MSG_PROFILE
},
34 {"profilelevel", MSG_REQ_PROFILELEVEL
},
35 {"debuglevel", MSG_REQ_DEBUGLEVEL
},
36 {"printer-notify", MSG_PRINTER_NOTIFY
},
37 {"close-share", MSG_SMB_FORCE_TDIS
},
45 static void usage(BOOL doexit
)
49 printf("Usage: smbcontrol -i -s configfile\n");
50 printf(" smbcontrol <destination> <message-type> <parameters>\n\n");
52 printf("<destination> <message-type> <parameters>\n\n");
54 printf("\t<destination> is one of \"nmbd\", \"smbd\" or a process ID\n");
55 printf("\t<message-type> is one of: ");
56 for (i
=0; msg_types
[i
].name
; i
++)
57 printf("%s%s", i
?", ":"",msg_types
[i
].name
);
62 static int pong_count
;
63 static BOOL got_level
;
64 static BOOL pong_registered
= False
;
65 static BOOL debuglevel_registered
= False
;
66 static BOOL profilelevel_registered
= False
;
69 /****************************************************************************
70 a useful function for testing the message system
71 ****************************************************************************/
72 void pong_function(int msg_type
, pid_t src
, void *buf
, size_t len
)
75 printf("PONG from PID %u\n",(unsigned int)src
);
78 /****************************************************************************
79 Prints out the current Debug level returned by MSG_DEBUGLEVEL
80 ****************************************************************************/
81 void debuglevel_function(int msg_type
, pid_t src
, void *buf
, size_t len
)
84 int debuglevel_class
[DBGC_LAST
];
86 memcpy(debuglevel_class
, buf
, len
);
88 printf("Current debug level of PID %u is %d ",(unsigned int)src
, debuglevel_class
[0]);
89 for (i
=1;i
<DBGC_LAST
;i
++)
90 if (debuglevel_class
[i
])
91 printf("%s:%d ", debug_classname_from_index(i
), debuglevel_class
[i
]);
97 /****************************************************************************
98 Prints out the current Profile level returned by MSG_PROFILELEVEL
99 ****************************************************************************/
100 void profilelevel_function(int msg_type
, pid_t src
, void *buf
, size_t len
)
104 memcpy(&level
, buf
, sizeof(int));
115 s
= "count and time";
118 printf("Profiling %s on PID %u\n",s
,(unsigned int)src
);
120 printf("Profiling not available on PID %u\n",(unsigned int)src
);
125 /****************************************************************************
126 send a message to a named destination
127 ****************************************************************************/
128 static BOOL
send_message(char *dest
, int msg_type
, void *buf
, int len
, BOOL duplicates
)
132 TDB_CONTEXT
*the_tdb
;
134 the_tdb
= tdb_open_log(lock_path("connections.tdb"), 0, TDB_DEFAULT
, O_RDWR
, 0);
136 fprintf(stderr
,"Failed to open connections database in send_message.\n");
140 /* "smbd" is the only broadcast operation */
141 if (strequal(dest
,"smbd")) {
142 retval
= message_send_all(the_tdb
,msg_type
, buf
, len
, duplicates
);
143 } else if (strequal(dest
,"nmbd")) {
144 pid
= pidfile_pid(dest
);
146 fprintf(stderr
,"Can't find pid for nmbd\n");
148 } else if (strequal(dest
,"self")) {
153 fprintf(stderr
,"Not a valid pid\n");
159 return message_send_pid(pid
, msg_type
, buf
, len
, duplicates
);
164 /****************************************************************************
165 evaluate a message type string
166 ****************************************************************************/
167 static int parse_type(char *mtype
)
170 for (i
=0;msg_types
[i
].name
;i
++) {
171 if (strequal(mtype
, msg_types
[i
].name
)) return msg_types
[i
].value
;
177 /****************************************************************************
179 ****************************************************************************/
180 static BOOL
do_command(char *dest
, char *msg_name
, int iparams
, char **params
)
185 int debuglevel_class
[DBGC_LAST
];
187 mtype
= parse_type(msg_name
);
189 fprintf(stderr
,"Couldn't resolve message type: %s\n", msg_name
);
195 if (!params
|| !params
[0]) {
196 fprintf(stderr
,"MSG_DEBUG needs a parameter\n");
200 ZERO_ARRAY(debuglevel_class
);
201 if (!debug_parse_params(params
, debuglevel_class
)) {
202 fprintf(stderr
, "MSG_DEBUG error. Expected <class name>:level\n");
205 send_message(dest
, MSG_DEBUG
, debuglevel_class
, sizeof(debuglevel_class
), False
);
209 if (!params
|| !params
[0]) {
210 fprintf(stderr
,"MSG_PROFILE needs a parameter\n");
213 if (strequal(params
[0], "off")) {
215 } else if (strequal(params
[0], "count")) {
217 } else if (strequal(params
[0], "on")) {
219 } else if (strequal(params
[0], "flush")) {
223 "MSG_PROFILE parameter must be off, count, on, or flush\n");
226 send_message(dest
, MSG_PROFILE
, &v
, sizeof(int), False
);
229 case MSG_FORCE_ELECTION
:
230 if (!strequal(dest
, "nmbd")) {
231 fprintf(stderr
,"force-election can only be sent to nmbd\n");
234 send_message(dest
, MSG_FORCE_ELECTION
, NULL
, 0, False
);
237 case MSG_REQ_PROFILELEVEL
:
238 if (!profilelevel_registered
) {
239 message_register(MSG_PROFILELEVEL
, profilelevel_function
);
240 profilelevel_registered
= True
;
243 retval
= send_message(dest
, MSG_REQ_PROFILELEVEL
, NULL
, 0, True
);
245 timeout_start
= time(NULL
);
248 if ((time(NULL
) - timeout_start
) > MAX_WAIT
) {
249 fprintf(stderr
,"profilelevel timeout\n");
256 case MSG_REQ_DEBUGLEVEL
:
257 if (!debuglevel_registered
) {
258 message_register(MSG_DEBUGLEVEL
, debuglevel_function
);
259 debuglevel_registered
= True
;
262 retval
= send_message(dest
, MSG_REQ_DEBUGLEVEL
, NULL
, 0, True
);
264 timeout_start
= time(NULL
);
267 if ((time(NULL
) - timeout_start
) > MAX_WAIT
) {
268 fprintf(stderr
,"debuglevel timeout\n");
275 case MSG_PRINTER_NOTIFY
:
276 if (!strequal(dest
, "smbd")) {
277 fprintf(stderr
,"printer-notify can only be sent to smbd\n");
280 if (!params
|| !params
[0]) {
281 fprintf(stderr
, "printer-notify needs a printer name\n");
284 retval
= send_message(dest
, MSG_PRINTER_NOTIFY
, params
[0],
285 strlen(params
[0]) + 1, False
);
288 case MSG_SMB_FORCE_TDIS
:
289 if (!strequal(dest
, "smbd")) {
290 fprintf(stderr
,"close-share can only be sent to smbd\n");
293 if (!params
|| !params
[0]) {
294 fprintf(stderr
, "close-share needs a share name or '*'\n");
297 retval
= send_message(dest
, MSG_SMB_FORCE_TDIS
, params
[0],
298 strlen(params
[0]) + 1, False
);
302 if (!pong_registered
) {
303 message_register(MSG_PONG
, pong_function
);
304 pong_registered
= True
;
306 if (!params
|| !params
[0]) {
307 fprintf(stderr
,"MSG_PING needs a parameter\n");
314 retval
= send_message(dest
, MSG_PING
, params
[1], strlen(params
[1]) + 1, True
);
316 retval
= send_message(dest
, MSG_PING
, NULL
, 0, True
);
317 if (retval
== False
) break;
320 timeout_start
= time(NULL
);
321 while (pong_count
< n
) {
323 if ((time(NULL
) - timeout_start
) > MAX_WAIT
) {
324 fprintf(stderr
,"PING timeout\n");
336 int main(int argc
, char *argv
[])
341 pstring servicesf
= CONFIGFILE
;
342 BOOL interactive
= False
;
345 setup_logging(argv
[0],True
);
347 AllowDebugChange
= False
;
349 charset_initialise();
351 if (argc
< 2) usage(True
);
353 while ((opt
= getopt(argc
, argv
,"is:")) != EOF
) {
359 pstrcpy(servicesf
, optarg
);
362 printf("Unknown option %c (%d)\n", (char)opt
, opt
);
367 lp_load(servicesf
,False
,False
,False
);
369 if (!message_init()) exit(1);
372 argv
= &argv
[optind
];
375 if (argc
< 2) usage(True
);
376 return (do_command(argv
[0],argv
[1], argc
-2, argc
> 2 ? &argv
[2] : 0));
383 printf("smbcontrol> ");
384 if (!fgets(temp
, sizeof(temp
)-1, stdin
)) break;
386 while ((myargc
< 4) &&
387 (myargv
[myargc
] = strtok(myargc
?NULL
:temp
," \t\n"))) {
391 if (strequal(myargv
[0],"q")) break;
394 else if (!do_command(myargv
[0],myargv
[1],myargc
-2,myargc
> 2 ? &myargv
[2] : 0))