149 lines
5.9 KiB
C
149 lines
5.9 KiB
C
#include <iconv.h>
|
||
#include <stdint.h>
|
||
|
||
#include <libreborn/string.h>
|
||
|
||
// Sanitize String
|
||
void sanitize_string(char **str, int max_length, unsigned int allow_newlines) {
|
||
// Store Message Length
|
||
int length = strlen(*str);
|
||
// Truncate Message
|
||
if (max_length != -1 && length > max_length) {
|
||
(*str)[max_length] = '\0';
|
||
length = max_length;
|
||
}
|
||
// Loop Through Message
|
||
if (!allow_newlines) {
|
||
for (int i = 0; i < length; i++) {
|
||
if ((*str)[i] == '\n' || (*str)[i] == '\r') {
|
||
// Replace Newline
|
||
(*str)[i] = ' ';
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// Minecraft-Flavored CP437
|
||
void safe_iconv(iconv_t cd, char *input, size_t input_size, char *output, size_t output_size) {
|
||
iconv(cd, &input, &input_size, &output, &output_size);
|
||
}
|
||
#define CP437_CHARACTERS 256
|
||
static const char *cp437_characters_map[CP437_CHARACTERS] = {
|
||
"\0", "☺", "☻", "♥", "♦", "♣", "♠", "•", "◘", "○", "\n", "♂", "♀", "\r", "♫", "☼",
|
||
"►", "◄", "↕", "‼", "¶", "§", "▬", "↨", "↑", "↓", "→", "←", "∟", "↔", "▲", "▼",
|
||
" ", "!", "\"", "#", "$", "%", "&", "'", "(", ")", "*", "+", ",", "-", ".", "/",
|
||
"0", "1", "2", "3", "4", "5", "6", "7", "8", "9", ":", ";", "<", "=", ">", "?",
|
||
"@", "A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O",
|
||
"P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z", "[", "\\", "]", "^", "_",
|
||
"`", "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o",
|
||
"p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z", "{", "|", "}", "~", "⌂",
|
||
"Ç", "ü", "é", "â", "ä", "à", "å", "ç", "ê", "ë", "è", "ï", "î", "ì", "Ä", "Å",
|
||
"É", "æ", "Æ", "ô", "ö", "ò", "û", "ù", "ÿ", "Ö", "Ü", "¢", "£", "¥", "₧", "ƒ",
|
||
"á", "í", "ó", "ú", "ñ", "Ñ", "ª", "º", "¿", "⌐", "¬", "½", "¼", "¡", "«", "»",
|
||
"░", "▒", "▓", "│", "┤", "╡", "╢", "╖", "╕", "╣", "║", "╗", "╝", "╜", "╛", "┐",
|
||
"└", "┴", "┬", "├", "─", "┼", "╞", "╟", "╚", "╔", "╩", "╦", "╠", "═", "╬", "╧",
|
||
"╨", "╤", "╥", "╙", "╘", "╒", "╓", "╫", "╪", "┘", "┌", "█", "▄", "▌", "▐", "▀",
|
||
"α", "ß", "Γ", "π", "Σ", "σ", "µ", "τ", "Φ", "Θ", "Ω", "δ", "∞", "φ", "ε", "∩",
|
||
"≡", "±", "≥", "≤", "⌠", "⌡", "÷", "≈", "°", "∙", "·", "√", "ⁿ", "²", "■", "©"
|
||
};
|
||
static uint32_t *get_cp437_characters_codepoint_map() {
|
||
static uint32_t map[CP437_CHARACTERS];
|
||
static int is_setup = 0;
|
||
if (!is_setup) {
|
||
// Build Map
|
||
iconv_t cd = iconv_open("UTF-32LE", "UTF-8");
|
||
if (cd != (iconv_t) -1) {
|
||
size_t str_size = 4;
|
||
uint32_t *str = (uint32_t *) malloc(str_size);
|
||
ALLOC_CHECK(str);
|
||
for (int i = 0; i < CP437_CHARACTERS; i++) {
|
||
// Convert to UTF-32, Then Extract Codepoint
|
||
safe_iconv(cd, (char *) cp437_characters_map[i], strlen(cp437_characters_map[i]), (char *) str, str_size);
|
||
// Extract
|
||
map[i] = str[0];
|
||
}
|
||
// Free
|
||
free(str);
|
||
iconv_close(cd);
|
||
} else {
|
||
IMPOSSIBLE();
|
||
}
|
||
is_setup = 1;
|
||
}
|
||
return map;
|
||
}
|
||
char *to_cp437(const char *input) {
|
||
// Convert To UTF-32 For Easier Parsing
|
||
size_t in_size = strlen(input);
|
||
size_t utf32_str_size = in_size * 4;
|
||
size_t real_utf32_str_size = utf32_str_size + 4 /* NULL-terminator */;
|
||
uint32_t *utf32_str = (uint32_t *) malloc(real_utf32_str_size);
|
||
ALLOC_CHECK(utf32_str);
|
||
memset(utf32_str, 0, real_utf32_str_size);
|
||
iconv_t cd = iconv_open("UTF-32LE", "UTF-8");
|
||
if (cd != (iconv_t) -1) {
|
||
safe_iconv(cd, (char *) input, in_size, (char *) utf32_str, utf32_str_size);
|
||
iconv_close(cd);
|
||
} else {
|
||
IMPOSSIBLE();
|
||
}
|
||
// Allocate String
|
||
size_t cp437_str_size;
|
||
for (cp437_str_size = 0; utf32_str[cp437_str_size] != 0; cp437_str_size++);
|
||
size_t real_cp437_str_size = cp437_str_size + 1 /* NULL-terminator */;
|
||
char *cp437_str = (char *) malloc(real_cp437_str_size);
|
||
ALLOC_CHECK(cp437_str);
|
||
memset(cp437_str, 0, real_cp437_str_size);
|
||
// Handle Characters
|
||
for (size_t i = 0; utf32_str[i] != 0; i++) {
|
||
uint32_t codepoint = utf32_str[i];
|
||
for (int j = 0; j < CP437_CHARACTERS; j++) {
|
||
uint32_t test_codepoint = get_cp437_characters_codepoint_map()[j];
|
||
if (codepoint == test_codepoint) {
|
||
cp437_str[i] = j;
|
||
break;
|
||
}
|
||
}
|
||
if (cp437_str[i] == '\0') {
|
||
cp437_str[i] = '?';
|
||
}
|
||
}
|
||
// Free
|
||
free(utf32_str);
|
||
// Return
|
||
return cp437_str;
|
||
}
|
||
char *from_cp437(const char *input) {
|
||
// Convert To UTF-32 For Easier Parsing
|
||
size_t in_size = strlen(input);
|
||
size_t utf32_str_size = in_size * 4;
|
||
size_t real_utf32_str_size = utf32_str_size + 4 /* NULL-terminator */;
|
||
uint32_t *utf32_str = (uint32_t *) malloc(real_utf32_str_size);
|
||
ALLOC_CHECK(utf32_str);
|
||
memset(utf32_str, 0, real_utf32_str_size);
|
||
// Handle Characters
|
||
for (size_t i = 0; input[i] != '\0'; i++) {
|
||
utf32_str[i] = get_cp437_characters_codepoint_map()[(uint32_t) input[i]];
|
||
}
|
||
// Convert To UTF-8
|
||
size_t out_size = utf32_str_size;
|
||
size_t real_out_size = utf32_str_size + 1 /* NULL-terminator */;
|
||
char *output = (char *) malloc(real_out_size);
|
||
ALLOC_CHECK(output);
|
||
memset(output, 0, real_out_size);
|
||
iconv_t cd = iconv_open("UTF-8", "UTF-32LE");
|
||
if (cd != (iconv_t) -1) {
|
||
safe_iconv(cd, (char *) utf32_str, utf32_str_size, output, out_size);
|
||
iconv_close(cd);
|
||
} else {
|
||
IMPOSSIBLE();
|
||
}
|
||
// Return
|
||
return output;
|
||
}
|
||
|
||
// Starts With
|
||
int starts_with(const char *str, const char *prefix) {
|
||
return strncmp(prefix, str, strlen(prefix)) == 0;
|
||
}
|