blob: c567b68b85fdc9d8eadf649ad0f902d3c592035c [file] [log] [blame]
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001#!/usr/bin/env python3
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08002#
3# Copyright (C) 2019 The Android Open Source Project
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9# http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16"""Call cargo -v, parse its output, and generate Android.bp.
17
18Usage: Run this script in a crate workspace root directory.
19The Cargo.toml file should work at least for the host platform.
20
21(1) Without other flags, "cargo2android.py --run"
22 calls cargo clean, calls cargo build -v, and generates Android.bp.
23 The cargo build only generates crates for the host,
24 without test crates.
25
26(2) To build crates for both host and device in Android.bp, use the
27 --device flag, for example:
28 cargo2android.py --run --device
29
Chih-Hung Hsiehe1b7bb62020-09-30 13:09:30 -070030 Note that cargo build is only called once with the default target
31 x86_64-unknown-linux-gnu.
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080032
33(3) To build default and test crates, for host and device, use both
34 --device and --tests flags:
35 cargo2android.py --run --device --tests
36
37 This is equivalent to using the --cargo flag to add extra builds:
38 cargo2android.py --run
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080039 --cargo "build --target x86_64-unknown-linux-gnu"
40 --cargo "build --tests --target x86_64-unknown-linux-gnu"
41
Jeff Vander Stoepcec8bac2021-01-15 14:15:37 +010042 Note that when there are tests for this module or for its reverse
43 dependencies, these tests will be added to the TEST_MAPPING file.
Chih-Hung Hsiehf7eff152020-07-16 15:36:22 -070044
Chih-Hung Hsieh35ca4bc2020-07-10 16:49:51 -070045If there are rustc warning messages, this script will add
46a warning comment to the owner crate module in Android.bp.
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080047"""
48
49from __future__ import print_function
Jeff Vander Stoepcec8bac2021-01-15 14:15:37 +010050from update_crate_tests import TestMapping
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080051
52import argparse
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -070053import glob
Joel Galenson0fbdafe2021-04-21 16:33:33 -070054import json
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080055import os
56import os.path
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -070057import platform
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080058import re
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -070059import shutil
Joel Galenson7e8247e2021-05-20 18:51:42 -070060import subprocess
Andrew Walbran80e90be2020-06-09 14:33:18 +010061import sys
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080062
Chih-Hung Hsiehe02dce12020-07-14 16:05:21 -070063# Some Rust packages include extra unwanted crates.
64# This set contains all such excluded crate names.
65EXCLUDED_CRATES = set(['protobuf_bin_gen_rust_do_not_use'])
66
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080067RENAME_MAP = {
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -070068 # This map includes all changes to the default rust module names
69 # to resolve name conflicts, avoid confusion, or work as plugin.
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080070 'libbacktrace': 'libbacktrace_rust',
Andrew Walbrane51f1042020-08-11 16:42:48 +010071 'libbase': 'libbase_rust',
Victor Hsieh21bea792020-12-04 10:59:16 -080072 'libfuse': 'libfuse_rust',
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080073 'libgcc': 'libgcc_rust',
74 'liblog': 'liblog_rust',
Chih-Hung Hsieh07119862020-07-24 15:34:06 -070075 'libminijail': 'libminijail_rust',
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080076 'libsync': 'libsync_rust',
77 'libx86_64': 'libx86_64_rust',
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -070078 'protoc_gen_rust': 'protoc-gen-rust',
79}
80
81RENAME_STEM_MAP = {
82 # This map includes all changes to the default rust module stem names,
83 # which is used for output files when different from the module name.
84 'protoc_gen_rust': 'protoc-gen-rust',
85}
86
87RENAME_DEFAULTS_MAP = {
88 # This map includes all changes to the default prefix of rust_default
89 # module names, to avoid conflict with existing Android modules.
90 'libc': 'rust_libc',
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080091}
92
93# Header added to all generated Android.bp files.
Joel Galenson56446742021-02-18 08:27:48 -080094ANDROID_BP_HEADER = (
95 '// This file is generated by cargo2android.py {args}.\n' +
96 '// Do not modify this file as changes will be overridden on upgrade.\n')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -080097
98CARGO_OUT = 'cargo.out' # Name of file to keep cargo build -v output.
99
Joel Galenson3f42f802021-04-07 12:42:17 -0700100# This should be kept in sync with tools/external_updater/crates_updater.py.
101ERRORS_LINE = 'Errors in ' + CARGO_OUT + ':'
102
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800103TARGET_TMP = 'target.tmp' # Name of temporary output directory.
104
105# Message to be displayed when this script is called without the --run flag.
106DRY_RUN_NOTE = (
107 'Dry-run: This script uses ./' + TARGET_TMP + ' for output directory,\n' +
108 'runs cargo clean, runs cargo build -v, saves output to ./cargo.out,\n' +
109 'and writes to Android.bp in the current and subdirectories.\n\n' +
110 'To do do all of the above, use the --run flag.\n' +
111 'See --help for other flags, and more usage notes in this script.\n')
112
113# Cargo -v output of a call to rustc.
114RUSTC_PAT = re.compile('^ +Running `rustc (.*)`$')
115
116# Cargo -vv output of a call to rustc could be split into multiple lines.
117# Assume that the first line will contain some CARGO_* env definition.
118RUSTC_VV_PAT = re.compile('^ +Running `.*CARGO_.*=.*$')
119# The combined -vv output rustc command line pattern.
120RUSTC_VV_CMD_ARGS = re.compile('^ *Running `.*CARGO_.*=.* rustc (.*)`$')
121
122# Cargo -vv output of a "cc" or "ar" command; all in one line.
123CC_AR_VV_PAT = re.compile(r'^\[([^ ]*)[^\]]*\] running:? "(cc|ar)" (.*)$')
124# Some package, such as ring-0.13.5, has pattern '... running "cc"'.
125
126# Rustc output of file location path pattern for a warning message.
127WARNING_FILE_PAT = re.compile('^ *--> ([^:]*):[0-9]+')
128
129# Rust package name with suffix -d1.d2.d3.
130VERSION_SUFFIX_PAT = re.compile(r'^(.*)-[0-9]+\.[0-9]+\.[0-9]+$')
131
132
133def altered_name(name):
134 return RENAME_MAP[name] if (name in RENAME_MAP) else name
135
136
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700137def altered_stem(name):
138 return RENAME_STEM_MAP[name] if (name in RENAME_STEM_MAP) else name
139
140
141def altered_defaults(name):
142 return RENAME_DEFAULTS_MAP[name] if (name in RENAME_DEFAULTS_MAP) else name
143
144
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800145def is_build_crate_name(name):
146 # We added special prefix to build script crate names.
147 return name.startswith('build_script_')
148
149
150def is_dependent_file_path(path):
151 # Absolute or dependent '.../' paths are not main files of this crate.
152 return path.startswith('/') or path.startswith('.../')
153
154
155def get_module_name(crate): # to sort crates in a list
156 return crate.module_name
157
158
159def pkg2crate_name(s):
160 return s.replace('-', '_').replace('.', '_')
161
162
163def file_base_name(path):
164 return os.path.splitext(os.path.basename(path))[0]
165
166
167def test_base_name(path):
168 return pkg2crate_name(file_base_name(path))
169
170
171def unquote(s): # remove quotes around str
172 if s and len(s) > 1 and s[0] == '"' and s[-1] == '"':
173 return s[1:-1]
174 return s
175
176
177def remove_version_suffix(s): # remove -d1.d2.d3 suffix
178 if VERSION_SUFFIX_PAT.match(s):
179 return VERSION_SUFFIX_PAT.match(s).group(1)
180 return s
181
182
183def short_out_name(pkg, s): # replace /.../pkg-*/out/* with .../out/*
184 return re.sub('^/.*/' + pkg + '-[0-9a-f]*/out/', '.../out/', s)
185
186
187def escape_quotes(s): # replace '"' with '\\"'
188 return s.replace('"', '\\"')
189
190
191class Crate(object):
192 """Information of a Rust crate to collect/emit for an Android.bp module."""
193
194 def __init__(self, runner, outf_name):
195 # Remembered global runner and its members.
196 self.runner = runner
197 self.debug = runner.args.debug
198 self.cargo_dir = '' # directory of my Cargo.toml
199 self.outf_name = outf_name # path to Android.bp
200 self.outf = None # open file handle of outf_name during dump*
201 # Variants/results that could be merged from multiple rustc lines.
202 self.host_supported = False
203 self.device_supported = False
204 self.has_warning = False
205 # Android module properties derived from rustc parameters.
206 self.module_name = '' # unique in Android build system
207 self.module_type = '' # rust_{binary,library,test}[_host] etc.
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700208 self.defaults = '' # rust_defaults used by rust_test* modules
Chih-Hung Hsiehf7eff152020-07-16 15:36:22 -0700209 self.default_srcs = False # use 'srcs' defined in self.defaults
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800210 self.root_pkg = '' # parent package name of a sub/test packge, from -L
211 self.srcs = list() # main_src or merged multiple source files
212 self.stem = '' # real base name of output file
213 # Kept parsed status
214 self.errors = '' # all errors found during parsing
215 self.line_num = 1 # runner told input source line number
216 self.line = '' # original rustc command line parameters
217 # Parameters collected from rustc command line.
218 self.crate_name = '' # follows --crate-name
219 self.main_src = '' # follows crate_name parameter, shortened
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700220 self.crate_types = list() # follows --crate-type
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800221 self.cfgs = list() # follows --cfg, without feature= prefix
222 self.features = list() # follows --cfg, name in 'feature="..."'
223 self.codegens = list() # follows -C, some ignored
224 self.externs = list() # follows --extern
225 self.core_externs = list() # first part of self.externs elements
226 self.static_libs = list() # e.g. -l static=host_cpuid
227 self.shared_libs = list() # e.g. -l dylib=wayland-client, -l z
228 self.cap_lints = '' # follows --cap-lints
229 self.emit_list = '' # e.g., --emit=dep-info,metadata,link
230 self.edition = '2015' # rustc default, e.g., --edition=2018
231 self.target = '' # follows --target
232
233 def write(self, s):
234 # convenient way to output one line at a time with EOL.
235 self.outf.write(s + '\n')
236
237 def same_flags(self, other):
238 # host_supported, device_supported, has_warning are not compared but merged
239 # target is not compared, to merge different target/host modules
240 # externs is not compared; only core_externs is compared
241 return (not self.errors and not other.errors and
242 self.edition == other.edition and
243 self.cap_lints == other.cap_lints and
244 self.emit_list == other.emit_list and
245 self.core_externs == other.core_externs and
246 self.codegens == other.codegens and
247 self.features == other.features and
248 self.static_libs == other.static_libs and
249 self.shared_libs == other.shared_libs and self.cfgs == other.cfgs)
250
251 def merge_host_device(self, other):
252 """Returns true if attributes are the same except host/device support."""
253 return (self.crate_name == other.crate_name and
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700254 self.crate_types == other.crate_types and
255 self.main_src == other.main_src and
256 # before merge, each test module has an unique module name and stem
257 (self.stem == other.stem or self.crate_types == ['test']) and
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800258 self.root_pkg == other.root_pkg and not self.skip_crate() and
259 self.same_flags(other))
260
261 def merge_test(self, other):
262 """Returns true if self and other are tests of same root_pkg."""
263 # Before merger, each test has its own crate_name.
264 # A merged test uses its source file base name as output file name,
265 # so a test is mergeable only if its base name equals to its crate name.
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700266 return (self.crate_types == other.crate_types and
Chih-Hung Hsieh185052a2020-05-07 14:48:57 -0700267 self.crate_types == ['test'] and self.root_pkg == other.root_pkg and
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800268 not self.skip_crate() and
269 other.crate_name == test_base_name(other.main_src) and
270 (len(self.srcs) > 1 or
271 (self.crate_name == test_base_name(self.main_src)) and
272 self.host_supported == other.host_supported and
273 self.device_supported == other.device_supported) and
274 self.same_flags(other))
275
276 def merge(self, other, outf_name):
277 """Try to merge crate into self."""
Chih-Hung Hsiehe1b7bb62020-09-30 13:09:30 -0700278 # Cargo build --tests could recompile a library for tests.
279 # We need to merge such duplicated calls to rustc, with
280 # the algorithm in merge_host_device.
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800281 should_merge_host_device = self.merge_host_device(other)
282 should_merge_test = False
283 if not should_merge_host_device:
284 should_merge_test = self.merge_test(other)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800285 if should_merge_host_device or should_merge_test:
286 self.runner.init_bp_file(outf_name)
287 with open(outf_name, 'a') as outf: # to write debug info
288 self.outf = outf
289 other.outf = outf
290 self.do_merge(other, should_merge_test)
291 return True
292 return False
293
294 def do_merge(self, other, should_merge_test):
295 """Merge attributes of other to self."""
296 if self.debug:
297 self.write('\n// Before merge definition (1):')
298 self.dump_debug_info()
299 self.write('\n// Before merge definition (2):')
300 other.dump_debug_info()
301 # Merge properties of other to self.
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800302 self.has_warning = self.has_warning or other.has_warning
303 if not self.target: # okay to keep only the first target triple
304 self.target = other.target
305 # decide_module_type sets up default self.stem,
306 # which can be changed if self is a merged test module.
307 self.decide_module_type()
308 if should_merge_test:
309 self.srcs.append(other.main_src)
310 # use a short unique name as the merged module name.
311 prefix = self.root_pkg + '_tests'
312 self.module_name = self.runner.claim_module_name(prefix, self, 0)
313 self.stem = self.module_name
314 # This normalized root_pkg name although might be the same
315 # as other module's crate_name, it is not actually used for
316 # output file name. A merged test module always have multiple
317 # source files and each source file base name is used as
318 # its output file name.
319 self.crate_name = pkg2crate_name(self.root_pkg)
320 if self.debug:
321 self.write('\n// After merge definition (1):')
322 self.dump_debug_info()
323
324 def find_cargo_dir(self):
325 """Deepest directory with Cargo.toml and contains the main_src."""
326 if not is_dependent_file_path(self.main_src):
327 dir_name = os.path.dirname(self.main_src)
328 while dir_name:
329 if os.path.exists(dir_name + '/Cargo.toml'):
330 self.cargo_dir = dir_name
331 return
332 dir_name = os.path.dirname(dir_name)
333
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -0700334 def add_codegens_flag(self, flag):
Chih-Hung Hsiehe1b7bb62020-09-30 13:09:30 -0700335 """Ignore options not used in Android."""
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -0700336 # 'prefer-dynamic' does not work with common flag -C lto
Chih-Hung Hsieh63459ed2020-08-26 11:51:15 -0700337 # 'embed-bitcode' is ignored; we might control LTO with other .bp flag
Chih-Hung Hsieh6c13b722020-09-11 21:24:03 -0700338 # 'codegen-units' is set in Android global config or by default
339 if not (flag.startswith('codegen-units=') or
340 flag.startswith('debuginfo=') or
Chih-Hung Hsieh63459ed2020-08-26 11:51:15 -0700341 flag.startswith('embed-bitcode=') or
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -0700342 flag.startswith('extra-filename=') or
343 flag.startswith('incremental=') or
344 flag.startswith('metadata=') or
345 flag == 'prefer-dynamic'):
346 self.codegens.append(flag)
347
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800348 def parse(self, line_num, line):
349 """Find important rustc arguments to convert to Android.bp properties."""
350 self.line_num = line_num
351 self.line = line
352 args = line.split() # Loop through every argument of rustc.
353 i = 0
354 while i < len(args):
355 arg = args[i]
356 if arg == '--crate-name':
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700357 i += 1
358 self.crate_name = args[i]
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800359 elif arg == '--crate-type':
360 i += 1
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700361 # cargo calls rustc with multiple --crate-type flags.
362 # rustc can accept:
363 # --crate-type [bin|lib|rlib|dylib|cdylib|staticlib|proc-macro]
364 self.crate_types.append(args[i])
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800365 elif arg == '--test':
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700366 self.crate_types.append('test')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800367 elif arg == '--target':
368 i += 1
369 self.target = args[i]
370 elif arg == '--cfg':
371 i += 1
372 if args[i].startswith('\'feature='):
373 self.features.append(unquote(args[i].replace('\'feature=', '')[:-1]))
374 else:
375 self.cfgs.append(args[i])
376 elif arg == '--extern':
377 i += 1
378 extern_names = re.sub('=/[^ ]*/deps/', ' = ', args[i])
379 self.externs.append(extern_names)
380 self.core_externs.append(re.sub(' = .*', '', extern_names))
381 elif arg == '-C': # codegen options
382 i += 1
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -0700383 self.add_codegens_flag(args[i])
384 elif arg.startswith('-C'):
385 # cargo has been passing "-C <xyz>" flag to rustc,
386 # but newer cargo could pass '-Cembed-bitcode=no' to rustc.
387 self.add_codegens_flag(arg[2:])
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800388 elif arg == '--cap-lints':
389 i += 1
390 self.cap_lints = args[i]
391 elif arg == '-L':
392 i += 1
393 if args[i].startswith('dependency=') and args[i].endswith('/deps'):
394 if '/' + TARGET_TMP + '/' in args[i]:
395 self.root_pkg = re.sub(
396 '^.*/', '', re.sub('/' + TARGET_TMP + '/.*/deps$', '', args[i]))
397 else:
398 self.root_pkg = re.sub('^.*/', '',
399 re.sub('/[^/]+/[^/]+/deps$', '', args[i]))
400 self.root_pkg = remove_version_suffix(self.root_pkg)
401 elif arg == '-l':
402 i += 1
403 if args[i].startswith('static='):
404 self.static_libs.append(re.sub('static=', '', args[i]))
405 elif args[i].startswith('dylib='):
406 self.shared_libs.append(re.sub('dylib=', '', args[i]))
407 else:
408 self.shared_libs.append(args[i])
409 elif arg == '--out-dir' or arg == '--color': # ignored
410 i += 1
411 elif arg.startswith('--error-format=') or arg.startswith('--json='):
412 _ = arg # ignored
413 elif arg.startswith('--emit='):
414 self.emit_list = arg.replace('--emit=', '')
415 elif arg.startswith('--edition='):
416 self.edition = arg.replace('--edition=', '')
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700417 elif not arg.startswith('-'):
418 # shorten imported crate main source paths like $HOME/.cargo/
419 # registry/src/github.com-1ecc6299db9ec823/memchr-2.3.3/src/lib.rs
420 self.main_src = re.sub(r'^/[^ ]*/registry/src/', '.../', args[i])
421 self.main_src = re.sub(r'^\.\.\./github.com-[0-9a-f]*/', '.../',
422 self.main_src)
423 self.find_cargo_dir()
Chih-Hung Hsieh07119862020-07-24 15:34:06 -0700424 if self.cargo_dir: # for a subdirectory
425 if self.runner.args.no_subdir: # all .bp content to /dev/null
426 self.outf_name = '/dev/null'
427 elif not self.runner.args.onefile:
428 # Write to Android.bp in the subdirectory with Cargo.toml.
429 self.outf_name = self.cargo_dir + '/Android.bp'
430 self.main_src = self.main_src[len(self.cargo_dir) + 1:]
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800431 else:
432 self.errors += 'ERROR: unknown ' + arg + '\n'
433 i += 1
434 if not self.crate_name:
435 self.errors += 'ERROR: missing --crate-name\n'
436 if not self.main_src:
437 self.errors += 'ERROR: missing main source file\n'
438 else:
439 self.srcs.append(self.main_src)
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700440 if not self.crate_types:
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800441 # Treat "--cfg test" as "--test"
442 if 'test' in self.cfgs:
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700443 self.crate_types.append('test')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800444 else:
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700445 self.errors += 'ERROR: missing --crate-type or --test\n'
446 elif len(self.crate_types) > 1:
447 if 'test' in self.crate_types:
448 self.errors += 'ERROR: cannot handle both --crate-type and --test\n'
449 if 'lib' in self.crate_types and 'rlib' in self.crate_types:
450 self.errors += 'ERROR: cannot generate both lib and rlib crate types\n'
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800451 if not self.root_pkg:
452 self.root_pkg = self.crate_name
Chih-Hung Hsiehe1b7bb62020-09-30 13:09:30 -0700453 self.device_supported = self.runner.args.device
454 self.host_supported = not self.runner.args.no_host
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800455 self.cfgs = sorted(set(self.cfgs))
456 self.features = sorted(set(self.features))
457 self.codegens = sorted(set(self.codegens))
458 self.externs = sorted(set(self.externs))
459 self.core_externs = sorted(set(self.core_externs))
460 self.static_libs = sorted(set(self.static_libs))
461 self.shared_libs = sorted(set(self.shared_libs))
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700462 self.crate_types = sorted(set(self.crate_types))
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800463 self.decide_module_type()
464 self.module_name = altered_name(self.stem)
465 return self
466
467 def dump_line(self):
468 self.write('\n// Line ' + str(self.line_num) + ' ' + self.line)
469
470 def feature_list(self):
471 """Return a string of main_src + "feature_list"."""
472 pkg = self.main_src
473 if pkg.startswith('.../'): # keep only the main package name
474 pkg = re.sub('/.*', '', pkg[4:])
Chih-Hung Hsieh07119862020-07-24 15:34:06 -0700475 elif pkg.startswith('/'): # use relative path for a local package
476 pkg = os.path.relpath(pkg)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800477 if not self.features:
478 return pkg
479 return pkg + ' "' + ','.join(self.features) + '"'
480
481 def dump_skip_crate(self, kind):
482 if self.debug:
483 self.write('\n// IGNORED: ' + kind + ' ' + self.main_src)
484 return self
485
486 def skip_crate(self):
487 """Return crate_name or a message if this crate should be skipped."""
Chih-Hung Hsiehe02dce12020-07-14 16:05:21 -0700488 if (is_build_crate_name(self.crate_name) or
489 self.crate_name in EXCLUDED_CRATES):
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800490 return self.crate_name
491 if is_dependent_file_path(self.main_src):
492 return 'dependent crate'
493 return ''
494
495 def dump(self):
496 """Dump all error/debug/module code to the output .bp file."""
497 self.runner.init_bp_file(self.outf_name)
498 with open(self.outf_name, 'a') as outf:
499 self.outf = outf
500 if self.errors:
501 self.dump_line()
502 self.write(self.errors)
503 elif self.skip_crate():
504 self.dump_skip_crate(self.skip_crate())
505 else:
506 if self.debug:
507 self.dump_debug_info()
508 self.dump_android_module()
509
510 def dump_debug_info(self):
511 """Dump parsed data, when cargo2android is called with --debug."""
512
513 def dump(name, value):
514 self.write('//%12s = %s' % (name, value))
515
516 def opt_dump(name, value):
517 if value:
518 dump(name, value)
519
520 def dump_list(fmt, values):
521 for v in values:
522 self.write(fmt % v)
523
524 self.dump_line()
525 dump('module_name', self.module_name)
526 dump('crate_name', self.crate_name)
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700527 dump('crate_types', self.crate_types)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800528 dump('main_src', self.main_src)
529 dump('has_warning', self.has_warning)
530 dump('for_host', self.host_supported)
531 dump('for_device', self.device_supported)
532 dump('module_type', self.module_type)
533 opt_dump('target', self.target)
534 opt_dump('edition', self.edition)
535 opt_dump('emit_list', self.emit_list)
536 opt_dump('cap_lints', self.cap_lints)
537 dump_list('// cfg = %s', self.cfgs)
538 dump_list('// cfg = \'feature "%s"\'', self.features)
539 # TODO(chh): escape quotes in self.features, but not in other dump_list
540 dump_list('// codegen = %s', self.codegens)
541 dump_list('// externs = %s', self.externs)
542 dump_list('// -l static = %s', self.static_libs)
543 dump_list('// -l (dylib) = %s', self.shared_libs)
544
545 def dump_android_module(self):
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700546 """Dump one or more Android module definition, depending on crate_types."""
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700547 if len(self.crate_types) == 1:
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700548 self.dump_single_type_android_module()
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700549 return
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700550 if 'test' in self.crate_types:
551 self.write('\nERROR: multiple crate types cannot include test type')
552 return
553 # Dump one Android module per crate_type.
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700554 for crate_type in self.crate_types:
555 self.decide_one_module_type(crate_type)
556 self.dump_one_android_module(crate_type)
557
Chih-Hung Hsiehe02dce12020-07-14 16:05:21 -0700558 def build_default_name(self):
559 """Return a short and readable name for the rust_defaults module."""
560 # Choices: (1) root_pkg + '_defaults',
561 # (2) root_pkg + '_defaults_' + crate_name
562 # (3) root_pkg + '_defaults_' + main_src_basename_path
563 # (4) root_pkg + '_defaults_' + a_positive_sequence_number
564 name1 = altered_defaults(self.root_pkg) + '_defaults'
565 if self.runner.try_claim_module_name(name1, self):
566 return name1
567 name2 = name1 + '_' + self.crate_name
568 if self.runner.try_claim_module_name(name2, self):
569 return name2
570 name3 = name1 + '_' + self.main_src_basename_path()
571 if self.runner.try_claim_module_name(name3, self):
572 return name3
573 return self.runner.claim_module_name(name1, self, 0)
574
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -0700575 def dump_srcs_list(self):
576 """Dump the srcs list, for defaults or regular modules."""
577 if len(self.srcs) > 1:
578 srcs = sorted(set(self.srcs)) # make a copy and dedup
579 else:
580 srcs = [self.main_src]
581 copy_out = self.runner.copy_out_module_name()
582 if copy_out:
583 srcs.append(':' + copy_out)
584 self.dump_android_property_list('srcs', '"%s"', srcs)
585
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700586 def dump_defaults_module(self):
587 """Dump a rust_defaults module to be shared by other modules."""
Chih-Hung Hsiehe02dce12020-07-14 16:05:21 -0700588 name = self.build_default_name()
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700589 self.defaults = name
590 self.write('\nrust_defaults {')
591 self.write(' name: "' + name + '",')
Chih-Hung Hsieh07119862020-07-24 15:34:06 -0700592 if self.runner.args.global_defaults:
593 self.write(' defaults: ["' + self.runner.args.global_defaults + '"],')
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700594 self.write(' crate_name: "' + self.crate_name + '",')
Chih-Hung Hsiehf7eff152020-07-16 15:36:22 -0700595 if len(self.srcs) == 1: # only one source file; share it in defaults
596 self.default_srcs = True
597 if self.has_warning and not self.cap_lints:
598 self.write(' // has rustc warnings')
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -0700599 self.dump_srcs_list()
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700600 if 'test' in self.crate_types:
601 self.write(' test_suites: ["general-tests"],')
602 self.write(' auto_gen_config: true,')
603 self.dump_edition_flags_libs()
604 self.write('}')
605
606 def dump_single_type_android_module(self):
607 """Dump one simple Android module, which has only one crate_type."""
608 crate_type = self.crate_types[0]
609 if crate_type != 'test':
610 # do not change self.stem or self.module_name
611 self.dump_one_android_module(crate_type)
612 return
613 # Dump one test module per source file, and separate host and device tests.
614 # crate_type == 'test'
Joel Galensonf6b3c912021-06-03 16:00:54 -0700615 self.srcs = [src for src in self.srcs if not src in self.runner.args.test_blocklist]
616 if ((self.host_supported and self.device_supported and len(self.srcs) > 0) or
617 len(self.srcs) > 1):
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700618 self.srcs = sorted(set(self.srcs))
619 self.dump_defaults_module()
620 saved_srcs = self.srcs
621 for src in saved_srcs:
622 self.srcs = [src]
623 saved_device_supported = self.device_supported
624 saved_host_supported = self.host_supported
625 saved_main_src = self.main_src
626 self.main_src = src
627 if saved_host_supported:
628 self.device_supported = False
629 self.host_supported = True
630 self.module_name = self.test_module_name()
631 self.decide_one_module_type(crate_type)
632 self.dump_one_android_module(crate_type)
633 if saved_device_supported:
634 self.device_supported = True
635 self.host_supported = False
636 self.module_name = self.test_module_name()
637 self.decide_one_module_type(crate_type)
638 self.dump_one_android_module(crate_type)
639 self.host_supported = saved_host_supported
640 self.device_supported = saved_device_supported
641 self.main_src = saved_main_src
642 self.srcs = saved_srcs
643
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700644 def dump_one_android_module(self, crate_type):
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800645 """Dump one Android module definition."""
646 if not self.module_type:
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700647 self.write('\nERROR: unknown crate_type ' + crate_type)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800648 return
649 self.write('\n' + self.module_type + ' {')
650 self.dump_android_core_properties()
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700651 if not self.defaults:
652 self.dump_edition_flags_libs()
653 if self.runner.args.host_first_multilib and self.host_supported and crate_type != 'test':
654 self.write(' compile_multilib: "first",')
Joel Galensond9c4de62021-04-23 10:26:40 -0700655 if self.runner.args.apex_available and crate_type == 'lib':
656 self.write(' apex_available: [')
657 for apex in self.runner.args.apex_available:
658 self.write(' "%s",' % apex)
659 self.write(' ],')
660 if self.runner.args.min_sdk_version and crate_type == 'lib':
661 self.write(' min_sdk_version: "%s",' % self.runner.args.min_sdk_version)
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700662 self.write('}')
663
664 def dump_android_flags(self):
665 """Dump Android module flags property."""
Thiébaud Weksteena5a728b2021-04-08 14:23:49 +0200666 if not self.codegens and not self.cap_lints:
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -0700667 return
668 self.write(' flags: [')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800669 if self.cap_lints:
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -0700670 self.write(' "--cap-lints ' + self.cap_lints + '",')
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -0700671 codegens_fmt = '"-C %s"'
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -0700672 self.dump_android_property_list_items(codegens_fmt, self.codegens)
673 self.write(' ],')
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700674
675 def dump_edition_flags_libs(self):
676 if self.edition:
677 self.write(' edition: "' + self.edition + '",')
678 self.dump_android_property_list('features', '"%s"', self.features)
Thiébaud Weksteena5a728b2021-04-08 14:23:49 +0200679 self.dump_android_property_list('cfgs', '"%s"', self.cfgs)
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700680 self.dump_android_flags()
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800681 if self.externs:
682 self.dump_android_externs()
683 self.dump_android_property_list('static_libs', '"lib%s"', self.static_libs)
684 self.dump_android_property_list('shared_libs', '"lib%s"', self.shared_libs)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800685
Chih-Hung Hsiehe02dce12020-07-14 16:05:21 -0700686 def main_src_basename_path(self):
687 return re.sub('/', '_', re.sub('.rs$', '', self.main_src))
688
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800689 def test_module_name(self):
690 """Return a unique name for a test module."""
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700691 # root_pkg+(_host|_device) + '_test_'+source_file_name
Chih-Hung Hsiehe02dce12020-07-14 16:05:21 -0700692 suffix = self.main_src_basename_path()
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700693 host_device = '_host'
694 if self.device_supported:
695 host_device = '_device'
696 return self.root_pkg + host_device + '_test_' + suffix
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800697
698 def decide_module_type(self):
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700699 # Use the first crate type for the default/first module.
700 crate_type = self.crate_types[0] if self.crate_types else ''
701 self.decide_one_module_type(crate_type)
702
703 def decide_one_module_type(self, crate_type):
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800704 """Decide which Android module type to use."""
705 host = '' if self.device_supported else '_host'
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700706 if crate_type == 'bin': # rust_binary[_host]
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800707 self.module_type = 'rust_binary' + host
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700708 # In rare cases like protobuf-codegen, the output binary name must
709 # be renamed to use as a plugin for protoc.
710 self.stem = altered_stem(self.crate_name)
711 self.module_name = altered_name(self.crate_name)
Chih-Hung Hsieh35ca4bc2020-07-10 16:49:51 -0700712 elif crate_type == 'lib': # rust_library[_host]
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700713 # TODO(chh): should this be rust_library[_host]?
714 # Assuming that Cargo.toml do not use both 'lib' and 'rlib',
715 # because we map them both to rlib.
Chih-Hung Hsieh35ca4bc2020-07-10 16:49:51 -0700716 self.module_type = 'rust_library' + host
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800717 self.stem = 'lib' + self.crate_name
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700718 self.module_name = altered_name(self.stem)
Chih-Hung Hsieh35ca4bc2020-07-10 16:49:51 -0700719 elif crate_type == 'rlib': # rust_library[_host]
720 self.module_type = 'rust_library' + host
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700721 self.stem = 'lib' + self.crate_name
722 self.module_name = altered_name(self.stem)
723 elif crate_type == 'dylib': # rust_library[_host]_dylib
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800724 self.module_type = 'rust_library' + host + '_dylib'
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700725 self.stem = 'lib' + self.crate_name
726 self.module_name = altered_name(self.stem) + '_dylib'
727 elif crate_type == 'cdylib': # rust_library[_host]_shared
Ivan Lozano0c057ad2020-12-15 10:41:26 -0500728 self.module_type = 'rust_ffi' + host + '_shared'
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700729 self.stem = 'lib' + self.crate_name
730 self.module_name = altered_name(self.stem) + '_shared'
731 elif crate_type == 'staticlib': # rust_library[_host]_static
Ivan Lozano0c057ad2020-12-15 10:41:26 -0500732 self.module_type = 'rust_ffi' + host + '_static'
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700733 self.stem = 'lib' + self.crate_name
734 self.module_name = altered_name(self.stem) + '_static'
735 elif crate_type == 'test': # rust_test[_host]
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800736 self.module_type = 'rust_test' + host
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700737 # Before do_merge, stem name is based on the --crate-name parameter.
738 # and test module name is based on stem.
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800739 self.stem = self.test_module_name()
740 # self.stem will be changed after merging with other tests.
741 # self.stem is NOT used for final test binary name.
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700742 # rust_test uses each source file base name as part of output file name.
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700743 # In do_merge, this function is called again, with a module_name.
744 # We make sure that the module name is unique in each package.
745 if self.module_name:
746 # Cargo uses "-C extra-filename=..." and "-C metadata=..." to add
747 # different suffixes and distinguish multiple tests of the same
748 # crate name. We ignore -C and use claim_module_name to get
749 # unique sequential suffix.
750 self.module_name = self.runner.claim_module_name(
751 self.module_name, self, 0)
752 # Now the module name is unique, stem should also match and unique.
753 self.stem = self.module_name
754 elif crate_type == 'proc-macro': # rust_proc_macro
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800755 self.module_type = 'rust_proc_macro'
756 self.stem = 'lib' + self.crate_name
Chih-Hung Hsieh8a1a2302020-04-03 14:33:33 -0700757 self.module_name = altered_name(self.stem)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800758 else: # unknown module type, rust_prebuilt_dylib? rust_library[_host]?
759 self.module_type = ''
760 self.stem = ''
761
762 def dump_android_property_list_items(self, fmt, values):
763 for v in values:
764 # fmt has quotes, so we need escape_quotes(v)
765 self.write(' ' + (fmt % escape_quotes(v)) + ',')
766
767 def dump_android_property_list(self, name, fmt, values):
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -0700768 if not values:
769 return
770 if len(values) > 1:
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800771 self.write(' ' + name + ': [')
772 self.dump_android_property_list_items(fmt, values)
773 self.write(' ],')
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -0700774 else:
775 self.write(' ' + name + ': [' +
776 (fmt % escape_quotes(values[0])) + '],')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800777
778 def dump_android_core_properties(self):
779 """Dump the module header, name, stem, etc."""
780 self.write(' name: "' + self.module_name + '",')
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700781 # see properties shared by dump_defaults_module
782 if self.defaults:
783 self.write(' defaults: ["' + self.defaults + '"],')
Chih-Hung Hsieh07119862020-07-24 15:34:06 -0700784 elif self.runner.args.global_defaults:
785 self.write(' defaults: ["' + self.runner.args.global_defaults + '"],')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800786 if self.stem != self.module_name:
787 self.write(' stem: "' + self.stem + '",')
Chih-Hung Hsiehf7eff152020-07-16 15:36:22 -0700788 if self.has_warning and not self.cap_lints and not self.default_srcs:
Chih-Hung Hsieh35ca4bc2020-07-10 16:49:51 -0700789 self.write(' // has rustc warnings')
Chih-Hung Hsiehe1b7bb62020-09-30 13:09:30 -0700790 if self.host_supported and self.device_supported and self.module_type != 'rust_proc_macro':
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800791 self.write(' host_supported: true,')
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700792 if not self.defaults:
793 self.write(' crate_name: "' + self.crate_name + '",')
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -0700794 if not self.default_srcs:
795 self.dump_srcs_list()
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700796 if 'test' in self.crate_types and not self.defaults:
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800797 # self.root_pkg can have multiple test modules, with different *_tests[n]
798 # names, but their executables can all be installed under the same _tests
799 # directory. When built from Cargo.toml, all tests should have different
Chih-Hung Hsieh185052a2020-05-07 14:48:57 -0700800 # file or crate names. So we used (root_pkg + '_tests') name as the
801 # relative_install_path.
802 # However, some package like 'slab' can have non-mergeable tests that
803 # must be separated by different module names. So, here we no longer
804 # emit relative_install_path.
805 # self.write(' relative_install_path: "' + self.root_pkg + '_tests",')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800806 self.write(' test_suites: ["general-tests"],')
807 self.write(' auto_gen_config: true,')
Joel Galensone261a152021-01-12 11:31:53 -0800808 if 'test' in self.crate_types and self.host_supported:
809 self.write(' test_options: {')
810 self.write(' unit_test: true,')
811 self.write(' },')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800812
813 def dump_android_externs(self):
814 """Dump the dependent rlibs and dylibs property."""
815 so_libs = list()
816 rust_libs = ''
817 deps_libname = re.compile('^.* = lib(.*)-[0-9a-f]*.(rlib|so|rmeta)$')
818 for lib in self.externs:
819 # normal value of lib: "libc = liblibc-*.rlib"
820 # strange case in rand crate: "getrandom_package = libgetrandom-*.rlib"
821 # we should use "libgetrandom", not "lib" + "getrandom_package"
822 groups = deps_libname.match(lib)
823 if groups is not None:
824 lib_name = groups.group(1)
825 else:
826 lib_name = re.sub(' .*$', '', lib)
Joel Galenson97e414a2021-05-27 09:42:32 -0700827 if lib_name in self.runner.args.dependency_blocklist:
828 continue
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800829 if lib.endswith('.rlib') or lib.endswith('.rmeta'):
830 # On MacOS .rmeta is used when Linux uses .rlib or .rmeta.
831 rust_libs += ' "' + altered_name('lib' + lib_name) + '",\n'
832 elif lib.endswith('.so'):
833 so_libs.append(lib_name)
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -0700834 elif lib != 'proc_macro': # --extern proc_macro is special and ignored
835 rust_libs += ' // ERROR: unknown type of lib ' + lib + '\n'
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800836 if rust_libs:
Chih-Hung Hsieh35ca4bc2020-07-10 16:49:51 -0700837 self.write(' rustlibs: [\n' + rust_libs + ' ],')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -0800838 # Are all dependent .so files proc_macros?
839 # TODO(chh): Separate proc_macros and dylib.
840 self.dump_android_property_list('proc_macros', '"lib%s"', so_libs)
841
842
843class ARObject(object):
844 """Information of an "ar" link command."""
845
846 def __init__(self, runner, outf_name):
847 # Remembered global runner and its members.
848 self.runner = runner
849 self.pkg = ''
850 self.outf_name = outf_name # path to Android.bp
851 # "ar" arguments
852 self.line_num = 1
853 self.line = ''
854 self.flags = '' # e.g. "crs"
855 self.lib = '' # e.g. "/.../out/lib*.a"
856 self.objs = list() # e.g. "/.../out/.../*.o"
857
858 def parse(self, pkg, line_num, args_line):
859 """Collect ar obj/lib file names."""
860 self.pkg = pkg
861 self.line_num = line_num
862 self.line = args_line
863 args = args_line.split()
864 num_args = len(args)
865 if num_args < 3:
866 print('ERROR: "ar" command has too few arguments', args_line)
867 else:
868 self.flags = unquote(args[0])
869 self.lib = unquote(args[1])
870 self.objs = sorted(set(map(unquote, args[2:])))
871 return self
872
873 def write(self, s):
874 self.outf.write(s + '\n')
875
876 def dump_debug_info(self):
877 self.write('\n// Line ' + str(self.line_num) + ' "ar" ' + self.line)
878 self.write('// ar_object for %12s' % self.pkg)
879 self.write('// flags = %s' % self.flags)
880 self.write('// lib = %s' % short_out_name(self.pkg, self.lib))
881 for o in self.objs:
882 self.write('// obj = %s' % short_out_name(self.pkg, o))
883
884 def dump_android_lib(self):
885 """Write cc_library_static into Android.bp."""
886 self.write('\ncc_library_static {')
887 self.write(' name: "' + file_base_name(self.lib) + '",')
888 self.write(' host_supported: true,')
889 if self.flags != 'crs':
890 self.write(' // ar flags = %s' % self.flags)
891 if self.pkg not in self.runner.pkg_obj2cc:
892 self.write(' ERROR: cannot find source files.\n}')
893 return
894 self.write(' srcs: [')
895 obj2cc = self.runner.pkg_obj2cc[self.pkg]
896 # Note: wflags are ignored.
897 dflags = list()
898 fflags = list()
899 for obj in self.objs:
900 self.write(' "' + short_out_name(self.pkg, obj2cc[obj].src) + '",')
901 # TODO(chh): union of dflags and flags of all obj
902 # Now, just a temporary hack that uses the last obj's flags
903 dflags = obj2cc[obj].dflags
904 fflags = obj2cc[obj].fflags
905 self.write(' ],')
906 self.write(' cflags: [')
907 self.write(' "-O3",') # TODO(chh): is this default correct?
908 self.write(' "-Wno-error",')
909 for x in fflags:
910 self.write(' "-f' + x + '",')
911 for x in dflags:
912 self.write(' "-D' + x + '",')
913 self.write(' ],')
914 self.write('}')
915
916 def dump(self):
917 """Dump error/debug/module info to the output .bp file."""
918 self.runner.init_bp_file(self.outf_name)
919 with open(self.outf_name, 'a') as outf:
920 self.outf = outf
921 if self.runner.args.debug:
922 self.dump_debug_info()
923 self.dump_android_lib()
924
925
926class CCObject(object):
927 """Information of a "cc" compilation command."""
928
929 def __init__(self, runner, outf_name):
930 # Remembered global runner and its members.
931 self.runner = runner
932 self.pkg = ''
933 self.outf_name = outf_name # path to Android.bp
934 # "cc" arguments
935 self.line_num = 1
936 self.line = ''
937 self.src = ''
938 self.obj = ''
939 self.dflags = list() # -D flags
940 self.fflags = list() # -f flags
941 self.iflags = list() # -I flags
942 self.wflags = list() # -W flags
943 self.other_args = list()
944
945 def parse(self, pkg, line_num, args_line):
946 """Collect cc compilation flags and src/out file names."""
947 self.pkg = pkg
948 self.line_num = line_num
949 self.line = args_line
950 args = args_line.split()
951 i = 0
952 while i < len(args):
953 arg = args[i]
954 if arg == '"-c"':
955 i += 1
956 if args[i].startswith('"-o'):
957 # ring-0.13.5 dumps: ... "-c" "-o/.../*.o" ".../*.c"
958 self.obj = unquote(args[i])[2:]
959 i += 1
960 self.src = unquote(args[i])
961 else:
962 self.src = unquote(args[i])
963 elif arg == '"-o"':
964 i += 1
965 self.obj = unquote(args[i])
966 elif arg == '"-I"':
967 i += 1
968 self.iflags.append(unquote(args[i]))
969 elif arg.startswith('"-D'):
970 self.dflags.append(unquote(args[i])[2:])
971 elif arg.startswith('"-f'):
972 self.fflags.append(unquote(args[i])[2:])
973 elif arg.startswith('"-W'):
974 self.wflags.append(unquote(args[i])[2:])
975 elif not (arg.startswith('"-O') or arg == '"-m64"' or arg == '"-g"' or
976 arg == '"-g3"'):
977 # ignore -O -m64 -g
978 self.other_args.append(unquote(args[i]))
979 i += 1
980 self.dflags = sorted(set(self.dflags))
981 self.fflags = sorted(set(self.fflags))
982 # self.wflags is not sorted because some are order sensitive
983 # and we ignore them anyway.
984 if self.pkg not in self.runner.pkg_obj2cc:
985 self.runner.pkg_obj2cc[self.pkg] = {}
986 self.runner.pkg_obj2cc[self.pkg][self.obj] = self
987 return self
988
989 def write(self, s):
990 self.outf.write(s + '\n')
991
992 def dump_debug_flags(self, name, flags):
993 self.write('// ' + name + ':')
994 for f in flags:
995 self.write('// %s' % f)
996
997 def dump(self):
998 """Dump only error/debug info to the output .bp file."""
999 if not self.runner.args.debug:
1000 return
1001 self.runner.init_bp_file(self.outf_name)
1002 with open(self.outf_name, 'a') as outf:
1003 self.outf = outf
1004 self.write('\n// Line ' + str(self.line_num) + ' "cc" ' + self.line)
1005 self.write('// cc_object for %12s' % self.pkg)
1006 self.write('// src = %s' % short_out_name(self.pkg, self.src))
1007 self.write('// obj = %s' % short_out_name(self.pkg, self.obj))
1008 self.dump_debug_flags('-I flags', self.iflags)
1009 self.dump_debug_flags('-D flags', self.dflags)
1010 self.dump_debug_flags('-f flags', self.fflags)
1011 self.dump_debug_flags('-W flags', self.wflags)
1012 if self.other_args:
1013 self.dump_debug_flags('other args', self.other_args)
1014
1015
1016class Runner(object):
1017 """Main class to parse cargo -v output and print Android module definitions."""
1018
1019 def __init__(self, args):
1020 self.bp_files = set() # Remember all output Android.bp files.
1021 self.root_pkg = '' # name of package in ./Cargo.toml
1022 # Saved flags, modes, and data.
1023 self.args = args
1024 self.dry_run = not args.run
1025 self.skip_cargo = args.skipcargo
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -07001026 self.cargo_path = './cargo' # path to cargo, will be set later
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001027 self.checked_out_files = False # to check only once
1028 self.build_out_files = [] # output files generated by build.rs
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001029 # All cc/ar objects, crates, dependencies, and warning files
1030 self.cc_objects = list()
1031 self.pkg_obj2cc = {}
1032 # pkg_obj2cc[cc_object[i].pkg][cc_objects[i].obj] = cc_objects[i]
1033 self.ar_objects = list()
1034 self.crates = list()
1035 self.dependencies = list() # dependent and build script crates
1036 self.warning_files = set()
1037 # Keep a unique mapping from (module name) to crate
1038 self.name_owners = {}
Chih-Hung Hsieh185052a2020-05-07 14:48:57 -07001039 # Save and dump all errors from cargo to Android.bp.
1040 self.errors = ''
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -07001041 self.setup_cargo_path()
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001042 # Default action is cargo clean, followed by build or user given actions.
1043 if args.cargo:
1044 self.cargo = ['clean'] + args.cargo
1045 else:
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001046 default_target = '--target x86_64-unknown-linux-gnu'
Chih-Hung Hsiehe1b7bb62020-09-30 13:09:30 -07001047 # Use the same target for both host and default device builds.
1048 # Same target is used as default in host x86_64 Android compilation.
1049 # Note: b/169872957, prebuilt cargo failed to build vsock
1050 # on x86_64-unknown-linux-musl systems.
1051 self.cargo = ['clean', 'build ' + default_target]
1052 if args.tests:
1053 self.cargo.append('build --tests ' + default_target)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001054
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -07001055 def setup_cargo_path(self):
1056 """Find cargo in the --cargo_bin or prebuilt rust bin directory."""
1057 if self.args.cargo_bin:
1058 self.cargo_path = os.path.join(self.args.cargo_bin, 'cargo')
1059 if not os.path.isfile(self.cargo_path):
1060 sys.exit('ERROR: cannot find cargo in ' + self.args.cargo_bin)
1061 print('WARNING: using cargo in ' + self.args.cargo_bin)
1062 return
1063 # We have only tested this on Linux.
1064 if platform.system() != 'Linux':
1065 sys.exit('ERROR: this script has only been tested on Linux with cargo.')
1066 # Assuming that this script is in development/scripts.
1067 my_dir = os.path.dirname(os.path.abspath(__file__))
1068 linux_dir = os.path.join(my_dir, '..', '..',
1069 'prebuilts', 'rust', 'linux-x86')
1070 if not os.path.isdir(linux_dir):
1071 sys.exit('ERROR: cannot find directory ' + linux_dir)
1072 rust_version = self.find_rust_version(my_dir, linux_dir)
1073 cargo_bin = os.path.join(linux_dir, rust_version, 'bin')
1074 self.cargo_path = os.path.join(cargo_bin, 'cargo')
1075 if not os.path.isfile(self.cargo_path):
1076 sys.exit('ERROR: cannot find cargo in ' + cargo_bin
1077 + '; please try --cargo_bin= flag.')
1078 return
1079
1080 def find_rust_version(self, my_dir, linux_dir):
1081 """Use my script directory, find prebuilt rust version."""
1082 # First look up build/soong/rust/config/global.go.
1083 path2global = os.path.join(my_dir, '..', '..',
1084 'build', 'soong', 'rust', 'config', 'global.go')
1085 if os.path.isfile(path2global):
1086 # try to find: RustDefaultVersion = "1.44.0"
1087 version_pat = re.compile(
1088 r'\s*RustDefaultVersion\s*=\s*"([0-9]+\.[0-9]+\.[0-9]+)".*$')
1089 with open(path2global, 'r') as inf:
1090 for line in inf:
1091 result = version_pat.match(line)
1092 if result:
1093 return result.group(1)
1094 print('WARNING: cannot find RustDefaultVersion in ' + path2global)
1095 # Otherwise, find the newest (largest) version number in linux_dir.
1096 rust_version = (0, 0, 0) # the prebuilt version to use
1097 version_pat = re.compile(r'([0-9]+)\.([0-9]+)\.([0-9]+)$')
1098 for dir_name in os.listdir(linux_dir):
1099 result = version_pat.match(dir_name)
1100 if not result:
1101 continue
1102 version = (result.group(1), result.group(2), result.group(3))
1103 if version > rust_version:
1104 rust_version = version
1105 return '.'.join(rust_version)
1106
Chih-Hung Hsieh60140752020-11-03 15:05:58 -08001107 def find_out_files(self):
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001108 # list1 has build.rs output for normal crates
1109 list1 = glob.glob(TARGET_TMP + '/*/*/build/' + self.root_pkg + '-*/out/*')
1110 # list2 has build.rs output for proc-macro crates
1111 list2 = glob.glob(TARGET_TMP + '/*/build/' + self.root_pkg + '-*/out/*')
Chih-Hung Hsieh60140752020-11-03 15:05:58 -08001112 return list1 + list2
1113
1114 def copy_out_files(self):
1115 """Copy build.rs output files to ./out and set up build_out_files."""
1116 if self.checked_out_files:
1117 return
1118 self.checked_out_files = True
1119 cargo_out_files = self.find_out_files()
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001120 out_files = set()
Chih-Hung Hsieh60140752020-11-03 15:05:58 -08001121 if cargo_out_files:
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001122 os.makedirs('out', exist_ok=True)
Chih-Hung Hsieh60140752020-11-03 15:05:58 -08001123 for path in cargo_out_files:
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001124 file_name = path.split('/')[-1]
1125 out_files.add(file_name)
1126 shutil.copy(path, 'out/' + file_name)
1127 self.build_out_files = sorted(out_files)
1128
Chih-Hung Hsieh60140752020-11-03 15:05:58 -08001129 def has_used_out_dir(self):
1130 """Returns true if env!("OUT_DIR") is found."""
1131 return 0 == os.system('grep -rl --exclude build.rs --include \\*.rs' +
1132 ' \'env!("OUT_DIR")\' * > /dev/null')
1133
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001134 def copy_out_module_name(self):
1135 if self.args.copy_out and self.build_out_files:
1136 return 'copy_' + self.root_pkg + '_build_out'
1137 else:
1138 return ''
1139
Haibo Huang0f72c952021-03-19 11:34:15 -07001140 def read_license(self, name):
1141 if not os.path.isfile(name):
1142 return ''
1143 license = ''
1144 with open(name, 'r') as intf:
1145 line = intf.readline()
1146 # Firstly skip ANDROID_BP_HEADER
1147 while line.startswith('//'):
1148 line = intf.readline()
Joel Galensond9d13b82021-04-05 11:27:55 -07001149 # Read all lines until we see a rust_* or genrule rule.
1150 while line != '' and not (line.startswith('rust_') or line.startswith('genrule {')):
Haibo Huang0f72c952021-03-19 11:34:15 -07001151 license += line
1152 line = intf.readline()
1153 return license.strip()
1154
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001155 def dump_copy_out_module(self, outf):
1156 """Output the genrule module to copy out/* to $(genDir)."""
1157 copy_out = self.copy_out_module_name()
1158 if not copy_out:
1159 return
1160 outf.write('\ngenrule {\n')
1161 outf.write(' name: "' + copy_out + '",\n')
1162 outf.write(' srcs: ["out/*"],\n')
1163 outf.write(' cmd: "cp $(in) $(genDir)",\n')
1164 if len(self.build_out_files) > 1:
1165 outf.write(' out: [\n')
1166 for f in self.build_out_files:
1167 outf.write(' "' + f + '",\n')
1168 outf.write(' ],\n')
1169 else:
1170 outf.write(' out: ["' + self.build_out_files[0] + '"],\n')
1171 outf.write('}\n')
1172
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001173 def init_bp_file(self, name):
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001174 # name could be Android.bp or sub_dir_path/Android.bp
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001175 if name not in self.bp_files:
1176 self.bp_files.add(name)
Haibo Huang0f72c952021-03-19 11:34:15 -07001177 license_section = self.read_license(name)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001178 with open(name, 'w') as outf:
Joel Galenson367360c2021-04-29 14:31:43 -07001179 print_args = filter(lambda x: x != "--no-test-mapping", sys.argv[1:])
1180 outf.write(ANDROID_BP_HEADER.format(args=' '.join(print_args)))
Haibo Huang0f72c952021-03-19 11:34:15 -07001181 outf.write('\n')
1182 outf.write(license_section)
1183 outf.write('\n')
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001184 # at most one copy_out module per .bp file
1185 self.dump_copy_out_module(outf)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001186
Chih-Hung Hsiehf7eff152020-07-16 15:36:22 -07001187 def dump_test_mapping_files(self):
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001188 """Dump all TEST_MAPPING files."""
Chih-Hung Hsiehf7eff152020-07-16 15:36:22 -07001189 if self.dry_run:
1190 print('Dry-run skip dump of TEST_MAPPING')
Joel Galensoncf4ebb02021-04-07 08:39:51 -07001191 elif self.args.no_test_mapping:
1192 print('Skipping generation of TEST_MAPPING')
Chih-Hung Hsiehf7eff152020-07-16 15:36:22 -07001193 else:
Jeff Vander Stoep1b24dc32021-02-03 18:52:42 +01001194 test_mapping = TestMapping(None)
Jeff Vander Stoepcec8bac2021-01-15 14:15:37 +01001195 for bp_file_name in self.bp_files:
1196 test_mapping.create_test_mapping(os.path.dirname(bp_file_name))
Chih-Hung Hsiehf7eff152020-07-16 15:36:22 -07001197 return self
1198
Chih-Hung Hsiehe02dce12020-07-14 16:05:21 -07001199 def try_claim_module_name(self, name, owner):
1200 """Reserve and return True if it has not been reserved yet."""
1201 if name not in self.name_owners or owner == self.name_owners[name]:
1202 self.name_owners[name] = owner
1203 return True
1204 return False
1205
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001206 def claim_module_name(self, prefix, owner, counter):
1207 """Return prefix if not owned yet, otherwise, prefix+str(counter)."""
1208 while True:
1209 name = prefix
1210 if counter > 0:
Chih-Hung Hsiehe02dce12020-07-14 16:05:21 -07001211 name += '_' + str(counter)
1212 if self.try_claim_module_name(name, owner):
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001213 return name
1214 counter += 1
1215
1216 def find_root_pkg(self):
1217 """Read name of [package] in ./Cargo.toml."""
1218 if not os.path.exists('./Cargo.toml'):
1219 return
1220 with open('./Cargo.toml', 'r') as inf:
1221 pkg_section = re.compile(r'^ *\[package\]')
1222 name = re.compile('^ *name *= * "([^"]*)"')
1223 in_pkg = False
1224 for line in inf:
1225 if in_pkg:
1226 if name.match(line):
1227 self.root_pkg = name.match(line).group(1)
1228 break
1229 else:
1230 in_pkg = pkg_section.match(line) is not None
1231
1232 def run_cargo(self):
1233 """Calls cargo -v and save its output to ./cargo.out."""
1234 if self.skip_cargo:
1235 return self
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001236 cargo_toml = './Cargo.toml'
1237 cargo_out = './cargo.out'
Chih-Hung Hsieh610a8942020-10-29 17:21:35 -07001238 # Do not use Cargo.lock, because .bp rules are designed to
1239 # run with "latest" crates avaialable on Android.
1240 cargo_lock = './Cargo.lock'
1241 cargo_lock_saved = './cargo.lock.saved'
1242 had_cargo_lock = os.path.exists(cargo_lock)
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001243 if not os.access(cargo_toml, os.R_OK):
1244 print('ERROR: Cannot find or read', cargo_toml)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001245 return self
Chih-Hung Hsieh610a8942020-10-29 17:21:35 -07001246 if not self.dry_run:
1247 if os.path.exists(cargo_out):
1248 os.remove(cargo_out)
1249 if not self.args.use_cargo_lock and had_cargo_lock: # save it
1250 os.rename(cargo_lock, cargo_lock_saved)
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001251 cmd_tail = ' --target-dir ' + TARGET_TMP + ' >> ' + cargo_out + ' 2>&1'
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -07001252 # set up search PATH for cargo to find the correct rustc
1253 saved_path = os.environ['PATH']
1254 os.environ['PATH'] = os.path.dirname(self.cargo_path) + ':' + saved_path
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001255 # Add [workspace] to Cargo.toml if it is not there.
1256 added_workspace = False
1257 if self.args.add_workspace:
1258 with open(cargo_toml, 'r') as in_file:
1259 cargo_toml_lines = in_file.readlines()
1260 found_workspace = '[workspace]\n' in cargo_toml_lines
1261 if found_workspace:
1262 print('### WARNING: found [workspace] in Cargo.toml')
1263 else:
1264 with open(cargo_toml, 'a') as out_file:
1265 out_file.write('[workspace]\n')
1266 added_workspace = True
1267 if self.args.verbose:
1268 print('### INFO: added [workspace] to Cargo.toml')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001269 for c in self.cargo:
1270 features = ''
Chih-Hung Hsieh6c8d52f2020-03-30 18:28:52 -07001271 if c != 'clean':
1272 if self.args.features is not None:
1273 features = ' --no-default-features'
1274 if self.args.features:
1275 features += ' --features ' + self.args.features
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -07001276 cmd_v_flag = ' -vv ' if self.args.vv else ' -v '
1277 cmd = self.cargo_path + cmd_v_flag
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001278 cmd += c + features + cmd_tail
1279 if self.args.rustflags and c != 'clean':
1280 cmd = 'RUSTFLAGS="' + self.args.rustflags + '" ' + cmd
1281 if self.dry_run:
1282 print('Dry-run skip:', cmd)
1283 else:
1284 if self.args.verbose:
1285 print('Running:', cmd)
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001286 with open(cargo_out, 'a') as out_file:
1287 out_file.write('### Running: ' + cmd + '\n')
Joel Galenson6bf54e32021-05-17 10:54:50 -07001288 ret = os.system(cmd)
1289 if ret != 0:
1290 print('*** There was an error while running cargo. ' +
1291 'See the cargo.out file for details.')
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001292 if added_workspace: # restore original Cargo.toml
1293 with open(cargo_toml, 'w') as out_file:
1294 out_file.writelines(cargo_toml_lines)
1295 if self.args.verbose:
1296 print('### INFO: restored original Cargo.toml')
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -07001297 os.environ['PATH'] = saved_path
Chih-Hung Hsieh610a8942020-10-29 17:21:35 -07001298 if not self.dry_run:
1299 if not had_cargo_lock: # restore to no Cargo.lock state
1300 os.remove(cargo_lock)
1301 elif not self.args.use_cargo_lock: # restore saved Cargo.lock
1302 os.rename(cargo_lock_saved, cargo_lock)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001303 return self
1304
1305 def dump_dependencies(self):
1306 """Append dependencies and their features to Android.bp."""
1307 if not self.dependencies:
1308 return
1309 dependent_list = list()
1310 for c in self.dependencies:
1311 dependent_list.append(c.feature_list())
1312 sorted_dependencies = sorted(set(dependent_list))
1313 self.init_bp_file('Android.bp')
1314 with open('Android.bp', 'a') as outf:
1315 outf.write('\n// dependent_library ["feature_list"]\n')
1316 for s in sorted_dependencies:
1317 outf.write('// ' + s + '\n')
1318
1319 def dump_pkg_obj2cc(self):
1320 """Dump debug info of the pkg_obj2cc map."""
1321 if not self.args.debug:
1322 return
1323 self.init_bp_file('Android.bp')
1324 with open('Android.bp', 'a') as outf:
1325 sorted_pkgs = sorted(self.pkg_obj2cc.keys())
1326 for pkg in sorted_pkgs:
1327 if not self.pkg_obj2cc[pkg]:
1328 continue
1329 outf.write('\n// obj => src for %s\n' % pkg)
1330 obj2cc = self.pkg_obj2cc[pkg]
1331 for obj in sorted(obj2cc.keys()):
1332 outf.write('// ' + short_out_name(pkg, obj) + ' => ' +
1333 short_out_name(pkg, obj2cc[obj].src) + '\n')
1334
Chih-Hung Hsiehec8846b2020-10-30 17:03:47 -07001335 def apply_patch(self):
1336 """Apply local patch file if it is given."""
1337 if self.args.patch:
1338 if self.dry_run:
1339 print('Dry-run skip patch file:', self.args.patch)
1340 else:
1341 if not os.path.exists(self.args.patch):
1342 self.append_to_bp('ERROR cannot find patch file: ' + self.args.patch)
1343 return self
1344 if self.args.verbose:
1345 print('### INFO: applying local patch file:', self.args.patch)
Joel Galenson7e8247e2021-05-20 18:51:42 -07001346 subprocess.run(['patch', '-s', '--no-backup-if-mismatch', './Android.bp',
1347 self.args.patch], check=True)
Chih-Hung Hsiehec8846b2020-10-30 17:03:47 -07001348 return self
1349
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001350 def gen_bp(self):
1351 """Parse cargo.out and generate Android.bp files."""
1352 if self.dry_run:
1353 print('Dry-run skip: read', CARGO_OUT, 'write Android.bp')
1354 elif os.path.exists(CARGO_OUT):
1355 self.find_root_pkg()
Chih-Hung Hsieh60140752020-11-03 15:05:58 -08001356 if self.args.copy_out:
1357 self.copy_out_files()
1358 elif self.find_out_files() and self.has_used_out_dir():
1359 print('WARNING: ' + self.root_pkg + ' has cargo output files; ' +
1360 'please rerun with the --copy-out flag.')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001361 with open(CARGO_OUT, 'r') as cargo_out:
1362 self.parse(cargo_out, 'Android.bp')
1363 self.crates.sort(key=get_module_name)
1364 for obj in self.cc_objects:
1365 obj.dump()
1366 self.dump_pkg_obj2cc()
1367 for crate in self.crates:
1368 crate.dump()
1369 dumped_libs = set()
1370 for lib in self.ar_objects:
1371 if lib.pkg == self.root_pkg:
1372 lib_name = file_base_name(lib.lib)
1373 if lib_name not in dumped_libs:
1374 dumped_libs.add(lib_name)
1375 lib.dump()
1376 if self.args.dependencies and self.dependencies:
1377 self.dump_dependencies()
Chih-Hung Hsieh185052a2020-05-07 14:48:57 -07001378 if self.errors:
Joel Galenson3f42f802021-04-07 12:42:17 -07001379 self.append_to_bp('\n' + ERRORS_LINE + '\n' + self.errors)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001380 return self
1381
1382 def add_ar_object(self, obj):
1383 self.ar_objects.append(obj)
1384
1385 def add_cc_object(self, obj):
1386 self.cc_objects.append(obj)
1387
1388 def add_crate(self, crate):
1389 """Merge crate with someone in crates, or append to it. Return crates."""
1390 if crate.skip_crate():
1391 if self.args.debug: # include debug info of all crates
1392 self.crates.append(crate)
1393 if self.args.dependencies: # include only dependent crates
1394 if (is_dependent_file_path(crate.main_src) and
1395 not is_build_crate_name(crate.crate_name)):
1396 self.dependencies.append(crate)
1397 else:
1398 for c in self.crates:
1399 if c.merge(crate, 'Android.bp'):
1400 return
Chih-Hung Hsieh185052a2020-05-07 14:48:57 -07001401 # If not merged, decide module type and name now.
1402 crate.decide_module_type()
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001403 self.crates.append(crate)
1404
1405 def find_warning_owners(self):
1406 """For each warning file, find its owner crate."""
1407 missing_owner = False
1408 for f in self.warning_files:
1409 cargo_dir = '' # find lowest crate, with longest path
1410 owner = None # owner crate of this warning
1411 for c in self.crates:
1412 if (f.startswith(c.cargo_dir + '/') and
1413 len(cargo_dir) < len(c.cargo_dir)):
1414 cargo_dir = c.cargo_dir
1415 owner = c
1416 if owner:
1417 owner.has_warning = True
1418 else:
1419 missing_owner = True
1420 if missing_owner and os.path.exists('Cargo.toml'):
1421 # owner is the root cargo, with empty cargo_dir
1422 for c in self.crates:
1423 if not c.cargo_dir:
1424 c.has_warning = True
1425
1426 def rustc_command(self, n, rustc_line, line, outf_name):
1427 """Process a rustc command line from cargo -vv output."""
1428 # cargo build -vv output can have multiple lines for a rustc command
1429 # due to '\n' in strings for environment variables.
1430 # strip removes leading spaces and '\n' at the end
1431 new_rustc = (rustc_line.strip() + line) if rustc_line else line
1432 # Use an heuristic to detect the completions of a multi-line command.
1433 # This might fail for some very rare case, but easy to fix manually.
1434 if not line.endswith('`\n') or (new_rustc.count('`') % 2) != 0:
1435 return new_rustc
1436 if RUSTC_VV_CMD_ARGS.match(new_rustc):
1437 args = RUSTC_VV_CMD_ARGS.match(new_rustc).group(1)
1438 self.add_crate(Crate(self, outf_name).parse(n, args))
1439 else:
1440 self.assert_empty_vv_line(new_rustc)
1441 return ''
1442
1443 def cc_ar_command(self, n, groups, outf_name):
1444 pkg = groups.group(1)
1445 line = groups.group(3)
1446 if groups.group(2) == 'cc':
1447 self.add_cc_object(CCObject(self, outf_name).parse(pkg, n, line))
1448 else:
1449 self.add_ar_object(ARObject(self, outf_name).parse(pkg, n, line))
1450
Chih-Hung Hsieh185052a2020-05-07 14:48:57 -07001451 def append_to_bp(self, line):
1452 self.init_bp_file('Android.bp')
1453 with open('Android.bp', 'a') as outf:
1454 outf.write(line)
1455
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001456 def assert_empty_vv_line(self, line):
1457 if line: # report error if line is not empty
Chih-Hung Hsieh185052a2020-05-07 14:48:57 -07001458 self.append_to_bp('ERROR -vv line: ' + line)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001459 return ''
1460
1461 def parse(self, inf, outf_name):
1462 """Parse rustc and warning messages in inf, return a list of Crates."""
1463 n = 0 # line number
1464 prev_warning = False # true if the previous line was warning: ...
1465 rustc_line = '' # previous line(s) matching RUSTC_VV_PAT
1466 for line in inf:
1467 n += 1
1468 if line.startswith('warning: '):
1469 prev_warning = True
1470 rustc_line = self.assert_empty_vv_line(rustc_line)
1471 continue
1472 new_rustc = ''
1473 if RUSTC_PAT.match(line):
1474 args_line = RUSTC_PAT.match(line).group(1)
1475 self.add_crate(Crate(self, outf_name).parse(n, args_line))
1476 self.assert_empty_vv_line(rustc_line)
1477 elif rustc_line or RUSTC_VV_PAT.match(line):
1478 new_rustc = self.rustc_command(n, rustc_line, line, outf_name)
1479 elif CC_AR_VV_PAT.match(line):
1480 self.cc_ar_command(n, CC_AR_VV_PAT.match(line), outf_name)
1481 elif prev_warning and WARNING_FILE_PAT.match(line):
1482 self.assert_empty_vv_line(rustc_line)
1483 fpath = WARNING_FILE_PAT.match(line).group(1)
1484 if fpath[0] != '/': # ignore absolute path
1485 self.warning_files.add(fpath)
Chih-Hung Hsieh185052a2020-05-07 14:48:57 -07001486 elif line.startswith('error: ') or line.startswith('error[E'):
Chih-Hung Hsiehec8846b2020-10-30 17:03:47 -07001487 if not self.args.ignore_cargo_errors:
1488 self.errors += line
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001489 prev_warning = False
1490 rustc_line = new_rustc
1491 self.find_warning_owners()
1492
1493
Joel Galenson0fbdafe2021-04-21 16:33:33 -07001494def get_parser():
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001495 """Parse main arguments."""
1496 parser = argparse.ArgumentParser('cargo2android')
1497 parser.add_argument(
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001498 '--add_workspace',
1499 action='store_true',
1500 default=False,
1501 help=('append [workspace] to Cargo.toml before calling cargo,' +
1502 ' to treat current directory as root of package source;' +
1503 ' otherwise the relative source file path in generated' +
1504 ' .bp file will be from the parent directory.'))
1505 parser.add_argument(
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001506 '--cargo',
1507 action='append',
1508 metavar='args_string',
1509 help=('extra cargo build -v args in a string, ' +
1510 'each --cargo flag calls cargo build -v once'))
1511 parser.add_argument(
Chih-Hung Hsieh776f6a12020-07-22 14:16:54 -07001512 '--cargo_bin',
1513 type=str,
1514 help='use cargo in the cargo_bin directory instead of the prebuilt one')
1515 parser.add_argument(
Chih-Hung Hsiehe2342ba2020-10-25 03:51:24 -07001516 '--copy-out',
1517 action='store_true',
1518 default=False,
1519 help=('only for root directory, ' +
1520 'copy build.rs output to ./out/* and add a genrule to copy ' +
1521 './out/* to genrule output; for crates with code pattern: ' +
1522 'include!(concat!(env!("OUT_DIR"), "/<some_file>.rs"))'))
1523 parser.add_argument(
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001524 '--debug',
1525 action='store_true',
1526 default=False,
1527 help='dump debug info into Android.bp')
1528 parser.add_argument(
1529 '--dependencies',
1530 action='store_true',
1531 default=False,
1532 help='dump debug info of dependent crates')
1533 parser.add_argument(
1534 '--device',
1535 action='store_true',
1536 default=False,
1537 help='run cargo also for a default device target')
1538 parser.add_argument(
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001539 '--features',
1540 type=str,
1541 help=('pass features to cargo build, ' +
1542 'empty string means no default features'))
1543 parser.add_argument(
1544 '--global_defaults',
1545 type=str,
1546 help='add a defaults name to every module')
Chih-Hung Hsieh3725e082020-07-12 00:51:20 -07001547 parser.add_argument(
1548 '--host-first-multilib',
1549 action='store_true',
1550 default=False,
1551 help=('add a compile_multilib:"first" property ' +
1552 'to Android.bp host modules.'))
1553 parser.add_argument(
Chih-Hung Hsiehec8846b2020-10-30 17:03:47 -07001554 '--ignore-cargo-errors',
1555 action='store_true',
1556 default=False,
1557 help='do not append cargo/rustc error messages to Android.bp')
1558 parser.add_argument(
Chih-Hung Hsieh07119862020-07-24 15:34:06 -07001559 '--no-host',
1560 action='store_true',
1561 default=False,
1562 help='do not run cargo for the host; only for the device target')
1563 parser.add_argument(
1564 '--no-subdir',
1565 action='store_true',
1566 default=False,
1567 help='do not output anything for sub-directories')
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001568 parser.add_argument(
1569 '--onefile',
1570 action='store_true',
1571 default=False,
1572 help=('output all into one ./Android.bp, default will generate ' +
1573 'one Android.bp per Cargo.toml in subdirectories'))
1574 parser.add_argument(
Chih-Hung Hsiehec8846b2020-10-30 17:03:47 -07001575 '--patch',
1576 type=str,
1577 help='apply the given patch file to generated ./Android.bp')
1578 parser.add_argument(
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001579 '--run',
1580 action='store_true',
1581 default=False,
1582 help='run it, default is dry-run')
1583 parser.add_argument('--rustflags', type=str, help='passing flags to rustc')
1584 parser.add_argument(
1585 '--skipcargo',
1586 action='store_true',
1587 default=False,
1588 help='skip cargo command, parse cargo.out, and generate Android.bp')
1589 parser.add_argument(
1590 '--tests',
1591 action='store_true',
1592 default=False,
1593 help='run cargo build --tests after normal build')
1594 parser.add_argument(
Chih-Hung Hsieh610a8942020-10-29 17:21:35 -07001595 '--use-cargo-lock',
1596 action='store_true',
1597 default=False,
1598 help=('run cargo build with existing Cargo.lock ' +
1599 '(used when some latest dependent crates failed)'))
1600 parser.add_argument(
Joel Galensond9c4de62021-04-23 10:26:40 -07001601 '--min-sdk-version',
1602 type=str,
1603 help='Minimum SDK version')
1604 parser.add_argument(
1605 '--apex-available',
1606 nargs='*',
1607 help='Mark the main library as apex_available with the given apexes.')
1608 parser.add_argument(
Joel Galenson97e414a2021-05-27 09:42:32 -07001609 '--dependency-blocklist',
1610 nargs='*',
1611 default=[],
1612 help='Do not emit the given dependencies.')
1613 parser.add_argument(
Joel Galensonf6b3c912021-06-03 16:00:54 -07001614 '--test-blocklist',
1615 nargs='*',
1616 default=[],
1617 help=('Do not emit the given tests. ' +
1618 'Pass the path to the test file to exclude.'))
1619 parser.add_argument(
Joel Galensoncf4ebb02021-04-07 08:39:51 -07001620 '--no-test-mapping',
1621 action='store_true',
1622 default=False,
1623 help='Do not generate a TEST_MAPPING file. Use only to speed up debugging.')
1624 parser.add_argument(
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001625 '--verbose',
1626 action='store_true',
1627 default=False,
1628 help='echo executed commands')
1629 parser.add_argument(
1630 '--vv',
1631 action='store_true',
1632 default=False,
1633 help='run cargo with -vv instead of default -v')
Joel Galenson0fbdafe2021-04-21 16:33:33 -07001634 parser.add_argument(
1635 '--dump-config-and-exit',
1636 type=str,
1637 help=('Dump command-line arguments (minus this flag) to a config file and exit. ' +
1638 'This is intended to help migrate from command line options to config files.'))
1639 parser.add_argument(
1640 '--config',
1641 type=str,
1642 help=('Load command-line options from the given config file. ' +
1643 'Options in this file will override those passed on the command line.'))
1644 return parser
1645
1646
1647def parse_args(parser):
1648 """Parses command-line options."""
1649 args = parser.parse_args()
1650 # Use the values specified in a config file if one was found.
1651 if args.config:
1652 with open(args.config, 'r') as f:
1653 config = json.load(f)
1654 args_dict = vars(args)
1655 for arg in config:
1656 args_dict[arg.replace('-', '_')] = config[arg]
1657 return args
1658
1659
1660def dump_config(parser, args):
1661 """Writes the non-default command-line options to the specified file."""
1662 args_dict = vars(args)
1663 # Filter out the arguments that have their default value.
Joel Galenson367360c2021-04-29 14:31:43 -07001664 # Also filter certain "temporary" arguments.
Joel Galenson0fbdafe2021-04-21 16:33:33 -07001665 non_default_args = {}
1666 for arg in args_dict:
Joel Galenson367360c2021-04-29 14:31:43 -07001667 if args_dict[arg] != parser.get_default(
1668 arg) and arg != 'dump_config_and_exit' and arg != 'no_test_mapping':
Joel Galenson0fbdafe2021-04-21 16:33:33 -07001669 non_default_args[arg.replace('_', '-')] = args_dict[arg]
1670 # Write to the specified file.
1671 with open(args.dump_config_and_exit, 'w') as f:
1672 json.dump(non_default_args, f, indent=2, sort_keys=True)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001673
1674
1675def main():
Joel Galenson0fbdafe2021-04-21 16:33:33 -07001676 parser = get_parser()
1677 args = parse_args(parser)
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001678 if not args.run: # default is dry-run
1679 print(DRY_RUN_NOTE)
Joel Galenson0fbdafe2021-04-21 16:33:33 -07001680 if args.dump_config_and_exit:
1681 dump_config(parser, args)
1682 else:
1683 Runner(args).run_cargo().gen_bp().apply_patch().dump_test_mapping_files()
Chih-Hung Hsiehe8887372019-11-05 10:34:17 -08001684
1685
1686if __name__ == '__main__':
1687 main()