1 // Copyright 2017 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
12 extern crate rustc_serialize;
14 use std::collections::{BTreeMap, HashMap};
17 use std::io::{self, Read, Write};
18 use std::path::{PathBuf, Path};
19 use std::process::{Command, Stdio};
21 static HOSTS: &'static [&'static str] = &[
22 "aarch64-unknown-linux-gnu",
23 "arm-unknown-linux-gnueabi",
24 "arm-unknown-linux-gnueabihf",
25 "armv7-unknown-linux-gnueabihf",
27 "i686-pc-windows-gnu",
28 "i686-pc-windows-msvc",
29 "i686-unknown-linux-gnu",
30 "mips-unknown-linux-gnu",
31 "mips64-unknown-linux-gnuabi64",
32 "mips64el-unknown-linux-gnuabi64",
33 "mipsel-unknown-linux-gnu",
34 "powerpc-unknown-linux-gnu",
35 "powerpc64-unknown-linux-gnu",
36 "powerpc64le-unknown-linux-gnu",
37 "s390x-unknown-linux-gnu",
38 "x86_64-apple-darwin",
39 "x86_64-pc-windows-gnu",
40 "x86_64-pc-windows-msvc",
41 "x86_64-unknown-freebsd",
42 "x86_64-unknown-linux-gnu",
43 "x86_64-unknown-netbsd",
46 static TARGETS: &'static [&'static str] = &[
48 "aarch64-unknown-fuchsia",
49 "aarch64-linux-android",
50 "aarch64-unknown-linux-gnu",
51 "arm-linux-androideabi",
52 "arm-unknown-linux-gnueabi",
53 "arm-unknown-linux-gnueabihf",
54 "arm-unknown-linux-musleabi",
55 "arm-unknown-linux-musleabihf",
57 "armv7-linux-androideabi",
58 "armv7-unknown-linux-gnueabihf",
59 "armv7-unknown-linux-musleabihf",
61 "asmjs-unknown-emscripten",
63 "i586-pc-windows-msvc",
64 "i586-unknown-linux-gnu",
67 "i686-pc-windows-gnu",
68 "i686-pc-windows-msvc",
69 "i686-unknown-freebsd",
70 "i686-unknown-linux-gnu",
71 "i686-unknown-linux-musl",
72 "mips-unknown-linux-gnu",
73 "mips-unknown-linux-musl",
74 "mips64-unknown-linux-gnuabi64",
75 "mips64el-unknown-linux-gnuabi64",
76 "mipsel-unknown-linux-gnu",
77 "mipsel-unknown-linux-musl",
78 "powerpc-unknown-linux-gnu",
79 "powerpc64-unknown-linux-gnu",
80 "powerpc64le-unknown-linux-gnu",
81 "s390x-unknown-linux-gnu",
82 "sparc64-unknown-linux-gnu",
83 "wasm32-unknown-emscripten",
84 "x86_64-apple-darwin",
86 "x86_64-pc-windows-gnu",
87 "x86_64-pc-windows-msvc",
88 "x86_64-rumprun-netbsd",
89 "x86_64-unknown-freebsd",
90 "x86_64-unknown-fuchsia",
91 "x86_64-unknown-linux-gnu",
92 "x86_64-unknown-linux-musl",
93 "x86_64-unknown-netbsd",
96 static MINGW: &'static [&'static str] = &[
97 "i686-pc-windows-gnu",
98 "x86_64-pc-windows-gnu",
102 manifest_version: String,
104 pkg: HashMap<String, Package>,
107 #[derive(RustcEncodable)]
110 target: HashMap<String, Target>,
113 #[derive(RustcEncodable)]
117 hash: Option<String>,
118 components: Option<Vec<Component>>,
119 extensions: Option<Vec<Component>>,
122 #[derive(RustcEncodable)]
129 ($e:expr) => (match $e {
131 Err(e) => panic!("{} failed with {}", stringify!($e), e),
136 rust_release: String,
137 cargo_release: String,
140 gpg_passphrase: String,
141 digests: HashMap<String, String>,
144 rust_version: String,
145 cargo_version: String,
149 let mut args = env::args().skip(1);
150 let input = PathBuf::from(args.next().unwrap());
151 let output = PathBuf::from(args.next().unwrap());
152 let date = args.next().unwrap();
153 let rust_release = args.next().unwrap();
154 let cargo_release = args.next().unwrap();
155 let s3_address = args.next().unwrap();
156 let mut passphrase = String::new();
157 t!(io::stdin().read_to_string(&mut passphrase));
160 rust_release: rust_release,
161 cargo_release: cargo_release,
164 gpg_passphrase: passphrase,
165 digests: HashMap::new(),
166 s3_address: s3_address,
168 rust_version: String::new(),
169 cargo_version: String::new(),
174 fn build(&mut self) {
175 self.rust_version = self.version("rust", "x86_64-unknown-linux-gnu");
176 self.cargo_version = self.version("cargo", "x86_64-unknown-linux-gnu");
178 self.digest_and_sign();
179 let Manifest { manifest_version, date, pkg } = self.build_manifest();
181 // Unfortunately we can't use derive(RustcEncodable) here because the
182 // version field is called `manifest-version`, not `manifest_version`.
183 // In lieu of that just create the table directly here with a `BTreeMap`
184 // and wrap it up in a `Value::Table`.
185 let mut manifest = BTreeMap::new();
186 manifest.insert("manifest-version".to_string(),
187 toml::Value::String(manifest_version));
188 manifest.insert("date".to_string(), toml::Value::String(date.clone()));
189 manifest.insert("pkg".to_string(), toml::encode(&pkg));
190 let manifest = toml::Value::Table(manifest).to_string();
192 let filename = format!("channel-rust-{}.toml", self.rust_release);
193 self.write_manifest(&manifest, &filename);
195 let filename = format!("channel-rust-{}-date.txt", self.rust_release);
196 self.write_date_stamp(&date, &filename);
198 if self.rust_release != "beta" && self.rust_release != "nightly" {
199 self.write_manifest(&manifest, "channel-rust-stable.toml");
200 self.write_date_stamp(&date, "channel-rust-stable-date.txt");
204 fn digest_and_sign(&mut self) {
205 for file in t!(self.input.read_dir()).map(|e| t!(e).path()) {
206 let filename = file.file_name().unwrap().to_str().unwrap();
207 let digest = self.hash(&file);
209 assert!(self.digests.insert(filename.to_string(), digest).is_none());
213 fn build_manifest(&mut self) -> Manifest {
214 let mut manifest = Manifest {
215 manifest_version: "2".to_string(),
216 date: self.date.to_string(),
220 self.package("rustc", &mut manifest.pkg, HOSTS);
221 self.package("cargo", &mut manifest.pkg, HOSTS);
222 self.package("rust-mingw", &mut manifest.pkg, MINGW);
223 self.package("rust-std", &mut manifest.pkg, TARGETS);
224 self.package("rust-docs", &mut manifest.pkg, TARGETS);
225 self.package("rust-src", &mut manifest.pkg, &["*"]);
227 if self.rust_release == "nightly" {
228 self.package("rust-analysis", &mut manifest.pkg, TARGETS);
231 let mut pkg = Package {
232 version: self.cached_version("rust").to_string(),
233 target: HashMap::new(),
236 let filename = self.filename("rust", host);
237 let digest = match self.digests.remove(&filename) {
238 Some(digest) => digest,
240 pkg.target.insert(host.to_string(), Target {
250 let mut components = Vec::new();
251 let mut extensions = Vec::new();
253 // rustc/rust-std/cargo/docs are all required, and so is rust-mingw
254 // if it's available for the target.
255 components.extend(vec![
256 Component { pkg: "rustc".to_string(), target: host.to_string() },
257 Component { pkg: "rust-std".to_string(), target: host.to_string() },
258 Component { pkg: "cargo".to_string(), target: host.to_string() },
259 Component { pkg: "rust-docs".to_string(), target: host.to_string() },
261 if host.contains("pc-windows-gnu") {
262 components.push(Component {
263 pkg: "rust-mingw".to_string(),
264 target: host.to_string(),
268 for target in TARGETS {
270 extensions.push(Component {
271 pkg: "rust-std".to_string(),
272 target: target.to_string(),
275 if self.rust_release == "nightly" {
276 extensions.push(Component {
277 pkg: "rust-analysis".to_string(),
278 target: target.to_string(),
282 extensions.push(Component {
283 pkg: "rust-src".to_string(),
284 target: "*".to_string(),
287 pkg.target.insert(host.to_string(), Target {
289 url: Some(self.url("rust", host)),
291 components: Some(components),
292 extensions: Some(extensions),
295 manifest.pkg.insert("rust".to_string(), pkg);
300 fn package(&mut self,
302 dst: &mut HashMap<String, Package>,
304 let targets = targets.iter().map(|name| {
305 let filename = self.filename(pkgname, name);
306 let digest = match self.digests.remove(&filename) {
307 Some(digest) => digest,
309 return (name.to_string(), Target {
319 (name.to_string(), Target {
321 url: Some(self.url(pkgname, name)),
328 dst.insert(pkgname.to_string(), Package {
329 version: self.cached_version(pkgname).to_string(),
334 fn url(&self, component: &str, target: &str) -> String {
338 self.filename(component, target))
341 fn filename(&self, component: &str, target: &str) -> String {
342 if component == "rust-src" {
343 format!("rust-src-{}.tar.gz", self.rust_release)
344 } else if component == "cargo" {
345 format!("cargo-{}-{}.tar.gz", self.cargo_release, target)
347 format!("{}-{}-{}.tar.gz", component, self.rust_release, target)
351 fn cached_version(&self, component: &str) -> &str {
352 if component == "cargo" {
359 fn version(&self, component: &str, target: &str) -> String {
360 let mut cmd = Command::new("tar");
361 let filename = self.filename(component, target);
363 .arg(self.input.join(&filename))
364 .arg(format!("{}/version", filename.replace(".tar.gz", "")))
366 let version = t!(cmd.output());
367 if !version.status.success() {
368 panic!("failed to learn version:\n\n{:?}\n\n{}\n\n{}",
370 String::from_utf8_lossy(&version.stdout),
371 String::from_utf8_lossy(&version.stderr));
373 String::from_utf8_lossy(&version.stdout).trim().to_string()
376 fn hash(&self, path: &Path) -> String {
377 let sha = t!(Command::new("shasum")
378 .arg("-a").arg("256")
379 .arg(path.file_name().unwrap())
380 .current_dir(path.parent().unwrap())
382 assert!(sha.status.success());
384 let filename = path.file_name().unwrap().to_str().unwrap();
385 let sha256 = self.output.join(format!("{}.sha256", filename));
386 t!(t!(File::create(&sha256)).write_all(&sha.stdout));
388 let stdout = String::from_utf8_lossy(&sha.stdout);
389 stdout.split_whitespace().next().unwrap().to_string()
392 fn sign(&self, path: &Path) {
393 let filename = path.file_name().unwrap().to_str().unwrap();
394 let asc = self.output.join(format!("{}.asc", filename));
395 println!("signing: {:?}", path);
396 let mut cmd = Command::new("gpg");
399 .arg("--passphrase-fd").arg("0")
401 .arg("--output").arg(&asc)
402 .arg("--detach-sign").arg(path)
403 .stdin(Stdio::piped());
404 let mut child = t!(cmd.spawn());
405 t!(child.stdin.take().unwrap().write_all(self.gpg_passphrase.as_bytes()));
406 assert!(t!(child.wait()).success());
409 fn write_manifest(&self, manifest: &str, name: &str) {
410 let dst = self.output.join(name);
411 t!(t!(File::create(&dst)).write_all(manifest.as_bytes()));
416 fn write_date_stamp(&self, date: &str, name: &str) {
417 let dst = self.output.join(name);
418 t!(t!(File::create(&dst)).write_all(date.as_bytes()));