• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright 2014 The Chromium Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can b
3# found in the LICENSE file.
4
5"""Enumerates the BoringSSL source in src/ and generates two gypi files:
6  boringssl.gypi and boringssl_tests.gypi."""
7
8import os
9import subprocess
10import sys
11
12
13# OS_ARCH_COMBOS maps from OS and platform to the OpenSSL assembly "style" for
14# that platform and the extension used by asm files.
15OS_ARCH_COMBOS = [
16    ('linux', 'arm', 'elf', [''], 'S'),
17    ('linux', 'x86', 'elf', ['-fPIC'], 'S'),
18    ('linux', 'x86_64', 'elf', [''], 'S'),
19    ('mac', 'x86', 'macosx', ['-fPIC'], 'S'),
20    ('mac', 'x86_64', 'macosx', [''], 'S'),
21    ('win', 'x86_64', 'masm', [''], 'asm'),
22]
23
24# NON_PERL_FILES enumerates assembly files that are not processed by the
25# perlasm system.
26NON_PERL_FILES = {
27    ('linux', 'arm'): [
28        'src/crypto/poly1305/poly1305_arm_asm.S',
29        'src/crypto/chacha/chacha_vec_arm.S',
30    ],
31}
32
33FILE_HEADER = """# Copyright (c) 2014 The Chromium Authors. All rights reserved.
34# Use of this source code is governed by a BSD-style license that can be
35# found in the LICENSE file.
36
37# This file is created by update_gypi_and_asm.py. Do not edit manually.
38
39"""
40
41
42def FindCMakeFiles(directory):
43  """Returns list of all CMakeLists.txt files recursively in directory."""
44  cmakefiles = []
45
46  for (path, _, filenames) in os.walk(directory):
47    for filename in filenames:
48      if filename == 'CMakeLists.txt':
49        cmakefiles.append(os.path.join(path, filename))
50
51  return cmakefiles
52
53
54def NoTests(dent, is_dir):
55  """Filter function that can be passed to FindCFiles in order to remove test
56  sources."""
57  if is_dir:
58    return dent != 'test'
59  return 'test.' not in dent and not dent.startswith('example_')
60
61
62def OnlyTests(dent, is_dir):
63  """Filter function that can be passed to FindCFiles in order to remove
64  non-test sources."""
65  if is_dir:
66    return True
67  return '_test.' in dent or dent.startswith('example_')
68
69
70def FindCFiles(directory, filter_func):
71  """Recurses through directory and returns a list of paths to all the C source
72  files that pass filter_func."""
73  cfiles = []
74
75  for (path, dirnames, filenames) in os.walk(directory):
76    for filename in filenames:
77      if filename.endswith('.c') and filter_func(filename, False):
78        cfiles.append(os.path.join(path, filename))
79        continue
80
81    for (i, dirname) in enumerate(dirnames):
82      if not filter_func(dirname, True):
83        del dirnames[i]
84
85  return cfiles
86
87
88def ExtractPerlAsmFromCMakeFile(cmakefile):
89  """Parses the contents of the CMakeLists.txt file passed as an argument and
90  returns a list of all the perlasm() directives found in the file."""
91  perlasms = []
92  with open(cmakefile) as f:
93    for line in f:
94      line = line.strip()
95      if not line.startswith('perlasm('):
96        continue
97      if not line.endswith(')'):
98        raise ValueError('Bad perlasm line in %s' % cmakefile)
99      # Remove "perlasm(" from start and ")" from end
100      params = line[8:-1].split()
101      if len(params) < 2:
102        raise ValueError('Bad perlasm line in %s' % cmakefile)
103      perlasms.append({
104          'extra_args': params[2:],
105          'input': os.path.join(os.path.dirname(cmakefile), params[1]),
106          'output': os.path.join(os.path.dirname(cmakefile), params[0]),
107      })
108
109  return perlasms
110
111
112def ReadPerlAsmOperations():
113  """Returns a list of all perlasm() directives found in CMake config files in
114  src/."""
115  perlasms = []
116  cmakefiles = FindCMakeFiles('src')
117
118  for cmakefile in cmakefiles:
119    perlasms.extend(ExtractPerlAsmFromCMakeFile(cmakefile))
120
121  return perlasms
122
123
124def PerlAsm(output_filename, input_filename, perlasm_style, extra_args):
125  """Runs the a perlasm script and puts the output into output_filename."""
126  base_dir = os.path.dirname(output_filename)
127  if not os.path.isdir(base_dir):
128    os.makedirs(base_dir)
129  output = subprocess.check_output(
130      ['perl', input_filename, perlasm_style] + extra_args)
131  with open(output_filename, 'w+') as out_file:
132    out_file.write(output)
133
134
135def ArchForAsmFilename(filename):
136  """Returns the architecture that a given asm file should be compiled for
137  based on substrings in the filename."""
138
139  if 'x86_64' in filename or 'avx2' in filename:
140    return 'x86_64'
141  elif ('x86' in filename and 'x86_64' not in filename) or '586' in filename:
142    return 'x86'
143  elif 'arm' in filename:
144    return 'arm'
145  else:
146    raise ValueError('Unknown arch for asm filename: ' + filename)
147
148
149def WriteAsmFiles(perlasms):
150  """Generates asm files from perlasm directives for each supported OS x
151  platform combination."""
152  asmfiles = {}
153
154  for osarch in OS_ARCH_COMBOS:
155    (osname, arch, perlasm_style, extra_args, asm_ext) = osarch
156    key = (osname, arch)
157    outDir = '%s-%s' % key
158
159    for perlasm in perlasms:
160      filename = os.path.basename(perlasm['input'])
161      output = perlasm['output']
162      if not output.startswith('src'):
163        raise ValueError('output missing src: %s' % output)
164      output = os.path.join(outDir, output[4:])
165      output = output.replace('${ASM_EXT}', asm_ext)
166
167      if arch == ArchForAsmFilename(filename):
168        PerlAsm(output, perlasm['input'], perlasm_style,
169                perlasm['extra_args'] + extra_args)
170        asmfiles.setdefault(key, []).append(output)
171
172  for (key, non_perl_asm_files) in NON_PERL_FILES.iteritems():
173    asmfiles.setdefault(key, []).extend(non_perl_asm_files)
174
175  return asmfiles
176
177
178def PrintVariableSection(out, name, files):
179  out.write('    \'%s\': [\n' % name)
180  for f in sorted(files):
181    out.write('      \'%s\',\n' % f)
182  out.write('    ],\n')
183
184
185def main():
186  crypto_c_files = FindCFiles(os.path.join('src', 'crypto'), NoTests)
187  ssl_c_files = FindCFiles(os.path.join('src', 'ssl'), NoTests)
188
189  with open('boringssl.gypi', 'w+') as gypi:
190    gypi.write(FILE_HEADER + '{\n  \'variables\': {\n')
191
192    PrintVariableSection(
193        gypi, 'boringssl_lib_sources', crypto_c_files + ssl_c_files)
194
195    perlasms = ReadPerlAsmOperations()
196
197    for ((osname, arch), asm_files) in sorted(
198        WriteAsmFiles(perlasms).iteritems()):
199      PrintVariableSection(gypi, 'boringssl_%s_%s_sources' %
200                           (osname, arch), asm_files)
201
202    gypi.write('  }\n}\n')
203
204  test_c_files = FindCFiles(os.path.join('src', 'crypto'), OnlyTests)
205  test_c_files += FindCFiles(os.path.join('src', 'ssl'), OnlyTests)
206
207  with open('boringssl_tests.gypi', 'w+') as test_gypi:
208    test_gypi.write(FILE_HEADER + '{\n  \'targets\': [\n')
209
210    test_names = []
211    for test in sorted(test_c_files):
212      test_name = 'boringssl_%s' % os.path.splitext(os.path.basename(test))[0]
213      test_gypi.write("""    {
214      'target_name': '%s',
215      'type': 'executable',
216      'dependencies': [
217        'boringssl.gyp:boringssl',
218      ],
219      'sources': [
220        '%s',
221      ],
222    },\n""" % (test_name, test))
223      test_names.append(test_name)
224
225    test_names.sort()
226
227    test_gypi.write("""  ],
228  'variables': {
229    'boringssl_test_targets': [\n""")
230
231    for test in test_names:
232      test_gypi.write("""      '%s',\n""" % test)
233
234    test_gypi.write('    ],\n  }\n}\n')
235
236  return 0
237
238
239if __name__ == '__main__':
240  sys.exit(main())
241