You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.1 KiB
56 lines
1.1 KiB
#include "src/lib/string.h"
|
|
#include "src/lib/syscall.h"
|
|
#include "src/lib/util.h"
|
|
#include "src/user/syscall.h"
|
|
|
|
#define BLOCK_SIZE 65536
|
|
|
|
exit_code_t pass(uint64_t fd) {
|
|
static char buffer[BLOCK_SIZE];
|
|
uint64_t bytes;
|
|
uint64_t total = 0;
|
|
while ((bytes = read(fd, BLOCK_SIZE, buffer))) {
|
|
if (bytes == (uint64_t)-1) {
|
|
ERR_S("wc: could not read file\n");
|
|
return EXIT_CODE_GENERAL_FAILURE;
|
|
} else {
|
|
total += bytes;
|
|
}
|
|
}
|
|
|
|
uint64_t l = string_format("%d\n", BLOCK_SIZE, buffer, total);
|
|
write(STDOUT, buffer, l);
|
|
|
|
return EXIT_CODE_OK;
|
|
}
|
|
|
|
exit_code_t wc(const char *path) {
|
|
uint64_t fd = open(path, OPEN_FILE);
|
|
|
|
if (fd == (uint64_t)-1) {
|
|
OUT_S("wc: path does not exist\n");
|
|
return EXIT_CODE_GENERAL_FAILURE;
|
|
}
|
|
|
|
exit_code_t code = pass(fd);
|
|
|
|
close(fd);
|
|
|
|
return code;
|
|
}
|
|
|
|
exit_code_t main(uint64_t argc, const char **argv) {
|
|
if (argc == 1) {
|
|
return pass(0);
|
|
}
|
|
|
|
uint64_t code = EXIT_CODE_OK;
|
|
for (uint64_t i = 1; i < argc; i++) {
|
|
exit_code_t c = wc(argv[i]);
|
|
if (c != EXIT_CODE_OK) {
|
|
code = c;
|
|
}
|
|
}
|
|
|
|
return code;
|
|
}
|
|
|