Implemented most disk functions (except trim)

src/disk.c

@@ -0,0 +1,209 @@
+#include "disk.h"
+
+#include <stdlib.h>
+#include <string.h>
+#include <errno.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include <json-c/json.h>
+
+Result create_empty_disk(const char* path, uint64_t size, mode_t permissions) {
+	// Convert the size to a string
+	char* size_str;
+	Result result = format(&size_str, "%lu", size);
+	if (result != SUCCESS)
+		return result;
+
+	// Create the disk
+	int exit_code;
+	char* stderr_buffer = NULL;
+	result = run_executable(&exit_code, NULL, &stderr_buffer, "qemu-img", "create", "-f", "qcow2", path, size_str, NULL);
+
+	// Free the size string
+	free(size_str);
+
+	if (result != SUCCESS) {
+		free(stderr_buffer);
+		return result;
+	}
+
+	// Check the exit code
+	if (exit_code != EXIT_SUCCESS) {
+		if (stderr_buffer == NULL)
+			log_message(LOG_LEVEL_ERROR, "Failed to create the disk '%s'.", path);
+		else {
+			// Remove newlines from the error message
+			for (size_t i = 0; i < strlen(stderr_buffer); i++)
+				if (stderr_buffer[i] == '\n')
+					stderr_buffer[i] = ' ';
+
+			log_message(LOG_LEVEL_ERROR, "Failed to create the disk '%s' (%s).", path, stderr_buffer);
+			free(stderr_buffer);
+		}
+
+		return FAILURE;
+	}
+
+	// Set the permissions of the disk
+	if (chmod(path, permissions) != 0) {
+		log_message(LOG_LEVEL_ERROR, "Failed to set the permissions of the disk '%s' (%s).", path, strerror(errno));
+
+		// Try to remove the disk if the permissions could not be set
+		unlink(path);
+
+		return FAILURE;
+	}
+
+	return SUCCESS;
+}
+
+Result create_backed_disk(const char* path, const char* backing_disk, mode_t permissions) {
+	// Create the disk
+	int exit_code;
+	char* stderr_buffer = NULL;
+	Result result = run_executable(&exit_code, NULL, &stderr_buffer, "qemu-img", "create", "-f", "qcow2", "-F", "qcow2", "-b", backing_disk, path, NULL);
+
+	if (result != SUCCESS) {
+		free(stderr_buffer);
+		return result;
+	}
+
+	// Check the exit code
+	if (exit_code != EXIT_SUCCESS) {
+		if (stderr_buffer == NULL)
+			log_message(LOG_LEVEL_ERROR, "Failed to create the disk '%s'.", path);
+		else {
+			// Remove newlines from the error message
+			for (size_t i = 0; i < strlen(stderr_buffer); i++)
+				if (stderr_buffer[i] == '\n')
+					stderr_buffer[i] = ' ';
+
+			log_message(LOG_LEVEL_ERROR, "Failed to create the disk '%s' (%s).", path, stderr_buffer);
+			free(stderr_buffer);
+		}
+
+		return FAILURE;
+	}
+
+	// Set the permissions of the disk
+	if (chmod(path, permissions) != 0) {
+		log_message(LOG_LEVEL_ERROR, "Failed to set the permissions of the disk '%s' (%s).", path, strerror(errno));
+
+		// Try to remove the disk if the permissions could not be set
+		unlink(path);
+
+		return FAILURE;
+	}
+
+	return SUCCESS;
+}
+
+Result rebase_disk(const char* path, const char* backing_disk) {
+	// Rebase the disk
+	int exit_code;
+	char* stderr_buffer = NULL;
+	Result result = run_executable(&exit_code, NULL, &stderr_buffer, "qemu-img", "rebase", "-u", "-b", backing_disk, path, NULL);
+
+	if (result != SUCCESS) {
+		free(stderr_buffer);
+		return result;
+	}
+
+	// Check the exit code
+	if (exit_code != EXIT_SUCCESS) {
+		if (stderr_buffer == NULL)
+			log_message(LOG_LEVEL_ERROR, "Failed to rebase the disk '%s'.", path);
+		else {
+			// Remove newlines from the error message
+			for (size_t i = 0; i < strlen(stderr_buffer); i++)
+				if (stderr_buffer[i] == '\n')
+					stderr_buffer[i] = ' ';
+
+			log_message(LOG_LEVEL_ERROR, "Failed to rebase the disk '%s' (%s).", path, stderr_buffer);
+			free(stderr_buffer);
+		}
+
+		return FAILURE;
+	}
+
+	return SUCCESS;
+}
+
+Result get_disk_info(const char* path, DiskInfo* out_info) {
+	// Initialize the output
+    out_info->backing_file = NULL;
+    out_info->actual_size = 0;
+    out_info->virtual_size = 0;
+
+	// Get the information about the disk
+	int exit_code;
+	char* stdout_buffer = NULL;
+	Result result = run_executable(&exit_code, &stdout_buffer, NULL, "qemu-img", "info", "--output", "json", path, NULL);
+	if (result != SUCCESS) {
+		free(stdout_buffer);
+		return result;
+	}
+
+	// Check the exit code
+	if (exit_code != EXIT_SUCCESS) {
+		if (stdout_buffer == NULL)
+			log_message(LOG_LEVEL_ERROR, "Failed to get information about the disk '%s'.", path);
+		else {
+			// Remove newlines from the error message
+			for (size_t i = 0; i < strlen(stdout_buffer); i++)
+				if (stdout_buffer[i] == '\n')
+					stdout_buffer[i] = ' ';
+
+			log_message(LOG_LEVEL_ERROR, "Failed to get information about the disk '%s' (%s).", path, stdout_buffer);
+			free(stdout_buffer);
+		}
+
+		return FAILURE;
+	}
+
+	// Parse the JSON output
+	json_object* root = json_tokener_parse(stdout_buffer);
+	if (root == NULL) {
+		log_message(LOG_LEVEL_ERROR, "Failed to parse the JSON output of 'qemu-img info'.");
+		free(stdout_buffer);
+		return FAILURE;
+	}
+
+	// Free the stdout buffer, as it is no longer needed
+	free(stdout_buffer);
+
+	// Get the virtual size
+	json_object* virtual_size_obj;
+	if (json_object_object_get_ex(root, "virtual-size", &virtual_size_obj))
+		out_info->virtual_size = json_object_get_int64(virtual_size_obj);
+
+	// Get the actual size
+	json_object* actual_size_obj;
+	if (json_object_object_get_ex(root, "actual-size", &actual_size_obj))
+		out_info->actual_size = json_object_get_int64(actual_size_obj);
+
+	// Get the backing file
+	json_object* backing_file_obj;
+	if (json_object_object_get_ex(root, "backing-filename", &backing_file_obj)) {
+		out_info->backing_file = strdup(json_object_get_string(backing_file_obj));
+
+		if (out_info->backing_file == NULL) {
+			log_message(LOG_LEVEL_ERROR, "Failed to allocate memory for the backing file path.");
+			json_object_put(root);
+			return OUT_OF_MEMORY;
+		}
+	}
+
+	// Free the JSON object
+	json_object_put(root);
+
+	return SUCCESS;
+}
+
+void free_disk_info(DiskInfo* info) {
+	free(info->backing_file);
+
+	info->backing_file = NULL;
+	info->actual_size = 0;
+	info->virtual_size = 0;
+}

src/sandbox.c

@@ -2,6 +2,7 @@
 
 #include "utils.h"
 #include "entry.h"
+#include "disk.h"
 
 #include <stdio.h>
 #include <stdlib.h>

src/sandbox.h

@@ -2,7 +2,6 @@
 
 #define VERSION "0.0.6"
 
-
 typedef struct {
 	int (*handler)(int argc, char* argv[]);
 	const char* name;