1*3f6f8362SLouise Kilheeney#!/usr/bin/env python3 2309bf90bSTal Shnaiderman# SPDX-License-Identifier: BSD-3-Clause 3309bf90bSTal Shnaiderman# Copyright(c) 2019 Intel Corporation 4309bf90bSTal Shnaiderman 5309bf90bSTal Shnaidermanimport sys 6309bf90bSTal Shnaidermanfrom os.path import dirname, basename, join, exists 7309bf90bSTal Shnaiderman 8309bf90bSTal Shnaiderman 9309bf90bSTal Shnaidermandef is_function_line(ln): 10309bf90bSTal Shnaiderman return ln.startswith('\t') and ln.endswith(';\n') and ":" not in ln 11309bf90bSTal Shnaiderman 12309bf90bSTal Shnaiderman# MinGW keeps the original .map file but replaces per_lcore* to __emutls_v.per_lcore* 13309bf90bSTal Shnaidermandef create_mingw_map_file(input_map, output_map): 14309bf90bSTal Shnaiderman with open(input_map) as f_in, open(output_map, 'w') as f_out: 15309bf90bSTal Shnaiderman f_out.writelines([lines.replace('per_lcore', '__emutls_v.per_lcore') for lines in f_in.readlines()]) 16309bf90bSTal Shnaiderman 17309bf90bSTal Shnaidermandef main(args): 18309bf90bSTal Shnaiderman if not args[1].endswith('version.map') or \ 19309bf90bSTal Shnaiderman not args[2].endswith('exports.def') and \ 20309bf90bSTal Shnaiderman not args[2].endswith('mingw.map'): 21309bf90bSTal Shnaiderman return 1 22309bf90bSTal Shnaiderman 23309bf90bSTal Shnaiderman if args[2].endswith('mingw.map'): 24309bf90bSTal Shnaiderman create_mingw_map_file(args[1], args[2]) 25309bf90bSTal Shnaiderman return 0 26309bf90bSTal Shnaiderman 27309bf90bSTal Shnaiderman# special case, allow override if an def file already exists alongside map file 28309bf90bSTal Shnaiderman override_file = join(dirname(args[1]), basename(args[2])) 29309bf90bSTal Shnaiderman if exists(override_file): 30309bf90bSTal Shnaiderman with open(override_file) as f_in: 31309bf90bSTal Shnaiderman functions = f_in.readlines() 32309bf90bSTal Shnaiderman 33309bf90bSTal Shnaiderman# generate def file from map file. 34309bf90bSTal Shnaiderman# This works taking indented lines only which end with a ";" and which don't 35309bf90bSTal Shnaiderman# have a colon in them, i.e. the lines defining functions only. 36309bf90bSTal Shnaiderman else: 37309bf90bSTal Shnaiderman with open(args[1]) as f_in: 38309bf90bSTal Shnaiderman functions = [ln[:-2] + '\n' for ln in sorted(f_in.readlines()) 39309bf90bSTal Shnaiderman if is_function_line(ln)] 40309bf90bSTal Shnaiderman functions = ["EXPORTS\n"] + functions 41309bf90bSTal Shnaiderman 42309bf90bSTal Shnaiderman with open(args[2], 'w') as f_out: 43309bf90bSTal Shnaiderman f_out.writelines(functions) 44309bf90bSTal Shnaiderman return 0 45309bf90bSTal Shnaiderman 46309bf90bSTal Shnaiderman 47309bf90bSTal Shnaidermanif __name__ == "__main__": 48309bf90bSTal Shnaiderman sys.exit(main(sys.argv)) 49