fstool 0.4.1

Build disk images and filesystems (ext2/3/4, MBR, GPT) from a directory tree and TOML spec, in the spirit of genext2fs.
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
//! ext4 HTree (`COMPAT_DIR_INDEX`) write-side support.
//!
//! ## What an indexed directory looks like on disk
//!
//! A regular ext4 directory is a flat sequence of `ext4_dir_entry_2`
//! records concatenated across one or more data blocks. Lookup is O(n).
//!
//! With `COMPAT_DIR_INDEX` the first data block of the directory is a
//! special **dx_root** block whose payload layers a hash-index table
//! over what otherwise look like two ordinary dirent records (`.` and
//! `..`). Bytes 0..12 hold `.`, 12..24 hold `..`, 24..32 hold a tiny
//! `dx_root_info` header, and 32.. is an array of [`DxEntry`] slots
//! `{hash, block}`. Slot 0 is overloaded as `{limit, count, leaf_block}`
//! (the `dx_countlimit` view); slots 1..count carry real `{hash, block}`
//! pairs.
//!
//! Lookups walk the dx_entry table by binary search on `hash`, then dive
//! into the leaf block whose range covers that hash. The leaf block is
//! a perfectly ordinary `ext4_dir_entry_2` block — readers that don't
//! grok HTree simply ignore the root's `dx_*` overlay (the `.` / `..`
//! façade lets them treat the root as a normal dir block whose data
//! ends after `..`) and linear-scan every leaf, which still finds every
//! entry. The cost is just O(n) where O(log n) was on offer.
//!
//! The "fake `.` / `..` façade" trick is also why setting
//! `COMPAT_DIR_INDEX` on a filesystem doesn't force *every* directory
//! to be indexed — un-indexed dirs are still valid; only those whose
//! inode carries `EXT4_INDEX_FL` are interpreted as HTree.
//!
//! ## Hashing
//!
//! Names hash through a half-rounds MD4 (Linux's `DX_HASH_HALF_MD4`),
//! which produces a 32-bit major and 32-bit minor hash. We emit
//! `DX_HASH_HALF_MD4_UNSIGNED` (`hash_version = 1` per the kernel's
//! enum but variant `1` matches the *unsigned* path under modern
//! e2fsprogs — see `linux/include/linux/dx_hash.h`). The low bit of
//! the major hash is reserved as a collision-chain marker and is
//! always cleared; the value `0xFFFF_FFFE` is a sentinel meaning EOF
//! so the hasher remaps it to `0xFFFF_FFFC` before returning.
//!
//! ## metadata_csum
//!
//! dx_root and dx_node blocks carry an 8-byte `dx_tail` at the very
//! end (`dt_reserved` + `dt_checksum`). The csum is over a *subset* of
//! the block — only the slots that are actually in use — not the full
//! 4 KiB. See [`stamp_dx_csum`].
//!
//! ## Scope (v1 writer)
//!
//! - Single-level only (`indirect_levels = 0`). With 4 KiB blocks and
//!   metadata_csum on, a dx_root holds 507 leaf slots × ~250 entries
//!   per leaf ≈ 127 K entries in a single directory — well past
//!   anything a normal rootfs throws at us. Multi-level (dx_node
//!   intermediate blocks) is deferred.
//! - HALF_MD4_UNSIGNED hash only. `tea`, `siphash`, and the legacy
//!   `dx_hack_hash` are not emitted on the write side (the reader,
//!   when added, will need to handle whatever it sees).

use super::constants::DENT_DIR;

/// Hash-version selector for the `dx_root_info.hash_version` field.
/// Matches the constants in `linux/fs/ext4/ext4.h` / `tune2fs(8)`.
pub const DX_HASH_LEGACY: u8 = 0;
pub const DX_HASH_HALF_MD4: u8 = 1;
pub const DX_HASH_TEA: u8 = 2;
pub const DX_HASH_LEGACY_UNSIGNED: u8 = 3;
pub const DX_HASH_HALF_MD4_UNSIGNED: u8 = 4;
pub const DX_HASH_TEA_UNSIGNED: u8 = 5;
pub const DX_HASH_SIPHASH: u8 = 6;

/// Size of the `dx_tail` checksum footer in bytes.
pub const DX_TAIL_LEN: usize = 8;

/// Size of one `dx_entry` slot (also the size of `dx_countlimit` since
/// they overlay).
pub const DX_ENTRY_LEN: usize = 8;

/// Static bytes consumed by the dx_root prefix before the dx_entry
/// table: `.` (12) + `..` (12) + `dx_root_info` (8).
pub const DX_ROOT_HEADER_LEN: usize = 32;

/// Static bytes consumed by the dx_node prefix before the dx_entry
/// table: a single 8-byte fake dirent (inode + rec_len + name_len +
/// file_type) with `name_len = 0` so no name bytes follow. `rec_len`
/// spans the whole block, so legacy linear-walk readers consume one
/// dirent and stop. The dx_entry table follows immediately after,
/// matching the kernel's `struct dx_node { struct fake_dirent fake;
/// struct dx_entry entries[0]; }` layout.
pub const DX_NODE_HEADER_LEN: usize = 8;

/// One slot in the dx_root / dx_node entry table.
#[derive(Debug, Clone, Copy)]
pub struct DxEntry {
    /// First major hash covered by the leaf at `block`. For slot 0 the
    /// hash field overlays `dx_countlimit` and is otherwise unused as a
    /// hash key (the leftmost leaf is implicit "hash ≥ 0").
    pub hash: u32,
    /// Logical block number of the leaf (or, for multi-level trees,
    /// the child dx_node) within the directory.
    pub block: u32,
}

/// Number of dx_entry slots a dx_root with the given block size can
/// hold, after subtracting the 32-byte prefix and the optional 8-byte
/// `dx_tail`.
pub fn dx_root_limit(block_size: u32, csum_tail: bool) -> usize {
    let mut entry_space = block_size as usize - DX_ROOT_HEADER_LEN;
    if csum_tail {
        entry_space -= DX_TAIL_LEN;
    }
    entry_space / DX_ENTRY_LEN
}

/// Number of dx_entry slots a dx_node holds, after subtracting the
/// 12-byte fake-dirent prefix and the optional 8-byte dx_tail.
/// Larger than `dx_root_limit` because dx_node skips the `..` +
/// dx_root_info block (20 bytes of overhead).
pub fn dx_node_limit(block_size: u32, csum_tail: bool) -> usize {
    let mut entry_space = block_size as usize - DX_NODE_HEADER_LEN;
    if csum_tail {
        entry_space -= DX_TAIL_LEN;
    }
    entry_space / DX_ENTRY_LEN
}

/// Maximum number of real dirent records that fit in a regular leaf
/// block of size `bs` with metadata_csum's 12-byte trailing dirent
/// taken into account. The leaf block format is identical to a plain
/// ext4 directory data block (no dx_* overlay), so the regular
/// `usable_dir_len` rule applies.
pub fn leaf_max_entries(_block_size: u32) -> usize {
    // We don't actually cap by count — what matters is byte
    // occupancy. This helper exists for documentation symmetry with
    // dx_root_limit and is unused inside this module; entry-packing
    // code consults `dir::usable_dir_len` directly.
    usize::MAX
}

// ─── DX_HASH_HALF_MD4 ───────────────────────────────────────────────

/// Compute the (major, minor) HTree hash of `name` under
/// `DX_HASH_HALF_MD4_UNSIGNED`. Matches the kernel's `ext4fs_dirhash`
/// for the unsigned variant: the seed defaults to the MD4 IV, the name
/// is padded into 32-byte chunks via `str2hashbuf`, each chunk is mixed
/// in with `half_md4_transform`, then the major hash is `buf[1]` and
/// the minor hash is `buf[2]`. The low bit of the major hash is
/// cleared (collision-chain marker reservation); the EOF sentinel
/// `0xFFFF_FFFE` is remapped to `0xFFFF_FFFC`.
pub fn half_md4_hash(name: &[u8]) -> (u32, u32) {
    // Default MD4 seed (no per-FS hash-seed override yet).
    let mut buf: [u32; 4] = [0x67452301, 0xefcdab89, 0x98badcfe, 0x10325476];
    let mut remaining = name;
    let mut in_buf = [0u32; 8];
    // Matches the kernel's `while (len > 0)` — empty names skip the
    // transform entirely so their hash is the raw MD4 IV's middle
    // words.
    while !remaining.is_empty() {
        str2hashbuf_unsigned(remaining, &mut in_buf, 8);
        half_md4_transform(&mut buf, &in_buf);
        if remaining.len() <= 32 {
            break;
        }
        remaining = &remaining[32..];
    }
    let mut hash = buf[1] & !1; // clear collision-chain marker
    if hash == 0xFFFF_FFFE {
        hash = 0xFFFF_FFFC;
    }
    (hash, buf[2])
}

/// Pack the next ≤ 32 bytes of `msg` into `out[..num]` as u32 words
/// for half-MD4 input. Matches the kernel's `str2hashbuf_unsigned`
/// exactly — including its quirk of carrying the held partial word
/// into the trailing slot (so a length-5 name still has its 5th byte
/// fold into the hash, just in the slot following the first full
/// word). The signed variant differs by interpreting bytes as `i8`;
/// modern Linux defaults to unsigned and so do we.
///
/// The pad word `pad = len_byte | len_byte << 8 | len_byte << 16 |
/// len_byte << 24` fills any position the actual name doesn't reach,
/// distinguishing two short names of different lengths from each
/// other (both would otherwise hash on identical prefixes).
fn str2hashbuf_unsigned(msg: &[u8], out: &mut [u32; 8], num: usize) {
    let mut len = msg.len();
    let len_byte = (len & 0xff) as u32;
    let pad = len_byte | (len_byte << 8) | (len_byte << 16) | (len_byte << 24);
    let mut val = pad;
    let mut written = 0usize;
    let mut num_remaining = num;
    if len > num * 4 {
        len = num * 4;
    }
    for (i, &b) in msg.iter().take(len).enumerate() {
        // Kernel: val = ((int) ucp[i]) + (val << 8)
        val = (b as u32).wrapping_add(val.wrapping_shl(8));
        if i % 4 == 3 {
            out[written] = val;
            written += 1;
            val = pad;
            num_remaining -= 1;
        }
    }
    // Trailing slots: write the held partial val once (preserving any
    // unflushed bytes from a non-aligned tail), then fill the rest
    // with pad. e2fsprogs 1.47 expresses this as:
    //   if (--num >= 0) *buf++ = val;
    //   while (--num >= 0) *buf++ = pad;
    if num_remaining > 0 {
        out[written] = val;
        written += 1;
        num_remaining -= 1;
    }
    while num_remaining > 0 {
        out[written] = pad;
        written += 1;
        num_remaining -= 1;
    }
}

/// Half-MD4 transform: three rounds of MD4 mixing using the input as
/// 8 u32 words (only half the standard MD4 input size — the "half"
/// in the name).
///
/// Reference: Linux `fs/ext4/hash.c::half_md4_transform`.
fn half_md4_transform(buf: &mut [u32; 4], i: &[u32; 8]) {
    let (mut a, mut b, mut c, mut d) = (buf[0], buf[1], buf[2], buf[3]);

    // Round 1: F(x,y,z) = (x & y) | (!x & z)
    a = round_f(a, b, c, d, i[0], 3);
    d = round_f(d, a, b, c, i[1], 7);
    c = round_f(c, d, a, b, i[2], 11);
    b = round_f(b, c, d, a, i[3], 19);
    a = round_f(a, b, c, d, i[4], 3);
    d = round_f(d, a, b, c, i[5], 7);
    c = round_f(c, d, a, b, i[6], 11);
    b = round_f(b, c, d, a, i[7], 19);

    // Round 2: G(x,y,z) = (x & y) | (x & z) | (y & z); +K2
    a = round_g(a, b, c, d, i[1], 3);
    d = round_g(d, a, b, c, i[3], 5);
    c = round_g(c, d, a, b, i[5], 9);
    b = round_g(b, c, d, a, i[7], 13);
    a = round_g(a, b, c, d, i[0], 3);
    d = round_g(d, a, b, c, i[2], 5);
    c = round_g(c, d, a, b, i[4], 9);
    b = round_g(b, c, d, a, i[6], 13);

    // Round 3: H(x,y,z) = x ^ y ^ z; +K3
    a = round_h(a, b, c, d, i[3], 3);
    d = round_h(d, a, b, c, i[7], 9);
    c = round_h(c, d, a, b, i[2], 11);
    b = round_h(b, c, d, a, i[6], 15);
    a = round_h(a, b, c, d, i[1], 3);
    d = round_h(d, a, b, c, i[5], 9);
    c = round_h(c, d, a, b, i[0], 11);
    b = round_h(b, c, d, a, i[4], 15);

    buf[0] = buf[0].wrapping_add(a);
    buf[1] = buf[1].wrapping_add(b);
    buf[2] = buf[2].wrapping_add(c);
    buf[3] = buf[3].wrapping_add(d);
}

#[inline]
fn round_f(a: u32, b: u32, c: u32, d: u32, x: u32, s: u32) -> u32 {
    let f = (b & c) | ((!b) & d);
    a.wrapping_add(f).wrapping_add(x).rotate_left(s)
}

#[inline]
fn round_g(a: u32, b: u32, c: u32, d: u32, x: u32, s: u32) -> u32 {
    let g = (b & c) | (b & d) | (c & d);
    const K2: u32 = 0x5a827999;
    a.wrapping_add(g)
        .wrapping_add(x)
        .wrapping_add(K2)
        .rotate_left(s)
}

#[inline]
fn round_h(a: u32, b: u32, c: u32, d: u32, x: u32, s: u32) -> u32 {
    let h = b ^ c ^ d;
    const K3: u32 = 0x6ed9eba1;
    a.wrapping_add(h)
        .wrapping_add(x)
        .wrapping_add(K3)
        .rotate_left(s)
}

// ─── dx_root encoding ───────────────────────────────────────────────

/// Build the dx_root block for an indexed directory.
///
/// `self_inode` / `parent_inode` populate the fake `.` / `..` entries
/// at the head of the block. `hash_version` is the algorithm we used
/// when bucketing names into `entries` (see [`half_md4_hash`]).
/// `entries[0]` is the dx_countlimit slot — its `hash` field is the
/// packed `(limit, count)` value, its `block` field is the leftmost
/// leaf block number. `entries[1..]` are real `{hash, block}` rows
/// sorted by ascending hash.
///
/// When `csum_tail` is set, the final 8 bytes are reserved for the
/// dx_tail (zero-stamped here; the actual CRC32C is computed at flush
/// time by [`compute_dx_csum`]).
#[allow(clippy::too_many_arguments)] // mirrors the e2fsprogs layout struct
pub fn make_dx_root_block(
    self_inode: u32,
    parent_inode: u32,
    block_size: u32,
    hash_version: u8,
    indirect_levels: u8,
    entries: &[DxEntry],
    with_filetype: bool,
    csum_tail: bool,
) -> Vec<u8> {
    assert!(
        !entries.is_empty(),
        "dx_root must have at least the countlimit slot"
    );
    let mut buf = vec![0u8; block_size as usize];

    // Fake "." dirent (offset 0..12).
    write_fake_dirent(&mut buf[0..12], self_inode, b".", with_filetype);

    // Fake ".." dirent (offset 12). rec_len spans from offset 12 to
    // the END of the block — always, regardless of csum_tail. The
    // dx_entry table and (when present) the 8-byte dx_tail both live
    // INSIDE this declared rec_len so a legacy linear-walk reader sees
    // ".." absorb everything and never tries to decode the dx_entries
    // (or dx_tail) as a separate dirent. The kernel's HTree-aware
    // code path indexes into the dx_entry array directly and ignores
    // the rec_len fiction.
    let _ = csum_tail; // (kept for API symmetry with regular dir block)
    let dotdot_rec_len = (block_size as usize - 12) as u16;
    let mut dotdot = vec![0u8; 12];
    dotdot[0..4].copy_from_slice(&parent_inode.to_le_bytes());
    dotdot[4..6].copy_from_slice(&dotdot_rec_len.to_le_bytes());
    if with_filetype {
        dotdot[6] = 2; // name_len
        dotdot[7] = DENT_DIR;
    } else {
        dotdot[6..8].copy_from_slice(&2u16.to_le_bytes());
    }
    dotdot[8] = b'.';
    dotdot[9] = b'.';
    buf[12..24].copy_from_slice(&dotdot);

    // dx_root_info (offset 24..32).
    //   0..4 reserved_zero (le32, must be 0)
    //   4    hash_version
    //   5    info_length (== 8)
    //   6    indirect_levels (== 0 in v1; multi-level deferred)
    //   7    unused_flags (0)
    buf[24..28].copy_from_slice(&0u32.to_le_bytes());
    buf[28] = hash_version;
    buf[29] = 8;
    buf[30] = indirect_levels;
    buf[31] = 0; // unused_flags

    // dx_entry table starting at offset 32. entries[0] is the
    // dx_countlimit slot: low 16 bits of the "hash" field are `limit`,
    // high 16 bits are `count`. Its `block` field is the leaf block
    // for the leftmost (hash=0) range.
    for (i, e) in entries.iter().enumerate() {
        let off = DX_ROOT_HEADER_LEN + i * DX_ENTRY_LEN;
        buf[off..off + 4].copy_from_slice(&e.hash.to_le_bytes());
        buf[off + 4..off + 8].copy_from_slice(&e.block.to_le_bytes());
    }

    // dx_tail placeholder when metadata_csum is on; the 4-byte CRC is
    // stamped at flush time.
    if csum_tail {
        let tail_off = block_size as usize - DX_TAIL_LEN;
        for b in buf[tail_off..].iter_mut() {
            *b = 0;
        }
    }
    buf
}

/// Build a dx_node intermediate block — used at `indirect_levels = 1`
/// to fan dx_root entries out to multiple leaves. Layout: a single
/// 12-byte fake dirent (inode=0, rec_len=block_size, name_len=0) so
/// legacy linear-walk readers stop immediately at the dirent
/// boundary, followed by a `dx_countlimit` slot and the real
/// `dx_entry` rows that map a hash range to a leaf block.
///
/// `entries[0]` is the dx_countlimit slot — its `hash` field is the
/// packed `(limit, count)` value, its `block` field is the leftmost
/// leaf at this node. `entries[1..]` are real `{hash, block}` rows
/// sorted by ascending hash. Same convention as
/// [`make_dx_root_block`].
pub fn make_dx_node_block(block_size: u32, entries: &[DxEntry], csum_tail: bool) -> Vec<u8> {
    assert!(
        !entries.is_empty(),
        "dx_node must have at least the countlimit slot"
    );
    let mut buf = vec![0u8; block_size as usize];
    // Single 8-byte fake dirent at offset 0, spanning the whole block
    // so a linear-walk reader consumes one dirent and stops. The
    // dx_entry table follows immediately at offset 8.
    buf[0..4].copy_from_slice(&0u32.to_le_bytes()); // inode = 0
    buf[4..6].copy_from_slice(&(block_size as u16).to_le_bytes()); // rec_len
    buf[6] = 0; // name_len
    buf[7] = 0; // file_type
    for (i, e) in entries.iter().enumerate() {
        let off = DX_NODE_HEADER_LEN + i * DX_ENTRY_LEN;
        buf[off..off + 4].copy_from_slice(&e.hash.to_le_bytes());
        buf[off + 4..off + 8].copy_from_slice(&e.block.to_le_bytes());
    }
    if csum_tail {
        let tail_off = block_size as usize - DX_TAIL_LEN;
        for b in buf[tail_off..].iter_mut() {
            *b = 0;
        }
    }
    buf
}

/// Pack the (limit, count) pair into the low/high halves of the
/// dx_countlimit slot's "hash" field. Use as `entries[0].hash`.
pub fn pack_countlimit(limit: u16, count: u16) -> u32 {
    (limit as u32) | ((count as u32) << 16)
}

/// Compute the dx_root / dx_node CRC32C over the in-use prefix of the
/// block: `count_offset + count * DX_ENTRY_LEN` bytes, then the 4-byte
/// `dt_reserved` field (zero), then 4 bytes of zero standing in for
/// `dt_checksum`. Mirrors the kernel's `ext4_dx_csum`.
pub fn compute_dx_csum(
    raw_update: impl Fn(u32, &[u8]) -> u32,
    seed: u32,
    inode_num: u32,
    inode_generation: u32,
    block: &[u8],
    count_offset: usize,
    count: usize,
) -> u32 {
    // Per-inode seed (kernel's `i_csum_seed`).
    let c = raw_update(seed, &inode_num.to_le_bytes());
    let c = raw_update(c, &inode_generation.to_le_bytes());
    // The in-use prefix.
    let used_len = count_offset + count * DX_ENTRY_LEN;
    let c = raw_update(c, &block[..used_len]);
    // dt_reserved (4 bytes of zero).
    let c = raw_update(c, &0u32.to_le_bytes());
    // dt_checksum placeholder (4 bytes of zero — checksum is computed
    // OVER the field treated as zero, then stamped back into it).
    raw_update(c, &0u32.to_le_bytes())
}

/// Stamp `csum` into the `dt_checksum` field at the very end of the
/// block. Pairs with [`compute_dx_csum`].
pub fn stamp_dx_csum(block: &mut [u8], csum: u32) {
    let n = block.len();
    block[n - 4..].copy_from_slice(&csum.to_le_bytes());
}

fn write_fake_dirent(dst: &mut [u8], inode: u32, name: &[u8], with_filetype: bool) {
    assert!(dst.len() >= 12, "fake dirent slot must be 12 bytes");
    dst[0..4].copy_from_slice(&inode.to_le_bytes());
    dst[4..6].copy_from_slice(&12u16.to_le_bytes());
    if with_filetype {
        dst[6] = name.len() as u8;
        dst[7] = DENT_DIR;
    } else {
        dst[6..8].copy_from_slice(&(name.len() as u16).to_le_bytes());
    }
    dst[8..8 + name.len()].copy_from_slice(name);
    for b in dst[8 + name.len()..12].iter_mut() {
        *b = 0;
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    /// dx_root layout sanity: ".", "..", info header.
    #[test]
    fn dx_root_header_layout() {
        let entries = vec![DxEntry {
            hash: pack_countlimit(508, 1),
            block: 1,
        }];
        let buf = make_dx_root_block(
            12,
            2,
            4096,
            DX_HASH_HALF_MD4_UNSIGNED,
            0,
            &entries,
            true,
            false,
        );
        // "." at offset 0
        assert_eq!(u32::from_le_bytes(buf[0..4].try_into().unwrap()), 12);
        assert_eq!(u16::from_le_bytes(buf[4..6].try_into().unwrap()), 12);
        assert_eq!(buf[6], 1); // name_len
        assert_eq!(buf[7], DENT_DIR);
        assert_eq!(buf[8], b'.');
        // ".." at offset 12, rec_len = 4096 - 12 (no csum tail).
        assert_eq!(u32::from_le_bytes(buf[12..16].try_into().unwrap()), 2);
        assert_eq!(
            u16::from_le_bytes(buf[16..18].try_into().unwrap()),
            (4096 - 12) as u16
        );
        // dx_root_info at offset 24.
        assert_eq!(u32::from_le_bytes(buf[24..28].try_into().unwrap()), 0);
        assert_eq!(buf[28], DX_HASH_HALF_MD4_UNSIGNED);
        assert_eq!(buf[29], 8); // info_length
        assert_eq!(buf[30], 0); // indirect_levels
        // entries[0] (the countlimit slot) at offset 32.
        let cl_hash = u32::from_le_bytes(buf[32..36].try_into().unwrap());
        assert_eq!(cl_hash & 0xffff, 508);
        assert_eq!(cl_hash >> 16, 1);
        assert_eq!(u32::from_le_bytes(buf[36..40].try_into().unwrap()), 1);
    }

    /// Csum-tail variant: ".." rec_len shrinks by 8 to leave room for dx_tail,
    /// and the dx_root_limit drops by one slot.
    #[test]
    fn dx_root_limit_metadata_csum() {
        assert_eq!(dx_root_limit(4096, false), (4096 - 32) / 8);
        assert_eq!(dx_root_limit(4096, true), (4096 - 32 - 8) / 8);
        assert_eq!(dx_root_limit(1024, true), (1024 - 32 - 8) / 8);
    }

    /// Empty-name hash should be deterministic and ≠ 0.
    #[test]
    fn half_md4_empty_name() {
        let (h, _) = half_md4_hash(b"");
        // Low bit always cleared.
        assert_eq!(h & 1, 0);
    }

    /// Two different names must produce different major hashes (with
    /// overwhelming probability — this is a smoke test, not a
    /// collision-resistance proof).
    #[test]
    fn half_md4_different_names_different_hashes() {
        let (h1, _) = half_md4_hash(b"foo");
        let (h2, _) = half_md4_hash(b"bar");
        let (h3, _) = half_md4_hash(b"foobar");
        assert_ne!(h1, h2);
        assert_ne!(h1, h3);
        assert_ne!(h2, h3);
    }

    /// All hashes must have the low bit clear (collision-chain marker).
    #[test]
    fn half_md4_low_bit_always_clear() {
        for n in 0u32..32 {
            let name = format!("name{n:04}");
            let (h, _) = half_md4_hash(name.as_bytes());
            assert_eq!(h & 1, 0, "hash {h:#x} of {name:?} has low bit set");
        }
    }

    /// EOF sentinel must not be produced.
    #[test]
    fn half_md4_eof_sentinel_remapped() {
        // We can't easily force the algorithm to produce 0xFFFFFFFE
        // for an arbitrary name, but the remap path is exercised
        // by the code itself; here we just confirm the post-condition.
        for n in 0u32..2000 {
            let name = format!("entry_{n}");
            let (h, _) = half_md4_hash(name.as_bytes());
            assert_ne!(h, 0xFFFF_FFFE);
        }
    }

    /// countlimit packing round-trips.
    #[test]
    fn countlimit_packing() {
        let v = pack_countlimit(508, 2);
        assert_eq!(v & 0xffff, 508);
        assert_eq!(v >> 16, 2);
    }

    /// dx_node_limit must be larger than dx_root_limit at the same block
    /// size: dx_node skips the 20 extra bytes of ./.. + dx_root_info,
    /// trading them for two more dx_entry slots at 4 KiB blocks.
    #[test]
    fn dx_node_limit_larger_than_root_limit() {
        for &bs in &[1024u32, 2048, 4096] {
            for &csum in &[false, true] {
                let r = dx_root_limit(bs, csum);
                let n = dx_node_limit(bs, csum);
                assert!(
                    n > r,
                    "dx_node should fit more slots than dx_root at bs={bs} csum={csum}: \
                     root={r} node={n}"
                );
            }
        }
    }

    /// Build a depth-1 layout — dx_root → dx_node → leaf — by hand and
    /// route a sample hash through it via the same byte-level scheme
    /// the writer uses, confirming the two-tier table walks land on
    /// the expected leaf block. Verifies dx_node header offset (12),
    /// dx_countlimit field layout, and the dx_lookup_logical contract
    /// in `mod.rs` (not directly callable here, so the test inlines
    /// the same arithmetic).
    #[test]
    fn dx_node_routing_two_level_layout() {
        let bs: u32 = 4096;
        // Sample hash boundaries: dx_root has 2 dx_node slots
        // (countlimit + slot1), dx_node has 3 leaf slots
        // (countlimit + slot1 + slot2).
        //
        //   dx_root entries:
        //     slot 0: countlimit, block = node0 (logical 1)
        //     slot 1: hash = 0x50000000, block = node1 (logical 2)
        //
        //   dx_node #0 entries (covers hashes 0..0x50000000):
        //     slot 0: countlimit, block = leaf_a (logical 3)
        //     slot 1: hash = 0x20000000, block = leaf_b (logical 4)
        //     slot 2: hash = 0x40000000, block = leaf_c (logical 5)
        //
        //   dx_node #1 entries (covers hashes 0x50000000..):
        //     slot 0: countlimit, block = leaf_d (logical 6)
        //     slot 1: hash = 0x80000000, block = leaf_e (logical 7)
        //     slot 2: hash = 0xC0000000, block = leaf_f (logical 8)
        let root_entries = vec![
            DxEntry {
                hash: pack_countlimit(dx_root_limit(bs, true) as u16, 2),
                block: 1,
            },
            DxEntry {
                hash: 0x50000000,
                block: 2,
            },
        ];
        let root = make_dx_root_block(2, 2, bs, DX_HASH_HALF_MD4, 1, &root_entries, true, true);
        // Confirm indirect_levels stamped at offset 30.
        assert_eq!(root[30], 1);

        let node0_entries = vec![
            DxEntry {
                hash: pack_countlimit(dx_node_limit(bs, true) as u16, 3),
                block: 3,
            },
            DxEntry {
                hash: 0x20000000,
                block: 4,
            },
            DxEntry {
                hash: 0x40000000,
                block: 5,
            },
        ];
        let node0 = make_dx_node_block(bs, &node0_entries, true);
        // dx_node starts with one 12-byte fake dirent: inode=0,
        // rec_len=bs, name_len=0.
        assert_eq!(u32::from_le_bytes(node0[0..4].try_into().unwrap()), 0);
        assert_eq!(
            u16::from_le_bytes(node0[4..6].try_into().unwrap()),
            bs as u16
        );

        let node1_entries = vec![
            DxEntry {
                hash: pack_countlimit(dx_node_limit(bs, true) as u16, 3),
                block: 6,
            },
            DxEntry {
                hash: 0x80000000,
                block: 7,
            },
            DxEntry {
                hash: 0xC0000000,
                block: 8,
            },
        ];
        let node1 = make_dx_node_block(bs, &node1_entries, true);

        // Inline routing arithmetic mirroring `dx_lookup_logical` in
        // `mod.rs`. Verifies the byte layout the writer emits matches
        // what the reader (kernel / e2fsck / our own walker) expects.
        let route = |buf: &[u8], header_len: usize, target: u32| -> u32 {
            let cl = u32::from_le_bytes(buf[header_len..header_len + 4].try_into().unwrap());
            let count = (cl >> 16) as usize;
            let mut chosen = 0usize;
            for slot in 1..count {
                let off = header_len + slot * 8;
                let h = u32::from_le_bytes(buf[off..off + 4].try_into().unwrap());
                if h <= target {
                    chosen = slot;
                } else {
                    break;
                }
            }
            let block_off = header_len + chosen * 8 + 4;
            u32::from_le_bytes(buf[block_off..block_off + 4].try_into().unwrap())
        };

        // Hash 0x10000000 → routes to dx_root slot 0 (countlimit, block=1, node0),
        // then to node0 slot 0 (countlimit, block=3, leaf_a).
        assert_eq!(route(&root, DX_ROOT_HEADER_LEN, 0x10000000), 1);
        assert_eq!(route(&node0, DX_NODE_HEADER_LEN, 0x10000000), 3);

        // Hash 0x30000000 → root slot 0 (node0), then node0 slot 1 (leaf_b).
        assert_eq!(route(&root, DX_ROOT_HEADER_LEN, 0x30000000), 1);
        assert_eq!(route(&node0, DX_NODE_HEADER_LEN, 0x30000000), 4);

        // Hash 0x50000000 → root slot 1 (node1, exact boundary), then
        // node1 slot 0 (leaf_d, since 0x50 < 0x80).
        assert_eq!(route(&root, DX_ROOT_HEADER_LEN, 0x50000000), 2);
        assert_eq!(route(&node1, DX_NODE_HEADER_LEN, 0x50000000), 6);

        // Hash 0xF0000000 → root slot 1 (node1), then node1 slot 2 (leaf_f).
        assert_eq!(route(&root, DX_ROOT_HEADER_LEN, 0xF0000000), 2);
        assert_eq!(route(&node1, DX_NODE_HEADER_LEN, 0xF0000000), 8);
    }

    /// Bit-exact agreement with `libext2fs::ext2fs_dirhash` for
    /// DX_HASH_HALF_MD4 (and DX_HASH_HALF_MD4_UNSIGNED — identical for
    /// ASCII names since signed/unsigned interpretation only differs
    /// on bytes ≥ 0x80). Reference values captured by a small C
    /// program linking against e2fsprogs 1.47.4. If this test ever
    /// breaks, the implementation is wrong — e2fsck will reject the
    /// on-disk index.
    #[test]
    fn half_md4_matches_libext2fs() {
        let cases: &[(&[u8], u32, u32)] = &[
            (b"entry_0000", 0x63df8060, 0x0cf0abb8),
            (b"entry_0001", 0xb88bbf6c, 0x2dae5e85),
            (b"entry_0002", 0xa4283f96, 0x870c4121),
            (b"foo", 0x74c657ac, 0x85a8d812),
            (b"bar", 0x4caaf2ba, 0x16c15fb9),
            (b"baz", 0xee788c74, 0xc5a8743c),
        ];
        for &(name, want_h, want_m) in cases {
            let (got_h, got_m) = half_md4_hash(name);
            assert_eq!(
                got_h,
                want_h,
                "major hash mismatch for {:?}: got {got_h:#010x}, want {want_h:#010x}",
                String::from_utf8_lossy(name)
            );
            assert_eq!(
                got_m,
                want_m,
                "minor hash mismatch for {:?}: got {got_m:#010x}, want {want_m:#010x}",
                String::from_utf8_lossy(name)
            );
        }
    }
}