Merge branch 'master' of git://git.denx.de/u-boot-spi
[platform/kernel/u-boot.git] / tools / microcode-tool.py
1 #!/usr/bin/env python
2 #
3 # Copyright (c) 2014 Google, Inc
4 #
5 # SPDX-License-Identifier:      GPL-2.0+
6 #
7 # Intel microcode update tool
8
9 from optparse import OptionParser
10 import os
11 import re
12 import struct
13 import sys
14
15 MICROCODE_DIR = 'arch/x86/dts/microcode'
16
17 class Microcode:
18     """Holds information about the microcode for a particular model of CPU.
19
20     Attributes:
21         name:  Name of the CPU this microcode is for, including any version
22                    information (e.g. 'm12206a7_00000029')
23         model: Model code string (this is cpuid(1).eax, e.g. '206a7')
24         words: List of hex words containing the microcode. The first 16 words
25                    are the public header.
26     """
27     def __init__(self, name, data):
28         self.name = name
29         # Convert data into a list of hex words
30         self.words = []
31         for value in ''.join(data).split(','):
32             hexval = value.strip()
33             if hexval:
34                 self.words.append(int(hexval, 0))
35
36         # The model is in the 4rd hex word
37         self.model = '%x' % self.words[3]
38
39 def ParseFile(fname):
40     """Parse a micrcode.dat file and return the component parts
41
42     Args:
43         fname: Filename to parse
44     Returns:
45         3-Tuple:
46             date:         String containing date from the file's header
47             license_text: List of text lines for the license file
48             microcodes:   List of Microcode objects from the file
49     """
50     re_date = re.compile('/\* *(.* [0-9]{4}) *\*/$')
51     re_license = re.compile('/[^-*+] *(.*)$')
52     re_name = re.compile('/\* *(.*)\.inc *\*/', re.IGNORECASE)
53     microcodes = {}
54     license_text = []
55     date = ''
56     data = []
57     name = None
58     with open(fname) as fd:
59         for line in fd:
60             line = line.rstrip()
61             m_date = re_date.match(line)
62             m_license = re_license.match(line)
63             m_name = re_name.match(line)
64             if m_name:
65                 if name:
66                     microcodes[name] = Microcode(name, data)
67                 name = m_name.group(1).lower()
68                 data = []
69             elif m_license:
70                 license_text.append(m_license.group(1))
71             elif m_date:
72                 date = m_date.group(1)
73             else:
74                 data.append(line)
75     if name:
76         microcodes[name] = Microcode(name, data)
77     return date, license_text, microcodes
78
79 def List(date, microcodes, model):
80     """List the available microcode chunks
81
82     Args:
83         date:           Date of the microcode file
84         microcodes:     Dict of Microcode objects indexed by name
85         model:          Model string to search for, or None
86     """
87     print 'Date: %s' % date
88     if model:
89         mcode_list, tried = FindMicrocode(microcodes, model.lower())
90         print 'Matching models %s:' % (', '.join(tried))
91     else:
92         print 'All models:'
93         mcode_list = [microcodes[m] for m in microcodes.keys()]
94     for mcode in mcode_list:
95         print '%-20s: model %s' % (mcode.name, mcode.model)
96
97 def FindMicrocode(microcodes, model):
98     """Find all the microcode chunks which match the given model.
99
100     This model is something like 306a9 (the value returned in eax from
101     cpuid(1) when running on Intel CPUs). But we allow a partial match,
102     omitting the last 1 or two characters to allow many families to have the
103     same microcode.
104
105     If the model name is ambiguous we return a list of matches.
106
107     Args:
108         microcodes: Dict of Microcode objects indexed by name
109         model:      String containing model name to find
110     Returns:
111         Tuple:
112             List of matching Microcode objects
113             List of abbreviations we tried
114     """
115     # Allow a full name to be used
116     mcode = microcodes.get(model)
117     if mcode:
118         return [mcode], []
119
120     tried = []
121     found = []
122     for i in range(3):
123         abbrev = model[:-i] if i else model
124         tried.append(abbrev)
125         for mcode in microcodes.values():
126             if mcode.model.startswith(abbrev):
127                 found.append(mcode)
128         if found:
129             break
130     return found, tried
131
132 def CreateFile(date, license_text, mcode, outfile):
133     """Create a microcode file in U-Boot's .dtsi format
134
135     Args:
136         date:       String containing date of original microcode file
137         license:    List of text lines for the license file
138         mcode:      Microcode object to write
139         outfile:    Filename to write to ('-' for stdout)
140     """
141     out = '''/*%s
142  * ---
143  * This is a device tree fragment. Use #include to add these properties to a
144  * node.
145  *
146  * Date: %s
147  */
148
149 compatible = "intel,microcode";
150 intel,header-version = <%d>;
151 intel,update-revision = <%#x>;
152 intel,date-code = <%#x>;
153 intel,processor-signature = <%#x>;
154 intel,checksum = <%#x>;
155 intel,loader-revision = <%d>;
156 intel,processor-flags = <%#x>;
157
158 /* The first 48-bytes are the public header which repeats the above data */
159 data = <%s
160 \t>;'''
161     words = ''
162     for i in range(len(mcode.words)):
163         if not (i & 3):
164             words += '\n'
165         val = mcode.words[i]
166         # Change each word so it will be little-endian in the FDT
167         # This data is needed before RAM is available on some platforms so we
168         # cannot do an endianness swap on boot.
169         val = struct.unpack("<I", struct.pack(">I", val))[0]
170         words += '\t%#010x' % val
171
172     # Take care to avoid adding a space before a tab
173     text = ''
174     for line in license_text:
175         if line[0] == '\t':
176             text += '\n *' + line
177         else:
178             text += '\n * ' + line
179     args = [text, date]
180     args += [mcode.words[i] for i in range(7)]
181     args.append(words)
182     if outfile == '-':
183         print out % tuple(args)
184     else:
185         if not outfile:
186             if not os.path.exists(MICROCODE_DIR):
187                 print >> sys.stderr, "Creating directory '%s'" % MICROCODE_DIR
188                 os.makedirs(MICROCODE_DIR)
189             outfile = os.path.join(MICROCODE_DIR, mcode.name + '.dtsi')
190             print >> sys.stderr, "Writing microcode for '%s' to '%s'" % (
191                      mcode.name, outfile)
192         with open(outfile, 'w') as fd:
193             print >> fd, out % tuple(args)
194
195 def MicrocodeTool():
196     """Run the microcode tool"""
197     commands = 'create,license,list'.split(',')
198     parser = OptionParser()
199     parser.add_option('-d', '--mcfile', type='string', action='store',
200                     help='Name of microcode.dat file')
201     parser.add_option('-m', '--model', type='string', action='store',
202                     help='Model name to extract')
203     parser.add_option('-o', '--outfile', type='string', action='store',
204                     help='Filename to use for output (- for stdout), default is'
205                     ' %s/<name>.dtsi' % MICROCODE_DIR)
206     parser.usage += """ command
207
208     Process an Intel microcode file (use -h for help). Commands:
209
210        create     Create microcode .dtsi file for a model
211        list       List available models in microcode file
212        license    Print the license
213
214     Typical usage:
215
216        ./tools/microcode-tool -d microcode.dat -m 306a create
217
218     This will find the appropriate file and write it to %s.""" % MICROCODE_DIR
219
220     (options, args) = parser.parse_args()
221     if not args:
222         parser.error('Please specify a command')
223     cmd = args[0]
224     if cmd not in commands:
225         parser.error("Unknown command '%s'" % cmd)
226
227     if not options.mcfile:
228         parser.error('You must specify a microcode file')
229     date, license_text, microcodes = ParseFile(options.mcfile)
230
231     if cmd == 'list':
232         List(date, microcodes, options.model)
233     elif cmd == 'license':
234         print '\n'.join(license_text)
235     elif cmd == 'create':
236         if not options.model:
237             parser.error('You must specify a model to create')
238         model = options.model.lower()
239         mcode_list, tried = FindMicrocode(microcodes, model)
240         if not mcode_list:
241             parser.error("Unknown model '%s' (%s) - try 'list' to list" %
242                         (model, ', '.join(tried)))
243         if len(mcode_list) > 1:
244             parser.error("Ambiguous model '%s' (%s) matched %s - try 'list' "
245                         "to list or specify a particular file" %
246                         (model, ', '.join(tried),
247                         ', '.join([m.name for m in mcode_list])))
248         CreateFile(date, license_text, mcode_list[0], options.outfile)
249     else:
250         parser.error("Unknown command '%s'" % cmd)
251
252 if __name__ == "__main__":
253     MicrocodeTool()