X-Git-Url: https://git.cryptolib.org/?a=blobdiff_plain;f=arcfour%2Farcfour.c;h=6c068a8023ee0acb487b33a313c050f6afa8e9b4;hb=b246a2a0589f234db6247255555df98f4c281c41;hp=e07193f2cbfbc9564adff1557c4f78b3b1ed414e;hpb=d32eba56ce10ea6b9eff123b50d9842673b38f2b;p=avr-crypto-lib.git

diff --git a/arcfour/arcfour.c b/arcfour/arcfour.c
index e07193f..6c068a8 100644
--- a/arcfour/arcfour.c
+++ b/arcfour/arcfour.c
@@ -33,20 +33,27 @@
  * length is length of key in bytes!
  */
 
-void arcfour_init(const void *key, uint8_t length_B, arcfour_ctx_t *ctx){
+void arcfour_init(const void *key, uint16_t length_b, arcfour_ctx_t *ctx){
 	uint8_t t;
-	uint16_t x,y=0;
-	for(x=0; x<= 255; ++x)
+	uint8_t length_B = length_b/8;
+	uint8_t x=0,y=0;
+	uint8_t *kptr=key;
+	do{
 		ctx->s[x]=x;
+	}while(++x);
 
-	for(x=0; x<= 255; ++x){
-		y += ctx->s[x] + ((uint8_t*)key)[x % length_B];
+	do{
+		y += ctx->s[x] + *kptr++;
+		if(x==length_B){
+			kptr = key;
+		}
 		y &= 0xff;
 		/* ctx->s[y] <--> ctx->s[x] */
 		t = ctx->s[y];
 		ctx->s[y] = ctx->s[x];
 		ctx->s[x] = t;
-	}
+	}while(++x);
+
 	ctx->i = ctx->j = 0;
 }