1use std::collections::HashMap;
7use std::fmt::Formatter;
8use std::io::{BufRead, Cursor};
9
10use chrono::{DateTime, Utc};
11use serde::{de, Deserialize, Deserializer};
12
13use crate::architectures::Architecture;
14use crate::archive::{Codename, Component, Suite};
15use crate::utils::{DateTimeVisitor, WhitespaceListVisitor};
16
17fn deserialize_datetime<'de, D>(deserializer: D) -> Result<DateTime<Utc>, D::Error>
19where
20 D: Deserializer<'de>,
21{
22 deserializer.deserialize_str(DateTimeVisitor("%a, %d %b %Y %H:%M:%S %Z"))
23}
24
25fn deserialize_datetime_option<'de, D>(deserializer: D) -> Result<Option<DateTime<Utc>>, D::Error>
27where
28 D: Deserializer<'de>,
29{
30 deserialize_datetime(deserializer).map(Some)
31}
32
33fn deserialize_architectures<'de, D>(deserializer: D) -> Result<Vec<Architecture>, D::Error>
35where
36 D: Deserializer<'de>,
37{
38 deserializer.deserialize_str(WhitespaceListVisitor::<Architecture>::new())
39}
40
41fn deserialize_components<'de, D>(deserializer: D) -> Result<Vec<Component>, D::Error>
43where
44 D: Deserializer<'de>,
45{
46 deserializer.deserialize_str(WhitespaceListVisitor::<Component>::new())
47}
48
49struct SHA256Visitor;
50
51impl de::Visitor<'_> for SHA256Visitor {
52 type Value = HashMap<String, FileInfo>;
53
54 fn expecting(&self, formatter: &mut Formatter) -> std::fmt::Result {
55 write!(formatter, "a list of files")
56 }
57
58 fn visit_str<E>(self, s: &str) -> Result<Self::Value, E>
59 where
60 E: de::Error,
61 {
62 let cursor = Cursor::new(s);
63 let mut ret: HashMap<String, FileInfo> = HashMap::default();
64 for line in cursor.lines() {
65 let Ok(line) = line else {
66 break;
67 };
68
69 let fields: Vec<_> = line.split_ascii_whitespace().collect();
70 if fields.len() != 3 {
71 return Err(E::invalid_value(de::Unexpected::Str(&line), &self));
72 }
73
74 let file = fields[2];
75 let file_size = fields[1].parse().map_err(E::custom)?;
76 let hash = hex::decode(fields[0]).map_err(E::custom)?;
77
78 ret.insert(
79 file.to_string(),
80 FileInfo {
81 file_size,
82 hash: hash
83 .try_into()
84 .map_err(|_| E::invalid_value(de::Unexpected::Str(fields[0]), &self))?,
85 },
86 );
87 }
88 Ok(ret)
89 }
90}
91
92fn deserialize_sha256<'de, D>(deserializer: D) -> Result<HashMap<String, FileInfo>, D::Error>
94where
95 D: Deserializer<'de>,
96{
97 deserializer.deserialize_str(SHA256Visitor)
98}
99
100#[derive(Debug, Deserialize, PartialEq, Eq)]
102pub struct FileInfo {
103 file_size: u64,
104 hash: [u8; 32],
105}
106
107#[derive(Debug, Deserialize, PartialEq, Eq, Copy, Clone)]
109#[serde(rename_all = "lowercase")]
110pub enum AcquireByHash {
111 Yes,
113 No,
115}
116
117#[derive(Debug, Deserialize, PartialEq, Eq)]
119#[serde(rename_all = "PascalCase")]
120pub struct Release {
121 pub origin: String,
123 pub label: String,
125 pub suite: Suite,
127 pub codename: Codename,
129 pub version: Option<String>,
131 #[serde(deserialize_with = "deserialize_datetime")]
133 pub date: DateTime<Utc>,
134 #[serde(
135 default,
136 deserialize_with = "deserialize_datetime_option",
137 rename = "Valid-Until"
138 )]
139 pub valid_until: Option<DateTime<Utc>>,
141 #[serde(rename = "Acquire-By-Hash")]
142 pub acquire_by_hash: Option<AcquireByHash>,
144 #[serde(deserialize_with = "deserialize_architectures")]
146 pub architectures: Vec<Architecture>,
147 #[serde(deserialize_with = "deserialize_components")]
149 pub components: Vec<Component>,
150 pub description: String,
152 #[serde(rename = "SHA256", deserialize_with = "deserialize_sha256")]
154 pub files: HashMap<String, FileInfo>,
155}
156
157impl Release {
158 pub fn lookup_url(&self, file: &str) -> Option<String> {
160 let info = self.files.get(file)?;
161
162 if self
163 .acquire_by_hash
164 .is_some_and(|by_hash| by_hash == AcquireByHash::Yes)
165 {
166 file.rsplit_once('/').map(|(component, _)| {
167 format!("{}/by-hash/SHA256/{}", component, hex::encode(info.hash))
168 })
169 } else {
170 Some(file.to_string())
171 }
172 }
173}
174
175pub fn from_reader(reader: impl BufRead) -> Result<Release, rfc822_like::de::Error> {
177 rfc822_like::from_reader(reader)
178}
179
180pub fn from_str(data: &str) -> Result<Release, rfc822_like::de::Error> {
182 rfc822_like::from_str(data)
183}
184
185#[cfg(test)]
186mod test {
187 use super::*;
188
189 #[test]
190 fn archive() {
191 let data = r"Origin: Debian-ramacher.at
192Label: Debian-ramacher.at
193Suite: unstable
194Codename: sid
195Version: 13.0
196Date: Sun, 17 Dec 2023 18:43:37 UTC
197Architectures: i386 amd64
198Components: main
199Description: Experimental and unfinished Debian packages (for unstable)
200MD5Sum:
201 628a4efab35e598c7b6debdb0ac85314 26187 main/binary-i386/Packages
202 6c849211e65839aac2682c461c82dbb3 7777 main/binary-i386/Packages.gz
203 05ee2bfa660c3acc3559928769c29730 191 main/binary-i386/Release
204 d41d8cd98f00b204e9800998ecf8427e 0 main/debian-installer/binary-i386/Packages
205 7029066c27ac6f5ef18d660d5741979a 20 main/debian-installer/binary-i386/Packages.gz
206 296265926c83b0d9d9d43fcc6c43496d 30187 main/binary-amd64/Packages
207 8dad6d33daa175a4a54b9d328e9bb491 8821 main/binary-amd64/Packages.gz
208 c0f8f3dd5202483a2b57bb348a3741a6 192 main/binary-amd64/Release
209 d41d8cd98f00b204e9800998ecf8427e 0 main/debian-installer/binary-amd64/Packages
210 7029066c27ac6f5ef18d660d5741979a 20 main/debian-installer/binary-amd64/Packages.gz
211 4b35b2727e9c1d87c775e35fd8d00cf4 15130 main/source/Sources
212 689c40d665e43a8f9a94d6e2b1dd47a4 4582 main/source/Sources.gz
213 3ce12e6e384a34e6e1850bcc192edf8c 193 main/source/Release
214SHA1:
215 da7a5b4f20e79cab9bacca996d83419d5224a709 26187 main/binary-i386/Packages
216 a0b5ae4166358c741f1c27bf457c3b31bcdb495a 7777 main/binary-i386/Packages.gz
217 046a2ee510a7ea14c8b718dd153077b0359b3509 191 main/binary-i386/Release
218 da39a3ee5e6b4b0d3255bfef95601890afd80709 0 main/debian-installer/binary-i386/Packages
219 46c6643f07aa7f6bfe7118de926b86defc5087c4 20 main/debian-installer/binary-i386/Packages.gz
220 d7fc79844dbc2702ca889a985f716374f7c8b9a5 30187 main/binary-amd64/Packages
221 21374a60ce3d47b87bac11b3b3a96795020a0d41 8821 main/binary-amd64/Packages.gz
222 01f970b6eae435dd8b6b1f8f61727db854212ce4 192 main/binary-amd64/Release
223 da39a3ee5e6b4b0d3255bfef95601890afd80709 0 main/debian-installer/binary-amd64/Packages
224 46c6643f07aa7f6bfe7118de926b86defc5087c4 20 main/debian-installer/binary-amd64/Packages.gz
225 12b46a55c05518bfcfb267908185f041a1b984ae 15130 main/source/Sources
226 5e2bfa609cbc328e07336f8e17707683fda37011 4582 main/source/Sources.gz
227 96d0688be60481ba7eb71007b609bdf1f8323725 193 main/source/Release
228SHA256:
229 efe2dafdf6a50f376af1dfc574d6bd3360558fde917555671b13832c89604d9f 26187 main/binary-i386/Packages
230 ba66d22607be572323b72ca152d6e635fab075d92a2265bbfe319337c35ccd13 7777 main/binary-i386/Packages.gz
231 e6be53e3210056ed6854cf2a362cb953eaa962ea811cfbe34cdad2807be61101 191 main/binary-i386/Release
232 e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855 0 main/debian-installer/binary-i386/Packages
233 59869db34853933b239f1e2219cf7d431da006aa919635478511fabbfc8849d2 20 main/debian-installer/binary-i386/Packages.gz
234 baf930986b322ef7ff8cc04fa57762c68e7f9d8b67a0423bd5441686cbf3e751 30187 main/binary-amd64/Packages
235 0ad7ab0202ece24b57051f16010c72479b97e905c659f975eac5d69284c562f3 8821 main/binary-amd64/Packages.gz
236 97e06eefea86617e4abc8a647d0faebd0eaca7c87031423a4ae1d38e8f1c97bb 192 main/binary-amd64/Release
237 e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855 0 main/debian-installer/binary-amd64/Packages
238 59869db34853933b239f1e2219cf7d431da006aa919635478511fabbfc8849d2 20 main/debian-installer/binary-amd64/Packages.gz
239 b0a524d1ba90e253c937859e3ce30bc49a291e33dbb8124706424cf5c06100a8 15130 main/source/Sources
240 2bc04b364bfc30657836faf8d1de7f6044652bcca6af6503ef404a086897267a 4582 main/source/Sources.gz
241 3637559f78ac17d0e55bce465d510ef912d539e4b810a66b32431dd76f5929d8 193 main/source/Release";
242 let release = from_str(data).unwrap();
243
244 assert_eq!(
245 release.architectures,
246 vec![Architecture::I386, Architecture::Amd64]
247 );
248 assert_eq!(release.components, vec![Component::Main]);
249 assert_eq!(release.suite, Suite::Unstable);
250 assert_eq!(release.codename, Codename::Sid);
251 assert!(release.files.contains_key("main/source/Release"));
252 assert_eq!(
253 release.files["main/source/Release"],
254 FileInfo {
255 file_size: 193,
256 hash: [
257 0x36, 0x37, 0x55, 0x9f, 0x78, 0xac, 0x17, 0xd0, 0xe5, 0x5b, 0xce, 0x46, 0x5d,
258 0x51, 0x0e, 0xf9, 0x12, 0xd5, 0x39, 0xe4, 0xb8, 0x10, 0xa6, 0x6b, 0x32, 0x43,
259 0x1d, 0xd7, 0x6f, 0x59, 0x29, 0xd8
260 ]
261 }
262 );
263 }
264}