blob: 983923d2b6f1e62bd5fdae081547fb3e54638710 [file] [log] [blame]
Nico Huberca41a6a2016-01-23 21:29:47 +01001#!/usr/bin/env python2
Yidi Lin8e76f342015-07-31 17:10:48 +08002
3#
4# This file is part of the coreboot project.
5#
6# Copyright (c) 2015 MediaTek Inc.
7# Author: Tristan Shieh <tristan.shieh@mediatek.com>
8#
9# This program is free software; you can redistribute it and/or modify
10# it under the terms of the GNU General Public License as published by
11# the Free Software Foundation; version 2 of the License.
12#
13# This program is distributed in the hope that it will be useful,
14# but WITHOUT ANY WARRANTY; without even the implied warranty of
15# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16# GNU General Public License for more details.
17#
18
19
20import struct
21import sys
22import hashlib
23
24def read(path):
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080025 with open(path, 'rb') as f:
Yidi Lin8e76f342015-07-31 17:10:48 +080026 return f.read()
27
28def write(path, data):
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080029 with open(path, 'wb') as f:
Yidi Lin8e76f342015-07-31 17:10:48 +080030 f.write(data)
31
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080032def padding(data, size, pattern='\0'):
Yidi Lin8e76f342015-07-31 17:10:48 +080033 return data + pattern * (size - len(data))
34
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080035def align(data, size, pattern='\0'):
Yidi Lin8e76f342015-07-31 17:10:48 +080036 return padding(data, (len(data) + (size - 1)) & ~(size - 1), pattern)
37
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080038def gen_gfh_info(chip, data):
39 entries = {
40 'mt8173': 0x000C1000,
41 'mt8183': 0x00201000
42 }
43
44 gfh_format = '<44I'
45 gfh_size = struct.calcsize(gfh_format)
46 load_addr = entries[chip] - gfh_size
47 load_size = gfh_size + len(data) + hashlib.sha256().digest_size
48
49 gfh = struct.pack(gfh_format,
50 0x014d4d4d, 0x00000038, 0x454c4946, 0x464e495f,
51 0x0000004f, 0x00000001, 0x01050001, load_addr,
52 load_size, 0x00020000, 0x000000a8, 0x00000020,
53 0x000000B0, 0x00000001, 0x014d4d4d, 0x0001000c,
54 0x00000001, 0x034d4d4d, 0x00070064, 0x00001182,
55 0x00000000, 0x00000000, 0x00000000, 0x00000000,
56 0x00000000, 0x00000000, 0x00000000, 0x00000000,
57 0x00000000, 0x00000000, 0x00000000, 0x00000000,
58 0x00000000, 0x00000000, 0x00000000, 0x00000000,
59 0x00000000, 0x00000000, 0x00006400, 0x00001388,
60 0x00000000, 0x00000000, 0x00000000, 0x00000000)
61
62 return gfh
Yidi Lin8e76f342015-07-31 17:10:48 +080063
64def gen_emmc_header(data):
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080065 header = (padding(struct.pack('<12sII', 'EMMC_BOOT', 1, 512), 512, '\xff') +
66 padding(struct.pack('<8sIIIIIIII', 'BRLYT', 1, 2048, 2048 + len(data),
67 0x42424242, 0x00010005, 2048, 2048 + len(data), 1) + '\0' * 140, 512,
68 '\xff') +
Yidi Lin8e76f342015-07-31 17:10:48 +080069 '\0' * 1024)
70 return header
71
72def gen_sf_header(data):
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080073 header = (padding(struct.pack('<12sII', 'SF_BOOT', 1, 512), 512, '\xff') +
74 padding(struct.pack('<8sIIIIIIII', 'BRLYT', 1, 2048, 2048 + len(data),
75 0x42424242, 0x00010007, 2048, 2048 + len(data), 1) + '\0' * 140, 512,
76 '\xff') +
Yidi Lin8e76f342015-07-31 17:10:48 +080077 '\0' * 1024)
78 return header
79
80gen_dev_header = {
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080081 'emmc': gen_emmc_header,
82 'sf': gen_sf_header
Yidi Lin8e76f342015-07-31 17:10:48 +080083}
84
85def gen_preloader(chip_ver, flash_type, data):
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080086 gfh_info = gen_gfh_info(chip_ver, data)
Yidi Lin8e76f342015-07-31 17:10:48 +080087 gfh_hash = hashlib.sha256(gfh_info + data).digest()
88
89 data = align(gfh_info + data + gfh_hash, 512, '\xff')
90 header = gen_dev_header[flash_type](data)
91 return header + data
92
93def main(argv):
94 if len(argv) != 5:
Tristan Shieh3ddf57e2018-05-31 09:20:53 +080095 print 'Usage: %s <chip> <flash_type> <input_file> <output_file>' % argv[0]
96 print '\t flash_type: emmc|sf'
97 print '\t chip : mt8173|mt8183'
Yidi Lin8e76f342015-07-31 17:10:48 +080098
99 exit(1)
100 write(argv[4], gen_preloader(argv[1], argv[2], read(argv[3])))
101
Tristan Shieh3ddf57e2018-05-31 09:20:53 +0800102if __name__ == '__main__':
Yidi Lin8e76f342015-07-31 17:10:48 +0800103 main(sys.argv)