From 0174ccc0d6f34e84b201031a4c3eb58763a19770 Mon Sep 17 00:00:00 2001 From: Simo Sorce Date: Fri, 24 Aug 2001 07:51:59 +0000 Subject: let admins also abort a shutdown (This used to be commit 3b40ec4f149a8813c1d68f184858e2ddd605d8fd) --- source3/rpc_server/srv_reg_nt.c | 35 ++++++++++++++++++++++++++--------- 1 file changed, 26 insertions(+), 9 deletions(-) (limited to 'source3/rpc_server/srv_reg_nt.c') diff --git a/source3/rpc_server/srv_reg_nt.c b/source3/rpc_server/srv_reg_nt.c index a467e529cb..a77f329d4a 100644 --- a/source3/rpc_server/srv_reg_nt.c +++ b/source3/rpc_server/srv_reg_nt.c @@ -192,26 +192,27 @@ uint32 _reg_shutdown(pipes_struct *p, REG_Q_SHUTDOWN *q_u, REG_R_SHUTDOWN *r_u) pstring shutdown_script; UNISTR2 unimsg = q_u->uni_msg; pstring message; + pstring chkmsg; fstring timeout; fstring r; fstring f; + /* message */ rpcstr_pull (message, unimsg.buffer, sizeof(message), unimsg.uni_str_len*2,0); + /* security check */ + alpha_strcpy (chkmsg, message, NULL, sizeof(message); + /* timeout */ snprintf(timeout, sizeof(timeout), "%d", q_u->timeout); - if ((q_u->flags) & 0x100) /* reboot */ - snprintf(r, sizeof(r), SHUTDOWN_R_STRING); - if ((q_u->flags) & 0x001) /* force */ - snprintf(f, sizeof(f), SHUTDOWN_F_STRING); + /* reboot */ + snprintf(r, sizeof(r), (q_u->flags & 0x100)?SHUTDOWN_R_STRING:""); + /* force */ + snprintf(f, sizeof(f), (q_u->flags & 0x001)?SHUTDOWN_F_STRING:""); pstrcpy(shutdown_script, lp_shutdown_script()); - if (!*shutdown_script) { - pstrcpy(shutdown_script, lp_shutdown_script()); - } - if(*shutdown_script) { int shutdown_ret; - all_string_sub(shutdown_script, "%m", message, sizeof(shutdown_script)); + all_string_sub(shutdown_script, "%m", chkmsg, sizeof(shutdown_script)); all_string_sub(shutdown_script, "%t", timeout, sizeof(shutdown_script)); all_string_sub(shutdown_script, "%r", r, sizeof(shutdown_script)); all_string_sub(shutdown_script, "%f", f, sizeof(shutdown_script)); @@ -221,3 +222,19 @@ uint32 _reg_shutdown(pipes_struct *p, REG_Q_SHUTDOWN *q_u, REG_R_SHUTDOWN *r_u) return status; } + +uint32 _reg_abort_shutdown(pipes_struct *p, REG_Q_ABORT_SHUTDOWN *q_u, REG_R_ABORT_SHUTDOWN *r_u) +{ + uint32 status = NT_STATUS_NOPROBLEMO; + pstring abort_shutdown_script; + + pstrcpy(abort_shutdown_script, lp_abort_shutdown_script()); + + if(*abort_shutdown_script) { + int abort_shutdown_ret; + abort_shutdown_ret = smbrun(abort_shutdown_script,NULL); + DEBUG(3,("_reg_abort_shutdown: Running the command `%s' gave %d\n",abort_shutdown_script,abort_shutdown_ret)); + } + + return status; +} -- cgit