[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]

[PATCH 3/4] chachapoly: add external files


From 3ae2f64ce8ff011f26b965322541c60e7d108a0e Mon Sep 17 00:00:00 2001
From: Meng Tan <mtan@xxxxxxxxxx>
Date: Mon, 18 Sep 2017 11:34:50 +0200
Subject: [PATCH 3/4] chachapoly: add external files

Signed-off-by: Meng Tan <mtan@xxxxxxxxxx>
---
 CMakeLists.txt              |   2 +-
 DefineOptions.cmake         |   1 +
 config.h.cmake              |   3 +
 include/libssh/buffer.h     |  47 ++++++++++
 include/libssh/chacha.h     |  35 +++++++
 include/libssh/chachapoly.h |  41 +++++++++
 include/libssh/poly1305.h   |  22 +++++
 src/CMakeLists.txt          |   9 ++
 src/external/chacha.c       | 219 ++++++++++++++++++++++++++++++++++++++++++++
 src/external/chachapoly.c   | 134 +++++++++++++++++++++++++++
 src/external/poly1305.c     | 158 ++++++++++++++++++++++++++++++++
 11 files changed, 670 insertions(+), 1 deletion(-)
 create mode 100644 include/libssh/chacha.h
 create mode 100644 include/libssh/chachapoly.h
 create mode 100644 include/libssh/poly1305.h
 create mode 100644 src/external/chacha.c
 create mode 100644 src/external/chachapoly.c
 create mode 100644 src/external/poly1305.c

diff --git a/CMakeLists.txt b/CMakeLists.txt
index b561947..61bd120 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -151,6 +151,7 @@ message(STATUS "********** ${PROJECT_NAME} build options : **********")
 message(STATUS "zlib support: ${WITH_ZLIB}")
 message(STATUS "libgcrypt support: ${WITH_GCRYPT}")
 message(STATUS "libnacl support: ${WITH_NACL}")
+message(STATUS "chachapoly support: ${WITH_CHACHAPOLY}")
 message(STATUS "SSH-1 support: ${WITH_SSH1}")
 message(STATUS "SFTP support: ${WITH_SFTP}")
 message(STATUS "Server support : ${WITH_SERVER}")
@@ -166,4 +167,3 @@ else (WITH_INTERNAL_DOC)
 endif (WITH_INTERNAL_DOC)
 message(STATUS "Benchmarks: ${WITH_BENCHMARKS}")
 message(STATUS "********************************************")
-
diff --git a/DefineOptions.cmake b/DefineOptions.cmake
index ab7819a..87e537d 100644
--- a/DefineOptions.cmake
+++ b/DefineOptions.cmake
@@ -14,6 +14,7 @@ option(WITH_CLIENT_TESTING "Build with client tests; requires a running sshd" OF
 option(WITH_BENCHMARKS "Build benchmarks tools" OFF)
 option(WITH_EXAMPLES "Build examples" ON)
 option(WITH_NACL "Build with libnacl (curve25519" ON)
+option(WITH_CHACHAPOLY "Build with chachapoly support" ON)
 if (WITH_ZLIB)
     set(WITH_LIBZ ON)
 else (WITH_ZLIB)
diff --git a/config.h.cmake b/config.h.cmake
index f8869df..cbe0bca 100644
--- a/config.h.cmake
+++ b/config.h.cmake
@@ -202,6 +202,9 @@
 /* Define to 1 if you want to enable NaCl support */
 #cmakedefine WITH_NACL 1
 
+/* Define to 1 if you want to enable chachapoly support */
+#cmakedefine WITH_CHACHAPOLY 1
+
 /*************************** ENDIAN *****************************/
 
 /* Define WORDS_BIGENDIAN to 1 if your processor stores words with the most
diff --git a/include/libssh/buffer.h b/include/libssh/buffer.h
index 81d9452..dc697e9 100644
--- a/include/libssh/buffer.h
+++ b/include/libssh/buffer.h
@@ -87,4 +87,51 @@ ssh_string ssh_buffer_get_mpint(ssh_buffer buffer);
 uint32_t ssh_buffer_pass_bytes_end(ssh_buffer buffer, uint32_t len);
 uint32_t ssh_buffer_pass_bytes(ssh_buffer buffer, uint32_t len);
 
+
+/* Macros for decoding/encoding integers */
+#define PEEK_U64(p)                                     \
+    (((u_int64_t)(((const u_char *)(p))[0]) << 56) |    \
+     ((u_int64_t)(((const u_char *)(p))[1]) << 48) |    \
+     ((u_int64_t)(((const u_char *)(p))[2]) << 40) |    \
+     ((u_int64_t)(((const u_char *)(p))[3]) << 32) |    \
+     ((u_int64_t)(((const u_char *)(p))[4]) << 24) |    \
+     ((u_int64_t)(((const u_char *)(p))[5]) << 16) |    \
+     ((u_int64_t)(((const u_char *)(p))[6]) << 8) |     \
+     (u_int64_t)(((const u_char *)(p))[7]))
+#define PEEK_U32(p)                                     \
+    (((u_int32_t)(((const u_char *)(p))[0]) << 24) |    \
+     ((u_int32_t)(((const u_char *)(p))[1]) << 16) |    \
+     ((u_int32_t)(((const u_char *)(p))[2]) << 8) |     \
+     (u_int32_t)(((const u_char *)(p))[3]))
+#define PEEK_U16(p)                                     \
+    (((u_int16_t)(((const u_char *)(p))[0]) << 8) |     \
+     (u_int16_t)(((const u_char *)(p))[1]))
+
+#define POKE_U64(p, v)                                  \
+    do {                                                \
+        const u_int64_t __v = (v);                      \
+        ((u_char *)(p))[0] = (__v >> 56) & 0xff;        \
+        ((u_char *)(p))[1] = (__v >> 48) & 0xff;        \
+        ((u_char *)(p))[2] = (__v >> 40) & 0xff;        \
+        ((u_char *)(p))[3] = (__v >> 32) & 0xff;        \
+        ((u_char *)(p))[4] = (__v >> 24) & 0xff;        \
+        ((u_char *)(p))[5] = (__v >> 16) & 0xff;        \
+        ((u_char *)(p))[6] = (__v >> 8) & 0xff;         \
+        ((u_char *)(p))[7] = __v & 0xff;                \
+    } while (0)
+#define POKE_U32(p, v)                                  \
+    do {                                                \
+        const u_int32_t __v = (v);                      \
+        ((u_char *)(p))[0] = (__v >> 24) & 0xff;        \
+        ((u_char *)(p))[1] = (__v >> 16) & 0xff;        \
+        ((u_char *)(p))[2] = (__v >> 8) & 0xff;         \
+        ((u_char *)(p))[3] = __v & 0xff;                \
+	} while (0)
+#define POKE_U16(p, v)                          \
+    do {                                        \
+        const u_int16_t __v = (v);              \
+        ((u_char *)(p))[0] = (__v >> 8) & 0xff; \
+        ((u_char *)(p))[1] = __v & 0xff;        \
+	} while (0)
+
 #endif /* BUFFER_H_ */
diff --git a/include/libssh/chacha.h b/include/libssh/chacha.h
new file mode 100644
index 0000000..5a04ae4
--- /dev/null
+++ b/include/libssh/chacha.h
@@ -0,0 +1,35 @@
+/* $OpenBSD: chacha.h,v 1.4 2016/08/27 04:04:56 guenther Exp $ */
+
+/*
+  chacha-merged.c version 20080118
+  D. J. Bernstein
+  Public domain.
+*/
+
+#ifndef CHACHA_H
+#define CHACHA_H
+
+#include <sys/types.h>
+#include <stdlib.h>
+
+struct chacha_ctx {
+    u_int input[16];
+};
+
+#define CHACHA_MINKEYLEN 	16
+#define CHACHA_NONCELEN		8
+#define CHACHA_CTRLEN		8
+#define CHACHA_STATELEN		(CHACHA_NONCELEN+CHACHA_CTRLEN)
+#define CHACHA_BLOCKLEN		64
+
+void chacha_keysetup(struct chacha_ctx *x, const u_char *k, u_int kbits);
+/* __attribute__((__bounded__(__minbytes__, 2, CHACHA_MINKEYLEN))); */
+void chacha_ivsetup(struct chacha_ctx *x, const u_char *iv, const u_char *ctr);
+/* __attribute__((__bounded__(__minbytes__, 2, CHACHA_NONCELEN))) */
+/* __attribute__((__bounded__(__minbytes__, 3, CHACHA_CTRLEN))); */
+void chacha_encrypt_bytes(struct chacha_ctx *x, const u_char *m,
+                          u_char *c, u_int bytes);
+/* __attribute__((__bounded__(__buffer__, 2, 4))) */
+/* __attribute__((__bounded__(__buffer__, 3, 4))); */
+
+#endif	/* CHACHA_H */
diff --git a/include/libssh/chachapoly.h b/include/libssh/chachapoly.h
new file mode 100644
index 0000000..df54b60
--- /dev/null
+++ b/include/libssh/chachapoly.h
@@ -0,0 +1,41 @@
+/* $OpenBSD: cipher-chachapoly.h,v 1.4 2014/06/24 01:13:21 djm Exp $ */
+
+/*
+ * Copyright (c) Damien Miller 2013 <djm@xxxxxxxxxxx>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+#ifndef CHACHA_POLY_AEAD_H
+#define CHACHA_POLY_AEAD_H
+
+#include <sys/types.h>
+#include "chacha.h"
+#include "poly1305.h"
+
+#define CHACHA_KEYLEN	32 /* Only 256 bit keys used here */
+
+struct chachapoly_ctx {
+    struct chacha_ctx main_ctx, header_ctx;
+};
+
+int	chachapoly_init(struct chachapoly_ctx *cpctx,
+                        const u_char *key, u_int keylen);
+/* __attribute__((__bounded__(__buffer__, 2, 3))); */
+int	chachapoly_crypt(struct chachapoly_ctx *cpctx, u_int seqnr,
+                         u_char *dest, const u_char *src, u_int len, u_int aadlen, u_int authlen,
+                         int do_encrypt);
+int	chachapoly_get_length(struct chachapoly_ctx *cpctx,
+                              u_int *plenp, u_int seqnr, const u_char *cp, u_int len);
+/* __attribute__((__bounded__(__buffer__, 4, 5))); */
+
+#endif /* CHACHA_POLY_AEAD_H */
diff --git a/include/libssh/poly1305.h b/include/libssh/poly1305.h
new file mode 100644
index 0000000..757b7ff
--- /dev/null
+++ b/include/libssh/poly1305.h
@@ -0,0 +1,22 @@
+/* $OpenBSD: poly1305.h,v 1.4 2014/05/02 03:27:54 djm Exp $ */
+
+/*
+ * Public Domain poly1305 from Andrew Moon
+ * poly1305-donna-unrolled.c from https://github.com/floodyberry/poly1305-donna
+ */
+
+#ifndef POLY1305_H
+#define POLY1305_H
+
+#include <sys/types.h>
+
+#define POLY1305_KEYLEN		32
+#define POLY1305_TAGLEN		16
+
+void poly1305_auth(u_char out[POLY1305_TAGLEN], const u_char *m, size_t inlen,
+                   const u_char key[POLY1305_KEYLEN]);
+/* __attribute__((__bounded__(__minbytes__, 1, POLY1305_TAGLEN))) */
+/* __attribute__((__bounded__(__buffer__, 2, 3))) */
+/* __attribute__((__bounded__(__minbytes__, 4, POLY1305_KEYLEN))); */
+
+#endif	/* POLY1305_H */
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index e5a9353..73ff36c 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -152,6 +152,15 @@ set(libssh_SRCS
   external/sc25519.c
 )
 
+if (WITH_CHACHAPOLY)
+  set(libssh_SRCS
+    ${libssh_SRCS}
+    external/chacha.c
+    external/poly1305.c
+    external/chachapoly.c
+    )
+endif (WITH_CHACHAPOLY)
+
 if (WITH_GCRYPT)
     set(libssh_SRCS
         ${libssh_SRCS}
diff --git a/src/external/chacha.c b/src/external/chacha.c
new file mode 100644
index 0000000..1bb3b34
--- /dev/null
+++ b/src/external/chacha.c
@@ -0,0 +1,219 @@
+/*
+  chacha-merged.c version 20080118
+  D. J. Bernstein
+  Public domain.
+*/
+
+/* #include "includes.h" */
+
+#include "libssh/chacha.h"
+
+/* $OpenBSD: chacha.c,v 1.1 2013/11/21 00:45:44 djm Exp $ */
+
+typedef unsigned char u8;
+typedef unsigned int u32;
+
+typedef struct chacha_ctx chacha_ctx;
+
+#define U8C(v) (v##U)
+#define U32C(v) (v##U)
+
+#define U8V(v) ((u8)(v) & U8C(0xFF))
+#define U32V(v) ((u32)(v) & U32C(0xFFFFFFFF))
+
+#define ROTL32(v, n)                            \
+    (U32V((v) << (n)) | ((v) >> (32 - (n))))
+
+#define U8TO32_LITTLE(p)                        \
+    (((u32)((p)[0])      ) |                    \
+     ((u32)((p)[1]) <<  8) |                    \
+     ((u32)((p)[2]) << 16) |                    \
+     ((u32)((p)[3]) << 24))
+
+#define U32TO8_LITTLE(p, v)                     \
+    do {                                        \
+        (p)[0] = U8V((v)      );                \
+        (p)[1] = U8V((v) >>  8);                \
+        (p)[2] = U8V((v) >> 16);                \
+        (p)[3] = U8V((v) >> 24);                \
+    } while (0)
+
+#define ROTATE(v,c) (ROTL32(v,c))
+#define XOR(v,w) ((v) ^ (w))
+#define PLUS(v,w) (U32V((v) + (w)))
+#define PLUSONE(v) (PLUS((v),1))
+
+#define QUARTERROUND(a,b,c,d)                   \
+    a = PLUS(a,b); d = ROTATE(XOR(d,a),16);     \
+    c = PLUS(c,d); b = ROTATE(XOR(b,c),12);     \
+    a = PLUS(a,b); d = ROTATE(XOR(d,a), 8);     \
+    c = PLUS(c,d); b = ROTATE(XOR(b,c), 7);
+
+static const char sigma[16] = "expand 32-byte k";
+static const char tau[16] = "expand 16-byte k";
+
+void
+chacha_keysetup(chacha_ctx *x,const u8 *k,u32 kbits)
+{
+    const char *constants;
+
+    x->input[4] = U8TO32_LITTLE(k + 0);
+    x->input[5] = U8TO32_LITTLE(k + 4);
+    x->input[6] = U8TO32_LITTLE(k + 8);
+    x->input[7] = U8TO32_LITTLE(k + 12);
+    if (kbits == 256) { /* recommended */
+        k += 16;
+        constants = sigma;
+    } else { /* kbits == 128 */
+        constants = tau;
+    }
+    x->input[8] = U8TO32_LITTLE(k + 0);
+    x->input[9] = U8TO32_LITTLE(k + 4);
+    x->input[10] = U8TO32_LITTLE(k + 8);
+    x->input[11] = U8TO32_LITTLE(k + 12);
+    x->input[0] = U8TO32_LITTLE(constants + 0);
+    x->input[1] = U8TO32_LITTLE(constants + 4);
+    x->input[2] = U8TO32_LITTLE(constants + 8);
+    x->input[3] = U8TO32_LITTLE(constants + 12);
+}
+
+void
+chacha_ivsetup(chacha_ctx *x, const u8 *iv, const u8 *counter)
+{
+    x->input[12] = counter == NULL ? 0 : U8TO32_LITTLE(counter + 0);
+    x->input[13] = counter == NULL ? 0 : U8TO32_LITTLE(counter + 4);
+    x->input[14] = U8TO32_LITTLE(iv + 0);
+    x->input[15] = U8TO32_LITTLE(iv + 4);
+}
+
+void
+chacha_encrypt_bytes(chacha_ctx *x,const u8 *m,u8 *c,u32 bytes)
+{
+    u32 x0, x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15;
+    u32 j0, j1, j2, j3, j4, j5, j6, j7, j8, j9, j10, j11, j12, j13, j14, j15;
+    u8 *ctarget = NULL;
+    u8 tmp[64];
+    u_int i;
+
+    if (!bytes) return;
+
+    j0 = x->input[0];
+    j1 = x->input[1];
+    j2 = x->input[2];
+    j3 = x->input[3];
+    j4 = x->input[4];
+    j5 = x->input[5];
+    j6 = x->input[6];
+    j7 = x->input[7];
+    j8 = x->input[8];
+    j9 = x->input[9];
+    j10 = x->input[10];
+    j11 = x->input[11];
+    j12 = x->input[12];
+    j13 = x->input[13];
+    j14 = x->input[14];
+    j15 = x->input[15];
+
+    for (;;) {
+        if (bytes < 64) {
+            for (i = 0;i < bytes;++i) tmp[i] = m[i];
+            m = tmp;
+            ctarget = c;
+            c = tmp;
+        }
+        x0 = j0;
+        x1 = j1;
+        x2 = j2;
+        x3 = j3;
+        x4 = j4;
+        x5 = j5;
+        x6 = j6;
+        x7 = j7;
+        x8 = j8;
+        x9 = j9;
+        x10 = j10;
+        x11 = j11;
+        x12 = j12;
+        x13 = j13;
+        x14 = j14;
+        x15 = j15;
+        for (i = 20;i > 0;i -= 2) {
+            QUARTERROUND( x0, x4, x8,x12)
+                QUARTERROUND( x1, x5, x9,x13)
+                QUARTERROUND( x2, x6,x10,x14)
+                QUARTERROUND( x3, x7,x11,x15)
+                QUARTERROUND( x0, x5,x10,x15)
+                QUARTERROUND( x1, x6,x11,x12)
+                QUARTERROUND( x2, x7, x8,x13)
+                QUARTERROUND( x3, x4, x9,x14)
+                }
+        x0 = PLUS(x0,j0);
+        x1 = PLUS(x1,j1);
+        x2 = PLUS(x2,j2);
+        x3 = PLUS(x3,j3);
+        x4 = PLUS(x4,j4);
+        x5 = PLUS(x5,j5);
+        x6 = PLUS(x6,j6);
+        x7 = PLUS(x7,j7);
+        x8 = PLUS(x8,j8);
+        x9 = PLUS(x9,j9);
+        x10 = PLUS(x10,j10);
+        x11 = PLUS(x11,j11);
+        x12 = PLUS(x12,j12);
+        x13 = PLUS(x13,j13);
+        x14 = PLUS(x14,j14);
+        x15 = PLUS(x15,j15);
+
+        x0 = XOR(x0,U8TO32_LITTLE(m + 0));
+        x1 = XOR(x1,U8TO32_LITTLE(m + 4));
+        x2 = XOR(x2,U8TO32_LITTLE(m + 8));
+        x3 = XOR(x3,U8TO32_LITTLE(m + 12));
+        x4 = XOR(x4,U8TO32_LITTLE(m + 16));
+        x5 = XOR(x5,U8TO32_LITTLE(m + 20));
+        x6 = XOR(x6,U8TO32_LITTLE(m + 24));
+        x7 = XOR(x7,U8TO32_LITTLE(m + 28));
+        x8 = XOR(x8,U8TO32_LITTLE(m + 32));
+        x9 = XOR(x9,U8TO32_LITTLE(m + 36));
+        x10 = XOR(x10,U8TO32_LITTLE(m + 40));
+        x11 = XOR(x11,U8TO32_LITTLE(m + 44));
+        x12 = XOR(x12,U8TO32_LITTLE(m + 48));
+        x13 = XOR(x13,U8TO32_LITTLE(m + 52));
+        x14 = XOR(x14,U8TO32_LITTLE(m + 56));
+        x15 = XOR(x15,U8TO32_LITTLE(m + 60));
+
+        j12 = PLUSONE(j12);
+        if (!j12) {
+            j13 = PLUSONE(j13);
+            /* stopping at 2^70 bytes per nonce is user's responsibility */
+        }
+
+        U32TO8_LITTLE(c + 0,x0);
+        U32TO8_LITTLE(c + 4,x1);
+        U32TO8_LITTLE(c + 8,x2);
+        U32TO8_LITTLE(c + 12,x3);
+        U32TO8_LITTLE(c + 16,x4);
+        U32TO8_LITTLE(c + 20,x5);
+        U32TO8_LITTLE(c + 24,x6);
+        U32TO8_LITTLE(c + 28,x7);
+        U32TO8_LITTLE(c + 32,x8);
+        U32TO8_LITTLE(c + 36,x9);
+        U32TO8_LITTLE(c + 40,x10);
+        U32TO8_LITTLE(c + 44,x11);
+        U32TO8_LITTLE(c + 48,x12);
+        U32TO8_LITTLE(c + 52,x13);
+        U32TO8_LITTLE(c + 56,x14);
+        U32TO8_LITTLE(c + 60,x15);
+
+        if (bytes <= 64) {
+            if (bytes < 64) {
+                for (i = 0;i < bytes;++i) ctarget[i] = c[i];
+            }
+            x->input[12] = j12;
+            x->input[13] = j13;
+            return;
+        }
+        bytes -= 64;
+        c += 64;
+        m += 64;
+    }
+}
diff --git a/src/external/chachapoly.c b/src/external/chachapoly.c
new file mode 100644
index 0000000..51590a6
--- /dev/null
+++ b/src/external/chachapoly.c
@@ -0,0 +1,134 @@
+/*
+ * Copyright (c) 2013 Damien Miller <djm@xxxxxxxxxxx>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+/* $OpenBSD: cipher-chachapoly.c,v 1.8 2016/08/03 05:41:57 djm Exp $ */
+
+/* #include "includes.h" */
+
+#include <sys/types.h>
+#include <stdarg.h> /* needed for log.h */
+#include <string.h>
+#include <stdio.h>  /* needed for misc.h */
+
+#include "libssh/pki_priv.h"
+#include "libssh/chachapoly.h"
+#include "libssh/buffer.h"
+
+#define SSH_ERR_INVALID_ARGUMENT -1
+#define SSH_ERR_INTERNAL_ERROR -2
+#define SSH_ERR_MAC_INVALID -3
+#define SSH_ERR_MESSAGE_INCOMPLETE -4
+
+int
+chachapoly_init(struct chachapoly_ctx *ctx,
+                const u_char *key, u_int keylen)
+{
+    if (keylen != (32 + 32)) /* 2 x 256 bit keys */
+        return SSH_ERR_INVALID_ARGUMENT;
+    chacha_keysetup(&ctx->main_ctx, key, 256);
+    chacha_keysetup(&ctx->header_ctx, key + 32, 256);
+    return 0;
+}
+static int
+timingsafe_bcmp(const void *b1, const void *b2, size_t n)
+{
+    const unsigned char *p1 = b1, *p2 = b2;
+    int ret = 0;
+
+    for (; n > 0; n--)
+        ret |= *p1++ ^ *p2++;
+    return (ret != 0);
+}
+
+/*
+ * chachapoly_crypt() operates as following:
+ * En/decrypt with header key 'aadlen' bytes from 'src', storing result
+ * to 'dest'. The ciphertext here is treated as additional authenticated
+ * data for MAC calculation.
+ * En/decrypt 'len' bytes at offset 'aadlen' from 'src' to 'dest'. Use
+ * POLY1305_TAGLEN bytes at offset 'len'+'aadlen' as the authentication
+ * tag. This tag is written on encryption and verified on decryption.
+ */
+int
+chachapoly_crypt(struct chachapoly_ctx *ctx, u_int seqnr, u_char *dest,
+                 const u_char *src, u_int len, u_int aadlen, u_int authlen, int do_encrypt)
+{
+    u_char seqbuf[8];
+    const u_char one[8] = { 1, 0, 0, 0, 0, 0, 0, 0 }; /* NB little-endian */
+    u_char expected_tag[POLY1305_TAGLEN], poly_key[POLY1305_KEYLEN];
+    int r = SSH_ERR_INTERNAL_ERROR;
+    (void)authlen;
+
+    /*
+     * Run ChaCha20 once to generate the Poly1305 key. The IV is the
+     * packet sequence number.
+     */
+    memset(poly_key, 0, sizeof(poly_key));
+    POKE_U64(seqbuf, seqnr);
+    chacha_ivsetup(&ctx->main_ctx, seqbuf, NULL);
+    chacha_encrypt_bytes(&ctx->main_ctx,
+                         poly_key, poly_key, sizeof(poly_key));
+
+    /* If decrypting, check tag before anything else */
+    if (!do_encrypt) {
+        const u_char *tag = src + aadlen + len;
+
+        poly1305_auth(expected_tag, src, aadlen + len, poly_key);
+        if (timingsafe_bcmp(expected_tag, tag, POLY1305_TAGLEN) != 0) {
+            r = SSH_ERR_MAC_INVALID;
+            goto out;
+        }
+    }
+
+    /* Crypt additional data */
+    if (aadlen) {
+        chacha_ivsetup(&ctx->header_ctx, seqbuf, NULL);
+        chacha_encrypt_bytes(&ctx->header_ctx, src, dest, aadlen);
+    }
+
+    /* Set Chacha's block counter to 1 */
+    chacha_ivsetup(&ctx->main_ctx, seqbuf, one);
+    chacha_encrypt_bytes(&ctx->main_ctx, src + aadlen,
+                         dest + aadlen, len);
+
+    /* If encrypting, calculate and append tag */
+    if (do_encrypt) {
+        poly1305_auth(dest + aadlen + len, dest, aadlen + len,
+                      poly_key);
+    }
+    r = 0;
+ out:
+    BURN_BUFFER(expected_tag, sizeof(expected_tag));
+    BURN_BUFFER(seqbuf, sizeof(seqbuf));
+    BURN_BUFFER(poly_key, sizeof(poly_key));
+    return r;
+}
+
+/* Decrypt and extract the encrypted packet length */
+int
+chachapoly_get_length(struct chachapoly_ctx *ctx,
+                      u_int *plenp, u_int seqnr, const u_char *cp, u_int len)
+{
+    u_char buf[4], seqbuf[8];
+
+    if (len < 4)
+        return SSH_ERR_MESSAGE_INCOMPLETE;
+    POKE_U64(seqbuf, seqnr);
+    chacha_ivsetup(&ctx->header_ctx, seqbuf, NULL);
+    chacha_encrypt_bytes(&ctx->header_ctx, cp, buf, 4);
+    *plenp = PEEK_U32(buf);
+    return 0;
+}
diff --git a/src/external/poly1305.c b/src/external/poly1305.c
new file mode 100644
index 0000000..e8f497e
--- /dev/null
+++ b/src/external/poly1305.c
@@ -0,0 +1,158 @@
+/*
+ * Public Domain poly1305 from Andrew Moon
+ * poly1305-donna-unrolled.c from https://github.com/floodyberry/poly1305-donna
+ */
+
+/* $OpenBSD: poly1305.c,v 1.3 2013/12/19 22:57:13 djm Exp $ */
+
+/* #include "includes.h" */
+
+#include <sys/types.h>
+#include <stdint.h>
+
+#include "libssh/poly1305.h"
+
+#define mul32x32_64(a,b) ((uint64_t)(a) * (b))
+
+#define U8TO32_LE(p)                            \
+    (((uint32_t)((p)[0])) |                     \
+     ((uint32_t)((p)[1]) <<  8) |               \
+     ((uint32_t)((p)[2]) << 16) |               \
+     ((uint32_t)((p)[3]) << 24))
+
+#define U32TO8_LE(p, v)                         \
+    do {                                        \
+        (p)[0] = (uint8_t)((v));                \
+        (p)[1] = (uint8_t)((v) >>  8);          \
+        (p)[2] = (uint8_t)((v) >> 16);          \
+        (p)[3] = (uint8_t)((v) >> 24);          \
+    } while (0)
+
+void
+poly1305_auth(unsigned char out[POLY1305_TAGLEN], const unsigned char *m, size_t inlen, const unsigned char key[POLY1305_KEYLEN]) {
+    uint32_t t0,t1,t2,t3;
+    uint32_t h0,h1,h2,h3,h4;
+    uint32_t r0,r1,r2,r3,r4;
+    uint32_t s1,s2,s3,s4;
+    uint32_t b, nb;
+    size_t j;
+    uint64_t t[5];
+    uint64_t f0,f1,f2,f3;
+    uint32_t g0,g1,g2,g3,g4;
+    uint64_t c;
+    unsigned char mp[16];
+
+    /* clamp key */
+    t0 = U8TO32_LE(key+0);
+    t1 = U8TO32_LE(key+4);
+    t2 = U8TO32_LE(key+8);
+    t3 = U8TO32_LE(key+12);
+
+    /* precompute multipliers */
+    r0 = t0 & 0x3ffffff; t0 >>= 26; t0 |= t1 << 6;
+    r1 = t0 & 0x3ffff03; t1 >>= 20; t1 |= t2 << 12;
+    r2 = t1 & 0x3ffc0ff; t2 >>= 14; t2 |= t3 << 18;
+    r3 = t2 & 0x3f03fff; t3 >>= 8;
+    r4 = t3 & 0x00fffff;
+
+    s1 = r1 * 5;
+    s2 = r2 * 5;
+    s3 = r3 * 5;
+    s4 = r4 * 5;
+
+    /* init state */
+    h0 = 0;
+    h1 = 0;
+    h2 = 0;
+    h3 = 0;
+    h4 = 0;
+
+    /* full blocks */
+    if (inlen < 16) goto poly1305_donna_atmost15bytes;
+ poly1305_donna_16bytes:
+    m += 16;
+    inlen -= 16;
+
+    t0 = U8TO32_LE(m-16);
+    t1 = U8TO32_LE(m-12);
+    t2 = U8TO32_LE(m-8);
+    t3 = U8TO32_LE(m-4);
+
+    h0 += t0 & 0x3ffffff;
+    h1 += ((((uint64_t)t1 << 32) | t0) >> 26) & 0x3ffffff;
+    h2 += ((((uint64_t)t2 << 32) | t1) >> 20) & 0x3ffffff;
+    h3 += ((((uint64_t)t3 << 32) | t2) >> 14) & 0x3ffffff;
+    h4 += (t3 >> 8) | (1 << 24);
+
+
+ poly1305_donna_mul:
+    t[0]  = mul32x32_64(h0,r0) + mul32x32_64(h1,s4) + mul32x32_64(h2,s3) + mul32x32_64(h3,s2) + mul32x32_64(h4,s1);
+    t[1]  = mul32x32_64(h0,r1) + mul32x32_64(h1,r0) + mul32x32_64(h2,s4) + mul32x32_64(h3,s3) + mul32x32_64(h4,s2);
+    t[2]  = mul32x32_64(h0,r2) + mul32x32_64(h1,r1) + mul32x32_64(h2,r0) + mul32x32_64(h3,s4) + mul32x32_64(h4,s3);
+    t[3]  = mul32x32_64(h0,r3) + mul32x32_64(h1,r2) + mul32x32_64(h2,r1) + mul32x32_64(h3,r0) + mul32x32_64(h4,s4);
+    t[4]  = mul32x32_64(h0,r4) + mul32x32_64(h1,r3) + mul32x32_64(h2,r2) + mul32x32_64(h3,r1) + mul32x32_64(h4,r0);
+
+    h0 = (uint32_t)t[0] & 0x3ffffff; c =           (t[0] >> 26);
+    t[1] += c;      h1 = (uint32_t)t[1] & 0x3ffffff; b = (uint32_t)(t[1] >> 26);
+    t[2] += b;      h2 = (uint32_t)t[2] & 0x3ffffff; b = (uint32_t)(t[2] >> 26);
+    t[3] += b;      h3 = (uint32_t)t[3] & 0x3ffffff; b = (uint32_t)(t[3] >> 26);
+    t[4] += b;      h4 = (uint32_t)t[4] & 0x3ffffff; b = (uint32_t)(t[4] >> 26);
+    h0 += b * 5;
+
+    if (inlen >= 16) goto poly1305_donna_16bytes;
+
+    /* final bytes */
+ poly1305_donna_atmost15bytes:
+    if (!inlen) goto poly1305_donna_finish;
+
+    for (j = 0; j < inlen; j++) mp[j] = m[j];
+    mp[j++] = 1;
+    for (; j < 16; j++)	mp[j] = 0;
+    inlen = 0;
+
+    t0 = U8TO32_LE(mp+0);
+    t1 = U8TO32_LE(mp+4);
+    t2 = U8TO32_LE(mp+8);
+    t3 = U8TO32_LE(mp+12);
+
+    h0 += t0 & 0x3ffffff;
+    h1 += ((((uint64_t)t1 << 32) | t0) >> 26) & 0x3ffffff;
+    h2 += ((((uint64_t)t2 << 32) | t1) >> 20) & 0x3ffffff;
+    h3 += ((((uint64_t)t3 << 32) | t2) >> 14) & 0x3ffffff;
+    h4 += (t3 >> 8);
+
+    goto poly1305_donna_mul;
+
+ poly1305_donna_finish:
+    b = h0 >> 26; h0 = h0 & 0x3ffffff;
+    h1 +=     b; b = h1 >> 26; h1 = h1 & 0x3ffffff;
+    h2 +=     b; b = h2 >> 26; h2 = h2 & 0x3ffffff;
+    h3 +=     b; b = h3 >> 26; h3 = h3 & 0x3ffffff;
+    h4 +=     b; b = h4 >> 26; h4 = h4 & 0x3ffffff;
+    h0 += b * 5; b = h0 >> 26; h0 = h0 & 0x3ffffff;
+    h1 +=     b;
+
+    g0 = h0 + 5; b = g0 >> 26; g0 &= 0x3ffffff;
+    g1 = h1 + b; b = g1 >> 26; g1 &= 0x3ffffff;
+    g2 = h2 + b; b = g2 >> 26; g2 &= 0x3ffffff;
+    g3 = h3 + b; b = g3 >> 26; g3 &= 0x3ffffff;
+    g4 = h4 + b - (1 << 26);
+
+    b = (g4 >> 31) - 1;
+    nb = ~b;
+    h0 = (h0 & nb) | (g0 & b);
+    h1 = (h1 & nb) | (g1 & b);
+    h2 = (h2 & nb) | (g2 & b);
+    h3 = (h3 & nb) | (g3 & b);
+    h4 = (h4 & nb) | (g4 & b);
+
+    f0 = ((h0      ) | (h1 << 26)) + (uint64_t)U8TO32_LE(&key[16]);
+    f1 = ((h1 >>  6) | (h2 << 20)) + (uint64_t)U8TO32_LE(&key[20]);
+    f2 = ((h2 >> 12) | (h3 << 14)) + (uint64_t)U8TO32_LE(&key[24]);
+    f3 = ((h3 >> 18) | (h4 <<  8)) + (uint64_t)U8TO32_LE(&key[28]);
+
+    U32TO8_LE(&out[ 0], f0); f1 += (f0 >> 32);
+    U32TO8_LE(&out[ 4], f1); f2 += (f1 >> 32);
+    U32TO8_LE(&out[ 8], f2); f3 += (f2 >> 32);
+    U32TO8_LE(&out[12], f3);
+}
-- 
2.1.4


Archive administrator: postmaster@lists.cynapses.org