BUILD A HOMELAB TO PREP FOR OSCP / SECURITY RESEARCH

def get_cpu_flags
  cpu_flags = `grep flags /proc/cpuinfo | cut -f 2 -d : | uniq`.strip
end

cpu_flags = get_cpu_flags

def file_exists(filename)
  File.file?(filename) # return true if filename exists
end

def convert_file(format, filename) # needs qemu-img binary installed on the system, returns a new raw image file
  `qemu-img convert -f #{format} -O raw "#{filename}" "#{filename}.img"` 
end 

# I am using a 56MB vmdk file to test the conversion process
def test_convert(format, filename)
  file_exists(filename)
  convert_file(format, filename)
end

test_convert('vmdk', '/storage/virtual_machines/DSL-4.4.10-disk1.vmdk')

class DownloadVulnHubTorrents
  require 'csv'

  def initialize
    @base_url = 'https://download.vulnhub.com'
  end

  # get the checksum file which has checksums + urls. we can automate the check of the files and compare with the checksums to make sure everything downloaded matches
  def download_checksum
    checksum_url = "#{@base_url}/checksum.txt"
    `wget #{checksum_url}` # download the checksum file
  end

  def gather_urls(filename)
    urls = `awk ' { print $2 } ' #{filename} | sed 's/^\./''/g' | grep -E 'ova|torrent|zip|tar|txt|gz|gzip|iso|7z|exe|text|img|png|jpg|jpeg|md|LICENSE|README'`
    CSV.parse(urls).flatten
  end
end

vulnhubber = DownloadVulnHubTorrents.new
vulnhubber.download_checksums
vulnhubber.gather_urls('/home/booboy/bin/mygit/homelab_scripts/checksum.txt')

require 'sshkey'

  def gen_ssh_keypair

    k = SSHKey.generate(
      type: "DSA",
      bits: 1024,
      comment: "foo@bar.com",
      passphrase: "foobar"
    )
  end

keypair = gen_ssh_keypair