libavutil: Added twofish symmetric block cipher
[ffmpeg.git] / libavutil / twofish.c
1 /*
2  * An implementation of the TwoFish algorithm
3  * Copyright (c) 2015 Supraja Meedinti
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 #include "twofish.h"
22 #include "common.h"
23 #include "intreadwrite.h"
24 #include "attributes.h"
25
26 #define LR(x, n) ((x) << (n) | (x) >> (32 - (n)))
27 #define RR(x, n) ((x) >> (n) | (x) << (32 - (n)))
28
29 typedef struct AVTWOFISH {
30     uint32_t K[40];
31     uint32_t S[4];
32     int ksize;
33 } AVTWOFISH;
34
35 static const uint8_t MD1[256] = {
36     0x00, 0x5b, 0xb6, 0xed, 0x05, 0x5e, 0xb3, 0xe8, 0x0a, 0x51, 0xbc, 0xe7, 0x0f, 0x54, 0xb9, 0xe2,
37     0x14, 0x4f, 0xa2, 0xf9, 0x11, 0x4a, 0xa7, 0xfc, 0x1e, 0x45, 0xa8, 0xf3, 0x1b, 0x40, 0xad, 0xf6,
38     0x28, 0x73, 0x9e, 0xc5, 0x2d, 0x76, 0x9b, 0xc0, 0x22, 0x79, 0x94, 0xcf, 0x27, 0x7c, 0x91, 0xca,
39     0x3c, 0x67, 0x8a, 0xd1, 0x39, 0x62, 0x8f, 0xd4, 0x36, 0x6d, 0x80, 0xdb, 0x33, 0x68, 0x85, 0xde,
40     0x50, 0x0b, 0xe6, 0xbd, 0x55, 0x0e, 0xe3, 0xb8, 0x5a, 0x01, 0xec, 0xb7, 0x5f, 0x04, 0xe9, 0xb2,
41     0x44, 0x1f, 0xf2, 0xa9, 0x41, 0x1a, 0xf7, 0xac, 0x4e, 0x15, 0xf8, 0xa3, 0x4b, 0x10, 0xfd, 0xa6,
42     0x78, 0x23, 0xce, 0x95, 0x7d, 0x26, 0xcb, 0x90, 0x72, 0x29, 0xc4, 0x9f, 0x77, 0x2c, 0xc1, 0x9a,
43     0x6c, 0x37, 0xda, 0x81, 0x69, 0x32, 0xdf, 0x84, 0x66, 0x3d, 0xd0, 0x8b, 0x63, 0x38, 0xd5, 0x8e,
44     0xa0, 0xfb, 0x16, 0x4d, 0xa5, 0xfe, 0x13, 0x48, 0xaa, 0xf1, 0x1c, 0x47, 0xaf, 0xf4, 0x19, 0x42,
45     0xb4, 0xef, 0x02, 0x59, 0xb1, 0xea, 0x07, 0x5c, 0xbe, 0xe5, 0x08, 0x53, 0xbb, 0xe0, 0x0d, 0x56,
46     0x88, 0xd3, 0x3e, 0x65, 0x8d, 0xd6, 0x3b, 0x60, 0x82, 0xd9, 0x34, 0x6f, 0x87, 0xdc, 0x31, 0x6a,
47     0x9c, 0xc7, 0x2a, 0x71, 0x99, 0xc2, 0x2f, 0x74, 0x96, 0xcd, 0x20, 0x7b, 0x93, 0xc8, 0x25, 0x7e,
48     0xf0, 0xab, 0x46, 0x1d, 0xf5, 0xae, 0x43, 0x18, 0xfa, 0xa1, 0x4c, 0x17, 0xff, 0xa4, 0x49, 0x12,
49     0xe4, 0xbf, 0x52, 0x09, 0xe1, 0xba, 0x57, 0x0c, 0xee, 0xb5, 0x58, 0x03, 0xeb, 0xb0, 0x5d, 0x06,
50     0xd8, 0x83, 0x6e, 0x35, 0xdd, 0x86, 0x6b, 0x30, 0xd2, 0x89, 0x64, 0x3f, 0xd7, 0x8c, 0x61, 0x3a,
51     0xcc, 0x97, 0x7a, 0x21, 0xc9, 0x92, 0x7f, 0x24, 0xc6, 0x9d, 0x70, 0x2b, 0xc3, 0x98, 0x75, 0x2e
52 };
53
54 static const uint8_t MD2[256] = {
55     0x00, 0xef, 0xb7, 0x58, 0x07, 0xe8, 0xb0, 0x5f, 0x0e, 0xe1, 0xb9, 0x56, 0x09, 0xe6, 0xbe, 0x51,
56     0x1c, 0xf3, 0xab, 0x44, 0x1b, 0xf4, 0xac, 0x43, 0x12, 0xfd, 0xa5, 0x4a, 0x15, 0xfa, 0xa2, 0x4d,
57     0x38, 0xd7, 0x8f, 0x60, 0x3f, 0xd0, 0x88, 0x67, 0x36, 0xd9, 0x81, 0x6e, 0x31, 0xde, 0x86, 0x69,
58     0x24, 0xcb, 0x93, 0x7c, 0x23, 0xcc, 0x94, 0x7b, 0x2a, 0xc5, 0x9d, 0x72, 0x2d, 0xc2, 0x9a, 0x75,
59     0x70, 0x9f, 0xc7, 0x28, 0x77, 0x98, 0xc0, 0x2f, 0x7e, 0x91, 0xc9, 0x26, 0x79, 0x96, 0xce, 0x21,
60     0x6c, 0x83, 0xdb, 0x34, 0x6b, 0x84, 0xdc, 0x33, 0x62, 0x8d, 0xd5, 0x3a, 0x65, 0x8a, 0xd2, 0x3d,
61     0x48, 0xa7, 0xff, 0x10, 0x4f, 0xa0, 0xf8, 0x17, 0x46, 0xa9, 0xf1, 0x1e, 0x41, 0xae, 0xf6, 0x19,
62     0x54, 0xbb, 0xe3, 0x0c, 0x53, 0xbc, 0xe4, 0x0b, 0x5a, 0xb5, 0xed, 0x02, 0x5d, 0xb2, 0xea, 0x05,
63     0xe0, 0x0f, 0x57, 0xb8, 0xe7, 0x08, 0x50, 0xbf, 0xee, 0x01, 0x59, 0xb6, 0xe9, 0x06, 0x5e, 0xb1,
64     0xfc, 0x13, 0x4b, 0xa4, 0xfb, 0x14, 0x4c, 0xa3, 0xf2, 0x1d, 0x45, 0xaa, 0xf5, 0x1a, 0x42, 0xad,
65     0xd8, 0x37, 0x6f, 0x80, 0xdf, 0x30, 0x68, 0x87, 0xd6, 0x39, 0x61, 0x8e, 0xd1, 0x3e, 0x66, 0x89,
66     0xc4, 0x2b, 0x73, 0x9c, 0xc3, 0x2c, 0x74, 0x9b, 0xca, 0x25, 0x7d, 0x92, 0xcd, 0x22, 0x7a, 0x95,
67     0x90, 0x7f, 0x27, 0xc8, 0x97, 0x78, 0x20, 0xcf, 0x9e, 0x71, 0x29, 0xc6, 0x99, 0x76, 0x2e, 0xc1,
68     0x8c, 0x63, 0x3b, 0xd4, 0x8b, 0x64, 0x3c, 0xd3, 0x82, 0x6d, 0x35, 0xda, 0x85, 0x6a, 0x32, 0xdd,
69     0xa8, 0x47, 0x1f, 0xf0, 0xaf, 0x40, 0x18, 0xf7, 0xa6, 0x49, 0x11, 0xfe, 0xa1, 0x4e, 0x16, 0xf9,
70     0xb4, 0x5b, 0x03, 0xec, 0xb3, 0x5c, 0x04, 0xeb, 0xba, 0x55, 0x0d, 0xe2, 0xbd, 0x52, 0x0a, 0xe5
71 };
72
73 static const uint8_t q0[256] = {
74     0xa9, 0x67, 0xb3, 0xe8, 0x04, 0xfd, 0xa3, 0x76, 0x9a, 0x92, 0x80, 0x78, 0xe4, 0xdd, 0xd1, 0x38,
75     0x0d, 0xc6, 0x35, 0x98, 0x18, 0xf7, 0xec, 0x6c, 0x43, 0x75, 0x37, 0x26, 0xfa, 0x13, 0x94, 0x48,
76     0xf2, 0xd0, 0x8b, 0x30, 0x84, 0x54, 0xdf, 0x23, 0x19, 0x5b, 0x3d, 0x59, 0xf3, 0xae, 0xa2, 0x82,
77     0x63, 0x01, 0x83, 0x2e, 0xd9, 0x51, 0x9b, 0x7c, 0xa6, 0xeb, 0xa5, 0xbe, 0x16, 0x0c, 0xe3, 0x61,
78     0xc0, 0x8c, 0x3a, 0xf5, 0x73, 0x2c, 0x25, 0x0b, 0xbb, 0x4e, 0x89, 0x6b, 0x53, 0x6a, 0xb4, 0xf1,
79     0xe1, 0xe6, 0xbd, 0x45, 0xe2, 0xf4, 0xb6, 0x66, 0xcc, 0x95, 0x03, 0x56, 0xd4, 0x1c, 0x1e, 0xd7,
80     0xfb, 0xc3, 0x8e, 0xb5, 0xe9, 0xcf, 0xbf, 0xba, 0xea, 0x77, 0x39, 0xaf, 0x33, 0xc9, 0x62, 0x71,
81     0x81, 0x79, 0x09, 0xad, 0x24, 0xcd, 0xf9, 0xd8, 0xe5, 0xc5, 0xb9, 0x4d, 0x44, 0x08, 0x86, 0xe7,
82     0xa1, 0x1d, 0xaa, 0xed, 0x06, 0x70, 0xb2, 0xd2, 0x41, 0x7b, 0xa0, 0x11, 0x31, 0xc2, 0x27, 0x90,
83     0x20, 0xf6, 0x60, 0xff, 0x96, 0x5c, 0xb1, 0xab, 0x9e, 0x9c, 0x52, 0x1b, 0x5f, 0x93, 0x0a, 0xef,
84     0x91, 0x85, 0x49, 0xee, 0x2d, 0x4f, 0x8f, 0x3b, 0x47, 0x87, 0x6d, 0x46, 0xd6, 0x3e, 0x69, 0x64,
85     0x2a, 0xce, 0xcb, 0x2f, 0xfc, 0x97, 0x05, 0x7a, 0xac, 0x7f, 0xd5, 0x1a, 0x4b, 0x0e, 0xa7, 0x5a,
86     0x28, 0x14, 0x3f, 0x29, 0x88, 0x3c, 0x4c, 0x02, 0xb8, 0xda, 0xb0, 0x17, 0x55, 0x1f, 0x8a, 0x7d,
87     0x57, 0xc7, 0x8d, 0x74, 0xb7, 0xc4, 0x9f, 0x72, 0x7e, 0x15, 0x22, 0x12, 0x58, 0x07, 0x99, 0x34,
88     0x6e, 0x50, 0xde, 0x68, 0x65, 0xbc, 0xdb, 0xf8, 0xc8, 0xa8, 0x2b, 0x40, 0xdc, 0xfe, 0x32, 0xa4,
89     0xca, 0x10, 0x21, 0xf0, 0xd3, 0x5d, 0x0f, 0x00, 0x6f, 0x9d, 0x36, 0x42, 0x4a, 0x5e, 0xc1, 0xe0
90 };
91
92 static const uint8_t q1[256] = {
93     0x75, 0xf3, 0xc6, 0xf4, 0xdb, 0x7b, 0xfb, 0xc8, 0x4a, 0xd3, 0xe6, 0x6b, 0x45, 0x7d, 0xe8, 0x4b,
94     0xd6, 0x32, 0xd8, 0xfd, 0x37, 0x71, 0xf1, 0xe1, 0x30, 0x0f, 0xf8, 0x1b, 0x87, 0xfa, 0x06, 0x3f,
95     0x5e, 0xba, 0xae, 0x5b, 0x8a, 0x00, 0xbc, 0x9d, 0x6d, 0xc1, 0xb1, 0x0e, 0x80, 0x5d, 0xd2, 0xd5,
96     0xa0, 0x84, 0x07, 0x14, 0xb5, 0x90, 0x2c, 0xa3, 0xb2, 0x73, 0x4c, 0x54, 0x92, 0x74, 0x36, 0x51,
97     0x38, 0xb0, 0xbd, 0x5a, 0xfc, 0x60, 0x62, 0x96, 0x6c, 0x42, 0xf7, 0x10, 0x7c, 0x28, 0x27, 0x8c,
98     0x13, 0x95, 0x9c, 0xc7, 0x24, 0x46, 0x3b, 0x70, 0xca, 0xe3, 0x85, 0xcb, 0x11, 0xd0, 0x93, 0xb8,
99     0xa6, 0x83, 0x20, 0xff, 0x9f, 0x77, 0xc3, 0xcc, 0x03, 0x6f, 0x08, 0xbf, 0x40, 0xe7, 0x2b, 0xe2,
100     0x79, 0x0c, 0xaa, 0x82, 0x41, 0x3a, 0xea, 0xb9, 0xe4, 0x9a, 0xa4, 0x97, 0x7e, 0xda, 0x7a, 0x17,
101     0x66, 0x94, 0xa1, 0x1d, 0x3d, 0xf0, 0xde, 0xb3, 0x0b, 0x72, 0xa7, 0x1c, 0xef, 0xd1, 0x53, 0x3e,
102     0x8f, 0x33, 0x26, 0x5f, 0xec, 0x76, 0x2a, 0x49, 0x81, 0x88, 0xee, 0x21, 0xc4, 0x1a, 0xeb, 0xd9,
103     0xc5, 0x39, 0x99, 0xcd, 0xad, 0x31, 0x8b, 0x01, 0x18, 0x23, 0xdd, 0x1f, 0x4e, 0x2d, 0xf9, 0x48,
104     0x4f, 0xf2, 0x65, 0x8e, 0x78, 0x5c, 0x58, 0x19, 0x8d, 0xe5, 0x98, 0x57, 0x67, 0x7f, 0x05, 0x64,
105     0xaf, 0x63, 0xb6, 0xfe, 0xf5, 0xb7, 0x3c, 0xa5, 0xce, 0xe9, 0x68, 0x44, 0xe0, 0x4d, 0x43, 0x69,
106     0x29, 0x2e, 0xac, 0x15, 0x59, 0xa8, 0x0a, 0x9e, 0x6e, 0x47, 0xdf, 0x34, 0x35, 0x6a, 0xcf, 0xdc,
107     0x22, 0xc9, 0xc0, 0x9b, 0x89, 0xd4, 0xed, 0xab, 0x12, 0xa2, 0x0d, 0x52, 0xbb, 0x02, 0x2f, 0xa9,
108     0xd7, 0x61, 0x1e, 0xb4, 0x50, 0x04, 0xf6, 0xc2, 0x16, 0x25, 0x86, 0x56, 0x55, 0x09, 0xbe, 0x91
109 };
110
111 struct AVTWOFISH *av_twofish_alloc(void)
112 {
113     return av_mallocz(sizeof(struct AVTWOFISH));
114 }
115
116 const int av_twofish_size = sizeof(AVTWOFISH);
117
118 static uint8_t gfmul(uint8_t a, uint8_t b)
119 {
120     uint8_t r = 0, t;
121     while (a && b) {
122         if (a & 1)
123             r = r ^ b;
124         t = b & 0x80;
125         b = b << 1;
126         if (t)
127             b = b ^ 0x4d;
128         a = a >> 1;
129     }
130     return r;
131 }
132
133 static uint32_t tf_RS(uint32_t k0, uint32_t k1)
134 {
135     uint8_t s[4], m[8];
136     AV_WL32(m, k0);
137     AV_WL32(m + 4, k1);
138     s[0] = gfmul(0x01, m[0]) ^ gfmul(0xa4, m[1]) ^ gfmul(0x55, m[2]) ^ gfmul(0x87, m[3]) ^ gfmul(0x5a, m[4]) ^ gfmul(0x58, m[5]) ^ gfmul(0xdb, m[6]) ^ gfmul(0x9e, m[7]);
139     s[1] = gfmul(0xa4, m[0]) ^ gfmul(0x56, m[1]) ^ gfmul(0x82, m[2]) ^ gfmul(0xf3, m[3]) ^ gfmul(0x1e, m[4]) ^ gfmul(0xc6, m[5]) ^ gfmul(0x68, m[6]) ^ gfmul(0xe5, m[7]);
140     s[2] = gfmul(0x02, m[0]) ^ gfmul(0xa1, m[1]) ^ gfmul(0xfc, m[2]) ^ gfmul(0xc1, m[3]) ^ gfmul(0x47, m[4]) ^ gfmul(0xae, m[5]) ^ gfmul(0x3d, m[6]) ^ gfmul(0x19, m[7]);
141     s[3] = gfmul(0xa4, m[0]) ^ gfmul(0x55, m[1]) ^ gfmul(0x87, m[2]) ^ gfmul(0x5a, m[3]) ^ gfmul(0x58, m[4]) ^ gfmul(0xdb, m[5]) ^ gfmul(0x9e, m[6]) ^ gfmul(0x03, m[7]);
142     return AV_RL32(s);
143 }
144
145 static uint32_t tf_h(uint32_t X, uint32_t L[4], int k)
146 {
147     uint8_t y[4], l[4];
148     AV_WL32(y, X);
149     if (k == 4) {
150         AV_WL32(l, L[3]);
151         y[0] = q1[y[0]] ^ l[0];
152         y[1] = q0[y[1]] ^ l[1];
153         y[2] = q0[y[2]] ^ l[2];
154         y[3] = q1[y[3]] ^ l[3];
155     }
156     if (k >= 3) {
157         AV_WL32(l, L[2]);
158         y[0] = q1[y[0]] ^ l[0];
159         y[1] = q1[y[1]] ^ l[1];
160         y[2] = q0[y[2]] ^ l[2];
161         y[3] = q0[y[3]] ^ l[3];
162     }
163     AV_WL32(l, L[1]);
164     y[0] = q1[q0[q0[y[0]] ^ l[0]] ^ (L[0] & 0xff)];
165     y[1] = q0[q0[q1[y[1]] ^ l[1]] ^ ((L[0] >> 8) & 0xff)];
166     y[2] = q1[q1[q0[y[2]] ^ l[2]] ^ ((L[0] >> 16) & 0xff)];
167     y[3] = q0[q1[q1[y[3]] ^ l[3]] ^ (L[0] >> 24)];
168
169     l[0] = y[0] ^ MD2[y[1]] ^ MD1[y[2]] ^ MD1[y[3]];
170     l[1] = MD1[y[0]] ^ MD2[y[1]] ^ MD2[y[2]] ^ y[3];
171     l[2] = MD2[y[0]] ^ MD1[y[1]] ^ y[2] ^ MD2[y[3]];
172     l[3] = MD2[y[0]] ^ y[1] ^ MD2[y[2]] ^ MD1[y[3]];
173
174     return AV_RL32(l);
175 }
176
177 static void twofish_encrypt(AVTWOFISH *cs, uint8_t *dst, const uint8_t *src)
178 {
179     uint32_t P[4], t0, t1;
180     int i;
181     P[0] = AV_RL32(src) ^ cs->K[0];
182     P[1] = AV_RL32(src + 4) ^ cs->K[1];
183     P[2] = AV_RL32(src + 8) ^ cs->K[2];
184     P[3] = AV_RL32(src + 12) ^ cs->K[3];
185     for (i = 0; i < 16; i += 2) {
186         t0 = tf_h(P[0], cs->S, cs->ksize);
187         t1 = tf_h(LR(P[1], 8), cs->S, cs->ksize);
188         P[2] = RR(P[2] ^ (t0 + t1 + cs->K[2 * i + 8]), 1);
189         P[3] = LR(P[3], 1) ^ (t0 + 2 * t1 + cs->K[2 * i + 9]);
190         t0 = tf_h(P[2], cs->S, cs->ksize);
191         t1 = tf_h(LR(P[3], 8), cs->S, cs->ksize);
192         P[0] = RR(P[0] ^ (t0 + t1 + cs->K[2 * i + 10]), 1);
193         P[1] = LR(P[1], 1) ^ (t0 + 2 * t1 + cs->K[2 * i + 11]);
194     }
195     P[2] ^= cs->K[4];
196     P[3] ^= cs->K[5];
197     P[0] ^= cs->K[6];
198     P[1] ^= cs->K[7];
199     AV_WL32(dst, P[2]);
200     AV_WL32(dst + 4, P[3]);
201     AV_WL32(dst + 8, P[0]);
202     AV_WL32(dst + 12, P[1]);
203 }
204
205 static void twofish_decrypt(AVTWOFISH *cs, uint8_t *dst, const uint8_t *src, uint8_t *iv)
206 {
207     uint32_t P[4], t0, t1;
208     int i;
209     P[2] = AV_RL32(src) ^ cs->K[4];
210     P[3] = AV_RL32(src + 4) ^ cs->K[5];
211     P[0] = AV_RL32(src + 8) ^ cs->K[6];
212     P[1] = AV_RL32(src + 12) ^ cs->K[7];
213     for (i = 15; i >= 0; i -= 2) {
214         t0 = tf_h(P[2], cs->S, cs->ksize);
215         t1 = tf_h(LR(P[3], 8), cs->S, cs->ksize);
216         P[0] = LR(P[0], 1) ^ (t0 + t1 + cs->K[2 * i + 8]);
217         P[1] = RR(P[1] ^ (t0 + 2 * t1 + cs->K[2 * i + 9]), 1);
218         t0 = tf_h(P[0], cs->S, cs->ksize);
219         t1 = tf_h(LR(P[1], 8), cs->S, cs->ksize);
220         P[2] = LR(P[2], 1) ^ (t0 + t1 + cs->K[2 * i + 6]);
221         P[3] = RR(P[3] ^ (t0 + 2 * t1 + cs->K[2 * i + 7]), 1);
222     }
223     P[0] ^= cs->K[0];
224     P[1] ^= cs->K[1];
225     P[2] ^= cs->K[2];
226     P[3] ^= cs->K[3];
227     if (iv) {
228         P[0] ^= AV_RL32(iv);
229         P[1] ^= AV_RL32(iv + 4);
230         P[2] ^= AV_RL32(iv + 8);
231         P[3] ^= AV_RL32(iv + 12);
232         memcpy(iv, src, 16);
233     }
234     AV_WL32(dst, P[2]);
235     AV_WL32(dst + 4, P[3]);
236     AV_WL32(dst + 8, P[0]);
237     AV_WL32(dst + 12, P[1]);
238 }
239
240 av_cold int av_twofish_init(AVTWOFISH *cs, const uint8_t *key, int key_bits)
241 {
242     int i;
243     uint8_t keypad[32];
244     uint32_t Key[8], Me[4], Mo[4], A, B;
245     const uint32_t rho = 0x01010101;
246     if (key_bits < 0)
247         return -1;
248     if (key_bits <= 128) {
249         cs->ksize = 2;
250     } else if (key_bits <= 192) {
251         cs->ksize = 3;
252     } else {
253         cs->ksize = 4;
254     }
255     memset(keypad, 0, sizeof(keypad));
256     if (key_bits <= 256) {
257         memcpy(keypad, key, key_bits >> 3);
258     } else {
259         memcpy(keypad, key, 32);
260     }
261     for (i = 0; i < 2 * cs->ksize ; i++)
262         Key[i] = AV_RL32(keypad + 4 * i);
263     for (i = 0; i < cs->ksize; i++) {
264         Me[i] = Key[2 * i];
265         Mo[i] = Key[2 * i + 1];
266         cs->S[cs->ksize - i - 1] = tf_RS(Me[i], Mo[i]);
267     }
268     for (i = 0; i < 20; i++) {
269         A = tf_h((2 * i) * rho, Me, cs->ksize);
270         B = tf_h((2 * i + 1) * rho, Mo, cs->ksize);
271         B = LR(B, 8);
272         cs->K[2 * i] = A + B;
273         cs->K[2 * i + 1] = LR((A + (2 * B)), 9);
274     }
275     if (cs->ksize << 6 != key_bits) {
276         return 1;
277     } else {
278         return 0;
279     }
280 }
281
282 void av_twofish_crypt(AVTWOFISH *cs, uint8_t *dst, const uint8_t *src, int count, uint8_t *iv, int decrypt)
283 {
284     int i;
285     while (count--) {
286         if (decrypt) {
287             twofish_decrypt(cs, dst, src, iv);
288         } else {
289             if (iv) {
290                 for (i = 0; i < 16; i++)
291                     dst[i] = src[i] ^ iv[i];
292                 twofish_encrypt(cs, dst, dst);
293                 memcpy(iv, dst, 16);
294             } else {
295                 twofish_encrypt(cs, dst, src);
296             }
297         }
298         src = src + 16;
299         dst = dst + 16;
300     }
301 }
302
303 #ifdef TEST
304 #include<stdio.h>
305 #include<stdlib.h>
306 #include"log.h"
307
308 int main(int argc, char *argv[])
309 {
310     uint8_t Key[32] = {0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0xfe, 0xdc, 0xba, 0x98, 0x76, 0x54, 0x32, 0x10, 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff
311     };
312     const uint8_t rct[6][16] = {
313         {0x9f, 0x58, 0x9f, 0x5c, 0xf6, 0x12, 0x2c, 0x32, 0xb6, 0xbf, 0xec, 0x2f, 0x2a, 0xe8, 0xc3, 0x5a},
314         {0xcf, 0xd1, 0xd2, 0xe5, 0xa9, 0xbe, 0x9c, 0xdf, 0x50, 0x1f, 0x13, 0xb8, 0x92, 0xbd, 0x22, 0x48},
315         {0x37, 0x52, 0x7b, 0xe0, 0x05, 0x23, 0x34, 0xb8, 0x9f, 0x0c, 0xfc, 0xca, 0xe8, 0x7c, 0xfa, 0x20},
316         {0x5d, 0x9d, 0x4e, 0xef, 0xfa, 0x91, 0x51, 0x57, 0x55, 0x24, 0xf1, 0x15, 0x81, 0x5a, 0x12, 0xe0},
317         {0xe7, 0x54, 0x49, 0x21, 0x2b, 0xee, 0xf9, 0xf4, 0xa3, 0x90, 0xbd, 0x86, 0x0a, 0x64, 0x09, 0x41},
318         {0x37, 0xfe, 0x26, 0xff, 0x1c, 0xf6, 0x61, 0x75, 0xf5, 0xdd, 0xf4, 0xc3, 0x3b, 0x97, 0xa2, 0x05}
319     };
320     uint8_t temp[32], iv[16], rpt[32] = {0};
321     const int kbits[3] = {128, 192, 256};
322     int i, j, err = 0;
323     AVTWOFISH *cs;
324     cs = av_twofish_alloc();
325     if (!cs)
326         return 1;
327     for (j = 1; j < 3; j++) {
328         av_twofish_init(cs, Key, kbits[j]);
329         av_twofish_crypt(cs, temp, rpt, 1, NULL, 0);
330         for (i = 0; i < 16; i++) {
331             if (rct[j][i] != temp[i]) {
332                 av_log(NULL, AV_LOG_ERROR, "%d %02x %02x\n", i, rct[j][i], temp[i]);
333                 err = 1;
334             }
335         }
336         av_twofish_crypt(cs, temp, rct[j], 1, NULL, 1);
337         for (i = 0; i < 16; i++) {
338             if (rpt[i] != temp[i]) {
339                 av_log(NULL, AV_LOG_ERROR, "%d %02x %02x\n", i, rpt[i], temp[i]);
340                 err = 1;
341             }
342         }
343     }
344     for (j = 0; j < 3; j++) {
345         memset(Key, 0, sizeof(Key));
346         memset(rpt, 0, sizeof(rpt));
347         for (i = 1; i < 50; i++) {
348             av_twofish_init(cs, Key, kbits[j]);
349             av_twofish_crypt(cs, temp, rpt, 1, NULL, 0);
350             memcpy(Key+16,Key,(kbits[j]-128) >> 3);
351             memcpy(Key,rpt,16);
352             memcpy(rpt,temp,16);
353         }
354         for (i = 0; i < 16; i++) {
355             if (rct[3 + j][i] != temp[i]) {
356                 av_log(NULL, AV_LOG_ERROR, "%d %02x %02x\n", i, rct[3 + j][i], temp[i]);
357                 err = 1;
358             }
359         }
360     }
361     memset(rpt, 0, sizeof(rpt));
362     memcpy(iv, "HALLO123HALLO123", 16);
363     av_twofish_crypt(cs, temp, rpt, 2, iv, 0);
364     memcpy(iv, "HALLO123HALLO123", 16);
365     av_twofish_crypt(cs, temp, temp, 2, iv, 1);
366     for (i = 0; i < 32; i++) {
367         if (rpt[i] != temp[i]) {
368             av_log(NULL, AV_LOG_ERROR, "%d %02x %02x\n", i, rpt[i], temp[i]);
369             err = 1;
370         }
371     }
372     av_free(cs);
373     return err;
374 }
375 #endif