diff options
Diffstat (limited to 'vendor/httparse/build.rs')
| -rw-r--r-- | vendor/httparse/build.rs | 133 |
1 files changed, 133 insertions, 0 deletions
diff --git a/vendor/httparse/build.rs b/vendor/httparse/build.rs new file mode 100644 index 00000000..bffcedb6 --- /dev/null +++ b/vendor/httparse/build.rs @@ -0,0 +1,133 @@ +use std::env; +use std::ffi::OsString; +use std::process::Command; + +fn main() { + // We check rustc version to enable features beyond MSRV, such as: + // - 1.59 => neon_intrinsics + let rustc = env::var_os("RUSTC").unwrap_or(OsString::from("rustc")); + let output = Command::new(rustc) + .arg("--version") + .output() + .expect("failed to check 'rustc --version'") + .stdout; + + let raw_version = String::from_utf8(output) + .expect("rustc version output should be utf-8"); + + let version = match Version::parse(&raw_version) { + Ok(version) => version, + Err(err) => { + println!("cargo:warning=failed to parse `rustc --version`: {}", err); + return; + } + }; + + enable_new_features(version); +} + +fn enable_new_features(version: Version) { + enable_simd(version); +} + +fn enable_simd(version: Version) { + if env::var_os("CARGO_FEATURE_STD").is_none() { + println!("cargo:warning=building for no_std disables httparse SIMD"); + return; + } + if env::var_os("CARGO_CFG_MIRI").is_some() { + println!("cargo:warning=building for Miri disables httparse SIMD"); + return; + } + + let env_disable = "CARGO_CFG_HTTPARSE_DISABLE_SIMD"; + if var_is(env_disable, "1") { + println!("cargo:warning=detected {} environment variable, disabling SIMD", env_disable); + return; + } + + // 1.59.0 is the first version to support neon_intrinsics + if version >= Version(1, 59, 0) { + println!("cargo:rustc-cfg=httparse_simd_neon_intrinsics"); + } + + println!("cargo:rustc-cfg=httparse_simd"); + + // cfg(target_feature) isn't stable yet, but CARGO_CFG_TARGET_FEATURE has + // a list... We aren't doing anything unsafe, since the is_x86_feature_detected + // macro still checks in the actual lib, BUT! + // + // By peeking at the list here, we can change up slightly how we do feature + // detection in the lib. If our features aren't in the feature list, we + // stick with a cached runtime detection strategy. + // + // But if the features *are* in the list, we benefit from removing our cache, + // since the compiler will eliminate several branches with its internal + // cfg(target_feature) usage. + + + let env_runtime_only = "CARGO_CFG_HTTPARSE_DISABLE_SIMD_COMPILETIME"; + if var_is(env_runtime_only, "1") { + println!("cargo:warning=detected {} environment variable, using runtime SIMD detection only", env_runtime_only); + return; + } + let feature_list = match env::var_os("CARGO_CFG_TARGET_FEATURE") { + Some(var) => match var.into_string() { + Ok(s) => s, + Err(_) => { + println!("cargo:warning=CARGO_CFG_TARGET_FEATURE was not valid utf-8"); + return; + }, + }, + None => { + println!("cargo:warning=CARGO_CFG_TARGET_FEATURE was not set"); + return + }, + }; + + let features = feature_list.split(',').map(|s| s.trim()); + if features.clone().any(|f| f == "sse4.2") { + println!("cargo:rustc-cfg=httparse_simd_target_feature_sse42"); + } + if features.clone().any(|f| f == "avx2") { + println!("cargo:rustc-cfg=httparse_simd_target_feature_avx2"); + } +} + +#[derive(Debug, Clone, Copy, PartialEq, PartialOrd)] +struct Version (u32, u32, u32); + +impl Version { + fn parse(s: &str) -> Result<Version, String> { + if !s.starts_with("rustc ") { + return Err(format!("unrecognized version string: {}", s)); + } + let s = s.trim_start_matches("rustc "); + + let mut iter = s + .split('.') + .take(3) + .map(|s| match s.find(|c: char| !c.is_ascii_digit()) { + Some(end) => &s[..end], + None => s, + }) + .map(|s| s.parse::<u32>().map_err(|e| e.to_string())); + + if iter.clone().count() != 3 { + return Err(format!("not enough version parts: {:?}", s)); + } + + let major = iter.next().unwrap()?; + let minor = iter.next().unwrap()?; + let patch = iter.next().unwrap()?; + + Ok(Version(major, minor, patch)) + } +} + +fn var_is(key: &str, val: &str) -> bool { + match env::var(key) { + Ok(v) => v == val, + Err(_) => false, + } +} |
