Return board info

Author: AT0myks

README.md

@@ -87,6 +87,8 @@ U-Boot compiler:      mipsel-24kec-linux-uclibc-gcc.br_real (Buildroot 2015.11.1
 U-Boot linker:        GNU ld (GNU Binutils) 2.24
 File system:          squashfs
 File system sections: fs
+Board vendor:         Novatek
+Board:                Novatek NA51023 evaluation board
 ```
 
 Or with JSON output:
@@ -110,6 +112,8 @@ $ reolinkfw info RLC-410-5MP_20_20052300.zip -j 2
     "uboot_compiler": "mipsel-24kec-linux-uclibc-gcc.br_real (Buildroot 2015.11.1-00003-gfd1edb1) 4.9.3",
     "uboot_linker": "GNU ld (GNU Binutils) 2.24",
     "linux_banner": "Linux version 4.1.0 (lwy@ubuntu) (gcc version 4.9.3 (Buildroot 2015.11.1-00003-gfd1edb1) ) #1 PREEMPT Tue Feb 26 18:19:48 CST 2019",
+    "board": "Novatek NA51023 evaluation board",
+    "board_vendor": "Novatek",
     "filesystems": [
       {
         "name": "fs",

reolinkfw/__init__.py

@@ -160,6 +160,12 @@ def fdt_json(self) -> Optional[dict[str, Any]]:
             return literal_eval(self.fdt.to_json().replace("null", "None"))
         return None
 
+    @property
+    def board(self) -> Optional[str]:
+        if self.fdt_json is not None:
+            return self.fdt_json["model"][1]
+        return None
+
     def _fdclose(self, fd: BinaryIO) -> None:
         self._open_files -= 1
         if self._closefd and not self._open_files:
@@ -294,6 +300,22 @@ def get_kernel_config(self) -> Optional[bytes]:
             return gzip.decompress(match.group(1))
         return None
 
+    def get_vendor(self) -> Optional[str]:
+        map_ = {
+            "novatek": "Novatek",
+            "sstar": "MStar/SigmaStar",
+            "hisilicon": "HiSilicon",
+        }
+        if self.fdt_json is not None:
+            key = self.fdt_json["compatible"][1].split(',')[0]
+            return map_.get(key.lower(), key)
+        with self.open(self["uboot"]) as f:
+            if re.match(b"GM[0-9]{4}", f.read(6)):
+                return "Grain Media"
+        if re.search(b"HISILICON LOGO MAGIC", self.uboot_section) is not None:
+            return "HiSilicon"
+        return None
+
     def get_fs_info(self) -> list[dict[str, str]]:
         result = []
         for section in self._fs_sections:
@@ -332,6 +354,8 @@ async def get_info(self) -> dict[str, Any]:
             "uboot_compiler": compiler,
             "uboot_linker": linker,
             "linux_banner": self.get_linux_banner(),
+            "board": self.board or "Unknown",
+            "board_vendor": self.get_vendor() or "Unknown",
             "filesystems": self.get_fs_info(),
             "sha256": ha
         }

reolinkfw/__main__.py

@@ -38,6 +38,8 @@ async def info(args: Namespace) -> None:
             print(f"{'U-Boot linker:':{width}}", info.uboot_linker or "Unknown")
             print(f"{'File system:':{width}}", ', '.join(sorted(fs_types)))
             print(f"{'File system sections:':{width}}", ', '.join(fs_names))
+            print(f"{'Board vendor:':{width}}", info.board_vendor or "Unknown")
+            print(f"{'Board:':{width}}", info.board or "Unknown")
             if idx != len(pak_infos) - 1:
                 print()
     else: