module MetasploitPayloads

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

# File lib/metasploit-payloads.rb, line 87
def self.list_meterpreter_extension_suffixes(extension_name=nil)
  list_meterpreter_dirs { |dir| meterpreter_enum_ext_suffixes(dir, extension_name) }
end

# File lib/metasploit-payloads.rb, line 76
def self.list_meterpreter_extensions(binary_suffix=nil)
  list_meterpreter_dirs { |dir| meterpreter_enum_ext(dir, binary_suffix) }
end

# File lib/metasploit-payloads.rb, line 115
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=nil)
  exts = []
  binary_suffix ||= '.*'
  ::Dir.entries(root_dir).each do |f|
    if ::File.readable?(::File.join(root_dir, f)) && \
       f =~ /#{EXTENSION_PREFIX}(\w+)\.#{binary_suffix}/
      exts.push($1)
    end
  end
  exts
end

# File lib/metasploit-payloads.rb, line 145
def self.meterpreter_enum_ext_suffixes(root_dir, extension_name=nil)
  suffixes = []
  extension_name ||= '\w+'
  ::Dir.entries(root_dir).each do |f|
    if ::File.readable?(::File.join(root_dir, f)) && \
       f =~ /#{EXTENSION_PREFIX}#{extension_name}\.(\w+(\.\w+)*)/
      suffixes.push($1)
    end
  end
  suffixes
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 101
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 108
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 169
def self.expand(root_dir, file_name)
  ::File.expand_path(::File.join(root_dir, file_name))
end

# File lib/metasploit-payloads.rb, line 187
def list_meterpreter_dirs(&block)
  things = [] # *things* is whatever is being enumerated (extension names, suffixes, etc.) as determined by the block
  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|
    next unless ::File.directory?(dir)

    # Merge in any that don't already exist in the collection.
    (yield dir).each do |e|
      things.push(e) unless things.include?(e)
    end
  end

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

# File lib/metasploit-payloads.rb, line 175
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