midgard/planetwars.dev
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

verify blob digest on upload

Browse source
This commit is contained in:
Ilion Beyst 2022-06-30 20:49:10 +02:00
parent 4d1c0a3289
commit 419029738d
1 changed files with 21 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
planetwars-server/src/modules/registry.rs
View file

@ -125,6 +125,15 @@ where
}
}
// Since async file io just calls spawn_blocking internally, it does not really make sense
// to make this an async function
fn file_sha256_digest(path: &std::path::Path) -> std::io::Result<String> {
let mut file = std::fs::File::open(path)?;
let mut hasher = Sha256::new();
let _n = std::io::copy(&mut file, &mut hasher)?;
Ok(format!("{:x}", hasher.finalize()))
}
async fn get_root(_auth: RegistryAuth) -> impl IntoResponse {
// root should return 200 OK to confirm api compliance
Response::builder()
@ -273,9 +282,15 @@ async fn put_upload(
file.write_all(&chunk).await.unwrap();
_len += chunk.len();
}
file.flush().await.unwrap();
let expected_digest = params.digest.strip_prefix("sha256:").unwrap();
let digest = file_sha256_digest(&upload_path).unwrap();
if digest != expected_digest {
// TODO: return a docker error body
return Err(StatusCode::BAD_REQUEST);
}
let digest = params.digest.strip_prefix("sha256:").unwrap();
// TODO: check the digest
let target_path = PathBuf::from(REGISTRY_PATH).join("sha256").join(&digest);
tokio::fs::rename(&upload_path, &target_path).await.unwrap();
@ -286,8 +301,9 @@ async fn put_upload(
format!("/v2/{}/blobs/{}", repository_name, digest),
)
.header("Docker-Upload-UUID", uuid)
// .header("Range", format!("0-{}", len))
.header("Docker-Content-Digest", digest)
// TODO: set content-range
// .header("Content-Range", format!("0-{}", len))
.header("Docker-Content-Digest", params.digest)
.body(Body::empty())
.unwrap())
}
@ -342,7 +358,7 @@ async fn put_manifest(
.unwrap();
while let Some(Ok(chunk)) = stream.next().await {
hasher.update(&chunk);
file.write(&chunk).await.unwrap();
file.write_all(&chunk).await.unwrap();
}
}
let digest = hasher.finalize();