[FFmpeg-cvslog] xtea-test: test inplace decryption

[Michael Niedermayer]

ffmpeg | branch: master | Michael Niedermayer <michaelni at gmx.at> | Sun Sep 30 17:36:01 2012 +0200| [a4dcdd04d7dc2fce0a0f683d361d6d4760a152f9] | committer: Michael Niedermayer

xtea-test: test inplace decryption

Based on test code by: Giorgio Vazzana <mywing81 at gmail.com>
Signed-off-by: Michael Niedermayer <michaelni at gmx.at>

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=a4dcdd04d7dc2fce0a0f683d361d6d4760a152f9
---

 libavutil/xtea.c |   16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/libavutil/xtea.c b/libavutil/xtea.c
index f2dfe99..c952042 100644
--- a/libavutil/xtea.c
+++ b/libavutil/xtea.c
@@ -231,8 +231,11 @@ static const uint8_t xtea_test_ct[XTEA_NUM_TESTS][8] = {
 int main(void)
 {
     AVXTEA ctx;
-    uint8_t buf[8];
+    uint8_t buf[8], iv[8];
     int i;
+    const uint8_t src[32] = "HelloWorldHelloWorldHelloWorld";
+    uint8_t  ct[32];
+    uint8_t  pl[32];
 
 #define CHECK(dst, src, ref, len, iv, dir, error) \
         av_xtea_crypt(&ctx, dst, src, len, iv, dir);\
@@ -247,7 +250,18 @@ int main(void)
         CHECK(buf, xtea_test_pt[i], xtea_test_ct[i], 1, NULL, 0, "Test encryption failed.\n");
         CHECK(buf, xtea_test_ct[i], xtea_test_pt[i], 1, NULL, 1, "Test decryption failed.\n");
 
+        /* encrypt */
+        memcpy(iv, "HALLO123", 8);
+        av_xtea_crypt(&ctx, ct, src, 4, iv, 0);
+
+        /* decrypt into pl */
+        memcpy(iv, "HALLO123", 8);
+        CHECK(pl, ct, src, 4, iv, 1, "Test IV decryption failed.\n");
+
+        memcpy(iv, "HALLO123", 8);
+        CHECK(ct, ct, src, 4, iv, 1, "Test IV inplace decryption failed.\n");
     }
+
     printf("Test encryption/decryption success.\n");
 
     return 0;