FFmpeg
xtea.c
Go to the documentation of this file.
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 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  crypt(ctx, dst, src, len, iv, dir);
69  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 }
81 
82 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];
90  if (!ctx)
91  return 1;
92 
93  for (i = 0; i < XTEA_NUM_TESTS; i++) {
95 
96  test_xtea(ctx, buf, xtea_test_pt[i], xtea_test_ct[i], 1, NULL, 0, "encryption", av_xtea_crypt);
97  test_xtea(ctx, buf, xtea_test_ct[i], xtea_test_pt[i], 1, NULL, 1, "decryption", av_xtea_crypt);
98 
99  for (j = 0; j < 4; j++)
100  AV_WL32(&buf[4*j], AV_RB32(&xtea_test_key[i][4*j]));
101  av_xtea_le_init(ctx, buf);
102  for (j = 0; j < 2; j++) {
103  AV_WL32(&ct[4*j], AV_RB32(&xtea_test_ct[i][4*j]));
104  AV_WL32(&pl[4*j], AV_RB32(&xtea_test_pt[i][4*j]));
105  }
106  test_xtea(ctx, buf, pl, ct, 1, NULL, 0, "encryption", av_xtea_le_crypt);
107  test_xtea(ctx, buf, ct, pl, 1, NULL, 1, "decryption", av_xtea_le_crypt);
108 
109  /* encrypt */
110  memcpy(iv, "HALLO123", 8);
111  av_xtea_crypt(ctx, ct, src, 4, iv, 0);
112 
113  /* decrypt into pl */
114  memcpy(iv, "HALLO123", 8);
115  test_xtea(ctx, pl, ct, src, 4, iv, 1, "CBC decryption", av_xtea_crypt);
116 
117  memcpy(iv, "HALLO123", 8);
118  test_xtea(ctx, ct, ct, src, 4, iv, 1, "CBC inplace decryption", av_xtea_crypt);
119  }
120 
121  printf("Test encryption/decryption success.\n");
122  av_free(ctx);
123 
124  return 0;
125 }
AV_WL32
#define AV_WL32(p, v)
Definition: intreadwrite.h:422
test
Definition: idctdsp.c:35
av_xtea_crypt
void av_xtea_crypt(AVXTEA *ctx, uint8_t *dst, const uint8_t *src, int count, uint8_t *iv, int decrypt)
Encrypt or decrypt a buffer using a previously initialized context, in big endian format.
Definition: xtea.c:243
XTEA_NUM_TESTS
#define XTEA_NUM_TESTS
Definition: xtea.c:28
av_xtea_init
void av_xtea_init(AVXTEA *ctx, const uint8_t key[16])
Initialize an AVXTEA context.
Definition: xtea.c:42
intreadwrite.h
ctx
AVFormatContext * ctx
Definition: movenc.c:49
NULL
#define NULL
Definition: coverity.c:32
dst
uint8_t ptrdiff_t const uint8_t ptrdiff_t int intptr_t intptr_t int int16_t * dst
Definition: dsp.h:83
AV_RB32
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_WB64 unsigned int_TMPL AV_RB32
Definition: bytestream.h:96
printf
printf("static const uint8_t my_array[100] = {\n")
av_xtea_le_init
void av_xtea_le_init(AVXTEA *ctx, const uint8_t key[16])
Initialize an AVXTEA context.
Definition: xtea.c:50
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
xtea_test_pt
static const uint8_t xtea_test_pt[XTEA_NUM_TESTS][8]
Definition: xtea.c:45
AVXTEA
Definition: xtea.h:35
xtea_test_key
static const uint8_t xtea_test_key[XTEA_NUM_TESTS][16]
Definition: xtea.c:30
av_xtea_alloc
AVXTEA * av_xtea_alloc(void)
Allocate an AVXTEA context.
Definition: xtea.c:37
len
int len
Definition: vorbis_enc_data.h:426
main
int main(void)
Definition: xtea.c:82
ref
static int ref[MAX_W *MAX_W]
Definition: jpeg2000dwt.c:112
mem.h
av_xtea_le_crypt
void av_xtea_le_crypt(AVXTEA *ctx, uint8_t *dst, const uint8_t *src, int count, uint8_t *iv, int decrypt)
Encrypt or decrypt a buffer using a previously initialized context, in little endian format.
Definition: xtea.c:249
xtea_test_ct
static const uint8_t xtea_test_ct[XTEA_NUM_TESTS][8]
Definition: xtea.c:54
xtea.h
Public header for libavutil XTEA algorithm.
av_free
#define av_free(p)
Definition: tableprint_vlc.h:33
test_xtea
static void test_xtea(AVXTEA *ctx, uint8_t *dst, const uint8_t *src, const uint8_t *ref, int len, uint8_t *iv, int dir, const char *test, void(*crypt)(AVXTEA *, uint8_t *, const uint8_t *, int, uint8_t *, int))
Definition: xtea.c:63
src
#define src
Definition: vp8dsp.c:248