/
build.rs
66 lines (56 loc) · 2.01 KB
/
build.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
use std::env;
use std::fs::{read_dir, read_to_string, OpenOptions};
use std::io::Write;
fn main() {
static_vcruntime::metabuild();
let export_file_path = env::var("EXPORTS_FILE").expect("No EXPORTS_FILE ?");
let mut out_file = env::var("OUT_DIR").unwrap();
if cfg!(target_env = "msvc") {
out_file.push_str("/proxy.c");
let mut data_file = OpenOptions::new()
.create(true)
.write(true)
.truncate(true)
.open(&out_file)
.expect("Failed :(");
for l in read_to_string(&export_file_path).unwrap().lines() {
let line = format!(
"#pragma comment(linker, \"/export:{}\")\n",
l.replace(" @", ",@")
);
data_file.write(line.as_bytes()).expect("failed writing");
}
} else if cfg!(target_env = "gnu") {
out_file.push_str("/proxy.def");
let mut data_file = OpenOptions::new()
.create(true)
.write(true)
.truncate(true)
.open(&out_file)
.expect("Failed :(");
data_file
.write("EXPORTS\n".as_bytes())
.expect("failed writing .def headers");
for l in read_to_string(&export_file_path).unwrap().lines() {
data_file
.write(format!("{}\n", l).as_bytes())
.expect("failed writing");
}
}
if cfg!(target_env = "msvc") {
cc::Build::new().file(&out_file).compile("proxy");
for f in read_dir(env::var("OUT_DIR").unwrap()).unwrap() {
let file_name = f.unwrap().file_name();
let file_name = file_name.to_string_lossy();
if file_name.ends_with(".o") {
println!("cargo:rustc-link-arg={}", file_name)
}
}
} else if cfg!(target_env = "gnu") {
println!("cargo:rustc-link-arg={}", out_file);
}
println!(
"cargo:rustc-env=OUTPUT_PATH={}",
env::var("OUTPUT_PATH").expect("No OUTPUT_PATH ?")
);
}