mirror of
https://git.ffmpeg.org/ffmpeg.git
synced 2026-04-18 00:20:21 +00:00
117897bcd0
Test the three public API functions: av_rc4_alloc, av_rc4_init, and av_rc4_crypt. Verifies keystream output against RFC 6229 test vectors for 40, 56, 64, and 128-bit keys, encrypt/decrypt round-trip, inplace operation, and the invalid key_bits error path. Coverage for libavutil/rc4.c: 0.00% -> 100.00%
123 lines
4 KiB
C
123 lines
4 KiB
C
/*
|
|
* This file is part of FFmpeg.
|
|
*
|
|
* FFmpeg is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with FFmpeg; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include "libavutil/attributes_internal.h"
|
|
#include "libavutil/macros.h"
|
|
#include "libavutil/mem.h"
|
|
#include "libavutil/rc4.h"
|
|
|
|
/* RFC 6229 test vectors */
|
|
static const struct {
|
|
int key_bits;
|
|
const uint8_t key[16];
|
|
const uint8_t keystream[8]; /* first 8 bytes of output */
|
|
} test_vectors[] = {
|
|
/* 40-bit key: 0x0102030405 */
|
|
{ 40,
|
|
{ 0x01, 0x02, 0x03, 0x04, 0x05 },
|
|
{ 0xb2, 0x39, 0x63, 0x05, 0xf0, 0x3d, 0xc0, 0x27 } },
|
|
/* 56-bit key: 0x01020304050607 */
|
|
{ 56,
|
|
{ 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07 },
|
|
{ 0x29, 0x3f, 0x02, 0xd4, 0x7f, 0x37, 0xc9, 0xb6 } },
|
|
/* 64-bit key: 0x0102030405060708 */
|
|
{ 64,
|
|
{ 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 },
|
|
{ 0x97, 0xab, 0x8a, 0x1b, 0xf0, 0xaf, 0xb9, 0x61 } },
|
|
/* 128-bit key: 0x0102030405060708090a0b0c0d0e0f10 */
|
|
{ 128,
|
|
{ 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
|
|
0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10 },
|
|
{ 0x9a, 0xc7, 0xcc, 0x9a, 0x60, 0x9d, 0x1e, 0xf7 } },
|
|
};
|
|
|
|
int main(void)
|
|
{
|
|
AVRC4 *ctx;
|
|
uint8_t buf[8], encrypted[8], decrypted[8];
|
|
|
|
ctx = av_rc4_alloc();
|
|
if (!ctx)
|
|
return 1;
|
|
|
|
/* test keystream output (src=NULL) against known vectors */
|
|
for (int i = 0; i < FF_ARRAY_ELEMS(test_vectors); i++) {
|
|
av_rc4_init(ctx, test_vectors[i].key, test_vectors[i].key_bits, 0);
|
|
av_rc4_crypt(ctx, buf, NULL, sizeof(buf), NULL, 0);
|
|
if (memcmp(buf, test_vectors[i].keystream, sizeof(buf))) {
|
|
printf("Keystream test %d (%d-bit key) failed.\n",
|
|
i, test_vectors[i].key_bits);
|
|
av_free(ctx);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
/* test encrypt then decrypt round-trip */
|
|
{
|
|
const uint8_t key[] = { 0x01, 0x02, 0x03, 0x04, 0x05 };
|
|
static attribute_nonstring const uint8_t plaintext[8] = "TestData";
|
|
|
|
av_rc4_init(ctx, key, 40, 0);
|
|
av_rc4_crypt(ctx, encrypted, plaintext, sizeof(plaintext), NULL, 0);
|
|
|
|
/* RC4 is symmetric: re-init and encrypt again to decrypt */
|
|
av_rc4_init(ctx, key, 40, 0);
|
|
av_rc4_crypt(ctx, decrypted, encrypted, sizeof(encrypted), NULL, 0);
|
|
|
|
if (memcmp(decrypted, plaintext, sizeof(plaintext))) {
|
|
printf("Round-trip test failed.\n");
|
|
av_free(ctx);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
/* test inplace encrypt/decrypt */
|
|
{
|
|
const uint8_t key[] = { 0xFE, 0xDC, 0xBA, 0x98, 0x76, 0x54, 0x32, 0x10 };
|
|
static attribute_nonstring const uint8_t plaintext[8] = "InPlace!";
|
|
|
|
memcpy(buf, plaintext, sizeof(plaintext));
|
|
av_rc4_init(ctx, key, 64, 0);
|
|
av_rc4_crypt(ctx, buf, buf, sizeof(buf), NULL, 0);
|
|
|
|
av_rc4_init(ctx, key, 64, 0);
|
|
av_rc4_crypt(ctx, buf, buf, sizeof(buf), NULL, 0);
|
|
|
|
if (memcmp(buf, plaintext, sizeof(plaintext))) {
|
|
printf("Inplace round-trip test failed.\n");
|
|
av_free(ctx);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
/* test invalid key_bits (not multiple of 8) */
|
|
if (av_rc4_init(ctx, (const uint8_t[]){ 0x01 }, 7, 0) >= 0) {
|
|
printf("Invalid key_bits should return error.\n");
|
|
av_free(ctx);
|
|
return 1;
|
|
}
|
|
|
|
printf("Test encryption/decryption success.\n");
|
|
av_free(ctx);
|
|
|
|
return 0;
|
|
}
|