mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-01-24 18:24:45 -05:00
d5871f5717
Similar to POSIX read, the basic read and write functions of AK::Stream do not have a lower limit of how much data they read or write (apart from "none at all"). Rename the functions to "read some [data]" and "write some [data]" (with "data" being omitted, since everything here is reading and writing data) to make them sufficiently distinct from the functions that ensure to use the entire buffer (which should be the go-to function for most usages). No functional changes, just a lot of new FIXMEs.
92 lines
3.2 KiB
C++
92 lines
3.2 KiB
C++
/*
|
|
* Copyright (c) 2021, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <LibCore/ArgsParser.h>
|
|
#include <LibCore/File.h>
|
|
#include <LibCore/System.h>
|
|
#include <LibCrypto/Checksum/Adler32.h>
|
|
#include <LibCrypto/Checksum/CRC32.h>
|
|
#include <LibMain/Main.h>
|
|
#include <string.h>
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
|
{
|
|
Vector<DeprecatedString> paths;
|
|
StringView opt_algorithm;
|
|
|
|
Core::ArgsParser args_parser;
|
|
args_parser.add_option(opt_algorithm, "Checksum algorithm (default 'crc32', use 'list' to list available algorithms)", "algorithm", '\0', nullptr);
|
|
args_parser.add_positional_argument(paths, "File", "file", Core::ArgsParser::Required::No);
|
|
args_parser.parse(arguments);
|
|
|
|
auto algorithm = opt_algorithm.is_empty() ? "crc32" : DeprecatedString(opt_algorithm).to_lowercase();
|
|
|
|
auto available_algorithms = Vector<DeprecatedString> { "crc32", "adler32" };
|
|
|
|
if (algorithm == "list") {
|
|
outln("Available algorithms:");
|
|
for (auto& available_algorithm : available_algorithms) {
|
|
outln(available_algorithm);
|
|
}
|
|
exit(0);
|
|
}
|
|
|
|
if (!available_algorithms.contains_slow(algorithm)) {
|
|
warnln("{}: Unknown checksum algorithm: {}", arguments.strings[0], algorithm);
|
|
exit(1);
|
|
}
|
|
|
|
if (paths.is_empty())
|
|
paths.append("-");
|
|
|
|
bool fail = false;
|
|
Array<u8, PAGE_SIZE> buffer;
|
|
|
|
for (auto& path : paths) {
|
|
auto file_or_error = Core::File::open_file_or_standard_stream(path, Core::File::OpenMode::Read);
|
|
auto filepath = (path == "-") ? "/dev/stdin" : path;
|
|
if (file_or_error.is_error()) {
|
|
warnln("{}: {}: {}", arguments.strings[0], filepath, file_or_error.error());
|
|
fail = true;
|
|
continue;
|
|
}
|
|
auto file = file_or_error.release_value();
|
|
size_t file_size = 0;
|
|
|
|
if (algorithm == "crc32") {
|
|
Crypto::Checksum::CRC32 crc32;
|
|
while (!file->is_eof()) {
|
|
auto data_or_error = file->read_some(buffer);
|
|
if (data_or_error.is_error()) {
|
|
warnln("{}: Failed to read {}: {}", arguments.strings[0], filepath, data_or_error.error());
|
|
fail = true;
|
|
continue;
|
|
}
|
|
file_size += data_or_error.value().size();
|
|
crc32.update(data_or_error.value());
|
|
}
|
|
outln("{:08x} {} {}", crc32.digest(), file_size, path);
|
|
} else if (algorithm == "adler32") {
|
|
Crypto::Checksum::Adler32 adler32;
|
|
while (!file->is_eof()) {
|
|
auto data_or_error = file->read_some(buffer);
|
|
if (data_or_error.is_error()) {
|
|
warnln("{}: Failed to read {}: {}", arguments.strings[0], filepath, data_or_error.error());
|
|
fail = true;
|
|
continue;
|
|
}
|
|
file_size += data_or_error.value().size();
|
|
adler32.update(data_or_error.value());
|
|
}
|
|
outln("{:08x} {} {}", adler32.digest(), file_size, path);
|
|
} else {
|
|
warnln("{}: Unknown checksum algorithm: {}", arguments.strings[0], algorithm);
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
return fail;
|
|
}
|