b43: Add support for removing unnecessary firmware files to reduce the image size.
[openwrt/svn-archive/archive.git] / 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