#include "grand.h"
#include "keysz.h"
#include "paranoia.h"
+#include "rsvr.h"
#include "salsa20.h"
#include "salsa20-core.h"
#if CPUFAM_X86 || CPUFAM_AMD64
extern core__functype salsa20_core_x86ish_sse2;
+extern core__functype salsa20_core_x86ish_avx;
#endif
#if CPUFAM_ARMEL
extern core__functype salsa20_core_arm_neon;
#endif
+#if CPUFAM_ARM64
+extern core__functype salsa20_core_arm64;
+#endif
+
static core__functype *pick_core(void)
{
#if CPUFAM_X86 || CPUFAM_AMD64
+ DISPATCH_PICK_COND(salsa20_core, salsa20_core_x86ish_avx,
+ cpu_feature_p(CPUFEAT_X86_AVX));
DISPATCH_PICK_COND(salsa20_core, salsa20_core_x86ish_sse2,
cpu_feature_p(CPUFEAT_X86_SSE2));
#endif
DISPATCH_PICK_COND(salsa20_core, salsa20_core_arm_neon,
cpu_feature_p(CPUFEAT_ARM_NEON));
#endif
+#if CPUFAM_ARM64
+ DISPATCH_PICK_COND(salsa20_core, salsa20_core_arm64, 1);
+#endif
DISPATCH_PICK_FALLBACK(salsa20_core, simple_core);
}
salsa20_setnonce(ctx, nonce ? nonce : zerononce);
}
-/* --- @salsa20_setnonce@ --- *
+/* --- @salsa20_setnonce{,_ietf}@ --- *
*
* Arguments: @salsa20_ctx *ctx@ = pointer to context
- * @const void *nonce@ = the nonce (@SALSA20_NONCESZ@ bytes)
+ * @const void *nonce@ = the nonce (@SALSA20_NONCESZ@ or
+ * @SALSA20_IETF_NONCESZ@ bytes)
*
* Returns: ---
*
salsa20_seek(ctx, 0);
}
-/* --- @salsa20_seek@, @salsa20_seeku64@ --- *
+void salsa20_setnonce_ietf(salsa20_ctx *ctx, const void *nonce)
+{
+ const octet *n = nonce;
+
+ ctx->a[ 5] = LOAD32_L(n + 0);
+ ctx->a[14] = LOAD32_L(n + 4);
+ ctx->a[11] = LOAD32_L(n + 8);
+ salsa20_seek_ietf(ctx, 0);
+}
+
+/* --- @salsa20_seek{,u64,_ietf}@ --- *
*
* Arguments: @salsa20_ctx *ctx@ = pointer to context
- * @unsigned long i@, @kludge64 i@ = new position to set
+ * @unsigned long i@, @kludge64 i@, @uint32@ = new position
*
* Returns: ---
*
void salsa20_seeku64(salsa20_ctx *ctx, kludge64 i)
{
ctx->a[8] = LO64(i); ctx->a[5] = HI64(i);
- ctx->bufi = SALSA20_OUTSZ;
+ ctx->off = 0;
}
-/* --- @salsa20_tell@, @salsa20_tellu64@ --- *
+void salsa20_seek_ietf(salsa20_ctx *ctx, uint32 i)
+ { ctx->a[8] = i; }
+
+/* --- @salsa20_tell{,u64,_ietf}@ --- *
*
* Arguments: @salsa20_ctx *ctx@ = pointer to context
*
kludge64 salsa20_tellu64(salsa20_ctx *ctx)
{ kludge64 i; SET64(i, ctx->a[5], ctx->a[8]); return (i); }
+uint32 salsa20_tell_ietf(salsa20_ctx *ctx)
+ { return (ctx->a[5]); }
+
/* --- @salsa20{,12,8}_encrypt@ --- *
*
* Arguments: @salsa20_ctx *ctx@ = pointer to context
* to @dest@.
*/
+static const rsvr_policy policy = { 0, SALSA20_OUTSZ, SALSA20_OUTSZ };
+
#define SALSA20_ENCRYPT(r, ctx, src, dest, sz) \
SALSA20_DECOR(salsa20, r, _encrypt)(ctx, src, dest, sz)
#define DEFENCRYPT(r) \
salsa20_matrix b; \
const octet *s = src; \
octet *d = dest; \
- size_t n; \
+ rsvr_plan plan; \
kludge64 pos, delta; \
\
- SALSA20_OUTBUF(ctx, d, s, sz); \
- if (!sz) return; \
- \
- if (!dest) { \
- n = sz/SALSA20_OUTSZ; \
- pos = salsa20_tellu64(ctx); \
- ASSIGN64(delta, n); \
- ADD64(pos, pos, delta); \
- salsa20_seeku64(ctx, pos); \
- sz = sz%SALSA20_OUTSZ; \
- } else if (!src) { \
- while (sz >= SALSA20_OUTSZ) { \
- core(r, ctx->a, b); \
- SALSA20_STEP(ctx->a); \
- SALSA20_GENFULL(b, d); \
- sz -= SALSA20_OUTSZ; \
+ rsvr_mkplan(&plan, &policy, ctx->off, sz); \
+ \
+ if (plan.head) { \
+ if (!ctx->off) { \
+ core(r, ctx->a, b); SALSA20_STEP(ctx->a); \
+ SALSA20_PREPBUF(ctx, b); \
} \
- } else { \
- while (sz >= SALSA20_OUTSZ) { \
- core(r, ctx->a, b); \
- SALSA20_STEP(ctx->a); \
- SALSA20_MIXFULL(b, d, s); \
- sz -= SALSA20_OUTSZ; \
+ SALSA20_OUTBUF(ctx, d, s, plan.head); \
+ } \
+ \
+ ctx->off -= plan.from_rsvr; \
+ \
+ if (!d) { \
+ if (plan.from_input) { \
+ pos = salsa20_tellu64(ctx); \
+ ASSIGN64(delta, plan.from_input/SALSA20_OUTSZ); \
+ ADD64(pos, pos, delta); \
+ salsa20_seeku64(ctx, pos); \
} \
+ } else if (!s) while (plan.from_input) { \
+ core(r, ctx->a, b); SALSA20_STEP(ctx->a); \
+ SALSA20_GENFULL(b, d); plan.from_input -= SALSA20_OUTSZ; \
+ } else while (plan.from_input) { \
+ core(r, ctx->a, b); SALSA20_STEP(ctx->a); \
+ SALSA20_MIXFULL(b, d, s); plan.from_input -= SALSA20_OUTSZ; \
} \
\
- if (sz) { \
- core(r, ctx->a, b); \
- SALSA20_STEP(ctx->a); \
+ if (plan.tail) { \
+ core(r, ctx->a, b); SALSA20_STEP(ctx->a); \
SALSA20_PREPBUF(ctx, b); \
- SALSA20_OUTBUF(ctx, d, s, sz); \
- assert(!sz); \
+ SALSA20_OUTBUF(ctx, d, s, plan.tail); \
} \
}
SALSA20_VARS(DEFENCRYPT)
}
SALSA20_VARS(DEFXNONCE)
-/* --- @xsalsa20{,12,8}_seek@, @xsalsa20{,12,8}_seeku64@ --- *
+/* --- @xsalsa20{,12,8}_seek{,u64}@ --- *
*
* Arguments: @xsalsa20R_ctx *ctx@ = pointer to context
* @unsigned long i@, @kludge64 i@ = new position to set
* different.
*/
-/* --- @xsalsa20{,12,8}_tell@, @xsalsa20{,12,8}_tellu64@ --- *
+/* --- @xsalsa20{,12,8}_tell{,u64}@ --- *
*
* Arguments: @salsa20_ctx *ctx@ = pointer to context
*
static void gsetiv(gcipher *c, const void *iv)
{ gctx *g = (gctx *)c; salsa20_setnonce(&g->ctx, iv); }
+static void gsetiv_ietf(gcipher *c, const void *iv)
+ { gctx *g = (gctx *)c; salsa20_setnonce_ietf(&g->ctx, iv); }
+
static void gdestroy(gcipher *c)
{ gctx *g = (gctx *)c; BURN(*g); S_DESTROY(g); }
+static gcipher *ginit(const void *k, size_t sz, const gcipher_ops *ops)
+{
+ gctx *g = S_CREATE(gctx);
+ g->c.ops = ops;
+ salsa20_init(&g->ctx, k, sz, 0);
+ return (&g->c);
+}
+
#define DEFGCIPHER(r) \
\
- static const gcipher_ops gops_##r; \
+ static const gcipher_ops gops_##r, gops_##r##_ietf; \
\
static gcipher *ginit_##r(const void *k, size_t sz) \
- { \
- gctx *g = S_CREATE(gctx); \
- g->c.ops = &gops_##r; \
- salsa20_init(&g->ctx, k, sz, 0); \
- return (&g->c); \
- } \
+ { return (ginit(k, sz, &gops_##r)); } \
+ \
+ static gcipher *ginit_##r##_ietf(const void *k, size_t sz) \
+ { return (ginit(k, sz, &gops_##r##_ietf)); } \
\
static void gencrypt_##r(gcipher *c, const void *s, \
void *t, size_t sz) \
gencrypt_##r, gencrypt_##r, gdestroy, gsetiv, 0 \
}; \
\
+ static const gcipher_ops gops_##r##_ietf = { \
+ &SALSA20_DECOR(salsa20, r, _ietf), \
+ gencrypt_##r, gencrypt_##r, gdestroy, gsetiv_ietf, 0 \
+ }; \
+ \
const gccipher SALSA20_DECOR(salsa20, r, ) = { \
SALSA20_NAME_##r, salsa20_keysz, \
SALSA20_NONCESZ, ginit_##r \
+ }; \
+ \
+ const gccipher SALSA20_DECOR(salsa20, r, _ietf) = { \
+ SALSA20_NAME_##r "-ietf", salsa20_keysz, \
+ SALSA20_IETF_NONCESZ, ginit_##r##_ietf \
};
SALSA20_VARS(DEFGCIPHER)
static void gr_seek(void *r, kludge64 pos)
{ grctx *g = r; salsa20_seeku64(&g->ctx, pos); }
+static void gr_seek_ietf(void *r, kludge64 pos)
+ { grctx *g = r; salsa20_seek_ietf(&g->ctx, LO64(pos)); }
+
static kludge64 gr_tell(void *r)
{ grctx *g = r; return (salsa20_tellu64(&g->ctx)); }
+static kludge64 gr_tell_ietf(void *r)
+{
+ grctx *g = r;
+ kludge64 pos;
+
+ SET64(pos, 0, salsa20_tell_ietf(&g->ctx));
+ return (pos);
+}
+
static void gr_setnonce(void *r, const void *n)
{ grctx *g = r; salsa20_setnonce(&g->ctx, n); }
+static void gr_setnonce_ietf(void *r, const void *n)
+ { grctx *g = r; salsa20_setnonce(&g->ctx, n); }
+
static void grdestroy(grand *r)
{ grctx *g = (grctx *)r; BURN(*g); S_DESTROY(g); }
+static grand *grinit(const void *k, size_t ksz, const void *n,
+ const grand_ops *ops, const grops *myops)
+{
+ grctx *g = S_CREATE(grctx);
+ g->r.r.ops = ops;
+ g->r.ops = myops;
+ salsa20_init(&g->ctx, k, ksz, 0);
+ if (n) myops->setnonce(g, n);
+ return (&g->r.r);
+}
+
#define DEFGRAND(rr) \
\
static void gr_generate_##rr(void *r, void *b, size_t sz) \
{ SALSA20_NONCESZ, gr_seek, gr_tell, \
gr_setnonce, gr_generate_##rr }; \
\
+ static const grops grops_##rr##_ietf = \
+ { SALSA20_IETF_NONCESZ, gr_seek_ietf, gr_tell_ietf, \
+ gr_setnonce_ietf, gr_generate_##rr }; \
+ \
static const grand_ops grops_rand_##rr = { \
SALSA20_NAME_##rr, GRAND_CRYPTO, 0, \
grmisc, grdestroy, grword, \
grbyte, grword, grand_defaultrange, grfill \
}; \
\
+ static const grand_ops grops_rand_##rr##_ietf = { \
+ SALSA20_NAME_##rr "-ietf", GRAND_CRYPTO, 0, \
+ grmisc, grdestroy, grword, \
+ grbyte, grword, grand_defaultrange, grfill \
+ }; \
+ \
grand *SALSA20_DECOR(salsa20, rr, _rand) \
(const void *k, size_t ksz, const void *n) \
+ { return (grinit(k, ksz, n, &grops_rand_##rr, &grops_##rr)); } \
+ \
+ grand *SALSA20_DECOR(salsa20, rr, _ietf_rand) \
+ (const void *k, size_t ksz, const void *n) \
{ \
- grctx *g = S_CREATE(grctx); \
- g->r.r.ops = &grops_rand_##rr; \
- g->r.ops = &grops_##rr; \
- salsa20_init(&g->ctx, k, ksz, n); \
- return (&g->r.r); \
+ return (grinit(k, ksz, n, \
+ &grops_rand_##rr##_ietf, \
+ &grops_##rr##_ietf)); \
}
+
SALSA20_VARS(DEFGRAND)
#define DEFXGRAND(rr) \
SALSA20_VARS(DEFVCORE)
#define SALSA20_CTX(r) salsa20_ctx
-#define SALSA20_INIT(r, ctx, k, ksz, n) salsa20_init(ctx, k, ksz, n)
-#define SALSA20_SEEKU64(r, ctx, i) salsa20_seeku64(ctx, i)
+
+#define SALSA20_TESTSETUP(r, ctx, k, ksz, n, nsz, p, psz) do { \
+ kludge64 pos64; \
+ salsa20_init(ctx, k, ksz, 0); \
+ if (nsz == 8) salsa20_setnonce(ctx, n); \
+ else if (nsz == 12) salsa20_setnonce_ietf(ctx, n); \
+ if (psz == 8) { LOAD64_(pos64, p); salsa20_seeku64(ctx, pos64); } \
+ else if (psz == 4) salsa20_seek_ietf(ctx, LOAD32(p)); \
+} while (0)
+
+#define XSALSA20_TESTSETUP(r, ctx, k, ksz, n, nsz, p, psz) do { \
+ kludge64 pos64; \
+ XSALSA20_INIT(r, ctx, k, ksz, 0); \
+ if (nsz == 24) XSALSA20_SETNONCE(r, ctx, n); \
+ if (psz == 8) { LOAD64_(pos64, p); XSALSA20_SEEKU64(r, ctx, pos64); } \
+} while (0)
#define DEFxVENC(base, BASE, r) \
static int v_encrypt_##base##_##r(dstr *v) \
{ \
BASE##_CTX(r) ctx; \
dstr d = DSTR_INIT; \
- kludge64 pos; \
const octet *p, *p0; \
octet *q; \
size_t sz, sz0, step; \
while (step < sz0 + skip) { \
step = step ? 3*step + 4 : 1; \
if (step > sz0 + skip) step = sz0 + skip; \
- BASE##_INIT(r, &ctx, v[0].buf, v[0].len, v[1].buf); \
- if (v[2].len) { \
- LOAD64_(pos, v[2].buf); \
- BASE##_SEEKU64(r, &ctx, pos); \
- } \
+ BASE##_TESTSETUP(r, &ctx, v[0].buf, v[0].len, \
+ v[1].buf, v[1].len, v[2].buf, v[2].len); \
\
for (sz = skip; sz >= step; sz -= step) \
BASE##_ENCRYPT(r, &ctx, 0, 0, step); \