source: mainline/kernel/tools/genmap.py@ c0ea739

#!/usr/bin/env python
#
# Copyright (c) 2006 Ondrej Palkovsky
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# - Redistributions of source code must retain the above copyright
#   notice, this list of conditions and the following disclaimer.
# - Redistributions in binary form must reproduce the above copyright
#   notice, this list of conditions and the following disclaimer in the
#   documentation and/or other materials provided with the distribution.
# - The name of the author may not be used to endorse or promote products
#   derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#

"""
Create binary symbol map out of linker map file
"""

import sys
import struct
import re

MAXSTRING = 63
symtabfmt = "<Q%ds" % (MAXSTRING + 1)

funcline = re.compile(r'([0-9a-f]+)\s+[lg]\s+.\s+\.text\s+([0-9a-f]+)\s+(.*)$')
bssline = re.compile(r'([0-9a-f]+)\s+[lg]\s+[a-zA-Z]\s+\.bss\s+([0-9a-f]+)\s+(.*)$')
dataline = re.compile(r'([0-9a-f]+)\s+[lg]\s+[a-zA-Z]\s+\.data\s+([0-9a-f]+)\s+(.*)$')
fileexp = re.compile(r'([^\s]+):\s+file format')
startfile = re.compile(r'\.(text|bss|data)\s+(0x[0-9a-f]+)\s+0x[0-9a-f]+\s+(.*)$')

def read_obdump(inp):
        "Parse input"

        funcs = {}
        data = {}
        bss = {}
        fname = ''

        for line in inp:
                line = line.strip()
                res = funcline.match(line)
                if (res):
                        funcs.setdefault(fname, []).append((int(res.group(1), 16), res.group(3)))
                        continue

                res = bssline.match(line)
                if (res):
                        start = int(res.group(1), 16)
                        end = int(res.group(2), 16)
                        if (end):
                                bss.setdefault(fname, []).append((start, res.group(3)))

                res = dataline.match(line)
                if (res):
                        start = int(res.group(1), 16)
                        end = int(res.group(2), 16)
                        if (end):
                                data.setdefault(fname, []).append((start, res.group(3)))

                res = fileexp.match(line)
                if (res):
                        fname = res.group(1)
                        continue

        return {'text' : funcs, 'bss' : bss, 'data' : data}

def generate(kmapf, obmapf, out):
        "Generate output file"

        obdump = read_obdump(obmapf)

        def key_sorter(x):
                return x[0]

        for line in kmapf:
                line = line.strip()
                res = startfile.match(line)

                if ((res) and (res.group(3) in obdump[res.group(1)])):
                        offset = int(res.group(2), 16)
                        fname = res.group(3)
                        symbols = obdump[res.group(1)][fname]
                        symbols.sort(key = key_sorter)
                        for addr, symbol in symbols:
                                value = fname + ':' + symbol
                                value_bytes = value.encode('ascii')
                                data = struct.pack(symtabfmt, addr + offset, value_bytes[:MAXSTRING])
                                out.write(data)

        out.write(struct.pack(symtabfmt, 0, b''))

def main():
        if (len(sys.argv) != 4):
                print("Usage: %s <kernel.map> <nm dump> <output.bin>" % sys.argv[0])
                return 1

        kmapf = open(sys.argv[1], 'r')
        obmapf = open(sys.argv[2], 'r')
        out = open(sys.argv[3], 'wb')

        generate(kmapf, obmapf, out)

        kmapf.close()
        obmapf.close()
        out.close()

if __name__ == '__main__':
        sys.exit(main())