|
| 1 | +//! LLVM utilities |
| 2 | +
|
| 3 | +use anyhow::{bail, ensure, Context, Result}; |
| 4 | +use std::collections::HashSet; |
| 5 | +use std::path::{Path, PathBuf}; |
| 6 | +use std::process::Command; |
| 7 | + |
| 8 | +/// Returns the set of additional libraries that need to be bundled with |
| 9 | +/// the given library, scanned recursively. |
| 10 | +/// |
| 11 | +/// Any libraries in `provided_libs_paths` will be treated as available, without |
| 12 | +/// being emitted. Any other library not in `search_paths` or `provided_libs_paths` |
| 13 | +/// will result in an error. |
| 14 | +pub fn list_needed_libs_recursively( |
| 15 | + lib: &Path, |
| 16 | + search_paths: &[&Path], |
| 17 | + provided_libs_paths: &[&Path], |
| 18 | +) -> Result<HashSet<PathBuf>> { |
| 19 | + // Create a view of all libraries that are available on Android |
| 20 | + let mut provided = HashSet::new(); |
| 21 | + for path in provided_libs_paths { |
| 22 | + for lib in find_libs_in_dir(path).with_context(|| { |
| 23 | + format!("Unable to list available libraries in `{}`", path.display()) |
| 24 | + })? { |
| 25 | + // libc++_shared is bundled with the NDK but not available on-device |
| 26 | + if lib != "libc++_shared.so" { |
| 27 | + provided.insert(lib); |
| 28 | + } |
| 29 | + } |
| 30 | + } |
| 31 | + |
| 32 | + let mut to_copy = HashSet::new(); |
| 33 | + |
| 34 | + let mut artifacts = vec![lib.to_path_buf()]; |
| 35 | + while let Some(artifact) = artifacts.pop() { |
| 36 | + for need in list_needed_libs(&artifact).with_context(|| { |
| 37 | + format!( |
| 38 | + "Unable to read needed libraries from `{}`", |
| 39 | + artifact.display() |
| 40 | + ) |
| 41 | + })? { |
| 42 | + // c++_shared is available in the NDK but not on-device. |
| 43 | + // Must be bundled with the apk if used: |
| 44 | + // https://developer.android.com/ndk/guides/cpp-support#libc |
| 45 | + let search_paths = if need == "libc++_shared.so" { |
| 46 | + provided_libs_paths |
| 47 | + } else { |
| 48 | + search_paths |
| 49 | + }; |
| 50 | + |
| 51 | + if provided.insert(need.clone()) { |
| 52 | + if let Some(path) = find_library_path(search_paths, &need).with_context(|| { |
| 53 | + format!( |
| 54 | + "Could not iterate one or more search directories in `{:?}` while searching for library `{}`", |
| 55 | + search_paths, need |
| 56 | + ) |
| 57 | + })? { |
| 58 | + to_copy.insert(path.clone()); |
| 59 | + artifacts.push(path); |
| 60 | + } else { |
| 61 | + bail!("Shared library `{}` not found", need); |
| 62 | + } |
| 63 | + } |
| 64 | + } |
| 65 | + } |
| 66 | + |
| 67 | + Ok(to_copy) |
| 68 | +} |
| 69 | + |
| 70 | +/// List all required shared libraries as per the dynamic section |
| 71 | +fn list_needed_libs(library_path: &Path) -> Result<HashSet<String>> { |
| 72 | + let mut readelf = Command::new("llvm-readelf"); |
| 73 | + let output = readelf.arg("--needed-libs").arg(library_path).output()?; |
| 74 | + ensure!( |
| 75 | + output.status.success(), |
| 76 | + "Failed to run `{:?}`: {}", |
| 77 | + readelf, |
| 78 | + output.status |
| 79 | + ); |
| 80 | + let output = std::str::from_utf8(&output.stdout).unwrap(); |
| 81 | + let output = output.strip_prefix("NeededLibraries [\n").unwrap(); |
| 82 | + let output = output.strip_suffix("]\n").unwrap(); |
| 83 | + let mut needed = HashSet::new(); |
| 84 | + |
| 85 | + for line in output.lines() { |
| 86 | + let lib = line.trim_start(); |
| 87 | + needed.insert(lib.to_string()); |
| 88 | + } |
| 89 | + Ok(needed) |
| 90 | +} |
| 91 | + |
| 92 | +/// List names of shared libraries inside directory |
| 93 | +fn find_libs_in_dir(path: &Path) -> Result<HashSet<String>> { |
| 94 | + let mut libs = HashSet::new(); |
| 95 | + let entries = std::fs::read_dir(path)?; |
| 96 | + for entry in entries { |
| 97 | + let entry = entry?; |
| 98 | + if !entry.path().is_dir() { |
| 99 | + if let Some(file_name) = entry.file_name().to_str() { |
| 100 | + if file_name.ends_with(".so") { |
| 101 | + libs.insert(file_name.to_string()); |
| 102 | + } |
| 103 | + } |
| 104 | + } |
| 105 | + } |
| 106 | + Ok(libs) |
| 107 | +} |
| 108 | + |
| 109 | +/// Resolves native library using search paths |
| 110 | +fn find_library_path(paths: &[&Path], library: &str) -> Result<Option<PathBuf>> { |
| 111 | + for path in paths { |
| 112 | + let lib_path = path.join(library); |
| 113 | + if lib_path.exists() { |
| 114 | + return Ok(Some(dunce::canonicalize(lib_path)?)); |
| 115 | + } |
| 116 | + } |
| 117 | + Ok(None) |
| 118 | +} |
0 commit comments