diff options
Diffstat (limited to '')
| -rw-r--r-- | src/collation-functions.cc | 155 |
1 files changed, 155 insertions, 0 deletions
diff --git a/src/collation-functions.cc b/src/collation-functions.cc new file mode 100644 index 0000000..8ad474b --- /dev/null +++ b/src/collation-functions.cc @@ -0,0 +1,155 @@ +/** + * Copyright (c) 2013, Timothy Stack + * + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * * Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * * Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * * Neither the name of Timothy Stack nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ''AS IS'' AND ANY + * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY + * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; + * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND + * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS + * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + * + * @file logfile_sub_source.hh + */ + +#include <arpa/inet.h> +#include <netinet/in.h> +#include <sqlite3.h> +#include <string.h> +#include <sys/socket.h> + +#include "base/strnatcmp.h" +#include "config.h" +#include "log_level.hh" + +constexpr int MAX_ADDR_LEN = 128; + +static int +try_inet_pton(int p_len, const char* p, char* n) +{ + static int ADDR_FAMILIES[] = {AF_INET, AF_INET6}; + + char buf[MAX_ADDR_LEN + 1]; + int retval = AF_MAX; + + strncpy(buf, p, p_len); + buf[p_len] = '\0'; + for (int family : ADDR_FAMILIES) { + if (inet_pton(family, buf, n) == 1) { + retval = family; + break; + } + } + + return retval; +} + +static int +convert_v6_to_v4(int family, char* n) +{ + struct in6_addr* ia = (struct in6_addr*) n; + + if (family == AF_INET6 + && (IN6_IS_ADDR_V4COMPAT(ia) || IN6_IS_ADDR_V4MAPPED(ia))) + { + family = AF_INET; + memmove(n, n + 12, sizeof(struct in_addr)); + } + + return family; +} + +static int +ipaddress(void* ptr, int a_len, const void* a_in, int b_len, const void* b_in) +{ + char a_addr[sizeof(struct in6_addr)], b_addr[sizeof(struct in6_addr)]; + const char *a_str = (const char*) a_in, *b_str = (const char*) b_in; + int a_family, b_family, retval; + + if ((a_len > MAX_ADDR_LEN) || (b_len > MAX_ADDR_LEN)) { + return strnatcasecmp(a_len, a_str, b_len, b_str); + } + + int v4res = 0; + if (ipv4cmp(a_len, a_str, b_len, b_str, &v4res)) { + return v4res; + } + + a_family = try_inet_pton(a_len, a_str, a_addr); + b_family = try_inet_pton(b_len, b_str, b_addr); + + if (a_family == AF_MAX && b_family == AF_MAX) { + return strnatcasecmp(a_len, a_str, b_len, b_str); + } else if (a_family == AF_MAX && b_family != AF_MAX) { + retval = -1; + } else if (a_family != AF_MAX && b_family == AF_MAX) { + retval = 1; + } else { + a_family = convert_v6_to_v4(a_family, a_addr); + b_family = convert_v6_to_v4(b_family, b_addr); + if (a_family == b_family) { + retval = memcmp(a_addr, + b_addr, + a_family == AF_INET ? sizeof(struct in_addr) + : sizeof(struct in6_addr)); + } else if (a_family == AF_INET) { + retval = -1; + } else { + retval = 1; + } + } + + return retval; +} + +static int +sql_strnatcmp( + void* ptr, int a_len, const void* a_in, int b_len, const void* b_in) +{ + return strnatcmp(a_len, (char*) a_in, b_len, (char*) b_in); +} + +static int +sql_strnatcasecmp( + void* ptr, int a_len, const void* a_in, int b_len, const void* b_in) +{ + return strnatcasecmp(a_len, (char*) a_in, b_len, (char*) b_in); +} + +static int +sql_loglevelcmp( + void* ptr, int a_len, const void* a_in, int b_len, const void* b_in) +{ + return levelcmp((const char*) a_in, a_len, (const char*) b_in, b_len); +} + +int +register_collation_functions(sqlite3* db) +{ + sqlite3_create_collation(db, "ipaddress", SQLITE_UTF8, nullptr, ipaddress); + sqlite3_create_collation( + db, "naturalcase", SQLITE_UTF8, nullptr, sql_strnatcmp); + sqlite3_create_collation( + db, "naturalnocase", SQLITE_UTF8, nullptr, sql_strnatcasecmp); + sqlite3_create_collation( + db, "loglevel", SQLITE_UTF8, nullptr, sql_loglevelcmp); + + return 0; +} |