catcrypt: Implement symmetric key-encapsulation and signature schemes.

[u/mdw/catacomb] / cbc-def.h

diff --git a/cbc-def.h b/cbc-def.h
index c0fc600..00695ae 100644 (file)
--- a/cbc-def.h
+++ b/cbc-def.h
@@ -1,6 +1,6 @@
 /* -*-c-*-
  *
- * $Id: cbc-def.h,v 1.5 2004/04/08 01:36:15 mdw Exp $
+ * $Id: cbc-def.h,v 1.6 2004/04/17 09:58:36 mdw Exp $
  *
  * Definitions for cipher block chaining mode
  *
@@ -179,8 +179,10 @@ void pre##_cbcencrypt(pre##_cbcctx *ctx,                           \
                                                                        \
     pre##_eblk(&ctx->ctx, ctx->iv, ctx->iv);                           \
     BLKC_STORE(PRE, b, ctx->iv);                                       \
-    for (i = 0; i < sz; i++)                                           \
-      d[i] = b[i] ^ s[i];                                              \
+    if (d) {                                                           \
+      for (i = 0; i < sz; i++)                                         \
+        d[i] = b[i] ^ (s ? s[i] : 0);                                  \
+    }                                                                  \
     memmove(b, b + sz, PRE##_BLKSZ - sz);                              \
     memcpy(b + PRE##_BLKSZ - sz, d, sz);                               \
     BLKC_LOAD(PRE, ctx->iv, b);                                                \
@@ -195,11 +197,15 @@ void pre##_cbcencrypt(pre##_cbcctx *ctx,                          \
    */                                                                  \
                                                                        \
   while (sz >= 2 * PRE##_BLKSZ || sz == PRE##_BLKSZ) {                 \
-    BLKC_XLOAD(PRE, ctx->iv, s);                                       \
+    if (s) {                                                           \
+      BLKC_XLOAD(PRE, ctx->iv, s);                                     \
+      s += PRE##_BLKSZ;                                                        \
+    }                                                                  \
     pre##_eblk(&ctx->ctx, ctx->iv, ctx->iv);                           \
-    BLKC_STORE(PRE, d, ctx->iv);                                       \
-    s += PRE##_BLKSZ;                                                  \
-    d += PRE##_BLKSZ;                                                  \
+    if (d) {                                                           \
+      BLKC_STORE(PRE, d, ctx->iv);                                     \
+      d += PRE##_BLKSZ;                                                        \
+    }                                                                  \
     sz -= PRE##_BLKSZ;                                                 \
   }                                                                    \
                                                                        \
@@ -224,7 +230,7 @@ void pre##_cbcencrypt(pre##_cbcctx *ctx,                            \
      * block.                                                          \
      */                                                                        \
                                                                        \
-    BLKC_XLOAD(PRE, ctx->iv, s);                                       \
+    if (s) BLKC_XLOAD(PRE, ctx->iv, s);                                        \
     pre##_eblk(&ctx->ctx, ctx->iv, ctx->iv);                           \
     BLKC_STORE(PRE, b, ctx->iv);                                       \
                                                                        \
@@ -235,16 +241,16 @@ void pre##_cbcencrypt(pre##_cbcctx *ctx,                          \
      * ciphertext block.                                               \
      */                                                                        \
                                                                        \
-    s += PRE##_BLKSZ;                                                  \
-    d += PRE##_BLKSZ;                                                  \
+    if (s) s += PRE##_BLKSZ;                                           \
+    if (d) d += PRE##_BLKSZ;                                           \
     for (i = 0; i < sz; i++) {                                         \
       register octet x = b[i];                                         \
-      b[i] ^= s[i];                                                    \
-      d[i] = x;                                                                \
+      if (s) b[i] ^= s[i];                                             \
+      if (d) d[i] = x;                                                 \
     }                                                                  \
     BLKC_LOAD(PRE, ctx->iv, b);                                                \
     pre##_eblk(&ctx->ctx, ctx->iv, ctx->iv);                           \
-    BLKC_STORE(PRE, d - PRE##_BLKSZ, ctx->iv);                         \
+    if (d) BLKC_STORE(PRE, d - PRE##_BLKSZ, ctx->iv);                  \
   }                                                                    \
                                                                        \
   /* --- Done --- */                                                   \