1use std::env;
2use std::ffi::OsStr;
3#[cfg(target_family = "unix")]
4use std::fs::OpenOptions;
5use std::io::Write;
6#[cfg(target_family = "unix")]
7use std::os::unix::fs::OpenOptionsExt;
8use std::path::{Path, PathBuf};
9use std::process::{self, Command};
10use std::str;
11
12use anyhow::{anyhow, bail, Context, Result};
13use fs_err as fs;
14use path_slash::PathBufExt;
15use serde::Deserialize;
16use target_lexicon::{Architecture, Environment, OperatingSystem, Triple};
17
18use crate::linux::ARM_FEATURES_H;
19use crate::macos::{LIBCHARSET_TBD, LIBICONV_TBD};
20
21#[derive(Clone, Debug, clap::Subcommand)]
23pub enum Zig {
24 #[command(name = "cc")]
26 Cc {
27 #[arg(num_args = 1.., trailing_var_arg = true)]
29 args: Vec<String>,
30 },
31 #[command(name = "c++")]
33 Cxx {
34 #[arg(num_args = 1.., trailing_var_arg = true)]
36 args: Vec<String>,
37 },
38 #[command(name = "ar")]
40 Ar {
41 #[arg(num_args = 1.., trailing_var_arg = true)]
43 args: Vec<String>,
44 },
45 #[command(name = "ranlib")]
47 Ranlib {
48 #[arg(num_args = 1.., trailing_var_arg = true)]
50 args: Vec<String>,
51 },
52 #[command(name = "lib")]
54 Lib {
55 #[arg(num_args = 1.., trailing_var_arg = true)]
57 args: Vec<String>,
58 },
59}
60
61struct TargetInfo {
62 target: Option<String>,
63 is_musl: bool,
64 is_windows_gnu: bool,
65 is_windows_msvc: bool,
66 is_arm: bool,
67 is_i386: bool,
68 is_riscv64: bool,
69 is_mips32: bool,
70 is_macos: bool,
71 is_ohos: bool,
72}
73
74impl TargetInfo {
75 fn new(target: Option<&String>) -> Self {
76 Self {
77 target: target.cloned(),
78 is_musl: target.map(|x| x.contains("musl")).unwrap_or_default(),
79 is_windows_gnu: target
80 .map(|x| x.contains("windows-gnu"))
81 .unwrap_or_default(),
82 is_windows_msvc: target
83 .map(|x| x.contains("windows-msvc"))
84 .unwrap_or_default(),
85 is_arm: target.map(|x| x.starts_with("arm")).unwrap_or_default(),
86 is_i386: target.map(|x| x.starts_with("i386")).unwrap_or_default(),
87 is_riscv64: target.map(|x| x.starts_with("riscv64")).unwrap_or_default(),
88 is_mips32: target
89 .map(|x| x.starts_with("mips") && !x.starts_with("mips64"))
90 .unwrap_or_default(),
91 is_macos: target.map(|x| x.contains("macos")).unwrap_or_default(),
92 is_ohos: target.map(|x| x.contains("ohos")).unwrap_or_default(),
93 }
94 }
95}
96
97impl Zig {
98 pub fn execute(&self) -> Result<()> {
100 match self {
101 Zig::Cc { args } => self.execute_compiler("cc", args),
102 Zig::Cxx { args } => self.execute_compiler("c++", args),
103 Zig::Ar { args } => self.execute_tool("ar", args),
104 Zig::Ranlib { args } => self.execute_compiler("ranlib", args),
105 Zig::Lib { args } => self.execute_compiler("lib", args),
106 }
107 }
108
109 pub fn execute_compiler(&self, cmd: &str, cmd_args: &[String]) -> Result<()> {
111 let target = cmd_args
112 .iter()
113 .position(|x| x == "-target")
114 .and_then(|index| cmd_args.get(index + 1));
115 let target_info = TargetInfo::new(target);
116
117 let rustc_ver = match env::var("CARGO_ZIGBUILD_RUSTC_VERSION") {
118 Ok(version) => version.parse()?,
119 Err(_) => rustc_version::version()?,
120 };
121 let zig_version = Zig::zig_version()?;
122
123 let mut new_cmd_args = Vec::with_capacity(cmd_args.len());
124 let mut skip_next_arg = false;
125 for arg in cmd_args {
126 if skip_next_arg {
127 skip_next_arg = false;
128 continue;
129 }
130 let args = if arg.starts_with('@') && arg.ends_with("linker-arguments") {
131 vec![self.process_linker_response_file(
132 arg,
133 &rustc_ver,
134 &zig_version,
135 &target_info,
136 )?]
137 } else {
138 self.filter_linker_arg(arg, &rustc_ver, &zig_version, &target_info)
139 };
140 for arg in args {
141 if arg == "-Wl,-exported_symbols_list" {
142 skip_next_arg = true;
144 } else {
145 new_cmd_args.push(arg);
146 }
147 }
148 }
149
150 if target_info.is_mips32 {
151 new_cmd_args.push("-Wl,-z,notext".to_string());
153 }
154
155 if self.has_undefined_dynamic_lookup(cmd_args) {
156 new_cmd_args.push("-Wl,-undefined=dynamic_lookup".to_string());
157 }
158 if target_info.is_macos {
159 if self.should_add_libcharset(cmd_args, &zig_version) {
160 new_cmd_args.push("-lcharset".to_string());
161 }
162 self.add_macos_specific_args(&mut new_cmd_args, &zig_version)?;
163 }
164
165 let mut child = Self::command()?
166 .arg(cmd)
167 .args(new_cmd_args)
168 .spawn()
169 .with_context(|| format!("Failed to run `zig {cmd}`"))?;
170 let status = child.wait().expect("Failed to wait on zig child process");
171 if !status.success() {
172 process::exit(status.code().unwrap_or(1));
173 }
174 Ok(())
175 }
176
177 fn process_linker_response_file(
178 &self,
179 arg: &str,
180 rustc_ver: &rustc_version::Version,
181 zig_version: &semver::Version,
182 target_info: &TargetInfo,
183 ) -> Result<String> {
184 let content_bytes = fs::read(arg.trim_start_matches('@'))?;
188 let content = if target_info.is_windows_msvc {
189 if content_bytes[0..2] != [255, 254] {
190 bail!(
191 "linker response file `{}` didn't start with a utf16 BOM",
192 &arg
193 );
194 }
195 let content_utf16: Vec<u16> = content_bytes[2..]
196 .chunks_exact(2)
197 .map(|a| u16::from_ne_bytes([a[0], a[1]]))
198 .collect();
199 String::from_utf16(&content_utf16).with_context(|| {
200 format!(
201 "linker response file `{}` didn't contain valid utf16 content",
202 &arg
203 )
204 })?
205 } else {
206 String::from_utf8(content_bytes).with_context(|| {
207 format!(
208 "linker response file `{}` didn't contain valid utf8 content",
209 &arg
210 )
211 })?
212 };
213 let mut link_args: Vec<_> = content
214 .split('\n')
215 .flat_map(|arg| self.filter_linker_arg(arg, &rustc_ver, &zig_version, &target_info))
216 .collect();
217 if self.has_undefined_dynamic_lookup(&link_args) {
218 link_args.push("-Wl,-undefined=dynamic_lookup".to_string());
219 }
220 if target_info.is_macos && self.should_add_libcharset(&link_args, &zig_version) {
221 link_args.push("-lcharset".to_string());
222 }
223 if target_info.is_windows_msvc {
224 let new_content = link_args.join("\n");
225 let mut out = Vec::with_capacity((1 + new_content.len()) * 2);
226 for c in std::iter::once(0xFEFF).chain(new_content.encode_utf16()) {
228 out.push(c as u8);
230 out.push((c >> 8) as u8);
231 }
232 fs::write(arg.trim_start_matches('@'), out)?;
233 } else {
234 fs::write(arg.trim_start_matches('@'), link_args.join("\n").as_bytes())?;
235 }
236 Ok(arg.to_string())
237 }
238
239 fn filter_linker_arg(
240 &self,
241 arg: &str,
242 rustc_ver: &rustc_version::Version,
243 zig_version: &semver::Version,
244 target_info: &TargetInfo,
245 ) -> Vec<String> {
246 if arg == "-lgcc_s" {
247 return vec!["-lunwind".to_string()];
249 } else if arg.starts_with("--target=") {
250 return vec![];
252 }
253 if (target_info.is_arm || target_info.is_windows_gnu)
254 && arg.ends_with(".rlib")
255 && arg.contains("libcompiler_builtins-")
256 {
257 return vec![];
259 }
260 if target_info.is_windows_gnu {
261 #[allow(clippy::if_same_then_else)]
262 if arg == "-lgcc_eh" {
263 return vec!["-lc++".to_string()];
266 } else if arg == "-Wl,-Bdynamic" && (zig_version.major, zig_version.minor) >= (0, 11) {
267 return vec!["-Wl,-search_paths_first".to_owned()];
272 } else if arg == "-lwindows" || arg == "-l:libpthread.a" || arg == "-lgcc" {
273 return vec![];
274 } else if arg == "-Wl,--disable-auto-image-base"
275 || arg == "-Wl,--dynamicbase"
276 || arg == "-Wl,--large-address-aware"
277 || (arg.starts_with("-Wl,")
278 && (arg.ends_with("/list.def") || arg.ends_with("\\list.def")))
279 {
280 return vec![];
285 } else if arg == "-lmsvcrt" {
286 return vec![];
287 }
288 } else if arg == "-Wl,--no-undefined-version" {
289 return vec![];
292 }
293 if target_info.is_musl || target_info.is_ohos {
294 if arg.ends_with(".o") && arg.contains("self-contained") && arg.contains("crt") {
296 return vec![];
297 } else if arg == "-Wl,-melf_i386" {
298 return vec![];
300 }
301 if rustc_ver.major == 1
302 && rustc_ver.minor < 59
303 && arg.ends_with(".rlib")
304 && arg.contains("liblibc-")
305 {
306 return vec![];
309 }
310 if arg == "-lc" {
311 return vec![];
312 }
313 }
314 if arg.starts_with("-march=") {
315 if target_info.is_arm || target_info.is_i386 {
317 return vec![];
318 } else if target_info.is_riscv64 {
319 return vec!["-march=generic_rv64".to_string()];
320 } else if arg.starts_with("-march=armv8-a") {
321 let mut args_march = if target_info
322 .target
323 .as_ref()
324 .map(|x| x.starts_with("aarch64-macos"))
325 .unwrap_or_default()
326 {
327 vec![arg.replace("armv8-a", "apple_m1")]
328 } else if target_info
329 .target
330 .as_ref()
331 .map(|x| x.starts_with("aarch64-linux"))
332 .unwrap_or_default()
333 {
334 vec![arg
335 .replace("armv8-a", "generic+v8a")
336 .replace("simd", "neon")]
337 } else {
338 vec![arg.to_string()]
339 };
340 if arg == "-march=armv8-a+crypto" {
341 args_march.append(&mut vec![
347 "-Xassembler".to_owned(),
348 "-march=armv8-a+crypto".to_owned(),
349 ]);
350 }
351 return args_march;
352 }
353 }
354 if target_info.is_macos {
355 if arg.starts_with("-Wl,-exported_symbols_list,") {
356 return vec![];
359 }
360 if arg == "-Wl,-dylib" {
361 return vec![];
363 }
364 }
365 vec![arg.to_string()]
366 }
367
368 fn has_undefined_dynamic_lookup(&self, args: &[String]) -> bool {
369 let undefined = args
370 .iter()
371 .position(|x| x == "-undefined")
372 .and_then(|i| args.get(i + 1));
373 matches!(undefined, Some(x) if x == "dynamic_lookup")
374 }
375
376 fn should_add_libcharset(&self, args: &[String], zig_version: &semver::Version) -> bool {
377 if (zig_version.major, zig_version.minor) >= (0, 12) {
379 args.iter().any(|x| x == "-liconv") && !args.iter().any(|x| x == "-lcharset")
380 } else {
381 false
382 }
383 }
384
385 fn add_macos_specific_args(
386 &self,
387 new_cmd_args: &mut Vec<String>,
388 zig_version: &semver::Version,
389 ) -> Result<()> {
390 let sdkroot = Self::macos_sdk_root();
391 if (zig_version.major, zig_version.minor) >= (0, 12) {
392 if let Some(ref sdkroot) = sdkroot {
394 new_cmd_args.push(format!("--sysroot={}", sdkroot.display()));
395 }
396 }
397 if let Some(ref sdkroot) = sdkroot {
398 new_cmd_args.extend_from_slice(&[
399 "-isystem".to_string(),
400 format!("{}", sdkroot.join("usr").join("include").display()),
401 format!("-L{}", sdkroot.join("usr").join("lib").display()),
402 format!(
403 "-F{}",
404 sdkroot
405 .join("System")
406 .join("Library")
407 .join("Frameworks")
408 .display()
409 ),
410 "-DTARGET_OS_IPHONE=0".to_string(),
411 ]);
412 }
413
414 let cache_dir = cache_dir();
416 let deps_dir = cache_dir.join("deps");
417 fs::create_dir_all(&deps_dir)?;
418 write_tbd_files(&deps_dir)?;
419 new_cmd_args.push("-L".to_string());
420 new_cmd_args.push(format!("{}", deps_dir.display()));
421 Ok(())
422 }
423
424 pub fn execute_tool(&self, cmd: &str, cmd_args: &[String]) -> Result<()> {
426 let mut child = Self::command()?
427 .arg(cmd)
428 .args(cmd_args)
429 .spawn()
430 .with_context(|| format!("Failed to run `zig {cmd}`"))?;
431 let status = child.wait().expect("Failed to wait on zig child process");
432 if !status.success() {
433 process::exit(status.code().unwrap_or(1));
434 }
435 Ok(())
436 }
437
438 pub fn command() -> Result<Command> {
440 let (zig, zig_args) = Self::find_zig()?;
441 let mut cmd = Command::new(zig);
442 cmd.args(zig_args);
443 Ok(cmd)
444 }
445
446 fn zig_version() -> Result<semver::Version> {
447 let output = Self::command()?.arg("version").output()?;
448 let version_str =
449 str::from_utf8(&output.stdout).context("`zig version` didn't return utf8 output")?;
450 let version = semver::Version::parse(version_str.trim())?;
451 Ok(version)
452 }
453
454 pub fn find_zig() -> Result<(PathBuf, Vec<String>)> {
456 Self::find_zig_python()
457 .or_else(|_| Self::find_zig_bin())
458 .context("Failed to find zig")
459 }
460
461 fn find_zig_bin() -> Result<(PathBuf, Vec<String>)> {
463 let zig_path = zig_path()?;
464 let output = Command::new(&zig_path).arg("version").output()?;
465
466 let version_str = str::from_utf8(&output.stdout).with_context(|| {
467 format!("`{} version` didn't return utf8 output", zig_path.display())
468 })?;
469 Self::validate_zig_version(version_str)?;
470 Ok((zig_path, Vec::new()))
471 }
472
473 fn find_zig_python() -> Result<(PathBuf, Vec<String>)> {
475 let python_path = python_path()?;
476 let output = Command::new(&python_path)
477 .args(["-m", "ziglang", "version"])
478 .output()?;
479
480 let version_str = str::from_utf8(&output.stdout).with_context(|| {
481 format!(
482 "`{} -m ziglang version` didn't return utf8 output",
483 python_path.display()
484 )
485 })?;
486 Self::validate_zig_version(version_str)?;
487 Ok((python_path, vec!["-m".to_string(), "ziglang".to_string()]))
488 }
489
490 fn validate_zig_version(version: &str) -> Result<()> {
491 let min_ver = semver::Version::new(0, 9, 0);
492 let version = semver::Version::parse(version.trim())?;
493 if version >= min_ver {
494 Ok(())
495 } else {
496 bail!(
497 "zig version {} is too old, need at least {}",
498 version,
499 min_ver
500 )
501 }
502 }
503
504 pub fn lib_dir() -> Result<PathBuf> {
506 let (zig, zig_args) = Self::find_zig()?;
507 let output = Command::new(zig).args(zig_args).arg("env").output()?;
508 let zig_env: ZigEnv = serde_json::from_slice(&output.stdout)?;
509 Ok(PathBuf::from(zig_env.lib_dir))
510 }
511
512 fn add_env_if_missing<K, V>(command: &mut Command, name: K, value: V)
513 where
514 K: AsRef<OsStr>,
515 V: AsRef<OsStr>,
516 {
517 let command_env_contains_no_key =
518 |name: &K| !command.get_envs().any(|(key, _)| name.as_ref() == key);
519
520 if command_env_contains_no_key(&name) && env::var_os(&name).is_none() {
521 command.env(name, value);
522 }
523 }
524
525 pub(crate) fn apply_command_env(
526 manifest_path: Option<&Path>,
527 release: bool,
528 cargo: &cargo_options::CommonOptions,
529 cmd: &mut Command,
530 enable_zig_ar: bool,
531 ) -> Result<()> {
532 let rust_targets = cargo
534 .target
535 .iter()
536 .map(|target| target.split_once('.').map(|(t, _)| t).unwrap_or(target))
537 .collect::<Vec<&str>>();
538 let rustc_meta = rustc_version::version_meta()?;
539 Self::add_env_if_missing(
540 cmd,
541 "CARGO_ZIGBUILD_RUSTC_VERSION",
542 rustc_meta.semver.to_string(),
543 );
544 let host_target = &rustc_meta.host;
545 for (parsed_target, raw_target) in rust_targets.iter().zip(&cargo.target) {
546 let env_target = parsed_target.replace('-', "_");
547 let zig_wrapper = prepare_zig_linker(raw_target)?;
548
549 if is_mingw_shell() {
550 let zig_cc = zig_wrapper.cc.to_slash_lossy();
551 let zig_cxx = zig_wrapper.cxx.to_slash_lossy();
552 Self::add_env_if_missing(cmd, format!("CC_{env_target}"), &*zig_cc);
553 Self::add_env_if_missing(cmd, format!("CXX_{env_target}"), &*zig_cxx);
554 if !parsed_target.contains("wasm") {
555 Self::add_env_if_missing(
556 cmd,
557 format!("CARGO_TARGET_{}_LINKER", env_target.to_uppercase()),
558 &*zig_cc,
559 );
560 }
561 } else {
562 Self::add_env_if_missing(cmd, format!("CC_{env_target}"), &zig_wrapper.cc);
563 Self::add_env_if_missing(cmd, format!("CXX_{env_target}"), &zig_wrapper.cxx);
564 if !parsed_target.contains("wasm") {
565 Self::add_env_if_missing(
566 cmd,
567 format!("CARGO_TARGET_{}_LINKER", env_target.to_uppercase()),
568 &zig_wrapper.cc,
569 );
570 }
571 }
572
573 Self::add_env_if_missing(cmd, format!("RANLIB_{env_target}"), &zig_wrapper.ranlib);
574 if enable_zig_ar {
577 if parsed_target.contains("msvc") {
578 Self::add_env_if_missing(cmd, format!("AR_{env_target}"), &zig_wrapper.lib);
579 } else {
580 Self::add_env_if_missing(cmd, format!("AR_{env_target}"), &zig_wrapper.ar);
581 }
582 }
583
584 Self::setup_os_deps(manifest_path, release, cargo)?;
585
586 let cmake_toolchain_file_env = format!("CMAKE_TOOLCHAIN_FILE_{env_target}");
587 if env::var_os(&cmake_toolchain_file_env).is_none()
588 && env::var_os(format!("CMAKE_TOOLCHAIN_FILE_{parsed_target}")).is_none()
589 && env::var_os("TARGET_CMAKE_TOOLCHAIN_FILE").is_none()
590 && env::var_os("CMAKE_TOOLCHAIN_FILE").is_none()
591 {
592 if let Ok(cmake_toolchain_file) =
593 Self::setup_cmake_toolchain(parsed_target, &zig_wrapper, enable_zig_ar)
594 {
595 cmd.env(cmake_toolchain_file_env, cmake_toolchain_file);
596 }
597 }
598
599 if raw_target.contains("windows-gnu") {
600 cmd.env("WINAPI_NO_BUNDLED_LIBRARIES", "1");
601 }
602
603 if raw_target.contains("apple-darwin") {
604 if let Some(sdkroot) = Self::macos_sdk_root() {
605 if env::var_os("PKG_CONFIG_SYSROOT_DIR").is_none() {
606 cmd.env("PKG_CONFIG_SYSROOT_DIR", sdkroot);
608 }
609 }
610 }
611
612 if host_target == parsed_target {
615 if !matches!(rustc_meta.channel, rustc_version::Channel::Nightly) {
616 cmd.env("__CARGO_TEST_CHANNEL_OVERRIDE_DO_NOT_USE_THIS", "nightly");
619 }
620 cmd.env("CARGO_UNSTABLE_TARGET_APPLIES_TO_HOST", "true");
621 cmd.env("CARGO_TARGET_APPLIES_TO_HOST", "false");
622 }
623
624 if let Ok(mut options) = Self::collect_zig_cc_options(&zig_wrapper, raw_target) {
626 if raw_target.contains("apple-darwin") {
627 options.push("-DTARGET_OS_IPHONE=0".to_string());
629 }
630
631 let escaped_options = shlex::try_join(options.iter().map(|s| &s[..]))?;
632
633 let bindgen_env = "BINDGEN_EXTRA_CLANG_ARGS";
635 let fallback_value = env::var(bindgen_env);
636 for target in [&env_target[..], parsed_target] {
637 let name = format!("{bindgen_env}_{target}");
638 if let Ok(mut value) = env::var(&name).or(fallback_value.clone()) {
639 if shlex::split(&value).is_none() {
640 value = shlex::try_quote(&value)?.into_owned();
642 }
643 if !value.is_empty() {
644 value.push(' ');
645 }
646 value.push_str(&escaped_options);
647 env::set_var(name, value);
648 } else {
649 env::set_var(name, escaped_options.clone());
650 }
651 }
652 }
653 }
654 Ok(())
655 }
656
657 fn collect_zig_cc_options(zig_wrapper: &ZigWrapper, raw_target: &str) -> Result<Vec<String>> {
661 #[derive(Debug, PartialEq, Eq)]
662 enum Kind {
663 Normal,
664 Framework,
665 }
666
667 #[derive(Debug)]
668 struct PerLanguageOptions {
669 glibc_minor_ver: Option<u32>,
670 include_paths: Vec<(Kind, String)>,
671 }
672
673 fn collect_per_language_options(
674 program: &Path,
675 ext: &str,
676 raw_target: &str,
677 ) -> Result<PerLanguageOptions> {
678 let empty_file_path = cache_dir().join(format!(".intentionally-empty-file.{ext}"));
680 if !empty_file_path.exists() {
681 fs::write(&empty_file_path, "")?;
682 }
683
684 let output = Command::new(program)
685 .arg("-E")
686 .arg(&empty_file_path)
687 .arg("-v")
688 .output();
689 fs::remove_file(&empty_file_path).ok();
690 let output = output?;
691 let stderr = String::from_utf8(output.stderr)?;
693 if !output.status.success() {
694 bail!(
695 "Failed to run `zig cc -v` with status {}: {}",
696 output.status,
697 stderr.trim(),
698 );
699 }
700
701 let glibc_minor_ver = if let Some(start) = stderr.find("__GLIBC_MINOR__=") {
705 let stderr = &stderr[start + 16..];
706 let end = stderr
707 .find(|c: char| !c.is_ascii_digit())
708 .unwrap_or(stderr.len());
709 stderr[..end].parse().ok()
710 } else {
711 None
712 };
713
714 let start = stderr
715 .find("#include <...> search starts here:")
716 .ok_or_else(|| anyhow!("Failed to parse `zig cc -v` output"))?
717 + 34;
718 let end = stderr
719 .find("End of search list.")
720 .ok_or_else(|| anyhow!("Failed to parse `zig cc -v` output"))?;
721
722 let mut include_paths = Vec::new();
723 for mut line in stderr[start..end].lines() {
724 line = line.trim();
725 let mut kind = Kind::Normal;
726 if line.ends_with(" (framework directory)") {
727 line = line[..line.len() - 22].trim();
728 kind = Kind::Framework;
729 } else if line.ends_with(" (headermap)") {
730 bail!("C/C++ search path includes header maps, which are not supported");
731 }
732 if !line.is_empty() {
733 include_paths.push((kind, line.to_owned()));
734 }
735 }
736
737 if raw_target.contains("ohos") {
739 let ndk = env::var("OHOS_NDK_HOME").expect("Can't get NDK path");
740 include_paths.push((Kind::Normal, format!("{}/native/sysroot/usr/include", ndk)));
741 }
742
743 Ok(PerLanguageOptions {
744 include_paths,
745 glibc_minor_ver,
746 })
747 }
748
749 let c_opts = collect_per_language_options(&zig_wrapper.cc, "c", raw_target)?;
750 let cpp_opts = collect_per_language_options(&zig_wrapper.cxx, "cpp", raw_target)?;
751
752 if c_opts.glibc_minor_ver != cpp_opts.glibc_minor_ver {
754 bail!(
755 "`zig cc` gives a different glibc minor version for C ({:?}) and C++ ({:?})",
756 c_opts.glibc_minor_ver,
757 cpp_opts.glibc_minor_ver,
758 );
759 }
760 let c_paths = c_opts.include_paths;
761 let mut cpp_paths = cpp_opts.include_paths;
762 let (cpp_pre_len, cpp_post_len) = if cpp_paths.starts_with(&c_paths) {
763 (0, cpp_paths.len() - c_paths.len())
764 } else if cpp_paths.ends_with(&c_paths) {
765 (cpp_paths.len() - c_paths.len(), 0)
766 } else {
767 bail!("C++ search path used by `zig cc` is unexpectedly different from C search path!");
768 };
769 if !cpp_paths[..cpp_pre_len]
770 .iter()
771 .chain(cpp_paths[cpp_paths.len() - cpp_post_len..].iter())
772 .all(|(kind, _)| *kind == Kind::Normal)
773 {
774 bail!("C++ search path used by `zig cc` contains additional special search paths!");
775 }
776
777 let mut args = Vec::new();
830
831 args.push("-nostdinc".to_owned());
834
835 if raw_target.contains("musl") || raw_target.contains("ohos") {
843 args.push("-D_LIBCPP_HAS_MUSL_LIBC".to_owned());
844 args.push("-D_LARGEFILE64_SOURCE".to_owned());
847 }
848 args.push("-D_LIBCPP_DISABLE_VISIBILITY_ANNOTATIONS".to_owned());
849 args.push("-D_LIBCXXABI_DISABLE_VISIBILITY_ANNOTATIONS".to_owned());
850 args.push("-D_LIBCPP_HAS_NO_VENDOR_AVAILABILITY_ANNOTATIONS".to_owned());
851 args.push("-D_LIBCPP_PSTL_CPU_BACKEND_SERIAL".to_owned());
852 args.push("-D_LIBCPP_ABI_VERSION=1".to_owned());
853 args.push("-D_LIBCPP_ABI_NAMESPACE=__1".to_owned());
854 args.push("-D_LIBCPP_HARDENING_MODE=_LIBCPP_HARDENING_MODE_FAST".to_owned());
855 if let Some(ver) = c_opts.glibc_minor_ver {
856 args.push(format!("-D__GLIBC_MINOR__={ver}"));
858 }
859
860 for (kind, path) in cpp_paths.drain(..cpp_pre_len) {
861 assert!(kind == Kind::Normal);
862 args.push("-cxx-isystem".to_owned());
868 args.push(path);
869 }
870
871 for (kind, path) in c_paths {
872 match kind {
873 Kind::Normal => {
874 args.push("-Xclang".to_owned());
876 args.push("-c-isystem".to_owned());
877 args.push("-Xclang".to_owned());
878 args.push(path.clone());
879 args.push("-cxx-isystem".to_owned());
880 args.push(path);
881 }
882 Kind::Framework => {
883 args.push("-iframework".to_owned());
884 args.push(path);
885 }
886 }
887 }
888
889 for (kind, path) in cpp_paths.drain(cpp_paths.len() - cpp_post_len..) {
890 assert!(kind == Kind::Normal);
891 args.push("-cxx-isystem".to_owned());
892 args.push(path);
893 }
894
895 Ok(args)
896 }
897
898 fn setup_os_deps(
899 manifest_path: Option<&Path>,
900 release: bool,
901 cargo: &cargo_options::CommonOptions,
902 ) -> Result<()> {
903 for target in &cargo.target {
904 if target.contains("apple") {
905 let target_dir = if let Some(target_dir) = cargo.target_dir.clone() {
906 target_dir.join(target)
907 } else {
908 let manifest_path = manifest_path.unwrap_or_else(|| Path::new("Cargo.toml"));
909 if !manifest_path.exists() {
910 continue;
912 }
913 let metadata = cargo_metadata::MetadataCommand::new()
914 .manifest_path(manifest_path)
915 .no_deps()
916 .exec()?;
917 metadata.target_directory.into_std_path_buf().join(target)
918 };
919 let profile = match cargo.profile.as_deref() {
920 Some("dev" | "test") => "debug",
921 Some("release" | "bench") => "release",
922 Some(profile) => profile,
923 None => {
924 if release {
925 "release"
926 } else {
927 "debug"
928 }
929 }
930 };
931 let deps_dir = target_dir.join(profile).join("deps");
932 fs::create_dir_all(&deps_dir)?;
933 if !target_dir.join("CACHEDIR.TAG").is_file() {
934 let _ = write_file(
936 &target_dir.join("CACHEDIR.TAG"),
937 "Signature: 8a477f597d28d172789f06886806bc55
938# This file is a cache directory tag created by cargo.
939# For information about cache directory tags see https://bford.info/cachedir/
940",
941 );
942 }
943 write_tbd_files(&deps_dir)?;
944 } else if target.contains("arm") && target.contains("linux") {
945 if let Ok(lib_dir) = Zig::lib_dir() {
947 let arm_features_h = lib_dir
948 .join("libc")
949 .join("glibc")
950 .join("sysdeps")
951 .join("arm")
952 .join("arm-features.h");
953 if !arm_features_h.is_file() {
954 fs::write(arm_features_h, ARM_FEATURES_H)?;
955 }
956 }
957 } else if target.contains("windows-gnu") {
958 if let Ok(lib_dir) = Zig::lib_dir() {
959 let lib_common = lib_dir.join("libc").join("mingw").join("lib-common");
960 let synchronization_def = lib_common.join("synchronization.def");
961 if !synchronization_def.is_file() {
962 let api_ms_win_core_synch_l1_2_0_def =
963 lib_common.join("api-ms-win-core-synch-l1-2-0.def");
964 fs::copy(api_ms_win_core_synch_l1_2_0_def, synchronization_def).ok();
966 }
967 }
968 }
969 }
970 Ok(())
971 }
972
973 fn setup_cmake_toolchain(
974 target: &str,
975 zig_wrapper: &ZigWrapper,
976 enable_zig_ar: bool,
977 ) -> Result<PathBuf> {
978 let cmake = cache_dir().join("cmake");
979 fs::create_dir_all(&cmake)?;
980
981 let toolchain_file = cmake.join(format!("{target}-toolchain.cmake"));
982 let triple: Triple = target.parse()?;
983 let os = triple.operating_system.to_string();
984 let arch = triple.architecture.to_string();
985 let (system_name, system_processor) = match (os.as_str(), arch.as_str()) {
986 ("darwin", "x86_64") => ("Darwin", "x86_64"),
987 ("darwin", "aarch64") => ("Darwin", "arm64"),
988 ("linux", arch) => {
989 let cmake_arch = match arch {
990 "powerpc" => "ppc",
991 "powerpc64" => "ppc64",
992 "powerpc64le" => "ppc64le",
993 _ => arch,
994 };
995 ("Linux", cmake_arch)
996 }
997 ("windows", "x86_64") => ("Windows", "AMD64"),
998 ("windows", "i686") => ("Windows", "X86"),
999 ("windows", "aarch64") => ("Windows", "ARM64"),
1000 (os, arch) => (os, arch),
1001 };
1002 let mut content = format!(
1003 r#"
1004set(CMAKE_SYSTEM_NAME {system_name})
1005set(CMAKE_SYSTEM_PROCESSOR {system_processor})
1006set(CMAKE_C_COMPILER {cc})
1007set(CMAKE_CXX_COMPILER {cxx})
1008set(CMAKE_RANLIB {ranlib})"#,
1009 system_name = system_name,
1010 system_processor = system_processor,
1011 cc = zig_wrapper.cc.to_slash_lossy(),
1012 cxx = zig_wrapper.cxx.to_slash_lossy(),
1013 ranlib = zig_wrapper.ranlib.to_slash_lossy(),
1014 );
1015 if enable_zig_ar {
1016 content.push_str(&format!(
1017 "\nset(CMAKE_AR {})\n",
1018 zig_wrapper.ar.to_slash_lossy()
1019 ));
1020 }
1021 write_file(&toolchain_file, &content)?;
1022 Ok(toolchain_file)
1023 }
1024
1025 #[cfg(target_os = "macos")]
1026 fn macos_sdk_root() -> Option<PathBuf> {
1027 match env::var_os("SDKROOT") {
1028 Some(sdkroot) => {
1029 if !sdkroot.is_empty() {
1030 Some(sdkroot.into())
1031 } else {
1032 None
1033 }
1034 }
1035 None => {
1036 let output = Command::new("xcrun")
1037 .args(["--sdk", "macosx", "--show-sdk-path"])
1038 .output();
1039 if let Ok(output) = output {
1040 if output.status.success() {
1041 if let Ok(stdout) = String::from_utf8(output.stdout) {
1042 let stdout = stdout.trim();
1043 if !stdout.is_empty() {
1044 return Some(stdout.into());
1045 }
1046 }
1047 }
1048 }
1049 None
1050 }
1051 }
1052 }
1053
1054 #[cfg(not(target_os = "macos"))]
1055 fn macos_sdk_root() -> Option<PathBuf> {
1056 match env::var_os("SDKROOT") {
1057 Some(sdkroot) if !sdkroot.is_empty() => Some(sdkroot.into()),
1058 _ => None,
1059 }
1060 }
1061}
1062
1063fn write_file(path: &Path, content: &str) -> Result<(), anyhow::Error> {
1064 let existing_content = fs::read_to_string(path).unwrap_or_default();
1065 if existing_content != content {
1066 fs::write(path, content)?;
1067 }
1068 Ok(())
1069}
1070
1071fn write_tbd_files(deps_dir: &Path) -> Result<(), anyhow::Error> {
1072 write_file(&deps_dir.join("libiconv.tbd"), LIBICONV_TBD)?;
1073 write_file(&deps_dir.join("libcharset.1.tbd"), LIBCHARSET_TBD)?;
1074 write_file(&deps_dir.join("libcharset.tbd"), LIBCHARSET_TBD)?;
1075 Ok(())
1076}
1077
1078fn cache_dir() -> PathBuf {
1079 env::var("CARGO_ZIGBUILD_CACHE_DIR")
1080 .ok()
1081 .map(|s| s.into())
1082 .or_else(dirs::cache_dir)
1083 .unwrap_or_else(|| env::current_dir().expect("Failed to get current dir"))
1085 .join(env!("CARGO_PKG_NAME"))
1086 .join(env!("CARGO_PKG_VERSION"))
1087}
1088
1089#[derive(Debug, Deserialize)]
1090struct ZigEnv {
1091 lib_dir: String,
1092}
1093
1094#[derive(Debug, Clone)]
1096pub struct ZigWrapper {
1097 pub cc: PathBuf,
1098 pub cxx: PathBuf,
1099 pub ar: PathBuf,
1100 pub ranlib: PathBuf,
1101 pub lib: PathBuf,
1102}
1103
1104#[derive(Debug, Clone, Default, PartialEq)]
1105struct TargetFlags {
1106 pub target_cpu: String,
1107 pub target_feature: String,
1108}
1109
1110impl TargetFlags {
1111 pub fn parse_from_encoded(encoded: &OsStr) -> Result<Self> {
1112 let mut parsed = Self::default();
1113
1114 let f = rustflags::from_encoded(encoded);
1115 for flag in f {
1116 if let rustflags::Flag::Codegen { opt, value } = flag {
1117 let key = opt.replace('-', "_");
1118 match key.as_str() {
1119 "target_cpu" => {
1120 if let Some(value) = value {
1121 parsed.target_cpu = value;
1122 }
1123 }
1124 "target_feature" => {
1125 if let Some(value) = value {
1127 if !parsed.target_feature.is_empty() {
1128 parsed.target_feature.push(',');
1129 }
1130 parsed.target_feature.push_str(&value);
1131 }
1132 }
1133 _ => {}
1134 }
1135 }
1136 }
1137 Ok(parsed)
1138 }
1139}
1140
1141#[allow(clippy::blocks_in_conditions)]
1150pub fn prepare_zig_linker(target: &str) -> Result<ZigWrapper> {
1151 let (rust_target, abi_suffix) = target.split_once('.').unwrap_or((target, ""));
1152 let abi_suffix = if abi_suffix.is_empty() {
1153 String::new()
1154 } else {
1155 if abi_suffix
1156 .split_once('.')
1157 .filter(|(x, y)| {
1158 !x.is_empty()
1159 && x.chars().all(|c| c.is_ascii_digit())
1160 && !y.is_empty()
1161 && y.chars().all(|c| c.is_ascii_digit())
1162 })
1163 .is_none()
1164 {
1165 bail!("Malformed zig target abi suffix.")
1166 }
1167 format!(".{abi_suffix}")
1168 };
1169 let triple: Triple = rust_target
1170 .parse()
1171 .with_context(|| format!("Unsupported Rust target '{rust_target}'"))?;
1172 let arch = triple.architecture.to_string();
1173 let target_env = match (triple.architecture, triple.environment) {
1174 (Architecture::Mips32(..), Environment::Gnu) => Environment::Gnueabihf,
1175 (Architecture::Powerpc, Environment::Gnu) => Environment::Gnueabihf,
1176 (_, Environment::GnuLlvm) => Environment::Gnu,
1177 (_, environment) => environment,
1178 };
1179 let file_ext = if cfg!(windows) { "bat" } else { "sh" };
1180 let file_target = target.trim_end_matches('.');
1181
1182 let mut cc_args = vec!["-g".to_owned()]; let zig_mcpu_default = match triple.operating_system {
1187 OperatingSystem::Linux => {
1188 match arch.as_str() {
1189 "arm" => match target_env {
1191 Environment::Gnueabi | Environment::Musleabi => "generic+v6+strict_align",
1192 Environment::Gnueabihf | Environment::Musleabihf => {
1193 "generic+v6+strict_align+vfp2-d32"
1194 }
1195 _ => "",
1196 },
1197 "armv5te" => "generic+soft_float+strict_align",
1198 "armv7" => "generic+v7a+vfp3-d32+thumb2-neon",
1199 arch_str @ ("i586" | "i686") => {
1200 if arch_str == "i586" {
1201 "pentium"
1202 } else {
1203 "pentium4"
1204 }
1205 }
1206 "riscv64gc" => "generic_rv64+m+a+f+d+c",
1207 "s390x" => "z10-vector",
1208 _ => "",
1209 }
1210 }
1211 _ => "",
1212 };
1213
1214 let zig_mcpu_override = {
1218 let cargo_config = cargo_config2::Config::load()?;
1219 let rust_flags = cargo_config.rustflags(rust_target)?.unwrap_or_default();
1220 let encoded_rust_flags = rust_flags.encode()?;
1221 let target_flags = TargetFlags::parse_from_encoded(OsStr::new(&encoded_rust_flags))?;
1222 target_flags.target_cpu.replace('-', "_")
1225 };
1226
1227 if !zig_mcpu_override.is_empty() {
1228 cc_args.push(format!("-mcpu={zig_mcpu_override}"));
1229 } else if !zig_mcpu_default.is_empty() {
1230 cc_args.push(format!("-mcpu={zig_mcpu_default}"));
1231 }
1232
1233 match triple.operating_system {
1234 OperatingSystem::Linux => {
1235 let zig_arch = match arch.as_str() {
1236 "arm" => "arm",
1238 "armv5te" => "arm",
1239 "armv7" => "arm",
1240 "i586" | "i686" => {
1241 let zig_version = Zig::zig_version()?;
1242 if zig_version.major == 0 && zig_version.minor >= 11 {
1243 "x86"
1244 } else {
1245 "i386"
1246 }
1247 }
1248 "riscv64gc" => "riscv64",
1249 "s390x" => "s390x",
1250 _ => arch.as_str(),
1251 };
1252 cc_args.push(format!("-target {zig_arch}-linux-{target_env}{abi_suffix}"));
1253 }
1254 OperatingSystem::MacOSX { .. } | OperatingSystem::Darwin(_) => {
1255 let zig_version = Zig::zig_version()?;
1256 if zig_version > semver::Version::new(0, 9, 1) {
1259 cc_args.push(format!("-target {arch}-macos-none{abi_suffix}"));
1260 } else {
1261 cc_args.push(format!("-target {arch}-macos-gnu{abi_suffix}"));
1262 }
1263 }
1264 OperatingSystem::Windows { .. } => {
1265 let zig_arch = match arch.as_str() {
1266 "i686" => {
1267 let zig_version = Zig::zig_version()?;
1268 if zig_version.major == 0 && zig_version.minor >= 11 {
1269 "x86"
1270 } else {
1271 "i386"
1272 }
1273 }
1274 arch => arch,
1275 };
1276 cc_args.push(format!(
1277 "-target {zig_arch}-windows-{target_env}{abi_suffix}"
1278 ));
1279 }
1280 OperatingSystem::Emscripten => {
1281 cc_args.push(format!("-target {arch}-emscripten{abi_suffix}"));
1282 }
1283 OperatingSystem::Wasi => {
1284 cc_args.push(format!("-target {arch}-wasi{abi_suffix}"));
1285 }
1286 OperatingSystem::WasiP1 => {
1287 cc_args.push(format!("-target {arch}-wasi.0.1.0{abi_suffix}"));
1288 }
1289 OperatingSystem::Unknown => {
1290 if triple.architecture == Architecture::Wasm32
1291 || triple.architecture == Architecture::Wasm64
1292 {
1293 cc_args.push(format!("-target {arch}-freestanding{abi_suffix}"));
1294 } else {
1295 bail!("unsupported target '{rust_target}'")
1296 }
1297 }
1298 _ => bail!(format!("unsupported target '{rust_target}'")),
1299 };
1300
1301 let zig_linker_dir = cache_dir();
1302 fs::create_dir_all(&zig_linker_dir)?;
1303
1304 if triple.operating_system == OperatingSystem::Linux {
1305 if matches!(
1306 triple.environment,
1307 Environment::Gnu
1308 | Environment::Gnuspe
1309 | Environment::Gnux32
1310 | Environment::Gnueabi
1311 | Environment::Gnuabi64
1312 | Environment::GnuIlp32
1313 | Environment::Gnueabihf
1314 ) {
1315 let glibc_version = if abi_suffix.is_empty() {
1316 (2, 17)
1317 } else {
1318 let mut parts = abi_suffix[1..].split('.');
1319 let major: usize = parts.next().unwrap().parse()?;
1320 let minor: usize = parts.next().unwrap().parse()?;
1321 (major, minor)
1322 };
1323 if glibc_version < (2, 28) {
1325 use crate::linux::{FCNTL_H, FCNTL_MAP};
1326
1327 let zig_version = Zig::zig_version()?;
1328 if zig_version.major == 0 && zig_version.minor < 11 {
1329 let fcntl_map = zig_linker_dir.join("fcntl.map");
1330 let existing_content = fs::read_to_string(&fcntl_map).unwrap_or_default();
1331 if existing_content != FCNTL_MAP {
1332 fs::write(&fcntl_map, FCNTL_MAP)?;
1333 }
1334 let fcntl_h = zig_linker_dir.join("fcntl.h");
1335 let existing_content = fs::read_to_string(&fcntl_h).unwrap_or_default();
1336 if existing_content != FCNTL_H {
1337 fs::write(&fcntl_h, FCNTL_H)?;
1338 }
1339
1340 cc_args.push(format!("-Wl,--version-script={}", fcntl_map.display()));
1341 cc_args.push(format!("-include {}", fcntl_h.display()));
1342 }
1343 }
1344 } else if matches!(
1345 triple.environment,
1346 Environment::Musl
1347 | Environment::Muslabi64
1348 | Environment::Musleabi
1349 | Environment::Musleabihf
1350 ) {
1351 use crate::linux::MUSL_WEAK_SYMBOLS_MAPPING_SCRIPT;
1352
1353 let zig_version = Zig::zig_version()?;
1354 let rustc_version = rustc_version::version_meta()?.semver;
1355
1356 if (zig_version.major, zig_version.minor) >= (0, 11)
1361 && (rustc_version.major, rustc_version.minor) < (1, 72)
1362 {
1363 let weak_symbols_map = zig_linker_dir.join("musl_weak_symbols_map.ld");
1364 fs::write(&weak_symbols_map, MUSL_WEAK_SYMBOLS_MAPPING_SCRIPT)?;
1365
1366 cc_args.push(format!("-Wl,-T,{}", weak_symbols_map.display()));
1367 }
1368 }
1369 }
1370
1371 let cc_args_str = cc_args.join(" ");
1372 let hash = crc::Crc::<u16>::new(&crc::CRC_16_IBM_SDLC).checksum(cc_args_str.as_bytes());
1373 let zig_cc = zig_linker_dir.join(format!("zigcc-{file_target}-{:x}.{file_ext}", hash));
1374 let zig_cxx = zig_linker_dir.join(format!("zigcxx-{file_target}-{:x}.{file_ext}", hash));
1375 let zig_ranlib = zig_linker_dir.join(format!("zigranlib.{file_ext}"));
1376 write_linker_wrapper(&zig_cc, "cc", &cc_args_str)?;
1377 write_linker_wrapper(&zig_cxx, "c++", &cc_args_str)?;
1378 write_linker_wrapper(&zig_ranlib, "ranlib", "")?;
1379
1380 let exe_ext = if cfg!(windows) { ".exe" } else { "" };
1381 let zig_ar = zig_linker_dir.join(format!("ar{exe_ext}"));
1382 symlink_wrapper(&zig_ar)?;
1383 let zig_lib = zig_linker_dir.join(format!("lib{exe_ext}"));
1384 symlink_wrapper(&zig_lib)?;
1385
1386 Ok(ZigWrapper {
1387 cc: zig_cc,
1388 cxx: zig_cxx,
1389 ar: zig_ar,
1390 ranlib: zig_ranlib,
1391 lib: zig_lib,
1392 })
1393}
1394
1395fn symlink_wrapper(target: &Path) -> Result<()> {
1396 let current_exe = if let Ok(exe) = env::var("CARGO_BIN_EXE_cargo-zigbuild") {
1397 PathBuf::from(exe)
1398 } else {
1399 env::current_exe()?
1400 };
1401 #[cfg(windows)]
1402 {
1403 if !target.exists() {
1404 if std::fs::hard_link(¤t_exe, target).is_err() {
1406 std::fs::copy(¤t_exe, target)?;
1408 }
1409 }
1410 }
1411
1412 #[cfg(unix)]
1413 {
1414 if !target.exists() {
1415 if fs::read_link(target).is_ok() {
1416 fs::remove_file(target)?;
1418 }
1419 std::os::unix::fs::symlink(current_exe, target)?;
1420 }
1421 }
1422 Ok(())
1423}
1424
1425#[cfg(target_family = "unix")]
1427fn write_linker_wrapper(path: &Path, command: &str, args: &str) -> Result<()> {
1428 let mut buf = Vec::<u8>::new();
1429 let current_exe = if let Ok(exe) = env::var("CARGO_BIN_EXE_cargo-zigbuild") {
1430 PathBuf::from(exe)
1431 } else {
1432 env::current_exe()?
1433 };
1434 writeln!(&mut buf, "#!/bin/sh")?;
1435 writeln!(
1436 &mut buf,
1437 "exec \"{}\" zig {} -- {} \"$@\"",
1438 current_exe.display(),
1439 command,
1440 args
1441 )?;
1442
1443 let existing_content = fs::read(path).unwrap_or_default();
1447 if existing_content != buf {
1448 OpenOptions::new()
1449 .create(true)
1450 .write(true)
1451 .truncate(true)
1452 .mode(0o700)
1453 .open(path)?
1454 .write_all(&buf)?;
1455 }
1456 Ok(())
1457}
1458
1459#[cfg(not(target_family = "unix"))]
1461fn write_linker_wrapper(path: &Path, command: &str, args: &str) -> Result<()> {
1462 let mut buf = Vec::<u8>::new();
1463 let current_exe = if let Ok(exe) = env::var("CARGO_BIN_EXE_cargo-zigbuild") {
1464 PathBuf::from(exe)
1465 } else {
1466 env::current_exe()?
1467 };
1468 let current_exe = if is_mingw_shell() {
1469 current_exe.to_slash_lossy().to_string()
1470 } else {
1471 current_exe.display().to_string()
1472 };
1473 writeln!(
1474 &mut buf,
1475 "\"{}\" zig {} -- {} %*",
1476 adjust_canonicalization(current_exe),
1477 command,
1478 args
1479 )?;
1480
1481 let existing_content = fs::read(path).unwrap_or_default();
1482 if existing_content != buf {
1483 fs::write(path, buf)?;
1484 }
1485 Ok(())
1486}
1487
1488pub(crate) fn is_mingw_shell() -> bool {
1489 env::var_os("MSYSTEM").is_some() && env::var_os("SHELL").is_some()
1490}
1491
1492#[cfg(target_os = "windows")]
1494pub fn adjust_canonicalization(p: String) -> String {
1495 const VERBATIM_PREFIX: &str = r#"\\?\"#;
1496 if p.starts_with(VERBATIM_PREFIX) {
1497 p[VERBATIM_PREFIX.len()..].to_string()
1498 } else {
1499 p
1500 }
1501}
1502
1503fn python_path() -> Result<PathBuf> {
1504 let python = env::var("CARGO_ZIGBUILD_PYTHON_PATH").unwrap_or_else(|_| "python3".to_string());
1505 Ok(which::which(python)?)
1506}
1507
1508fn zig_path() -> Result<PathBuf> {
1509 let zig = env::var("CARGO_ZIGBUILD_ZIG_PATH").unwrap_or_else(|_| "zig".to_string());
1510 Ok(which::which(zig)?)
1511}
1512
1513#[cfg(test)]
1514mod tests {
1515 use super::*;
1516
1517 #[test]
1518 fn test_target_flags() {
1519 let cases = [
1520 ("-C target-feature=-crt-static", "", "-crt-static"),
1522 ("-C target-cpu=native", "native", ""),
1523 (
1524 "--deny warnings --codegen target-feature=+crt-static",
1525 "",
1526 "+crt-static",
1527 ),
1528 ("-C target_cpu=skylake-avx512", "skylake-avx512", ""),
1529 ("-Ctarget_cpu=x86-64-v3", "x86-64-v3", ""),
1530 (
1531 "-C target-cpu=native --cfg foo -C target-feature=-avx512bf16,-avx512bitalg",
1532 "native",
1533 "-avx512bf16,-avx512bitalg",
1534 ),
1535 (
1536 "--target x86_64-unknown-linux-gnu --codegen=target-cpu=x --codegen=target-cpu=x86-64",
1537 "x86-64",
1538 "",
1539 ),
1540 (
1541 "-Ctarget-feature=+crt-static -Ctarget-feature=+avx",
1542 "",
1543 "+crt-static,+avx",
1544 ),
1545 ];
1546
1547 for (input, expected_target_cpu, expected_target_feature) in cases.iter() {
1548 let args = cargo_config2::Flags::from_space_separated(input);
1549 let encoded_rust_flags = args.encode().unwrap();
1550 let flags = TargetFlags::parse_from_encoded(OsStr::new(&encoded_rust_flags)).unwrap();
1551 assert_eq!(flags.target_cpu, *expected_target_cpu, "{}", input);
1552 assert_eq!(flags.target_feature, *expected_target_feature, "{}", input);
1553 }
1554 }
1555}