initial commit

2026-06-09 17:46:46 +02:00
commit 28b42492e6
18 changed files with 195 additions and 0 deletions
@@ -0,0 +1,63 @@
+import sys
+import re
+
+
+class Scorpion:
+    def __init__(self, args):
+        self.args = self.args_init(args[1:])
+        self.signatures = {
+            "jpg": [b"\xFF\xD8\xFF"],
+            "png": [b"\x89\x50\x4E\x47\x0D\x0A\x1A\x0A"],
+            "gif": [b"GIF87a",b"GIF89a"],
+            "bmp": [b"BM"]
+        }
+        self.png_cluster = {
+            "IHDR": {"Width": 4, "Height": 4, "Bit Depth": 1, "Color Type": 1, "Compression": 1, "Filter": 1, "Interlace": 1}
+        }
+    def args_init(self, args: list[str]) -> set[str]:
+        argset = set()
+        pattern = re.compile(r'^.*\.(?:jpe?g|png|gif|bmp)$', re.IGNORECASE)
+        for arg in args:
+            print(arg)
+            if not pattern.match(arg):
+                print("Error : Invalid file")
+                return
+            argset.add(arg)
+        return argset
+
+    def hexdump(self, data: bytes, width: int = 16):
+        for i in range(0, len(data), width):
+            chunk = data[i:i+width]
+            hex_bytes = " ".join(f"{b:02X}" for b in chunk)
+            ascii_part = "".join(chr(b) if 32 <= b < 127 else "." for b in chunk)
+            print(f"{i:08X}  {hex_bytes:<{width*3}}  {ascii_part}")
+
+    def get_file_type(self, data: bytes):
+        for filetype, sigs in self.signatures.items():
+            if any(data.startswith(sig) for sig in sigs):
+                return filetype
+        return None
+
+    def run(self):
+        for arg in self.args:
+            with open(arg, "rb") as f:
+                data = f.read()
+                filetype = self.get_file_type(data)
+                self.hexdump(data)
+                match filetype:
+                    case "jpg":
+                        return None
+                    case "png":
+                        self.read_png(data)
+                    case "gif":
+                        return None
+                    case "bmp":
+                        return None
+                    case _:
+                        return None
+    def read_png(self, data):
+        for cluster, cluster_values in self.png_cluster.values():
+            
+if __name__ == "__main__":
+    scorpion = Scorpion(sys.argv)
+    scorpion.run()