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
|
/* -------------------------------------------
Copyright (C) 2025, Amlal EL Mahrouss, all rights reserved.
FILE: DiskImage.cc
PURPOSE: Disk Imaging framework.
------------------------------------------- */
#include <DiskImage.fwrk/headers/DiskImage.h>
#include <FirmwareKit/EPM.h>
#include <FSKit/NeFS.h>
/// @brief EPM format disk
/// @param img disk image structure.
/// @return Status code upon completion.
SInt32 DIFormatPartitionEPM(struct DI_DISK_IMAGE& img) noexcept
{
if (!img.sector_sz || (img.sector_sz % 512 != 0))
return kDIFailureStatus;
if (*img.out_name == 0 ||
*img.disk_name == 0)
return kDIFailureStatus;
struct ::EPM_PART_BLOCK block
{
};
block.NumBlocks = img.block_cnt;
block.SectorSz = img.sector_sz;
block.Version = kEPMRevisionBcd;
block.LbaStart = sizeof(struct ::EPM_PART_BLOCK);
block.LbaEnd = img.disk_sz - block.LbaStart;
block.FsVersion = kNeFSVersionInteger;
::MmCopyMemory(block.Name, (VoidPtr)img.disk_name, ::MmStrLen(img.disk_name));
::MmCopyMemory(block.Magic, (VoidPtr)kEPMMagic86, ::MmStrLen(kEPMMagic86));
IOObject handle = IoOpenFile(img.out_name, nullptr);
if (!handle)
return kDIFailureStatus;
::IoWriteFile(handle, (Char*)&block, sizeof(struct ::EPM_PART_BLOCK));
::IoCloseFile(handle);
handle = nullptr;
return kDISuccessStatus;
}
/// @brief NeFS format over EPM.
/// @param img disk image structure.
/// @return Status code upon completion.
SInt32 DIFormatFilesystemNeFS(struct DI_DISK_IMAGE& img) noexcept
{
if (!img.sector_sz || (img.sector_sz % 512 != 0))
return kDIFailureStatus;
if (*img.out_name == 0 ||
*img.disk_name == 0)
return kDIFailureStatus;
struct ::NEFS_ROOT_PARTITION_BLOCK rpb
{
};
::MmCopyMemory(rpb.PartitionName, (VoidPtr)img.disk_name, ::MmStrLen(img.disk_name));
::MmCopyMemory(rpb.Ident, (VoidPtr)kNeFSIdent, ::MmStrLen(kNeFSIdent));
rpb.Version = kNeFSVersionInteger;
rpb.EpmBlock = kEPMBootBlockLba;
rpb.StartCatalog = kNeFSCatalogStartAddress;
rpb.CatalogCount = 0;
rpb.DiskSize = img.disk_sz;
rpb.SectorSize = img.sector_sz;
rpb.SectorCount = rpb.DiskSize / rpb.SectorSize;
rpb.FreeSectors = rpb.SectorCount;
rpb.FreeCatalog = rpb.DiskSize / sizeof(NEFS_CATALOG_STRUCT);
IOObject handle = IoOpenFile(img.out_name, nullptr);
if (!handle)
return kDIFailureStatus;
UInt64 p_prev = ::IoTellFile(handle);
::IoWriteFile(handle, (Char*)&rpb, sizeof(struct ::NEFS_ROOT_PARTITION_BLOCK));
::IoSeekFile(handle, p_prev);
::IoCloseFile(handle);
handle = nullptr;
return kDISuccessStatus;
}
|
