291 lines
8.8 KiB
C
291 lines
8.8 KiB
C
/**
|
|
* sesimos - secure, simple, modern web server
|
|
* @brief Utilities
|
|
* @file src/lib/utils.c
|
|
* @author Lorenz Stechauner
|
|
* @date 2020-12-03
|
|
*/
|
|
|
|
#include "utils.h"
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <time.h>
|
|
#include <sys/stat.h>
|
|
#include <errno.h>
|
|
#include <unistd.h>
|
|
#include <dirent.h>
|
|
|
|
|
|
static const char base64_encode_table[64] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
|
static const int base64_mod_table[3] = {0, 2, 1};
|
|
|
|
|
|
char *format_duration(unsigned long micros, char *buf) {
|
|
if (micros < 10000) {
|
|
sprintf(buf, "%.1f ms", (double) micros / 1000);
|
|
} else if (micros < 1000000 - 1000) {
|
|
sprintf(buf, "%.0f ms", (double) micros / 1000);
|
|
} else if (micros < 60000000 - 1000000) {
|
|
sprintf(buf, "%.1f s", (double) micros / 1000000);
|
|
} else if (micros < 6000000000) {
|
|
sprintf(buf, "%li:%02li min", micros / 1000000 / 60, micros / 1000000 % 60);
|
|
} else {
|
|
sprintf(buf, "%.0f min", (double) micros / 1000000 / 60);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
int url_encode_component(const void *in, size_t size_in, char *out, size_t size_out) {
|
|
int size = 0;
|
|
|
|
// Encode control characters
|
|
for (int i = 0; i < size_in; i++) {
|
|
unsigned char ch = ((unsigned char *) in)[i];
|
|
if (ch == ' ') {
|
|
ch = '+';
|
|
} else if (
|
|
ch <= 0x20 || ch >= 0x7F ||
|
|
!((ch >= 'A' && ch <= 'Z') || (ch >= 'a' && ch <= 'z') || (ch >= '0' && ch <= '9') ||
|
|
ch == '-' || ch == '_' || ch == '.' || ch == '!' || ch == '~' || ch == '*' || ch == '\'' ||
|
|
ch == '(' || ch == ')')
|
|
) {
|
|
size += 3;
|
|
if (size < size_out) sprintf(out + size - 3, "%%%02X", ch);
|
|
ch = 0;
|
|
}
|
|
|
|
if (ch != 0) {
|
|
size++;
|
|
if (size < size_out) out[size - 1] = (char) ch;
|
|
}
|
|
}
|
|
|
|
// Set terminating null byte
|
|
if (size_out > 0) out[size < size_out ? size : size_out - 1] = 0;
|
|
|
|
// Return theoretical size
|
|
return size;
|
|
}
|
|
|
|
int url_encode(const void *in, size_t size_in, char *out, size_t size_out) {
|
|
int size = 0;
|
|
|
|
// Encode control characters
|
|
for (int i = 0; i < size_in; i++) {
|
|
unsigned char ch = ((unsigned char *) in)[i];
|
|
if (ch <= 0x20 || ch >= 0x7F) {
|
|
size += 3;
|
|
if (size < size_out) sprintf(out + size - 3, "%%%02X", ch);
|
|
} else {
|
|
size++;
|
|
if (size < size_out) out[size - 1] = (char) ch;
|
|
}
|
|
}
|
|
|
|
// Set terminating null byte
|
|
if (size_out > 0) out[size < size_out ? size : size_out - 1] = 0;
|
|
|
|
// Return theoretical size
|
|
return size;
|
|
}
|
|
|
|
int url_decode(const char *str, char *dec, long *size) {
|
|
char *ptr = dec;
|
|
char ch, buf[3];
|
|
memset(dec, 0, *size);
|
|
for (int i = 0; i < strlen(str); i++, ptr++) {
|
|
if ((ptr - dec) >= *size) {
|
|
return -1;
|
|
}
|
|
ch = str[i];
|
|
if (ch == '+') {
|
|
ch = ' ';
|
|
} else if (ch == '%') {
|
|
memcpy(buf, str + i + 1, 2);
|
|
buf[2] = 0;
|
|
ch = (char) strtol(buf, NULL, 16);
|
|
i += 2;
|
|
} else if (ch == '?') {
|
|
strcpy(ptr, str + i);
|
|
break;
|
|
}
|
|
ptr[0] = ch;
|
|
}
|
|
*size = ptr - dec;
|
|
return 0;
|
|
}
|
|
|
|
int mime_is_compressible(const char *type) {
|
|
if (type == NULL) return 0;
|
|
char type_parsed[64];
|
|
snprintf(type_parsed, sizeof(type_parsed), "%s", type);
|
|
char *pos = strchr(type_parsed, ';');
|
|
if (pos != NULL) pos[0] = 0;
|
|
return
|
|
strstarts(type_parsed, "text/") ||
|
|
strstarts(type_parsed, "message/") ||
|
|
strends(type_parsed, "+xml") ||
|
|
strends(type_parsed, "+json") ||
|
|
streq(type_parsed, "application/javascript") ||
|
|
streq(type_parsed, "application/json") ||
|
|
streq(type_parsed, "application/xml") ||
|
|
streq(type_parsed, "application/x-www-form-urlencoded") ||
|
|
streq(type_parsed, "application/x-tex") ||
|
|
streq(type_parsed, "application/x-httpd-php") ||
|
|
streq(type_parsed, "application/x-latex") ||
|
|
streq(type_parsed, "application/vnd.ms-fontobject") ||
|
|
streq(type_parsed, "application/x-font-ttf") ||
|
|
streq(type_parsed, "application/x-javascript") ||
|
|
streq(type_parsed, "font/eot") ||
|
|
streq(type_parsed, "font/opentype") ||
|
|
streq(type_parsed, "image/bmp") ||
|
|
streq(type_parsed, "image/gif") ||
|
|
streq(type_parsed, "image/vnd.microsoft.icon") ||
|
|
streq(type_parsed, "image/vnd.microsoft.iconbinary") ||
|
|
streq(type_parsed, "image/x-icon");
|
|
}
|
|
|
|
int strcpy_rem_webroot(char *dst, const char *src, long len, const char *webroot) {
|
|
memcpy(dst, src, len);
|
|
dst[len] = 0;
|
|
if (webroot == NULL)
|
|
return 0;
|
|
|
|
char *pos;
|
|
const unsigned long webroot_len = strlen(webroot);
|
|
if (webroot_len == 0)
|
|
return 0;
|
|
|
|
while ((pos = strstr(dst, webroot)) != NULL) {
|
|
strcpy(pos, pos + webroot_len);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int str_trim(char **start, char **end) {
|
|
if (start == NULL || end == NULL || *start == NULL || *end == NULL)
|
|
return -1;
|
|
|
|
(*end)--;
|
|
while (*start[0] == ' ' || *start[0] == '\t' || *start[0] == '\r' || *start[0] == '\n') (*start)++;
|
|
while (*end[0] == ' ' || *end[0] == '\t' || *end[0] == '\r' || *end[0] == '\n') (*end)--;
|
|
(*end)++;
|
|
return 0;
|
|
}
|
|
|
|
int str_trim_lws(char **start, char **end) {
|
|
if (start == NULL || end == NULL || *start == NULL || *end == NULL)
|
|
return -1;
|
|
|
|
(*end)--;
|
|
while (*start[0] == ' ' || *start[0] == '\t') (*start)++;
|
|
while (*end[0] == ' ' || *end[0] == '\t') (*end)--;
|
|
(*end)++;
|
|
return 0;
|
|
}
|
|
|
|
int streq(const char *restrict str1, const char *restrict str2) {
|
|
return str1 != NULL && str2 != NULL && strcmp(str1, str2) == 0;
|
|
}
|
|
|
|
int strcontains(const char *restrict haystack, const char *restrict needle) {
|
|
return haystack != NULL && needle != NULL && strstr(haystack, needle) != NULL;
|
|
}
|
|
|
|
int strstarts(const char *restrict str, const char *restrict prefix) {
|
|
if (str == NULL || prefix == NULL) return 0;
|
|
unsigned long l1 = strlen(str), l2 = strlen(prefix);
|
|
return l2 <= l1 && strncmp(str, prefix, l2) == 0;
|
|
}
|
|
|
|
int strends(const char *restrict str, const char *restrict suffix) {
|
|
if (str == NULL || suffix == NULL) return 0;
|
|
unsigned long l1 = strlen(str), l2 = strlen(suffix);
|
|
return l2 <= l1 && strcmp(str + l1 - l2, suffix) == 0;
|
|
}
|
|
|
|
int base64_encode(void *data, unsigned long data_len, char *output, unsigned long *output_len) {
|
|
unsigned long out_len = 4 * ((data_len + 2) / 3);
|
|
if (output_len != NULL) *output_len = out_len;
|
|
|
|
for (int i = 0, j = 0; i < data_len;) {
|
|
unsigned int octet_a = (i < data_len) ? ((unsigned char *) data)[i++] : 0;
|
|
unsigned int octet_b = (i < data_len) ? ((unsigned char *) data)[i++] : 0;
|
|
unsigned int octet_c = (i < data_len) ? ((unsigned char *) data)[i++] : 0;
|
|
unsigned int triple = (octet_a << 0x10) + (octet_b << 0x08) + octet_c;
|
|
output[j++] = base64_encode_table[(triple >> 3 * 6) & 0x3F];
|
|
output[j++] = base64_encode_table[(triple >> 2 * 6) & 0x3F];
|
|
output[j++] = base64_encode_table[(triple >> 1 * 6) & 0x3F];
|
|
output[j++] = base64_encode_table[(triple >> 0 * 6) & 0x3F];
|
|
}
|
|
|
|
for (int i = 0; i < base64_mod_table[data_len % 3]; i++)
|
|
output[out_len - 1 - i] = '=';
|
|
output[out_len] = 0;
|
|
|
|
return 0;
|
|
}
|
|
|
|
long clock_micros(void) {
|
|
struct timespec time;
|
|
clock_gettime(CLOCK_MONOTONIC, &time);
|
|
return time.tv_sec * 1000000 + time.tv_nsec / 1000;
|
|
}
|
|
|
|
long clock_cpu(void) {
|
|
struct timespec time;
|
|
clock_gettime(CLOCK_THREAD_CPUTIME_ID, &time);
|
|
return time.tv_sec * 1000000000 + time.tv_nsec;
|
|
}
|
|
|
|
long stat_mtime(const char *filename) {
|
|
struct stat stat_buf;
|
|
stat(filename, &stat_buf);
|
|
return stat_buf.st_mtime;
|
|
}
|
|
|
|
int rm_rf(const char *path) {
|
|
struct stat stat_buf;
|
|
if (lstat(path, &stat_buf) != 0)
|
|
return (errno == ENOENT) ? 0 : -1;
|
|
|
|
if (S_ISREG(stat_buf.st_mode)) {
|
|
// regular file
|
|
return unlink(path);
|
|
} else if (S_ISLNK(stat_buf.st_mode)) {
|
|
// link
|
|
return unlink(path);
|
|
} else if (S_ISDIR(stat_buf.st_mode)) {
|
|
// directory
|
|
char buf[FILENAME_MAX];
|
|
DIR *dir;
|
|
|
|
// open directory
|
|
if ((dir = opendir(path)) == NULL)
|
|
return -1;
|
|
|
|
// read directory
|
|
for (struct dirent *ent; (ent = readdir(dir)) != NULL;) {
|
|
if (streq(ent->d_name, ".") || streq(ent->d_name, ".."))
|
|
continue;
|
|
|
|
snprintf(buf, sizeof(buf), "%s/%s", path, ent->d_name);
|
|
if (rm_rf(buf) != 0) {
|
|
closedir(dir);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
// close and remove directory
|
|
closedir(dir);
|
|
return rmdir(path);
|
|
} else {
|
|
// other - not supported
|
|
errno = ENOTSUP;
|
|
return -1;
|
|
}
|
|
}
|