Anti-Virus

term.clear()
term.setCursorPos(1,1)

--[[

Give credit to Shinjiteru for the base-code!

Types of viruses:

R.A.K - Random access keys. These hook a math.random to startup and keep rebooting the computer until it gets a certain number.
 They will often display it's payload or allow the user to continue after it gets the number.

Virus - Just your average malware. Often infect other files and display it's payload later.

Attackware - These are usually made for revenge. They often flood rednet with long, spam messages or try to DDOS a certain url.

R.A.T - Remote admin tools. They allow for the attacker to remotely access the system that is infected.

Worm - ANY of the types of malware that spread. They often use an exploit, mass email, or infect floppy disks.

Keyloggers - These log the keys that you type into the computer, and often upload them to pastebin as items such as key.log, etc

Spyware - They can spy on ANYTHING you do. Keys, programs ran, programs edited, etc.

P.U.P - Not guaranteed to be a virus, but can sneak out malicious functions.

Trojan - Programs that download other malware or P.U.Ps.

Adware - None found yet. Advertise other malware to you, this will likely become a problem in Firewolf 3.0 with http support.

Crapware - Don't really damage your computer, but can be really annoying.

Scareware - Same as crapware, except they are made to scare you in some way, such as saying "Deleting ROM", Deleting "bios.lua", or displaying a scary image.

-- ]]
UpdateURL = "https://snipt.net/raw/bf1103f7d45ff3f74a05a2075e5b5671/"
VirusDefinitionsURL = "https://snipt.net/raw/59d2a60b6183005904b7192e1fc3914e/"

function split(pString, pPattern)
   local Table = {n = 0}  -- NOTE: use {n = 0} in Lua-5.0
   local fpat = "(.-)" .. pPattern
   local last_end = 1
   local s, e, cap = pString:find(fpat, 1)
   while s do
      if s ~= 1 or cap ~= "" then
     table.insert(Table,cap)
      end
      last_end = e+1
      s, e, cap = pString:find(fpat, last_end)
   end
   if last_end <= #pString then
      cap = pString:sub(last_end)
      table.insert(Table, cap)
   end
   return Table
end

function isColor ()
    if term.isColor then return term.isColor() else return false end
end

function isEven ( nNum ) xn=0 while xn < nNum do xn = xn + 2 if xn == nNum then return true elseif xn > nNum then return false end end return false end


function cPrint ( nString )
    if isEven(nString:len()) then nString = nString.." " end
    ox, oy = term.getCursorPos()
    scrWid, scrHei = term.getSize()
    term.setCursorPos(scrWid/2-nString:len()/2,oy)
    write(nString)
    term.setCursorPos(1,oy+1)
end

function fPrint ( nString )
    if isEven(nString:len()) then nString = nString.." " end
    ox, oy = term.getCursorPos()
    scrWid, scrHei = term.getSize()
    write(string.rep(" ",scrWid/2-nString:len()/2)..nString..string.rep(" ",scrWid/2-nString:len()/2))
    term.setCursorPos(1,oy+1)
end

function cLine ( nCol , rCol )
    ox, oy = term.getCursorPos()
    scrWid, scrHei = term.getSize()
    term.setBackgroundColor(nCol)
    write(string.rep(" ",scrWid))
    term.setBackgroundColor(rCol)
    term.setCursorPos(1,oy+1)
end

function tLine ( nT )
    ox, oy = term.getCursorPos()
    scrWid, scrHei = term.getSize()
    write(string.rep(nT:sub(1,1),scrWid))
    term.setCursorPos(1,oy+1)
end

function doError ( nErr )
    if isColor() then
        term.setBackgroundColor(colors.red)
        term.clear()
        term.setCursorPos(1,1)
        cPrint("ERROR")
        cLine(colors.white,colors.red)
        term.setCursorPos(3,4)
        write("Err: "..nErr)
        sleep(2)
        term.setBackgroundColor(colors.white)
        term.setCursorPos(3,6)
        write("Press any key to reboot.")
        os.pullEvent("key")
        os.reboot()
    else
        term.clear()
        term.setCursorPos(1,1)
        cPrint("ERROR")
        tLine("-")
        term.setCursorPos(3,4)
        write("Err: "..nErr)
        sleep(2)
        term.setCursorPos(3,6)
        write("Press any key to reboot.")
        os.pullEvent("key")
        os.reboot()
    end
end

resp = http.get(VirusDefinitionsURL)
if resp then
    defs = split(resp.readAll(),"\n")
else
    doError("Could not get Definitions!")
end

nDefs = {}

for n,m in ipairs(defs) do
    nDefs[n] = split(m,"|")
end

function rVirus ( nID )
    -- Name , Type , CatchString , Age , Common-ness , Danger , Estimated Total Infections , Curing Function , Sonar Pattern (NORTAN METHOD)
    return { name = nDefs[nID][1] , nType = nDefs[nID][2] , def = nDefs[nID][3] , age = nDefs[nID][4] , recurrency = nDefs[nID][5] , threatLevel = nDefs[nID][6] , infectionsToDate = nDefs[nID][7] , cure = nDefs[nID][8] , sonarPattern = nDefs[nID][9] }
end

curIt = ""
curInf = 0
curInfs = {}
curInfsT = {}

function scanDisplay ()
    while true do
        if isColor() then
            term.setBackgroundColor(colors.purple)
            term.clear()
            term.setCursorPos(1,1)
            cPrint("MALSCAN")
            cLine(colors.white,colors.purple)
            term.setCursorPos(3,4)
            write("Currently Scanning: "..curIt)
            term.setCursorPos(3,5)
            write("Infections: ")
            if curInf > 0 then
                term.setTextColor(colors.red)
                write(curInf)
                term.setTextColor(colors.white)
            else
                write(curInf)
            end
            term.setCursorPos(3,7)
            write("Stay Calm! Scanning!")
        else
            term.clear()
            term.setCursorPos(1,1)
            cPrint("MALSCAN")
            tLine("-")
            term.setCursorPos(3,4)
            write("Currently Scanning: "..curIt)
            term.setCursorPos(3,5)
            write("Infections: "..curInf)
            term.setCursorPos(3,7)
            write("Stay calm! Scanning!")
        end
        sleep(0.5)
    end
end

readStack = {

}

-- Prepare the readStack

-- >:( I know there is an easier way! But.... I can't think of that right now, too many ideas! XD
function toStack6 (nS5)
    for n5,m5 in ipairs(fs.list(nS5)) do
        readStack[#readStack+1] = nS5..m5
    end
end

function toStack5 (nS4)
    for n4,m4 in ipairs(fs.list(nS4)) do
        readStack[#readStack+1] = nS4..m4
        if fs.isDir(m4) then
            nCarry5 = n4
            toStack6(m4.."/")
        end
    end
end

function toStack4 (nS3)
    for n3,m3 in ipairs(fs.list(nS3)) do
        readStack[#readStack+1] = nS3..m3
        if fs.isDir(m3) then
            nCarry4 = n3
            toStack5(m3.."/")
        end
        nCarry4 = 0
    end
end

function toStack3 (nS2)
    for n2,m2 in ipairs(fs.list(nS2)) do
        readStack[#readStack+1] = nS2..m2
        if fs.isDir(m2) then
            nCarry3 = n2
            toStack4(m2.."/")
        end
        nCarry3 = 0
    end
end

function toStack2 (nS1)
    for n1,m1 in ipairs(fs.list(nS1)) do
        readStack[#readStack+1] = nS1..m1
        if fs.isDir(m1) then
            nCarry2 = n1
            toStack3(m1.."/")
        end
        nCarry2 = 0
    end
end

-- Teehee! ( >_> ) ffs world, I was thinking: "How can I prevent overlapping table variables??" So I came up with this BRILLIANT bugged out carry in / out system, BUT, I wasted 40 minutes on trying to solve this riddle when it was as simple as, table[#table+1]=var. >_< *facepalm*
nCarry1 = 0
nCarry2 = 0
nCarry3 = 0
nCarry4 = 0
nCarry5 = 0
nCarry6 = 0

function toStack1 (nS)
    for n,m in ipairs(fs.list(nS)) do
        readStack[#readStack+1] = nS..m
        if fs.isDir(m) then
            nCarry1 = n
            toStack2(m.."/")
        end
        nCarry1 = 0
    end
end

toStack1("/")

fileTable = {}

function detectFile_MALDEF ( nData )
    if nData == nil then nData = "" ad=io.open(".errors.ffs","a") ad:write("Nil catchy! >:(\n") ad:close() end -- Error catch~ for debugging.
    for i=1, #nDefs do
        if string.find ( nData, rVirus(i).def ) ~= nil then
            return i
        end
    end
    return false
end

function scanProc ()
    -- WARNING: Caution of Remote Code Injection! INFECTED_FILE -> PASTEBIN -> INFECTED_CLIENT -> LOADSTRING(VIRUS)()
    -- Note to self: Perform a google search on common CC viruses (exact code) and have google tell you where they are stored.
    for n,m in ipairs(readStack) do
        if fs.exists(m) then if fs.isDir(m) == false then if fs.getSize(m) > 0 then if fs.isReadOnly(m) == false and m ~= shell.getRunningProgram() and m ~= "/"..shell.getRunningProgram() then
            curIt = m
            ell=fs.open(m,"r")
            md=ell:readAll()
            ell:close()
            di = detectFile_MALDEF(md)
            if di ~= false then
                curInfs[#curInfs+1]=n
                curInf = curInf + 1
            end
        end end end end
    end
end

function scan ()
    parallel.waitForAny (
        scanDisplay,
        scanProc
    )
    if isColor() then
        if curInf > 0 then
            term.setBackgroundColor(colors.red)
            term.clear()
            term.setCursorPos(1,1)
            cPrint("VirusScope has found threats.")
            cLine(colors.white,colors.red)
            term.setCursorPos(3,4)
            write("Threats detected: "..curInf)
            term.setCursorPos(3,6)
            for n,m in ipairs(curInfs) do
                fs.delete(readStack[m])
            end
            write("All threats are deleted!")
            -- Subject to change in the future, we might just store all deleted files in a . folder and have them restorable. Not right now though.
            aFi = io.open(".avl","a")
            for n,m in ipairs(curInfsT) do
                aFi:write(m.."\n")
            end
            aFi:close()
            term.setCursorPos(3,7)
            write("Press any key to reboot!")
            os.pullEvent("key")
            os.reboot()
        else
            term.setBackgroundColor(colors.white)
            term.clear()
            term.setCursorPos(1,1)
            term.setTextColor(colors.black)
            cPrint("ALL CLEAR")
            cLine(colors.lime,colors.white)
            term.setCursorPos(3,4)
            write("Infections: 0")
            term.setCursorPos(3,6)
            sleep(2)
            write("Press any key to reboot!")
            os.pullEvent("key")
            os.reboot()
        end
    else
        if curInfs > 0 then
            term.clear()
            term.setCursorPos(1,1)
            cPrint("THREATS FOUND")
            tLine("-")
            term.setCursorPos(3,4)
            for n,m in ipairs(curInfs) do
                fs.delete(readStack[m])
            end
            fileHandle = io.open(".avl","a")
            for n,m in ipairs(curInfsT) do
                fileHandle:write(m.."\n")
            end
            fileHandle:close()
            write("Threats: "..curInf)
            term.setCursorPos(3,6)
            write("All viruses deleted!")
            term.setCursorPos(3,7)
            sleep(2)
            write("Press any key to reboot!")
            os.pullEvent("key")
            os.reboot()
        else
            term.clear()
            term.setCursorPos(1,1)
            cPrint("ALL CLEAR")
            tLine("-")
            term.setCursorPos(3,4)
            write("Infections: 0")
            term.setCursorPos(3,6)
            sleep(2)
            write("Press any key to reboot!")
            os.pullEvent("key")
            os.reboot()
        end
    end
end

function wipe ()
    for n,m in ipairs(fs.list("")) do
        if m ~= "rom" and m ~= shell.getRunningProgram() then
            fs.delete(m)
        end
    end
end

function update ()
    resp = http.get(UpdateURL)
    if resp then
        aFs=io.open(shell.getRunningProgram(),"w")
        aFs:write(resp.readAll())
        aFs:close()
        shell.run(shell.getRunningProgram())
    else
        doError("Could not reach update server!")
    end
end

function recov ()
term.clear()
cPrint("Flare and all infected files has been deleted.")
fs.delete("worm")
fs.delete("cubedos/main")
fs.delete("cubedos/block")
fs.delete("edit")
fs.delete("delete")
fs.delete("edit")
fs.delete("pastebin")
fs.delete("lua")
fs.delete("cd")
fs.delete("dir")
fs.delete("cube-dos")

end

function bscan ()
    if fs.exists("/startup") then fs.move("/startup","/startup.bk") end -- Stupid FS api... >_>
    fs.copy(shell.getRunningProgram(),"/startup")
    os.reboot()
end


bEndIt = false

function bEnd ()
    bEndIt = true
end

menuOptions = {
{ "Scan" , scan },
{ "Wipe Computer" , wipe },
{ "Update" , update },
{ "Recover from Flare" , recov },
{ "Boot Scan" , bscan },
{ "Exit AV" , bEnd }
}


function doMenuProcessing ()
    while bEndIt == false do
        eve = { os.pullEvent() }
        if eve[1] == "key" and eve[2] == keys.down then
            if menuIndex < 6 then menuIndex = menuIndex + 1 mDraw() end
        end
        if eve[1] == "key" and eve[2] == keys.up then
            if menuIndex > 1 then menuIndex = menuIndex - 1 mDraw() end
        end
        if eve[1] == "key" and eve[2] == keys.enter then
            menuOptions[menuIndex][2]()
        end
        if eve[1] == "mouse_click" then
            if eve[4] == 4 then menuOptions[1][2]() end
            if eve[4] == 6 then menuOptions[2][2]() end
            if eve[4] == 8 then menuOptions[3][2]() end
            if eve[4] == 10 then menuOptions[4][2]() end
            if eve[4] == 12 then menuOptions[5][2]() end
            if eve[4] == 14 then menuOptions[6][2]() end
        end
        if bEndIt then break end
    end
end

menuIndex = 1

function mDraw ()
    if isColor() then
        term.setBackgroundColor(colors.black)
        term.setTextColor(colors.black)
        term.clear()
        term.setCursorPos(1,1)
        cPrint("VirusScope")
        cLine(colors.green,colors.black)
        term.setCursorPos(1,4)
        term.setBackgroundColor(colors.lightGray)
        term.setTextColor(colors.black)
        if menuIndex == 1 then fPrint("["..menuOptions[1][1].."]") else fPrint(menuOptions[1][1]) end
        term.setCursorPos(1,6)
        term.setBackgroundColor(colors.lightGray)
        term.setTextColor(colors.black)
        if menuIndex == 2 then fPrint("["..menuOptions[2][1].."]") else fPrint(menuOptions[2][1]) end
        term.setCursorPos(1,8)
        term.setBackgroundColor(colors.lightGray)
        term.setTextColor(colors.black)
        if menuIndex == 3 then fPrint("["..menuOptions[3][1].."]") else fPrint(menuOptions[3][1]) end
        term.setCursorPos(1,10)
        term.setBackgroundColor(colors.lightGray)
        term.setTextColor(colors.black)
        if menuIndex == 4 then fPrint("["..menuOptions[4][1].."]") else fPrint(menuOptions[4][1]) end
        term.setCursorPos(1,12)
        term.setBackgroundColor(colors.lightGray)
        term.setTextColor(colors.black)
        if menuIndex == 5 then fPrint("["..menuOptions[5][1].."]") else fPrint(menuOptions[5][1]) end
        term.setCursorPos(1,14)
        term.setBackgroundColor(colors.lightGray)
        term.setTextColor(colors.black)
        if menuIndex == 6 then fPrint("["..menuOptions[6][1].."]") else fPrint(menuOptions[6][1]) end
        doMenuProcessing()
    else
        term.clear()
        term.setCursorPos(1,1)
        cPrint("Shinjiteru MalScan")
        tLine("-")
        term.setCursorPos(1,4)
        if menuIndex == 1 then fPrint("["..menuOptions[1][1].."]") else fPrint(menuOptions[1][1]) end
        term.setCursorPos(1,6)
        if menuIndex == 2 then fPrint("["..menuOptions[2][1].."]") else fPrint(menuOptions[2][1]) end
        term.setCursorPos(1,8)
        if menuIndex == 3 then fPrint("["..menuOptions[3][1].."]") else fPrint(menuOptions[3][1]) end
        term.setCursorPos(1,10)
        if menuIndex == 4 then fPrint("["..menuOptions[4][1].."]") else fPrint(menuOptions[4][1]) end
        term.setCursorPos(1,12)
        if menuIndex == 5 then fPrint("["..menuOptions[5][1].."]") else fPrint(menuOptions[5][1]) end
        term.setCursorPos(1,14)
        if menuIndex == 6 then fPrint("["..menuOptions[6][1].."]") else fPrint(menuOptions[6][1]) end
        doMenuProcessing()
    end
end

mDraw()
doMenuProcessing()

term.setBackgroundColor(colors.black)
term.setTextColor(colors.white)
term.clear()
term.setCursorPos(1,1)