pub/, progs/: Implement Bernstein's Ed25519 signature scheme.

[catacomb] / progs / perftest.c

diff --git a/progs/perftest.c b/progs/perftest.c
index 97a2e0f..3377487 100644 (file)
--- a/progs/perftest.c
+++ b/progs/perftest.c
@@ -62,11 +62,15 @@
 #include "pgen.h"
 #include "ec.h"
 #include "group.h"
+#include "x25519.h"
+#include "ed25519.h"
 
 #include "cc.h"
 #include "gcipher.h"
 #include "ghash.h"
 #include "gmac.h"
+#include "poly1305.h"
+
 #include "ectab.h"
 #include "ptab.h"
 
@@ -267,6 +271,74 @@ static void grsim_run(void *cc)
   G_DESTROY(c->g, x);
 }
 
+/* --- x25519 --- */
+
+typedef struct x25519_jobctx {
+  octet k[X25519_KEYSZ];
+  octet p[X25519_PUBSZ];
+} x25519_jobctx;
+
+static void *x25519_jobinit(opts *o)
+{
+  x25519_jobctx *c = CREATE(x25519_jobctx);
+  rand_get(RAND_GLOBAL, c->k, sizeof(c->k));
+  rand_get(RAND_GLOBAL, c->p, sizeof(c->p));
+  return (c);
+}
+
+static void x25519_jobrun(void *cc)
+  { x25519_jobctx *c = cc; octet z[X25519_OUTSZ]; x25519(z, c->k, c->p); }
+
+/* --- Ed25519 --- */
+
+typedef struct ed25519_signctx {
+  octet k[ED25519_KEYSZ];
+  octet K[ED25519_PUBSZ];
+  octet m[64];
+} ed25519_signctx;
+
+typedef struct ed25519_vrfctx {
+  octet K[ED25519_PUBSZ];
+  octet m[64];
+  octet sig[ED25519_SIGSZ];
+} ed25519_vrfctx;
+
+static void *ed25519_signinit(opts *o)
+{
+  ed25519_signctx *c = CREATE(ed25519_signctx);
+
+  rand_get(RAND_GLOBAL, c->k, sizeof(c->k));
+  rand_get(RAND_GLOBAL, c->m, sizeof(c->m));
+  ed25519_pubkey(c->K, c->k, sizeof(c->k));
+  return (c);
+}
+
+static void ed25519_signrun(void *cc)
+{
+  ed25519_signctx *c = cc;
+  octet sig[ED25519_SIGSZ];
+
+  ed25519_sign(sig, c->k, sizeof(c->k), c->K, c->m, sizeof(c->m));
+}
+
+static void *ed25519_vrfinit(opts *o)
+{
+  octet k[ED25519_KEYSZ];
+  ed25519_vrfctx *c = CREATE(ed25519_vrfctx);
+
+  rand_get(RAND_GLOBAL, k, sizeof(k));
+  rand_get(RAND_GLOBAL, c->m, sizeof(c->m));
+  ed25519_pubkey(c->K, k, sizeof(k));
+  ed25519_sign(c->sig, k, sizeof(k), c->K, c->m, sizeof(c->m));
+  return (c);
+}
+
+static void ed25519_vrfrun(void *cc)
+{
+  ed25519_vrfctx *c = cc;
+  ed25519_verify(c->K, c->m, sizeof(c->m), c->sig);
+}
+
 /* --- RSA --- */
 
 typedef struct rsapriv_ctx {
@@ -429,6 +501,40 @@ static void hash_run(void *cc)
   GH_DESTROY(h);
 }
 
+/* --- Poly1305 --- */
+
+typedef struct poly1305_jobctx {
+  poly1305_key k;
+  octet s[POLY1305_MASKSZ];
+  octet *m;
+  size_t sz;
+  size_t n;
+} poly1305_jobctx;
+
+static void *poly1305_jobinit(opts *o)
+{
+  octet k[POLY1305_KEYSZ];
+  poly1305_jobctx *c = CREATE(poly1305_jobctx);
+  rand_get(RAND_GLOBAL, k, sizeof(k));
+  poly1305_keyinit(&c->k, k, sizeof(k));
+  rand_get(RAND_GLOBAL, c->s, sizeof(c->s));
+  c->sz = o->gbits ? o->gbits : 65536;
+  c->n = o->n ? o->n : 16;
+  c->m = xmalloc(c->sz);
+  return (c);
+}
+
+static void poly1305_jobrun(void *cc)
+{
+  poly1305_jobctx *c = cc;
+  poly1305_ctx ctx;
+  octet t[POLY1305_TAGSZ];
+  size_t i;
+  poly1305_macinit(&ctx, &c->k, c->s);
+  for (i = 0; i < c->n; i++) poly1305_hash(&ctx, c->m, c->sz);
+  poly1305_done(&ctx, t);
+}
+
 /* --- Job table --- */
 
 typedef struct jobops {
@@ -449,9 +555,13 @@ static const jobops jobtab[] = {
   { "rsa-priv",                        rsapriv_init,           rsapriv_run },
   { "rsa-priv-blind",          rsaprivblind_init,      rsapriv_run },
   { "rsa-pub",                 rsapub_init,            rsapub_run },
+  { "x25519",                  x25519_jobinit,         x25519_jobrun },
+  { "ed25519-sign",            ed25519_signinit,       ed25519_signrun },
+  { "ed25519-vrf",             ed25519_vrfinit,        ed25519_vrfrun },
   { "ksched",                  ksched_init,            ksched_run },
   { "enc",                     enc_init,               enc_run },
   { "hash",                    hash_init,              hash_run },
+  { "poly1305",                        poly1305_jobinit,       poly1305_jobrun },
   { 0,                         0,                      0 }
 };