Merge commit 'a6a750c7ef240b72ce01e9653343a0ddf247d196'
[ffmpeg.git] / libavutil / tests / cast5.c
1 /*
2  * An implementation of the CAST128 algorithm as mentioned in RFC2144
3  * Copyright (c) 2014 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
22 #include "libavutil/cast5.h"
23 #include "libavutil/log.h"
24
25 int main(int argc, char** argv)
26 {
27
28     static const uint8_t Key[3][16] = {
29         {0x01, 0x23, 0x45, 0x67, 0x12, 0x34, 0x56, 0x78, 0x23, 0x45, 0x67, 0x89, 0x34, 0x56, 0x78, 0x9a},
30         {0x01, 0x23, 0x45, 0x67, 0x12, 0x34, 0x56, 0x78, 0x23, 0x45},
31         {0x01, 0x23, 0x45, 0x67, 0x12}
32     };
33     static const uint8_t rpt[8] = {0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef};
34     static const uint8_t rct[3][8] = {
35         {0x23, 0x8b, 0x4f, 0xe5, 0x84, 0x7e, 0x44, 0xb2},
36         {0xeb, 0x6a, 0x71, 0x1a, 0x2c, 0x02, 0x27, 0x1b},
37         {0x7a, 0xc8, 0x16, 0xd1, 0x6e, 0x9b, 0x30, 0x2e}
38     };
39     static const uint8_t rct2[2][16] = {
40         {0xee, 0xa9, 0xd0, 0xa2, 0x49, 0xfd, 0x3b, 0xa6, 0xb3, 0x43, 0x6f, 0xb8, 0x9d, 0x6d, 0xca, 0x92},
41         {0xb2, 0xc9, 0x5e, 0xb0, 0x0c, 0x31, 0xad, 0x71, 0x80, 0xac, 0x05, 0xb8, 0xe8, 0x3d, 0x69, 0x6e}
42     };
43     static const uint8_t iv[8] = {0xee, 0xa9, 0xd0, 0xa2, 0x49, 0xfd, 0x3b, 0xa6};
44     static uint8_t rpt2[2][16];
45     int i, j, err = 0;
46     static const int key_bits[3] = {128, 80, 40};
47     uint8_t temp[8];
48     struct AVCAST5 *cs;
49     cs = av_cast5_alloc();
50     if (!cs)
51         return 1;
52     for (j = 0; j < 3; j++){
53
54         av_cast5_init(cs, Key[j], key_bits[j]);
55         av_cast5_crypt(cs, temp, rpt, 1, 0);
56         for (i = 0;i < 8; i++){
57             if (rct[j][i] != temp[i]){
58                 av_log(NULL, AV_LOG_ERROR, "%d %02x %02x\n", i, rct[j][i], temp[i]);
59                 err = 1;
60             }
61         }
62
63         av_cast5_crypt(cs, temp, rct[j], 1, 1);
64         for (i =0; i < 8; i++) {
65             if (rpt[i] != temp[i]) {
66                 av_log(NULL, AV_LOG_ERROR, "%d %02x %02x\n", i, rpt[i], temp[i]);
67                 err = 1;
68             }
69         }
70     }
71     memcpy(rpt2[0], Key[0], 16);
72     memcpy(rpt2[1], Key[0], 16);
73     for (i = 0; i < 1000000; i++){
74         av_cast5_init(cs, rpt2[1], 128);
75         av_cast5_crypt(cs, rpt2[0], rpt2[0], 2, 0);
76         av_cast5_init(cs, rpt2[0], 128);
77         av_cast5_crypt(cs, rpt2[1], rpt2[1], 2, 0);
78     }
79     for (j = 0; j < 2; j++) {
80         for (i = 0; i < 16; i++) {
81             if (rct2[j][i] != rpt2[j][i]) {
82                 av_log(NULL, AV_LOG_ERROR, "%d %02x %02x\n", i, rct2[j][i], rpt2[j][i]);
83                 err = 1;
84             }
85         }
86     }
87     for (j = 0; j < 3; j++) {
88
89         av_cast5_init(cs, Key[j], key_bits[j]);
90         memcpy(temp, iv, 8);
91         av_cast5_crypt2(cs, rpt2[0], rct2[0], 2, temp, 0);
92         memcpy(temp, iv, 8);
93         av_cast5_crypt2(cs, rpt2[0], rpt2[0], 2, temp, 1);
94         for (i = 0; i < 16; i++) {
95             if (rct2[0][i] != rpt2[0][i]) {
96                 av_log(NULL, AV_LOG_ERROR, "%d %02x %02x\n", i, rct2[0][i], rpt2[0][i]);
97                 err = 1;
98             }
99         }
100     }
101     av_free(cs);
102     return err;
103 }