##
# WARNING: Metasploit no longer maintains or accepts meterpreter scripts.
# If you'd like to improve this script, please try to port it as a post
# module instead. Thank you.
##


# Author: Carlos Perez <carlos_perez [at] darkoperator.com and Shai rod (@NightRang3r)
#-------------------------------------------------------------------------------
################## Variable Declarations ##################

@client = client
srv_name           = nil
returned_value     = nil
srv_startup        = "Auto"
srv_display_name   = ""
srv_command        = nil
srv_list           = false
srv_start          = false
srv_stop           = false
srv_create         = false
srv_info           = false
srv_change_startup = false
srv_delete         = false


@exec_opts = Rex::Parser::Arguments.new(
  "-h" => [ false , "Help menu." ],
  "-l" => [ false , "List Services"],
  "-S" => [ false , "Start Service"],
  "-K" => [ false , "Stop Service"],
  "-C" => [ false , "Create Service, service will be set to auto start"],
  "-c" => [ false , "Change Service StartUp. Default <Auto>" ],
  "-i" => [ false , "Get Service Information"],
  "-n" => [ true  , "Service Name"],
  "-s" => [ true  , "Startup Parameter for service. Specify Auto, Manual or Disabled"],
  "-d" => [ true  , "Display Name of Service"],
  "-p" => [ true  , "Service command"],
  "-D" => [ false , "Delete Service"]
  )
meter_type = client.platform

################## Function Declarations ##################

# Usage Message Function
#-------------------------------------------------------------------------------
def usage
  print_line "Meterpreter Script for managing Windows Services."
  print_line(@exec_opts.usage)
  raise Rex::Script::Completed
end

# Wrong Meterpreter Version Message Function
#-------------------------------------------------------------------------------
def wrong_meter_version(meter = meter_type)
  print_error("#{meter} version of Meterpreter is not supported with this Script!")
  raise Rex::Script::Completed
end

# Check if sufficient privileges are present for certain actions
def priv_check
  if not is_uac_enabled? or is_admin?
    return true
  else
    print_error("Insufficient Privileges")
    raise Rex::Script::Completed
  end

end

################## Main ##################
# Check for Version of Meterpreter
wrong_meter_version(meter_type) if meter_type != 'windows'

@exec_opts.parse(args) { |opt, idx, val|
  case opt
  when "-h"
    usage
  when "-l"
    srv_list = true
  when "-n"
    srv_name = val
  when "-S"
    srv_start = true
  when "-K"
    srv_stop = true
  when "-i"
    srv_info = true
  when "-c"
    srv_change_startup = true
  when "-C"
    srv_create = true
  when "-d"
    srv_display_name = val
  when "-p"
    srv_command = val
  when "-D"
    srv_delete = true
  end
}

# List Services
if srv_list
  print_status("Service List:")
  service_list.each do |s|
    print_good("\t#{s}")
  end
  raise Rex::Script::Completed

# Start a service
elsif srv_start
  priv_check
  if srv_name
    begin
      returned_value = service_start(srv_name)
      if returned_value == 0
        print_good("Service #{srv_name} Started")
      elsif returned_value == 1
        print_good("Service #{srv_name} already Running")
      elsif returned_value == 2
        print_error("Service #{srv_name} is Disabled could not be started.")
      end

    rescue
      print_error("A Service Name must be provided, service names are case sensitive.")
    end
  else
    print_error("No Service Name was provided!")
  end
  raise Rex::Script::Completed

# Stop a Service
elsif srv_stop
  priv_check
  if srv_name
    begin
      returned_value = service_stop(srv_name)
      if returned_value == 0
        print_good("Service #{srv_name} Stopped")
      elsif returned_value == 1
        print_good("Service #{srv_name} already Stopped")
      elsif returned_value == 2
        print_error("Service #{srv_name} can not be stopped.")
      end

    rescue
      print_error("A Service Name must be provided, service names are case sensitive.")
    end
  else
    print_error("No Service Name was provided!")
  end
  raise Rex::Script::Completed

# Get service info
elsif srv_info
  srv_conf = {}
  if srv_name
    begin
      srv_conf = service_info(srv_name)
      print_status("Service Information for #{srv_name}:")
      print_good("\tName: #{srv_conf['Name']}")
      print_good("\tStartup: #{srv_conf['Startup']}")
      print_good("\tCommand: #{srv_conf['Command']}")
      print_good("\tCredentials: #{srv_conf['Credentials']}")
    rescue
      print_error("A Service Name must be provided, service names are case sensitive.")
    end
  else
    print_error("No Service Name was provided!")
  end
  raise Rex::Script::Completed

# Change startup of a service
elsif srv_change_startup
  priv_check
  if srv_name
    begin
      print_status("Changing Service #{srv_name} Startup to #{srv_startup}")
      service_change_startup(srv_name,srv_startup)
      print_good("Service Startup changed!")

    rescue
      print_error("A Service Name must be provided, service names are case sensitive.")
    end
  else
    print_error("No Service Name was provided!")
  end
  raise Rex::Script::Completed

# Create a service
elsif srv_create
  priv_check
  if srv_name and srv_command
    begin
      print_status("Creating Service #{srv_name}")
      service_create(srv_name,srv_display_name,srv_command)
      print_good("\tService Created!")
      print_good("\tDisplay Name: #{srv_display_name}")
      print_good("\tCommand: #{srv_command}")
      print_good("\tSet to Auto Star.")
    rescue::Exception => e
      print_error("Error: #{e}")
    end
  else
    print_error("No Service Name and Service Command where provided!")
  end

# Delete a service
elsif srv_delete
  priv_check
  if srv_name
    begin
      print_status("Deleting Service #{srv_name}")
      service_delete(srv_name)
      print_good("\tService #{srv_name} Delete")
    rescue::Exception => e
      print_error("A Service Name must be provided, service names are case sensitive.")
      print_error("Error: #{e}")
    end
  else
    print_error("No Service Name and Service Command where provided!")
  end
  raise Rex::Script::Completed
else
  usage
end