fix: am self-update copiait l'archive (zip/tar.gz) au lieu du binaire extrait

Les assets de release sont des archives (am-windows-x86_64.zip, am-linux-*.tar.gz) mais le self-update copiait les octets bruts téléchargés vers le binaire — le 'nouveau binaire' était un zip → 'Exec format error' au lancement. Le download des releases était en fait cassé depuis le début (jamais déclenché : l'asset Windows manquait).

- self_update: download → download::extract_archive (sniff par contenu, déjà utilisé pour les catalogues) → find_binary (préfère am/am.exe/*.exe, sinon le plus gros fichier) → copie du binaire extrait
- 2 tests : préférence am.exe dans un layout zip, fallback plus gros fichier
- E2E réel : le zip v1.0.0 extrait → 'am 1.0.0' exécutable (magic MZ)
This commit is contained in:
2026-08-20 08:15:11 -04:00
parent 2d90c48532
commit ac6f2ffaa5
+74 -4
View File
@@ -4,7 +4,7 @@ use super::*;
use crate::download;
use anyhow::{anyhow, bail, Context, Result};
use std::cmp::Ordering;
use std::path::Path;
use std::path::{Path, PathBuf};
pub fn run(app: &App, check: bool, to: Option<&Path>) -> Result<i32> {
let Some(repo) = app.config.settings.self_update_repo.clone() else {
@@ -51,19 +51,26 @@ pub fn run(app: &App, check: bool, to: Option<&Path>) -> Result<i32> {
let current_exe = std::env::current_exe()?;
let dest = to.map(|p| p.to_path_buf()).unwrap_or_else(|| current_exe.clone());
let tmpdir = tempfile::tempdir()?;
let tmp = tmpdir.path().join("am.new");
let tmp = tmpdir.path().join("am.dl");
download::download(&picked.url, &tmp, None, true)?;
// Release assets are archives (zip on Windows, tar.gz elsewhere):
// unpack them and locate the binary inside.
let staged = if download::extract_archive(&tmp, tmpdir.path())? {
find_binary(tmpdir.path())?
} else {
tmp
};
if dest == current_exe && cfg!(windows) {
// The running executable is locked on Windows.
let new_path = current_exe.with_extension("exe.new");
std::fs::copy(&tmp, &new_path)
std::fs::copy(&staged, &new_path)
.with_context(|| format!("cannot write {}", new_path.display()))?;
app.log.success(&format!(
"downloaded {tag} to {} — replace the running binary manually",
new_path.display()
));
} else {
std::fs::copy(&tmp, &dest)
std::fs::copy(&staged, &dest)
.with_context(|| format!("cannot write {}", dest.display()))?;
#[cfg(unix)]
{
@@ -79,3 +86,66 @@ pub fn run(app: &App, check: bool, to: Option<&Path>) -> Result<i32> {
Ok(0)
}
}
/// Locate the binary inside an unpacked release archive.
fn find_binary(dir: &Path) -> Result<PathBuf> {
let mut entries: Vec<PathBuf> = Vec::new();
collect_files(dir, &mut entries);
if entries.is_empty() {
bail!("no binary found in the release archive");
}
// Prefer a file named exactly `am` / `am.exe`, otherwise the largest
// file (the archive only contains the binary).
entries.sort_by_key(|p| {
let name = p
.file_name()
.map(|n| n.to_string_lossy().to_lowercase())
.unwrap_or_default();
let prefer = name == "am" || name == "am.exe" || name.ends_with(".exe");
(
!prefer,
std::cmp::Reverse(std::fs::metadata(p).map(|m| m.len()).unwrap_or(0)),
)
});
Ok(entries.remove(0))
}
fn collect_files(dir: &Path, out: &mut Vec<PathBuf>) {
if let Ok(rd) = std::fs::read_dir(dir) {
for e in rd.flatten() {
let p = e.path();
if p.is_dir() {
collect_files(&p, out);
} else {
out.push(p);
}
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn find_binary_prefers_am_exe_in_zip_layout() {
let dir = tempfile::tempdir().unwrap();
std::fs::write(dir.path().join("README.md"), "docs").unwrap();
std::fs::write(dir.path().join("am-windows-x86_64.exe"), b"exe").unwrap();
std::fs::write(dir.path().join("am-linux-x86_64"), b"elf").unwrap();
let found = find_binary(dir.path()).unwrap();
assert_eq!(
found.file_name().unwrap().to_str().unwrap(),
"am-windows-x86_64.exe"
);
}
#[test]
fn find_binary_falls_back_to_largest_file() {
let dir = tempfile::tempdir().unwrap();
std::fs::write(dir.path().join("am"), b"small").unwrap();
std::fs::write(dir.path().join("data.bin"), vec![0u8; 4096]).unwrap();
let found = find_binary(dir.path()).unwrap();
assert_eq!(found.file_name().unwrap().to_str().unwrap(), "am");
}
}