module MetasploitPayloads

# File lib/metasploit-payloads.rb, line 98
def self.data_directory
  ::File.realpath(::File.join(::File.dirname(__FILE__), '..', 'data'))
end

# File lib/metasploit-payloads.rb, line 73
def self.list_meterpreter_extensions(binary_suffix)
  extensions = []

  root_dirs = [local_meterpreter_dir]

  # Find the valid extensions in the data folder first, if MSF
  # is installed.
  if metasploit_installed?
    root_dirs.unshift(msf_meterpreter_dir)
    root_dirs.unshift(user_meterpreter_dir)
  end

  root_dirs.each do |dir|
    # Merge in any that don't already exist in the collection.
    meterpreter_enum_ext(dir, binary_suffix).each do |e|
      extensions.push(e) unless extensions.include?(e)
    end
  end

  extensions
end

# File lib/metasploit-payloads.rb, line 119
def self.local_meterpreter_dir
  ::File.join(data_directory, METERPRETER_SUBFOLDER)
end

# File lib/metasploit-payloads.rb, line 126
def self.meterpreter_enum_ext(root_dir, binary_suffix)
  exts = []
  ::Dir.entries(root_dir).each do |f|
    if ::File.readable?(::File.join(root_dir, f)) &&           f =~ /#{EXTENSION_PREFIX}(.*)\.#{binary_suffix}/
      exts.push($1)
    end
  end
  exts
end

# File lib/metasploit-payloads.rb, line 17
def self.meterpreter_ext_path(ext_name, binary_suffix)
  path(METERPRETER_SUBFOLDER, "#{EXTENSION_PREFIX}#{ext_name}.#{binary_suffix}")
end

# File lib/metasploit-payloads.rb, line 41
def self.meterpreter_path(name, binary_suffix)
  path(METERPRETER_SUBFOLDER, "#{name}.#{binary_suffix}".downcase)
end

# File lib/metasploit-payloads.rb, line 105
def self.msf_meterpreter_dir
  ::File.join(Msf::Config.data_directory, METERPRETER_SUBFOLDER)
end

# File lib/metasploit-payloads.rb, line 48
def self.path(*path_parts)
  gem_path = expand(data_directory, ::File.join(path_parts))
  if metasploit_installed?
    user_path = expand(Msf::Config.config_directory, ::File.join(USER_DATA_SUBFOLDER, path_parts))
    msf_path = expand(Msf::Config.data_directory, ::File.join(path_parts))
  end
  readable_path(gem_path, user_path, msf_path)
end

# File lib/metasploit-payloads.rb, line 60
def self.read(*path_parts)
  file_path = path(path_parts)
  if file_path.nil?
    full_path = ::File.join(path_parts)
    fail RuntimeError, "#{full_path} not found", caller
  end

  ::File.binread(file_path)
end

# File lib/metasploit-payloads.rb, line 21
def self.readable_path(gem_path, *extra_paths)
  # Try the MSF path first to see if the file exists, allowing the MSF data
  # folder to override what is in the gem. This is very helpful for
  # testing/development without having to move the binaries to the gem folder
  # each time. We only do this is MSF is installed.
  extra_paths.each do |extra_path|
    if ::File.readable? extra_path
      warn_local_path(extra_path) if ::File.readable? gem_path
      return extra_path
    end
  end

  return gem_path if ::File.readable? gem_path

  nil
end

# File lib/metasploit-payloads.rb, line 112
def self.user_meterpreter_dir
  ::File.join(Msf::Config.config_directory, USER_DATA_SUBFOLDER, METERPRETER_SUBFOLDER)
end

# File lib/metasploit-payloads/version.rb, line 4
def self.version
  VERSION
end

# File lib/metasploit-payloads.rb, line 149
def self.expand(root_dir, file_name)
  ::File.expand_path(::File.join(root_dir, file_name))
end

# File lib/metasploit-payloads.rb, line 142
def self.metasploit_installed?
  defined? Msf::Config
end

# File lib/metasploit-payloads.rb, line 155
def self.warn_local_path(path)
  unless @local_paths.include?(path)
    STDERR.puts("WARNING: Local file #{path} is being used")
    if @local_paths.empty?
      STDERR.puts('WARNING: Local files may be incompatible with the Metasploit Framework')
    end
    @local_paths << path
  end
end