Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * This file is part of FFmpeg. | ||
3 | * | ||
4 | * FFmpeg is free software; you can redistribute it and/or | ||
5 | * modify it under the terms of the GNU Lesser General Public | ||
6 | * License as published by the Free Software Foundation; either | ||
7 | * version 2.1 of the License, or (at your option) any later version. | ||
8 | * | ||
9 | * FFmpeg is distributed in the hope that it will be useful, | ||
10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
12 | * Lesser General Public License for more details. | ||
13 | * | ||
14 | * You should have received a copy of the GNU Lesser General Public | ||
15 | * License along with FFmpeg; if not, write to the Free Software | ||
16 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
17 | */ | ||
18 | |||
19 | #include <stdint.h> | ||
20 | #include <stdio.h> | ||
21 | #include <stdlib.h> | ||
22 | #include <string.h> | ||
23 | |||
24 | #include "libavutil/intreadwrite.h" | ||
25 | #include "libavutil/mem.h" | ||
26 | #include "libavutil/xtea.h" | ||
27 | |||
28 | #define XTEA_NUM_TESTS 6 | ||
29 | |||
30 | static const uint8_t xtea_test_key[XTEA_NUM_TESTS][16] = { | ||
31 | { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, | ||
32 | 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f }, | ||
33 | { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, | ||
34 | 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f }, | ||
35 | { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, | ||
36 | 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f }, | ||
37 | { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | ||
38 | 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | ||
39 | { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | ||
40 | 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | ||
41 | { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | ||
42 | 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 } | ||
43 | }; | ||
44 | |||
45 | static const uint8_t xtea_test_pt[XTEA_NUM_TESTS][8] = { | ||
46 | { 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48 }, | ||
47 | { 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 }, | ||
48 | { 0x5a, 0x5b, 0x6e, 0x27, 0x89, 0x48, 0xd7, 0x7f }, | ||
49 | { 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48 }, | ||
50 | { 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 }, | ||
51 | { 0x70, 0xe1, 0x22, 0x5d, 0x6e, 0x4e, 0x76, 0x55 } | ||
52 | }; | ||
53 | |||
54 | static const uint8_t xtea_test_ct[XTEA_NUM_TESTS][8] = { | ||
55 | { 0x49, 0x7d, 0xf3, 0xd0, 0x72, 0x61, 0x2c, 0xb5 }, | ||
56 | { 0xe7, 0x8f, 0x2d, 0x13, 0x74, 0x43, 0x41, 0xd8 }, | ||
57 | { 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 }, | ||
58 | { 0xa0, 0x39, 0x05, 0x89, 0xf8, 0xb8, 0xef, 0xa5 }, | ||
59 | { 0xed, 0x23, 0x37, 0x5a, 0x82, 0x1a, 0x8c, 0x2d }, | ||
60 | { 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 } | ||
61 | }; | ||
62 | |||
63 | 36 | static void test_xtea(AVXTEA *ctx, uint8_t *dst, const uint8_t *src, | |
64 | const uint8_t *ref, int len, uint8_t *iv, int dir, | ||
65 | const char *test, | ||
66 | void (*crypt)(AVXTEA *, uint8_t *, const uint8_t *, int, uint8_t *, int)) | ||
67 | { | ||
68 | 36 | crypt(ctx, dst, src, len, iv, dir); | |
69 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 36 times.
|
36 | if (memcmp(dst, ref, 8*len)) { |
70 | int i; | ||
71 | ✗ | printf("%s failed\ngot ", test); | |
72 | ✗ | for (i = 0; i < 8*len; i++) | |
73 | ✗ | printf("%02x ", dst[i]); | |
74 | ✗ | printf("\nexpected "); | |
75 | ✗ | for (i = 0; i < 8*len; i++) | |
76 | ✗ | printf("%02x ", ref[i]); | |
77 | ✗ | printf("\n"); | |
78 | ✗ | exit(1); | |
79 | } | ||
80 | 36 | } | |
81 | |||
82 | 1 | int main(void) | |
83 | { | ||
84 | uint8_t buf[16], iv[8]; | ||
85 | int i, j; | ||
86 | static const uint8_t src[32] = "HelloWorldHelloWorldHelloWorld"; | ||
87 | uint8_t ct[32]; | ||
88 | uint8_t pl[32]; | ||
89 | 1 | AVXTEA *ctx = av_xtea_alloc(); | |
90 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (!ctx) |
91 | ✗ | return 1; | |
92 | |||
93 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 1 times.
|
7 | for (i = 0; i < XTEA_NUM_TESTS; i++) { |
94 | 6 | av_xtea_init(ctx, xtea_test_key[i]); | |
95 | |||
96 | 6 | test_xtea(ctx, buf, xtea_test_pt[i], xtea_test_ct[i], 1, NULL, 0, "encryption", av_xtea_crypt); | |
97 | 6 | test_xtea(ctx, buf, xtea_test_ct[i], xtea_test_pt[i], 1, NULL, 1, "decryption", av_xtea_crypt); | |
98 | |||
99 |
2/2✓ Branch 0 taken 24 times.
✓ Branch 1 taken 6 times.
|
30 | for (j = 0; j < 4; j++) |
100 | 24 | AV_WL32(&buf[4*j], AV_RB32(&xtea_test_key[i][4*j])); | |
101 | 6 | av_xtea_le_init(ctx, buf); | |
102 |
2/2✓ Branch 0 taken 12 times.
✓ Branch 1 taken 6 times.
|
18 | for (j = 0; j < 2; j++) { |
103 | 12 | AV_WL32(&ct[4*j], AV_RB32(&xtea_test_ct[i][4*j])); | |
104 | 12 | AV_WL32(&pl[4*j], AV_RB32(&xtea_test_pt[i][4*j])); | |
105 | } | ||
106 | 6 | test_xtea(ctx, buf, pl, ct, 1, NULL, 0, "encryption", av_xtea_le_crypt); | |
107 | 6 | test_xtea(ctx, buf, ct, pl, 1, NULL, 1, "decryption", av_xtea_le_crypt); | |
108 | |||
109 | /* encrypt */ | ||
110 | 6 | memcpy(iv, "HALLO123", 8); | |
111 | 6 | av_xtea_crypt(ctx, ct, src, 4, iv, 0); | |
112 | |||
113 | /* decrypt into pl */ | ||
114 | 6 | memcpy(iv, "HALLO123", 8); | |
115 | 6 | test_xtea(ctx, pl, ct, src, 4, iv, 1, "CBC decryption", av_xtea_crypt); | |
116 | |||
117 | 6 | memcpy(iv, "HALLO123", 8); | |
118 | 6 | test_xtea(ctx, ct, ct, src, 4, iv, 1, "CBC inplace decryption", av_xtea_crypt); | |
119 | } | ||
120 | |||
121 | 1 | printf("Test encryption/decryption success.\n"); | |
122 | 1 | av_free(ctx); | |
123 | |||
124 | 1 | return 0; | |
125 | } | ||
126 |