waf-abi: auto-generate per-symbol versions from ABI files
[nivanova/samba-autobuild/.git] / buildtools / wafsamba / samba_abi.py
1 # functions for handling ABI checking of libraries
2
3 import Options, Utils, os, Logs, samba_utils, sys, Task, fnmatch, re, Build
4 from TaskGen import feature, before, after
5
6 # these type maps cope with platform specific names for common types
7 # please add new type mappings into the list below
8 abi_type_maps = {
9     '_Bool' : 'bool',
10     'struct __va_list_tag *' : 'va_list'
11     }
12
13 def normalise_signature(sig):
14     '''normalise a signature from gdb'''
15     sig = sig.strip()
16     sig = re.sub('^\$[0-9]+\s=\s\{*', '', sig)
17     sig = re.sub('\}(\s0x[0-9a-f]+\s<\w+>)?$', '', sig)
18     sig = re.sub('0x[0-9a-f]+', '0xXXXX', sig)
19
20     for t in abi_type_maps:
21         # we need to cope with non-word characters in mapped types
22         m = t
23         m = m.replace('*', '\*')
24         if m[-1].isalnum() or m[-1] == '_':
25             m += '\\b'
26         if m[0].isalnum() or m[0] == '_':
27             m = '\\b' + m
28         sig = re.sub(m, abi_type_maps[t], sig)
29     return sig
30
31 def normalise_varargs(sig):
32     '''cope with older versions of gdb'''
33     sig = re.sub(',\s\.\.\.', '', sig)
34     return sig
35
36 def parse_sigs(sigs, abi_match):
37     '''parse ABI signatures file'''
38     abi_match = samba_utils.TO_LIST(abi_match)
39     ret = {}
40     a = sigs.split('\n')
41     for s in a:
42         if s.find(':') == -1:
43             continue
44         sa = s.split(':')
45         if abi_match:
46             matched = False
47             for p in abi_match:
48                 if p[0] == '!' and fnmatch.fnmatch(sa[0], p[1:]):
49                     break
50                 elif fnmatch.fnmatch(sa[0], p):
51                     matched = True
52                     break
53             if not matched:
54                 continue
55         ret[sa[0]] = normalise_signature(sa[1])
56     return ret
57
58 def save_sigs(sig_file, parsed_sigs):
59     '''save ABI signatures to a file'''
60     sigs = ''
61     for s in sorted(parsed_sigs.keys()):
62         sigs += '%s: %s\n' % (s, parsed_sigs[s])
63     return samba_utils.save_file(sig_file, sigs, create_dir=True)
64
65
66 def abi_check_task(self):
67     '''check if the ABI has changed'''
68     abi_gen = self.ABI_GEN
69
70     libpath = self.inputs[0].abspath(self.env)
71     libname = os.path.basename(libpath)
72
73     sigs = Utils.cmd_output([abi_gen, libpath])
74     parsed_sigs = parse_sigs(sigs, self.ABI_MATCH)
75
76     sig_file = self.ABI_FILE
77
78     old_sigs = samba_utils.load_file(sig_file)
79     if old_sigs is None or Options.options.ABI_UPDATE:
80         if not save_sigs(sig_file, parsed_sigs):
81             raise Utils.WafError('Failed to save ABI file "%s"' % sig_file)
82         Logs.warn('Generated ABI signatures %s' % sig_file)
83         return
84
85     parsed_old_sigs = parse_sigs(old_sigs, self.ABI_MATCH)
86
87     # check all old sigs
88     got_error = False
89     for s in parsed_old_sigs:
90         if not s in parsed_sigs:
91             Logs.error('%s: symbol %s has been removed - please update major version\n\tsignature: %s' % (
92                 libname, s, parsed_old_sigs[s]))
93             got_error = True
94         elif normalise_varargs(parsed_old_sigs[s]) != normalise_varargs(parsed_sigs[s]):
95             Logs.error('%s: symbol %s has changed - please update major version\n\told_signature: %s\n\tnew_signature: %s' % (
96                 libname, s, parsed_old_sigs[s], parsed_sigs[s]))
97             got_error = True
98
99     for s in parsed_sigs:
100         if not s in parsed_old_sigs:
101             Logs.error('%s: symbol %s has been added - please mark it _PRIVATE_ or update minor version\n\tsignature: %s' % (
102                 libname, s, parsed_sigs[s]))
103             got_error = True
104
105     if got_error:
106         raise Utils.WafError('ABI for %s has changed - please fix library version then build with --abi-update\nSee http://wiki.samba.org/index.php/Waf#ABI_Checking for more information' % libname)
107
108
109 t = Task.task_type_from_func('abi_check', abi_check_task, color='BLUE', ext_in='.bin')
110 t.quiet = True
111 # allow "waf --abi-check" to force re-checking the ABI
112 if '--abi-check' in sys.argv:
113     Task.always_run(t)
114
115 @after('apply_link')
116 @feature('abi_check')
117 def abi_check(self):
118     '''check that ABI matches saved signatures'''
119     env = self.bld.env
120     if not env.ABI_CHECK or self.abi_directory is None:
121         return
122
123     # if the platform doesn't support -fvisibility=hidden then the ABI
124     # checks become fairly meaningless
125     if not env.HAVE_VISIBILITY_ATTR:
126         return
127
128     topsrc = self.bld.srcnode.abspath()
129     abi_gen = os.path.join(topsrc, 'buildtools/scripts/abi_gen.sh')
130
131     abi_file = "%s/%s-%s.sigs" % (self.abi_directory, self.name, self.vnum)
132
133     tsk = self.create_task('abi_check', self.link_task.outputs[0])
134     tsk.ABI_FILE = abi_file
135     tsk.ABI_MATCH = self.abi_match
136     tsk.ABI_GEN = abi_gen
137
138
139
140 def abi_process_file(fname, version, symmap):
141     '''process one ABI file, adding new symbols to the symmap'''
142     f = open(fname, mode='r')
143     for line in f:
144         symname = line.split(":")[0]
145         if not symname in symmap:
146             symmap[symname] = version
147     f.close()
148
149 def abi_write_vscript(vscript, libname, vnum, symmap):
150     '''write a vscript file for a library in --version-script format'''
151
152     libname = libname.replace("-", "_").replace("+","_").upper()
153
154     invmap = {}
155     for s in symmap:
156         invmap.setdefault(symmap[s], []).append(s)
157
158     f = open(vscript, mode='w')
159     last_key = ""
160     for k in sorted(invmap):
161         symver = "%s_%s" % (libname, k)
162         if symver == version:
163             break
164         f.write("%s {\n\tglobal: \n" % symver)
165         for s in invmap[k]:
166             f.write("\t\t%s;\n" % s);
167         f.write("}%s;\n\n" % last_key)
168         last_key = " %s" % symver
169     f.write("%s { global: *;};\n" % version)
170     f.close()
171
172
173 def abi_build_vscript(task):
174     '''generate a vscript file for our public libraries'''
175
176     tgt = task.outputs[0].bldpath(task.env)
177
178     symmap = {}
179
180     for f in task.inputs:
181         fname = f.abspath(task.env)
182         basename = os.path.basename(fname)
183         version = basename[len(task.env.LIBNAME)+1:-len(".sigs")]
184         abi_process_file(fname, version, symmap)
185     abi_write_vscript(tgt, task.env.LIBNAME, task.env.VNUM, symmap)
186
187
188 def ABI_VSCRIPT(bld, libname, abi_directory, vnum, vscript):
189     '''generate a vscript file for our public libraries'''
190     if abi_directory:
191         source = bld.path.ant_glob('%s/%s-[0-9]*.sigs' % (abi_directory, libname))
192         source = sorted(source.split())
193     else:
194         source = ''
195
196     libname = libname.replace("-", "_").replace("+","_").upper()
197     version = version.replace("-", "_").replace("+","_").upper()
198
199     t = bld.SAMBA_GENERATOR(vscript,
200                             rule=abi_build_vscript,
201                             source=source,
202                             group='vscripts',
203                             target=vscript)
204     t.env.VNUM = vnum
205     t.env.LIBNAME = libname
206     t.vars = [vnum, vscript]
207 Build.BuildContext.ABI_VSCRIPT = ABI_VSCRIPT