diff options
Diffstat (limited to 'src/modules/rlm_totp')
-rw-r--r-- | src/modules/rlm_totp/.gitignore | 3 | ||||
-rw-r--r-- | src/modules/rlm_totp/Makefile | 33 | ||||
-rw-r--r-- | src/modules/rlm_totp/README.md | 9 | ||||
-rw-r--r-- | src/modules/rlm_totp/all.mk | 2 | ||||
-rw-r--r-- | src/modules/rlm_totp/rlm_totp.c | 615 | ||||
-rw-r--r-- | src/modules/rlm_totp/sha1.txt | 6 |
6 files changed, 668 insertions, 0 deletions
diff --git a/src/modules/rlm_totp/.gitignore b/src/modules/rlm_totp/.gitignore new file mode 100644 index 0000000..fb9b99c --- /dev/null +++ b/src/modules/rlm_totp/.gitignore @@ -0,0 +1,3 @@ +totp +src +freeradius-devel diff --git a/src/modules/rlm_totp/Makefile b/src/modules/rlm_totp/Makefile new file mode 100644 index 0000000..595cea6 --- /dev/null +++ b/src/modules/rlm_totp/Makefile @@ -0,0 +1,33 @@ +# +# TOTP isn't simple, so we need test cases. +# +all: totp + +include ../../../Make.inc + +# +# Hack up stuff so we can build in a subdirectory. +# +.PHONY: src +src: + @ln -sf ../../../src + +.PHONY: freeradius-devel +freeradius-devel: + @ln -sf ../../../src/include freeradius-devel + +# +# ./totp decode <base32> +# +# ./totp totp <time> <sha1key> <8-character-challenge> +# +totp: rlm_totp.c | src freeradius-devel + @$(CC) -DTESTING $(CFLAGS) $(CPPFLAGS) $(OPENSSL_CPPFLAGS) -o $@ $(LDFLAGS) $(LIBS) ../../../build/lib/.libs/libfreeradius-radius.a rlm_totp.c + +# +# Test vectors from RFC 6238, Appendix B +# +test: totp + @while IFS= read -r line; do \ + ./totp totp $$line || exit 1; \ + done < sha1.txt diff --git a/src/modules/rlm_totp/README.md b/src/modules/rlm_totp/README.md new file mode 100644 index 0000000..2dd5711 --- /dev/null +++ b/src/modules/rlm_totp/README.md @@ -0,0 +1,9 @@ +# rlm_totp +## Metadata +<dl> + <dt>category</dt><dd>authentication</dd> +</dl> + +## Summary + +Implement the TOTP algorithm as given in RFC 6238. diff --git a/src/modules/rlm_totp/all.mk b/src/modules/rlm_totp/all.mk new file mode 100644 index 0000000..4f58d0e --- /dev/null +++ b/src/modules/rlm_totp/all.mk @@ -0,0 +1,2 @@ +TARGET := rlm_totp.a +SOURCES := rlm_totp.c diff --git a/src/modules/rlm_totp/rlm_totp.c b/src/modules/rlm_totp/rlm_totp.c new file mode 100644 index 0000000..d58e1ee --- /dev/null +++ b/src/modules/rlm_totp/rlm_totp.c @@ -0,0 +1,615 @@ +/* + * This program is is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or (at + * your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + */ + +/** + * $Id$ + * @file rlm_totp.c + * @brief Execute commands and parse the results. + * + * @copyright 2021 The FreeRADIUS server project + * @copyright 2021 Network RADIUS SARL (legal@networkradius.com) + */ +RCSID("$Id$") + +#include <freeradius-devel/radiusd.h> +#include <freeradius-devel/modules.h> +#include <freeradius-devel/dlist.h> +#include <freeradius-devel/rad_assert.h> + +#include <ctype.h> + +typedef struct { + uint8_t const *key; + size_t keylen; + char const *passwd; + time_t when; + bool unlisted; + void *instance; + fr_dlist_t dlist; +} totp_dedup_t; + +#ifdef HAVE_PTHREAD_H +#include <pthread.h> + +#define PTHREAD_MUTEX_LOCK(_x) pthread_mutex_lock(&((_x)->mutex)) +#define PTHREAD_MUTEX_UNLOCK(_x) pthread_mutex_unlock(&((_x)->mutex)) +#else +#define PTHREAD_MUTEX_LOCK(_x) +#define PTHREAD_MUTEX_UNLOCK(_x) +#endif + + +/* Define a structure for the configuration variables */ +typedef struct rlm_totp_t { + char const *name; //!< name of this instance */ + uint32_t time_step; //!< seconds + uint32_t otp_length; //!< forced to 6 or 8 + uint32_t lookback_steps; //!< number of steps to look back + uint32_t lookback_interval; //!< interval in seconds between steps + uint32_t lookforward_steps; //!< number of steps to look forwards + rbtree_t *dedup_tree; + fr_dlist_t dedup_list; +#ifdef HAVE_PTHREAD_H + pthread_mutex_t mutex; +#endif +} rlm_totp_t; + +#ifndef TESTING +/* Map configuration file names to internal variables */ +static const CONF_PARSER module_config[] = { + { "time_step", FR_CONF_OFFSET(PW_TYPE_INTEGER, rlm_totp_t, time_step), "30" }, + { "otp_length", FR_CONF_OFFSET(PW_TYPE_INTEGER, rlm_totp_t, otp_length), "6" }, + { "lookback_steps", FR_CONF_OFFSET(PW_TYPE_INTEGER, rlm_totp_t, lookback_steps), "1" }, + { "lookback_interval", FR_CONF_OFFSET(PW_TYPE_INTEGER, rlm_totp_t, lookback_interval), "30" }, + { "lookforward_steps", FR_CONF_OFFSET(PW_TYPE_INTEGER, rlm_totp_t, lookforward_steps), "0" }, + CONF_PARSER_TERMINATOR +}; + +#define TIME_STEP (inst->time_step) +#define OTP_LEN (inst->otp_length) +#define BACK_STEPS (steps) +#define BACK_STEP_SECS (inst->lookback_interval) +#else +#define TIME_STEP (30) +#define OTP_LEN (8) +#define BACK_STEPS (1) +#define BACK_STEP_SECS (30) +#endif + +/* + * RFC 4648 base32 decoding. + */ +static const uint8_t alphabet[UINT8_MAX] = { + ['A'] = 1, + ['B'] = 2, + ['C'] = 3, + ['D'] = 4, + ['E'] = 5, + ['F'] = 6, + ['G'] = 7, + ['H'] = 8, + ['I'] = 9, + ['J'] = 10, + ['K'] = 11, + ['L'] = 12, + ['M'] = 13, + ['N'] = 14, + ['O'] = 15, + ['P'] = 16, + ['Q'] = 17, + ['R'] = 18, + ['S'] = 19, + ['T'] = 20, + ['U'] = 21, + ['V'] = 22, + ['W'] = 23, + ['X'] = 24, + ['Y'] = 25, + ['Z'] = 26, + ['2'] = 27, + ['3'] = 28, + ['4'] = 29, + ['5'] = 30, + ['6'] = 31, + ['7'] = 32, +}; + +static ssize_t base32_decode(uint8_t *out, size_t outlen, char const *in) +{ + uint8_t *p, *end, *b; + char const *q; + + p = out; + end = p + outlen; + + memset(out, 0, outlen); + + /* + * Convert ASCII to binary. + */ + for (q = in; *q != '\0'; q++) { + /* + * Padding at the end, stop. + */ + if (*q == '=') { + break; + } + + if (!alphabet[*((uint8_t const *) q)]) return -1; + + *(p++) = alphabet[*((uint8_t const *) q)] - 1; + + if (p == end) return -1; /* too much data */ + } + + /* + * Reset to the end of the actual data we have + */ + end = p; + + /* + * Convert input 5-bit groups into output 8-bit groups. + * We do this in 8-byte blocks. + * + * 00011111 00022222 00033333 00044444 00055555 00066666 00077777 00088888 + * + * Will get converted to + * + * 11111222 22333334 44445555 56666677 77788888 + */ + for (p = b = out; p < end; p += 8) { + b[0] = p[0] << 3; + b[0] |= p[1] >> 2; + + b[1] = p[1] << 6; + b[1] |= p[2] << 1; + b[1] |= p[3] >> 4; + + b[2] = p[3] << 4; + b[2] |= p[4] >> 1; + + b[3] = p[4] << 7; + b[3] |= p[5] << 2; + b[3] |= p[6] >> 3; + + b[4] = p[6] << 5; + b[4] |= p[7]; + + b += 5; + + /* + * Clear out the remaining 3 octets of this block. + */ + b[0] = 0; + b[1] = 0; + b[2] = 0; + } + + return b - out; +} + + +#ifndef TESTING +#define TESTING_UNUSED + +#else /* TESTING */ +#undef RDEBUG3 +#define RDEBUG3(fmt, ...) printf(fmt "\n", ## __VA_ARGS__) +#define TESTING_UNUSED UNUSED +#endif + +#ifndef TESTING +static int mod_bootstrap(CONF_SECTION *conf, void *instance) +{ + rlm_totp_t *inst = instance; + + inst->name = cf_section_name2(conf); + if (!inst->name) { + inst->name = cf_section_name1(conf); + } + + return 0; +} + +static int dedup_cmp(void const *one, void const *two) +{ + int rcode; + totp_dedup_t const *a = one; + totp_dedup_t const *b = two; + + if (a->keylen < b->keylen) return -1; + if (a->keylen > b->keylen) return +1; + + rcode = memcmp(a->key , b->key, a->keylen); + if (rcode != 0) return rcode; + + /* + * The user can enter multiple keys + */ + return strcmp(a->passwd, b->passwd); +} + +static void dedup_free(void *data) +{ + totp_dedup_t *dedup = data; +#ifdef HAVE_PTHREAD_H + rlm_totp_t *inst = dedup->instance; +#endif + + if (!dedup->unlisted) { + PTHREAD_MUTEX_LOCK(inst); + fr_dlist_entry_unlink(&dedup->dlist); + PTHREAD_MUTEX_UNLOCK(inst); + } + + free(dedup); +} + +/* + * Do any per-module initialization that is separate to each + * configured instance of the module. e.g. set up connections + * to external databases, read configuration files, set up + * dictionary entries, etc. + * + * If configuration information is given in the config section + * that must be referenced in later calls, store a handle to it + * in *instance otherwise put a null pointer there. + */ +static int mod_instantiate(UNUSED CONF_SECTION *conf, void *instance) +{ + rlm_totp_t *inst = instance; + + FR_INTEGER_BOUND_CHECK("time_step", inst->time_step, >=, 5); + FR_INTEGER_BOUND_CHECK("time_step", inst->time_step, <=, 120); + + FR_INTEGER_BOUND_CHECK("lookback_steps", inst->lookback_steps, >=, 1); + FR_INTEGER_BOUND_CHECK("lookback_steps", inst->lookback_steps, <=, 10); + + FR_INTEGER_BOUND_CHECK("lookforward_steps", inst->lookforward_steps, <=, 10); + + FR_INTEGER_BOUND_CHECK("lookback_interval", inst->lookback_interval, <=, inst->time_step); + + FR_INTEGER_BOUND_CHECK("otp_length", inst->otp_length, >=, 6); + FR_INTEGER_BOUND_CHECK("otp_length", inst->otp_length, <=, 8); + + if (inst->otp_length == 7) inst->otp_length = 8; + + inst->dedup_tree = rbtree_create(instance, dedup_cmp, dedup_free, 0); + if (!inst->dedup_tree) return -1; + + fr_dlist_entry_init(&inst->dedup_list); +#ifdef HAVE_PTHREAD_H + (void) pthread_mutex_init(&inst->mutex, NULL); +#endif + + return 0; +} + +#ifdef HAVE_PTHREAD_H +static int mod_detach(void *instance) +{ + rlm_totp_t *inst = instance; + + pthread_mutex_destroy(&inst->mutex); + return 0; +} +#endif +#endif + +/* + * Implement RFC 6238 TOTP algorithm. + * + * Appendix B has test vectors. Note that the test vectors are + * for 8-character challenges, and not for 6 character + * challenges! + */ +static int totp_cmp(TESTING_UNUSED REQUEST *request, time_t now, uint8_t const *key, size_t keylen, char const *totp, TESTING_UNUSED void *instance) +{ +#ifndef TESTING + rlm_totp_t *inst = instance; + uint32_t steps = inst->lookback_steps > inst->lookforward_steps ? inst->lookback_steps : inst->lookforward_steps; +#endif + time_t diff, then; + unsigned int i; + uint8_t offset; + uint32_t challenge; + uint64_t padded; + char buffer[9]; + uint8_t data[8]; + uint8_t digest[SHA1_DIGEST_LENGTH]; + + /* + * First try to authenticate against the current OTP, then step + * back in increments of BACK_STEP_SECS, up to BACK_STEPS times, + * to authenticate properly in cases of long transit delay, as + * described in RFC 6238, secion 5.2. + */ + + for (i = 0, diff = 0; i <= BACK_STEPS; i++, diff += BACK_STEP_SECS) { +#ifndef TESTING + if (i > inst->lookback_steps) goto forwards; +#endif + then = now - diff; +#ifndef TESTING + repeat: +#endif + padded = (uint64_t) then / TIME_STEP; + data[0] = padded >> 56; + data[1] = padded >> 48; + data[2] = padded >> 40; + data[3] = padded >> 32; + data[4] = padded >> 24; + data[5] = padded >> 16; + data[6] = padded >> 8; + data[7] = padded & 0xff; + + /* + * Encrypt the network order time with the key. + */ + fr_hmac_sha1(digest, data, 8, key, keylen); + + /* + * Take the least significant 4 bits. + */ + offset = digest[SHA1_DIGEST_LENGTH - 1] & 0x0f; + + /* + * Grab the 32bits at "offset", and drop the high bit. + */ + challenge = (digest[offset] & 0x7f) << 24; + challenge |= digest[offset + 1] << 16; + challenge |= digest[offset + 2] << 8; + challenge |= digest[offset + 3]; + + /* + * The token is the last 6 digits in the number (or 8 for testing).. + */ + snprintf(buffer, sizeof(buffer), ((OTP_LEN == 6) ? "%06u" : "%08u"), + challenge % ((OTP_LEN == 6) ? 1000000 : 100000000)); + + RDEBUG3("Now: %zu, Then: %zu", (size_t) now, (size_t) then); + RDEBUG3("Expected %s", buffer); + RDEBUG3("Received %s", totp); + + if (rad_digest_cmp((uint8_t const *) buffer, (uint8_t const *) totp, OTP_LEN) == 0) return 0; + +#ifndef TESTING + /* + * We've tested backwards, now do the equivalent time slot forwards + */ + if ((then < now) && (i <= inst->lookforward_steps)) { + forwards: + then = now + diff; + goto repeat; + } +#endif + } + return 1; +} + +#ifndef TESTING + +static inline CC_HINT(nonnull) totp_dedup_t *fr_dlist_head(fr_dlist_t const *head) +{ + if (head->prev == head) return NULL; + + return (totp_dedup_t *) (((uintptr_t) head->next) - offsetof(totp_dedup_t, dlist)); +} + + +static bool totp_reused(void *instance, time_t now, uint8_t const *key, size_t keylen, char const *passwd) +{ + rlm_totp_t *inst = instance; + totp_dedup_t *dedup, my_dedup; + + my_dedup.key = key; + my_dedup.keylen = keylen; + my_dedup.passwd = passwd; + + PTHREAD_MUTEX_LOCK(inst); + + /* + * Expire the oldest entries before searching for an entry in the tree. + */ + while (true) { + dedup = fr_dlist_head(&inst->dedup_list); + if (!dedup) break; + + if ((now - dedup->when) < (inst->lookback_steps * inst->lookback_interval)) break; + + dedup->unlisted = true; + fr_dlist_entry_unlink(&dedup->dlist); + (void) rbtree_deletebydata(inst->dedup_tree, dedup); + } + + /* + * Was this key and TOTP reused? + */ + dedup = rbtree_finddata(inst->dedup_tree, &my_dedup); + if (dedup) { + PTHREAD_MUTEX_UNLOCK(inst); + return true; + } + + dedup = calloc(sizeof(*dedup), 1); + if (!dedup) { + PTHREAD_MUTEX_UNLOCK(inst); + return false; + } + + dedup->key = key; + dedup->keylen = keylen; + dedup->passwd = passwd; + dedup->when = now; + dedup->instance = inst; + + fr_dlist_insert_tail(&inst->dedup_list, &dedup->dlist); + (void) rbtree_insert(inst->dedup_tree, dedup); + PTHREAD_MUTEX_UNLOCK(inst); + + return false; +} + +/* + * Do the authentication + */ +static rlm_rcode_t CC_HINT(nonnull) mod_authenticate(void *instance, REQUEST *request) +{ + VALUE_PAIR *vp, *password; + uint8_t const *key; + size_t keylen; + uint8_t buffer[80]; /* multiple of 5*8 characters */ + uint64_t now = time(NULL); + + password = fr_pair_find_by_num(request->packet->vps, PW_TOTP_PASSWORD, 0, TAG_ANY); + if (!password) return RLM_MODULE_NOOP; + + if ((password->vp_length != 6) && (password->vp_length != 8)) { + RDEBUG("TOTP-Password has incorrect length %d", (int) password->vp_length); + return RLM_MODULE_FAIL; + } + + /* + * Look for the raw key first. + */ + vp = fr_pair_find_by_num(request->config, PW_TOTP_KEY, 0, TAG_ANY); + if (vp) { + key = vp->vp_octets; + keylen = vp->vp_length; + + } else { + ssize_t len; + + vp = fr_pair_find_by_num(request->config, PW_TOTP_SECRET, 0, TAG_ANY); + if (!vp) { + RDEBUG("TOTP mod_authenticate() did not receive a TOTP-Secret"); + return RLM_MODULE_NOOP; + } + len = base32_decode(buffer, sizeof(buffer), vp->vp_strvalue); + if (len < 0) { + REDEBUG("TOTP-Secret cannot be decoded"); + return RLM_MODULE_FAIL; + } + + key = buffer; + keylen = len; + } + + vp = fr_pair_find_by_num(request->config, PW_TOTP_TIME_OFFSET, 0, TAG_ANY); + if (vp && (vp->vp_signed > -600) && (vp->vp_signed < 600)) { + RDEBUG("Using TOTP-Time-Offset = %d", vp->vp_signed); + now += vp->vp_signed; + } + + if (totp_cmp(request, now, key, keylen, password->vp_strvalue, instance) == 0) { + /* + * Forbid using a key more than once. + */ + if (totp_reused(instance, now, key, keylen, password->vp_strvalue)) return RLM_MODULE_REJECT; + + return RLM_MODULE_OK; + } + + /* + * Bad keys don't affect the cache. + */ + return RLM_MODULE_REJECT; +} + + +/* + * The module name should be the only globally exported symbol. + * That is, everything else should be 'static'. + * + * If the module needs to temporarily modify it's instantiation + * data, the type should be changed to RLM_TYPE_THREAD_UNSAFE. + * The server will then take care of ensuring that the module + * is single-threaded. + */ +extern module_t rlm_totp; +module_t rlm_totp = { + .magic = RLM_MODULE_INIT, + .name = "totp", + .type = RLM_TYPE_THREAD_SAFE, + .inst_size = sizeof(rlm_totp_t), + .config = module_config, + .bootstrap = mod_bootstrap, + .instantiate = mod_instantiate, +#ifdef HAVE_PTHREAD_H + .detach = mod_detach, +#endif + .methods = { + [MOD_AUTHENTICATE] = mod_authenticate, + }, +}; + +#else /* TESTING */ +/* + * ./totp decode KEY_BASE32 + * + * ./totp totp now KEY TOTP + */ +int main(int argc, char **argv) +{ + size_t len; + uint8_t *p; + uint8_t key[80]; + + if (argc < 2) return 0; + + if (strcmp(argv[1], "decode") == 0) { + if (argc < 3) return 0; + + len = base32_decode(key, sizeof(key), argv[2]); + printf("Decoded %ld %s\n", len, key); + + for (p = key; p < (key + len); p++) { + printf("%02x ", *p); + }; + printf("\n"); + + return 0; + } + + /* + * TOTP <time> <key> <expected-token> + */ + if (strcmp(argv[1], "totp") == 0) { + uint64_t now; + + if (argc < 5) return 0; + + if (strcmp(argv[2], "now") == 0) { + now = time(NULL); + } else { + (void) sscanf(argv[2], "%llu", &now); + } + + printf ("=== Time = %llu, TIME_STEP = %d, BACK_STEPS = %d, BACK_STEP_SECS = %d ===\n", + now, TIME_STEP, BACK_STEPS, BACK_STEP_SECS); + + if (totp_cmp(NULL, (time_t) now, (uint8_t const *) argv[3], + strlen(argv[3]), argv[4], NULL) == 0) { + return 0; + } + printf("Fail\n"); + return 1; + } + + fprintf(stderr, "Unknown command %s\n", argv[1]); + return 1; +} +#endif diff --git a/src/modules/rlm_totp/sha1.txt b/src/modules/rlm_totp/sha1.txt new file mode 100644 index 0000000..da85ab1 --- /dev/null +++ b/src/modules/rlm_totp/sha1.txt @@ -0,0 +1,6 @@ +59 12345678901234567890 94287082 +1111111109 12345678901234567890 07081804 +1111111111 12345678901234567890 14050471 +1234567890 12345678901234567890 89005924 +2000000000 12345678901234567890 69279037 +20000000000 12345678901234567890 65353130 |