===================================================================
RCS file: /cvsrepo/anoncvs/cvs/src/usr.bin/ssh/cipher.c,v
retrieving revision 1.39
retrieving revision 1.40
diff -u -r1.39 -r1.40
--- src/usr.bin/ssh/cipher.c	2000/12/06 23:05:42	1.39
+++ src/usr.bin/ssh/cipher.c	2000/12/09 13:41:52	1.40
@@ -35,7 +35,7 @@
  */
 
 #include "includes.h"
-RCSID("$OpenBSD: cipher.c,v 1.39 2000/12/06 23:05:42 markus Exp $");
+RCSID("$OpenBSD: cipher.c,v 1.40 2000/12/09 13:41:52 markus Exp $");
 
 #include "ssh.h"
 #include "xmalloc.h"
@@ -285,45 +285,40 @@
 /* RIJNDAEL */
 
 #define RIJNDAEL_BLOCKSIZE 16
-
 void
 rijndael_setkey(CipherContext *cc, const u_char *key, u_int keylen)
 {
-	if (rijndael_makekey(&cc->u.rijndael.enc, RIJNDAEL_ENCRYPT,
-		    8*keylen, (char *)key) == -1)
-		fatal("rijndael_setkey: RIJNDAEL_ENCRYPT");
-	if (rijndael_makekey(&cc->u.rijndael.dec, RIJNDAEL_DECRYPT,
-		    8*keylen, (char *)key) == -1)
-		fatal("rijndael_setkey: RIJNDAEL_DECRYPT");
+	rijndael_set_key(&cc->u.rijndael.enc, (u4byte *)key, 8*keylen, 1);
+	rijndael_set_key(&cc->u.rijndael.dec, (u4byte *)key, 8*keylen, 0);
 }
 void
 rijndael_setiv(CipherContext *cc, const u_char *iv, u_int ivlen)
 {
-	if (iv == NULL || ivlen != RIJNDAEL_BLOCKSIZE) 
-		fatal("bad/no IV for %s.", cc->cipher->name);
-	memcpy(cc->u.rijndael.iv, iv, RIJNDAEL_BLOCKSIZE);
+	if (iv == NULL) 
+		fatal("no IV for %s.", cc->cipher->name);
+	memcpy((u_char *)cc->u.rijndael.iv, iv, RIJNDAEL_BLOCKSIZE);
 }
-
 void
 rijndael_cbc_encrypt(CipherContext *cc, u_char *dest, const u_char *src,
     u_int len)
 {
-	rijndael_key *ctx = &cc->u.rijndael.enc;
-	u_char *iv = cc->u.rijndael.iv;
-	u_char in[RIJNDAEL_BLOCKSIZE];
-	u_char *cprev, *cnow, *plain;
-	int i, j, blocks = len / RIJNDAEL_BLOCKSIZE;
+	rijndael_ctx *ctx = &cc->u.rijndael.enc;
+	u4byte *iv = cc->u.rijndael.iv;
+	u4byte in[4];
+	u4byte *cprev, *cnow, *plain;
+	int i, blocks = len / RIJNDAEL_BLOCKSIZE;
 	if (len == 0)
 		return;
 	if (len % RIJNDAEL_BLOCKSIZE)
 		fatal("rijndael_cbc_encrypt: bad len %d", len);
-	cnow  = dest;
-	plain = (u_char *) src;
+	cnow  = (u4byte*) dest;
+	plain = (u4byte*) src;
 	cprev = iv;
-	for(i = 0; i < blocks; i++, plain+=RIJNDAEL_BLOCKSIZE,
-	    cnow+=RIJNDAEL_BLOCKSIZE) {
-		for (j = 0; j < RIJNDAEL_BLOCKSIZE; j++)
-			in[j] = plain[j] ^ cprev[j];
+	for(i = 0; i < blocks; i++, plain+=4, cnow+=4) {
+		in[0] = plain[0] ^ cprev[0];
+		in[1] = plain[1] ^ cprev[1];
+		in[2] = plain[2] ^ cprev[2];
+		in[3] = plain[3] ^ cprev[3];
 		rijndael_encrypt(ctx, in, cnow);
 		cprev = cnow;
 	}
@@ -334,25 +329,25 @@
 rijndael_cbc_decrypt(CipherContext *cc, u_char *dest, const u_char *src,
     u_int len)
 {
-	rijndael_key *ctx = &cc->u.rijndael.dec;
-	u_char *iv = cc->u.rijndael.iv;
-	u_char ivsaved[RIJNDAEL_BLOCKSIZE];
-	u_char *cnow  = (u_char *) (src+len-RIJNDAEL_BLOCKSIZE);
-	u_char *plain = dest+len-RIJNDAEL_BLOCKSIZE;
-	u_char *ivp;
-	int i, j, blocks = len / RIJNDAEL_BLOCKSIZE;
+	rijndael_ctx *ctx = &cc->u.rijndael.dec;
+	u4byte *iv = cc->u.rijndael.iv;
+	u4byte ivsaved[4];
+	u4byte *cnow =  (u4byte*) (src+len-RIJNDAEL_BLOCKSIZE);
+	u4byte *plain = (u4byte*) (dest+len-RIJNDAEL_BLOCKSIZE);
+	u4byte *ivp;
+	int i, blocks = len / RIJNDAEL_BLOCKSIZE;
 	if (len == 0)
 		return;
 	if (len % RIJNDAEL_BLOCKSIZE)
 		fatal("rijndael_cbc_decrypt: bad len %d", len);
 	memcpy(ivsaved, cnow, RIJNDAEL_BLOCKSIZE);
-	for(i = blocks; i > 0; i--, cnow-=RIJNDAEL_BLOCKSIZE,
-	    plain-=RIJNDAEL_BLOCKSIZE) {
+	for(i = blocks; i > 0; i--, cnow-=4, plain-=4) {
 		rijndael_decrypt(ctx, cnow, plain);
-		//rijndael_decrypt(cnow, plain, ctx->keySched, ctx->ROUNDS);
-		ivp = (i == 1) ? iv : cnow-RIJNDAEL_BLOCKSIZE;
-		for (j = 0; j < RIJNDAEL_BLOCKSIZE; j++)
-			plain[j] ^= ivp[j];
+		ivp =  (i == 1) ? iv : cnow-4;
+		plain[0] ^= ivp[0];
+		plain[1] ^= ivp[1];
+		plain[2] ^= ivp[2];
+		plain[3] ^= ivp[3];
 	}
 	memcpy(iv, ivsaved, RIJNDAEL_BLOCKSIZE);
 }