238 lines
7.4 KiB
C
238 lines
7.4 KiB
C
#include "config.h"
|
|
#include "delta.h"
|
|
#include "log.h"
|
|
#include "protocol.h"
|
|
#include "utils.h"
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
Config* config_create(char* version, char* send_directory, char* receive_directory,
|
|
bool save_to_disk, bool use_multithreading, bool use_chunk_serialization,
|
|
bool use_compression, bool use_metadata, int compression_level,
|
|
bool use_sendfile, unsigned long long chunk_size) {
|
|
|
|
Config* config = malloc(sizeof(Config));
|
|
config->version = version;
|
|
config->send_directory = send_directory;
|
|
config->receive_root_directory = receive_directory;
|
|
config->save_to_disk = save_to_disk;
|
|
config->use_multithreading = use_multithreading;
|
|
config->use_chunk_serialization = use_chunk_serialization;
|
|
config->use_compression = use_compression;
|
|
config->use_metadata = use_metadata;
|
|
config->show_progress = false;
|
|
config->dry_run = false;
|
|
config->use_delete = false;
|
|
config->compression_level = compression_level;
|
|
config->use_sendfile = use_sendfile;
|
|
config->chunk_size = chunk_size > 0 ? chunk_size : DEFAULT_CHUNK_SIZE;
|
|
config->ssh_port = 22;
|
|
config->transport = TRANSPORT_TCP;
|
|
config->ssh_destination = NULL;
|
|
config->exclude_patterns = NULL;
|
|
config->exclude_count = 0;
|
|
config->include_patterns = NULL;
|
|
config->include_count = 0;
|
|
config->max_size = 0;
|
|
config->min_size = 0;
|
|
config->use_incremental = false;
|
|
config->use_delta = false;
|
|
config->delta_block_size = DELTA_BLOCK_SIZE_DEFAULT;
|
|
config->delta_max_file_size = DELTA_MAX_FILE_SIZE;
|
|
config->use_tls = false;
|
|
config->tls_cert = NULL;
|
|
config->tls_key = NULL;
|
|
config->tls_ca = NULL;
|
|
config->server_host = str_dup("127.0.0.1");
|
|
config->server_port = 8080;
|
|
return config;
|
|
}
|
|
|
|
bool is_remote_dest(const char* s) {
|
|
if (s == NULL)
|
|
return false;
|
|
const char* colon = strchr(s, ':');
|
|
if (colon == NULL)
|
|
return false;
|
|
if (colon == s)
|
|
return false;
|
|
for (const char* p = s; p < colon; p++) {
|
|
if (*p == '/')
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void config_parse_ssh_dest(Config* config) {
|
|
if (!is_remote_dest(config->receive_root_directory))
|
|
return;
|
|
config->transport = TRANSPORT_SSH;
|
|
config->ssh_destination = str_dup(config->receive_root_directory);
|
|
const char* colon = strchr(config->receive_root_directory, ':');
|
|
char* path = str_dup(colon + 1);
|
|
free(config->receive_root_directory);
|
|
config->receive_root_directory = path;
|
|
}
|
|
|
|
void config_delete(Config* config) {
|
|
free(config->version);
|
|
free(config->send_directory);
|
|
free(config->receive_root_directory);
|
|
free(config->ssh_destination);
|
|
for (int i = 0; i < config->exclude_count; i++)
|
|
free(config->exclude_patterns[i]);
|
|
free(config->exclude_patterns);
|
|
for (int i = 0; i < config->include_count; i++)
|
|
free(config->include_patterns[i]);
|
|
free(config->include_patterns);
|
|
free(config->tls_cert);
|
|
free(config->tls_key);
|
|
free(config->tls_ca);
|
|
free(config->server_host);
|
|
free(config);
|
|
}
|
|
|
|
bool config_send(int file_descriptor, const Config* config) {
|
|
if (!send_str(file_descriptor, config->version))
|
|
return false;
|
|
if (!send_str(file_descriptor, config->send_directory))
|
|
return false;
|
|
if (!send_str(file_descriptor, config->receive_root_directory))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->save_to_disk))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->use_multithreading))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->use_chunk_serialization))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->use_compression))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->use_metadata))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->compression_level))
|
|
return false;
|
|
if (!send_n_data(file_descriptor, &config->chunk_size, sizeof(config->chunk_size)))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->use_sendfile))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->use_delete))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->use_incremental))
|
|
return false;
|
|
if (!send_int(file_descriptor, config->use_delta))
|
|
return false;
|
|
if (!send_int(file_descriptor, (int)config->delta_block_size))
|
|
return false;
|
|
if (!send_n_data(file_descriptor, &config->delta_max_file_size, sizeof(unsigned long long)))
|
|
return false;
|
|
Status status;
|
|
if (!receive_status(file_descriptor, &status))
|
|
return false;
|
|
if (status != STATUS_OK) {
|
|
log_message(LOG_LEVEL_ERROR, "Error transmitting config");
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
Config* config_receive(int file_descriptor) {
|
|
Config* config = (Config*)malloc(sizeof(Config));
|
|
if (config == NULL)
|
|
return NULL;
|
|
memset(config, 0, sizeof(*config));
|
|
config->version = receive_str(file_descriptor);
|
|
if (!config->version) {
|
|
free(config);
|
|
return NULL;
|
|
}
|
|
if (strcmp(config->version, PROTOCOL_VERSION) != 0) {
|
|
fprintf(stderr, "Protocol version mismatch: client=%s, server=%s\n", config->version,
|
|
PROTOCOL_VERSION);
|
|
free(config->version);
|
|
free(config);
|
|
send_status(file_descriptor, STATUS_ERROR);
|
|
return NULL;
|
|
}
|
|
config->send_directory = receive_str(file_descriptor);
|
|
if (!config->send_directory) {
|
|
free(config->version);
|
|
free(config);
|
|
return NULL;
|
|
}
|
|
config->receive_root_directory = receive_str(file_descriptor);
|
|
if (!config->receive_root_directory) {
|
|
free(config->version);
|
|
free(config->send_directory);
|
|
free(config);
|
|
return NULL;
|
|
}
|
|
int tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->save_to_disk = tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->use_multithreading = tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->use_chunk_serialization = tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->use_compression = tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->use_metadata = tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->compression_level = tmp;
|
|
if (!receive_n_data(file_descriptor, &config->chunk_size, sizeof(config->chunk_size)))
|
|
goto error;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->use_sendfile = tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->use_delete = tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->use_incremental = tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->use_delta = tmp;
|
|
if (!receive_int(file_descriptor, &tmp))
|
|
goto error;
|
|
config->delta_block_size = (uint32_t)tmp;
|
|
if (!receive_n_data(file_descriptor, &config->delta_max_file_size, sizeof(unsigned long long)))
|
|
goto error;
|
|
config->show_progress = false;
|
|
config->dry_run = false;
|
|
config->ssh_port = 22;
|
|
config->transport = TRANSPORT_TCP;
|
|
config->ssh_destination = NULL;
|
|
config->exclude_patterns = NULL;
|
|
config->exclude_count = 0;
|
|
config->include_patterns = NULL;
|
|
config->include_count = 0;
|
|
config->max_size = 0;
|
|
config->min_size = 0;
|
|
config->use_tls = false;
|
|
config->tls_cert = NULL;
|
|
config->tls_key = NULL;
|
|
config->tls_ca = NULL;
|
|
config->server_host = str_dup("127.0.0.1");
|
|
config->server_port = 8080;
|
|
if (!send_status(file_descriptor, STATUS_OK))
|
|
goto error;
|
|
return config;
|
|
|
|
error:
|
|
free(config->version);
|
|
free(config->send_directory);
|
|
free(config->receive_root_directory);
|
|
free(config->server_host);
|
|
free(config);
|
|
return NULL;
|
|
}
|