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
|
/* -------------------------------------------
Copyright (C) 2025, Amlal El Mahrouss, licensed under the Apache 2.0 license.
FILE: CommandLine.cc
PURPOSE: DIUTIL CLI.
------------------------------------------- */
#include <DiskImage.fwrk/headers/DiskImage.h>
static const Char kDiskName[kDIDiskNameLen] = "Empty Disk";
static SInt32 kDiskSectorSz = kDISectorSz;
static SizeT kDiskSz = gib_cast(4);
static const Char kOutDisk[kDIOutNameLen] = "disk.eimg";
/// @brief Disk Utility entrypoint.
SInt32 _NeMain(SInt32 argc, Char** argv) {
for (SInt32 arg = 0; arg < argc; ++arg) {
const Char* arg_s = argv[arg];
if (MmCmpMemory((VoidPtr) arg_s, (VoidPtr) "-diutil-output-name",
MmStrLen("-diutil-output-name") == 0)) {
if ((arg + 1) < argc) {
MmCopyMemory((VoidPtr) kOutDisk, argv[arg + 1], kDIDiskNameLen);
}
} else if (MmCmpMemory((VoidPtr) arg_s, (VoidPtr) "-diutil-output-size",
MmStrLen("-diutil-output-size") == 0)) {
if ((arg + 1) < argc) {
kDiskSz = StrMathToNumber(argv[arg + 1], nullptr, 10);
}
} else if (MmCmpMemory((VoidPtr) arg_s, (VoidPtr) "-diutil-sector-size",
MmStrLen("-diutil-sector-size") == 0)) {
if ((arg + 1) < argc) {
kDiskSectorSz = StrMathToNumber(argv[arg + 1], nullptr, 10);
}
} else if (MmCmpMemory((VoidPtr) arg_s, (VoidPtr) "-diutil-part-name",
MmStrLen("-diutil-part-name") == 0)) {
if ((arg + 1) < argc) {
MmCopyMemory((VoidPtr) kDiskName, argv[arg + 1], kDIDiskNameLen);
}
}
}
// create disk image, by appending an EPM partition to it.
DI::DI_DISK_IMAGE img{};
img.disk_sz = kDiskSz;
img.sector_sz = kDiskSectorSz;
img.block_cnt = 0;
MmCopyMemory((VoidPtr) img.disk_name, (VoidPtr) kDiskName, kDIDiskNameLen);
MmCopyMemory((VoidPtr) img.out_name, (VoidPtr) kOutDisk, kDIDiskNameLen);
// format disk image to explicit partition map.
return DI::DIFormatPartitionEPM(img);
}
|
