/* Samba Unix/Linux SMB client library Distributed SMB/CIFS Server Management Utility Copyright (C) Gerald (Jerry) Carter 2005 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 "utils/net.h" struct svc_state_msg { uint32 flag; const char *message; }; static struct svc_state_msg state_msg_table[] = { { SVCCTL_STOPPED, "stopped" }, { SVCCTL_START_PENDING, "start pending" }, { SVCCTL_STOP_PENDING, "stop pending" }, { SVCCTL_RUNNING, "running" }, { SVCCTL_CONTINUE_PENDING, "resume pending" }, { SVCCTL_PAUSE_PENDING, "pause pending" }, { SVCCTL_PAUSED, "paused" }, { 0, NULL } }; /******************************************************************** ********************************************************************/ const char *svc_status_string( uint32 state ) { fstring msg; int i; fstr_sprintf( msg, "Unknown State [%d]", state ); for ( i=0; state_msg_table[i].message; i++ ) { if ( state_msg_table[i].flag == state ) { fstrcpy( msg, state_msg_table[i].message ); break; } } return talloc_strdup(talloc_tos(), msg); } /******************************************************************** ********************************************************************/ static WERROR query_service_state(struct rpc_pipe_client *pipe_hnd, TALLOC_CTX *mem_ctx, POLICY_HND *hSCM, const char *service, uint32 *state ) { POLICY_HND hService; SERVICE_STATUS service_status; WERROR result = WERR_GENERAL_FAILURE; NTSTATUS status; /* now cycle until the status is actually 'watch_state' */ status = rpccli_svcctl_OpenServiceW(pipe_hnd, mem_ctx, hSCM, service, SC_RIGHT_SVC_QUERY_STATUS, &hService, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result) ) { d_fprintf(stderr, "Failed to open service. [%s]\n", dos_errstr(result)); return result; } status = rpccli_svcctl_QueryServiceStatus(pipe_hnd, mem_ctx, &hService, &service_status, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result) ) { *state = service_status.state; } rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hService, NULL); return result; } /******************************************************************** ********************************************************************/ static WERROR watch_service_state(struct rpc_pipe_client *pipe_hnd, TALLOC_CTX *mem_ctx, POLICY_HND *hSCM, const char *service, uint32 watch_state, uint32 *final_state ) { uint32 i; uint32 state = 0; WERROR result = WERR_GENERAL_FAILURE; i = 0; while ( (state != watch_state ) && i<30 ) { /* get the status */ result = query_service_state(pipe_hnd, mem_ctx, hSCM, service, &state ); if ( !W_ERROR_IS_OK(result) ) { break; } d_printf("."); i++; sys_usleep( 100 ); } d_printf("\n"); *final_state = state; return result; } /******************************************************************** ********************************************************************/ static WERROR control_service(struct rpc_pipe_client *pipe_hnd, TALLOC_CTX *mem_ctx, POLICY_HND *hSCM, const char *service, uint32 control, uint32 watch_state ) { POLICY_HND hService; WERROR result = WERR_GENERAL_FAILURE; NTSTATUS status; SERVICE_STATUS service_status; uint32 state = 0; /* Open the Service */ status = rpccli_svcctl_OpenServiceW(pipe_hnd, mem_ctx, hSCM, service, (SC_RIGHT_SVC_STOP|SC_RIGHT_SVC_PAUSE_CONTINUE), &hService, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result) ) { d_fprintf(stderr, "Failed to open service. [%s]\n", dos_errstr(result)); goto done; } /* get the status */ status = rpccli_svcctl_ControlService(pipe_hnd, mem_ctx, &hService, control, &service_status, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result) ) { d_fprintf(stderr, "Control service request failed. [%s]\n", dos_errstr(result)); goto done; } /* loop -- checking the state until we are where we want to be */ result = watch_service_state(pipe_hnd, mem_ctx, hSCM, service, watch_state, &state ); d_printf("%s service is %s.\n", service, svc_status_string(state)); done: rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hService, NULL); return result; } /******************************************************************** ********************************************************************/ static NTSTATUS rpc_service_list_internal(struct net_context *c, const DOM_SID *domain_sid, const char *domain_name, struct cli_state *cli, struct rpc_pipe_client *pipe_hnd, TALLOC_CTX *mem_ctx, int argc, const char **argv ) { POLICY_HND hSCM; ENUM_SERVICES_STATUS *services; WERROR result = WERR_GENERAL_FAILURE; NTSTATUS status; fstring servicename; fstring displayname; uint32 num_services = 0; int i; if (argc != 0 ) { d_printf("Usage: net rpc service list\n"); return NT_STATUS_OK; } status = rpccli_svcctl_OpenSCManagerW(pipe_hnd, mem_ctx, pipe_hnd->srv_name_slash, NULL, SC_RIGHT_MGR_ENUMERATE_SERVICE, &hSCM, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result)) { d_fprintf(stderr, "Failed to open Service Control Manager. [%s]\n", dos_errstr(result)); return werror_to_ntstatus(result); } result = rpccli_svcctl_enumerate_services(pipe_hnd, mem_ctx, &hSCM, SVCCTL_TYPE_WIN32, SVCCTL_STATE_ALL, &num_services, &services ); if ( !W_ERROR_IS_OK(result) ) { d_fprintf(stderr, "Failed to enumerate services. [%s]\n", dos_errstr(result)); goto done; } if ( num_services == 0 ) d_printf("No services returned\n"); for ( i=0; i<num_services; i++ ) { rpcstr_pull( servicename, services[i].servicename.buffer, sizeof(servicename), -1, STR_TERMINATE ); rpcstr_pull( displayname, services[i].displayname.buffer, sizeof(displayname), -1, STR_TERMINATE ); d_printf("%-20s \"%s\"\n", servicename, displayname); } done: rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hSCM, NULL); return werror_to_ntstatus(result); } /******************************************************************** ********************************************************************/ static NTSTATUS rpc_service_status_internal(struct net_context *c, const DOM_SID *domain_sid, const char *domain_name, struct cli_state *cli, struct rpc_pipe_client *pipe_hnd, TALLOC_CTX *mem_ctx, int argc, const char **argv ) { POLICY_HND hSCM, hService; WERROR result = WERR_GENERAL_FAILURE; NTSTATUS status; SERVICE_STATUS service_status; struct QUERY_SERVICE_CONFIG config; uint32_t buf_size = sizeof(config); uint32_t ret_size = 0; if (argc != 1 ) { d_printf("Usage: net rpc service status <service>\n"); return NT_STATUS_OK; } /* Open the Service Control Manager */ status = rpccli_svcctl_OpenSCManagerW(pipe_hnd, mem_ctx, pipe_hnd->srv_name_slash, NULL, SC_RIGHT_MGR_ENUMERATE_SERVICE, &hSCM, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result)) { d_fprintf(stderr, "Failed to open Service Control Manager. [%s]\n", dos_errstr(result)); return werror_to_ntstatus(result); } /* Open the Service */ status = rpccli_svcctl_OpenServiceW(pipe_hnd, mem_ctx, &hSCM, argv[0], (SC_RIGHT_SVC_QUERY_STATUS|SC_RIGHT_SVC_QUERY_CONFIG), &hService, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result) ) { d_fprintf(stderr, "Failed to open service. [%s]\n", dos_errstr(result)); goto done; } /* get the status */ status = rpccli_svcctl_QueryServiceStatus(pipe_hnd, mem_ctx, &hService, &service_status, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result) ) { d_fprintf(stderr, "Query status request failed. [%s]\n", dos_errstr(result)); goto done; } d_printf("%s service is %s.\n", argv[0], svc_status_string(service_status.state)); /* get the config */ status = rpccli_svcctl_QueryServiceConfigW(pipe_hnd, mem_ctx, &hService, &config, buf_size, &ret_size, &result); if (W_ERROR_EQUAL(result, WERR_INSUFFICIENT_BUFFER)) { buf_size = ret_size; status = rpccli_svcctl_QueryServiceConfigW(pipe_hnd, mem_ctx, &hService, &config, buf_size, &ret_size, &result); } if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result) ) { d_fprintf(stderr, "Query config request failed. [%s]\n", dos_errstr(result)); goto done; } /* print out the configuration information for the service */ d_printf("Configuration details:\n"); d_printf("\tControls Accepted = 0x%x\n", service_status.controls_accepted); d_printf("\tService Type = 0x%x\n", config.service_type); d_printf("\tStart Type = 0x%x\n", config.start_type); d_printf("\tError Control = 0x%x\n", config.error_control); d_printf("\tTag ID = 0x%x\n", config.tag_id); if (config.executablepath) { d_printf("\tExecutable Path = %s\n", config.executablepath); } if (config.loadordergroup) { d_printf("\tLoad Order Group = %s\n", config.loadordergroup); } if (config.dependencies) { d_printf("\tDependencies = %s\n", config.dependencies); } if (config.startname) { d_printf("\tStart Name = %s\n", config.startname); } if (config.displayname) { d_printf("\tDisplay Name = %s\n", config.displayname); } done: rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hService, NULL); rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hSCM, NULL); return werror_to_ntstatus(result); } /******************************************************************** ********************************************************************/ static NTSTATUS rpc_service_stop_internal(struct net_context *c, const DOM_SID *domain_sid, const char *domain_name, struct cli_state *cli, struct rpc_pipe_client *pipe_hnd, TALLOC_CTX *mem_ctx, int argc, const char **argv ) { POLICY_HND hSCM; WERROR result = WERR_GENERAL_FAILURE; NTSTATUS status; fstring servicename; if (argc != 1 ) { d_printf("Usage: net rpc service status <service>\n"); return NT_STATUS_OK; } fstrcpy( servicename, argv[0] ); /* Open the Service Control Manager */ status = rpccli_svcctl_OpenSCManagerW(pipe_hnd, mem_ctx, pipe_hnd->srv_name_slash, NULL, SC_RIGHT_MGR_ENUMERATE_SERVICE, &hSCM, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result)) { d_fprintf(stderr, "Failed to open Service Control Manager. [%s]\n", dos_errstr(result)); return werror_to_ntstatus(result); } result = control_service(pipe_hnd, mem_ctx, &hSCM, servicename, SVCCTL_CONTROL_STOP, SVCCTL_STOPPED ); rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hSCM, NULL); return werror_to_ntstatus(result); } /******************************************************************** ********************************************************************/ static NTSTATUS rpc_service_pause_internal(struct net_context *c, const DOM_SID *domain_sid, const char *domain_name, struct cli_state *cli, struct rpc_pipe_client *pipe_hnd, TALLOC_CTX *mem_ctx, int argc, const char **argv ) { POLICY_HND hSCM; WERROR result = WERR_GENERAL_FAILURE; NTSTATUS status; fstring servicename; if (argc != 1 ) { d_printf("Usage: net rpc service status <service>\n"); return NT_STATUS_OK; } fstrcpy( servicename, argv[0] ); /* Open the Service Control Manager */ status = rpccli_svcctl_OpenSCManagerW(pipe_hnd, mem_ctx, pipe_hnd->srv_name_slash, NULL, SC_RIGHT_MGR_ENUMERATE_SERVICE, &hSCM, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result)) { d_fprintf(stderr, "Failed to open Service Control Manager. [%s]\n", dos_errstr(result)); return werror_to_ntstatus(result); } result = control_service(pipe_hnd, mem_ctx, &hSCM, servicename, SVCCTL_CONTROL_PAUSE, SVCCTL_PAUSED ); rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hSCM, NULL); return werror_to_ntstatus(result); } /******************************************************************** ********************************************************************/ static NTSTATUS rpc_service_resume_internal(struct net_context *c, const DOM_SID *domain_sid, const char *domain_name, struct cli_state *cli, struct rpc_pipe_client *pipe_hnd, TALLOC_CTX *mem_ctx, int argc, const char **argv ) { POLICY_HND hSCM; WERROR result = WERR_GENERAL_FAILURE; NTSTATUS status; fstring servicename; if (argc != 1 ) { d_printf("Usage: net rpc service status <service>\n"); return NT_STATUS_OK; } fstrcpy( servicename, argv[0] ); /* Open the Service Control Manager */ status = rpccli_svcctl_OpenSCManagerW(pipe_hnd, mem_ctx, pipe_hnd->srv_name_slash, NULL, SC_RIGHT_MGR_ENUMERATE_SERVICE, &hSCM, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result)) { d_fprintf(stderr, "Failed to open Service Control Manager. [%s]\n", dos_errstr(result)); return werror_to_ntstatus(result); } result = control_service(pipe_hnd, mem_ctx, &hSCM, servicename, SVCCTL_CONTROL_CONTINUE, SVCCTL_RUNNING ); rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hSCM, NULL); return werror_to_ntstatus(result); } /******************************************************************** ********************************************************************/ static NTSTATUS rpc_service_start_internal(struct net_context *c, const DOM_SID *domain_sid, const char *domain_name, struct cli_state *cli, struct rpc_pipe_client *pipe_hnd, TALLOC_CTX *mem_ctx, int argc, const char **argv ) { POLICY_HND hSCM, hService; WERROR result = WERR_GENERAL_FAILURE; NTSTATUS status; uint32 state = 0; if (argc != 1 ) { d_printf("Usage: net rpc service status <service>\n"); return NT_STATUS_OK; } /* Open the Service Control Manager */ status = rpccli_svcctl_OpenSCManagerW(pipe_hnd, mem_ctx, pipe_hnd->srv_name_slash, NULL, SC_RIGHT_MGR_ENUMERATE_SERVICE, &hSCM, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result)) { d_fprintf(stderr, "Failed to open Service Control Manager. [%s]\n", dos_errstr(result)); return werror_to_ntstatus(result); } /* Open the Service */ status = rpccli_svcctl_OpenServiceW(pipe_hnd, mem_ctx, &hSCM, argv[0], SC_RIGHT_SVC_START, &hService, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result) ) { d_fprintf(stderr, "Failed to open service. [%s]\n", dos_errstr(result)); goto done; } /* get the status */ status = rpccli_svcctl_StartServiceW(pipe_hnd, mem_ctx, &hService, 0, NULL, &result); if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(result) ) { d_fprintf(stderr, "Query status request failed. [%s]\n", dos_errstr(result)); goto done; } result = watch_service_state(pipe_hnd, mem_ctx, &hSCM, argv[0], SVCCTL_RUNNING, &state ); if ( W_ERROR_IS_OK(result) && (state == SVCCTL_RUNNING) ) d_printf("Successfully started service: %s\n", argv[0] ); else d_fprintf(stderr, "Failed to start service: %s [%s]\n", argv[0], dos_errstr(result) ); done: rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hService, NULL); rpccli_svcctl_CloseServiceHandle(pipe_hnd, mem_ctx, &hSCM, NULL); return werror_to_ntstatus(result); } /******************************************************************** ********************************************************************/ static int rpc_service_list(struct net_context *c, int argc, const char **argv ) { if (c->display_usage) { d_printf("Usage:\n" "net rpc service list\n" " View configured Win32 services\n"); return 0; } return run_rpc_command(c, NULL, &ndr_table_svcctl.syntax_id, 0, rpc_service_list_internal, argc, argv ); } /******************************************************************** ********************************************************************/ static int rpc_service_start(struct net_context *c, int argc, const char **argv ) { if (c->display_usage) { d_printf("Usage:\n" "net rpc service start <service>\n" " Start a Win32 service\n"); return 0; } return run_rpc_command(c, NULL, &ndr_table_svcctl.syntax_id, 0, rpc_service_start_internal, argc, argv ); } /******************************************************************** ********************************************************************/ static int rpc_service_stop(struct net_context *c, int argc, const char **argv ) { if (c->display_usage) { d_printf("Usage:\n" "net rpc service stop <service>\n" " Stop a Win32 service\n"); return 0; } return run_rpc_command(c, NULL, &ndr_table_svcctl.syntax_id, 0, rpc_service_stop_internal, argc, argv ); } /******************************************************************** ********************************************************************/ static int rpc_service_resume(struct net_context *c, int argc, const char **argv ) { if (c->display_usage) { d_printf("Usage:\n" "net rpc service resume <service>\n" " Resume a Win32 service\n"); return 0; } return run_rpc_command(c, NULL, &ndr_table_svcctl.syntax_id, 0, rpc_service_resume_internal, argc, argv ); } /******************************************************************** ********************************************************************/ static int rpc_service_pause(struct net_context *c, int argc, const char **argv ) { if (c->display_usage) { d_printf("Usage:\n" "net rpc service pause <service>\n" " Pause a Win32 service\n"); return 0; } return run_rpc_command(c, NULL, &ndr_table_svcctl.syntax_id, 0, rpc_service_pause_internal, argc, argv ); } /******************************************************************** ********************************************************************/ static int rpc_service_status(struct net_context *c, int argc, const char **argv ) { if (c->display_usage) { d_printf("Usage:\n" "net rpc service status <service>\n" " Show the current status of a service\n"); return 0; } return run_rpc_command(c, NULL, &ndr_table_svcctl.syntax_id, 0, rpc_service_status_internal, argc, argv ); } /******************************************************************** ********************************************************************/ int net_rpc_service(struct net_context *c, int argc, const char **argv) { struct functable func[] = { { "list", rpc_service_list, NET_TRANSPORT_RPC, "View configured Win32 services", "net rpc service list\n" " View configured Win32 services" }, { "start", rpc_service_start, NET_TRANSPORT_RPC, "Start a service", "net rpc service start\n" " Start a service" }, { "stop", rpc_service_stop, NET_TRANSPORT_RPC, "Stop a service", "net rpc service stop\n" " Stop a service" }, { "pause", rpc_service_pause, NET_TRANSPORT_RPC, "Pause a service", "net rpc service pause\n" " Pause a service" }, { "resume", rpc_service_resume, NET_TRANSPORT_RPC, "Resume a paused service", "net rpc service resume\n" " Resume a service" }, { "status", rpc_service_status, NET_TRANSPORT_RPC, "View current status of a service", "net rpc service status\n" " View current status of a service" }, {NULL, NULL, 0, NULL, NULL} }; return net_run_function(c, argc, argv, "net rpc service",func); }