2019-11-25 02:42:45 +00:00
|
|
|
//! Tests for the jobserver protocol.
|
|
|
|
|
2022-08-03 13:59:35 +00:00
|
|
|
use cargo_util::is_ci;
|
2017-05-30 04:09:53 +00:00
|
|
|
use std::net::TcpListener;
|
2022-08-03 13:59:35 +00:00
|
|
|
use std::process::Command;
|
2018-08-28 09:20:03 +00:00
|
|
|
use std::thread;
|
2017-05-30 04:09:53 +00:00
|
|
|
|
2022-03-26 10:08:11 +00:00
|
|
|
use cargo_test_support::install::{assert_has_installed_exe, cargo_home};
|
2019-09-12 17:14:29 +00:00
|
|
|
use cargo_test_support::{cargo_exe, project};
|
2017-05-30 04:09:53 +00:00
|
|
|
|
2022-03-26 10:08:11 +00:00
|
|
|
const EXE_CONTENT: &str = r#"
|
|
|
|
use std::env;
|
|
|
|
|
|
|
|
fn main() {
|
|
|
|
let var = env::var("CARGO_MAKEFLAGS").unwrap();
|
|
|
|
let arg = var.split(' ')
|
|
|
|
.find(|p| p.starts_with("--jobserver"))
|
|
|
|
.unwrap();
|
|
|
|
let val = &arg[arg.find('=').unwrap() + 1..];
|
|
|
|
validate(val);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(unix)]
|
|
|
|
fn validate(s: &str) {
|
2023-02-25 10:32:28 +00:00
|
|
|
use std::fs::{self, File};
|
2022-03-26 10:08:11 +00:00
|
|
|
use std::io::*;
|
|
|
|
use std::os::unix::prelude::*;
|
|
|
|
|
2023-02-25 10:32:28 +00:00
|
|
|
if let Some((r, w)) = s.split_once(',') {
|
|
|
|
// `--jobserver-auth=R,W`
|
|
|
|
unsafe {
|
|
|
|
let mut read = File::from_raw_fd(r.parse().unwrap());
|
|
|
|
let mut write = File::from_raw_fd(w.parse().unwrap());
|
2022-03-26 10:08:11 +00:00
|
|
|
|
2023-02-25 10:32:28 +00:00
|
|
|
let mut buf = [0];
|
|
|
|
assert_eq!(read.read(&mut buf).unwrap(), 1);
|
|
|
|
assert_eq!(write.write(&buf).unwrap(), 1);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// `--jobserver-auth=fifo:PATH` is the default since GNU Make 4.4
|
|
|
|
let (_, path) = s.split_once(':').expect("fifo:PATH");
|
|
|
|
assert!(fs::metadata(path).unwrap().file_type().is_fifo());
|
2022-03-26 10:08:11 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
fn validate(_: &str) {
|
|
|
|
// a little too complicated for a test...
|
|
|
|
}
|
|
|
|
"#;
|
|
|
|
|
2019-06-05 18:52:53 +00:00
|
|
|
#[cargo_test]
|
2017-05-30 04:09:53 +00:00
|
|
|
fn jobserver_exists() {
|
2018-07-20 11:47:47 +00:00
|
|
|
let p = project()
|
2022-03-26 10:08:11 +00:00
|
|
|
.file("build.rs", EXE_CONTENT)
|
2018-12-08 11:19:47 +00:00
|
|
|
.file("src/lib.rs", "")
|
2017-07-22 03:12:21 +00:00
|
|
|
.build();
|
2017-05-30 04:09:53 +00:00
|
|
|
|
2020-08-07 19:50:25 +00:00
|
|
|
// Explicitly use `-j2` to ensure that there's eventually going to be a
|
2020-08-09 18:43:52 +00:00
|
|
|
// token to read from `validate` above, since running the build script
|
2020-08-07 19:50:25 +00:00
|
|
|
// itself consumes a token.
|
2023-02-16 00:00:50 +00:00
|
|
|
p.cargo("check -j2").run();
|
2017-05-30 04:09:53 +00:00
|
|
|
}
|
|
|
|
|
2022-03-26 10:08:11 +00:00
|
|
|
#[cargo_test]
|
|
|
|
fn external_subcommand_inherits_jobserver() {
|
|
|
|
let make = if cfg!(windows) {
|
|
|
|
"mingw32-make"
|
|
|
|
} else {
|
|
|
|
"make"
|
|
|
|
};
|
|
|
|
if Command::new(make).arg("--version").output().is_err() {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
let name = "cargo-jobserver-check";
|
|
|
|
let p = project()
|
|
|
|
.file(
|
|
|
|
"Cargo.toml",
|
|
|
|
&format!(
|
|
|
|
r#"
|
|
|
|
[package]
|
|
|
|
name = "{name}"
|
|
|
|
version = "0.0.1"
|
|
|
|
"#
|
|
|
|
),
|
|
|
|
)
|
|
|
|
.file("src/main.rs", EXE_CONTENT)
|
|
|
|
.file(
|
|
|
|
"Makefile",
|
|
|
|
"\
|
|
|
|
all:
|
|
|
|
\t+$(CARGO) jobserver-check
|
|
|
|
",
|
|
|
|
)
|
|
|
|
.build();
|
|
|
|
|
|
|
|
p.cargo("install --path .").run();
|
|
|
|
assert_has_installed_exe(cargo_home(), name);
|
|
|
|
|
|
|
|
p.process(make).env("CARGO", cargo_exe()).arg("-j2").run();
|
|
|
|
}
|
|
|
|
|
2022-08-03 13:59:35 +00:00
|
|
|
#[cargo_test]
|
2017-05-30 04:09:53 +00:00
|
|
|
fn makes_jobserver_used() {
|
2022-08-03 13:59:35 +00:00
|
|
|
let make = if cfg!(windows) {
|
|
|
|
"mingw32-make"
|
|
|
|
} else {
|
|
|
|
"make"
|
|
|
|
};
|
|
|
|
if !is_ci() && Command::new(make).arg("--version").output().is_err() {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2018-07-20 11:47:47 +00:00
|
|
|
let p = project()
|
2018-03-14 15:17:44 +00:00
|
|
|
.file(
|
|
|
|
"Cargo.toml",
|
|
|
|
r#"
|
2020-09-27 00:59:58 +00:00
|
|
|
[package]
|
|
|
|
name = "foo"
|
|
|
|
version = "0.0.1"
|
|
|
|
authors = []
|
|
|
|
|
|
|
|
[dependencies]
|
|
|
|
d1 = { path = "d1" }
|
|
|
|
d2 = { path = "d2" }
|
|
|
|
d3 = { path = "d3" }
|
|
|
|
"#,
|
2018-12-08 11:19:47 +00:00
|
|
|
)
|
|
|
|
.file("src/lib.rs", "")
|
2018-03-14 15:17:44 +00:00
|
|
|
.file(
|
|
|
|
"d1/Cargo.toml",
|
|
|
|
r#"
|
2020-09-27 00:59:58 +00:00
|
|
|
[package]
|
|
|
|
name = "d1"
|
|
|
|
version = "0.0.1"
|
|
|
|
authors = []
|
|
|
|
build = "../dbuild.rs"
|
|
|
|
"#,
|
2018-12-08 11:19:47 +00:00
|
|
|
)
|
|
|
|
.file("d1/src/lib.rs", "")
|
2018-03-14 15:17:44 +00:00
|
|
|
.file(
|
|
|
|
"d2/Cargo.toml",
|
|
|
|
r#"
|
2020-09-27 00:59:58 +00:00
|
|
|
[package]
|
|
|
|
name = "d2"
|
|
|
|
version = "0.0.1"
|
|
|
|
authors = []
|
|
|
|
build = "../dbuild.rs"
|
|
|
|
"#,
|
2018-12-08 11:19:47 +00:00
|
|
|
)
|
|
|
|
.file("d2/src/lib.rs", "")
|
2018-03-14 15:17:44 +00:00
|
|
|
.file(
|
|
|
|
"d3/Cargo.toml",
|
|
|
|
r#"
|
2020-09-27 00:59:58 +00:00
|
|
|
[package]
|
|
|
|
name = "d3"
|
|
|
|
version = "0.0.1"
|
|
|
|
authors = []
|
|
|
|
build = "../dbuild.rs"
|
|
|
|
"#,
|
2018-12-08 11:19:47 +00:00
|
|
|
)
|
|
|
|
.file("d3/src/lib.rs", "")
|
2018-03-14 15:17:44 +00:00
|
|
|
.file(
|
|
|
|
"dbuild.rs",
|
|
|
|
r#"
|
2020-09-27 00:59:58 +00:00
|
|
|
use std::net::TcpStream;
|
|
|
|
use std::env;
|
|
|
|
use std::io::Read;
|
|
|
|
|
|
|
|
fn main() {
|
|
|
|
let addr = env::var("ADDR").unwrap();
|
|
|
|
let mut stream = TcpStream::connect(addr).unwrap();
|
|
|
|
let mut v = Vec::new();
|
|
|
|
stream.read_to_end(&mut v).unwrap();
|
|
|
|
}
|
|
|
|
"#,
|
2018-12-08 11:19:47 +00:00
|
|
|
)
|
|
|
|
.file(
|
2018-03-14 15:17:44 +00:00
|
|
|
"Makefile",
|
|
|
|
"\
|
2017-05-30 04:09:53 +00:00
|
|
|
all:
|
|
|
|
\t+$(CARGO) build
|
2018-03-14 15:17:44 +00:00
|
|
|
",
|
2018-12-08 11:19:47 +00:00
|
|
|
)
|
|
|
|
.build();
|
2017-05-30 04:09:53 +00:00
|
|
|
|
|
|
|
let l = TcpListener::bind("127.0.0.1:0").unwrap();
|
|
|
|
let addr = l.local_addr().unwrap();
|
|
|
|
|
|
|
|
let child = thread::spawn(move || {
|
|
|
|
let a1 = l.accept().unwrap();
|
|
|
|
let a2 = l.accept().unwrap();
|
|
|
|
l.set_nonblocking(true).unwrap();
|
|
|
|
|
|
|
|
for _ in 0..1000 {
|
|
|
|
assert!(l.accept().is_err());
|
|
|
|
thread::yield_now();
|
|
|
|
}
|
|
|
|
|
|
|
|
drop(a1);
|
|
|
|
l.set_nonblocking(false).unwrap();
|
|
|
|
let a3 = l.accept().unwrap();
|
|
|
|
|
|
|
|
drop((a2, a3));
|
|
|
|
});
|
|
|
|
|
2022-08-03 13:59:35 +00:00
|
|
|
p.process(make)
|
2018-08-28 20:38:26 +00:00
|
|
|
.env("CARGO", cargo_exe())
|
|
|
|
.env("ADDR", addr.to_string())
|
|
|
|
.arg("-j2")
|
|
|
|
.run();
|
2017-05-30 04:09:53 +00:00
|
|
|
child.join().unwrap();
|
|
|
|
}
|
|
|
|
|
2019-06-05 18:52:53 +00:00
|
|
|
#[cargo_test]
|
2017-05-30 04:09:53 +00:00
|
|
|
fn jobserver_and_j() {
|
2022-08-03 13:59:35 +00:00
|
|
|
let make = if cfg!(windows) {
|
|
|
|
"mingw32-make"
|
|
|
|
} else {
|
|
|
|
"make"
|
|
|
|
};
|
|
|
|
if !is_ci() && Command::new(make).arg("--version").output().is_err() {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2018-07-20 11:47:47 +00:00
|
|
|
let p = project()
|
2017-07-22 03:12:21 +00:00
|
|
|
.file("src/lib.rs", "")
|
2018-03-14 15:17:44 +00:00
|
|
|
.file(
|
|
|
|
"Makefile",
|
|
|
|
"\
|
2017-05-30 04:09:53 +00:00
|
|
|
all:
|
|
|
|
\t+$(CARGO) build -j2
|
2018-03-14 15:17:44 +00:00
|
|
|
",
|
2018-12-08 11:19:47 +00:00
|
|
|
)
|
|
|
|
.build();
|
2017-05-30 04:09:53 +00:00
|
|
|
|
2022-08-03 13:59:35 +00:00
|
|
|
p.process(make)
|
2018-08-28 20:38:26 +00:00
|
|
|
.env("CARGO", cargo_exe())
|
|
|
|
.arg("-j2")
|
|
|
|
.with_stderr(
|
2018-03-14 15:17:44 +00:00
|
|
|
"\
|
2017-05-30 04:09:53 +00:00
|
|
|
warning: a `-j` argument was passed to Cargo but Cargo is also configured \
|
|
|
|
with an external jobserver in its environment, ignoring the `-j` parameter
|
|
|
|
[COMPILING] [..]
|
|
|
|
[FINISHED] [..]
|
2018-03-14 15:17:44 +00:00
|
|
|
",
|
2018-12-08 11:19:47 +00:00
|
|
|
)
|
|
|
|
.run();
|
2017-05-30 04:09:53 +00:00
|
|
|
}
|