jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2022, Liav A. <liavalb@hotmail.co.il>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include <AK/DeprecatedString.h>
8#include <AK/Hex.h>
9#include <AK/StringView.h>
10#include <LibCore/ArgsParser.h>
11#include <LibCore/DirIterator.h>
12#include <LibCore/File.h>
13#include <LibCore/System.h>
14#include <LibMain/Main.h>
15
16static constexpr StringView format_row = "{:10s}\t{:10s}\t{:10s}\t{:10s}"sv;
17
18ErrorOr<int> serenity_main(Main::Arguments arguments)
19{
20 TRY(Core::System::pledge("stdio rpath"));
21 TRY(Core::System::unveil("/sys/devices/storage", "r"));
22 TRY(Core::System::unveil(nullptr, nullptr));
23
24 Core::ArgsParser args_parser;
25 args_parser.set_general_help("List Storage (Block) devices.");
26 args_parser.parse(arguments);
27
28 Core::DirIterator di("/sys/devices/storage/", Core::DirIterator::SkipParentAndBaseDir);
29 if (di.has_error()) {
30 auto error = di.error();
31 warnln("Failed to open /sys/devices/storage - {}", error);
32 return error;
33 }
34
35 outln(format_row, "LUN"sv, "Command set"sv, "Block Size"sv, "Last LBA"sv);
36
37 TRY(Core::System::pledge("stdio rpath"));
38
39 while (di.has_next()) {
40 auto dir = di.next_path();
41 auto command_set_filename = DeprecatedString::formatted("/sys/devices/storage/{}/command_set", dir);
42 auto command_set_file = Core::File::open(command_set_filename, Core::File::OpenMode::Read);
43 if (command_set_file.is_error()) {
44 dbgln("Error: Could not open {}: {}", command_set_filename, command_set_file.error());
45 continue;
46 }
47 auto last_lba_filename = DeprecatedString::formatted("/sys/devices/storage/{}/last_lba", dir);
48 auto last_lba_file = Core::File::open(last_lba_filename, Core::File::OpenMode::Read);
49 if (last_lba_file.is_error()) {
50 dbgln("Error: Could not open {}: {}", last_lba_filename, last_lba_file.error());
51 continue;
52 }
53 auto sector_size_filename = DeprecatedString::formatted("/sys/devices/storage/{}/sector_size", dir);
54 auto sector_size_file = Core::File::open(sector_size_filename, Core::File::OpenMode::Read);
55 if (sector_size_file.is_error()) {
56 dbgln("Error: Could not open {}: {}", sector_size_filename, sector_size_file.error());
57 continue;
58 }
59
60 auto maybe_command_set = command_set_file.value()->read_until_eof();
61 if (maybe_command_set.is_error()) {
62 dbgln("Error: Could not read {}: {}", command_set_filename, maybe_command_set.error());
63 continue;
64 }
65 DeprecatedString command_set = StringView(maybe_command_set.value().bytes());
66
67 auto maybe_last_lba = last_lba_file.value()->read_until_eof();
68 if (maybe_last_lba.is_error()) {
69 dbgln("Error: Could not read {}: {}", last_lba_filename, maybe_last_lba.error());
70 continue;
71 }
72 DeprecatedString last_lba = StringView(maybe_last_lba.value().bytes());
73
74 auto maybe_sector_size = sector_size_file.value()->read_until_eof();
75 if (maybe_sector_size.is_error()) {
76 dbgln("Error: Could not read {}: {}", sector_size_filename, maybe_sector_size.error());
77 continue;
78 }
79 DeprecatedString sector_size = StringView(maybe_sector_size.value().bytes());
80
81 outln(format_row, dir, command_set, sector_size, last_lba);
82 }
83
84 return 0;
85}