windows Namespace Reference

Functions
bool	check_admin (bool &result)
bool	ensure_admin (std::string const &arguments)
bool	install_service (std::string const &service_name, std::string const &arguments)
bool	start_service (std::string const &service_name)
bool	stop_service (std::string const &service_name)
bool	uninstall_service (std::string const &service_name)

Function Documentation

check_admin()

bool windows::check_admin

(

bool &

result

)

Definition at line 112 of file windows_service.cpp.

{
  BOOL is_admin = FALSE;
  PSID p_administrators_group = nullptr;
 
  SID_IDENTIFIER_AUTHORITY nt_authority = SECURITY_NT_AUTHORITY;
 
  if (!AllocateAndInitializeSid(
        &nt_authority
      , 2
      , SECURITY_BUILTIN_DOMAIN_RID
      , DOMAIN_ALIAS_RID_ADMINS
      , 0, 0, 0, 0, 0, 0
      , &p_administrators_group
      ))
  {
    tools::fail_msg_writer() << "Security Identifier creation failed: " << get_last_error();
    return false;
  }
 
  if (!CheckTokenMembership(
        nullptr
      , p_administrators_group
      , &is_admin
      ))
  {
    tools::fail_msg_writer() << "Permissions check failed: " << get_last_error();
    return false;
  }
 
  result = is_admin ? true : false;
 
  return true;
}

Here is the call graph for this function:

Here is the caller graph for this function:

ensure_admin()

bool windows::ensure_admin

(

std::string const &

arguments

)

Definition at line 147 of file windows_service.cpp.

{
  bool admin;
 
  if (!check_admin(admin))
  {
    return false;
  }
 
  if (admin)
  {
    return true;
  }
  else
  {
    std::string command = epee::string_tools::get_current_module_path();
    relaunch_as_admin(command, arguments);
    return false;
  }
}

Here is the call graph for this function:

install_service()

bool windows::install_service	(	std::string const &	service_name,
		std::string const &	arguments
	)

Definition at line 170 of file windows_service.cpp.

{
  std::string command = epee::string_tools::get_current_module_path();
  std::string full_command = command + arguments;
 
  service_handle p_manager{
    OpenSCManager(
        nullptr
      , nullptr
      , SC_MANAGER_CONNECT | SC_MANAGER_CREATE_SERVICE
      )
  , &::CloseServiceHandle
  };
  if (p_manager == nullptr)
  {
    tools::fail_msg_writer() << "Couldn't connect to service manager: " << get_last_error();
    return false;
  }
 
  service_handle p_service{
    CreateService(
        p_manager.get()
      , service_name.c_str()
      , service_name.c_str()
      , 0
      //, GENERIC_EXECUTE | GENERIC_READ
      , SERVICE_WIN32_OWN_PROCESS
      , SERVICE_DEMAND_START
      , SERVICE_ERROR_NORMAL
      , full_command.c_str()
      , nullptr
      , nullptr
      , ""
      //, "NT AUTHORITY\\LocalService"
      , nullptr // Implies LocalSystem account
      , nullptr
      )
  , &::CloseServiceHandle
  };
  if (p_service == nullptr)
  {
    tools::fail_msg_writer() << "Couldn't create service: " << get_last_error();
    return false;
  }
 
  tools::success_msg_writer() << "Service installed";
 
  pause_to_display_admin_window_messages();
 
  return true;
}

Here is the call graph for this function:

start_service()

bool windows::start_service

(

std::string const &

service_name

)

Definition at line 225 of file windows_service.cpp.

{
  tools::msg_writer() << "Starting service";
 
  SERVICE_STATUS_PROCESS service_status = {};
  DWORD unused = 0;
 
  service_handle p_manager{
    OpenSCManager(
        nullptr
      , nullptr
      , SC_MANAGER_CONNECT
      )
  , &::CloseServiceHandle
  };
  if (p_manager == nullptr)
  {
    tools::fail_msg_writer() << "Couldn't connect to service manager: " << get_last_error();
    return false;
  }
 
  service_handle p_service{
    OpenService(
        p_manager.get()
      , service_name.c_str()
      //, SERVICE_START | SERVICE_QUERY_STATUS
      , SERVICE_START
      )
  , &::CloseServiceHandle
  };
  if (p_service == nullptr)
  {
    tools::fail_msg_writer() << "Couldn't find service: " << get_last_error();
    return false;
  }
 
  if (!StartService(
      p_service.get()
    , 0
    , nullptr
    ))
  {
    tools::fail_msg_writer() << "Service start request failed: " << get_last_error();
    return false;
  }
 
  tools::success_msg_writer() << "Service started";
 
  pause_to_display_admin_window_messages();
 
  return true;
}

Here is the call graph for this function:

stop_service()

bool windows::stop_service

(

std::string const &

service_name

)

Definition at line 280 of file windows_service.cpp.

{
  tools::msg_writer() << "Stopping service";
 
  service_handle p_manager{
    OpenSCManager(
        nullptr
      , nullptr
      , SC_MANAGER_CONNECT
      )
  , &::CloseServiceHandle
  };
  if (p_manager == nullptr)
  {
    tools::fail_msg_writer() << "Couldn't connect to service manager: " << get_last_error();
    return false;
  }
 
  service_handle p_service{
    OpenService(
        p_manager.get()
      , service_name.c_str()
      , SERVICE_STOP | SERVICE_QUERY_STATUS
      )
  , &::CloseServiceHandle
  };
  if (p_service == nullptr)
  {
    tools::fail_msg_writer() << "Couldn't find service: " << get_last_error();
    return false;
  }
 
  SERVICE_STATUS status = {};
  if (!ControlService(p_service.get(), SERVICE_CONTROL_STOP, &status))
  {
    tools::fail_msg_writer() << "Couldn't request service stop: " << get_last_error();
    return false;
  }
 
  tools::success_msg_writer() << "Service stopped";
 
  pause_to_display_admin_window_messages();
 
  return true;
}

Here is the call graph for this function:

uninstall_service()

bool windows::uninstall_service

(

std::string const &

service_name

)

Definition at line 328 of file windows_service.cpp.

{
  service_handle p_manager{
    OpenSCManager(
        nullptr
      , nullptr
      , SC_MANAGER_CONNECT
      )
  , &::CloseServiceHandle
  };
  if (p_manager == nullptr)
  {
    tools::fail_msg_writer() << "Couldn't connect to service manager: " << get_last_error();
    return false;
  }
 
  service_handle p_service{
    OpenService(
        p_manager.get()
      , service_name.c_str()
      , SERVICE_QUERY_STATUS | DELETE
      )
  , &::CloseServiceHandle
  };
  if (p_service == nullptr)
  {
    tools::fail_msg_writer() << "Couldn't find service: " << get_last_error();
    return false;
  }
 
  SERVICE_STATUS status = {};
  if (!DeleteService(p_service.get()))
  {
    tools::fail_msg_writer() << "Couldn't uninstall service: " << get_last_error();
    return false;
  }
 
  tools::success_msg_writer() << "Service uninstalled";
 
  pause_to_display_admin_window_messages();
 
  return true;
}

Here is the call graph for this function: