package/mac80211/files/host_bin/b43-fwsquash.py

   1 #!/usr/bin/env python
   2 #
   3 # b43 firmware file squasher
   4 # Removes unnecessary firmware files
   5 #
   6 # Copyright (c) 2009 Michael Buesch <mb@bu3sch.de>
   7 #
   8 # Licensed under the GNU/GPL version 2 or (at your option) any later version.
   9 #
  10
  11 import sys
  12 import os
  13
  14 def usage():
  15         print "Usage: %s PHYTYPES COREREVS /path/to/extracted/firmware" % sys.argv[0]
  16         print ""
  17         print "PHYTYPES is a comma separated list of:"
  18         print "A         => A-PHY"
  19         print "AG        => Dual A-PHY G-PHY"
  20         print "G         => G-PHY"
  21         print "LP        => LP-PHY"
  22         print "N         => N-PHY"
  23         print ""
  24         print "COREREVS is a comma separated list of core revision numbers."
  25
  26 if len(sys.argv) != 4:
  27         usage()
  28         sys.exit(1)
  29
  30 phytypes = sys.argv[1]
  31 corerevs = sys.argv[2]
  32 fwpath = sys.argv[3]
  33
  34 phytypes = phytypes.split(',')
  35 try:
  36         corerevs = map(lambda r: int(r), corerevs.split(','))
  37 except ValueError:
  38         print "ERROR: \"%s\" is not a valid COREREVS string\n" % corerevs
  39         usage()
  40         sys.exit(1)
  41
  42
  43 fwfiles = os.listdir(fwpath)
  44 fwfiles = filter(lambda str: str.endswith(".fw"), fwfiles)
  45 if not fwfiles:
  46         print "ERROR: No firmware files found in %s" % fwpath
  47         sys.exit(1)
  48
  49 required_fwfiles = []
  50
  51 def revs_match(revs_a, revs_b):
  52         for rev in revs_a:
  53                 if rev in revs_b:
  54                         return True
  55         return False
  56
  57 def phytypes_match(types_a, types_b):
  58         for type in types_a:
  59                 type = type.strip().upper()
  60                 if type in types_b:
  61                         return True
  62         return False
  63
  64 revmapping = {
  65         "ucode5.fw"             : (5,6,7,8,9,10,),
  66         "ucode11.fw"            : (11,12,),
  67         "ucode13.fw"            : (13,),
  68         "pcm4.fw"               : (1,2,3,4,),
  69         "pcm5.fw"               : (5,6,7,8,9,10,),
  70 }
  71
  72 initvalmapping = {
  73         "a0g1initvals5.fw"      : ( (5,6,7,8,9,10,),    ("AG",), ),
  74         "a0g0initvals5.fw"      : ( (5,6,7,8,9,10,),    ("A", "AG",), ),
  75         "b0g0initvals5.fw"      : ( (5,6,7,8,9,10,),    ("G",), ),
  76         "b0g0initvals13.fw"     : ( (13,),              ("G",), ),
  77         "n0initvals11.fw"       : ( (11,12,),           ("N",), ),
  78         "a0g1bsinitvals5.fw"    : ( (5,6,7,8,9,10,),    ("AG",), ),
  79         "a0g0bsinitvals5.fw"    : ( (5,6,7,8,9,10,),    ("A", "AG"), ),
  80         "b0g0bsinitvals5.fw"    : ( (5,6,7,8,9,10,),    ("G",), ),
  81         "n0bsinitvals11.fw"     : ( (11,12,),           ("N",), ),
  82 }
  83
  84 for f in fwfiles:
  85         if f in revmapping:
  86                 if revs_match(corerevs, revmapping[f]):
  87                         required_fwfiles += [f]
  88                 continue
  89         if f in initvalmapping:
  90                 if revs_match(corerevs, initvalmapping[f][0]) and\
  91                    phytypes_match(phytypes, initvalmapping[f][1]):
  92                         required_fwfiles += [f]
  93                 continue
  94         print "WARNING: Firmware file %s not found in the mapping lists" % f
  95
  96 for f in fwfiles:
  97         if f not in required_fwfiles:
  98                 print "Deleting %s" % f
  99                 os.unlink(fwpath + '/' + f)
 100