1use std::collections::{HashMap, HashSet};
2use std::error::Error;
3use std::fmt::{Display, Formatter};
4
5#[derive(Debug, Clone)]
6pub enum KnoloError {
7 InvalidPack(String),
8}
9
10impl Display for KnoloError {
11 fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
12 match self {
13 KnoloError::InvalidPack(msg) => write!(f, "invalid pack: {msg}"),
14 }
15 }
16}
17
18impl Error for KnoloError {}
19
20#[derive(Debug, Clone)]
21pub struct PackMeta {
22 pub version: u32,
23 pub stats: PackStats,
24}
25
26#[derive(Debug, Clone)]
27pub struct PackStats {
28 pub docs: usize,
29 pub blocks: usize,
30 pub terms: usize,
31 pub avg_block_len: Option<f64>,
32}
33
34#[derive(Debug, Clone)]
35pub struct Pack {
36 pub meta: PackMeta,
37 pub lexicon: HashMap<String, u32>,
38 pub postings: Vec<u32>,
39 pub blocks: Vec<String>,
40 pub headings: Vec<Option<String>>,
41 pub doc_ids: Vec<Option<String>>,
42 pub namespaces: Vec<Option<String>>,
43 pub block_token_lens: Vec<usize>,
44 pub metadata_json: String,
45 pub claims_json: Option<String>,
46}
47
48#[derive(Debug, Clone)]
49pub struct QueryOptions {
50 pub top_k: usize,
51 pub min_score: f64,
52 pub namespace: Option<Vec<String>>,
53 pub source: Option<Vec<String>>,
54}
55
56impl Default for QueryOptions {
57 fn default() -> Self {
58 Self {
59 top_k: 10,
60 min_score: 0.0,
61 namespace: None,
62 source: None,
63 }
64 }
65}
66
67#[derive(Debug, Clone, PartialEq)]
68pub struct Hit {
69 pub block_id: usize,
70 pub score: f64,
71 pub text: String,
72 pub source: Option<String>,
73 pub namespace: Option<String>,
74}
75
76pub fn mount_pack_from_bytes(bytes: &[u8]) -> Result<Pack, KnoloError> {
77 let mut cursor = 0usize;
78
79 let meta_len = read_u32(bytes, &mut cursor)? as usize;
80 let meta_json = read_slice(bytes, &mut cursor, meta_len)?;
81 let metadata_json = std::str::from_utf8(meta_json).map_err(|_| KnoloError::InvalidPack("meta utf8".into()))?.to_string();
82 let meta = parse_meta(&metadata_json)?;
83
84 let lex_len = read_u32(bytes, &mut cursor)? as usize;
85 let lex_json = read_slice(bytes, &mut cursor, lex_len)?;
86 let lexicon = parse_lexicon(std::str::from_utf8(lex_json).map_err(|_| KnoloError::InvalidPack("lexicon utf8".into()))?)?;
87
88 let post_count = read_u32(bytes, &mut cursor)? as usize;
89 let postings = read_u32_array(bytes, &mut cursor, post_count)?;
90
91 let blocks_len = read_u32(bytes, &mut cursor)? as usize;
92 let blocks_json = read_slice(bytes, &mut cursor, blocks_len)?;
93 let blocks_str = std::str::from_utf8(blocks_json).map_err(|_| KnoloError::InvalidPack("blocks utf8".into()))?;
94 let parsed_blocks = parse_blocks(blocks_str)?;
95
96 Ok(Pack {
97 meta,
98 lexicon,
99 postings,
100 blocks: parsed_blocks.texts,
101 headings: parsed_blocks.headings,
102 doc_ids: parsed_blocks.doc_ids,
103 namespaces: parsed_blocks.namespaces,
104 block_token_lens: parsed_blocks.lens,
105 metadata_json,
106 claims_json: None,
107 })
108}
109
110pub fn query(pack: &Pack, q: &str, opts: QueryOptions) -> Vec<Hit> {
111 if q.trim().is_empty() {
112 return vec![];
113 }
114 let tokens = tokenize(q);
115 if tokens.is_empty() {
116 return vec![];
117 }
118
119 let term_ids = tokens
120 .iter()
121 .filter_map(|t| pack.lexicon.get(t).copied())
122 .collect::<HashSet<_>>();
123 if term_ids.is_empty() {
124 return vec![];
125 }
126
127 let namespace_filter = normalize_filter(opts.namespace.as_ref());
128 let source_filter = normalize_filter(opts.source.as_ref());
129
130 let mut candidates: HashMap<usize, HashMap<u32, f64>> = HashMap::new();
131 let mut dfs: HashMap<u32, usize> = HashMap::new();
132 let uses_offset_block_ids = pack.meta.version >= 3;
133
134 let mut i = 0usize;
135 while i < pack.postings.len() {
136 let tid = pack.postings[i];
137 i += 1;
138 if tid == 0 {
139 continue;
140 }
141 let relevant = term_ids.contains(&tid);
142 let mut term_df = 0usize;
143
144 if i >= pack.postings.len() { break; }
145 let mut encoded_bid = pack.postings[i];
146 i += 1;
147
148 while encoded_bid != 0 && i < pack.postings.len() {
149 let bid = if uses_offset_block_ids {
150 encoded_bid.saturating_sub(1) as usize
151 } else {
152 encoded_bid as usize
153 };
154
155 let mut tf = 0usize;
156 while i < pack.postings.len() {
157 let pos = pack.postings[i];
158 i += 1;
159 if pos == 0 {
160 break;
161 }
162 tf += 1;
163 }
164
165 term_df += 1;
166 if relevant && bid < pack.blocks.len() {
167 let entry = candidates.entry(bid).or_default();
168 *entry.entry(tid).or_insert(0.0) += tf as f64;
169 }
170
171 if i >= pack.postings.len() { break; }
172 encoded_bid = pack.postings[i];
173 i += 1;
174 }
175
176 if relevant {
177 dfs.insert(tid, term_df);
178 }
179 }
180
181 if !namespace_filter.is_empty() {
182 candidates.retain(|bid, _| {
183 pack.namespaces
184 .get(*bid)
185 .and_then(|n| n.clone())
186 .map(|n| namespace_filter.contains(&normalize(&n)))
187 .unwrap_or(false)
188 });
189 }
190
191 if !source_filter.is_empty() {
192 candidates.retain(|bid, _| {
193 pack.doc_ids
194 .get(*bid)
195 .and_then(|n| n.clone())
196 .map(|n| source_filter.contains(&normalize(&n)))
197 .unwrap_or(false)
198 });
199 }
200
201 let doc_count = pack.meta.stats.blocks.max(1) as f64;
202 let avg_len = pack
203 .meta
204 .stats
205 .avg_block_len
206 .unwrap_or_else(|| {
207 if pack.block_token_lens.is_empty() {
208 1.0
209 } else {
210 pack.block_token_lens.iter().sum::<usize>() as f64 / pack.block_token_lens.len() as f64
211 }
212 })
213 .max(1.0);
214
215 let mut scored = candidates
216 .into_iter()
217 .map(|(bid, tf_map)| {
218 let mut score = 0.0;
219 let len = *pack.block_token_lens.get(bid).unwrap_or(&1) as f64;
220 for (tid, tf) in tf_map {
221 let df = *dfs.get(&tid).unwrap_or(&0) as f64;
222 let idf = (1.0 + (doc_count - df + 0.5) / (df + 0.5)).ln();
223 let k1 = 1.5;
224 let b = 0.75;
225 let numer = tf * (k1 + 1.0);
226 let denom = tf + k1 * (1.0 - b + b * (len / avg_len));
227 score += idf * (numer / denom);
228 }
229 (bid, score)
230 })
231 .filter(|(_, score)| *score >= opts.min_score)
232 .collect::<Vec<_>>();
233
234 scored.sort_by(|a, b| b.1.partial_cmp(&a.1).unwrap_or(std::cmp::Ordering::Equal));
235
236 scored
237 .into_iter()
238 .take(opts.top_k.max(1))
239 .map(|(bid, score)| Hit {
240 block_id: bid,
241 score,
242 text: pack.blocks.get(bid).cloned().unwrap_or_default(),
243 source: pack.doc_ids.get(bid).and_then(|s| s.clone()),
244 namespace: pack.namespaces.get(bid).and_then(|s| s.clone()),
245 })
246 .collect()
247}
248
249struct ParsedBlocks {
250 texts: Vec<String>,
251 headings: Vec<Option<String>>,
252 doc_ids: Vec<Option<String>>,
253 namespaces: Vec<Option<String>>,
254 lens: Vec<usize>,
255}
256
257fn parse_meta(json: &str) -> Result<PackMeta, KnoloError> {
258 Ok(PackMeta {
259 version: parse_u32_field(json, "version")?,
260 stats: PackStats {
261 docs: parse_u32_field(json, "docs")? as usize,
262 blocks: parse_u32_field(json, "blocks")? as usize,
263 terms: parse_u32_field(json, "terms")? as usize,
264 avg_block_len: parse_f64_field(json, "avgBlockLen"),
265 },
266 })
267}
268
269fn parse_lexicon(json: &str) -> Result<HashMap<String, u32>, KnoloError> {
270 let mut map = HashMap::new();
271 let s = compact(json);
272 let mut i = 0usize;
273 while let Some(start) = s[i..].find("[\"") {
274 let abs = i + start + 2;
275 let rest = &s[abs..];
276 let end = rest.find('"').ok_or_else(|| KnoloError::InvalidPack("lexicon key".into()))?;
277 let key = rest[..end].to_string();
278 let rest2 = &rest[end + 1..];
279 let comma = rest2.find(',').ok_or_else(|| KnoloError::InvalidPack("lexicon comma".into()))?;
280 let rest3 = &rest2[comma + 1..];
281 let mut n = String::new();
282 for ch in rest3.chars() {
283 if ch.is_ascii_digit() {
284 n.push(ch);
285 } else {
286 break;
287 }
288 }
289 if !n.is_empty() {
290 map.insert(key, n.parse::<u32>().map_err(|_| KnoloError::InvalidPack("lexicon tid".into()))?);
291 }
292 i = abs + end + 1;
293 }
294 Ok(map)
295}
296
297fn parse_blocks(json: &str) -> Result<ParsedBlocks, KnoloError> {
298 let s = compact(json);
299 if s.starts_with("[\"") {
300 let mut texts = Vec::new();
301 let mut i = 2usize;
302 while i < s.len() {
303 if let Some(end) = s[i..].find('"') {
304 let piece = &s[i..i + end];
305 texts.push(unescape(piece));
306 i += end + 1;
307 if let Some(next) = s[i..].find('"') {
308 i += next + 1;
309 } else {
310 break;
311 }
312 } else {
313 break;
314 }
315 }
316 let lens = texts.iter().map(|t| tokenize(t).len()).collect::<Vec<_>>();
317 return Ok(ParsedBlocks {
318 headings: vec![None; texts.len()],
319 doc_ids: vec![None; texts.len()],
320 namespaces: vec![None; texts.len()],
321 lens,
322 texts,
323 });
324 }
325
326 let objects = split_top_level_objects(&s)?;
327 let mut texts = Vec::new();
328 let mut headings = Vec::new();
329 let mut doc_ids = Vec::new();
330 let mut namespaces = Vec::new();
331 let mut lens = Vec::new();
332
333 for obj in objects {
334 let text = parse_string_or_null(&obj, "text").unwrap_or_default();
335 let len = parse_u32_field_optional(&obj, "len").map(|v| v as usize).unwrap_or_else(|| tokenize(&text).len());
336 texts.push(text);
337 headings.push(parse_string_or_null(&obj, "heading"));
338 doc_ids.push(parse_string_or_null(&obj, "docId"));
339 namespaces.push(parse_string_or_null(&obj, "namespace"));
340 lens.push(len);
341 }
342
343 Ok(ParsedBlocks { texts, headings, doc_ids, namespaces, lens })
344}
345
346fn split_top_level_objects(s: &str) -> Result<Vec<String>, KnoloError> {
347 let mut out = Vec::new();
348 let mut depth = 0i32;
349 let mut start = None;
350 let chars: Vec<char> = s.chars().collect();
351 for (i, ch) in chars.iter().enumerate() {
352 if *ch == '{' {
353 if depth == 0 {
354 start = Some(i);
355 }
356 depth += 1;
357 } else if *ch == '}' {
358 depth -= 1;
359 if depth == 0 {
360 if let Some(st) = start {
361 out.push(chars[st..=i].iter().collect());
362 }
363 start = None;
364 }
365 }
366 }
367 if out.is_empty() {
368 return Err(KnoloError::InvalidPack("blocks objects".into()));
369 }
370 Ok(out)
371}
372
373fn parse_string_or_null(obj: &str, key: &str) -> Option<String> {
374 let needle = format!("\"{}\":", key);
375 let idx = obj.find(&needle)? + needle.len();
376 let tail = &obj[idx..];
377 if tail.starts_with("null") {
378 return None;
379 }
380 if !tail.starts_with('"') {
381 return None;
382 }
383 let rest = &tail[1..];
384 let end = rest.find('"')?;
385 Some(unescape(&rest[..end]))
386}
387
388fn parse_u32_field(json: &str, key: &str) -> Result<u32, KnoloError> {
389 parse_u32_field_optional(json, key).ok_or_else(|| KnoloError::InvalidPack(format!("missing {key}")))
390}
391
392fn parse_u32_field_optional(json: &str, key: &str) -> Option<u32> {
393 let needle = format!("\"{}\":", key);
394 let idx = json.find(&needle)? + needle.len();
395 let tail = &json[idx..];
396 let mut n = String::new();
397 for ch in tail.chars() {
398 if ch.is_ascii_digit() {
399 n.push(ch);
400 } else if !n.is_empty() {
401 break;
402 }
403 }
404 n.parse().ok()
405}
406
407fn parse_f64_field(json: &str, key: &str) -> Option<f64> {
408 let needle = format!("\"{}\":", key);
409 let idx = json.find(&needle)? + needle.len();
410 let tail = &json[idx..];
411 let mut n = String::new();
412 for ch in tail.chars() {
413 if ch.is_ascii_digit() || ch == '.' {
414 n.push(ch);
415 } else if !n.is_empty() {
416 break;
417 }
418 }
419 n.parse().ok()
420}
421
422fn normalize_filter(values: Option<&Vec<String>>) -> HashSet<String> {
423 values
424 .map(|arr| arr.iter().map(|s| normalize(s)).collect::<HashSet<_>>())
425 .unwrap_or_default()
426}
427
428fn normalize(s: &str) -> String {
429 s.to_lowercase().trim().to_string()
430}
431
432fn tokenize(text: &str) -> Vec<String> {
433 let mut out = Vec::new();
434 let mut cur = String::new();
435 for ch in text.chars() {
436 if ch.is_alphanumeric() {
437 cur.push(ch.to_ascii_lowercase());
438 } else if !cur.is_empty() {
439 out.push(std::mem::take(&mut cur));
440 }
441 }
442 if !cur.is_empty() {
443 out.push(cur);
444 }
445 out
446}
447
448fn compact(s: &str) -> String {
449 let mut out = String::with_capacity(s.len());
450 let mut in_string = false;
451 let mut escaped = false;
452
453 for ch in s.chars() {
454 if in_string {
455 out.push(ch);
456 if escaped {
457 escaped = false;
458 } else if ch == '\\' {
459 escaped = true;
460 } else if ch == '"' {
461 in_string = false;
462 }
463 continue;
464 }
465
466 if ch.is_whitespace() {
467 continue;
468 }
469
470 out.push(ch);
471 if ch == '"' {
472 in_string = true;
473 }
474 }
475
476 out
477}
478
479fn unescape(s: &str) -> String {
480 s.replace("\\\"", "\"")
481}
482
483fn read_u32(bytes: &[u8], cursor: &mut usize) -> Result<u32, KnoloError> {
484 let chunk = read_slice(bytes, cursor, 4)?;
485 Ok(u32::from_le_bytes([chunk[0], chunk[1], chunk[2], chunk[3]]))
486}
487
488fn read_u32_array(bytes: &[u8], cursor: &mut usize, len: usize) -> Result<Vec<u32>, KnoloError> {
489 let mut out = Vec::with_capacity(len);
490 for _ in 0..len {
491 out.push(read_u32(bytes, cursor)?);
492 }
493 Ok(out)
494}
495
496fn read_slice<'a>(bytes: &'a [u8], cursor: &mut usize, len: usize) -> Result<&'a [u8], KnoloError> {
497 let end = cursor.saturating_add(len);
498 if end > bytes.len() {
499 return Err(KnoloError::InvalidPack("unexpected end-of-buffer".into()));
500 }
501 let slice = &bytes[*cursor..end];
502 *cursor = end;
503 Ok(slice)
504}
505
506pub const KNOWLEDGE_IMAGE_V5_MAGIC: &[u8; 8] = b"KNLOV5\0\0";
511const SUPERBLOCK_MAGIC: &[u8; 8] = b"KNLOSB1\0";
512const SEGMENT_MAGIC: &[u8; 4] = b"KSEG";
513const V5_HEADER_SIZE: usize = 16;
514const V5_SUPERBLOCK_SIZE: usize = 128;
515const V5_SEGMENT_HEADER_SIZE: usize = 48;
516const V5_DATA_START: usize = V5_HEADER_SIZE + V5_SUPERBLOCK_SIZE * 2;
517const V5_MAX_SEGMENT: u64 = 512 * 1024 * 1024;
518
519#[derive(Debug, Clone, PartialEq)]
520pub struct KnowledgeObjectV1 {
521 pub id: String,
522 pub kind: String,
523 pub bytes: Vec<u8>,
524 pub meta: CborValue,
525}
526
527#[derive(Debug, Clone, PartialEq)]
528pub struct KnowledgeEventV1 {
529 pub id: String,
530 pub transaction_id: String,
531 pub actor: String,
532 pub actor_counter: u64,
533 pub kind: String,
534 pub target: String,
535 pub payload: String,
536}
537
538#[derive(Debug, Clone, PartialEq)]
539pub struct KnowledgeCommitV1 {
540 pub state_root: String,
541 pub commit_digest: String,
542 pub parents: Vec<String>,
543 pub object_root: String,
544 pub event_root: String,
545 pub policy_root: String,
546 pub sequence: u64,
547 pub actor: String,
548}
549
550#[derive(Debug, Clone, PartialEq)]
551pub struct KnowledgeImageSegment {
552 pub kind: u8,
553 pub schema: u8,
554 pub flags: u16,
555 pub offset: usize,
556 pub length: usize,
557 pub payload_length: usize,
558 pub digest: String,
559}
560
561#[derive(Debug, Clone, PartialEq)]
562pub struct KnowledgeImage {
563 pub state_root: String,
564 pub commit_digest: String,
565 pub commit: KnowledgeCommitV1,
566 pub objects: Vec<KnowledgeObjectV1>,
567 pub events: Vec<KnowledgeEventV1>,
568 pub segments: Vec<KnowledgeImageSegment>,
569 pub active_superblock: char,
570}
571
572#[derive(Debug, Clone, PartialEq)]
573pub struct KnowledgeImageVerification {
574 pub valid: bool,
575 pub state_root: String,
576 pub commit_digest: String,
577 pub active_superblock: char,
578 pub segments: Vec<KnowledgeImageSegment>,
579}
580
581#[derive(Debug, Clone, PartialEq)]
582pub struct KnowledgeRuntimeDiagnosticsImageV1 {
583 pub state_root: String,
584 pub commit_digest: String,
585 pub sequence: u64,
586 pub object_count: usize,
587 pub event_count: usize,
588 pub segment_count: usize,
589 pub active_superblock: char,
590}
591
592#[derive(Debug, Clone, PartialEq)]
593pub struct KnowledgeRuntimeDiagnosticsV1 {
594 pub version: u64,
595 pub valid: bool,
596 pub image: KnowledgeRuntimeDiagnosticsImageV1,
597 pub diagnostics_root: String,
598}
599
600#[derive(Debug, Clone, PartialEq)]
601pub struct KnowledgeStudioCapabilitiesV1 {
602 pub inspect_image: bool,
603 pub verify_image: bool,
604 pub inspect_query_index: bool,
605 pub inspect_query_history: bool,
606 pub inspect_run: bool,
607 pub inspect_replay: bool,
608 pub mutate_image: bool,
609}
610
611#[derive(Debug, Clone, PartialEq)]
612pub struct KnowledgeStudioManagementV1 {
613 pub version: u64,
614 pub surface: String,
615 pub valid: bool,
616 pub read_only: bool,
617 pub diagnostics: KnowledgeRuntimeDiagnosticsV1,
618 pub capabilities: KnowledgeStudioCapabilitiesV1,
619 pub management_root: String,
620}
621
622#[derive(Debug, Clone, PartialEq)]
623pub struct MigrationMapping {
624 pub legacy_block_id: usize,
625 pub source_object: String,
626 pub chunk_object: String,
627}
628
629#[derive(Debug, Clone, PartialEq)]
630pub struct MigrationResult {
631 pub image: Vec<u8>,
632 pub receipt: Vec<u8>,
633 pub state_root: String,
634 pub mappings: Vec<MigrationMapping>,
635}
636
637#[derive(Debug, Clone, PartialEq)]
638pub enum CborValue {
639 Null,
640 Bool(bool),
641 UInt(u64),
642 NInt(i64),
643 Bytes(Vec<u8>),
644 Text(String),
645 Array(Vec<CborValue>),
646 Map(Vec<(String, CborValue)>),
647}
648
649pub fn mount_knowledge_image(bytes: &[u8]) -> Result<KnowledgeImage, KnoloError> {
650 let parsed = parse_v5_image(bytes)?;
651 Ok(parsed)
652}
653
654pub fn inspect_knowledge_image(bytes: &[u8]) -> Result<KnowledgeImageVerification, KnoloError> {
655 let image = mount_knowledge_image(bytes)?;
656 Ok(KnowledgeImageVerification {
657 valid: true,
658 state_root: image.state_root,
659 commit_digest: image.commit_digest,
660 active_superblock: image.active_superblock,
661 segments: image.segments,
662 })
663}
664
665pub fn verify_knowledge_image(bytes: &[u8]) -> Result<KnowledgeImageVerification, KnoloError> {
666 inspect_knowledge_image(bytes)
667}
668
669pub fn inspect_knowledge_runtime_v5(bytes: &[u8]) -> Result<KnowledgeRuntimeDiagnosticsV1, KnoloError> {
670 let image = mount_knowledge_image(bytes)?;
671 let diagnostics = KnowledgeRuntimeDiagnosticsV1 {
672 version: 1,
673 valid: true,
674 image: KnowledgeRuntimeDiagnosticsImageV1 {
675 state_root: image.state_root,
676 commit_digest: image.commit_digest,
677 sequence: image.commit.sequence,
678 object_count: image.objects.len(),
679 event_count: image.events.len(),
680 segment_count: image.segments.len(),
681 active_superblock: image.active_superblock,
682 },
683 diagnostics_root: String::new(),
684 };
685 let diagnostics_root = runtime_diagnostics_root_v5(&diagnostics);
686 Ok(KnowledgeRuntimeDiagnosticsV1 { diagnostics_root, ..diagnostics })
687}
688
689pub fn runtime_diagnostics_root_v5(diagnostics: &KnowledgeRuntimeDiagnosticsV1) -> String {
690 digest_domain("runtime-diagnostics", &encode_cbor(&runtime_diagnostics_body(diagnostics)))
691}
692
693pub fn inspect_knowledge_studio_management_v5(bytes: &[u8]) -> Result<KnowledgeStudioManagementV1, KnoloError> {
694 let diagnostics = inspect_knowledge_runtime_v5(bytes)?;
695 let management = KnowledgeStudioManagementV1 {
696 version: 1,
697 surface: "studio-management".into(),
698 valid: true,
699 read_only: true,
700 diagnostics,
701 capabilities: KnowledgeStudioCapabilitiesV1 {
702 inspect_image: true,
703 verify_image: true,
704 inspect_query_index: false,
705 inspect_query_history: false,
706 inspect_run: false,
707 inspect_replay: false,
708 mutate_image: false,
709 },
710 management_root: String::new(),
711 };
712 let management_root = studio_management_root_v5(&management);
713 Ok(KnowledgeStudioManagementV1 { management_root, ..management })
714}
715
716pub fn studio_management_root_v5(management: &KnowledgeStudioManagementV1) -> String {
717 digest_domain("studio-management", &encode_cbor(&CborValue::Map(vec![
718 ("capabilities".into(), studio_capabilities_value(&management.capabilities)),
719 ("diagnostics".into(), runtime_diagnostics_value(&management.diagnostics)),
720 ("readOnly".into(), CborValue::Bool(management.read_only)),
721 ("surface".into(), CborValue::Text(management.surface.clone())),
722 ("valid".into(), CborValue::Bool(management.valid)),
723 ("version".into(), CborValue::UInt(management.version)),
724 ])))
725}
726
727fn runtime_diagnostics_body(diagnostics: &KnowledgeRuntimeDiagnosticsV1) -> CborValue {
728 CborValue::Map(vec![
729 ("image".into(), CborValue::Map(vec![
730 ("activeSuperblock".into(), CborValue::Text(diagnostics.image.active_superblock.to_string())),
731 ("commitDigest".into(), CborValue::Text(diagnostics.image.commit_digest.clone())),
732 ("eventCount".into(), CborValue::UInt(diagnostics.image.event_count as u64)),
733 ("objectCount".into(), CborValue::UInt(diagnostics.image.object_count as u64)),
734 ("segmentCount".into(), CborValue::UInt(diagnostics.image.segment_count as u64)),
735 ("sequence".into(), CborValue::UInt(diagnostics.image.sequence)),
736 ("stateRoot".into(), CborValue::Text(diagnostics.image.state_root.clone())),
737 ])),
738 ("valid".into(), CborValue::Bool(diagnostics.valid)),
739 ("version".into(), CborValue::UInt(diagnostics.version)),
740 ])
741}
742
743fn runtime_diagnostics_value(diagnostics: &KnowledgeRuntimeDiagnosticsV1) -> CborValue {
744 let mut entries = match runtime_diagnostics_body(diagnostics) {
745 CborValue::Map(entries) => entries,
746 _ => Vec::new(),
747 };
748 entries.push(("diagnosticsRoot".into(), CborValue::Text(diagnostics.diagnostics_root.clone())));
749 CborValue::Map(entries)
750}
751
752fn studio_capabilities_value(capabilities: &KnowledgeStudioCapabilitiesV1) -> CborValue {
753 CborValue::Map(vec![
754 ("inspectImage".into(), CborValue::Bool(capabilities.inspect_image)),
755 ("inspectQueryHistory".into(), CborValue::Bool(capabilities.inspect_query_history)),
756 ("inspectQueryIndex".into(), CborValue::Bool(capabilities.inspect_query_index)),
757 ("inspectReplay".into(), CborValue::Bool(capabilities.inspect_replay)),
758 ("inspectRun".into(), CborValue::Bool(capabilities.inspect_run)),
759 ("mutateImage".into(), CborValue::Bool(capabilities.mutate_image)),
760 ("verifyImage".into(), CborValue::Bool(capabilities.verify_image)),
761 ])
762}
763
764pub fn state_root(image: &KnowledgeImage) -> &str {
765 &image.state_root
766}
767
768#[derive(Debug, Clone, PartialEq)]
769pub struct KnowledgeQueryFilterV1 {
770 pub field: String,
771 pub value: CborValue,
772}
773
774#[derive(Debug, Clone, PartialEq)]
775pub struct KnowledgeQueryPlanV1 {
776 pub kind: Option<String>,
777 pub filters: Vec<KnowledgeQueryFilterV1>,
778 pub search: Option<String>,
779 pub limit: usize,
780}
781
782#[derive(Debug, Clone, PartialEq)]
783pub struct KnowledgeQueryHitV1 {
784 pub object_id: String,
785 pub kind: String,
786}
787
788#[derive(Debug, Clone, PartialEq)]
789pub struct KnowledgeQueryResultV1 {
790 pub state_root: String,
791 pub plan_root: String,
792 pub hits: Vec<KnowledgeQueryHitV1>,
793 pub result_root: String,
794}
795
796#[derive(Debug, Clone, PartialEq)]
797pub struct KnowledgePolicyRuleV1 {
798 pub effect: String,
799 pub action: String,
800 pub principal: Option<String>,
801 pub kind: Option<String>,
802}
803
804#[derive(Debug, Clone, PartialEq)]
805pub struct KnowledgePolicyV1 {
806 pub default: String,
807 pub rules: Vec<KnowledgePolicyRuleV1>,
808}
809
810#[derive(Debug, Clone, PartialEq)]
811pub struct KnowledgeAuthorizationResultV1 {
812 pub state_root: String,
813 pub plan_root: String,
814 pub policy_root: String,
815 pub authorization_root: String,
816 pub principal: String,
817 pub action: String,
818 pub decision: String,
819 pub allowed_object_ids: Vec<String>,
820 pub denied_object_ids: Vec<String>,
821}
822
823#[derive(Debug, Clone, PartialEq)]
824pub struct KnowledgeDelegationV1 {
825 pub version: u64,
826 pub delegator: String,
827 pub delegatee: String,
828 pub action: String,
829 pub issued_at: u64,
830 pub expires_at: u64,
831 pub algorithm: String,
832 pub key_id: Option<String>,
833 pub signature: Vec<u8>,
834}
835
836#[derive(Debug, Clone, PartialEq)]
837pub struct KnowledgeAuthorityEnvelopeV1 {
838 pub version: u64,
839 pub issuer: String,
840 pub subject: String,
841 pub authorization_root: String,
842 pub keyring_root: Option<String>,
843 pub issued_at: u64,
844 pub expires_at: u64,
845 pub algorithm: String,
846 pub key_id: Option<String>,
847 pub delegations: Vec<KnowledgeDelegationV1>,
848 pub signature: Vec<u8>,
849}
850
851#[derive(Debug, Clone, PartialEq)]
852pub struct KnowledgeAuthorityVerificationV1 {
853 pub envelope_root: String,
854 pub issuer: String,
855 pub subject: String,
856 pub authorization_root: String,
857 pub keyring_root: Option<String>,
858 pub delegation_depth: usize,
859}
860
861pub fn knowledge_policy_root_v5(policy: &KnowledgePolicyV1) -> Result<String, KnoloError> {
862 if policy.default != "allow" && policy.default != "deny" { return Err(KnoloError::InvalidPack("invalid V5 policy default".into())); }
863 let mut rules = policy.rules.clone();
864 for rule in &rules {
865 if !matches!(rule.effect.as_str(), "allow" | "deny") || !matches!(rule.action.as_str(), "query" | "read") { return Err(KnoloError::InvalidPack("invalid V5 policy rule".into())); }
866 if rule.principal.as_deref() == Some("") || rule.kind.as_deref() == Some("") { return Err(KnoloError::InvalidPack("invalid V5 policy selector".into())); }
867 }
868 rules.sort_by(|left, right| policy_rule_key(left).cmp(&policy_rule_key(right)));
869 let body = if rules.is_empty() {
870 CborValue::Map(vec![("default".into(), CborValue::Text(policy.default.clone()))])
871 } else {
872 CborValue::Map(vec![
873 ("default".into(), CborValue::Text(policy.default.clone())),
874 ("rules".into(), CborValue::Array(rules.iter().map(policy_rule_value).collect())),
875 ])
876 };
877 Ok(digest_domain("policy", &encode_cbor(&body)))
878}
879
880pub fn evaluate_knowledge_query_policy_v5(
881 image: &KnowledgeImage,
882 query_result: &KnowledgeQueryResultV1,
883 policy: &KnowledgePolicyV1,
884 principal: &str,
885 action: &str,
886) -> Result<KnowledgeAuthorizationResultV1, KnoloError> {
887 if principal.is_empty() || !matches!(action, "query" | "read") { return Err(KnoloError::InvalidPack("invalid V5 authorization input".into())); }
888 if query_result.state_root != image.state_root { return Err(KnoloError::InvalidPack("V5 query state root mismatch".into())); }
889 let policy_root = knowledge_policy_root_v5(policy)?;
890 if policy_root != image.commit.policy_root { return Err(KnoloError::InvalidPack("V5 policy root mismatch".into())); }
891 let mut allowed_object_ids = Vec::new();
892 let mut denied_object_ids = Vec::new();
893 for hit in &query_result.hits {
894 if authorize_policy_hit(hit, policy, principal, action) { allowed_object_ids.push(hit.object_id.clone()); } else { denied_object_ids.push(hit.object_id.clone()); }
895 }
896 let decision = if denied_object_ids.is_empty() { "allow" } else if allowed_object_ids.is_empty() { "deny" } else { "partial" };
897 let authorization_value = CborValue::Map(vec![
898 ("action".into(), CborValue::Text(action.into())),
899 ("allowedObjectIds".into(), CborValue::Array(allowed_object_ids.iter().map(|id| CborValue::Text(id.clone())).collect())),
900 ("decision".into(), CborValue::Text(decision.into())),
901 ("deniedObjectIds".into(), CborValue::Array(denied_object_ids.iter().map(|id| CborValue::Text(id.clone())).collect())),
902 ("planRoot".into(), CborValue::Text(query_result.plan_root.clone())),
903 ("policyRoot".into(), CborValue::Text(policy_root.clone())),
904 ("principal".into(), CborValue::Text(principal.into())),
905 ("stateRoot".into(), CborValue::Text(image.state_root.clone())),
906 ]);
907 let authorization_root = digest_domain("authorization", &encode_cbor(&authorization_value));
908 Ok(KnowledgeAuthorizationResultV1 { state_root: image.state_root.clone(), plan_root: query_result.plan_root.clone(), policy_root, authorization_root, principal: principal.into(), action: action.into(), decision: decision.into(), allowed_object_ids, denied_object_ids })
909}
910
911pub fn delegation_payload_v1(delegation: &KnowledgeDelegationV1) -> Vec<u8> {
912 let mut entries = vec![
913 ("action".into(), CborValue::Text(delegation.action.clone())),
914 ("algorithm".into(), CborValue::Text(delegation.algorithm.clone())),
915 ("delegatee".into(), CborValue::Text(delegation.delegatee.clone())),
916 ("delegator".into(), CborValue::Text(delegation.delegator.clone())),
917 ("expiresAt".into(), CborValue::UInt(delegation.expires_at)),
918 ("issuedAt".into(), CborValue::UInt(delegation.issued_at)),
919 ("version".into(), CborValue::UInt(delegation.version)),
920 ];
921 if let Some(key_id) = &delegation.key_id { entries.push(("keyId".into(), CborValue::Text(key_id.clone()))); }
922 encode_cbor(&CborValue::Map(entries))
923}
924
925pub fn authority_envelope_payload_v1(envelope: &KnowledgeAuthorityEnvelopeV1) -> Vec<u8> {
926 let delegation_roots = envelope.delegations.iter().map(|delegation| digest_domain("delegation-payload", &delegation_payload_v1(delegation))).map(CborValue::Text).collect();
927 let mut entries = vec![
928 ("algorithm".into(), CborValue::Text(envelope.algorithm.clone())),
929 ("authorizationRoot".into(), CborValue::Text(envelope.authorization_root.clone())),
930 ("delegations".into(), CborValue::Array(delegation_roots)),
931 ("expiresAt".into(), CborValue::UInt(envelope.expires_at)),
932 ("issuedAt".into(), CborValue::UInt(envelope.issued_at)),
933 ("issuer".into(), CborValue::Text(envelope.issuer.clone())),
934 ("subject".into(), CborValue::Text(envelope.subject.clone())),
935 ("version".into(), CborValue::UInt(envelope.version)),
936 ];
937 if let Some(keyring_root) = &envelope.keyring_root { entries.push(("keyringRoot".into(), CborValue::Text(keyring_root.clone()))); }
938 if let Some(key_id) = &envelope.key_id { entries.push(("keyId".into(), CborValue::Text(key_id.clone()))); }
939 encode_cbor(&CborValue::Map(entries))
940}
941
942pub fn delegation_root_v1(delegation: &KnowledgeDelegationV1) -> String {
943 digest_domain("delegation", &encode_cbor(&CborValue::Map(vec![
944 ("payload".into(), CborValue::Bytes(delegation_payload_v1(delegation))),
945 ("signature".into(), CborValue::Bytes(delegation.signature.clone())),
946 ])))
947}
948
949pub fn authority_envelope_root_v1(envelope: &KnowledgeAuthorityEnvelopeV1) -> String {
950 digest_domain("authority-envelope", &encode_cbor(&CborValue::Map(vec![
951 ("payload".into(), CborValue::Bytes(authority_envelope_payload_v1(envelope))),
952 ("signature".into(), CborValue::Bytes(envelope.signature.clone())),
953 ])))
954}
955
956pub fn authority_session_root_v1(state_root: &str, plan_root: &str, result_root: &str, authorization_root: &str, envelope_root: &str, keyring_root: Option<&str>) -> String {
957 let mut entries = vec![
958 ("authorizationRoot".into(), CborValue::Text(authorization_root.into())),
959 ("envelopeRoot".into(), CborValue::Text(envelope_root.into())),
960 ("keyringRoot".into(), keyring_root.map_or(CborValue::Null, |value| CborValue::Text(value.into()))),
961 ("planRoot".into(), CborValue::Text(plan_root.into())),
962 ("resultRoot".into(), CborValue::Text(result_root.into())),
963 ("stateRoot".into(), CborValue::Text(state_root.into())),
964 ("version".into(), CborValue::UInt(1)),
965 ];
966 digest_domain("authority-session", &encode_cbor(&CborValue::Map(std::mem::take(&mut entries))))
967}
968
969pub fn sync_request_payload_v1(request_id: &str, sender: &str, summary_root: &str, want_object_ids: &[String], want_event_ids: &[String], algorithm: &str, key_id: Option<&str>, keyring_root: Option<&str>, nonce: &[u8], issued_at: u64, expires_at: u64) -> Vec<u8> {
970 encode_cbor(&CborValue::Map(vec![
971 ("algorithm".into(), CborValue::Text(algorithm.into())),
972 ("expiresAt".into(), CborValue::UInt(expires_at)),
973 ("issuedAt".into(), CborValue::UInt(issued_at)),
974 ("keyId".into(), key_id.map_or(CborValue::Null, |value| CborValue::Text(value.into()))),
975 ("keyringRoot".into(), keyring_root.map_or(CborValue::Null, |value| CborValue::Text(value.into()))),
976 ("kind".into(), CborValue::Text("sync-request".into())),
977 ("nonce".into(), CborValue::Bytes(nonce.to_vec())),
978 ("requestId".into(), CborValue::Text(request_id.into())),
979 ("sender".into(), CborValue::Text(sender.into())),
980 ("summary".into(), CborValue::Text(summary_root.into())),
981 ("version".into(), CborValue::UInt(1)),
982 ("wantEventIds".into(), CborValue::Array(want_event_ids.iter().map(|value| CborValue::Text(value.clone())).collect())),
983 ("wantObjectIds".into(), CborValue::Array(want_object_ids.iter().map(|value| CborValue::Text(value.clone())).collect())),
984 ]))
985}
986
987pub fn sync_request_root_v1(request_id: &str, sender: &str, summary_root: &str, want_object_ids: &[String], want_event_ids: &[String], algorithm: &str, key_id: Option<&str>, keyring_root: Option<&str>, nonce: &[u8], issued_at: u64, expires_at: u64, signature: &[u8]) -> String {
988 digest_domain("sync-request", &encode_cbor(&CborValue::Map(vec![
989 ("payload".into(), CborValue::Bytes(sync_request_payload_v1(request_id, sender, summary_root, want_object_ids, want_event_ids, algorithm, key_id, keyring_root, nonce, issued_at, expires_at))),
990 ("signature".into(), CborValue::Bytes(signature.to_vec())),
991 ])))
992}
993
994pub fn sync_response_payload_v1(request_root: &str, responder: &str, summary_root: &str, relation: &str, object_ids: &[String], event_ids: &[String], algorithm: &str, key_id: Option<&str>, keyring_root: Option<&str>, issued_at: u64, expires_at: u64) -> Vec<u8> {
995 encode_cbor(&CborValue::Map(vec![
996 ("algorithm".into(), CborValue::Text(algorithm.into())),
997 ("eventIds".into(), CborValue::Array(event_ids.iter().map(|value| CborValue::Text(value.clone())).collect())),
998 ("expiresAt".into(), CborValue::UInt(expires_at)),
999 ("issuedAt".into(), CborValue::UInt(issued_at)),
1000 ("keyId".into(), key_id.map_or(CborValue::Null, |value| CborValue::Text(value.into()))),
1001 ("keyringRoot".into(), keyring_root.map_or(CborValue::Null, |value| CborValue::Text(value.into()))),
1002 ("kind".into(), CborValue::Text("sync-response".into())),
1003 ("objectIds".into(), CborValue::Array(object_ids.iter().map(|value| CborValue::Text(value.clone())).collect())),
1004 ("relation".into(), CborValue::Text(relation.into())),
1005 ("requestRoot".into(), CborValue::Text(request_root.into())),
1006 ("responder".into(), CborValue::Text(responder.into())),
1007 ("summary".into(), CborValue::Text(summary_root.into())),
1008 ("version".into(), CborValue::UInt(1)),
1009 ]))
1010}
1011
1012pub fn sync_response_root_v1(request_root: &str, responder: &str, summary_root: &str, relation: &str, object_ids: &[String], event_ids: &[String], algorithm: &str, key_id: Option<&str>, keyring_root: Option<&str>, issued_at: u64, expires_at: u64, signature: &[u8]) -> String {
1013 digest_domain("sync-response", &encode_cbor(&CborValue::Map(vec![
1014 ("payload".into(), CborValue::Bytes(sync_response_payload_v1(request_root, responder, summary_root, relation, object_ids, event_ids, algorithm, key_id, keyring_root, issued_at, expires_at))),
1015 ("signature".into(), CborValue::Bytes(signature.to_vec())),
1016 ])))
1017}
1018
1019pub fn sync_summary_root_v1(state_root: &str, commit_digest: &str, sequence: u64, parents: &[String], object_root: &str, event_root: &str, keyring_root: Option<&str>) -> String {
1020 digest_domain("sync-summary", &encode_cbor(&CborValue::Map(vec![
1021 ("commitDigest".into(), CborValue::Text(commit_digest.into())),
1022 ("eventRoot".into(), CborValue::Text(event_root.into())),
1023 ("keyringRoot".into(), keyring_root.map_or(CborValue::Null, |value| CborValue::Text(value.into()))),
1024 ("objectRoot".into(), CborValue::Text(object_root.into())),
1025 ("parents".into(), CborValue::Array(parents.iter().map(|value| CborValue::Text(value.clone())).collect())),
1026 ("sequence".into(), CborValue::UInt(sequence)),
1027 ("stateRoot".into(), CborValue::Text(state_root.into())),
1028 ("version".into(), CborValue::UInt(1)),
1029 ])))
1030}
1031
1032#[derive(Debug, Clone, PartialEq)]
1033pub struct KnowledgeAuthorityKeyV1 {
1034 pub version: u64,
1035 pub principal: String,
1036 pub key_id: String,
1037 pub algorithm: String,
1038 pub public_key: Vec<u8>,
1039 pub not_before: Option<u64>,
1040 pub not_after: Option<u64>,
1041 pub revoked_at: Option<u64>,
1042}
1043
1044#[derive(Debug, Clone, PartialEq)]
1045pub struct KnowledgeKeyRotationRecordV1 {
1046 pub version: u64,
1047 pub kind: String,
1048 pub issuer: String,
1049 pub issuer_key_id: String,
1050 pub principal: String,
1051 pub previous_key_id: Option<String>,
1052 pub key_id: String,
1053 pub algorithm: String,
1054 pub public_key: Vec<u8>,
1055 pub not_before: u64,
1056 pub not_after: Option<u64>,
1057 pub revoked_at: Option<u64>,
1058 pub issued_at: u64,
1059 pub expires_at: u64,
1060 pub signature: Vec<u8>,
1061}
1062
1063#[derive(Debug, Clone, PartialEq)]
1064pub struct KnowledgeAuthorityKeyringV1 {
1065 pub version: u64,
1066 pub sequence: u64,
1067 pub keys: Vec<KnowledgeAuthorityKeyV1>,
1068 pub rotations: Vec<KnowledgeKeyRotationRecordV1>,
1069}
1070
1071pub fn authority_key_payload_v1(key: &KnowledgeAuthorityKeyV1) -> Vec<u8> {
1072 encode_cbor(&authority_key_value(key))
1073}
1074
1075pub fn key_rotation_payload_v1(record: &KnowledgeKeyRotationRecordV1) -> Vec<u8> {
1076 encode_cbor(&key_rotation_value(record, false))
1077}
1078
1079pub fn key_rotation_root_v1(record: &KnowledgeKeyRotationRecordV1) -> String {
1080 digest_domain("key-rotation", &encode_cbor(&CborValue::Map(vec![
1081 ("payload".into(), CborValue::Bytes(key_rotation_payload_v1(record))),
1082 ("signature".into(), CborValue::Bytes(record.signature.clone())),
1083 ])))
1084}
1085
1086pub fn authority_keyring_root_v1(keyring: &KnowledgeAuthorityKeyringV1) -> String {
1087 let mut keys = keyring.keys.iter().collect::<Vec<_>>();
1088 keys.sort_by(|left, right| key_sort_key(left).cmp(&key_sort_key(right)));
1089 digest_domain("authority-keyring", &encode_cbor(&CborValue::Map(vec![
1090 ("keys".into(), CborValue::Array(keys.iter().map(|key| CborValue::Bytes(authority_key_payload_v1(key))).collect())),
1091 ("rotations".into(), CborValue::Array(keyring.rotations.iter().map(|record| CborValue::Text(key_rotation_root_v1(record))).collect())),
1092 ("sequence".into(), CborValue::UInt(keyring.sequence)),
1093 ("version".into(), CborValue::UInt(keyring.version)),
1094 ])))
1095}
1096
1097fn authority_key_value(key: &KnowledgeAuthorityKeyV1) -> CborValue {
1098 let mut entries = vec![
1099 ("algorithm".into(), CborValue::Text(key.algorithm.clone())),
1100 ("keyId".into(), CborValue::Text(key.key_id.clone())),
1101 ("principal".into(), CborValue::Text(key.principal.clone())),
1102 ("publicKey".into(), CborValue::Bytes(key.public_key.clone())),
1103 ("version".into(), CborValue::UInt(key.version)),
1104 ];
1105 if let Some(value) = key.not_after { entries.push(("notAfter".into(), CborValue::UInt(value))); }
1106 if let Some(value) = key.not_before { entries.push(("notBefore".into(), CborValue::UInt(value))); }
1107 if let Some(value) = key.revoked_at { entries.push(("revokedAt".into(), CborValue::UInt(value))); }
1108 CborValue::Map(entries)
1109}
1110
1111fn key_rotation_value(record: &KnowledgeKeyRotationRecordV1, include_signature: bool) -> CborValue {
1112 let mut entries = vec![
1113 ("algorithm".into(), CborValue::Text(record.algorithm.clone())),
1114 ("expiresAt".into(), CborValue::UInt(record.expires_at)),
1115 ("issuedAt".into(), CborValue::UInt(record.issued_at)),
1116 ("issuer".into(), CborValue::Text(record.issuer.clone())),
1117 ("issuerKeyId".into(), CborValue::Text(record.issuer_key_id.clone())),
1118 ("keyId".into(), CborValue::Text(record.key_id.clone())),
1119 ("kind".into(), CborValue::Text(record.kind.clone())),
1120 ("notBefore".into(), CborValue::UInt(record.not_before)),
1121 ("publicKey".into(), CborValue::Bytes(record.public_key.clone())),
1122 ("principal".into(), CborValue::Text(record.principal.clone())),
1123 ("version".into(), CborValue::UInt(record.version)),
1124 ];
1125 if include_signature { entries.push(("signature".into(), CborValue::Bytes(record.signature.clone()))); }
1126 if let Some(value) = record.not_after { entries.push(("notAfter".into(), CborValue::UInt(value))); }
1127 if let Some(value) = &record.previous_key_id { entries.push(("previousKeyId".into(), CborValue::Text(value.clone()))); }
1128 if let Some(value) = record.revoked_at { entries.push(("revokedAt".into(), CborValue::UInt(value))); }
1129 CborValue::Map(entries)
1130}
1131
1132fn key_sort_key(key: &KnowledgeAuthorityKeyV1) -> Vec<u8> {
1133 let mut out = key.principal.as_bytes().to_vec();
1134 out.push(0);
1135 out.extend_from_slice(key.key_id.as_bytes());
1136 out
1137}
1138
1139pub fn verify_knowledge_authority_envelope_v5<Resolve, Verify>(
1140 image: &KnowledgeImage,
1141 query_result: &KnowledgeQueryResultV1,
1142 policy: &KnowledgePolicyV1,
1143 authorization: &KnowledgeAuthorizationResultV1,
1144 envelope: &KnowledgeAuthorityEnvelopeV1,
1145 now: u64,
1146 resolve_key: Resolve,
1147 verify_signature: Verify,
1148) -> Result<KnowledgeAuthorityVerificationV1, KnoloError>
1149where
1150 Resolve: Fn(&str, &str, Option<&str>) -> Option<Vec<u8>>,
1151 Verify: Fn(&str, &[u8], &[u8], &[u8]) -> bool,
1152{
1153 verify_knowledge_authority_envelope_with_keyring_root_v5(image, query_result, policy, authorization, envelope, now, None, resolve_key, verify_signature)
1154}
1155
1156pub fn verify_knowledge_authority_envelope_with_keyring_root_v5<Resolve, Verify>(
1157 image: &KnowledgeImage,
1158 query_result: &KnowledgeQueryResultV1,
1159 policy: &KnowledgePolicyV1,
1160 authorization: &KnowledgeAuthorizationResultV1,
1161 envelope: &KnowledgeAuthorityEnvelopeV1,
1162 now: u64,
1163 expected_keyring_root: Option<&str>,
1164 resolve_key: Resolve,
1165 verify_signature: Verify,
1166) -> Result<KnowledgeAuthorityVerificationV1, KnoloError>
1167where
1168 Resolve: Fn(&str, &str, Option<&str>) -> Option<Vec<u8>>,
1169 Verify: Fn(&str, &[u8], &[u8], &[u8]) -> bool,
1170{
1171 let expected_authorization = evaluate_knowledge_query_policy_v5(image, query_result, policy, &authorization.principal, &authorization.action)?;
1172 if &expected_authorization != authorization { return Err(KnoloError::InvalidPack("V5 authorization result is not reproducible".into())); }
1173 if envelope.version != 1 || envelope.issuer.is_empty() || envelope.subject.is_empty() || envelope.algorithm.is_empty() || envelope.key_id.as_deref() == Some("") || envelope.issued_at > now || now >= envelope.expires_at { return Err(KnoloError::InvalidPack("invalid V5 authority envelope window".into())); }
1174 if envelope.keyring_root.is_some() && envelope.keyring_root.as_deref() != expected_keyring_root { return Err(KnoloError::InvalidPack("V5 authority keyring root mismatch".into())); }
1175 if authorization.state_root != image.state_root || envelope.authorization_root != authorization.authorization_root || envelope.subject != authorization.principal { return Err(KnoloError::InvalidPack("V5 authority binding mismatch".into())); }
1176 if envelope.delegations.len() > 8 { return Err(KnoloError::InvalidPack("V5 authority delegation depth exceeded".into())); }
1177 verify_authority_signature(&envelope.issuer, &envelope.algorithm, envelope.key_id.as_deref(), &authority_envelope_payload_v1(envelope), &envelope.signature, &resolve_key, &verify_signature)?;
1178 let mut previous = envelope.issuer.clone();
1179 for delegation in &envelope.delegations {
1180 if delegation.version != 1 || delegation.delegator.is_empty() || delegation.delegatee.is_empty() || delegation.algorithm.is_empty() || delegation.key_id.as_deref() == Some("") || delegation.action != authorization.action || delegation.issued_at > now || now >= delegation.expires_at || delegation.delegator != previous { return Err(KnoloError::InvalidPack("invalid V5 delegation chain".into())); }
1181 verify_authority_signature(&delegation.delegator, &delegation.algorithm, delegation.key_id.as_deref(), &delegation_payload_v1(delegation), &delegation.signature, &resolve_key, &verify_signature)?;
1182 previous = delegation.delegatee.clone();
1183 }
1184 if !envelope.delegations.is_empty() && previous != envelope.subject { return Err(KnoloError::InvalidPack("V5 delegation chain does not reach subject".into())); }
1185 Ok(KnowledgeAuthorityVerificationV1 { envelope_root: authority_envelope_root_v1(envelope), issuer: envelope.issuer.clone(), subject: envelope.subject.clone(), authorization_root: authorization.authorization_root.clone(), keyring_root: envelope.keyring_root.clone(), delegation_depth: envelope.delegations.len() })
1186}
1187
1188fn verify_authority_signature<Resolve, Verify>(principal: &str, algorithm: &str, key_id: Option<&str>, message: &[u8], signature: &[u8], resolve_key: &Resolve, verify_signature: &Verify) -> Result<(), KnoloError>
1189where
1190 Resolve: Fn(&str, &str, Option<&str>) -> Option<Vec<u8>>,
1191 Verify: Fn(&str, &[u8], &[u8], &[u8]) -> bool,
1192{
1193 let key = resolve_key(principal, algorithm, key_id).ok_or_else(|| KnoloError::InvalidPack("missing V5 authority key".into()))?;
1194 if !verify_signature(algorithm, &key, message, signature) { return Err(KnoloError::InvalidPack("V5 authority signature verification failed".into())); }
1195 Ok(())
1196}
1197
1198fn authorize_policy_hit(hit: &KnowledgeQueryHitV1, policy: &KnowledgePolicyV1, principal: &str, action: &str) -> bool {
1199 let matches = policy.rules.iter().filter(|rule| rule.action == action && rule.principal.as_deref().is_none_or(|value| value == principal) && rule.kind.as_deref().is_none_or(|value| value.to_lowercase() == hit.kind)).collect::<Vec<_>>();
1200 if matches.iter().any(|rule| rule.effect == "deny") { return false; }
1201 if matches.iter().any(|rule| rule.effect == "allow") { return true; }
1202 policy.default == "allow"
1203}
1204
1205fn policy_rule_value(rule: &KnowledgePolicyRuleV1) -> CborValue {
1206 let mut entries = vec![("action".into(), CborValue::Text(rule.action.clone())), ("effect".into(), CborValue::Text(rule.effect.clone()))];
1207 if let Some(kind) = &rule.kind { entries.push(("kind".into(), CborValue::Text(kind.to_lowercase()))); }
1208 if let Some(principal) = &rule.principal { entries.push(("principal".into(), CborValue::Text(principal.clone()))); }
1209 CborValue::Map(entries)
1210}
1211
1212fn policy_rule_key(rule: &KnowledgePolicyRuleV1) -> String { format!("{}\0{}\0{}\0{}", rule.effect, rule.action, rule.principal.as_deref().unwrap_or(""), rule.kind.as_deref().unwrap_or("").to_lowercase()) }
1213
1214pub fn parse_knowledge_query_v5(expression: &str) -> Result<KnowledgeQueryPlanV1, KnoloError> {
1215 let tokens = lex_knowledge_query(expression)?;
1216 let mut cursor = 0usize;
1217 expect_query_word(&tokens, &mut cursor, "FROM")?;
1218 let kind_token = query_word(&tokens, &mut cursor, "FROM requires an object kind or *")?;
1219 let kind = if kind_token == "*" { None } else { Some(normalize_query_text(&kind_token)) };
1220 if kind.as_deref() == Some("") { return Err(KnoloError::InvalidPack("empty V5 EQL object kind".into())); }
1221
1222 let mut filters = Vec::new();
1223 if query_is_word(tokens.get(cursor), "WHERE") {
1224 cursor += 1;
1225 loop {
1226 let field = query_word(&tokens, &mut cursor, "WHERE requires a field")?;
1227 let field = normalize_query_field(&field)?;
1228 if !matches!(tokens.get(cursor), Some(QueryToken::Equals)) {
1229 return Err(KnoloError::InvalidPack("V5 EQL WHERE only supports =".into()));
1230 }
1231 cursor += 1;
1232 let value = parse_query_literal(tokens.get(cursor))?;
1233 cursor += 1;
1234 filters.push(KnowledgeQueryFilterV1 { field, value });
1235 if query_is_word(tokens.get(cursor), "AND") {
1236 cursor += 1;
1237 continue;
1238 }
1239 break;
1240 }
1241 }
1242
1243 let mut search = None;
1244 if query_is_word(tokens.get(cursor), "SEARCH") {
1245 cursor += 1;
1246 let value = match tokens.get(cursor) {
1247 Some(QueryToken::String(value)) => normalize_query_text(value),
1248 _ => return Err(KnoloError::InvalidPack("V5 EQL SEARCH requires a quoted string".into())),
1249 };
1250 cursor += 1;
1251 if value.is_empty() { return Err(KnoloError::InvalidPack("empty V5 EQL SEARCH".into())); }
1252 search = Some(value);
1253 }
1254
1255 let mut limit = 100usize;
1256 if query_is_word(tokens.get(cursor), "LIMIT") {
1257 cursor += 1;
1258 let value = query_word(&tokens, &mut cursor, "LIMIT requires a positive integer")?;
1259 limit = value.parse::<usize>().map_err(|_| KnoloError::InvalidPack("invalid V5 EQL LIMIT".into()))?;
1260 if !(1..=1000).contains(&limit) { return Err(KnoloError::InvalidPack("V5 EQL LIMIT must be between 1 and 1000".into())); }
1261 }
1262 if cursor != tokens.len() { return Err(KnoloError::InvalidPack("unexpected V5 EQL token".into())); }
1263
1264 filters.sort_by(|left, right| left.field.cmp(&right.field).then_with(|| query_scalar_key(&left.value).cmp(&query_scalar_key(&right.value))));
1265 Ok(KnowledgeQueryPlanV1 { kind, filters, search, limit })
1266}
1267
1268pub fn query_knowledge_image_v5(image: &KnowledgeImage, expression: &str) -> Result<KnowledgeQueryResultV1, KnoloError> {
1269 let plan = parse_knowledge_query_v5(expression)?;
1270 query_knowledge_plan_v5(image, &plan)
1271}
1272
1273fn query_knowledge_plan_v5(image: &KnowledgeImage, plan: &KnowledgeQueryPlanV1) -> Result<KnowledgeQueryResultV1, KnoloError> {
1274 let plan_root = digest_domain("query-plan", &encode_cbor(&query_plan_value(plan)));
1275 let mut objects = image.objects.iter().filter(|object| query_matches_object(object, plan)).collect::<Vec<_>>();
1276 objects.sort_by(|left, right| left.id.cmp(&right.id));
1277 let hits = objects.into_iter().take(plan.limit).map(|object| KnowledgeQueryHitV1 { object_id: object.id.clone(), kind: object.kind.clone() }).collect::<Vec<_>>();
1278 let object_ids = CborValue::Array(hits.iter().map(|hit| CborValue::Text(hit.object_id.clone())).collect());
1279 let result_value = CborValue::Map(vec![
1280 ("objectIds".into(), object_ids),
1281 ("planRoot".into(), CborValue::Text(plan_root.clone())),
1282 ("stateRoot".into(), CborValue::Text(image.state_root.clone())),
1283 ]);
1284 let result_root = digest_domain("query-result", &encode_cbor(&result_value));
1285 Ok(KnowledgeQueryResultV1 { state_root: image.state_root.clone(), plan_root, hits, result_root })
1286}
1287
1288fn query_plan_value(plan: &KnowledgeQueryPlanV1) -> CborValue {
1289 CborValue::Map(vec![
1290 ("filters".into(), CborValue::Array(plan.filters.iter().map(|filter| CborValue::Map(vec![
1291 ("field".into(), CborValue::Text(filter.field.clone())),
1292 ("op".into(), CborValue::Text("=".into())),
1293 ("value".into(), filter.value.clone()),
1294 ])).collect())),
1295 ("kind".into(), plan.kind.clone().map(CborValue::Text).unwrap_or(CborValue::Null)),
1296 ("limit".into(), CborValue::UInt(plan.limit as u64)),
1297 ("search".into(), plan.search.clone().map(CborValue::Text).unwrap_or(CborValue::Null)),
1298 ("source".into(), CborValue::Text("knowledge-image-v5".into())),
1299 ("version".into(), CborValue::UInt(1)),
1300 ])
1301}
1302
1303fn query_matches_object(object: &KnowledgeObjectV1, plan: &KnowledgeQueryPlanV1) -> bool {
1304 if plan.kind.as_deref().is_some_and(|kind| kind != object.kind) { return false; }
1305 for filter in &plan.filters {
1306 let actual = if filter.field == "id" { Some(CborValue::Text(object.id.clone())) } else if filter.field == "kind" { Some(CborValue::Text(object.kind.clone())) } else { object.meta_map_value(&filter.field[5..]).cloned() };
1307 if !query_scalars_equal(actual.as_ref(), Some(&filter.value)) { return false; }
1308 }
1309 if let Some(search) = &plan.search {
1310 let text = normalize_query_text(&String::from_utf8_lossy(&object.bytes));
1311 if !search.split(' ').all(|term| text.contains(term)) { return false; }
1312 }
1313 true
1314}
1315
1316trait KnowledgeObjectMeta {
1317 fn meta_map_value(&self, key: &str) -> Option<&CborValue>;
1318}
1319
1320impl KnowledgeObjectMeta for KnowledgeObjectV1 {
1321 fn meta_map_value(&self, key: &str) -> Option<&CborValue> {
1322 match &self.meta { CborValue::Map(entries) => entries.iter().find(|(name, _)| name == key).map(|(_, value)| value), _ => None }
1323 }
1324}
1325
1326#[derive(Debug, Clone, PartialEq)]
1327enum QueryToken { Word(String), String(String), Equals }
1328
1329fn lex_knowledge_query(expression: &str) -> Result<Vec<QueryToken>, KnoloError> {
1330 let chars = expression.chars().collect::<Vec<_>>();
1331 let mut tokens = Vec::new();
1332 let mut cursor = 0usize;
1333 while cursor < chars.len() {
1334 while chars.get(cursor).is_some_and(|ch| ch.is_whitespace()) { cursor += 1; }
1335 if cursor >= chars.len() { break; }
1336 if chars[cursor] == '=' { tokens.push(QueryToken::Equals); cursor += 1; continue; }
1337 if chars[cursor] == '"' {
1338 cursor += 1;
1339 let mut value = String::new();
1340 let mut closed = false;
1341 while cursor < chars.len() {
1342 let ch = chars[cursor]; cursor += 1;
1343 if ch == '"' { closed = true; break; }
1344 if ch == '\\' {
1345 let escaped = *chars.get(cursor).ok_or_else(|| KnoloError::InvalidPack("unterminated V5 EQL string".into()))?; cursor += 1;
1346 if escaped != '"' && escaped != '\\' { return Err(KnoloError::InvalidPack("unsupported V5 EQL escape".into())); }
1347 value.push(escaped);
1348 } else { value.push(ch); }
1349 }
1350 if !closed { return Err(KnoloError::InvalidPack("unterminated V5 EQL string".into())); }
1351 tokens.push(QueryToken::String(value));
1352 continue;
1353 }
1354 let start = cursor;
1355 while cursor < chars.len() && !chars[cursor].is_whitespace() && chars[cursor] != '=' { cursor += 1; }
1356 let value = chars[start..cursor].iter().collect::<String>();
1357 if value.is_empty() || !value.chars().all(|ch| ch.is_ascii_alphanumeric() || matches!(ch, '_' | '.' | '*' | '-')) { return Err(KnoloError::InvalidPack("invalid V5 EQL token".into())); }
1358 tokens.push(QueryToken::Word(value));
1359 }
1360 Ok(tokens)
1361}
1362
1363fn expect_query_word(tokens: &[QueryToken], cursor: &mut usize, expected: &str) -> Result<(), KnoloError> {
1364 if !query_is_word(tokens.get(*cursor), expected) { return Err(KnoloError::InvalidPack(format!("V5 EQL query must start with {expected}"))); }
1365 *cursor += 1;
1366 Ok(())
1367}
1368
1369fn query_word(tokens: &[QueryToken], cursor: &mut usize, message: &str) -> Result<String, KnoloError> {
1370 match tokens.get(*cursor) { Some(QueryToken::Word(value)) => { *cursor += 1; Ok(value.clone()) }, _ => Err(KnoloError::InvalidPack(message.into())) }
1371}
1372
1373fn query_is_word(token: Option<&QueryToken>, expected: &str) -> bool { matches!(token, Some(QueryToken::Word(value)) if value.eq_ignore_ascii_case(expected)) }
1374
1375fn normalize_query_field(field: &str) -> Result<String, KnoloError> {
1376 let lower = field.to_ascii_lowercase();
1377 if lower == "id" || lower == "kind" { return Ok(lower); }
1378 if lower.strip_prefix("meta.").is_some_and(|key| !key.is_empty() && key.chars().all(|ch| ch.is_ascii_alphanumeric() || matches!(ch, '_' | '-'))) { return Ok(lower); }
1379 Err(KnoloError::InvalidPack(format!("unsupported V5 EQL field: {field}")))
1380}
1381
1382fn parse_query_literal(token: Option<&QueryToken>) -> Result<CborValue, KnoloError> {
1383 match token {
1384 Some(QueryToken::String(value)) => Ok(CborValue::Text(normalize_query_text(value))),
1385 Some(QueryToken::Word(value)) if value == "true" => Ok(CborValue::Bool(true)),
1386 Some(QueryToken::Word(value)) if value == "false" => Ok(CborValue::Bool(false)),
1387 Some(QueryToken::Word(value)) if value == "null" => Ok(CborValue::Null),
1388 Some(QueryToken::Word(value)) if value.parse::<i64>().is_ok() => {
1389 let number = value.parse::<i64>().unwrap();
1390 if number >= 0 { Ok(CborValue::UInt(number as u64)) } else { Ok(CborValue::NInt(number)) }
1391 }
1392 _ => Err(KnoloError::InvalidPack("V5 EQL WHERE requires a scalar literal".into())),
1393 }
1394}
1395
1396fn normalize_query_text(value: &str) -> String { value.to_lowercase().split_whitespace().collect::<Vec<_>>().join(" ") }
1397fn query_scalars_equal(actual: Option<&CborValue>, expected: Option<&CborValue>) -> bool {
1398 match (actual, expected) {
1399 (Some(CborValue::Text(left)), Some(CborValue::Text(right))) => normalize_query_text(left) == *right,
1400 _ => actual == expected,
1401 }
1402}
1403fn query_scalar_key(value: &CborValue) -> String {
1404 match value {
1405 CborValue::Null => "null:".into(),
1406 CborValue::Bool(value) => format!("boolean:{value}"),
1407 CborValue::UInt(value) => format!("number:{value}"),
1408 CborValue::NInt(value) => format!("number:{value}"),
1409 CborValue::Text(value) => format!("string:{value}"),
1410 other => format!("{other:?}"),
1411 }
1412}
1413
1414pub fn migrate_v4_to_v5(bytes: &[u8]) -> Result<MigrationResult, KnoloError> {
1415 let pack = if bytes.starts_with(b"KNLOV4\0\0") {
1416 parse_v4_pack(bytes)?
1417 } else {
1418 mount_pack_from_bytes(bytes)?
1419 };
1420 build_migration_image(&pack, &digest_from_raw(&sha256(bytes)))
1421}
1422
1423fn parse_v4_pack(bytes: &[u8]) -> Result<Pack, KnoloError> {
1424 if bytes.len() < 192 || &bytes[0..8] != b"KNLOV4\0\0" { return Err(KnoloError::InvalidPack("invalid V4 pack".into())); }
1425 let header_length = u32::from_le_bytes(bytes[12..16].try_into().unwrap()) as usize;
1426 let manifest_length = u32::from_le_bytes(bytes[20..24].try_into().unwrap()) as usize;
1427 let directory_length = u32::from_le_bytes(bytes[24..28].try_into().unwrap()) as usize;
1428 if header_length > bytes.len() || 192usize.checked_add(manifest_length).and_then(|n| n.checked_add(directory_length)) != Some(header_length) { return Err(KnoloError::InvalidPack("invalid V4 directory bounds".into())); }
1429 let manifest = bytes.get(192..192 + manifest_length).ok_or_else(|| KnoloError::InvalidPack("V4 manifest exceeds file".into()))?;
1430 if read_ascii_digest(&bytes[32..112]) != digest_from_raw(&sha256(manifest)) || read_ascii_digest(&bytes[112..192]) != digest_from_raw(&sha256(&bytes[header_length..])) { return Err(KnoloError::InvalidPack("V4 pack or manifest digest mismatch".into())); }
1431 let directory = std::str::from_utf8(&bytes[192 + manifest_length..header_length]).map_err(|_| KnoloError::InvalidPack("V4 directory is not UTF-8".into()))?;
1432 let (metadata_offset, metadata_length, metadata_digest) = v4_section_bounds(directory, "metadata")?;
1433 let metadata_end = metadata_offset.checked_add(metadata_length).ok_or_else(|| KnoloError::InvalidPack("V4 metadata overflow".into()))?;
1434 if metadata_end > bytes.len() || metadata_digest != digest_from_raw(&sha256(&bytes[metadata_offset..metadata_end])) { return Err(KnoloError::InvalidPack("V4 metadata digest mismatch".into())); }
1435 let metadata_json = std::str::from_utf8(&bytes[metadata_offset..metadata_end]).map_err(|_| KnoloError::InvalidPack("V4 metadata is not UTF-8".into()))?.to_string();
1436 let meta = parse_meta(&metadata_json)?;
1437 let claims_json = match v4_section_bounds(directory, "claims") {
1438 Ok((offset, length, digest)) => {
1439 let end = offset.checked_add(length).ok_or_else(|| KnoloError::InvalidPack("V4 claims overflow".into()))?;
1440 if end > bytes.len() || digest != digest_from_raw(&sha256(&bytes[offset..end])) { return Err(KnoloError::InvalidPack("V4 claims digest mismatch".into())); }
1441 Some(std::str::from_utf8(&bytes[offset..end]).map_err(|_| KnoloError::InvalidPack("V4 claims are not UTF-8".into()))?.to_string())
1442 }
1443 Err(_) => None,
1444 };
1445 let (chunks_offset, chunks_length, chunks_digest) = v4_section_bounds(directory, "chunks")?;
1446 let chunks_end = chunks_offset.checked_add(chunks_length).ok_or_else(|| KnoloError::InvalidPack("V4 chunks overflow".into()))?;
1447 if chunks_end > bytes.len() { return Err(KnoloError::InvalidPack("V4 chunks exceed file".into())); }
1448 if chunks_digest != digest_from_raw(&sha256(&bytes[chunks_offset..chunks_end])) { return Err(KnoloError::InvalidPack("V4 chunks digest mismatch".into())); }
1449 let parsed = parse_blocks(std::str::from_utf8(&bytes[chunks_offset..chunks_end]).map_err(|_| KnoloError::InvalidPack("V4 chunks are not UTF-8".into()))?)?;
1450 let count = parsed.texts.len();
1451 Ok(Pack { meta: PackMeta { stats: PackStats { docs: count, blocks: count, ..meta.stats }, ..meta }, lexicon: HashMap::new(), postings: Vec::new(), blocks: parsed.texts, headings: parsed.headings, doc_ids: parsed.doc_ids, namespaces: parsed.namespaces, block_token_lens: parsed.lens, metadata_json, claims_json })
1452}
1453
1454fn v4_section_bounds(directory: &str, name: &str) -> Result<(usize, usize, String), KnoloError> {
1455 let marker = format!("\"name\":\"{name}\"");
1456 let start = directory.find(&marker).ok_or_else(|| KnoloError::InvalidPack(format!("missing V4 section: {name}")))?;
1457 let entry = &directory[start..];
1458 let offset = json_number_after(entry, "\"offset\":")? as usize;
1459 let length = json_number_after(entry, "\"length\":")? as usize;
1460 let digest = json_string_after(entry, "\"digest\":\"")?;
1461 Ok((offset, length, digest))
1462}
1463
1464fn json_number_after(input: &str, marker: &str) -> Result<u64, KnoloError> {
1465 let start = input.find(marker).ok_or_else(|| KnoloError::InvalidPack("missing V4 directory number".into()))? + marker.len();
1466 let digits = input[start..].chars().take_while(|ch| ch.is_ascii_digit()).collect::<String>();
1467 digits.parse::<u64>().map_err(|_| KnoloError::InvalidPack("invalid V4 directory number".into()))
1468}
1469
1470fn json_string_after(input: &str, marker: &str) -> Result<String, KnoloError> {
1471 let start = input.find(marker).ok_or_else(|| KnoloError::InvalidPack("missing V4 directory digest".into()))? + marker.len();
1472 let end = input[start..].find('"').ok_or_else(|| KnoloError::InvalidPack("invalid V4 directory digest".into()))?;
1473 Ok(input[start..start + end].to_string())
1474}
1475
1476fn json_field_raw(input: &str, key: &str) -> Option<String> {
1477 let marker = format!("\"{key}\":");
1478 let start = input.find(&marker)? + marker.len();
1479 let bytes = input.as_bytes();
1480 let first = *bytes.get(start)?;
1481 if first == b'{' || first == b'[' {
1482 let open = first;
1483 let close = if open == b'{' { b'}' } else { b']' };
1484 let mut depth = 0usize;
1485 let mut in_string = false;
1486 let mut escaped = false;
1487 for index in start..bytes.len() {
1488 let byte = bytes[index];
1489 if in_string {
1490 if escaped { escaped = false; } else if byte == b'\\' { escaped = true; } else if byte == b'"' { in_string = false; }
1491 continue;
1492 }
1493 if byte == b'"' { in_string = true; continue; }
1494 if byte == open { depth += 1; }
1495 if byte == close { depth -= 1; if depth == 0 { return Some(input[start..=index].to_string()); } }
1496 }
1497 None
1498 } else {
1499 let end = input[start..].find(',').map(|offset| start + offset).unwrap_or(input.len());
1500 Some(input[start..end].trim().to_string())
1501 }
1502}
1503
1504fn read_ascii_digest(bytes: &[u8]) -> String {
1505 let end = bytes.iter().position(|byte| *byte == 0).unwrap_or(bytes.len());
1506 String::from_utf8_lossy(&bytes[..end]).to_string()
1507}
1508
1509#[derive(Clone)]
1510struct MigrationObject { id: String, kind: String, bytes: Vec<u8>, meta: CborValue }
1511
1512fn build_migration_image(pack: &Pack, source_digest: &str) -> Result<MigrationResult, KnoloError> {
1513 let actor = "knolo-v4-migrator";
1514 let mut objects = Vec::new();
1515 let mut mappings = Vec::new();
1516 for (index, text) in pack.blocks.iter().enumerate() {
1517 let source_bytes = text.as_bytes().to_vec();
1518 let source_meta = CborValue::Map(vec![
1519 ("docId".into(), pack.doc_ids.get(index).and_then(|value| value.clone()).map(CborValue::Text).unwrap_or(CborValue::Null)),
1520 ("legacyBlockId".into(), CborValue::UInt(index as u64)),
1521 ("namespace".into(), pack.namespaces.get(index).and_then(|value| value.clone()).map(CborValue::Text).unwrap_or(CborValue::Null)),
1522 ]);
1523 let source_body = CborValue::Map(vec![("bytes".into(), CborValue::Bytes(source_bytes.clone())), ("kind".into(), CborValue::Text("source".into())), ("meta".into(), source_meta.clone())]);
1524 let source_id = digest_domain("object", &encode_cbor(&source_body));
1525 let chunk_meta = CborValue::Map(vec![
1526 ("docId".into(), pack.doc_ids.get(index).and_then(|value| value.clone()).map(CborValue::Text).unwrap_or(CborValue::Null)),
1527 ("heading".into(), pack.headings.get(index).and_then(|value| value.clone()).map(CborValue::Text).unwrap_or(CborValue::Null)),
1528 ("legacyBlockId".into(), CborValue::UInt(index as u64)),
1529 ("namespace".into(), pack.namespaces.get(index).and_then(|value| value.clone()).map(CborValue::Text).unwrap_or(CborValue::Null)),
1530 ("sourceObject".into(), CborValue::Text(source_id.clone())),
1531 ("span".into(), CborValue::Map(vec![("end".into(), CborValue::UInt(text.len() as u64)), ("start".into(), CborValue::UInt(0))])),
1532 ]);
1533 let chunk_body = CborValue::Map(vec![("bytes".into(), CborValue::Bytes(source_bytes.clone())), ("kind".into(), CborValue::Text("chunk".into())), ("meta".into(), chunk_meta.clone())]);
1534 let chunk_id = digest_domain("object", &encode_cbor(&chunk_body));
1535 objects.push(MigrationObject { id: source_id.clone(), kind: "source".into(), bytes: source_bytes.clone(), meta: source_meta });
1536 objects.push(MigrationObject { id: chunk_id.clone(), kind: "chunk".into(), bytes: source_bytes, meta: chunk_meta });
1537 mappings.push(MigrationMapping { legacy_block_id: index, source_object: source_id, chunk_object: chunk_id });
1538 }
1539 if let Some(claims_json) = &pack.claims_json {
1540 let bytes = claims_json.as_bytes().to_vec();
1541 let meta = CborValue::Map(vec![("encoding".into(), CborValue::Text("json-v4".into())), ("version".into(), CborValue::UInt(1))]);
1542 let body = CborValue::Map(vec![("bytes".into(), CborValue::Bytes(bytes.clone())), ("kind".into(), CborValue::Text("claims".into())), ("meta".into(), meta.clone())]);
1543 objects.push(MigrationObject { id: digest_domain("object", &encode_cbor(&body)), kind: "claims".into(), bytes, meta });
1544 }
1545 if let Some(agents_json) = json_field_raw(&pack.metadata_json, "agents") {
1546 let bytes = agents_json.as_bytes().to_vec();
1547 let meta = CborValue::Map(vec![("encoding".into(), CborValue::Text("json-v4".into())), ("version".into(), CborValue::UInt(1))]);
1548 let body = CborValue::Map(vec![("bytes".into(), CborValue::Bytes(bytes.clone())), ("kind".into(), CborValue::Text("agents".into())), ("meta".into(), meta.clone())]);
1549 objects.push(MigrationObject { id: digest_domain("object", &encode_cbor(&body)), kind: "agents".into(), bytes, meta });
1550 }
1551 let metadata_meta = CborValue::Map(vec![("sourceDigest".into(), CborValue::Text(source_digest.into())), ("sourceVersion".into(), CborValue::UInt(pack.meta.version as u64))]); let metadata_bytes = encode_cbor(&metadata_meta); let metadata_body = CborValue::Map(vec![("bytes".into(), CborValue::Bytes(metadata_bytes.clone())), ("kind".into(), CborValue::Text("metadata".into())), ("meta".into(), metadata_meta.clone())]);
1552 objects.push(MigrationObject { id: digest_domain("object", &encode_cbor(&metadata_body)), kind: "metadata".into(), bytes: metadata_bytes, meta: metadata_meta });
1553 objects.sort_by(|left, right| left.id.cmp(&right.id));
1554 let object_values = objects.iter().map(|object| CborValue::Map(vec![("bytes".into(), CborValue::Bytes(object.bytes.clone())), ("id".into(), CborValue::Text(object.id.clone())), ("kind".into(), CborValue::Text(object.kind.clone())), ("meta".into(), object.meta.clone())])).collect::<Vec<_>>();
1555 let object_payload = encode_cbor(&CborValue::Array(object_values));
1556 let object_segment_digest = digest_domain("segment", &object_payload);
1557 let object_ids = objects.iter().map(|object| CborValue::Text(object.id.clone())).collect::<Vec<_>>();
1558 let parents = CborValue::Array(Vec::new());
1559 let transaction_id = digest_domain("transaction", &encode_cbor(&CborValue::Map(vec![("actor".into(), CborValue::Text(actor.into())), ("objects".into(), CborValue::Array(object_ids.clone())), ("parents".into(), parents.clone())])));
1560 let mut events = Vec::new();
1561 for (index, object) in objects.iter().enumerate() {
1562 let event_kind = if object.kind == "chunk" { "document.put".to_string() } else { format!("{}.put", object.kind) };
1563 let event_body = CborValue::Map(vec![("actor".into(), CborValue::Text(actor.into())), ("actorCounter".into(), CborValue::UInt(index as u64 + 1)), ("kind".into(), CborValue::Text(event_kind.into())), ("parents".into(), if index == 0 { CborValue::Array(Vec::new()) } else { CborValue::Array(vec![CborValue::Text(transaction_id.clone())]) }), ("payload".into(), CborValue::Text(object.id.clone())), ("provenance".into(), CborValue::Map(vec![("objectId".into(), CborValue::Text(object.id.clone()))])), ("target".into(), CborValue::Text(object.id.clone())), ("transactionId".into(), CborValue::Text(transaction_id.clone())), ("version".into(), CborValue::UInt(1))]);
1564 let event_id = digest_domain("event", &encode_cbor(&event_body));
1565 events.push((event_id, event_body));
1566 }
1567 events.sort_by(|left, right| left.0.cmp(&right.0));
1568 let event_values = events.iter().map(|(id, body)| { let mut entries = match body { CborValue::Map(entries) => entries.clone(), _ => Vec::new() }; entries.push(("id".into(), CborValue::Text(id.clone()))); CborValue::Map(entries) }).collect::<Vec<_>>();
1569 let event_payload = encode_cbor(&CborValue::Array(event_values));
1570 let event_segment_digest = digest_domain("segment", &event_payload);
1571 let event_ids = events.iter().map(|(id, _)| CborValue::Text(id.clone())).collect::<Vec<_>>();
1572 let object_root = digest_domain("object-root", &encode_cbor(&CborValue::Array(object_ids.clone())));
1573 let event_root = digest_domain("event-root", &encode_cbor(&CborValue::Array(event_ids.clone())));
1574 let lexical_view = digest_domain("view", &encode_cbor(&CborValue::Map(vec![("kind".into(), CborValue::Text("lexical".into())), ("objectIds".into(), CborValue::Array(object_ids.clone()))])));
1575 let commit_value = CborValue::Map(vec![("actor".into(), CborValue::Text(actor.into())), ("eventRoot".into(), CborValue::Text(event_root.clone())), ("eventSegmentDigest".into(), CborValue::Text(event_segment_digest)), ("objectRoot".into(), CborValue::Text(object_root.clone())), ("objectSegmentDigest".into(), CborValue::Text(object_segment_digest)), ("parents".into(), parents), ("policyRoot".into(), CborValue::Text(digest_domain("policy", &encode_cbor(&CborValue::Map(vec![("default".into(), CborValue::Text("deny".into()))]))))), ("runtimeContract".into(), CborValue::Text(digest_domain("runtime", &encode_cbor(&CborValue::Map(vec![("codec".into(), CborValue::Text("cbor-v1".into())), ("format".into(), CborValue::UInt(5))]))))), ("schemaRoot".into(), CborValue::Text(digest_domain("schema", &encode_cbor(&CborValue::Map(vec![("version".into(), CborValue::UInt(1))]))))), ("sequence".into(), CborValue::UInt(1)), ("transactionRoot".into(), CborValue::Text(digest_domain("transaction-root", &encode_cbor(&CborValue::Map(vec![("eventIds".into(), CborValue::Array(event_ids)), ("objectIds".into(), CborValue::Array(object_ids)), ("transactionId".into(), CborValue::Text(transaction_id))]))))), ("version".into(), CborValue::UInt(1)), ("views".into(), CborValue::Map(vec![("lexical".into(), CborValue::Text(lexical_view))]))]);
1576 let commit_payload = encode_cbor(&commit_value); let commit_digest = digest_domain("commit", &commit_payload); let state = digest_domain("state", &digest_raw(&commit_digest));
1577 let object_segment = encode_v5_segment(1, &object_payload); let event_segment = encode_v5_segment(2, &event_payload); let commit_segment = encode_v5_segment(3, &commit_payload); let commit_offset = V5_DATA_START + object_segment.len() + event_segment.len();
1578 let mut image = vec![0u8; V5_DATA_START + object_segment.len() + event_segment.len() + commit_segment.len()]; image[0..8].copy_from_slice(KNOWLEDGE_IMAGE_V5_MAGIC); image[8..10].copy_from_slice(&5u16.to_le_bytes()); image[12..14].copy_from_slice(&(V5_SUPERBLOCK_SIZE as u16).to_le_bytes()); image[V5_DATA_START..V5_DATA_START + object_segment.len()].copy_from_slice(&object_segment); image[V5_DATA_START + object_segment.len()..commit_offset].copy_from_slice(&event_segment); image[commit_offset..].copy_from_slice(&commit_segment); let superblock = encode_v5_superblock(1, commit_offset, commit_segment.len(), &commit_digest, &state); image[V5_HEADER_SIZE..V5_HEADER_SIZE + V5_SUPERBLOCK_SIZE].copy_from_slice(&superblock);
1579 let receipt_body = CborValue::Map(vec![("kind".into(), CborValue::Text("v4-to-v5-migration".into())), ("objectMappings".into(), CborValue::Array(mappings.iter().map(|mapping| CborValue::Map(vec![("chunkObject".into(), CborValue::Text(mapping.chunk_object.clone())), ("legacyBlockId".into(), CborValue::UInt(mapping.legacy_block_id as u64)), ("sourceObject".into(), CborValue::Text(mapping.source_object.clone()))])).collect())), ("sourceDigest".into(), CborValue::Text(source_digest.into())), ("sourceVersion".into(), CborValue::UInt(pack.meta.version as u64)), ("stateRoot".into(), CborValue::Text(state.clone())), ("version".into(), CborValue::UInt(1))]); let receipt_digest = digest_domain("receipt", &encode_cbor(&receipt_body)); let mut receipt_entries = match receipt_body { CborValue::Map(entries) => entries, _ => Vec::new() }; receipt_entries.push(("receiptDigest".into(), CborValue::Text(receipt_digest))); let receipt = encode_cbor(&CborValue::Map(receipt_entries));
1580 Ok(MigrationResult { image, receipt, state_root: state, mappings })
1581}
1582
1583fn encode_v5_segment(kind: u8, payload: &[u8]) -> Vec<u8> { let mut out = vec![0u8; V5_SEGMENT_HEADER_SIZE + payload.len()]; out[0..4].copy_from_slice(SEGMENT_MAGIC); out[4] = kind; out[5] = 1; out[8..16].copy_from_slice(&(payload.len() as u64).to_le_bytes()); out[16..48].copy_from_slice(&digest_raw(&digest_domain("segment", payload))); out[V5_SEGMENT_HEADER_SIZE..].copy_from_slice(payload); out }
1584fn encode_v5_superblock(generation: u64, commit_offset: usize, commit_length: usize, commit_digest: &str, state: &str) -> Vec<u8> { let mut out = vec![0u8; V5_SUPERBLOCK_SIZE]; out[0..8].copy_from_slice(SUPERBLOCK_MAGIC); out[8..16].copy_from_slice(&generation.to_le_bytes()); out[16..24].copy_from_slice(&(commit_offset as u64).to_le_bytes()); out[24..32].copy_from_slice(&(commit_length as u64).to_le_bytes()); out[32..64].copy_from_slice(&digest_raw(commit_digest)); out[64..96].copy_from_slice(&digest_raw(state)); let digest = digest_raw(&digest_domain("superblock", &out[0..96])); out[96..128].copy_from_slice(&digest); out }
1585
1586fn parse_v5_image(bytes: &[u8]) -> Result<KnowledgeImage, KnoloError> {
1587 if bytes.len() < V5_DATA_START { return Err(KnoloError::InvalidPack("V5 image is truncated".into())); }
1588 if &bytes[0..8] != KNOWLEDGE_IMAGE_V5_MAGIC { return Err(KnoloError::InvalidPack("invalid V5 image magic".into())); }
1589 if le_u16(bytes, 8)? != 5 || le_u16(bytes, 12)? as usize != V5_SUPERBLOCK_SIZE { return Err(KnoloError::InvalidPack("unsupported V5 image header".into())); }
1590
1591 let a = read_v5_superblock(bytes, V5_HEADER_SIZE, 'A');
1592 let b = read_v5_superblock(bytes, V5_HEADER_SIZE + V5_SUPERBLOCK_SIZE, 'B');
1593 let mut candidates = [a, b].into_iter().flatten().collect::<Vec<_>>();
1594 if candidates.is_empty() { return Err(KnoloError::InvalidPack("no valid V5 superblock".into())); }
1595
1596 let mut segments = Vec::new();
1597 let mut offset = V5_DATA_START;
1598 let mut required_seen = [false; 3];
1599 while offset < bytes.len() {
1600 let segment = read_v5_segment(bytes, offset)?;
1601 if !matches!(segment.kind, 1..=3) && segment.kind < 128 { return Err(KnoloError::InvalidPack("unknown non-optional V5 segment".into())); }
1602 if segment.kind <= 3 && segment.schema != 1 { return Err(KnoloError::InvalidPack("unsupported required V5 segment schema".into())); }
1603 if (1..=3).contains(&segment.kind) {
1604 if required_seen[segment.kind as usize - 1] { return Err(KnoloError::InvalidPack("duplicate required V5 segment".into())); }
1605 required_seen[segment.kind as usize - 1] = true;
1606 }
1607 offset += segment.length;
1608 segments.push(segment);
1609 }
1610 if offset != bytes.len() || required_seen.iter().any(|seen| !seen) { return Err(KnoloError::InvalidPack("missing or misaligned V5 segment".into())); }
1611
1612 let commit_segment = segments.iter().find(|segment| segment.kind == 3).ok_or_else(|| KnoloError::InvalidPack("missing V5 commit segment".into()))?;
1613 let object_segment = segments.iter().find(|segment| segment.kind == 1).ok_or_else(|| KnoloError::InvalidPack("missing V5 object segment".into()))?;
1614 let event_segment = segments.iter().find(|segment| segment.kind == 2).ok_or_else(|| KnoloError::InvalidPack("missing V5 event segment".into()))?;
1615 let commit_payload = &bytes[commit_segment.offset + V5_SEGMENT_HEADER_SIZE..commit_segment.offset + commit_segment.length];
1616 let commit_value = decode_cbor_exact(commit_payload)?;
1617 if encode_cbor(&commit_value) != commit_payload { return Err(KnoloError::InvalidPack("non-canonical V5 commit CBOR".into())); }
1618 let mut commit = parse_commit(&commit_value)?;
1619 let commit_digest = digest_domain("commit", commit_payload);
1620 let state = digest_domain("state", &digest_raw(&commit_digest));
1621 candidates.sort_by(|left, right| right.generation.cmp(&left.generation));
1622 let active = candidates.into_iter().find(|candidate| candidate.commit_offset == commit_segment.offset && candidate.commit_length == commit_segment.length && candidate.commit_digest == commit_digest && candidate.state_root == state).ok_or_else(|| KnoloError::InvalidPack("no V5 superblock points to a valid commit".into()))?;
1623 commit.state_root = state.clone();
1624 commit.commit_digest = commit_digest.clone();
1625 if commit_field(&commit_value, "objectSegmentDigest")? != object_segment.digest || commit_field(&commit_value, "eventSegmentDigest")? != event_segment.digest { return Err(KnoloError::InvalidPack("V5 segment digest mismatch".into())); }
1626
1627 let objects = parse_objects(&bytes[object_segment.offset + V5_SEGMENT_HEADER_SIZE..object_segment.offset + object_segment.length])?;
1628 let events = parse_events(&bytes[event_segment.offset + V5_SEGMENT_HEADER_SIZE..event_segment.offset + event_segment.length])?;
1629 let object_ids = CborValue::Array(objects.iter().map(|object| CborValue::Text(object.id.clone())).collect());
1630 let event_ids = CborValue::Array(events.iter().map(|event| CborValue::Text(event.id.clone())).collect());
1631 if digest_domain("object-root", &encode_cbor(&object_ids)) != commit.object_root || digest_domain("event-root", &encode_cbor(&event_ids)) != commit.event_root { return Err(KnoloError::InvalidPack("V5 object/event root mismatch".into())); }
1632
1633 Ok(KnowledgeImage { state_root: state, commit_digest, commit, objects, events, segments, active_superblock: active.slot })
1634}
1635
1636#[derive(Debug, Clone)]
1637struct V5Superblock { generation: u64, commit_offset: usize, commit_length: usize, commit_digest: String, state_root: String, slot: char }
1638
1639fn read_v5_superblock(bytes: &[u8], offset: usize, slot: char) -> Option<V5Superblock> {
1640 let raw = bytes.get(offset..offset + V5_SUPERBLOCK_SIZE)?;
1641 if raw.get(0..8)? != SUPERBLOCK_MAGIC { return None; }
1642 let generation = u64::from_le_bytes(raw.get(8..16)?.try_into().ok()?);
1643 let commit_offset = usize::try_from(u64::from_le_bytes(raw.get(16..24)?.try_into().ok()?)).ok()?;
1644 let commit_length = usize::try_from(u64::from_le_bytes(raw.get(24..32)?.try_into().ok()?)).ok()?;
1645 if commit_length < V5_SEGMENT_HEADER_SIZE || commit_offset.checked_add(commit_length)? > bytes.len() { return None; }
1646 let commit_digest = digest_from_raw(raw.get(32..64)?);
1647 let state_root = digest_from_raw(raw.get(64..96)?);
1648 if digest_from_raw(raw.get(96..128)?) != digest_domain("superblock", raw.get(0..96)?) { return None; }
1649 Some(V5Superblock { generation, commit_offset, commit_length, commit_digest, state_root, slot })
1650}
1651
1652fn read_v5_segment(bytes: &[u8], offset: usize) -> Result<KnowledgeImageSegment, KnoloError> {
1653 let header = read_slice(bytes, &mut offset.clone(), V5_SEGMENT_HEADER_SIZE)?;
1654 if header.get(0..4) != Some(SEGMENT_MAGIC) { return Err(KnoloError::InvalidPack("invalid V5 segment magic".into())); }
1655 let kind = header[4]; let schema = header[5]; let flags = u16::from_le_bytes([header[6], header[7]]);
1656 let payload_length = usize::try_from(u64::from_le_bytes(header[8..16].try_into().unwrap())).map_err(|_| KnoloError::InvalidPack("V5 segment length overflow".into()))?;
1657 if payload_length as u64 > V5_MAX_SEGMENT { return Err(KnoloError::InvalidPack("V5 segment exceeds safety limit".into())); }
1658 let length = V5_SEGMENT_HEADER_SIZE.checked_add(payload_length).ok_or_else(|| KnoloError::InvalidPack("V5 segment length overflow".into()))?;
1659 let payload = bytes.get(offset + V5_SEGMENT_HEADER_SIZE..offset + length).ok_or_else(|| KnoloError::InvalidPack("V5 segment exceeds file".into()))?;
1660 let digest = digest_from_raw(&header[16..48]);
1661 if digest != digest_domain("segment", payload) { return Err(KnoloError::InvalidPack("V5 segment digest mismatch".into())); }
1662 Ok(KnowledgeImageSegment { kind, schema, flags, offset, length, payload_length, digest })
1663}
1664
1665fn parse_objects(payload: &[u8]) -> Result<Vec<KnowledgeObjectV1>, KnoloError> {
1666 let value = decode_cbor_exact(payload)?;
1667 if encode_cbor(&value) != payload { return Err(KnoloError::InvalidPack("non-canonical V5 object CBOR".into())); }
1668 let entries = match value { CborValue::Array(items) => items, _ => return Err(KnoloError::InvalidPack("invalid V5 object segment".into())) };
1669 entries.into_iter().map(|entry| {
1670 let map = as_map(&entry)?;
1671 let id = map_text(map, "id")?; let kind = map_text(map, "kind")?; let bytes = map_bytes(map, "bytes")?; let meta = map_value(map, "meta")?.clone();
1672 let body = CborValue::Map(vec![("bytes".into(), CborValue::Bytes(bytes.clone())), ("kind".into(), CborValue::Text(kind.clone())), ("meta".into(), meta.clone())]);
1673 if digest_domain("object", &encode_cbor(&body)) != id { return Err(KnoloError::InvalidPack("V5 object identity mismatch".into())); }
1674 Ok(KnowledgeObjectV1 { id, kind, bytes, meta })
1675 }).collect()
1676}
1677
1678fn parse_events(payload: &[u8]) -> Result<Vec<KnowledgeEventV1>, KnoloError> {
1679 let value = decode_cbor_exact(payload)?;
1680 if encode_cbor(&value) != payload { return Err(KnoloError::InvalidPack("non-canonical V5 event CBOR".into())); }
1681 let entries = match value { CborValue::Array(items) => items, _ => return Err(KnoloError::InvalidPack("invalid V5 event segment".into())) };
1682 entries.into_iter().map(|entry| {
1683 let map = as_map(&entry)?; let id = map_text(map, "id")?; let version = map_uint(map, "version")?; if version != 1 { return Err(KnoloError::InvalidPack("unsupported V5 event version".into())); }
1684 let transaction_id = map_text(map, "transactionId")?; let actor = map_text(map, "actor")?; let actor_counter = map_uint(map, "actorCounter")?; let kind = map_text(map, "kind")?; let target = map_text(map, "target")?; let payload_id = map_text(map, "payload")?;
1685 let body = CborValue::Map(vec![("actor".into(), CborValue::Text(actor.clone())), ("actorCounter".into(), CborValue::UInt(actor_counter)), ("kind".into(), CborValue::Text(kind.clone())), ("parents".into(), map_value(map, "parents")?.clone()), ("payload".into(), CborValue::Text(payload_id.clone())), ("provenance".into(), map_value(map, "provenance")?.clone()), ("target".into(), CborValue::Text(target.clone())), ("transactionId".into(), CborValue::Text(transaction_id.clone())), ("version".into(), CborValue::UInt(1))]);
1686 if digest_domain("event", &encode_cbor(&body)) != id { return Err(KnoloError::InvalidPack("V5 event identity mismatch".into())); }
1687 Ok(KnowledgeEventV1 { id, transaction_id, actor, actor_counter, kind, target, payload: payload_id })
1688 }).collect()
1689}
1690
1691fn parse_commit(value: &CborValue) -> Result<KnowledgeCommitV1, KnoloError> {
1692 let map = as_map(value)?; let version = map_uint(map, "version")?; if version != 1 { return Err(KnoloError::InvalidPack("unsupported V5 commit version".into())); }
1693 Ok(KnowledgeCommitV1 { state_root: String::new(), commit_digest: String::new(), parents: map_strings(map, "parents")?, object_root: map_text(map, "objectRoot")?, event_root: map_text(map, "eventRoot")?, policy_root: map_text(map, "policyRoot")?, sequence: map_uint(map, "sequence")?, actor: map_text(map, "actor")? })
1694}
1695
1696fn commit_field(value: &CborValue, field: &str) -> Result<String, KnoloError> { map_text(as_map(value)?, field) }
1697
1698fn as_map(value: &CborValue) -> Result<&[(String, CborValue)], KnoloError> { match value { CborValue::Map(entries) => Ok(entries), _ => Err(KnoloError::InvalidPack("expected V5 CBOR map".into())) } }
1699fn map_value<'a>(map: &'a [(String, CborValue)], key: &str) -> Result<&'a CborValue, KnoloError> { map.iter().find(|(name, _)| name == key).map(|(_, value)| value).ok_or_else(|| KnoloError::InvalidPack(format!("missing V5 field: {key}"))) }
1700fn map_text(map: &[(String, CborValue)], key: &str) -> Result<String, KnoloError> { match map_value(map, key)? { CborValue::Text(value) => Ok(value.clone()), _ => Err(KnoloError::InvalidPack(format!("V5 field is not text: {key}"))) } }
1701fn map_uint(map: &[(String, CborValue)], key: &str) -> Result<u64, KnoloError> { match map_value(map, key)? { CborValue::UInt(value) => Ok(*value), _ => Err(KnoloError::InvalidPack(format!("V5 field is not unsigned: {key}"))) } }
1702fn map_bytes(map: &[(String, CborValue)], key: &str) -> Result<Vec<u8>, KnoloError> { match map_value(map, key)? { CborValue::Bytes(value) => Ok(value.clone()), _ => Err(KnoloError::InvalidPack(format!("V5 field is not bytes: {key}"))) } }
1703fn map_strings(map: &[(String, CborValue)], key: &str) -> Result<Vec<String>, KnoloError> { match map_value(map, key)? { CborValue::Array(values) => values.iter().map(|value| match value { CborValue::Text(text) => Ok(text.clone()), _ => Err(KnoloError::InvalidPack("V5 array value is not text".into())) }).collect(), _ => Err(KnoloError::InvalidPack(format!("V5 field is not an array: {key}"))) } }
1704
1705fn decode_cbor_exact(bytes: &[u8]) -> Result<CborValue, KnoloError> { let mut cursor = 0; let value = decode_cbor(bytes, &mut cursor)?; if cursor != bytes.len() { return Err(KnoloError::InvalidPack("trailing V5 CBOR bytes".into())); } Ok(value) }
1706fn decode_cbor(bytes: &[u8], cursor: &mut usize) -> Result<CborValue, KnoloError> {
1707 let initial = *bytes.get(*cursor).ok_or_else(|| KnoloError::InvalidPack("truncated V5 CBOR".into()))?; *cursor += 1; let major = initial >> 5; let ai = initial & 31;
1708 let length = |cursor: &mut usize| read_cbor_length(bytes, cursor, ai);
1709 match major { 0 => Ok(CborValue::UInt(length(cursor)?)), 1 => { let value = length(cursor)?; if value > i64::MAX as u64 { return Err(KnoloError::InvalidPack("V5 negative integer overflow".into())); } Ok(CborValue::NInt(-1 - value as i64)) }, 2 => { let len = usize::try_from(length(cursor)?).map_err(|_| KnoloError::InvalidPack("V5 bytes length overflow".into()))?; Ok(CborValue::Bytes(read_slice(bytes, cursor, len)?.to_vec())) }, 3 => { let len = usize::try_from(length(cursor)?).map_err(|_| KnoloError::InvalidPack("V5 text length overflow".into()))?; let value = std::str::from_utf8(read_slice(bytes, cursor, len)?).map_err(|_| KnoloError::InvalidPack("V5 text is not UTF-8".into()))?; Ok(CborValue::Text(value.to_string())) }, 4 => { let len = usize::try_from(length(cursor)?).map_err(|_| KnoloError::InvalidPack("V5 array length overflow".into()))?; let mut values = Vec::with_capacity(len); for _ in 0..len { values.push(decode_cbor(bytes, cursor)?); } Ok(CborValue::Array(values)) }, 5 => { let len = usize::try_from(length(cursor)?).map_err(|_| KnoloError::InvalidPack("V5 map length overflow".into()))?; let mut values = Vec::with_capacity(len); for _ in 0..len { let key = match decode_cbor(bytes, cursor)? { CborValue::Text(value) => value, _ => return Err(KnoloError::InvalidPack("V5 map key is not text".into())) }; if values.iter().any(|(name, _): &(String, CborValue)| name == &key) { return Err(KnoloError::InvalidPack("duplicate V5 map key".into())); } values.push((key, decode_cbor(bytes, cursor)?)); } Ok(CborValue::Map(values)) }, 7 if ai == 20 => Ok(CborValue::Bool(false)), 7 if ai == 21 => Ok(CborValue::Bool(true)), 7 if ai == 22 => Ok(CborValue::Null), _ => Err(KnoloError::InvalidPack("unsupported V5 CBOR type".into())) }
1710}
1711fn read_cbor_length(bytes: &[u8], cursor: &mut usize, ai: u8) -> Result<u64, KnoloError> { match ai { 0..=23 => Ok(ai as u64), 24 => Ok(read_slice(bytes, cursor, 1)?[0] as u64), 25 => Ok(u16::from_be_bytes(read_slice(bytes, cursor, 2)?.try_into().unwrap()) as u64), 26 => Ok(u32::from_be_bytes(read_slice(bytes, cursor, 4)?.try_into().unwrap()) as u64), 27 => Ok(u64::from_be_bytes(read_slice(bytes, cursor, 8)?.try_into().unwrap())), _ => Err(KnoloError::InvalidPack("indefinite V5 CBOR is not allowed".into())) } }
1712
1713fn encode_cbor(value: &CborValue) -> Vec<u8> { let mut out = Vec::new(); encode_cbor_into(value, &mut out); out }
1714fn encode_cbor_into(value: &CborValue, out: &mut Vec<u8>) { match value { CborValue::Null => out.push(0xf6), CborValue::Bool(false) => out.push(0xf4), CborValue::Bool(true) => out.push(0xf5), CborValue::UInt(value) => encode_cbor_length(0, *value, out), CborValue::NInt(value) => encode_cbor_length(1, (-1 - *value) as u64, out), CborValue::Bytes(value) => { encode_cbor_length(2, value.len() as u64, out); out.extend_from_slice(value); }, CborValue::Text(value) => { encode_cbor_length(3, value.len() as u64, out); out.extend_from_slice(value.as_bytes()); }, CborValue::Array(values) => { encode_cbor_length(4, values.len() as u64, out); for value in values { encode_cbor_into(value, out); } }, CborValue::Map(values) => { let mut sorted = values.iter().collect::<Vec<_>>(); sorted.sort_by(|left, right| left.0.as_bytes().cmp(right.0.as_bytes())); encode_cbor_length(5, sorted.len() as u64, out); for (key, value) in sorted { encode_cbor_into(&CborValue::Text(key.clone()), out); encode_cbor_into(value, out); } } } }
1715fn encode_cbor_length(major: u8, value: u64, out: &mut Vec<u8>) { if value < 24 { out.push((major << 5) | value as u8); } else if value <= u8::MAX as u64 { out.extend_from_slice(&[(major << 5) | 24, value as u8]); } else if value <= u16::MAX as u64 { out.push((major << 5) | 25); out.extend_from_slice(&(value as u16).to_be_bytes()); } else if value <= u32::MAX as u64 { out.push((major << 5) | 26); out.extend_from_slice(&(value as u32).to_be_bytes()); } else { out.push((major << 5) | 27); out.extend_from_slice(&value.to_be_bytes()); } }
1716
1717fn digest_domain(domain: &str, payload: &[u8]) -> String { let mut input = format!("knolo:{domain}:v1\0").into_bytes(); input.extend_from_slice(payload); digest_from_raw(&sha256(&input)) }
1718fn digest_from_raw(raw: &[u8]) -> String { format!("sha256-{}", raw.iter().map(|byte| format!("{byte:02x}")).collect::<String>()) }
1719fn digest_raw(digest: &str) -> Vec<u8> { let hex = digest.strip_prefix("sha256-").unwrap_or(""); (0..hex.len()).step_by(2).filter_map(|i| u8::from_str_radix(hex.get(i..i + 2)?, 16).ok()).collect() }
1720
1721fn le_u16(bytes: &[u8], offset: usize) -> Result<u16, KnoloError> { Ok(u16::from_le_bytes(read_slice(bytes, &mut offset.clone(), 2)?.try_into().unwrap())) }
1722
1723fn sha256(input: &[u8]) -> [u8; 32] {
1725 const K: [u32; 64] = [
1726 0x428a2f98,0x71374491,0xb5c0fbcf,0xe9b5dba5,0x3956c25b,0x59f111f1,0x923f82a4,0xab1c5ed5,
1727 0xd807aa98,0x12835b01,0x243185be,0x550c7dc3,0x72be5d74,0x80deb1fe,0x9bdc06a7,0xc19bf174,
1728 0xe49b69c1,0xefbe4786,0x0fc19dc6,0x240ca1cc,0x2de92c6f,0x4a7484aa,0x5cb0a9dc,0x76f988da,
1729 0x983e5152,0xa831c66d,0xb00327c8,0xbf597fc7,0xc6e00bf3,0xd5a79147,0x06ca6351,0x14292967,
1730 0x27b70a85,0x2e1b2138,0x4d2c6dfc,0x53380d13,0x650a7354,0x766a0abb,0x81c2c92e,0x92722c85,
1731 0xa2bfe8a1,0xa81a664b,0xc24b8b70,0xc76c51a3,0xd192e819,0xd6990624,0xf40e3585,0x106aa070,
1732 0x19a4c116,0x1e376c08,0x2748774c,0x34b0bcb5,0x391c0cb3,0x4ed8aa4a,0x5b9cca4f,0x682e6ff3,
1733 0x748f82ee,0x78a5636f,0x84c87814,0x8cc70208,0x90befffa,0xa4506ceb,0xbef9a3f7,0xc67178f2,
1734 ];
1735 let mut data = input.to_vec(); let bit_len = (data.len() as u64) * 8; data.push(0x80); while data.len() % 64 != 56 { data.push(0); } data.extend_from_slice(&bit_len.to_be_bytes());
1736 let mut h: [u32; 8] = [0x6a09e667,0xbb67ae85,0x3c6ef372,0xa54ff53a,0x510e527f,0x9b05688c,0x1f83d9ab,0x5be0cd19];
1737 for chunk in data.chunks_exact(64) { let mut w = [0u32; 64]; for i in 0..16 { w[i] = u32::from_be_bytes(chunk[i*4..i*4+4].try_into().unwrap()); } for i in 16..64 { let s0 = w[i-15].rotate_right(7) ^ w[i-15].rotate_right(18) ^ (w[i-15] >> 3); let s1 = w[i-2].rotate_right(17) ^ w[i-2].rotate_right(19) ^ (w[i-2] >> 10); w[i] = w[i-16].wrapping_add(s0).wrapping_add(w[i-7]).wrapping_add(s1); } let mut a=h[0]; let mut b=h[1]; let mut c=h[2]; let mut d=h[3]; let mut e=h[4]; let mut f=h[5]; let mut g=h[6]; let mut hh=h[7]; for i in 0..64 { let s1=e.rotate_right(6)^e.rotate_right(11)^e.rotate_right(25); let ch=(e&f)^((!e)&g); let t1=hh.wrapping_add(s1).wrapping_add(ch).wrapping_add(K[i]).wrapping_add(w[i]); let s0=a.rotate_right(2)^a.rotate_right(13)^a.rotate_right(22); let maj=(a&b)^(a&c)^(b&c); let t2=s0.wrapping_add(maj); hh=g; g=f; f=e; e=d.wrapping_add(t1); d=c; c=b; b=a; a=t1.wrapping_add(t2); } h[0]=h[0].wrapping_add(a); h[1]=h[1].wrapping_add(b); h[2]=h[2].wrapping_add(c); h[3]=h[3].wrapping_add(d); h[4]=h[4].wrapping_add(e); h[5]=h[5].wrapping_add(f); h[6]=h[6].wrapping_add(g); h[7]=h[7].wrapping_add(hh); }
1738 let mut out = [0u8; 32]; for (i, value) in h.iter().enumerate() { out[i*4..i*4+4].copy_from_slice(&value.to_be_bytes()); } out
1739}