1use std::collections::{HashMap, HashSet};
39use std::hash::{Hash, Hasher};
40use std::sync::Arc;
41
42use crate::core::compiler::unit_dependencies::build_unit_dependencies;
43use crate::core::compiler::unit_graph::{self, UnitDep, UnitGraph};
44use crate::core::compiler::UserIntent;
45use crate::core::compiler::{apply_env_config, standard_lib, CrateType, TargetInfo};
46use crate::core::compiler::{BuildConfig, BuildContext, BuildRunner, Compilation};
47use crate::core::compiler::{CompileKind, CompileTarget, RustcTargetData, Unit};
48use crate::core::compiler::{DefaultExecutor, Executor, UnitInterner};
49use crate::core::profiles::Profiles;
50use crate::core::resolver::features::{self, CliFeatures, FeaturesFor};
51use crate::core::resolver::{HasDevUnits, Resolve};
52use crate::core::{PackageId, PackageSet, SourceId, TargetKind, Workspace};
53use crate::drop_println;
54use crate::ops;
55use crate::ops::resolve::WorkspaceResolve;
56use crate::util::context::{GlobalContext, WarningHandling};
57use crate::util::interning::InternedString;
58use crate::util::{CargoResult, StableHasher};
59
60mod compile_filter;
61pub use compile_filter::{CompileFilter, FilterRule, LibRule};
62
63mod unit_generator;
64use unit_generator::UnitGenerator;
65
66mod packages;
67
68pub use packages::Packages;
69
70#[derive(Debug, Clone)]
79pub struct CompileOptions {
80 pub build_config: BuildConfig,
82 pub cli_features: CliFeatures,
84 pub spec: Packages,
86 pub filter: CompileFilter,
89 pub target_rustdoc_args: Option<Vec<String>>,
91 pub target_rustc_args: Option<Vec<String>>,
94 pub target_rustc_crate_types: Option<Vec<String>>,
96 pub rustdoc_document_private_items: bool,
99 pub honor_rust_version: Option<bool>,
102}
103
104impl CompileOptions {
105 pub fn new(gctx: &GlobalContext, intent: UserIntent) -> CargoResult<CompileOptions> {
106 let jobs = None;
107 let keep_going = false;
108 Ok(CompileOptions {
109 build_config: BuildConfig::new(gctx, jobs, keep_going, &[], intent)?,
110 cli_features: CliFeatures::new_all(false),
111 spec: ops::Packages::Packages(Vec::new()),
112 filter: CompileFilter::Default {
113 required_features_filterable: false,
114 },
115 target_rustdoc_args: None,
116 target_rustc_args: None,
117 target_rustc_crate_types: None,
118 rustdoc_document_private_items: false,
119 honor_rust_version: None,
120 })
121 }
122}
123
124pub fn compile<'a>(ws: &Workspace<'a>, options: &CompileOptions) -> CargoResult<Compilation<'a>> {
128 let exec: Arc<dyn Executor> = Arc::new(DefaultExecutor);
129 compile_with_exec(ws, options, &exec)
130}
131
132pub fn compile_with_exec<'a>(
137 ws: &Workspace<'a>,
138 options: &CompileOptions,
139 exec: &Arc<dyn Executor>,
140) -> CargoResult<Compilation<'a>> {
141 ws.emit_warnings()?;
142 let compilation = compile_ws(ws, options, exec)?;
143 if ws.gctx().warning_handling()? == WarningHandling::Deny && compilation.warning_count > 0 {
144 anyhow::bail!("warnings are denied by `build.warnings` configuration")
145 }
146 Ok(compilation)
147}
148
149#[tracing::instrument(skip_all)]
151pub fn compile_ws<'a>(
152 ws: &Workspace<'a>,
153 options: &CompileOptions,
154 exec: &Arc<dyn Executor>,
155) -> CargoResult<Compilation<'a>> {
156 let interner = UnitInterner::new();
157 let bcx = create_bcx(ws, options, &interner)?;
158 if options.build_config.unit_graph {
159 unit_graph::emit_serialized_unit_graph(&bcx.roots, &bcx.unit_graph, ws.gctx())?;
160 return Compilation::new(&bcx);
161 }
162 crate::core::gc::auto_gc(bcx.gctx);
163 let build_runner = BuildRunner::new(&bcx)?;
164 if options.build_config.dry_run {
165 build_runner.dry_run()
166 } else {
167 build_runner.compile(exec)
168 }
169}
170
171pub fn print<'a>(
175 ws: &Workspace<'a>,
176 options: &CompileOptions,
177 print_opt_value: &str,
178) -> CargoResult<()> {
179 let CompileOptions {
180 ref build_config,
181 ref target_rustc_args,
182 ..
183 } = *options;
184 let gctx = ws.gctx();
185 let rustc = gctx.load_global_rustc(Some(ws))?;
186 for (index, kind) in build_config.requested_kinds.iter().enumerate() {
187 if index != 0 {
188 drop_println!(gctx);
189 }
190 let target_info = TargetInfo::new(gctx, &build_config.requested_kinds, &rustc, *kind)?;
191 let mut process = rustc.process();
192 apply_env_config(gctx, &mut process)?;
193 process.args(&target_info.rustflags);
194 if let Some(args) = target_rustc_args {
195 process.args(args);
196 }
197 if let CompileKind::Target(t) = kind {
198 process.arg("--target").arg(t.rustc_target());
199 }
200 process.arg("--print").arg(print_opt_value);
201 process.exec()?;
202 }
203 Ok(())
204}
205
206#[tracing::instrument(skip_all)]
211pub fn create_bcx<'a, 'gctx>(
212 ws: &'a Workspace<'gctx>,
213 options: &'a CompileOptions,
214 interner: &'a UnitInterner,
215) -> CargoResult<BuildContext<'a, 'gctx>> {
216 let CompileOptions {
217 ref build_config,
218 ref spec,
219 ref cli_features,
220 ref filter,
221 ref target_rustdoc_args,
222 ref target_rustc_args,
223 ref target_rustc_crate_types,
224 rustdoc_document_private_items,
225 honor_rust_version,
226 } = *options;
227 let gctx = ws.gctx();
228
229 match build_config.intent {
231 UserIntent::Test | UserIntent::Build | UserIntent::Check { .. } | UserIntent::Bench => {
232 if ws.gctx().get_env("RUST_FLAGS").is_ok() {
233 gctx.shell().warn(
234 "Cargo does not read `RUST_FLAGS` environment variable. Did you mean `RUSTFLAGS`?",
235 )?;
236 }
237 }
238 UserIntent::Doc { .. } | UserIntent::Doctest => {
239 if ws.gctx().get_env("RUSTDOC_FLAGS").is_ok() {
240 gctx.shell().warn(
241 "Cargo does not read `RUSTDOC_FLAGS` environment variable. Did you mean `RUSTDOCFLAGS`?"
242 )?;
243 }
244 }
245 }
246 gctx.validate_term_config()?;
247
248 let mut target_data = RustcTargetData::new(ws, &build_config.requested_kinds)?;
249
250 let specs = spec.to_package_id_specs(ws)?;
251 let has_dev_units = {
252 let any_pkg_has_scrape_enabled = ws
256 .members_with_features(&specs, cli_features)?
257 .iter()
258 .any(|(pkg, _)| {
259 pkg.targets()
260 .iter()
261 .any(|target| target.is_example() && target.doc_scrape_examples().is_enabled())
262 });
263
264 if filter.need_dev_deps(build_config.intent)
265 || (build_config.intent.is_doc() && any_pkg_has_scrape_enabled)
266 {
267 HasDevUnits::Yes
268 } else {
269 HasDevUnits::No
270 }
271 };
272 let dry_run = false;
273 let resolve = ops::resolve_ws_with_opts(
274 ws,
275 &mut target_data,
276 &build_config.requested_kinds,
277 cli_features,
278 &specs,
279 has_dev_units,
280 crate::core::resolver::features::ForceAllTargets::No,
281 dry_run,
282 )?;
283 let WorkspaceResolve {
284 mut pkg_set,
285 workspace_resolve,
286 targeted_resolve: resolve,
287 resolved_features,
288 } = resolve;
289
290 let std_resolve_features = if let Some(crates) = &gctx.cli_unstable().build_std {
291 let (std_package_set, std_resolve, std_features) = standard_lib::resolve_std(
292 ws,
293 &mut target_data,
294 &build_config,
295 crates,
296 &build_config.requested_kinds,
297 )?;
298 pkg_set.add_set(std_package_set);
299 Some((std_resolve, std_features))
300 } else {
301 None
302 };
303
304 let to_build_ids = resolve.specs_to_ids(&specs)?;
308 let mut to_builds = pkg_set.get_many(to_build_ids)?;
312
313 to_builds.sort_by_key(|p| p.package_id());
317
318 for pkg in to_builds.iter() {
319 pkg.manifest().print_teapot(gctx);
320
321 if build_config.intent.is_any_test()
322 && !ws.is_member(pkg)
323 && pkg.dependencies().iter().any(|dep| !dep.is_transitive())
324 {
325 anyhow::bail!(
326 "package `{}` cannot be tested because it requires dev-dependencies \
327 and is not a member of the workspace",
328 pkg.name()
329 );
330 }
331 }
332
333 let (extra_args, extra_args_name) = match (target_rustc_args, target_rustdoc_args) {
334 (Some(args), _) => (Some(args.clone()), "rustc"),
335 (_, Some(args)) => (Some(args.clone()), "rustdoc"),
336 _ => (None, ""),
337 };
338
339 if extra_args.is_some() && to_builds.len() != 1 {
340 panic!(
341 "`{}` should not accept multiple `-p` flags",
342 extra_args_name
343 );
344 }
345
346 let profiles = Profiles::new(ws, build_config.requested_profile)?;
347 profiles.validate_packages(
348 ws.profiles(),
349 &mut gctx.shell(),
350 workspace_resolve.as_ref().unwrap_or(&resolve),
351 )?;
352
353 let explicit_host_kind = CompileKind::Target(CompileTarget::new(&target_data.rustc.host)?);
357 let explicit_host_kinds: Vec<_> = build_config
358 .requested_kinds
359 .iter()
360 .map(|kind| match kind {
361 CompileKind::Host => explicit_host_kind,
362 CompileKind::Target(t) => CompileKind::Target(*t),
363 })
364 .collect();
365
366 let generator = UnitGenerator {
372 ws,
373 packages: &to_builds,
374 spec,
375 target_data: &target_data,
376 filter,
377 requested_kinds: &build_config.requested_kinds,
378 explicit_host_kind,
379 intent: build_config.intent,
380 resolve: &resolve,
381 workspace_resolve: &workspace_resolve,
382 resolved_features: &resolved_features,
383 package_set: &pkg_set,
384 profiles: &profiles,
385 interner,
386 has_dev_units,
387 };
388 let mut units = generator.generate_root_units()?;
389
390 if let Some(args) = target_rustc_crate_types {
391 override_rustc_crate_types(&mut units, args, interner)?;
392 }
393
394 let should_scrape = build_config.intent.is_doc() && gctx.cli_unstable().rustdoc_scrape_examples;
395 let mut scrape_units = if should_scrape {
396 generator.generate_scrape_units(&units)?
397 } else {
398 Vec::new()
399 };
400
401 let std_roots = if let Some(crates) = gctx.cli_unstable().build_std.as_ref() {
402 let (std_resolve, std_features) = std_resolve_features.as_ref().unwrap();
403 standard_lib::generate_std_roots(
404 &crates,
405 &units,
406 std_resolve,
407 std_features,
408 &explicit_host_kinds,
409 &pkg_set,
410 interner,
411 &profiles,
412 &target_data,
413 )?
414 } else {
415 Default::default()
416 };
417
418 let mut unit_graph = build_unit_dependencies(
419 ws,
420 &pkg_set,
421 &resolve,
422 &resolved_features,
423 std_resolve_features.as_ref(),
424 &units,
425 &scrape_units,
426 &std_roots,
427 build_config.intent,
428 &target_data,
429 &profiles,
430 interner,
431 )?;
432
433 if build_config.intent.wants_deps_docs() {
436 remove_duplicate_doc(build_config, &units, &mut unit_graph);
437 }
438
439 let host_kind_requested = build_config
440 .requested_kinds
441 .iter()
442 .any(CompileKind::is_host);
443 (units, scrape_units, unit_graph) = rebuild_unit_graph_shared(
447 interner,
448 unit_graph,
449 &units,
450 &scrape_units,
451 host_kind_requested.then_some(explicit_host_kind),
452 );
453
454 let mut extra_compiler_args = HashMap::new();
455 if let Some(args) = extra_args {
456 if units.len() != 1 {
457 anyhow::bail!(
458 "extra arguments to `{}` can only be passed to one \
459 target, consider filtering\nthe package by passing, \
460 e.g., `--lib` or `--bin NAME` to specify a single target",
461 extra_args_name
462 );
463 }
464 extra_compiler_args.insert(units[0].clone(), args);
465 }
466
467 for unit in units
468 .iter()
469 .filter(|unit| unit.mode.is_doc() || unit.mode.is_doc_test())
470 .filter(|unit| rustdoc_document_private_items || unit.target.is_bin())
471 {
472 let mut args = vec!["--document-private-items".into()];
476 if unit.target.is_bin() {
477 args.push("-Arustdoc::private-intra-doc-links".into());
481 }
482 extra_compiler_args
483 .entry(unit.clone())
484 .or_default()
485 .extend(args);
486 }
487
488 if honor_rust_version.unwrap_or(true) {
489 let rustc_version = target_data.rustc.version.clone().into();
490
491 let mut incompatible = Vec::new();
492 let mut local_incompatible = false;
493 for unit in unit_graph.keys() {
494 let Some(pkg_msrv) = unit.pkg.rust_version() else {
495 continue;
496 };
497
498 if pkg_msrv.is_compatible_with(&rustc_version) {
499 continue;
500 }
501
502 local_incompatible |= unit.is_local();
503 incompatible.push((unit, pkg_msrv));
504 }
505 if !incompatible.is_empty() {
506 use std::fmt::Write as _;
507
508 let plural = if incompatible.len() == 1 { "" } else { "s" };
509 let mut message = format!(
510 "rustc {rustc_version} is not supported by the following package{plural}:\n"
511 );
512 incompatible.sort_by_key(|(unit, _)| (unit.pkg.name(), unit.pkg.version()));
513 for (unit, msrv) in incompatible {
514 let name = &unit.pkg.name();
515 let version = &unit.pkg.version();
516 writeln!(&mut message, " {name}@{version} requires rustc {msrv}").unwrap();
517 }
518 if ws.is_ephemeral() {
519 if ws.ignore_lock() {
520 writeln!(
521 &mut message,
522 "Try re-running `cargo install` with `--locked`"
523 )
524 .unwrap();
525 }
526 } else if !local_incompatible {
527 writeln!(
528 &mut message,
529 "Either upgrade rustc or select compatible dependency versions with
530`cargo update <name>@<current-ver> --precise <compatible-ver>`
531where `<compatible-ver>` is the latest version supporting rustc {rustc_version}",
532 )
533 .unwrap();
534 }
535 return Err(anyhow::Error::msg(message));
536 }
537 }
538
539 let bcx = BuildContext::new(
540 ws,
541 pkg_set,
542 build_config,
543 profiles,
544 extra_compiler_args,
545 target_data,
546 units,
547 unit_graph,
548 scrape_units,
549 )?;
550
551 Ok(bcx)
552}
553
554fn rebuild_unit_graph_shared(
586 interner: &UnitInterner,
587 unit_graph: UnitGraph,
588 roots: &[Unit],
589 scrape_units: &[Unit],
590 to_host: Option<CompileKind>,
591) -> (Vec<Unit>, Vec<Unit>, UnitGraph) {
592 let mut result = UnitGraph::new();
593 let mut memo = HashMap::new();
596 let new_roots = roots
597 .iter()
598 .map(|root| {
599 traverse_and_share(
600 interner,
601 &mut memo,
602 &mut result,
603 &unit_graph,
604 root,
605 false,
606 to_host,
607 )
608 })
609 .collect();
610 let new_scrape_units = scrape_units
614 .iter()
615 .map(|unit| memo.get(unit).unwrap().clone())
616 .collect();
617 (new_roots, new_scrape_units, result)
618}
619
620fn traverse_and_share(
626 interner: &UnitInterner,
627 memo: &mut HashMap<Unit, Unit>,
628 new_graph: &mut UnitGraph,
629 unit_graph: &UnitGraph,
630 unit: &Unit,
631 unit_is_for_host: bool,
632 to_host: Option<CompileKind>,
633) -> Unit {
634 if let Some(new_unit) = memo.get(unit) {
635 return new_unit.clone();
637 }
638 let mut dep_hash = StableHasher::new();
639 let new_deps: Vec<_> = unit_graph[unit]
640 .iter()
641 .map(|dep| {
642 let new_dep_unit = traverse_and_share(
643 interner,
644 memo,
645 new_graph,
646 unit_graph,
647 &dep.unit,
648 dep.unit_for.is_for_host(),
649 to_host,
650 );
651 new_dep_unit.hash(&mut dep_hash);
652 UnitDep {
653 unit: new_dep_unit,
654 ..dep.clone()
655 }
656 })
657 .collect();
658 let new_dep_hash = Hasher::finish(&dep_hash);
661
662 let canonical_kind = match to_host {
669 Some(to_host) if to_host == unit.kind => CompileKind::Host,
670 _ => unit.kind,
671 };
672
673 let mut profile = unit.profile.clone();
674 if profile.strip.is_deferred() {
675 if !profile.debuginfo.is_turned_on()
679 && new_deps
680 .iter()
681 .all(|dep| !dep.unit.profile.debuginfo.is_turned_on())
682 {
683 profile.strip = profile.strip.strip_debuginfo();
684 }
685 }
686
687 if unit_is_for_host
691 && to_host.is_some()
692 && profile.debuginfo.is_deferred()
693 && !unit.artifact.is_true()
694 {
695 let canonical_debuginfo = profile.debuginfo.finalize();
699 let mut canonical_profile = profile.clone();
700 canonical_profile.debuginfo = canonical_debuginfo;
701 let unit_probe = interner.intern(
702 &unit.pkg,
703 &unit.target,
704 canonical_profile,
705 to_host.unwrap(),
706 unit.mode,
707 unit.features.clone(),
708 unit.rustflags.clone(),
709 unit.rustdocflags.clone(),
710 unit.links_overrides.clone(),
711 unit.is_std,
712 unit.dep_hash,
713 unit.artifact,
714 unit.artifact_target_for_features,
715 );
716
717 profile.debuginfo = if unit_graph.contains_key(&unit_probe) {
719 canonical_debuginfo
722 } else {
723 canonical_debuginfo.weaken()
726 }
727 }
728
729 let new_unit = interner.intern(
730 &unit.pkg,
731 &unit.target,
732 profile,
733 canonical_kind,
734 unit.mode,
735 unit.features.clone(),
736 unit.rustflags.clone(),
737 unit.rustdocflags.clone(),
738 unit.links_overrides.clone(),
739 unit.is_std,
740 new_dep_hash,
741 unit.artifact,
742 None,
745 );
746 assert!(memo.insert(unit.clone(), new_unit.clone()).is_none());
747 new_graph.entry(new_unit.clone()).or_insert(new_deps);
748 new_unit
749}
750
751fn remove_duplicate_doc(
767 build_config: &BuildConfig,
768 root_units: &[Unit],
769 unit_graph: &mut UnitGraph,
770) {
771 let mut all_docs: HashMap<String, Vec<Unit>> = HashMap::new();
774 for unit in unit_graph.keys() {
775 if unit.mode.is_doc() {
776 all_docs
777 .entry(unit.target.crate_name())
778 .or_default()
779 .push(unit.clone());
780 }
781 }
782 let mut removed_units: HashSet<Unit> = HashSet::new();
785 let mut remove = |units: Vec<Unit>, reason: &str, cb: &dyn Fn(&Unit) -> bool| -> Vec<Unit> {
786 let (to_remove, remaining_units): (Vec<Unit>, Vec<Unit>) = units
787 .into_iter()
788 .partition(|unit| cb(unit) && !root_units.contains(unit));
789 for unit in to_remove {
790 tracing::debug!(
791 "removing duplicate doc due to {} for package {} target `{}`",
792 reason,
793 unit.pkg,
794 unit.target.name()
795 );
796 unit_graph.remove(&unit);
797 removed_units.insert(unit);
798 }
799 remaining_units
800 };
801 for (_crate_name, mut units) in all_docs {
803 if units.len() == 1 {
804 continue;
805 }
806 if build_config
808 .requested_kinds
809 .iter()
810 .all(CompileKind::is_host)
811 {
812 units = remove(units, "host/target merger", &|unit| unit.kind.is_host());
817 if units.len() == 1 {
818 continue;
819 }
820 }
821 let mut source_map: HashMap<(InternedString, SourceId, CompileKind), Vec<Unit>> =
823 HashMap::new();
824 for unit in units {
825 let pkg_id = unit.pkg.package_id();
826 source_map
828 .entry((pkg_id.name(), pkg_id.source_id(), unit.kind))
829 .or_default()
830 .push(unit);
831 }
832 let mut remaining_units = Vec::new();
833 for (_key, mut units) in source_map {
834 if units.len() > 1 {
835 units.sort_by(|a, b| a.pkg.version().partial_cmp(b.pkg.version()).unwrap());
836 let newest_version = units.last().unwrap().pkg.version().clone();
838 let keep_units = remove(units, "older version", &|unit| {
839 unit.pkg.version() < &newest_version
840 });
841 remaining_units.extend(keep_units);
842 } else {
843 remaining_units.extend(units);
844 }
845 }
846 if remaining_units.len() == 1 {
847 continue;
848 }
849 }
852 for unit_deps in unit_graph.values_mut() {
854 unit_deps.retain(|unit_dep| !removed_units.contains(&unit_dep.unit));
855 }
856 let mut visited = HashSet::new();
858 fn visit(unit: &Unit, graph: &UnitGraph, visited: &mut HashSet<Unit>) {
859 if !visited.insert(unit.clone()) {
860 return;
861 }
862 for dep in &graph[unit] {
863 visit(&dep.unit, graph, visited);
864 }
865 }
866 for unit in root_units {
867 visit(unit, unit_graph, &mut visited);
868 }
869 unit_graph.retain(|unit, _| visited.contains(unit));
870}
871
872fn override_rustc_crate_types(
876 units: &mut [Unit],
877 args: &[String],
878 interner: &UnitInterner,
879) -> CargoResult<()> {
880 if units.len() != 1 {
881 anyhow::bail!(
882 "crate types to rustc can only be passed to one \
883 target, consider filtering\nthe package by passing, \
884 e.g., `--lib` or `--example` to specify a single target"
885 );
886 }
887
888 let unit = &units[0];
889 let override_unit = |f: fn(Vec<CrateType>) -> TargetKind| {
890 let crate_types = args.iter().map(|s| s.into()).collect();
891 let mut target = unit.target.clone();
892 target.set_kind(f(crate_types));
893 interner.intern(
894 &unit.pkg,
895 &target,
896 unit.profile.clone(),
897 unit.kind,
898 unit.mode,
899 unit.features.clone(),
900 unit.rustflags.clone(),
901 unit.rustdocflags.clone(),
902 unit.links_overrides.clone(),
903 unit.is_std,
904 unit.dep_hash,
905 unit.artifact,
906 unit.artifact_target_for_features,
907 )
908 };
909 units[0] = match unit.target.kind() {
910 TargetKind::Lib(_) => override_unit(TargetKind::Lib),
911 TargetKind::ExampleLib(_) => override_unit(TargetKind::ExampleLib),
912 _ => {
913 anyhow::bail!(
914 "crate types can only be specified for libraries and example libraries.\n\
915 Binaries, tests, and benchmarks are always the `bin` crate type"
916 );
917 }
918 };
919
920 Ok(())
921}
922
923pub fn resolve_all_features(
929 resolve_with_overrides: &Resolve,
930 resolved_features: &features::ResolvedFeatures,
931 package_set: &PackageSet<'_>,
932 package_id: PackageId,
933) -> HashSet<String> {
934 let mut features: HashSet<String> = resolved_features
935 .activated_features(package_id, FeaturesFor::NormalOrDev)
936 .iter()
937 .map(|s| s.to_string())
938 .collect();
939
940 for (dep_id, deps) in resolve_with_overrides.deps(package_id) {
943 let is_proc_macro = package_set
944 .get_one(dep_id)
945 .expect("packages downloaded")
946 .proc_macro();
947 for dep in deps {
948 let features_for = FeaturesFor::from_for_host(is_proc_macro || dep.is_build());
949 for feature in resolved_features
950 .activated_features_unverified(dep_id, features_for)
951 .unwrap_or_default()
952 {
953 features.insert(format!("{}/{}", dep.name_in_toml(), feature));
954 }
955 }
956 }
957
958 features
959}