client/pyscripts/spi_flash_decode.py

   1 #!/usr/bin/env python3
   2
   3 import re
   4 import pm3
   5 # optional color support
   6 try:
   7     # pip install ansicolors
   8     from colors import color
   9 except ModuleNotFoundError:
  10     def color(s, fg=None):
  11         _ = fg
  12         return str(s)
  13
  14 spi = {
  15     0x85:{
  16         "manufacturer": "Puya",
  17         0x60: {
  18             0x15: {
  19                 "part": "P25Q16H",
  20                 "size": "16mbits",
  21                 "sizeB": "2MB",
  22                 },
  23             },
  24         },
  25     0xEF:{
  26         "manufacturer": "Winbond",
  27         0x30: {
  28             0x11: {
  29                 "part": "W25X10BV",
  30                 "size": "1mbits",
  31                 "sizeB": "128KB",
  32                 },
  33             0x12: {
  34                 "part": "W25X20BV",
  35                 "size": "2mbits",
  36                 "sizeB": "256KB",
  37                 },
  38             0x13: {
  39                 "part": "W25X40BV",
  40                 "size": "4mbits",
  41                 "sizeB": "512KB",
  42                 },
  43             },
  44         0x40: {
  45             0x13: {
  46                 "part": "W25Q40BV",
  47                 "size": "4mbits",
  48                 "sizeB": "512KB",
  49                 },
  50             0x14: {
  51                 "part": "W25Q80BV",
  52                 "size": "8mbits",
  53                 "sizeB": "1MB",
  54                 },
  55             0x15: {
  56                 "part": "W25Q16BV",
  57                 "size": "16mbits",
  58                 "sizeB": "2MB",
  59                 },
  60             0x16: {
  61                 "part": "W25Q32BV",
  62                 "size": "32mbits",
  63                 "sizeB": "4MB",
  64                 },
  65             },
  66         0x70: {
  67             0x22: {
  68                 "part": "W25Q02JV-IM",
  69                 "size": "2mbits",
  70                 "sizeB": "256KB",
  71                 },
  72             },
  73         },
  74     }
  75
  76 p = pm3.pm3()
  77
  78 p.console("hw status")
  79
  80 rex = re.compile("...\\s([0-9a-fA-F]{2})\\s/\\s([0-9a-fA-F]{4})")
  81 for line in p.grabbed_output.split('\n'):
  82     # [#]   JEDEC Mfr ID / Dev ID... 85 / 6015
  83     if " JEDEC " not in line:
  84         continue
  85     match = re.findall(rex, line)
  86     mid = int(match[0][0], 16)
  87     did = int(match[0][1], 16)
  88     did_h = did >> 8
  89     did_l = did & 0xff
  90     t = None
  91     if mid in spi:
  92         mfr = spi[mid]['manufacturer']
  93         if did_h in spi[mid]:
  94             if did_l in spi[mid][did_h]:
  95                 t = spi[mid][did_h][did_l]
  96                 print("\n Manufacturer... " + color(f"{mfr}", fg="green") +
  97                      "\n Device......... " + color(f"{t['part']}", fg="green") +
  98                      "\n Size........... " + color(f"{t['size']} ({t['sizeB']})", fg="yellow")
  99                      )
 100             else:
 101                 print("\n Manufacturer... " + color(f"{mfr}", fg="green") +
 102                      "\n Device ID...... " + color(f"{did:04X}h (unknown)", fg="red"))
 103         else:
 104             print("\n Manufacturer... " + color(f"{mfr}", fg="green") +
 105                  "\n Device ID...... " + color(f"{did:04X}h (unknown)", fg="red"))
 106     else:
 107         print("\n Manufacturer... " + color(f"{mid:02X}h (unknown)", fg="red") +
 108              "\n Device ID...... " + color(f"{did:04X}h (unknown)", fg="red"))