123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445 |
- /*
- * Copyright 2023 Vsevolod Stakhov
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
- #include "config.h"
- #include "cryptobox.h"
- #include "base64.h"
- #include "platform_config.h"
- #include "str_util.h"
- #include "util.h"
- #include "contrib/libottery/ottery.h"
-
- extern unsigned cpu_config;
- const uint8_t
- base64_table_dec[256] =
- {
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 62,
- 255,
- 255,
- 255,
- 63,
- 52,
- 53,
- 54,
- 55,
- 56,
- 57,
- 58,
- 59,
- 60,
- 61,
- 255,
- 255,
- 255,
- 254,
- 255,
- 255,
- 255,
- 0,
- 1,
- 2,
- 3,
- 4,
- 5,
- 6,
- 7,
- 8,
- 9,
- 10,
- 11,
- 12,
- 13,
- 14,
- 15,
- 16,
- 17,
- 18,
- 19,
- 20,
- 21,
- 22,
- 23,
- 24,
- 25,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 26,
- 27,
- 28,
- 29,
- 30,
- 31,
- 32,
- 33,
- 34,
- 35,
- 36,
- 37,
- 38,
- 39,
- 40,
- 41,
- 42,
- 43,
- 44,
- 45,
- 46,
- 47,
- 48,
- 49,
- 50,
- 51,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- 255,
- };
-
- static const char base64_alphabet[] = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz+/";
-
- typedef struct base64_impl {
- unsigned short enabled;
- unsigned short min_len;
- unsigned int cpu_flags;
- const char *desc;
- int (*decode)(const char *in, size_t inlen,
- unsigned char *out, size_t *outlen);
- } base64_impl_t;
-
- #define BASE64_DECLARE(ext) \
- int base64_decode_##ext(const char *in, size_t inlen, unsigned char *out, size_t *outlen);
- #define BASE64_IMPL(cpuflags, min_len, desc, ext) \
- { \
- 0, (min_len), (cpuflags), desc, base64_decode_##ext \
- }
-
- BASE64_DECLARE(ref);
- #define BASE64_REF BASE64_IMPL(0, 0, "ref", ref)
-
- #ifdef RSPAMD_HAS_TARGET_ATTR
- #if defined(HAVE_SSE42) && defined(__x86_64__)
- int base64_decode_sse42(const char *in, size_t inlen,
- unsigned char *out, size_t *outlen) __attribute__((__target__("sse4.2")));
-
- BASE64_DECLARE(sse42);
- #define BASE64_SSE42 BASE64_IMPL(CPUID_SSE42, 24, "sse42", sse42)
- #endif
- #endif
-
- #ifdef RSPAMD_HAS_TARGET_ATTR
- #if defined(HAVE_AVX2) && defined(__x86_64__)
- int base64_decode_avx2(const char *in, size_t inlen,
- unsigned char *out, size_t *outlen) __attribute__((__target__("avx2")));
-
- BASE64_DECLARE(avx2);
- #define BASE64_AVX2 BASE64_IMPL(CPUID_AVX2, 128, "avx2", avx2)
- #endif
- #endif
-
- static base64_impl_t base64_list[] = {
- BASE64_REF,
- #ifdef BASE64_SSE42
- BASE64_SSE42,
- #endif
- #ifdef BASE64_AVX2
- BASE64_AVX2,
- #endif
- };
-
- static const base64_impl_t *base64_ref = &base64_list[0];
-
- const char *
- base64_load(void)
- {
- unsigned int i;
- const base64_impl_t *opt_impl = base64_ref;
-
- /* Enable reference */
- base64_list[0].enabled = true;
-
- if (cpu_config != 0) {
- for (i = 1; i < G_N_ELEMENTS(base64_list); i++) {
- if (base64_list[i].cpu_flags & cpu_config) {
- base64_list[i].enabled = true;
- opt_impl = &base64_list[i];
- }
- }
- }
-
-
- return opt_impl->desc;
- }
-
- gboolean
- rspamd_cryptobox_base64_decode(const char *in, gsize inlen,
- unsigned char *out, gsize *outlen)
- {
- const base64_impl_t *opt_impl = base64_ref;
-
- for (int i = G_N_ELEMENTS(base64_list) - 1; i > 0; i--) {
- if (base64_list[i].enabled && base64_list[i].min_len <= inlen) {
- opt_impl = &base64_list[i];
- break;
- }
- }
-
- return opt_impl->decode(in, inlen, out, outlen);
- }
-
- double
- base64_test(bool generic, size_t niters, size_t len, size_t str_len)
- {
- size_t cycles;
- unsigned char *in, *out, *tmp;
- double t1, t2, total = 0;
- gsize outlen;
-
- g_assert(len > 0);
- in = g_malloc(len);
- tmp = g_malloc(len);
- ottery_rand_bytes(in, len);
-
- out = rspamd_encode_base64_fold(in, len, str_len, &outlen,
- RSPAMD_TASK_NEWLINES_CRLF);
-
- if (generic) {
- base64_list[0].decode(out, outlen, tmp, &len);
- }
- else {
- rspamd_cryptobox_base64_decode(out, outlen, tmp, &len);
- }
-
- g_assert(memcmp(in, tmp, len) == 0);
-
- for (cycles = 0; cycles < niters; cycles++) {
- t1 = rspamd_get_ticks(TRUE);
- if (generic) {
- base64_list[0].decode(out, outlen, tmp, &len);
- }
- else {
- rspamd_cryptobox_base64_decode(out, outlen, tmp, &len);
- }
- t2 = rspamd_get_ticks(TRUE);
- total += t2 - t1;
- }
-
- g_free(in);
- g_free(tmp);
- g_free(out);
-
- return total;
- }
-
-
- gboolean
- rspamd_cryptobox_base64_is_valid(const char *in, gsize inlen)
- {
- const unsigned char *p, *end;
-
- if (inlen == 0) {
- return FALSE;
- }
-
- p = in;
- end = in + inlen;
-
- while (p < end && *p != '=') {
- if (!g_ascii_isspace(*p)) {
- if (base64_table_dec[*p] == 255) {
- return FALSE;
- }
- }
- p++;
- }
-
- return TRUE;
- }
|