H1K0-dev/tanabata
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
tanabata/tdbms/server/tdbms-server.c

1009 lines
34 KiB
C
Raw Blame History

#include <stdlib.h>
#include <stdint.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <pthread.h>
#include <stdarg.h>
#include <sys/time.h>
#include "../../include/tdbms.h"
#include "../../include/tanabata.h"
// TDBMS configuration file
#define TDBMS_CONFIG_FILE "/etc/tanabata/tdbms.conf"
// TDBMS variables directory
#define TDBMS_VAR_DIR "/var/lib/tanabata/tdbms"
// TDBMS default socket domain
#define DEFAULT_SOCK_DOMAIN "UNIX"
// TDBMS degault socket address
#define DEFAULT_SOCK_ADDR "/tmp/tdbms.sock"
// Maximum number of queued clients
#define DEFAULT_CLIENTS_MAX 0x100
// Maximum count of critical errors in row
#define DEFAULT_CRITICAL_MAX 0b10000
// Default log file path
#define DEFAULT_LOG_FILE "/var/log/tanabata/tdbms.log"
// Log levels
#define LOG_INFO 2
#define LOG_WARNING 3
#define LOG_ERROR 4
#define LOG_CRITICAL 5
#define LOG_FATAL 6
typedef struct tdb {
char *name; // TDB name
char *path; // TDB location path
Tanabata *database; // TDB data
} TDB;
char *sock_domain = NULL;
char *sock_addr = NULL;
uint16_t max_clients = DEFAULT_CLIENTS_MAX;
uint16_t max_criticals = DEFAULT_CRITICAL_MAX;
int server_fd;
TDB *db_list = NULL;
uint16_t db_count = 0;
FILE *logfile = NULL;
pthread_mutex_t mutex_log = PTHREAD_MUTEX_INITIALIZER;
pthread_mutex_t mutex_connection = PTHREAD_MUTEX_INITIALIZER;
pthread_mutex_t mutex_request = PTHREAD_MUTEX_INITIALIZER;
pthread_t *threads = NULL;
// Log formatted message
void logtf(int level, const char *fmt, ...) {
va_list args;
va_start(args, fmt);
struct timeval now;
gettimeofday(&now, NULL);
char *slevel;
switch (level) {
case LOG_INFO:
slevel = "INFO";
break;
case LOG_WARNING:
slevel = "WARNING";
break;
case LOG_ERROR:
slevel = "ERROR";
break;
case LOG_CRITICAL:
slevel = "CRITICAL";
break;
case LOG_FATAL:
slevel = "FATAL";
break;
default:
logtf(LOG_ERROR, "Invalid log level %i", level);
return;
}
size_t level_len = strlen(slevel);
char *fmt_log = malloc(33 + level_len + strlen(fmt));
strftime(fmt_log, 20, "%F %T", localtime(&now.tv_sec));
sprintf(fmt_log, "%s.%06li | %s | %s\n", fmt_log, now.tv_usec, slevel, fmt);
pthread_mutex_lock(&mutex_log);
if (logfile != NULL) {
vfprintf(logfile, fmt_log, args);
} else {
vfprintf(stdout, fmt_log, args);
}
fflush(logfile);
pthread_mutex_unlock(&mutex_log);
free(fmt_log);
}
// Load configuration
void config_load() {
FILE *config = fopen(TDBMS_CONFIG_FILE, "r");
if (config == NULL) {
config = fopen(TDBMS_CONFIG_FILE, "w");
if (config == NULL) {
logtf(LOG_FATAL, "failed to create config file");
exit(1);
}
fprintf(config, "sockdomn=%s\nsockaddr=%s\nmax_clients=0x%x\nmax_criticals=0x%x\nlogfile=%s\n",
DEFAULT_SOCK_DOMAIN, DEFAULT_SOCK_ADDR, DEFAULT_CLIENTS_MAX,
DEFAULT_CRITICAL_MAX, DEFAULT_LOG_FILE);
fclose(config);
return;
}
char buffer[BUFSIZ];
char *value;
while (fgets(buffer, BUFSIZ, config) != NULL) {
value = buffer;
value[strlen(value) - 1] = 0;
if (strncmp(buffer, "sockdomn", 8) == 0) {
for (value += 8; *value == ' ' || *value == '='; value++);
sock_domain = realloc(sock_domain, strlen(value) + 1);
strcpy(sock_domain, value);
} else if (strncmp(buffer, "sockaddr", 8) == 0) {
for (value += 8; *value == ' ' || *value == '='; value++);
sock_addr = realloc(sock_addr, strlen(value) + 1);
strcpy(sock_addr, value);
} else if (strncmp(buffer, "max_clients", 11) == 0) {
for (value += 11; *value == ' ' || *value == '='; value++);
char *endptr;
max_clients = strtoul(value, &endptr, 0);
if (*endptr != 0) {
logtf(LOG_FATAL, "invalid config file");
exit(1);
}
} else if (strncmp(buffer, "max_criticals", 13) == 0) {
for (value += 13; *value == ' ' || *value == '='; value++);
char *endptr;
max_criticals = strtoul(value, &endptr, 0);
if (*endptr != 0) {
logtf(LOG_FATAL, "invalid config file");
exit(1);
}
} else if (strncmp(buffer, "logfile", 7) == 0) {
for (value += 7; *value == ' ' || *value == '='; value++);
if ((logfile = fopen(value, "a")) == NULL) {
logtf(LOG_ERROR, "failed to load custom log file, use default");
}
} else if (*buffer != 0) {
logtf(LOG_WARNING, "unexpected parameter '%s'", value);
}
}
fclose(config);
}
// Load database list
void dblist_load() {
FILE *fdblist = fopen(TDBMS_VAR_DIR"/.dblist", "rb");
if (fdblist == NULL) {
fdblist = fopen(TDBMS_VAR_DIR"/.dblist", "wb");
if (fdblist == NULL) {
logtf(LOG_FATAL, "failed to create database list file");
exit(1);
}
fclose(fdblist);
return;
}
TDB newbie;
char *name = NULL, *path = NULL;
size_t MAX_PATH_LEN = SIZE_MAX;
for (db_count = 0; db_count < UINT16_MAX;) {
if (getdelim(&name, &MAX_PATH_LEN, ' ', fdblist) == -1 ||
getdelim(&path, &MAX_PATH_LEN, '\n', fdblist) == -1) {
break;
}
newbie.name = malloc(strlen(name));
newbie.path = malloc(strlen(path));
newbie.database = NULL;
name[strlen(name) - 1] = 0;
path[strlen(path) - 1] = 0;
strcpy(newbie.name, name);
strcpy(newbie.path, path);
db_count++;
db_list = realloc(db_list, db_count * sizeof(TDB));
db_list[db_count - 1] = newbie;
}
fclose(fdblist);
}
// Save database list
void dblist_save() {
FILE *fdblist = fopen(TDBMS_VAR_DIR"/.dblist", "wb");
if (fdblist == NULL) {
logtf(LOG_CRITICAL, "failed to save database list file");
return;
}
TDB *tdb = db_list;
for (uint16_t i = 0; i < db_count; i++) {
fprintf(fdblist, "%s %s\n", tdb->name, tdb->path);
tdb++;
}
fclose(fdblist);
}
// Open TDBMS server socket
int socket_open() {
int socket_fd;
struct sockaddr_un sockaddr;
int domain;
if (strcmp(sock_domain, "UNIX") == 0) {
domain = AF_UNIX;
} else {
logtf(LOG_FATAL, "unexpected socket domain '%s'", sock_domain);
exit(1);
}
if (strlen(sock_addr) + 1 > sizeof(sockaddr.sun_path)) {
logtf(LOG_FATAL, "too long socket address");
exit(1);
}
socket_fd = socket(domain, SOCK_STREAM, 0);
if (socket_fd < 0) {
logtf(LOG_FATAL, "failed to initialize socket");
exit(-1);
}
bzero(&sockaddr, sizeof(sockaddr));
sockaddr.sun_family = domain;
strcpy(sockaddr.sun_path, sock_addr);
unlink(sock_addr);
if (bind(socket_fd, (const struct sockaddr *) &sockaddr, sizeof(sockaddr)) < 0) {
logtf(LOG_FATAL, "failed to bind socket");
exit(-1);
}
if (listen(socket_fd, DEFAULT_CLIENTS_MAX) < 0) {
logtf(LOG_FATAL, "failed to listen to socket");
exit(-1);
}
return socket_fd;
}
// Create an escaped string for JSON
char *escape(const char *plain) {
char *escaped = malloc(BUFSIZ);
size_t size = BUFSIZ;
const char *input;
char *output;
for (input = plain, output = escaped; *input != 0; input++, output++) {
if (output - escaped + 3 >= size) {
off_t offset = output - escaped;
size += BUFSIZ;
escaped = realloc(escaped, size);
output = escaped + offset;
}
switch (*input) {
case '\\':
case '"':
*output = '\\';
output++;
*output = *input;
break;
case '/':
strcpy(output, "\\/");
output++;
break;
case '\b':
strcpy(output, "\\b");
output++;
break;
case '\t':
strcpy(output, "\\t");
output++;
break;
case '\n':
strcpy(output, "\\n");
output++;
break;
case '\f':
strcpy(output, "\\f");
output++;
break;
default:
*output = *input;
}
}
*output = 0;
return escaped;
}
// Execute request
int execute(char *request, char **response) {
char request_code = *request;
char *request_db_name = request + 1;
char *request_body = request_db_name + strlen(request_db_name) + 1;
*response = malloc(BUFSIZ);
**response = 0;
TDB *tdb;
Tanabata *tanabata;
for (tdb = db_list + db_count - 1; tdb >= db_list; tdb--) {
if (strcmp(tdb->name, request_db_name) == 0) {
break;
}
}
if (tdb < db_list) {
tdb = NULL;
tanabata = NULL;
} else {
if (tdb->database == NULL &&
request_code != trc_db_init && request_code != trc_db_edit && request_code != trc_db_remove_soft) {
tdb->database = malloc(sizeof(Tanabata));
tanabata_init(tdb->database);
if (tanabata_open(tdb->database, tdb->path) != 0) {
return 1;
}
}
tanabata = tdb->database;
}
sprintf(*response, "{\"status\":true,\"data\":[");
char *buffer = *response + strlen(*response);
if (request_code == trc_db_stats) {
if (*request_db_name != 0) {
if (tanabata == NULL) {
sprintf(buffer, "{\"loaded\":false}]}");
return 0;
}
sprintf(buffer, "{"
"\"loaded\":true,\"changed\":%s,"
"\"sasahyou\":{\"cts\":%lu,\"mts\":%lu,\"size\":%lu,\"holes\":%lu},"
"\"sappyou\":{\"cts\":%lu,\"mts\":%lu,\"size\":%lu,\"holes\":%lu},"
"\"shoppyou\":{\"cts\":%lu,\"mts\":%lu,\"size\":%lu,\"holes\":%lu}"
"}]}",
(tanabata->sasahyou_mod != tanabata->sasahyou.modified_ts ||
tanabata->sappyou_mod != tanabata->sappyou.modified_ts ||
tanabata->shoppyou_mod != tanabata->shoppyou.modified_ts) ? "true" : "false",
tanabata->sasahyou.created_ts, tanabata->sasahyou.modified_ts, tanabata->sasahyou.size,
tanabata->sasahyou.hole_cnt,
tanabata->sappyou.created_ts, tanabata->sappyou.modified_ts, tanabata->sappyou.size,
tanabata->sappyou.hole_cnt,
tanabata->shoppyou.created_ts, tanabata->shoppyou.modified_ts, tanabata->shoppyou.size,
tanabata->shoppyou.hole_cnt);
return 0;
}
if (db_count == 0) {
**response = 0;
return 0;
}
size_t resp_size = BUFSIZ;
buffer = malloc(BUFSIZ);
TDB *temp = db_list;
for (uint16_t i = 0; i < db_count; i++, temp++) {
if (temp->database == NULL) {
sprintf(buffer, "{\"name\":\"%s\",\"loaded\":false},", temp->name);
} else {
tanabata = temp->database;
sprintf(buffer, "{"
"\"loaded\":true,\"unsaved\":%s,"
"\"sasahyou_cts\":%lu,\"sasahyou_mts\":%lu,\"sasahyou_size\":%lu,\"sasahyou_holes\":%lu,"
"\"sappyou_cts\":%lu,\"sappyou_mts\":%lu,\"sappyou_size\":%lu,\"sappyou_holes\":%lu,"
"\"shoppyou_cts\":%lu,\"shoppyou_mts\":%lu,\"shoppyou_size\":%lu,\"shoppyou_holes\":%lu"
"},",
(tanabata->sasahyou_mod != tanabata->sasahyou.modified_ts ||
tanabata->sappyou_mod != tanabata->sappyou.modified_ts ||
tanabata->shoppyou_mod != tanabata->shoppyou.modified_ts) ? "true" : "false",
tanabata->sasahyou.created_ts, tanabata->sasahyou.modified_ts, tanabata->sasahyou.size,
tanabata->sasahyou.hole_cnt,
tanabata->sappyou.created_ts, tanabata->sappyou.modified_ts, tanabata->sappyou.size,
tanabata->sappyou.hole_cnt,
tanabata->shoppyou.created_ts, tanabata->shoppyou.modified_ts, tanabata->shoppyou.size,
tanabata->shoppyou.hole_cnt);
}
if (strlen(*response) + strlen(buffer) >= resp_size) {
resp_size += BUFSIZ;
*response = realloc(*response, resp_size);
}
strcat(*response, buffer);
}
sprintf(buffer, "]}");
if (strlen(*response) + 2 >= resp_size) {
*response = realloc(*response, resp_size + 2);
}
strcpy(*response + strlen(*response) - 1, buffer);
free(buffer);
return 0;
}
if (request_code == trc_db_init) {
if (db_count == UINT16_MAX || tdb != NULL) {
return 1;
}
**response = 0;
db_count++;
db_list = reallocarray(db_list, db_count, sizeof(TDB));
tdb = db_list + db_count - 1;
tdb->path = malloc(strlen(request_body) + 1);
tdb->name = malloc(strlen(request_db_name) + 1);
tdb->database = malloc(sizeof(Tanabata));
tanabata_init(tdb->database);
strcpy(tdb->name, request_db_name);
strcpy(tdb->path, request_body);
dblist_save();
return 0;
}
if (request_code == trc_db_load) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
return tanabata_open(tanabata, tdb->path);
}
if (request_code == trc_db_save) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
if (strlen(request_body) > 0) {
return tanabata_dump(tanabata, request_body);
}
return tanabata_dump(tanabata, tdb->path);
}
if (request_code == trc_db_edit) {
if (tdb == NULL) {
return 1;
}
**response = 0;
buffer = request_body;
off_t offset;
while (*buffer != 0) {
if (strncmp(buffer, "name", 4) == 0) {
for (buffer += 4; *buffer == ' ' || *buffer == '='; buffer++);
for (offset = 0; buffer[offset] != '\n' && buffer[offset] != 0; offset++);
tdb->name = realloc(tdb->name, offset);
strncpy(tdb->name, buffer, offset);
tdb->name[offset] = 0;
buffer += offset;
}
if (strncmp(buffer, "path", 4) == 0) {
for (buffer += 4; *buffer == ' ' || *buffer == '='; buffer++);
for (offset = 0; buffer[offset] != '\n' && buffer[offset] != 0; offset++);
tdb->path = realloc(tdb->path, offset);
strncpy(tdb->path, buffer, offset);
tdb->path[offset] = 0;
buffer += offset;
} else {
return 1;
}
}
dblist_save();
return 0;
}
if (request_code == trc_db_remove_soft) {
if (tdb == NULL) {
return 1;
}
**response = 0;
free(tdb->name);
free(tdb->path);
if (tdb->database != NULL) {
tanabata_free(tdb->database);
free(tdb->database);
}
db_count--;
for (uint16_t i = tdb - db_list; i < db_count; i++) {
db_list[i] = db_list[i + 1];
}
db_list = reallocarray(db_list, db_count, sizeof(TDB));
dblist_save();
return 0;
}
if (request_code == trc_db_remove_hard) {
if (tdb == NULL) {
return 1;
}
**response = 0;
size_t pathlen = strlen(tdb->path);
char *remove_path = malloc(pathlen + 10);
strcpy(remove_path, tdb->path);
free(tdb->name);
free(tdb->path);
if (tdb->database != NULL) {
tanabata_free(tdb->database);
free(tdb->database);
}
db_count--;
for (uint16_t i = tdb - db_list; i < db_count; i++) {
db_list[i] = db_list[i + 1];
}
db_list = reallocarray(db_list, db_count, sizeof(TDB));
strcpy(remove_path + pathlen, "/sasahyou");
unlink(remove_path);
strcpy(remove_path + pathlen, "/sappyou");
unlink(remove_path);
strcpy(remove_path + pathlen, "/shoppyou");
unlink(remove_path);
free(remove_path);
dblist_save();
return 0;
}
if (request_code == trc_db_weed) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
return tanabata_weed(tanabata);
}
if (request_code == trc_sasa_get) {
if (tanabata == NULL) {
return 1;
}
if (*request_body != 0) {
char *endptr;
uint64_t sasa_id = strtoull(request_body, &endptr, 0);
if (*endptr != 0) {
return 1;
}
Sasa temp = tanabata_sasa_get(tanabata, sasa_id);
if (temp.id == HOLE_ID) {
return 1;
}
sprintf(buffer, "{\"id\":%lu,\"cts\":%lu,\"path\":\"%s\"}]}",
temp.id, temp.created_ts, temp.path);
return 0;
}
if (tanabata->sasahyou.size - tanabata->sasahyou.hole_cnt == 0) {
strcat(*response, "]}");
return 0;
}
size_t resp_size = BUFSIZ;
buffer = malloc(BUFSIZ);
Sasa *temp = tanabata->sasahyou.database;
for (uint64_t i = 0; i < tanabata->sasahyou.size; i++, temp++) {
if (temp->id == HOLE_ID) {
continue;
}
sprintf(buffer, "{\"id\":%lu,\"cts\":%lu,\"path\":\"%s\"},",
temp->id, temp->created_ts, temp->path);
if (strlen(*response) + strlen(buffer) >= resp_size) {
resp_size += BUFSIZ;
*response = realloc(*response, resp_size);
}
strcat(*response, buffer);
}
sprintf(buffer, "]}");
if (strlen(*response) + 2 >= resp_size) {
*response = realloc(*response, resp_size + 2);
}
strcpy(*response + strlen(*response) - 1, buffer);
free(buffer);
return 0;
}
if (request_code == trc_sasa_get_by_tanzaku) {
if (tanabata == NULL || *request_body == 0) {
return 1;
}
char *endptr;
uint64_t tanzaku_id = strtoull(request_body, &endptr, 0);
if (*endptr != 0) {
return 1;
}
Sasa *list = tanabata_sasa_get_by_tanzaku(tanabata, tanzaku_id);
if (list == NULL) {
return 1;
}
if (list->id == HOLE_ID) {
**response = 0;
return 0;
}
size_t resp_size = BUFSIZ;
buffer = malloc(BUFSIZ);
for (Sasa *temp = list; temp->id != HOLE_ID; temp++) {
sprintf(buffer, "{\"id\":%lu,\"cts\":%lu,\"path\":\"%s\"},",
temp->id, temp->created_ts, temp->path);
if (strlen(*response) + strlen(buffer) >= resp_size) {
resp_size += BUFSIZ;
*response = realloc(*response, resp_size);
}
strcat(*response, buffer);
}
free(list);
sprintf(buffer, "]}");
if (strlen(*response) + 2 >= resp_size) {
*response = realloc(*response, resp_size + 2);
}
strcpy(*response + strlen(*response) - 1, buffer);
free(buffer);
return 0;
}
if (request_code == trc_sasa_add) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
return tanabata_sasa_add(tanabata, request_body);
}
if (request_code == trc_sasa_update) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t sasa_id = strtoull(request_body, &endptr, 0);
if (*endptr != ' ') {
return 1;
}
return tanabata_sasa_upd(tanabata, sasa_id, endptr + 1);
}
if (request_code == trc_sasa_remove) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t sasa_id = strtoull(request_body, &endptr, 0);
if (*endptr != 0) {
return 1;
}
return tanabata_sasa_rem(tanabata, sasa_id);
}
if (request_code == trc_sasa_remove_by_tanzaku) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t tanzaku_id = strtoull(request_body, &endptr, 0);
if (*endptr != 0) {
return 1;
}
Sasa *list = tanabata_sasa_get_by_tanzaku(tanabata, tanzaku_id);
if (list == NULL) {
return 1;
}
for (Sasa *temp = list; temp->id != HOLE_ID; temp++) {
tanabata_sasa_rem(tanabata, temp->id);
}
free(list);
return 0;
}
if (request_code == trc_tanzaku_get) {
if (tanabata == NULL) {
return 1;
}
if (*request_body != 0) {
char *endptr;
uint64_t tanzaku_id = strtoull(request_body, &endptr, 0);
if (*endptr != 0) {
return 1;
}
Tanzaku temp = tanabata_tanzaku_get(tanabata, tanzaku_id);
if (temp.id == HOLE_ID) {
return 1;
}
char *escaped_name = escape(temp.name),
*escaped_description = escape(temp.description);
sprintf(buffer, "{\"id\":%lu,\"cts\":%lu,\"mts\":%lu,\"name\":\"%s\",\"desc\":\"%s\"}]}",
temp.id, temp.created_ts, temp.modified_ts, escaped_name, escaped_description);
free(escaped_name);
free(escaped_description);
return 0;
}
if (tanabata->sappyou.size - tanabata->sappyou.hole_cnt == 0) {
strcat(*response, "]}");
return 0;
}
size_t resp_size = BUFSIZ;
buffer = malloc(BUFSIZ);
Tanzaku *temp = tanabata->sappyou.database;
for (uint64_t i = 0; i < tanabata->sappyou.size; i++, temp++) {
if (temp->id == HOLE_ID) {
continue;
}
char *escaped_name = escape(temp->name),
*escaped_description = escape(temp->description);
sprintf(buffer, "{\"id\":%lu,\"cts\":%lu,\"mts\":%lu,\"name\":\"%s\",\"desc\":\"%s\"},",
temp->id, temp->created_ts, temp->modified_ts, escaped_name, escaped_description);
free(escaped_name);
free(escaped_description);
if (strlen(*response) + strlen(buffer) >= resp_size) {
resp_size += BUFSIZ;
*response = realloc(*response, resp_size);
}
strcat(*response, buffer);
}
sprintf(buffer, "]}");
if (strlen(*response) + 2 >= resp_size) {
*response = realloc(*response, resp_size + 2);
}
strcpy(*response + strlen(*response) - 1, buffer);
free(buffer);
return 0;
}
if (request_code == trc_tanzaku_get_by_sasa) {
if (tanabata == NULL || *request_body == 0) {
return 1;
}
char *endptr;
uint64_t sasa_id = strtoull(request_body, &endptr, 0);
if (*endptr != 0) {
return 1;
}
Tanzaku *list = tanabata_tanzaku_get_by_sasa(tanabata, sasa_id);
if (list == NULL) {
return 1;
}
if (list->id == HOLE_ID) {
**response = 0;
return 0;
}
size_t resp_size = BUFSIZ;
buffer = malloc(BUFSIZ);
for (Tanzaku *temp = list; temp->id != HOLE_ID; temp++) {
if (temp->id == HOLE_ID) {
continue;
}
char *escaped_name = escape(temp->name),
*escaped_description = escape(temp->description);
sprintf(buffer, "{\"id\":%lu,\"cts\":%lu,\"mts\":%lu,\"name\":\"%s\",\"desc\":\"%s\"},",
temp->id, temp->created_ts, temp->modified_ts, escaped_name, escaped_description);
free(escaped_name);
free(escaped_description);
if (strlen(*response) + strlen(buffer) >= resp_size) {
resp_size += BUFSIZ;
*response = realloc(*response, resp_size);
}
strcat(*response, buffer);
}
free(list);
sprintf(buffer, "]}");
if (strlen(*response) + 2 >= resp_size) {
*response = realloc(*response, resp_size + 2);
}
strcpy(*response + strlen(*response) - 1, buffer);
free(buffer);
return 0;
}
if (request_code == trc_tanzaku_add) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *name = request_body;
for (; *request_body != '\n' && *request_body != 0; request_body++);
if (*request_body == 0) {
return 1;
}
*request_body = 0;
char *description = request_body + 1;
return tanabata_tanzaku_add(tanabata, name, description);
}
if (request_code == trc_tanzaku_update) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t tanzaku_id = strtoull(request_body, &endptr, 0);
if (*endptr != ' ') {
return 1;
}
char *name = endptr + 1;
for (endptr++; *endptr != '\n' && *endptr != 0; endptr++);
if (*endptr == 0) {
return 1;
}
*endptr = 0;
char *description = endptr + 1;
return tanabata_tanzaku_upd(tanabata, tanzaku_id, name, description);
}
if (request_code == trc_tanzaku_remove) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t tanzaku_id = strtoull(request_body, &endptr, 0);
if (*endptr != 0) {
return 1;
}
return tanabata_tanzaku_rem(tanabata, tanzaku_id);
}
if (request_code == trc_tanzaku_remove_by_sasa) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t sasa_id = strtoull(request_body, &endptr, 0);
if (*endptr != 0) {
return 1;
}
Tanzaku *list = tanabata_tanzaku_get_by_sasa(tanabata, sasa_id);
if (list == NULL) {
return 1;
}
for (Tanzaku *temp = list; temp->id != HOLE_ID; temp++) {
tanabata_tanzaku_rem(tanabata, temp->id);
}
free(list);
return 0;
}
if (request_code == trc_kazari_get) {
if (tanabata == NULL) {
return 1;
}
if (tanabata->shoppyou.size - tanabata->shoppyou.hole_cnt == 0) {
sprintf(*response, "{\"status\":true,\"data\":[]}");
return 0;
}
size_t resp_size = BUFSIZ;
buffer = malloc(BUFSIZ);
Kazari *temp = tanabata->shoppyou.database;
for (uint64_t i = 0; i < tanabata->shoppyou.size; i++, temp++) {
if (temp->sasa_id == HOLE_ID || temp->tanzaku_id == HOLE_ID) {
continue;
}
sprintf(buffer, "{\"cts\":%lu,\"sasa_id\":%lu,\"tanzaku_id\":%lu},",
temp->created_ts, temp->sasa_id, temp->tanzaku_id);
if (strlen(*response) + strlen(buffer) >= resp_size) {
resp_size += BUFSIZ;
*response = realloc(*response, resp_size);
}
strcat(*response, buffer);
}
sprintf(buffer, "]}");
if (strlen(*response) + 2 >= resp_size) {
*response = realloc(*response, resp_size + 2);
}
strcpy(*response + strlen(*response) - 1, buffer);
free(buffer);
return 0;
}
if (request_code == trc_kazari_add) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t sasa_id = strtoull(request_body, &endptr, 0);
if (*endptr != ' ') {
return 1;
}
uint64_t tanzaku_id = strtoull(endptr + 1, &endptr, 0);
if (*endptr != 0) {
return 1;
}
return tanabata_kazari_add(tanabata, sasa_id, tanzaku_id);
}
if (request_code == trc_kazari_add_single_sasa_to_multiple_tanzaku) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t sasa_id = strtoull(request_body, &endptr, 0), tanzaku_id;
if (*endptr != ' ') {
return 1;
}
for (endptr++; endptr != 0;) {
tanzaku_id = strtoull(endptr, &endptr, 0);
if (*endptr != ' ' && *endptr != 0 || tanabata_kazari_add(tanabata, sasa_id, tanzaku_id)) {
return 1;
}
}
return 0;
}
if (request_code == trc_kazari_add_single_tanzaku_to_multiple_sasa) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t tanzaku_id = strtoull(request_body, &endptr, 0), sasa_id;
if (*endptr != ' ') {
return 1;
}
for (endptr++; endptr != 0;) {
sasa_id = strtoull(endptr, &endptr, 0);
if (*endptr != ' ' && *endptr != 0 || tanabata_kazari_add(tanabata, sasa_id, tanzaku_id)) {
return 1;
}
}
return 0;
}
if (request_code == trc_kazari_remove) {
if (tanabata == NULL) {
return 1;
}
**response = 0;
char *endptr;
uint64_t sasa_id = strtoull(request_body, &endptr, 0);
if (*endptr != ' ') {
return 1;
}
uint64_t tanzaku_id = strtoull(endptr + 1, &endptr, 0);
if (*endptr != 0) {
return 1;
}
return tanabata_kazari_rem(tanabata, sasa_id, tanzaku_id);
}
return 1;
}
// Client thread
void *client_thread(void *arg) {
uint16_t thread_id = 1 + (pthread_t *) arg - threads;
int client_fd;
unsigned int critical_count = 0;
char buffer[BUFSIZ];
ssize_t nread, nwrite;
size_t req_size, resp_size;
char *request = NULL, *response = NULL;
for (; critical_count < max_criticals;) {
pthread_mutex_lock(&mutex_connection);
if ((client_fd = accept(server_fd, NULL, NULL)) < 0) {
logtf(LOG_CRITICAL, "thread %i: failed to accept connection", thread_id);
critical_count++;
continue;
}
logtf(LOG_INFO, "thread %i: client connected", thread_id);
pthread_mutex_unlock(&mutex_connection);
critical_count = 0;
for (;;) {
req_size = BUFSIZ;
request = malloc(req_size);
for (off_t offset = 0; (nread = read(client_fd, buffer, BUFSIZ)) > 0;) {
if (offset + nread > req_size) {
req_size += BUFSIZ;
request = realloc(request, req_size);
}
memcpy(request + offset, buffer, nread);
offset += nread;
if (request[offset - 1] == EOT) {
break;
}
}
if (nread < 0) {
logtf(LOG_ERROR, "thread %i: failed to read request", thread_id);
} else if (nread > 0) {
logtf(LOG_INFO, "thread %i: request got", thread_id);
pthread_mutex_lock(&mutex_request);
if (execute(request, &response) != 0) {
logtf(LOG_INFO, "thread %i: request executed, status 'false'", thread_id);
strcpy(response, "{\"status\":false,\"data\":[]}");
} else {
logtf(LOG_INFO, "thread %i: request executed, status 'true'", thread_id);
if (*response == 0) {
strcpy(response, "{\"status\":true,\"data\":[]}");
}
}
pthread_mutex_unlock(&mutex_request);
resp_size = strlen(response) + 1;
for (char *buf = response; (nwrite = write(client_fd, buf, resp_size)) > 0;) {
buf += nwrite;
resp_size -= nwrite;
if (resp_size == 0) {
nwrite = write(client_fd, "\4", 1);
break;
}
}
if (nwrite <= 0) {
logtf(LOG_ERROR, "thread %i: failed to send response", thread_id);
}
logtf(LOG_INFO, "thread %i: response sent", thread_id);
}
free(request);
free(response);
request = NULL;
response = NULL;
if (nread == 0) {
break;
}
}
close(client_fd);
logtf(LOG_INFO, "thread %i: connection closed", thread_id);
}
logtf(LOG_INFO, "thread %i: thread finished", thread_id);
return NULL;
}
int main() {
logtf(LOG_INFO, "TDBMS server started");
sock_domain = malloc(strlen(DEFAULT_SOCK_DOMAIN) + 1);
strcpy(sock_domain, DEFAULT_SOCK_DOMAIN);
sock_addr = malloc(strlen(DEFAULT_SOCK_ADDR) + 1);
strcpy(sock_addr, DEFAULT_SOCK_ADDR);
config_load();
if (logfile == NULL &&
(logfile = fopen(DEFAULT_LOG_FILE, "a")) == NULL) {
logtf(LOG_FATAL, "failed to load default log file");
return 1;
}
logtf(LOG_INFO, "configuration loaded");
dblist_load();
server_fd = socket_open();
threads = malloc(max_clients * sizeof(pthread_t));
for (uint16_t i = 0; i < max_clients; i++) {
if (pthread_create(threads + i, NULL, client_thread, threads + i) != 0) {
logtf(LOG_CRITICAL, "failed to create a thread");
}
}
for (unsigned int i = 0; i < max_clients; i++) {
if (pthread_join(threads[i], NULL) != 0) {
logtf(LOG_CRITICAL, "failed to join a thread");
}
}
close(server_fd);
return 0;
}
Reference in New Issue View Git Blame Copy Permalink