X-Git-Url: https://git.cryptolib.org/?a=blobdiff_plain;f=threefish256_enc_small.S;h=da0835995af1c3feee3e1eda908ef94760f1fdc4;hb=ff9f1b74e757f2f67b11c51f65b12fffaa02e18d;hp=c1b1152d2a08e3db108a2dfb069da0550936e62f;hpb=d1d3ca64870a4cc4ee0659db7b4387e1dd51549f;p=avr-crypto-lib.git

diff --git a/threefish256_enc_small.S b/threefish256_enc_small.S
index c1b1152..da08359 100644
--- a/threefish256_enc_small.S
+++ b/threefish256_enc_small.S
@@ -104,6 +104,17 @@ threefish256_init:
 	st Z+, A7
 	/* now the tweak */
 	movw r26, r22
+	tst r27
+	brne 3f
+	tst r26
+	brne 3f
+	ldi r26, 3*8
+1:
+	st Z+, r1
+	dec r26
+	brne 1b
+	rjmp 9f
+3:	
 	ld A0, X+
 	ld A1, X+
 	ld A2, X+
@@ -152,6 +163,7 @@ threefish256_init:
 	st Z+, A5
 	st Z+, A6
 	st Z+, A7
+9:
 	pop_range 14, 17
 	ret