{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.11.13","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"none","dataSources":[{"sourceId":4117,"databundleVersionId":46665,"sourceType":"competition"}],"dockerImageVersionId":31192,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":false}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import os\nimport pandas as pd\nimport subprocess\nfrom tqdm import tqdm\nfrom zipfile import ZipFile\n\n# =========================\n# PATHS (Kaggle competition)\n# =========================\nINPUT_DIR = \"/kaggle/input/malware-classification\"\nARCHIVE_PATH = os.path.join(INPUT_DIR, \"train.7z\")\nLABEL_CSV = os.path.join(INPUT_DIR, \"trainLabels.csv\")\n\nWORK_DIR = \"/kaggle/working\"\nOUTPUT_DIR = os.path.join(WORK_DIR, \"asm_labeled\")\nRAMNIT_DIR = os.path.join(OUTPUT_DIR, \"Ramnit\")\nZIP_PATH = os.path.join(WORK_DIR, \"Ramnit_ASM.zip\")\n\n# =========================\n# CREATE OUTPUT FOLDER\n# =========================\nos.makedirs(RAMNIT_DIR, exist_ok=True)\nprint(\"✅ Output folder ready: Ramnit\")\n\n# =========================\n# READ & FILTER CSV\n# =========================\ndf = pd.read_csv(LABEL_CSV)\ndf = df[df[\"Class\"] == 1]   # Ramnit only\n\nprint(f\"📊 Total Ramnit samples: {len(df)}\")\n\n# =========================\n# EXTRACTION FROM train.7z\n# =========================\nextracted = 0\nfailed = 0\n\nfor _, row in tqdm(\n    df.iterrows(),\n    total=len(df),\n    desc=\"Extracting Ramnit ASM files\",\n    unit=\"file\"\n):\n    file_id = row[\"Id\"]\n    asm_in_7z = f\"train/{file_id}.asm\"\n\n    cmd = [\n        \"7z\", \"x\",\n        ARCHIVE_PATH,\n        asm_in_7z,\n        f\"-o{RAMNIT_DIR}\",\n        \"-y\"\n    ]\n\n    result = subprocess.run(\n        cmd,\n        stdout=subprocess.DEVNULL,\n        stderr=subprocess.DEVNULL\n    )\n\n    if result.returncode == 0:\n        extracted += 1\n    else:\n        failed += 1\n\n# =========================\n# ZIP THE OUTPUT FOLDER\n# =========================\nprint(\"\\n📦 Creating ZIP archive...\")\n\nwith ZipFile(ZIP_PATH, 'w') as zipf:\n    for root, _, files in os.walk(RAMNIT_DIR):\n        for file in files:\n            file_path = os.path.join(root, file)\n            arcname = os.path.relpath(file_path, OUTPUT_DIR)\n            zipf.write(file_path, arcname)\n\n# =========================\n# FINAL SUMMARY\n# =========================\nprint(\"\\n✅ PROCESS COMPLETE\")\nprint(f\"📁 Ramnit ASM extracted : {extracted}\")\nprint(f\"❌ Failed extractions  : {failed}\")\nprint(f\"📦 ZIP file created    : {ZIP_PATH}\")\nprint(\"\\n🎉 RAMNIT DATASET READY FOR DOWNLOAD!\")\n","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true,"execution":{"iopub.status.busy":"2025-12-13T17:31:23.805894Z","iopub.execute_input":"2025-12-13T17:31:23.806121Z","iopub.status.idle":"2025-12-13T20:07:32.094445Z","shell.execute_reply.started":"2025-12-13T17:31:23.806106Z","shell.execute_reply":"2025-12-13T20:07:32.090022Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"!7z a -t7z /kaggle/working/Ramnit_ASM.7z /kaggle/working/asm_labeled/Ramnit\n","metadata":{"trusted":true,"execution":{"iopub.status.busy":"2025-12-13T20:13:37.480940Z","iopub.execute_input":"2025-12-13T20:13:37.481797Z","iopub.status.idle":"2025-12-13T20:34:22.213602Z","shell.execute_reply.started":"2025-12-13T20:13:37.481763Z","shell.execute_reply":"2025-12-13T20:34:22.212191Z"}},"outputs":[],"execution_count":null},{"cell_type":"code","source":"","metadata":{"trusted":true},"outputs":[],"execution_count":null}]}