blob: ad52f39df020666f5e8a2dabedb5928fab89bb51 (
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
65
66
67
68
69
70
71
72
73
|
// Copyright 2024-2026, Amlal El Mahrouss (amlal@nekernel.org)
// Licensed under the Apache License, Version 2.0 (see LICENSE file)
// Official repository: https://github.com/ne-foss-org/nekernel
#include <tools/libmkfs/mkfs.hpp>
#include <tools/libmkfs/openhefs.hpp>
#include <cstdlib>
#include <fstream>
/// @note decimal base.
static uint16_t kNumericalBase = 10;
static auto kMinArgs = 2;
int main(int argc, char** argv) {
if (argc < kMinArgs) {
mkfs::console_out() << "chk: hefs: usage: chk.hefs -in=<input_device> -b=<origin>"
<< "\n";
return EXIT_FAILURE;
}
auto args = mkfs::detail::build_args(argc, argv);
auto opt_disk = mkfs::get_option<char>(args, "in");
auto origin = mkfs::get_option<char>(args, "b");
if (opt_disk.empty()) {
mkfs::console_out() << "chk: hefs: error: OpenHeFS partition is empty! Exiting..."
<< "\n";
return EXIT_FAILURE;
}
auto out_origin = 0UL;
if (!mkfs::detail::parse_signed(origin, out_origin, kNumericalBase)) {
mkfs::console_out() << "chk: hefs: error: Invalid -org=<dec> argument.\n";
return EXIT_FAILURE;
}
std::ifstream output_device(opt_disk, std::ios::binary);
if (!output_device.good()) {
mkfs::console_out() << "chk: hefs: error: Unable to open output device: " << opt_disk << "\n";
return EXIT_FAILURE;
}
output_device.seekg(out_origin);
if (!output_device.good()) {
mkfs::console_out() << "chk: hefs: error: Failed seek to origin.\n";
return EXIT_FAILURE;
}
// @note use modern intializer list here to empty out the fields according to the struct layout.
mkfs::hefs::BootNode boot_node{};
output_device.read(reinterpret_cast<char*>(&boot_node), sizeof(boot_node));
if (strncmp(boot_node.magic, kOpenHeFSMagic, kOpenHeFSMagicLen) != 0 ||
boot_node.sectorCount < 1 || boot_node.sectorSize < kMkFsSectorSz) {
mkfs::console_out() << "chk: hefs: error: Device does not contain an OpenHeFS disk: " << opt_disk
<< "\n";
return EXIT_FAILURE;
}
if (boot_node.badSectors > kMkFsMaxBadSectors) {
mkfs::console_out() << "chk: hefs: error: OpenHeFS disk has too much bad sectors: " << opt_disk
<< "\n";
return EXIT_FAILURE;
}
mkfs::console_out() << "chk: hefs: OpenHeFS partition is healthy. Exiting...\n";
return EXIT_SUCCESS;
}
|
