Added commands for managing disks

This commit is contained in:
2024-02-18 14:05:04 +01:00
parent a149a838d1
commit 3e08d6a438
8 changed files with 403 additions and 35 deletions

View File

@@ -11,14 +11,14 @@
#include <sys/wait.h>
#include <sys/types.h>
LogLevel log_level = LOG_LEVEL_INFO;
LogLevel LOG_LEVEL = LOG_LEVEL_INFO;
void SetLogLevel(LogLevel level) {
log_level = level;
LOG_LEVEL = level;
}
void Log(LogLevel level, const char* format, ...) {
if (level < log_level)
if (level < LOG_LEVEL)
return;
va_list args;
@@ -92,6 +92,67 @@ Status Format(char** _string, const char* fmt, ...) {
return SUCCESS;
}
Status FormatSize(uint64_t size, char** _size_str) {
*_size_str = NULL;
// Determine the unit
const char* unit;
double value = size;
if (value >= 1024ULL * 1024ULL * 1024ULL * 1024ULL) {
unit = "TiB";
value /= 1024.0 * 1024.0 * 1024.0 * 1024.0;
} else if (value >= 1024ULL * 1024ULL * 1024ULL) {
unit = "GiB";
value /= 1024.0 * 1024.0 * 1024.0;
} else if (value >= 1024ULL * 1024ULL) {
unit = "MiB";
value /= 1024.0 * 1024.0;
} else if (value >= 1024ULL) {
unit = "KiB";
value /= 1024.0;
} else {
unit = "B";
}
// Format the size
return Format(_size_str, "%.2f %s", value, unit);
}
Status ParseSize(const char* size_str, uint64_t* _size) {
*_size = 0;
// Parse the size
char* endptr;
uint64_t size = strtoull(size_str, &endptr, 10);
if (endptr == size_str) {
Log(LOG_LEVEL_ERROR, "Failed to parse the size string.");
return FAILURE;
}
if (*endptr == '\0') {
*_size = size;
return SUCCESS;
}
// Determine the unit
if (strcmp(endptr, "TiB") == 0 || strcmp(endptr, "T") == 0 || strcmp(endptr, "TB") == 0)
size *= 1024ULL * 1024ULL * 1024ULL * 1024ULL;
else if (strcmp(endptr, "GiB") == 0 || strcmp(endptr, "G") == 0 || strcmp(endptr, "GB") == 0)
size *= 1024ULL * 1024ULL * 1024ULL;
else if (strcmp(endptr, "MiB") == 0 || strcmp(endptr, "M") == 0 || strcmp(endptr, "MB") == 0)
size *= 1024ULL * 1024ULL;
else if (strcmp(endptr, "KiB") == 0 || strcmp(endptr, "K") == 0 || strcmp(endptr, "KB") == 0)
size *= 1024ULL;
else if (strcmp(endptr, "B") != 0) {
Log(LOG_LEVEL_ERROR, "Invalid size unit '%s'.", endptr);
return FAILURE;
}
*_size = size;
return SUCCESS;
}
Status RunExecutable(int* _exit_code, char** _stdout, char** _stderr, const char* executable, ...) {
if (_exit_code != NULL)
*_exit_code = -1;