File: buildrom.py

package info (click to toggle)
seabios 1.16.3-2
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 3,400 kB
  • sloc: ansic: 55,254; cpp: 1,826; asm: 904; python: 821; yacc: 604; makefile: 517; perl: 416; lex: 363; sh: 287
file content (54 lines) | stat: -rwxr-xr-x 1,372 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#!/usr/bin/env python
# Fill in checksum/size of an option rom, and pad it to proper length.
#
# Copyright (C) 2009  Kevin O'Connor <kevin@koconnor.net>
#
# This file may be distributed under the terms of the GNU GPLv3 license.

import sys, struct

def alignpos(pos, alignbytes):
    mask = alignbytes - 1
    return (pos + mask) & ~mask

def checksum(data):
    if (sys.version_info > (3, 0)):
        cksum = sum(data)
    else:
        cksum = sum(map(ord, data))
    return struct.pack('<B', (0x100 - cksum) & 0xff)

def main():
    inname = sys.argv[1]
    outname = sys.argv[2]

    # Read data in
    f = open(inname, 'rb')
    data = f.read()
    f.close()
    count = len(data)

    # Pad to a 512 byte boundary
    data += b"\0" * (alignpos(count, 512) - count)
    count = len(data)

    # Check if a pci header is present
    pcidata = ord(data[24:25]) + (ord(data[25:26]) << 8)
    if pcidata != 0:
        blocks = struct.pack('<H', int(count/512))
        data = data[:pcidata + 16] + blocks + data[pcidata + 18:]

    # Fill in size field; clear checksum field
    blocks = struct.pack('<B', int(count/512))
    data = data[:2] + blocks + data[3:6] + b"\0" + data[7:]

    # Checksum rom
    data = data[:6] + checksum(data) + data[7:]

    # Write new rom
    f = open(outname, 'wb')
    f.write(data)
    f.close()

if __name__ == '__main__':
    main()