summaryrefslogtreecommitdiffhomepage
path: root/public/frameworks/DiskImage.fwrk/src/DiskImage+NeFS.cc
blob: 4928c8787d515dfc16b8caa7f02c396ef7c21699 (plain)
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
/* -------------------------------------------

	Copyright (C) 2025, Amlal El Mahrouss, all rights reserved.

	FILE: DiskImage+NeFS.cc
	PURPOSE: Disk Imaging framework.

   ------------------------------------------- */

#include <DiskImage.fwrk/headers/DiskImage.h>

#include <FirmwareKit/EPM.h>
#include <FSKit/NeFS.h>

/// @brief NeFS format over EPM.
/// @param img disk image structure.
/// @return Status code upon completion.
SInt32 DI::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);

	IORef 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;
}