added poly1305 random message tester
authorSadoon Albader <sadoon@soulserv.xyz>
Tue, 3 Oct 2023 16:33:17 +0000 (19:33 +0300)
committerLuke Kenneth Casson Leighton <lkcl@lkcl.net>
Fri, 22 Dec 2023 19:26:21 +0000 (19:26 +0000)
crypto/poly1305/.gitignore [new file with mode: 0644]
crypto/poly1305/Makefile [new file with mode: 0644]
crypto/poly1305/poly1305-donna-64.h [new file with mode: 0644]
crypto/poly1305/poly1305-donna.c [new file with mode: 0644]
crypto/poly1305/poly1305-donna.h [new file with mode: 0644]
crypto/poly1305/poly1305-rand-test.c [new file with mode: 0644]

diff --git a/crypto/poly1305/.gitignore b/crypto/poly1305/.gitignore
new file mode 100644 (file)
index 0000000..b34a840
--- /dev/null
@@ -0,0 +1,2 @@
+*.o
+poly1305-rand-test
diff --git a/crypto/poly1305/Makefile b/crypto/poly1305/Makefile
new file mode 100644 (file)
index 0000000..51f988f
--- /dev/null
@@ -0,0 +1,25 @@
+CC=gcc
+CFLAGS=-O2 -DPOLY1305_64BIT
+SRCS=poly1305-rand-test.c poly1305-donna.c
+OBJS=$(SRCS:.c=.o)
+
+BIN=poly1305-rand-test
+
+.PHONY: depend clean
+
+all: $(BIN)
+
+$(BIN): $(OBJS)
+       $(CC) $(CFLAGS) $(INCLUDES) -o $(BIN) $(OBJS) $(LFLAGS) $(LIBS)
+
+.c.o:
+       $(CC) $(CFLAGS) $(INCLUDES) -c $< -o $@
+
+
+clean:
+       rm -rf $(MAIN) $(OBJS) $(BIN)
+install:
+       cp ./$(BIN) /usr/bin/$(BIN)
+
+depend: $(SRCS)
+       makedepend $(INCLUDES) $^
diff --git a/crypto/poly1305/poly1305-donna-64.h b/crypto/poly1305/poly1305-donna-64.h
new file mode 100644 (file)
index 0000000..016f5b3
--- /dev/null
@@ -0,0 +1,224 @@
+/*
+       poly1305 implementation using 64 bit * 64 bit = 128 bit multiplication and 128 bit addition
+*/
+
+#if defined(_MSC_VER)
+       #include <intrin.h>
+
+       typedef struct uint128_t {
+               unsigned long long lo;
+               unsigned long long hi;
+       } uint128_t;
+
+       #define MUL(out, x, y) out.lo = _umul128((x), (y), &out.hi)
+       #define ADD(out, in) { unsigned long long t = out.lo; out.lo += in.lo; out.hi += (out.lo < t) + in.hi; }
+       #define ADDLO(out, in) { unsigned long long t = out.lo; out.lo += in; out.hi += (out.lo < t); }
+       #define SHR(in, shift) (__shiftright128(in.lo, in.hi, (shift)))
+       #define LO(in) (in.lo)
+
+       #define POLY1305_NOINLINE __declspec(noinline)
+#elif defined(__GNUC__)
+       #if defined(__SIZEOF_INT128__)
+               typedef unsigned __int128 uint128_t;
+       #else
+               typedef unsigned uint128_t __attribute__((mode(TI)));
+       #endif
+
+       #define MUL(out, x, y) out = ((uint128_t)x * y)
+       #define ADD(out, in) out += in
+       #define ADDLO(out, in) out += in
+       #define SHR(in, shift) (unsigned long long)(in >> (shift))
+       #define LO(in) (unsigned long long)(in)
+
+       #define POLY1305_NOINLINE __attribute__((noinline))
+#endif
+
+#define poly1305_block_size 16
+
+/* 17 + sizeof(size_t) + 8*sizeof(unsigned long long) */
+typedef struct poly1305_state_internal_t {
+       unsigned long long r[3];
+       unsigned long long h[3];
+       unsigned long long pad[2];
+       size_t leftover;
+       unsigned char buffer[poly1305_block_size];
+       unsigned char final;
+} poly1305_state_internal_t;
+
+/* interpret eight 8 bit unsigned integers as a 64 bit unsigned integer in little endian */
+static unsigned long long
+U8TO64(const unsigned char *p) {
+       return
+               (((unsigned long long)(p[0] & 0xff)      ) |
+                ((unsigned long long)(p[1] & 0xff) <<  8) |
+                ((unsigned long long)(p[2] & 0xff) << 16) |
+                ((unsigned long long)(p[3] & 0xff) << 24) |
+                ((unsigned long long)(p[4] & 0xff) << 32) |
+                ((unsigned long long)(p[5] & 0xff) << 40) |
+                ((unsigned long long)(p[6] & 0xff) << 48) |
+                ((unsigned long long)(p[7] & 0xff) << 56));
+}
+
+/* store a 64 bit unsigned integer as eight 8 bit unsigned integers in little endian */
+static void
+U64TO8(unsigned char *p, unsigned long long v) {
+       p[0] = (v      ) & 0xff;
+       p[1] = (v >>  8) & 0xff;
+       p[2] = (v >> 16) & 0xff;
+       p[3] = (v >> 24) & 0xff;
+       p[4] = (v >> 32) & 0xff;
+       p[5] = (v >> 40) & 0xff;
+       p[6] = (v >> 48) & 0xff;
+       p[7] = (v >> 56) & 0xff;
+}
+
+void
+poly1305_init(poly1305_context *ctx, const unsigned char key[32]) {
+       poly1305_state_internal_t *st = (poly1305_state_internal_t *)ctx;
+       unsigned long long t0,t1;
+
+       /* r &= 0xffffffc0ffffffc0ffffffc0fffffff */
+       t0 = U8TO64(&key[0]);
+       t1 = U8TO64(&key[8]);
+
+       st->r[0] = ( t0                    ) & 0xffc0fffffff;
+       st->r[1] = ((t0 >> 44) | (t1 << 20)) & 0xfffffc0ffff;
+       st->r[2] = ((t1 >> 24)             ) & 0x00ffffffc0f;
+
+       /* h = 0 */
+       st->h[0] = 0;
+       st->h[1] = 0;
+       st->h[2] = 0;
+
+       /* save pad for later */
+       st->pad[0] = U8TO64(&key[16]);
+       st->pad[1] = U8TO64(&key[24]);
+
+       st->leftover = 0;
+       st->final = 0;
+}
+
+static void
+poly1305_blocks(poly1305_state_internal_t *st, const unsigned char *m, size_t bytes) {
+       const unsigned long long hibit = (st->final) ? 0 : ((unsigned long long)1 << 40); /* 1 << 128 */
+       unsigned long long r0,r1,r2;
+       unsigned long long s1,s2;
+       unsigned long long h0,h1,h2;
+       unsigned long long c;
+       uint128_t d0,d1,d2,d;
+
+       r0 = st->r[0];
+       r1 = st->r[1];
+       r2 = st->r[2];
+
+       h0 = st->h[0];
+       h1 = st->h[1];
+       h2 = st->h[2];
+
+       s1 = r1 * (5 << 2);
+       s2 = r2 * (5 << 2);
+
+       while (bytes >= poly1305_block_size) {
+               unsigned long long t0,t1;
+
+               /* h += m[i] */
+               t0 = U8TO64(&m[0]);
+               t1 = U8TO64(&m[8]);
+
+               h0 += (( t0                    ) & 0xfffffffffff);
+               h1 += (((t0 >> 44) | (t1 << 20)) & 0xfffffffffff);
+               h2 += (((t1 >> 24)             ) & 0x3ffffffffff) | hibit;
+
+               /* h *= r */
+               MUL(d0, h0, r0); MUL(d, h1, s2); ADD(d0, d); MUL(d, h2, s1); ADD(d0, d);
+               MUL(d1, h0, r1); MUL(d, h1, r0); ADD(d1, d); MUL(d, h2, s2); ADD(d1, d);
+               MUL(d2, h0, r2); MUL(d, h1, r1); ADD(d2, d); MUL(d, h2, r0); ADD(d2, d);
+
+               /* (partial) h %= p */
+                             c = SHR(d0, 44); h0 = LO(d0) & 0xfffffffffff;
+               ADDLO(d1, c); c = SHR(d1, 44); h1 = LO(d1) & 0xfffffffffff;
+               ADDLO(d2, c); c = SHR(d2, 42); h2 = LO(d2) & 0x3ffffffffff;
+               h0  += c * 5; c = (h0 >> 44);  h0 =    h0  & 0xfffffffffff;
+               h1  += c;
+
+               m += poly1305_block_size;
+               bytes -= poly1305_block_size;
+       }
+
+       st->h[0] = h0;
+       st->h[1] = h1;
+       st->h[2] = h2;
+}
+
+
+POLY1305_NOINLINE void
+poly1305_finish(poly1305_context *ctx, unsigned char mac[16]) {
+       poly1305_state_internal_t *st = (poly1305_state_internal_t *)ctx;
+       unsigned long long h0,h1,h2,c;
+       unsigned long long g0,g1,g2;
+       unsigned long long t0,t1;
+
+       /* process the remaining block */
+       if (st->leftover) {
+               size_t i = st->leftover;
+               st->buffer[i] = 1;
+               for (i = i + 1; i < poly1305_block_size; i++)
+                       st->buffer[i] = 0;
+               st->final = 1;
+               poly1305_blocks(st, st->buffer, poly1305_block_size);
+       }
+
+       /* fully carry h */
+       h0 = st->h[0];
+       h1 = st->h[1];
+       h2 = st->h[2];
+
+                    c = (h1 >> 44); h1 &= 0xfffffffffff;
+       h2 += c;     c = (h2 >> 42); h2 &= 0x3ffffffffff;
+       h0 += c * 5; c = (h0 >> 44); h0 &= 0xfffffffffff;
+       h1 += c;     c = (h1 >> 44); h1 &= 0xfffffffffff;
+       h2 += c;     c = (h2 >> 42); h2 &= 0x3ffffffffff;
+       h0 += c * 5; c = (h0 >> 44); h0 &= 0xfffffffffff;
+       h1 += c;
+
+       /* compute h + -p */
+       g0 = h0 + 5; c = (g0 >> 44); g0 &= 0xfffffffffff;
+       g1 = h1 + c; c = (g1 >> 44); g1 &= 0xfffffffffff;
+       g2 = h2 + c - ((unsigned long long)1 << 42);
+
+       /* select h if h < p, or h + -p if h >= p */
+       c = (g2 >> ((sizeof(unsigned long long) * 8) - 1)) - 1;
+       g0 &= c;
+       g1 &= c;
+       g2 &= c;
+       c = ~c;
+       h0 = (h0 & c) | g0;
+       h1 = (h1 & c) | g1;
+       h2 = (h2 & c) | g2;
+
+       /* h = (h + pad) */
+       t0 = st->pad[0];
+       t1 = st->pad[1];
+
+       h0 += (( t0                    ) & 0xfffffffffff)    ; c = (h0 >> 44); h0 &= 0xfffffffffff;
+       h1 += (((t0 >> 44) | (t1 << 20)) & 0xfffffffffff) + c; c = (h1 >> 44); h1 &= 0xfffffffffff;
+       h2 += (((t1 >> 24)             ) & 0x3ffffffffff) + c;                 h2 &= 0x3ffffffffff;
+
+       /* mac = h % (2^128) */
+       h0 = ((h0      ) | (h1 << 44));
+       h1 = ((h1 >> 20) | (h2 << 24));
+
+       U64TO8(&mac[0], h0);
+       U64TO8(&mac[8], h1);
+
+       /* zero out the state */
+       st->h[0] = 0;
+       st->h[1] = 0;
+       st->h[2] = 0;
+       st->r[0] = 0;
+       st->r[1] = 0;
+       st->r[2] = 0;
+       st->pad[0] = 0;
+       st->pad[1] = 0;
+}
+
diff --git a/crypto/poly1305/poly1305-donna.c b/crypto/poly1305/poly1305-donna.c
new file mode 100644 (file)
index 0000000..c1e3c74
--- /dev/null
@@ -0,0 +1,201 @@
+#include "poly1305-donna.h"
+
+#if defined(POLY1305_8BIT)
+#include "poly1305-donna-8.h"
+#elif defined(POLY1305_16BIT)
+#include "poly1305-donna-16.h"
+#elif defined(POLY1305_32BIT)
+#include "poly1305-donna-32.h"
+#elif defined(POLY1305_64BIT)
+#include "poly1305-donna-64.h"
+#else
+
+/* auto detect between 32bit / 64bit */
+#define HAS_SIZEOF_INT128_64BIT (defined(__SIZEOF_INT128__) && defined(__LP64__))
+#define HAS_MSVC_64BIT (defined(_MSC_VER) && defined(_M_X64))
+#define HAS_GCC_4_4_64BIT (defined(__GNUC__) && defined(__LP64__) && ((__GNUC__ > 4) || ((__GNUC__ == 4) && (__GNUC_MINOR__ >= 4))))
+
+#if (HAS_SIZEOF_INT128_64BIT || HAS_MSVC_64BIT || HAS_GCC_4_4_64BIT)
+#include "poly1305-donna-64.h"
+#else
+#include "poly1305-donna-32.h"
+#endif
+
+#endif
+
+void
+poly1305_update(poly1305_context *ctx, const unsigned char *m, size_t bytes) {
+       poly1305_state_internal_t *st = (poly1305_state_internal_t *)ctx;
+       size_t i;
+
+       /* handle leftover */
+       if (st->leftover) {
+               size_t want = (poly1305_block_size - st->leftover);
+               if (want > bytes)
+                       want = bytes;
+               for (i = 0; i < want; i++)
+                       st->buffer[st->leftover + i] = m[i];
+               bytes -= want;
+               m += want;
+               st->leftover += want;
+               if (st->leftover < poly1305_block_size)
+                       return;
+               poly1305_blocks(st, st->buffer, poly1305_block_size);
+               st->leftover = 0;
+       }
+
+       /* process full blocks */
+       if (bytes >= poly1305_block_size) {
+               size_t want = (bytes & ~(poly1305_block_size - 1));
+               poly1305_blocks(st, m, want);
+               m += want;
+               bytes -= want;
+       }
+
+       /* store leftover */
+       if (bytes) {
+               for (i = 0; i < bytes; i++)
+                       st->buffer[st->leftover + i] = m[i];
+               st->leftover += bytes;
+       }
+}
+
+void
+poly1305_auth(unsigned char mac[16], const unsigned char *m, size_t bytes, const unsigned char key[32]) {
+       poly1305_context ctx;
+       poly1305_init(&ctx, key);
+       poly1305_update(&ctx, m, bytes);
+       poly1305_finish(&ctx, mac);
+}
+
+int
+poly1305_verify(const unsigned char mac1[16], const unsigned char mac2[16]) {
+       size_t i;
+       unsigned int dif = 0;
+       for (i = 0; i < 16; i++)
+               dif |= (mac1[i] ^ mac2[i]);
+       dif = (dif - 1) >> ((sizeof(unsigned int) * 8) - 1);
+       return (dif & 1);
+}
+
+
+/* test a few basic operations */
+int
+poly1305_power_on_self_test(void) {
+       /* example from nacl */
+       static const unsigned char nacl_key[32] = {
+               0xee,0xa6,0xa7,0x25,0x1c,0x1e,0x72,0x91,
+               0x6d,0x11,0xc2,0xcb,0x21,0x4d,0x3c,0x25,
+               0x25,0x39,0x12,0x1d,0x8e,0x23,0x4e,0x65,
+               0x2d,0x65,0x1f,0xa4,0xc8,0xcf,0xf8,0x80,
+       };
+
+       static const unsigned char nacl_msg[131] = {
+               0x8e,0x99,0x3b,0x9f,0x48,0x68,0x12,0x73,
+               0xc2,0x96,0x50,0xba,0x32,0xfc,0x76,0xce,
+               0x48,0x33,0x2e,0xa7,0x16,0x4d,0x96,0xa4,
+               0x47,0x6f,0xb8,0xc5,0x31,0xa1,0x18,0x6a,
+               0xc0,0xdf,0xc1,0x7c,0x98,0xdc,0xe8,0x7b,
+               0x4d,0xa7,0xf0,0x11,0xec,0x48,0xc9,0x72,
+               0x71,0xd2,0xc2,0x0f,0x9b,0x92,0x8f,0xe2,
+               0x27,0x0d,0x6f,0xb8,0x63,0xd5,0x17,0x38,
+               0xb4,0x8e,0xee,0xe3,0x14,0xa7,0xcc,0x8a,
+               0xb9,0x32,0x16,0x45,0x48,0xe5,0x26,0xae,
+               0x90,0x22,0x43,0x68,0x51,0x7a,0xcf,0xea,
+               0xbd,0x6b,0xb3,0x73,0x2b,0xc0,0xe9,0xda,
+               0x99,0x83,0x2b,0x61,0xca,0x01,0xb6,0xde,
+               0x56,0x24,0x4a,0x9e,0x88,0xd5,0xf9,0xb3,
+               0x79,0x73,0xf6,0x22,0xa4,0x3d,0x14,0xa6,
+               0x59,0x9b,0x1f,0x65,0x4c,0xb4,0x5a,0x74,
+               0xe3,0x55,0xa5
+       };
+
+       static const unsigned char nacl_mac[16] = {
+               0xf3,0xff,0xc7,0x70,0x3f,0x94,0x00,0xe5,
+               0x2a,0x7d,0xfb,0x4b,0x3d,0x33,0x05,0xd9
+       };
+
+       /* generates a final value of (2^130 - 2) == 3 */
+       static const unsigned char wrap_key[32] = {
+               0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+               0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+               0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+               0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+       };
+
+       static const unsigned char wrap_msg[16] = {
+               0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,
+               0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff
+       };
+
+       static const unsigned char wrap_mac[16] = {
+               0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+               0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+       };
+
+       /*
+               mac of the macs of messages of length 0 to 256, where the key and messages
+               have all their values set to the length
+       */
+       static const unsigned char total_key[32] = {
+               0x01,0x02,0x03,0x04,0x05,0x06,0x07,
+               0xff,0xfe,0xfd,0xfc,0xfb,0xfa,0xf9,
+               0xff,0xff,0xff,0xff,0xff,0xff,0xff,
+               0xff,0xff,0xff,0xff,0xff,0xff,0xff
+       };
+
+       static const unsigned char total_mac[16] = {
+               0x64,0xaf,0xe2,0xe8,0xd6,0xad,0x7b,0xbd,
+               0xd2,0x87,0xf9,0x7c,0x44,0x62,0x3d,0x39
+       };
+
+       poly1305_context ctx;
+       poly1305_context total_ctx;
+       unsigned char all_key[32];
+       unsigned char all_msg[256];
+       unsigned char mac[16];
+       size_t i, j;
+       int result = 1;
+
+       for (i = 0; i < sizeof(mac); i++)
+               mac[i] = 0;
+       poly1305_auth(mac, nacl_msg, sizeof(nacl_msg), nacl_key);
+       result &= poly1305_verify(nacl_mac, mac);
+
+       for (i = 0; i < sizeof(mac); i++)
+               mac[i] = 0;
+       poly1305_init(&ctx, nacl_key);
+       poly1305_update(&ctx, nacl_msg +   0, 32);
+       poly1305_update(&ctx, nacl_msg +  32, 64);
+       poly1305_update(&ctx, nacl_msg +  96, 16);
+       poly1305_update(&ctx, nacl_msg + 112,  8);
+       poly1305_update(&ctx, nacl_msg + 120,  4);
+       poly1305_update(&ctx, nacl_msg + 124,  2);
+       poly1305_update(&ctx, nacl_msg + 126,  1);
+       poly1305_update(&ctx, nacl_msg + 127,  1);
+       poly1305_update(&ctx, nacl_msg + 128,  1);
+       poly1305_update(&ctx, nacl_msg + 129,  1);
+       poly1305_update(&ctx, nacl_msg + 130,  1);
+       poly1305_finish(&ctx, mac);
+       result &= poly1305_verify(nacl_mac, mac);
+
+       for (i = 0; i < sizeof(mac); i++)
+               mac[i] = 0;
+       poly1305_auth(mac, wrap_msg, sizeof(wrap_msg), wrap_key);
+       result &= poly1305_verify(wrap_mac, mac);
+
+       poly1305_init(&total_ctx, total_key);
+       for (i = 0; i < 256; i++) {
+               /* set key and message to 'i,i,i..' */
+               for (j = 0; j < sizeof(all_key); j++)
+                       all_key[j] = i;
+               for (j = 0; j < i; j++)
+                       all_msg[j] = i;
+               poly1305_auth(mac, all_msg, i, all_key);
+               poly1305_update(&total_ctx, mac, 16);
+       }
+       poly1305_finish(&total_ctx, mac);
+       result &= poly1305_verify(total_mac, mac);
+
+       return result;
+}
diff --git a/crypto/poly1305/poly1305-donna.h b/crypto/poly1305/poly1305-donna.h
new file mode 100644 (file)
index 0000000..94e2353
--- /dev/null
@@ -0,0 +1,20 @@
+#ifndef POLY1305_DONNA_H
+#define POLY1305_DONNA_H
+
+#include <stddef.h>
+
+typedef struct poly1305_context {
+       size_t aligner;
+       unsigned char opaque[136];
+} poly1305_context;
+
+void poly1305_init(poly1305_context *ctx, const unsigned char key[32]);
+void poly1305_update(poly1305_context *ctx, const unsigned char *m, size_t bytes);
+void poly1305_finish(poly1305_context *ctx, unsigned char mac[16]);
+void poly1305_auth(unsigned char mac[16], const unsigned char *m, size_t bytes, const unsigned char key[32]);
+
+int poly1305_verify(const unsigned char mac1[16], const unsigned char mac2[16]);
+int poly1305_power_on_self_test(void);
+
+#endif /* POLY1305_DONNA_H */
+
diff --git a/crypto/poly1305/poly1305-rand-test.c b/crypto/poly1305/poly1305-rand-test.c
new file mode 100644 (file)
index 0000000..ad52706
--- /dev/null
@@ -0,0 +1,39 @@
+#include "poly1305-donna.h"
+#include <stdio.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include <time.h>
+
+uint8_t *msg, *key, *mac;
+size_t msg_size;
+
+#define KEY_SIZE 32
+#define MAC_SIZE 16
+
+int main(int argc, char **argv){
+
+    srandom(clock());
+
+    key = (uint8_t*)malloc(KEY_SIZE);
+    mac = (uint8_t*)malloc(MAC_SIZE);
+
+    for (int i = 0; i < KEY_SIZE; i++) key[i] = random() & 0x000000FF;
+
+    msg_size = random() & 0x00000FFF;
+    msg = (uint8_t*)malloc(msg_size);
+
+    for (int i = 0; i < msg_size; i++) msg[i] = random() & 0x000000FF;
+
+    poly1305_auth(mac, msg, sizeof(msg), key);
+
+    for (int i = 0; i < msg_size; i++)
+        printf("%u%s", msg[i], (i == msg_size-1) ? "\n" : "," );
+    for (int i = 0; i < KEY_SIZE; i++)
+        printf("%u%s", key[i], (i == KEY_SIZE-1) ? "\n" : "," );
+    for (int i = 0; i < MAC_SIZE; i++)
+        printf("%u%s", mac[i], (i == MAC_SIZE-1) ? "\n" : "," );
+
+    return 0;
+}
+
+