more options to edid editing

1 files changed, 99 insertions, 7 deletions

diff --git a/kvmd/apps/edidconf/__init__.py b/kvmd/apps/edidconf/__init__.py
index baf2ed43..f482c338 100644
--- a/kvmd/apps/edidconf/__init__.py
+++ b/kvmd/apps/edidconf/__init__.py
@@ -28,12 +28,15 @@ from typing import List
 from typing import Optional
 
 from ...validators.basic import valid_bool
+from ...validators.basic import valid_int_f0
 
 from .. import init
 
 
 # =====
 class _Edid:
+    # https://en.wikipedia.org/wiki/Extended_Display_Identification_Data
+
     def __init__(self, path: str) -> None:
         with open(path) as file:
             self.__load_from_hex(file.read())
@@ -53,15 +56,89 @@ class _Edid:
         else:
             print(text)
 
-    def is_audio_enabled(self) -> bool:
+    # =====
+
+    def get_mfc_id(self) -> str:
+        raw = self.__data[8] << 8 | self.__data[9]
+        return bytes([
+            ((raw >> 10) & 0b11111) + 0x40,
+            ((raw >> 5) & 0b11111) + 0x40,
+            (raw & 0b11111) + 0x40,
+        ]).decode("ascii")
+
+    def set_mfc_id(self, mfc_id: str) -> None:
+        assert len(mfc_id) == 3, "Mfc ID must be 3 characters long"
+        data = mfc_id.upper().encode("ascii")
+        for byte in data:
+            assert 0x41 <= byte <= 0x5A, "Mfc ID must contain only A-Z characters"
+        raw = (
+            (data[2] - 0x40)
+            | ((data[1] - 0x40) << 5)
+            | ((data[0] - 0x40) << 10)
+        )
+        self.__data[8] = (raw >> 8) & 0xFF
+        self.__data[9] = raw & 0xFF
+
+    # =====
+
+    def get_product_id(self) -> int:
+        return (self.__data[10] | self.__data[11] << 8)
+
+    def set_product_id(self, product_id: int) -> None:
+        assert 0 <= product_id <= 0xFFFF, f"Product ID should be from 0 to {0xFFFF}"
+        self.__data[10] = product_id & 0xFF
+        self.__data[11] = (product_id >> 8) & 0xFF
+
+    # =====
+
+    def get_serial(self) -> int:
+        return (
+            self.__data[12]
+            | self.__data[13] << 8
+            | self.__data[14] << 16
+            | self.__data[15] << 24
+        )
+
+    def set_serial(self, serial: int) -> None:
+        assert 0 <= serial <= 0xFFFFFFFF, f"Serial should be from 0 to {0xFFFFFFFF}"
+        self.__data[12] = serial & 0xFF
+        self.__data[13] = (serial >> 8) & 0xFF
+        self.__data[14] = (serial >> 16) & 0xFF
+        self.__data[15] = (serial >> 24) & 0xFF
+
+    # =====
+
+    def get_monitor_name(self) -> str:
+        index = self.__find_dtd_value(0xFC)
+        assert index > 0, "Can't find DTD Monitor name"
+        return bytes(self.__data[index:index + 13]).decode("cp437").strip()
+
+    def set_monitor_name(self, name: str) -> None:
+        index = self.__find_dtd_value(0xFC)
+        assert index > 0, "Can't find DTD Monitor name"
+        encoded = (name[:13] + "\n" + " " * 12)[:13].encode("cp437")
+        for (offset, byte) in enumerate(encoded):
+            self.__data[index + offset] = byte
+
+    # =====
+
+    def get_audio(self) -> bool:
         return bool(self.__data[131] & 0b01000000)
 
-    def set_audio_enabled(self, enabled: bool) -> None:
+    def set_audio(self, enabled: bool) -> None:
         if enabled:
             self.__data[131] |= 0b01000000
         else:
             self.__data[131] &= (0xFF - 0b01000000)  # ~X
 
+    # =====
+
+    def __find_dtd_value(self, dtype: int) -> int:
+        for index in [54, 72, 90, 108]:
+            if self.__data[index + 3] == dtype:
+                return index + 5
+        return -1
+
     def __load_from_hex(self, text: str) -> None:
         text = re.sub(r"\s", "", text)
         self.__data = [
@@ -92,8 +169,16 @@ def main(argv: Optional[List[str]]=None) -> None:
                         help="EDID hex text file path", metavar="<file>")
     parser.add_argument("--stdout", action="store_true",
                         help="Write to stdout instead of the rewriting the source file")
-    parser.add_argument("--set-audio", type=valid_bool, dest="set_audio", default=None,
+    parser.add_argument("--set-audio", type=valid_bool, default=None,
                         help="Enable or disable basic audio", metavar="<yes|no>")
+    parser.add_argument("--set-mfc-id", default=None,
+                        help="Set manufacturer ID (https://uefi.org/pnp_id_list)", metavar="<ABC>")
+    parser.add_argument("--set-product-id", type=valid_int_f0, default=None,
+                        help="Set product ID (decimal)", metavar="<uint>")
+    parser.add_argument("--set-serial", type=valid_int_f0, default=None,
+                        help="Set serial number (decimal)", metavar="<uint>")
+    parser.add_argument("--set-monitor-name", default=None,
+                        help="Set monitor name in DTD/MND (ASCII, max 13 characters)", metavar="<str>")
     parser.add_argument("--show-info", action="store_true",
                         help="Write summary info to stderr")
     options = parser.parse_args(argv[1:])
@@ -101,12 +186,19 @@ def main(argv: Optional[List[str]]=None) -> None:
     edid = _Edid(options.path)
     changed = False
 
-    if options.set_audio is not None:
-        edid.set_audio_enabled(options.set_audio)
-        changed = True
+    for cmd in dir(options):
+        if cmd.startswith("set_"):
+            value = getattr(options, cmd)
+            if value is not None:
+                getattr(edid, cmd)(value)
+                changed = True
 
     if changed:
         edid.write("" if options.stdout else options.path)
 
     if options.show_info:
-        print(f"Audio: {'yes' if edid.is_audio_enabled() else 'no'}", file=sys.stderr)
+        print("Mfc ID:      ", edid.get_mfc_id())
+        print("Product ID:  ", edid.get_product_id())
+        print("Serial:      ", edid.get_serial())
+        print("Monitor name:", edid.get_monitor_name())
+        print("Audio:       ", ("yes" if edid.get_audio() else "no"), file=sys.stderr)