This avoids spurious library rebuilds when only the test program code is changed and simplifies the build system.tags/n3.1
@@ -0,0 +1,52 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <math.h> | |||
#include <stdint.h> | |||
#include <stdio.h> | |||
#include "iirfilter.h" | |||
#define FILT_ORDER 4 | |||
#define SIZE 1024 | |||
int main(void) | |||
{ | |||
struct FFIIRFilterCoeffs *fcoeffs = NULL; | |||
struct FFIIRFilterState *fstate = NULL; | |||
float cutoff_coeff = 0.4; | |||
int16_t x[SIZE], y[SIZE]; | |||
int i; | |||
fcoeffs = ff_iir_filter_init_coeffs(NULL, FF_FILTER_TYPE_BUTTERWORTH, | |||
FF_FILTER_MODE_LOWPASS, FILT_ORDER, | |||
cutoff_coeff, 0.0, 0.0); | |||
fstate = ff_iir_filter_init_state(FILT_ORDER); | |||
for (i = 0; i < SIZE; i++) | |||
x[i] = lrint(0.75 * INT16_MAX * sin(0.5 * M_PI * i * i / SIZE)); | |||
ff_iir_filter(fcoeffs, fstate, SIZE, x, 1, y, 1); | |||
for (i = 0; i < SIZE; i++) | |||
printf("%6d %6d\n", x[i], y[i]); | |||
ff_iir_filter_free_coeffs(fcoeffs); | |||
ff_iir_filter_free_state(fstate); | |||
return 0; | |||
} |
@@ -315,35 +315,3 @@ av_cold void ff_iir_filter_free_coeffs(struct FFIIRFilterCoeffs *coeffs) | |||
} | |||
av_free(coeffs); | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
#define FILT_ORDER 4 | |||
#define SIZE 1024 | |||
int main(void) | |||
{ | |||
struct FFIIRFilterCoeffs *fcoeffs = NULL; | |||
struct FFIIRFilterState *fstate = NULL; | |||
float cutoff_coeff = 0.4; | |||
int16_t x[SIZE], y[SIZE]; | |||
int i; | |||
fcoeffs = ff_iir_filter_init_coeffs(NULL, FF_FILTER_TYPE_BUTTERWORTH, | |||
FF_FILTER_MODE_LOWPASS, FILT_ORDER, | |||
cutoff_coeff, 0.0, 0.0); | |||
fstate = ff_iir_filter_init_state(FILT_ORDER); | |||
for (i = 0; i < SIZE; i++) | |||
x[i] = lrint(0.75 * INT16_MAX * sin(0.5 * M_PI * i * i / SIZE)); | |||
ff_iir_filter(fcoeffs, fstate, SIZE, x, 1, y, 1); | |||
for (i = 0; i < SIZE; i++) | |||
printf("%6d %6d\n", x[i], y[i]); | |||
ff_iir_filter_free_coeffs(fcoeffs); | |||
ff_iir_filter_free_state(fstate); | |||
return 0; | |||
} | |||
#endif /* TEST */ |
@@ -0,0 +1,64 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <stdint.h> | |||
#include <string.h> | |||
#include "libavutil/lfg.h" | |||
#include "libavutil/log.h" | |||
#include "rangecoder.h" | |||
#define SIZE 10240 | |||
int main(void) | |||
{ | |||
RangeCoder c; | |||
uint8_t b[9 * SIZE]; | |||
uint8_t r[9 * SIZE]; | |||
int i; | |||
uint8_t state[10]; | |||
AVLFG prng; | |||
av_lfg_init(&prng, 1); | |||
ff_init_range_encoder(&c, b, SIZE); | |||
ff_build_rac_states(&c, 0.05 * (1LL << 32), 128 + 64 + 32 + 16); | |||
memset(state, 128, sizeof(state)); | |||
for (i = 0; i < SIZE; i++) | |||
r[i] = av_lfg_get(&prng) % 7; | |||
for (i = 0; i < SIZE; i++) | |||
put_rac(&c, state, r[i] & 1); | |||
ff_rac_terminate(&c); | |||
ff_init_range_decoder(&c, b, SIZE); | |||
memset(state, 128, sizeof(state)); | |||
for (i = 0; i < SIZE; i++) | |||
if ((r[i] & 1) != get_rac(&c, state)) { | |||
av_log(NULL, AV_LOG_ERROR, "rac failure at %d\n", i); | |||
return 1; | |||
} | |||
return 0; | |||
} |
@@ -113,47 +113,3 @@ int ff_rac_terminate(RangeCoder *c) | |||
return c->bytestream - c->bytestream_start; | |||
} | |||
#ifdef TEST | |||
#define SIZE 10240 | |||
#include "libavutil/lfg.h" | |||
#include "libavutil/log.h" | |||
int main(void) | |||
{ | |||
RangeCoder c; | |||
uint8_t b[9 * SIZE]; | |||
uint8_t r[9 * SIZE]; | |||
int i; | |||
uint8_t state[10]; | |||
AVLFG prng; | |||
av_lfg_init(&prng, 1); | |||
ff_init_range_encoder(&c, b, SIZE); | |||
ff_build_rac_states(&c, 0.05 * (1LL << 32), 128 + 64 + 32 + 16); | |||
memset(state, 128, sizeof(state)); | |||
for (i = 0; i < SIZE; i++) | |||
r[i] = av_lfg_get(&prng) % 7; | |||
for (i = 0; i < SIZE; i++) | |||
put_rac(&c, state, r[i] & 1); | |||
ff_rac_terminate(&c); | |||
ff_init_range_decoder(&c, b, SIZE); | |||
memset(state, 128, sizeof(state)); | |||
for (i = 0; i < SIZE; i++) | |||
if ((r[i] & 1) != get_rac(&c, state)) { | |||
av_log(NULL, AV_LOG_ERROR, "rac failure at %d\n", i); | |||
return 1; | |||
} | |||
return 0; | |||
} | |||
#endif /* TEST */ |
@@ -0,0 +1,92 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <stdio.h> | |||
#include "libavutil/common.h" | |||
#include "libavutil/lfg.h" | |||
#include "timefilter.h" | |||
#define LFG_MAX ((1LL << 32) - 1) | |||
int main(void) | |||
{ | |||
AVLFG prng; | |||
double n0, n1; | |||
#define SAMPLES 1000 | |||
double ideal[SAMPLES]; | |||
double samples[SAMPLES]; | |||
for (n0 = 0; n0 < 40; n0 = 2 * n0 + 1) { | |||
for (n1 = 0; n1 < 10; n1 = 2 * n1 + 1) { | |||
double best_error = 1000000000; | |||
double bestpar0 = 1; | |||
double bestpar1 = 0.001; | |||
int better, i; | |||
av_lfg_init(&prng, 123); | |||
for (i = 0; i < SAMPLES; i++) { | |||
ideal[i] = 10 + i + n1 * i / (1000); | |||
samples[i] = ideal[i] + n0 * (av_lfg_get(&prng) - LFG_MAX / 2) / (LFG_MAX * 10LL); | |||
} | |||
do { | |||
double par0, par1; | |||
better = 0; | |||
for (par0 = bestpar0 * 0.8; par0 <= bestpar0 * 1.21; par0 += bestpar0 * 0.05) { | |||
for (par1 = bestpar1 * 0.8; par1 <= bestpar1 * 1.21; par1 += bestpar1 * 0.05) { | |||
double error = 0; | |||
TimeFilter *tf = ff_timefilter_new(1, par0, par1); | |||
if (!tf) { | |||
printf("Could not allocate memory for timefilter.\n"); | |||
exit(1); | |||
} | |||
for (i = 0; i < SAMPLES; i++) { | |||
double filtered; | |||
filtered = ff_timefilter_update(tf, samples[i], 1); | |||
error += (filtered - ideal[i]) * (filtered - ideal[i]); | |||
} | |||
ff_timefilter_destroy(tf); | |||
if (error < best_error) { | |||
best_error = error; | |||
bestpar0 = par0; | |||
bestpar1 = par1; | |||
better = 1; | |||
} | |||
} | |||
} | |||
} while (better); | |||
#if 0 | |||
double lastfil = 9; | |||
TimeFilter *tf = ff_timefilter_new(1, bestpar0, bestpar1); | |||
for (i = 0; i < SAMPLES; i++) { | |||
double filtered; | |||
filtered = ff_timefilter_update(tf, samples[i], 1); | |||
printf("%f %f %f %f\n", i - samples[i] + 10, filtered - samples[i], | |||
samples[FFMAX(i, 1)] - samples[FFMAX(i - 1, 0)], filtered - lastfil); | |||
lastfil = filtered; | |||
} | |||
ff_timefilter_destroy(tf); | |||
#else | |||
printf(" [%f %f %9f]", bestpar0, bestpar1, best_error); | |||
#endif | |||
} | |||
printf("\n"); | |||
} | |||
return 0; | |||
} |
@@ -77,74 +77,3 @@ double ff_timefilter_update(TimeFilter *self, double system_time, double period) | |||
} | |||
return self->cycle_time; | |||
} | |||
#ifdef TEST | |||
#include "libavutil/lfg.h" | |||
#define LFG_MAX ((1LL << 32) - 1) | |||
int main(void) | |||
{ | |||
AVLFG prng; | |||
double n0, n1; | |||
#define SAMPLES 1000 | |||
double ideal[SAMPLES]; | |||
double samples[SAMPLES]; | |||
for (n0 = 0; n0 < 40; n0 = 2 * n0 + 1) { | |||
for (n1 = 0; n1 < 10; n1 = 2 * n1 + 1) { | |||
double best_error = 1000000000; | |||
double bestpar0 = 1; | |||
double bestpar1 = 0.001; | |||
int better, i; | |||
av_lfg_init(&prng, 123); | |||
for (i = 0; i < SAMPLES; i++) { | |||
ideal[i] = 10 + i + n1 * i / (1000); | |||
samples[i] = ideal[i] + n0 * (av_lfg_get(&prng) - LFG_MAX / 2) / (LFG_MAX * 10LL); | |||
} | |||
do { | |||
double par0, par1; | |||
better = 0; | |||
for (par0 = bestpar0 * 0.8; par0 <= bestpar0 * 1.21; par0 += bestpar0 * 0.05) { | |||
for (par1 = bestpar1 * 0.8; par1 <= bestpar1 * 1.21; par1 += bestpar1 * 0.05) { | |||
double error = 0; | |||
TimeFilter *tf = ff_timefilter_new(1, par0, par1); | |||
if (!tf) { | |||
printf("Could not allocate memory for timefilter.\n"); | |||
exit(1); | |||
} | |||
for (i = 0; i < SAMPLES; i++) { | |||
double filtered; | |||
filtered = ff_timefilter_update(tf, samples[i], 1); | |||
error += (filtered - ideal[i]) * (filtered - ideal[i]); | |||
} | |||
ff_timefilter_destroy(tf); | |||
if (error < best_error) { | |||
best_error = error; | |||
bestpar0 = par0; | |||
bestpar1 = par1; | |||
better = 1; | |||
} | |||
} | |||
} | |||
} while (better); | |||
#if 0 | |||
double lastfil = 9; | |||
TimeFilter *tf = ff_timefilter_new(1, bestpar0, bestpar1); | |||
for (i = 0; i < SAMPLES; i++) { | |||
double filtered; | |||
filtered = ff_timefilter_update(tf, samples[i], 1); | |||
printf("%f %f %f %f\n", i - samples[i] + 10, filtered - samples[i], | |||
samples[FFMAX(i, 1)] - samples[FFMAX(i - 1, 0)], filtered - lastfil); | |||
lastfil = filtered; | |||
} | |||
ff_timefilter_destroy(tf); | |||
#else | |||
printf(" [%f %f %9f]", bestpar0, bestpar1, best_error); | |||
#endif | |||
} | |||
printf("\n"); | |||
} | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,158 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include "rtmpdh.c" | |||
#include <stdio.h> | |||
static int test_random_shared_secret(void) | |||
{ | |||
FF_DH *peer1 = NULL, *peer2 = NULL; | |||
int ret; | |||
uint8_t pubkey1[128], pubkey2[128]; | |||
uint8_t sharedkey1[128], sharedkey2[128]; | |||
peer1 = ff_dh_init(1024); | |||
peer2 = ff_dh_init(1024); | |||
if (!peer1 || !peer2) { | |||
ret = AVERROR(ENOMEM); | |||
goto fail; | |||
} | |||
if ((ret = ff_dh_generate_public_key(peer1)) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_generate_public_key(peer2)) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_write_public_key(peer1, pubkey1, sizeof(pubkey1))) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_write_public_key(peer2, pubkey2, sizeof(pubkey2))) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_compute_shared_secret_key(peer1, pubkey2, sizeof(pubkey2), | |||
sharedkey1, sizeof(sharedkey1))) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_compute_shared_secret_key(peer2, pubkey1, sizeof(pubkey1), | |||
sharedkey2, sizeof(sharedkey2))) < 0) | |||
goto fail; | |||
if (memcmp(sharedkey1, sharedkey2, sizeof(sharedkey1))) { | |||
printf("Mismatched generated shared key\n"); | |||
ret = AVERROR_INVALIDDATA; | |||
} else { | |||
printf("Generated shared key ok\n"); | |||
} | |||
fail: | |||
ff_dh_free(peer1); | |||
ff_dh_free(peer2); | |||
return ret; | |||
} | |||
static const char *private_key = | |||
"976C18FCADC255B456564F74F3EEDA59D28AF6B744D743F2357BFD2404797EF896EF1A" | |||
"7C1CBEAAA3AB60AF3192D189CFF3F991C9CBBFD78119FCA2181384B94011943B6D6F28" | |||
"9E1B708E2D1A0C7771169293F03DA27E561F15F16F0AC9BC858C77A80FA98FD088A232" | |||
"19D08BE6F165DE0B02034B18705829FAD0ACB26A5B75EF"; | |||
static const char *public_key = | |||
"F272ECF8362257C5D2C3CC2229CF9C0A03225BC109B1DBC76A68C394F256ACA3EF5F64" | |||
"FC270C26382BF315C19E97A76104A716FC998A651E8610A3AE6CF65D8FAE5D3F32EEA0" | |||
"0B32CB9609B494116A825D7142D17B88E3D20EDD98743DE29CF37A23A9F6A58B960591" | |||
"3157D5965FCB46DDA73A1F08DD897BAE88DFE6FC937CBA"; | |||
static const uint8_t public_key_bin[] = { | |||
0xf2, 0x72, 0xec, 0xf8, 0x36, 0x22, 0x57, 0xc5, 0xd2, 0xc3, 0xcc, 0x22, | |||
0x29, 0xcf, 0x9c, 0x0a, 0x03, 0x22, 0x5b, 0xc1, 0x09, 0xb1, 0xdb, 0xc7, | |||
0x6a, 0x68, 0xc3, 0x94, 0xf2, 0x56, 0xac, 0xa3, 0xef, 0x5f, 0x64, 0xfc, | |||
0x27, 0x0c, 0x26, 0x38, 0x2b, 0xf3, 0x15, 0xc1, 0x9e, 0x97, 0xa7, 0x61, | |||
0x04, 0xa7, 0x16, 0xfc, 0x99, 0x8a, 0x65, 0x1e, 0x86, 0x10, 0xa3, 0xae, | |||
0x6c, 0xf6, 0x5d, 0x8f, 0xae, 0x5d, 0x3f, 0x32, 0xee, 0xa0, 0x0b, 0x32, | |||
0xcb, 0x96, 0x09, 0xb4, 0x94, 0x11, 0x6a, 0x82, 0x5d, 0x71, 0x42, 0xd1, | |||
0x7b, 0x88, 0xe3, 0xd2, 0x0e, 0xdd, 0x98, 0x74, 0x3d, 0xe2, 0x9c, 0xf3, | |||
0x7a, 0x23, 0xa9, 0xf6, 0xa5, 0x8b, 0x96, 0x05, 0x91, 0x31, 0x57, 0xd5, | |||
0x96, 0x5f, 0xcb, 0x46, 0xdd, 0xa7, 0x3a, 0x1f, 0x08, 0xdd, 0x89, 0x7b, | |||
0xae, 0x88, 0xdf, 0xe6, 0xfc, 0x93, 0x7c, 0xba | |||
}; | |||
static const uint8_t peer_public_key[] = { | |||
0x58, 0x66, 0x05, 0x49, 0x94, 0x23, 0x2b, 0x66, 0x52, 0x13, 0xff, 0x46, | |||
0xf2, 0xb3, 0x79, 0xa9, 0xee, 0xae, 0x1a, 0x13, 0xf0, 0x71, 0x52, 0xfb, | |||
0x93, 0x4e, 0xee, 0x97, 0x05, 0x73, 0x50, 0x7d, 0xaf, 0x02, 0x07, 0x72, | |||
0xac, 0xdc, 0xa3, 0x95, 0x78, 0xee, 0x9a, 0x19, 0x71, 0x7e, 0x99, 0x9f, | |||
0x2a, 0xd4, 0xb3, 0xe2, 0x0c, 0x1d, 0x1a, 0x78, 0x4c, 0xde, 0xf1, 0xad, | |||
0xb4, 0x60, 0xa8, 0x51, 0xac, 0x71, 0xec, 0x86, 0x70, 0xa2, 0x63, 0x36, | |||
0x92, 0x7c, 0xe3, 0x87, 0xee, 0xe4, 0xf1, 0x62, 0x24, 0x74, 0xb4, 0x04, | |||
0xfa, 0x5c, 0xdf, 0xba, 0xfa, 0xa3, 0xc2, 0xbb, 0x62, 0x27, 0xd0, 0xf4, | |||
0xe4, 0x43, 0xda, 0x8a, 0x88, 0x69, 0x60, 0xe2, 0xdb, 0x75, 0x2a, 0x98, | |||
0x9d, 0xb5, 0x50, 0xe3, 0x99, 0xda, 0xe0, 0xa6, 0x14, 0xc9, 0x80, 0x12, | |||
0xf9, 0x3c, 0xac, 0x06, 0x02, 0x7a, 0xde, 0x74 | |||
}; | |||
static const uint8_t shared_secret[] = { | |||
0xb2, 0xeb, 0xcb, 0x71, 0xf3, 0x61, 0xfb, 0x5b, 0x4e, 0x5c, 0x4c, 0xcf, | |||
0x5c, 0x08, 0x5f, 0x96, 0x26, 0x77, 0x1d, 0x31, 0xf1, 0xe1, 0xf7, 0x4b, | |||
0x92, 0xac, 0x82, 0x2a, 0x88, 0xc7, 0x83, 0xe1, 0xc7, 0xf3, 0xd3, 0x1a, | |||
0x7d, 0xc8, 0x31, 0xe3, 0x97, 0xe4, 0xec, 0x31, 0x0e, 0x8f, 0x73, 0x1a, | |||
0xe4, 0xf6, 0xd8, 0xc8, 0x94, 0xff, 0xa0, 0x03, 0x84, 0x03, 0x0f, 0xa5, | |||
0x30, 0x5d, 0x67, 0xe0, 0x7a, 0x3b, 0x5f, 0xed, 0x4c, 0xf5, 0xbc, 0x18, | |||
0xea, 0xd4, 0x77, 0xa9, 0x07, 0xb3, 0x54, 0x0b, 0x02, 0xd9, 0xc6, 0xb8, | |||
0x66, 0x5e, 0xec, 0xa4, 0xcd, 0x47, 0xed, 0xc9, 0x38, 0xc6, 0x91, 0x08, | |||
0xf3, 0x85, 0x9b, 0x69, 0x16, 0x78, 0x0d, 0xb7, 0x74, 0x51, 0xaa, 0x5b, | |||
0x4d, 0x74, 0xe4, 0x29, 0x2e, 0x9e, 0x8e, 0xf7, 0xe5, 0x42, 0x83, 0xb0, | |||
0x65, 0xb0, 0xce, 0xc6, 0xb2, 0x8f, 0x5b, 0xb0 | |||
}; | |||
static int test_ref_data(void) | |||
{ | |||
FF_DH *dh; | |||
int ret = AVERROR(ENOMEM); | |||
uint8_t pubkey_test[128]; | |||
uint8_t sharedkey_test[128]; | |||
dh = ff_dh_init(1024); | |||
if (!dh) | |||
goto fail; | |||
bn_hex2bn(dh->priv_key, private_key, ret); | |||
if (!ret) | |||
goto fail; | |||
bn_hex2bn(dh->pub_key, public_key, ret); | |||
if (!ret) | |||
goto fail; | |||
if ((ret = ff_dh_write_public_key(dh, pubkey_test, sizeof(pubkey_test))) < 0) | |||
goto fail; | |||
if (memcmp(pubkey_test, public_key_bin, sizeof(pubkey_test))) { | |||
printf("Mismatched generated public key\n"); | |||
ret = AVERROR_INVALIDDATA; | |||
goto fail; | |||
} else { | |||
printf("Generated public key ok\n"); | |||
} | |||
if ((ret = ff_dh_compute_shared_secret_key(dh, peer_public_key, sizeof(peer_public_key), | |||
sharedkey_test, sizeof(sharedkey_test))) < 0) | |||
goto fail; | |||
if (memcmp(shared_secret, sharedkey_test, sizeof(sharedkey_test))) { | |||
printf("Mismatched generated shared key\n"); | |||
ret = AVERROR_INVALIDDATA; | |||
} else { | |||
printf("Generated shared key ok\n"); | |||
} | |||
fail: | |||
ff_dh_free(dh); | |||
return ret; | |||
} | |||
int main(void) | |||
{ | |||
if (test_random_shared_secret() < 0) | |||
return 1; | |||
if (test_ref_data() < 0) | |||
return 1; | |||
return 0; | |||
} |
@@ -360,145 +360,3 @@ fail: | |||
return ret; | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
static int test_random_shared_secret(void) | |||
{ | |||
FF_DH *peer1 = NULL, *peer2 = NULL; | |||
int ret; | |||
uint8_t pubkey1[128], pubkey2[128]; | |||
uint8_t sharedkey1[128], sharedkey2[128]; | |||
peer1 = ff_dh_init(1024); | |||
peer2 = ff_dh_init(1024); | |||
if (!peer1 || !peer2) { | |||
ret = AVERROR(ENOMEM); | |||
goto fail; | |||
} | |||
if ((ret = ff_dh_generate_public_key(peer1)) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_generate_public_key(peer2)) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_write_public_key(peer1, pubkey1, sizeof(pubkey1))) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_write_public_key(peer2, pubkey2, sizeof(pubkey2))) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_compute_shared_secret_key(peer1, pubkey2, sizeof(pubkey2), | |||
sharedkey1, sizeof(sharedkey1))) < 0) | |||
goto fail; | |||
if ((ret = ff_dh_compute_shared_secret_key(peer2, pubkey1, sizeof(pubkey1), | |||
sharedkey2, sizeof(sharedkey2))) < 0) | |||
goto fail; | |||
if (memcmp(sharedkey1, sharedkey2, sizeof(sharedkey1))) { | |||
printf("Mismatched generated shared key\n"); | |||
ret = AVERROR_INVALIDDATA; | |||
} else { | |||
printf("Generated shared key ok\n"); | |||
} | |||
fail: | |||
ff_dh_free(peer1); | |||
ff_dh_free(peer2); | |||
return ret; | |||
} | |||
static const char *private_key = | |||
"976C18FCADC255B456564F74F3EEDA59D28AF6B744D743F2357BFD2404797EF896EF1A" | |||
"7C1CBEAAA3AB60AF3192D189CFF3F991C9CBBFD78119FCA2181384B94011943B6D6F28" | |||
"9E1B708E2D1A0C7771169293F03DA27E561F15F16F0AC9BC858C77A80FA98FD088A232" | |||
"19D08BE6F165DE0B02034B18705829FAD0ACB26A5B75EF"; | |||
static const char *public_key = | |||
"F272ECF8362257C5D2C3CC2229CF9C0A03225BC109B1DBC76A68C394F256ACA3EF5F64" | |||
"FC270C26382BF315C19E97A76104A716FC998A651E8610A3AE6CF65D8FAE5D3F32EEA0" | |||
"0B32CB9609B494116A825D7142D17B88E3D20EDD98743DE29CF37A23A9F6A58B960591" | |||
"3157D5965FCB46DDA73A1F08DD897BAE88DFE6FC937CBA"; | |||
static const uint8_t public_key_bin[] = { | |||
0xf2, 0x72, 0xec, 0xf8, 0x36, 0x22, 0x57, 0xc5, 0xd2, 0xc3, 0xcc, 0x22, | |||
0x29, 0xcf, 0x9c, 0x0a, 0x03, 0x22, 0x5b, 0xc1, 0x09, 0xb1, 0xdb, 0xc7, | |||
0x6a, 0x68, 0xc3, 0x94, 0xf2, 0x56, 0xac, 0xa3, 0xef, 0x5f, 0x64, 0xfc, | |||
0x27, 0x0c, 0x26, 0x38, 0x2b, 0xf3, 0x15, 0xc1, 0x9e, 0x97, 0xa7, 0x61, | |||
0x04, 0xa7, 0x16, 0xfc, 0x99, 0x8a, 0x65, 0x1e, 0x86, 0x10, 0xa3, 0xae, | |||
0x6c, 0xf6, 0x5d, 0x8f, 0xae, 0x5d, 0x3f, 0x32, 0xee, 0xa0, 0x0b, 0x32, | |||
0xcb, 0x96, 0x09, 0xb4, 0x94, 0x11, 0x6a, 0x82, 0x5d, 0x71, 0x42, 0xd1, | |||
0x7b, 0x88, 0xe3, 0xd2, 0x0e, 0xdd, 0x98, 0x74, 0x3d, 0xe2, 0x9c, 0xf3, | |||
0x7a, 0x23, 0xa9, 0xf6, 0xa5, 0x8b, 0x96, 0x05, 0x91, 0x31, 0x57, 0xd5, | |||
0x96, 0x5f, 0xcb, 0x46, 0xdd, 0xa7, 0x3a, 0x1f, 0x08, 0xdd, 0x89, 0x7b, | |||
0xae, 0x88, 0xdf, 0xe6, 0xfc, 0x93, 0x7c, 0xba | |||
}; | |||
static const uint8_t peer_public_key[] = { | |||
0x58, 0x66, 0x05, 0x49, 0x94, 0x23, 0x2b, 0x66, 0x52, 0x13, 0xff, 0x46, | |||
0xf2, 0xb3, 0x79, 0xa9, 0xee, 0xae, 0x1a, 0x13, 0xf0, 0x71, 0x52, 0xfb, | |||
0x93, 0x4e, 0xee, 0x97, 0x05, 0x73, 0x50, 0x7d, 0xaf, 0x02, 0x07, 0x72, | |||
0xac, 0xdc, 0xa3, 0x95, 0x78, 0xee, 0x9a, 0x19, 0x71, 0x7e, 0x99, 0x9f, | |||
0x2a, 0xd4, 0xb3, 0xe2, 0x0c, 0x1d, 0x1a, 0x78, 0x4c, 0xde, 0xf1, 0xad, | |||
0xb4, 0x60, 0xa8, 0x51, 0xac, 0x71, 0xec, 0x86, 0x70, 0xa2, 0x63, 0x36, | |||
0x92, 0x7c, 0xe3, 0x87, 0xee, 0xe4, 0xf1, 0x62, 0x24, 0x74, 0xb4, 0x04, | |||
0xfa, 0x5c, 0xdf, 0xba, 0xfa, 0xa3, 0xc2, 0xbb, 0x62, 0x27, 0xd0, 0xf4, | |||
0xe4, 0x43, 0xda, 0x8a, 0x88, 0x69, 0x60, 0xe2, 0xdb, 0x75, 0x2a, 0x98, | |||
0x9d, 0xb5, 0x50, 0xe3, 0x99, 0xda, 0xe0, 0xa6, 0x14, 0xc9, 0x80, 0x12, | |||
0xf9, 0x3c, 0xac, 0x06, 0x02, 0x7a, 0xde, 0x74 | |||
}; | |||
static const uint8_t shared_secret[] = { | |||
0xb2, 0xeb, 0xcb, 0x71, 0xf3, 0x61, 0xfb, 0x5b, 0x4e, 0x5c, 0x4c, 0xcf, | |||
0x5c, 0x08, 0x5f, 0x96, 0x26, 0x77, 0x1d, 0x31, 0xf1, 0xe1, 0xf7, 0x4b, | |||
0x92, 0xac, 0x82, 0x2a, 0x88, 0xc7, 0x83, 0xe1, 0xc7, 0xf3, 0xd3, 0x1a, | |||
0x7d, 0xc8, 0x31, 0xe3, 0x97, 0xe4, 0xec, 0x31, 0x0e, 0x8f, 0x73, 0x1a, | |||
0xe4, 0xf6, 0xd8, 0xc8, 0x94, 0xff, 0xa0, 0x03, 0x84, 0x03, 0x0f, 0xa5, | |||
0x30, 0x5d, 0x67, 0xe0, 0x7a, 0x3b, 0x5f, 0xed, 0x4c, 0xf5, 0xbc, 0x18, | |||
0xea, 0xd4, 0x77, 0xa9, 0x07, 0xb3, 0x54, 0x0b, 0x02, 0xd9, 0xc6, 0xb8, | |||
0x66, 0x5e, 0xec, 0xa4, 0xcd, 0x47, 0xed, 0xc9, 0x38, 0xc6, 0x91, 0x08, | |||
0xf3, 0x85, 0x9b, 0x69, 0x16, 0x78, 0x0d, 0xb7, 0x74, 0x51, 0xaa, 0x5b, | |||
0x4d, 0x74, 0xe4, 0x29, 0x2e, 0x9e, 0x8e, 0xf7, 0xe5, 0x42, 0x83, 0xb0, | |||
0x65, 0xb0, 0xce, 0xc6, 0xb2, 0x8f, 0x5b, 0xb0 | |||
}; | |||
static int test_ref_data(void) | |||
{ | |||
FF_DH *dh; | |||
int ret = AVERROR(ENOMEM); | |||
uint8_t pubkey_test[128]; | |||
uint8_t sharedkey_test[128]; | |||
dh = ff_dh_init(1024); | |||
if (!dh) | |||
goto fail; | |||
bn_hex2bn(dh->priv_key, private_key, ret); | |||
if (!ret) | |||
goto fail; | |||
bn_hex2bn(dh->pub_key, public_key, ret); | |||
if (!ret) | |||
goto fail; | |||
if ((ret = ff_dh_write_public_key(dh, pubkey_test, sizeof(pubkey_test))) < 0) | |||
goto fail; | |||
if (memcmp(pubkey_test, public_key_bin, sizeof(pubkey_test))) { | |||
printf("Mismatched generated public key\n"); | |||
ret = AVERROR_INVALIDDATA; | |||
goto fail; | |||
} else { | |||
printf("Generated public key ok\n"); | |||
} | |||
if ((ret = ff_dh_compute_shared_secret_key(dh, peer_public_key, sizeof(peer_public_key), | |||
sharedkey_test, sizeof(sharedkey_test))) < 0) | |||
goto fail; | |||
if (memcmp(shared_secret, sharedkey_test, sizeof(sharedkey_test))) { | |||
printf("Mismatched generated shared key\n"); | |||
ret = AVERROR_INVALIDDATA; | |||
} else { | |||
printf("Generated shared key ok\n"); | |||
} | |||
fail: | |||
ff_dh_free(dh); | |||
return ret; | |||
} | |||
int main(void) | |||
{ | |||
if (test_random_shared_secret() < 0) | |||
return 1; | |||
if (test_ref_data() < 0) | |||
return 1; | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,167 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; 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 "rtpdec.h" | |||
#include "srtp.h" | |||
static const char *aes128_80_key = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmn"; | |||
static const uint8_t rtp_aes128_80[] = { | |||
// RTP header | |||
0x80, 0xe0, 0x12, 0x34, 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x62, 0x69, 0x76, 0xca, 0xc5, | |||
// HMAC | |||
0xa1, 0xac, 0x1b, 0xb4, 0xa0, 0x1c, 0xd5, 0x49, 0x28, 0x99, | |||
}; | |||
static const uint8_t rtcp_aes128_80[] = { | |||
// RTCP header | |||
0x81, 0xc9, 0x00, 0x07, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x8a, 0xac, 0xdc, 0xa5, 0x4c, 0xf6, 0x78, 0xa6, 0x62, 0x8f, 0x24, 0xda, | |||
0x6c, 0x09, 0x3f, 0xa9, 0x28, 0x7a, 0xb5, 0x7f, 0x1f, 0x0f, 0xc9, 0x35, | |||
// RTCP index | |||
0x80, 0x00, 0x00, 0x03, | |||
// HMAC | |||
0xe9, 0x3b, 0xc0, 0x5c, 0x0c, 0x06, 0x9f, 0xab, 0xc0, 0xde, | |||
}; | |||
static const char *aes128_32_key = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmn"; | |||
static const uint8_t rtp_aes128_32[] = { | |||
// RTP header | |||
0x80, 0xe0, 0x12, 0x34, 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x62, 0x69, 0x76, 0xca, 0xc5, | |||
// HMAC | |||
0xa1, 0xac, 0x1b, 0xb4, | |||
}; | |||
static const uint8_t rtcp_aes128_32[] = { | |||
// RTCP header | |||
0x81, 0xc9, 0x00, 0x07, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x35, 0xe9, 0xb5, 0xff, 0x0d, 0xd1, 0xde, 0x70, 0x74, 0x10, 0xaa, 0x1b, | |||
0xb2, 0x8d, 0xf0, 0x20, 0x02, 0x99, 0x6b, 0x1b, 0x0b, 0xd0, 0x47, 0x34, | |||
// RTCP index | |||
0x80, 0x00, 0x00, 0x04, | |||
// HMAC | |||
0x5b, 0xd2, 0xa9, 0x9d, | |||
}; | |||
static const char *aes128_80_32_key = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmn"; | |||
static const uint8_t rtp_aes128_80_32[] = { | |||
// RTP header | |||
0x80, 0xe0, 0x12, 0x34, 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x62, 0x69, 0x76, 0xca, 0xc5, | |||
// HMAC | |||
0xa1, 0xac, 0x1b, 0xb4, | |||
}; | |||
static const uint8_t rtcp_aes128_80_32[] = { | |||
// RTCP header | |||
0x81, 0xc9, 0x00, 0x07, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0xd6, 0xae, 0xc1, 0x58, 0x63, 0x70, 0xc9, 0x88, 0x66, 0x26, 0x1c, 0x53, | |||
0xff, 0x5d, 0x5d, 0x2b, 0x0f, 0x8c, 0x72, 0x3e, 0xc9, 0x1d, 0x43, 0xf9, | |||
// RTCP index | |||
0x80, 0x00, 0x00, 0x05, | |||
// HMAC | |||
0x09, 0x16, 0xb4, 0x27, 0x9a, 0xe9, 0x92, 0x26, 0x4e, 0x10, | |||
}; | |||
static void print_data(const uint8_t *buf, int len) | |||
{ | |||
int i; | |||
for (i = 0; i < len; i++) | |||
printf("%02x", buf[i]); | |||
printf("\n"); | |||
} | |||
static int test_decrypt(struct SRTPContext *srtp, const uint8_t *in, int len, | |||
uint8_t *out) | |||
{ | |||
memcpy(out, in, len); | |||
if (!ff_srtp_decrypt(srtp, out, &len)) { | |||
print_data(out, len); | |||
return len; | |||
} else | |||
return -1; | |||
} | |||
static void test_encrypt(const uint8_t *data, int in_len, const char *suite, | |||
const char *key) | |||
{ | |||
struct SRTPContext enc = { 0 }, dec = { 0 }; | |||
int len; | |||
char buf[RTP_MAX_PACKET_LENGTH]; | |||
ff_srtp_set_crypto(&enc, suite, key); | |||
ff_srtp_set_crypto(&dec, suite, key); | |||
len = ff_srtp_encrypt(&enc, data, in_len, buf, sizeof(buf)); | |||
if (!ff_srtp_decrypt(&dec, buf, &len)) { | |||
if (len == in_len && !memcmp(buf, data, len)) | |||
printf("Decrypted content matches input\n"); | |||
else | |||
printf("Decrypted content doesn't match input\n"); | |||
} else { | |||
printf("Decryption failed\n"); | |||
} | |||
ff_srtp_free(&enc); | |||
ff_srtp_free(&dec); | |||
} | |||
int main(void) | |||
{ | |||
static const char *aes128_80_suite = "AES_CM_128_HMAC_SHA1_80"; | |||
static const char *aes128_32_suite = "AES_CM_128_HMAC_SHA1_32"; | |||
static const char *aes128_80_32_suite = "SRTP_AES128_CM_HMAC_SHA1_32"; | |||
static const char *test_key = "abcdefghijklmnopqrstuvwxyz1234567890ABCD"; | |||
uint8_t buf[RTP_MAX_PACKET_LENGTH]; | |||
struct SRTPContext srtp = { 0 }; | |||
int len; | |||
ff_srtp_set_crypto(&srtp, aes128_80_suite, aes128_80_key); | |||
len = test_decrypt(&srtp, rtp_aes128_80, sizeof(rtp_aes128_80), buf); | |||
test_encrypt(buf, len, aes128_80_suite, test_key); | |||
test_encrypt(buf, len, aes128_32_suite, test_key); | |||
test_encrypt(buf, len, aes128_80_32_suite, test_key); | |||
test_decrypt(&srtp, rtcp_aes128_80, sizeof(rtcp_aes128_80), buf); | |||
test_encrypt(buf, len, aes128_80_suite, test_key); | |||
test_encrypt(buf, len, aes128_32_suite, test_key); | |||
test_encrypt(buf, len, aes128_80_32_suite, test_key); | |||
ff_srtp_free(&srtp); | |||
memset(&srtp, 0, sizeof(srtp)); // Clear the context | |||
ff_srtp_set_crypto(&srtp, aes128_32_suite, aes128_32_key); | |||
test_decrypt(&srtp, rtp_aes128_32, sizeof(rtp_aes128_32), buf); | |||
test_decrypt(&srtp, rtcp_aes128_32, sizeof(rtcp_aes128_32), buf); | |||
ff_srtp_free(&srtp); | |||
memset(&srtp, 0, sizeof(srtp)); // Clear the context | |||
ff_srtp_set_crypto(&srtp, aes128_80_32_suite, aes128_80_32_key); | |||
test_decrypt(&srtp, rtp_aes128_80_32, sizeof(rtp_aes128_80_32), buf); | |||
test_decrypt(&srtp, rtcp_aes128_80_32, sizeof(rtcp_aes128_80_32), buf); | |||
ff_srtp_free(&srtp); | |||
return 0; | |||
} |
@@ -323,150 +323,3 @@ int ff_srtp_encrypt(struct SRTPContext *s, const uint8_t *in, int len, | |||
len += hmac_size; | |||
return buf + len - out; | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
static const char *aes128_80_key = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmn"; | |||
static const uint8_t rtp_aes128_80[] = { | |||
// RTP header | |||
0x80, 0xe0, 0x12, 0x34, 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x62, 0x69, 0x76, 0xca, 0xc5, | |||
// HMAC | |||
0xa1, 0xac, 0x1b, 0xb4, 0xa0, 0x1c, 0xd5, 0x49, 0x28, 0x99, | |||
}; | |||
static const uint8_t rtcp_aes128_80[] = { | |||
// RTCP header | |||
0x81, 0xc9, 0x00, 0x07, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x8a, 0xac, 0xdc, 0xa5, 0x4c, 0xf6, 0x78, 0xa6, 0x62, 0x8f, 0x24, 0xda, | |||
0x6c, 0x09, 0x3f, 0xa9, 0x28, 0x7a, 0xb5, 0x7f, 0x1f, 0x0f, 0xc9, 0x35, | |||
// RTCP index | |||
0x80, 0x00, 0x00, 0x03, | |||
// HMAC | |||
0xe9, 0x3b, 0xc0, 0x5c, 0x0c, 0x06, 0x9f, 0xab, 0xc0, 0xde, | |||
}; | |||
static const char *aes128_32_key = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmn"; | |||
static const uint8_t rtp_aes128_32[] = { | |||
// RTP header | |||
0x80, 0xe0, 0x12, 0x34, 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x62, 0x69, 0x76, 0xca, 0xc5, | |||
// HMAC | |||
0xa1, 0xac, 0x1b, 0xb4, | |||
}; | |||
static const uint8_t rtcp_aes128_32[] = { | |||
// RTCP header | |||
0x81, 0xc9, 0x00, 0x07, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x35, 0xe9, 0xb5, 0xff, 0x0d, 0xd1, 0xde, 0x70, 0x74, 0x10, 0xaa, 0x1b, | |||
0xb2, 0x8d, 0xf0, 0x20, 0x02, 0x99, 0x6b, 0x1b, 0x0b, 0xd0, 0x47, 0x34, | |||
// RTCP index | |||
0x80, 0x00, 0x00, 0x04, | |||
// HMAC | |||
0x5b, 0xd2, 0xa9, 0x9d, | |||
}; | |||
static const char *aes128_80_32_key = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmn"; | |||
static const uint8_t rtp_aes128_80_32[] = { | |||
// RTP header | |||
0x80, 0xe0, 0x12, 0x34, 0x12, 0x34, 0x56, 0x78, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0x62, 0x69, 0x76, 0xca, 0xc5, | |||
// HMAC | |||
0xa1, 0xac, 0x1b, 0xb4, | |||
}; | |||
static const uint8_t rtcp_aes128_80_32[] = { | |||
// RTCP header | |||
0x81, 0xc9, 0x00, 0x07, 0x12, 0x34, 0x56, 0x78, | |||
// encrypted payload | |||
0xd6, 0xae, 0xc1, 0x58, 0x63, 0x70, 0xc9, 0x88, 0x66, 0x26, 0x1c, 0x53, | |||
0xff, 0x5d, 0x5d, 0x2b, 0x0f, 0x8c, 0x72, 0x3e, 0xc9, 0x1d, 0x43, 0xf9, | |||
// RTCP index | |||
0x80, 0x00, 0x00, 0x05, | |||
// HMAC | |||
0x09, 0x16, 0xb4, 0x27, 0x9a, 0xe9, 0x92, 0x26, 0x4e, 0x10, | |||
}; | |||
static void print_data(const uint8_t *buf, int len) | |||
{ | |||
int i; | |||
for (i = 0; i < len; i++) | |||
printf("%02x", buf[i]); | |||
printf("\n"); | |||
} | |||
static int test_decrypt(struct SRTPContext *srtp, const uint8_t *in, int len, | |||
uint8_t *out) | |||
{ | |||
memcpy(out, in, len); | |||
if (!ff_srtp_decrypt(srtp, out, &len)) { | |||
print_data(out, len); | |||
return len; | |||
} else | |||
return -1; | |||
} | |||
static void test_encrypt(const uint8_t *data, int in_len, const char *suite, | |||
const char *key) | |||
{ | |||
struct SRTPContext enc = { 0 }, dec = { 0 }; | |||
int len; | |||
char buf[RTP_MAX_PACKET_LENGTH]; | |||
ff_srtp_set_crypto(&enc, suite, key); | |||
ff_srtp_set_crypto(&dec, suite, key); | |||
len = ff_srtp_encrypt(&enc, data, in_len, buf, sizeof(buf)); | |||
if (!ff_srtp_decrypt(&dec, buf, &len)) { | |||
if (len == in_len && !memcmp(buf, data, len)) | |||
printf("Decrypted content matches input\n"); | |||
else | |||
printf("Decrypted content doesn't match input\n"); | |||
} else { | |||
printf("Decryption failed\n"); | |||
} | |||
ff_srtp_free(&enc); | |||
ff_srtp_free(&dec); | |||
} | |||
int main(void) | |||
{ | |||
static const char *aes128_80_suite = "AES_CM_128_HMAC_SHA1_80"; | |||
static const char *aes128_32_suite = "AES_CM_128_HMAC_SHA1_32"; | |||
static const char *aes128_80_32_suite = "SRTP_AES128_CM_HMAC_SHA1_32"; | |||
static const char *test_key = "abcdefghijklmnopqrstuvwxyz1234567890ABCD"; | |||
uint8_t buf[RTP_MAX_PACKET_LENGTH]; | |||
struct SRTPContext srtp = { 0 }; | |||
int len; | |||
ff_srtp_set_crypto(&srtp, aes128_80_suite, aes128_80_key); | |||
len = test_decrypt(&srtp, rtp_aes128_80, sizeof(rtp_aes128_80), buf); | |||
test_encrypt(buf, len, aes128_80_suite, test_key); | |||
test_encrypt(buf, len, aes128_32_suite, test_key); | |||
test_encrypt(buf, len, aes128_80_32_suite, test_key); | |||
test_decrypt(&srtp, rtcp_aes128_80, sizeof(rtcp_aes128_80), buf); | |||
test_encrypt(buf, len, aes128_80_suite, test_key); | |||
test_encrypt(buf, len, aes128_32_suite, test_key); | |||
test_encrypt(buf, len, aes128_80_32_suite, test_key); | |||
ff_srtp_free(&srtp); | |||
memset(&srtp, 0, sizeof(srtp)); // Clear the context | |||
ff_srtp_set_crypto(&srtp, aes128_32_suite, aes128_32_key); | |||
test_decrypt(&srtp, rtp_aes128_32, sizeof(rtp_aes128_32), buf); | |||
test_decrypt(&srtp, rtcp_aes128_32, sizeof(rtcp_aes128_32), buf); | |||
ff_srtp_free(&srtp); | |||
memset(&srtp, 0, sizeof(srtp)); // Clear the context | |||
ff_srtp_set_crypto(&srtp, aes128_80_32_suite, aes128_80_32_key); | |||
test_decrypt(&srtp, rtp_aes128_80_32, sizeof(rtp_aes128_80_32), buf); | |||
test_decrypt(&srtp, rtcp_aes128_80_32, sizeof(rtcp_aes128_80_32), buf); | |||
ff_srtp_free(&srtp); | |||
return 0; | |||
} | |||
#endif /* TEST */ |
@@ -0,0 +1,51 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <string.h> | |||
#include "log.h" | |||
#include "timer.h" | |||
#include "adler32.h" | |||
#define LEN 7001 | |||
static volatile int checksum; | |||
int main(int argc, char **argv) | |||
{ | |||
int i; | |||
char data[LEN]; | |||
av_log_set_level(AV_LOG_DEBUG); | |||
for (i = 0; i < LEN; i++) | |||
data[i] = ((i * i) >> 3) + 123 * i; | |||
if (argc > 1 && !strcmp(argv[1], "-t")) { | |||
for (i = 0; i < 1000; i++) { | |||
START_TIMER; | |||
checksum = av_adler32_update(1, data, LEN); | |||
STOP_TIMER("adler"); | |||
} | |||
} else { | |||
checksum = av_adler32_update(1, data, LEN); | |||
} | |||
av_log(NULL, AV_LOG_DEBUG, "%X (expected 50E6E508)\n", checksum); | |||
return checksum == 0x50e6e508 ? 0 : 1; | |||
} |
@@ -63,36 +63,3 @@ unsigned long av_adler32_update(unsigned long adler, const uint8_t * buf, | |||
} | |||
return (s2 << 16) | s1; | |||
} | |||
#ifdef TEST | |||
#include <string.h> | |||
#include "log.h" | |||
#include "timer.h" | |||
#define LEN 7001 | |||
static volatile int checksum; | |||
int main(int argc, char **argv) | |||
{ | |||
int i; | |||
char data[LEN]; | |||
av_log_set_level(AV_LOG_DEBUG); | |||
for (i = 0; i < LEN; i++) | |||
data[i] = ((i * i) >> 3) + 123 * i; | |||
if (argc > 1 && !strcmp(argv[1], "-t")) { | |||
for (i = 0; i < 1000; i++) { | |||
START_TIMER; | |||
checksum = av_adler32_update(1, data, LEN); | |||
STOP_TIMER("adler"); | |||
} | |||
} else { | |||
checksum = av_adler32_update(1, data, LEN); | |||
} | |||
av_log(NULL, AV_LOG_DEBUG, "%X (expected 50E6E508)\n", checksum); | |||
return checksum == 0x50e6e508 ? 0 : 1; | |||
} | |||
#endif |
@@ -0,0 +1,92 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include "aes.c" | |||
#include <string.h> | |||
#include "lfg.h" | |||
#include "log.h" | |||
int main(int argc, char **argv) | |||
{ | |||
int i, j; | |||
AVAES b; | |||
uint8_t rkey[2][16] = { | |||
{ 0 }, | |||
{ 0x10, 0xa5, 0x88, 0x69, 0xd7, 0x4b, 0xe5, 0xa3, | |||
0x74, 0xcf, 0x86, 0x7c, 0xfb, 0x47, 0x38, 0x59 } | |||
}; | |||
uint8_t pt[16], rpt[2][16] = { | |||
{ 0x6a, 0x84, 0x86, 0x7c, 0xd7, 0x7e, 0x12, 0xad, | |||
0x07, 0xea, 0x1b, 0xe8, 0x95, 0xc5, 0x3f, 0xa3 }, | |||
{ 0 } | |||
}; | |||
uint8_t rct[2][16] = { | |||
{ 0x73, 0x22, 0x81, 0xc0, 0xa0, 0xaa, 0xb8, 0xf7, | |||
0xa5, 0x4a, 0x0c, 0x67, 0xa0, 0xc4, 0x5e, 0xcf }, | |||
{ 0x6d, 0x25, 0x1e, 0x69, 0x44, 0xb0, 0x51, 0xe0, | |||
0x4e, 0xaa, 0x6f, 0xb4, 0xdb, 0xf7, 0x84, 0x65 } | |||
}; | |||
uint8_t temp[16]; | |||
int err = 0; | |||
av_log_set_level(AV_LOG_DEBUG); | |||
for (i = 0; i < 2; i++) { | |||
av_aes_init(&b, rkey[i], 128, 1); | |||
av_aes_crypt(&b, temp, rct[i], 1, NULL, 1); | |||
for (j = 0; j < 16; j++) { | |||
if (rpt[i][j] != temp[j]) { | |||
av_log(NULL, AV_LOG_ERROR, "%d %02X %02X\n", | |||
j, rpt[i][j], temp[j]); | |||
err = 1; | |||
} | |||
} | |||
} | |||
if (argc > 1 && !strcmp(argv[1], "-t")) { | |||
AVAES ae, ad; | |||
AVLFG prng; | |||
av_aes_init(&ae, "PI=3.141592654..", 128, 0); | |||
av_aes_init(&ad, "PI=3.141592654..", 128, 1); | |||
av_lfg_init(&prng, 1); | |||
for (i = 0; i < 10000; i++) { | |||
for (j = 0; j < 16; j++) | |||
pt[j] = av_lfg_get(&prng); | |||
{ | |||
START_TIMER; | |||
av_aes_crypt(&ae, temp, pt, 1, NULL, 0); | |||
if (!(i & (i - 1))) | |||
av_log(NULL, AV_LOG_ERROR, "%02X %02X %02X %02X\n", | |||
temp[0], temp[5], temp[10], temp[15]); | |||
av_aes_crypt(&ad, temp, temp, 1, NULL, 1); | |||
STOP_TIMER("aes"); | |||
} | |||
for (j = 0; j < 16; j++) { | |||
if (pt[j] != temp[j]) { | |||
av_log(NULL, AV_LOG_ERROR, "%d %d %02X %02X\n", | |||
i, j, pt[j], temp[j]); | |||
} | |||
} | |||
} | |||
} | |||
return err; | |||
} |
@@ -262,78 +262,3 @@ int av_aes_init(AVAES *a, const uint8_t *key, int key_bits, int decrypt) | |||
return 0; | |||
} | |||
#ifdef TEST | |||
#include <string.h> | |||
#include "lfg.h" | |||
#include "log.h" | |||
int main(int argc, char **argv) | |||
{ | |||
int i, j; | |||
AVAES b; | |||
uint8_t rkey[2][16] = { | |||
{ 0 }, | |||
{ 0x10, 0xa5, 0x88, 0x69, 0xd7, 0x4b, 0xe5, 0xa3, | |||
0x74, 0xcf, 0x86, 0x7c, 0xfb, 0x47, 0x38, 0x59 } | |||
}; | |||
uint8_t pt[16], rpt[2][16] = { | |||
{ 0x6a, 0x84, 0x86, 0x7c, 0xd7, 0x7e, 0x12, 0xad, | |||
0x07, 0xea, 0x1b, 0xe8, 0x95, 0xc5, 0x3f, 0xa3 }, | |||
{ 0 } | |||
}; | |||
uint8_t rct[2][16] = { | |||
{ 0x73, 0x22, 0x81, 0xc0, 0xa0, 0xaa, 0xb8, 0xf7, | |||
0xa5, 0x4a, 0x0c, 0x67, 0xa0, 0xc4, 0x5e, 0xcf }, | |||
{ 0x6d, 0x25, 0x1e, 0x69, 0x44, 0xb0, 0x51, 0xe0, | |||
0x4e, 0xaa, 0x6f, 0xb4, 0xdb, 0xf7, 0x84, 0x65 } | |||
}; | |||
uint8_t temp[16]; | |||
int err = 0; | |||
av_log_set_level(AV_LOG_DEBUG); | |||
for (i = 0; i < 2; i++) { | |||
av_aes_init(&b, rkey[i], 128, 1); | |||
av_aes_crypt(&b, temp, rct[i], 1, NULL, 1); | |||
for (j = 0; j < 16; j++) { | |||
if (rpt[i][j] != temp[j]) { | |||
av_log(NULL, AV_LOG_ERROR, "%d %02X %02X\n", | |||
j, rpt[i][j], temp[j]); | |||
err = 1; | |||
} | |||
} | |||
} | |||
if (argc > 1 && !strcmp(argv[1], "-t")) { | |||
AVAES ae, ad; | |||
AVLFG prng; | |||
av_aes_init(&ae, "PI=3.141592654..", 128, 0); | |||
av_aes_init(&ad, "PI=3.141592654..", 128, 1); | |||
av_lfg_init(&prng, 1); | |||
for (i = 0; i < 10000; i++) { | |||
for (j = 0; j < 16; j++) | |||
pt[j] = av_lfg_get(&prng); | |||
{ | |||
START_TIMER; | |||
av_aes_crypt(&ae, temp, pt, 1, NULL, 0); | |||
if (!(i & (i - 1))) | |||
av_log(NULL, AV_LOG_ERROR, "%02X %02X %02X %02X\n", | |||
temp[0], temp[5], temp[10], temp[15]); | |||
av_aes_crypt(&ad, temp, temp, 1, NULL, 1); | |||
STOP_TIMER("aes"); | |||
} | |||
for (j = 0; j < 16; j++) { | |||
if (pt[j] != temp[j]) { | |||
av_log(NULL, AV_LOG_ERROR, "%d %d %02X %02X\n", | |||
i, j, pt[j], temp[j]); | |||
} | |||
} | |||
} | |||
} | |||
return err; | |||
} | |||
#endif |
@@ -0,0 +1,35 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <assert.h> | |||
#include "atomic.h" | |||
int main(void) | |||
{ | |||
volatile int val = 1; | |||
int res; | |||
res = avpriv_atomic_int_add_and_fetch(&val, 1); | |||
assert(res == 2); | |||
avpriv_atomic_int_set(&val, 3); | |||
res = avpriv_atomic_int_get(&val); | |||
assert(res == 3); | |||
return 0; | |||
} |
@@ -107,21 +107,3 @@ void *avpriv_atomic_ptr_cas(void * volatile *ptr, void *oldval, void *newval) | |||
#endif /* HAVE_PTHREADS */ | |||
#endif /* !HAVE_ATOMICS_NATIVE */ | |||
#ifdef TEST | |||
#include <assert.h> | |||
int main(void) | |||
{ | |||
volatile int val = 1; | |||
int res; | |||
res = avpriv_atomic_int_add_and_fetch(&val, 1); | |||
assert(res == 2); | |||
avpriv_atomic_int_set(&val, 3); | |||
res = avpriv_atomic_int_get(&val); | |||
assert(res == 3); | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,69 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <stdio.h> | |||
#include "common.h" | |||
#include "mem.h" | |||
#include "avstring.h" | |||
int main(void) | |||
{ | |||
int i; | |||
const char *strings[] = { | |||
"''", | |||
"", | |||
":", | |||
"\\", | |||
"'", | |||
" '' :", | |||
" '' '' :", | |||
"foo '' :", | |||
"'foo'", | |||
"foo ", | |||
" ' foo ' ", | |||
"foo\\", | |||
"foo': blah:blah", | |||
"foo\\: blah:blah", | |||
"foo\'", | |||
"'foo : ' :blahblah", | |||
"\\ :blah", | |||
" foo", | |||
" foo ", | |||
" foo \\ ", | |||
"foo ':blah", | |||
" foo bar : blahblah", | |||
"\\f\\o\\o", | |||
"'foo : \\ \\ ' : blahblah", | |||
"'\\fo\\o:': blahblah", | |||
"\\'fo\\o\\:': foo ' :blahblah" | |||
}; | |||
printf("Testing av_get_token()\n"); | |||
for (i = 0; i < FF_ARRAY_ELEMS(strings); i++) { | |||
const char *p = strings[i]; | |||
char *q; | |||
printf("|%s|", p); | |||
q = av_get_token(&p, ":"); | |||
printf(" -> |%s|", q); | |||
printf(" + |%s|\n", p); | |||
av_free(q); | |||
} | |||
return 0; | |||
} |
@@ -229,55 +229,3 @@ int av_match_name(const char *name, const char *names) | |||
} | |||
return !av_strcasecmp(name, names); | |||
} | |||
#ifdef TEST | |||
int main(void) | |||
{ | |||
int i; | |||
const char *strings[] = { | |||
"''", | |||
"", | |||
":", | |||
"\\", | |||
"'", | |||
" '' :", | |||
" '' '' :", | |||
"foo '' :", | |||
"'foo'", | |||
"foo ", | |||
" ' foo ' ", | |||
"foo\\", | |||
"foo': blah:blah", | |||
"foo\\: blah:blah", | |||
"foo\'", | |||
"'foo : ' :blahblah", | |||
"\\ :blah", | |||
" foo", | |||
" foo ", | |||
" foo \\ ", | |||
"foo ':blah", | |||
" foo bar : blahblah", | |||
"\\f\\o\\o", | |||
"'foo : \\ \\ ' : blahblah", | |||
"'\\fo\\o:': blahblah", | |||
"\\'fo\\o\\:': foo ' :blahblah" | |||
}; | |||
printf("Testing av_get_token()\n"); | |||
for (i = 0; i < FF_ARRAY_ELEMS(strings); i++) { | |||
const char *p = strings[i]; | |||
char *q; | |||
printf("|%s|", p); | |||
q = av_get_token(&p, ":"); | |||
printf(" -> |%s|", q); | |||
printf(" + |%s|\n", p); | |||
av_free(q); | |||
} | |||
return 0; | |||
} | |||
#endif /* TEST */ |
@@ -0,0 +1,84 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; 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 "common.h" | |||
#include "base64.h" | |||
#define MAX_DATA_SIZE 1024 | |||
#define MAX_ENCODED_SIZE 2048 | |||
static int test_encode_decode(const uint8_t *data, unsigned int data_size, | |||
const char *encoded_ref) | |||
{ | |||
char encoded[MAX_ENCODED_SIZE]; | |||
uint8_t data2[MAX_DATA_SIZE]; | |||
int data2_size, max_data2_size = MAX_DATA_SIZE; | |||
if (!av_base64_encode(encoded, MAX_ENCODED_SIZE, data, data_size)) { | |||
printf("Failed: cannot encode the input data\n"); | |||
return 1; | |||
} | |||
if (encoded_ref && strcmp(encoded, encoded_ref)) { | |||
printf("Failed: encoded string differs from reference\n" | |||
"Encoded:\n%s\nReference:\n%s\n", encoded, encoded_ref); | |||
return 1; | |||
} | |||
if ((data2_size = av_base64_decode(data2, encoded, max_data2_size)) < 0) { | |||
printf("Failed: cannot decode the encoded string\n" | |||
"Encoded:\n%s\n", encoded); | |||
return 1; | |||
} | |||
if (memcmp(data2, data, data_size)) { | |||
printf("Failed: encoded/decoded data differs from original data\n"); | |||
return 1; | |||
} | |||
printf("Passed!\n"); | |||
return 0; | |||
} | |||
int main(void) | |||
{ | |||
int i, error_count = 0; | |||
struct test { | |||
const uint8_t *data; | |||
const char *encoded_ref; | |||
} tests[] = { | |||
{ "", ""}, | |||
{ "1", "MQ=="}, | |||
{ "22", "MjI="}, | |||
{ "333", "MzMz"}, | |||
{ "4444", "NDQ0NA=="}, | |||
{ "55555", "NTU1NTU="}, | |||
{ "666666", "NjY2NjY2"}, | |||
{ "abc:def", "YWJjOmRlZg=="}, | |||
}; | |||
printf("Encoding/decoding tests\n"); | |||
for (i = 0; i < FF_ARRAY_ELEMS(tests); i++) | |||
error_count += test_encode_decode(tests[i].data, strlen(tests[i].data), tests[i].encoded_ref); | |||
if (error_count) | |||
printf("Error Count: %d.\n", error_count); | |||
return !!error_count; | |||
} |
@@ -98,68 +98,3 @@ char *av_base64_encode(char *out, int out_size, const uint8_t *in, int in_size) | |||
return ret; | |||
} | |||
#ifdef TEST | |||
#define MAX_DATA_SIZE 1024 | |||
#define MAX_ENCODED_SIZE 2048 | |||
static int test_encode_decode(const uint8_t *data, unsigned int data_size, | |||
const char *encoded_ref) | |||
{ | |||
char encoded[MAX_ENCODED_SIZE]; | |||
uint8_t data2[MAX_DATA_SIZE]; | |||
int data2_size, max_data2_size = MAX_DATA_SIZE; | |||
if (!av_base64_encode(encoded, MAX_ENCODED_SIZE, data, data_size)) { | |||
printf("Failed: cannot encode the input data\n"); | |||
return 1; | |||
} | |||
if (encoded_ref && strcmp(encoded, encoded_ref)) { | |||
printf("Failed: encoded string differs from reference\n" | |||
"Encoded:\n%s\nReference:\n%s\n", encoded, encoded_ref); | |||
return 1; | |||
} | |||
if ((data2_size = av_base64_decode(data2, encoded, max_data2_size)) < 0) { | |||
printf("Failed: cannot decode the encoded string\n" | |||
"Encoded:\n%s\n", encoded); | |||
return 1; | |||
} | |||
if (memcmp(data2, data, data_size)) { | |||
printf("Failed: encoded/decoded data differs from original data\n"); | |||
return 1; | |||
} | |||
printf("Passed!\n"); | |||
return 0; | |||
} | |||
int main(void) | |||
{ | |||
int i, error_count = 0; | |||
struct test { | |||
const uint8_t *data; | |||
const char *encoded_ref; | |||
} tests[] = { | |||
{ "", ""}, | |||
{ "1", "MQ=="}, | |||
{ "22", "MjI="}, | |||
{ "333", "MzMz"}, | |||
{ "4444", "NDQ0NA=="}, | |||
{ "55555", "NTU1NTU="}, | |||
{ "666666", "NjY2NjY2"}, | |||
{ "abc:def", "YWJjOmRlZg=="}, | |||
}; | |||
printf("Encoding/decoding tests\n"); | |||
for (i = 0; i < FF_ARRAY_ELEMS(tests); i++) | |||
error_count += test_encode_decode(tests[i].data, strlen(tests[i].data), tests[i].encoded_ref); | |||
if (error_count) | |||
printf("Error Count: %d.\n", error_count); | |||
return !!error_count; | |||
} | |||
#endif |
@@ -0,0 +1,190 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; 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 <stdlib.h> | |||
#include <string.h> | |||
#include "blowfish.h" | |||
#define NUM_VARIABLE_KEY_TESTS 34 | |||
/* plaintext bytes -- left halves */ | |||
static const uint32_t plaintext_l[NUM_VARIABLE_KEY_TESTS] = { | |||
0x00000000, 0xFFFFFFFF, 0x10000000, 0x11111111, 0x11111111, | |||
0x01234567, 0x00000000, 0x01234567, 0x01A1D6D0, 0x5CD54CA8, | |||
0x0248D438, 0x51454B58, 0x42FD4430, 0x059B5E08, 0x0756D8E0, | |||
0x762514B8, 0x3BDD1190, 0x26955F68, 0x164D5E40, 0x6B056E18, | |||
0x004BD6EF, 0x480D3900, 0x437540C8, 0x072D43A0, 0x02FE5577, | |||
0x1D9D5C50, 0x30553228, 0x01234567, 0x01234567, 0x01234567, | |||
0xFFFFFFFF, 0x00000000, 0x00000000, 0xFFFFFFFF | |||
}; | |||
/* plaintext bytes -- right halves */ | |||
static const uint32_t plaintext_r[NUM_VARIABLE_KEY_TESTS] = { | |||
0x00000000, 0xFFFFFFFF, 0x00000001, 0x11111111, 0x11111111, | |||
0x89ABCDEF, 0x00000000, 0x89ABCDEF, 0x39776742, 0x3DEF57DA, | |||
0x06F67172, 0x2DDF440A, 0x59577FA2, 0x51CF143A, 0x774761D2, | |||
0x29BF486A, 0x49372802, 0x35AF609A, 0x4F275232, 0x759F5CCA, | |||
0x09176062, 0x6EE762F2, 0x698F3CFA, 0x77075292, 0x8117F12A, | |||
0x18F728C2, 0x6D6F295A, 0x89ABCDEF, 0x89ABCDEF, 0x89ABCDEF, | |||
0xFFFFFFFF, 0x00000000, 0x00000000, 0xFFFFFFFF | |||
}; | |||
/* key bytes for variable key tests */ | |||
static const uint8_t variable_key[NUM_VARIABLE_KEY_TESTS][8] = { | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF }, | |||
{ 0x30, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11 }, | |||
{ 0x01, 0x23, 0x45, 0x67, 0x89, 0xAB, 0xCD, 0xEF }, | |||
{ 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11 }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0xFE, 0xDC, 0xBA, 0x98, 0x76, 0x54, 0x32, 0x10 }, | |||
{ 0x7C, 0xA1, 0x10, 0x45, 0x4A, 0x1A, 0x6E, 0x57 }, | |||
{ 0x01, 0x31, 0xD9, 0x61, 0x9D, 0xC1, 0x37, 0x6E }, | |||
{ 0x07, 0xA1, 0x13, 0x3E, 0x4A, 0x0B, 0x26, 0x86 }, | |||
{ 0x38, 0x49, 0x67, 0x4C, 0x26, 0x02, 0x31, 0x9E }, | |||
{ 0x04, 0xB9, 0x15, 0xBA, 0x43, 0xFE, 0xB5, 0xB6 }, | |||
{ 0x01, 0x13, 0xB9, 0x70, 0xFD, 0x34, 0xF2, 0xCE }, | |||
{ 0x01, 0x70, 0xF1, 0x75, 0x46, 0x8F, 0xB5, 0xE6 }, | |||
{ 0x43, 0x29, 0x7F, 0xAD, 0x38, 0xE3, 0x73, 0xFE }, | |||
{ 0x07, 0xA7, 0x13, 0x70, 0x45, 0xDA, 0x2A, 0x16 }, | |||
{ 0x04, 0x68, 0x91, 0x04, 0xC2, 0xFD, 0x3B, 0x2F }, | |||
{ 0x37, 0xD0, 0x6B, 0xB5, 0x16, 0xCB, 0x75, 0x46 }, | |||
{ 0x1F, 0x08, 0x26, 0x0D, 0x1A, 0xC2, 0x46, 0x5E }, | |||
{ 0x58, 0x40, 0x23, 0x64, 0x1A, 0xBA, 0x61, 0x76 }, | |||
{ 0x02, 0x58, 0x16, 0x16, 0x46, 0x29, 0xB0, 0x07 }, | |||
{ 0x49, 0x79, 0x3E, 0xBC, 0x79, 0xB3, 0x25, 0x8F }, | |||
{ 0x4F, 0xB0, 0x5E, 0x15, 0x15, 0xAB, 0x73, 0xA7 }, | |||
{ 0x49, 0xE9, 0x5D, 0x6D, 0x4C, 0xA2, 0x29, 0xBF }, | |||
{ 0x01, 0x83, 0x10, 0xDC, 0x40, 0x9B, 0x26, 0xD6 }, | |||
{ 0x1C, 0x58, 0x7F, 0x1C, 0x13, 0x92, 0x4F, 0xEF }, | |||
{ 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01 }, | |||
{ 0x1F, 0x1F, 0x1F, 0x1F, 0x0E, 0x0E, 0x0E, 0x0E }, | |||
{ 0xE0, 0xFE, 0xE0, 0xFE, 0xF1, 0xFE, 0xF1, 0xFE }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF }, | |||
{ 0x01, 0x23, 0x45, 0x67, 0x89, 0xAB, 0xCD, 0xEF }, | |||
{ 0xFE, 0xDC, 0xBA, 0x98, 0x76, 0x54, 0x32, 0x10 } | |||
}; | |||
/* ciphertext bytes -- left halves */ | |||
static const uint32_t ciphertext_l[NUM_VARIABLE_KEY_TESTS] = { | |||
0x4EF99745, 0x51866FD5, 0x7D856F9A, 0x2466DD87, 0x61F9C380, | |||
0x7D0CC630, 0x4EF99745, 0x0ACEAB0F, 0x59C68245, 0xB1B8CC0B, | |||
0x1730E577, 0xA25E7856, 0x353882B1, 0x48F4D088, 0x432193B7, | |||
0x13F04154, 0x2EEDDA93, 0xD887E039, 0x5F99D04F, 0x4A057A3B, | |||
0x452031C1, 0x7555AE39, 0x53C55F9C, 0x7A8E7BFA, 0xCF9C5D7A, | |||
0xD1ABB290, 0x55CB3774, 0xFA34EC48, 0xA7907951, 0xC39E072D, | |||
0x014933E0, 0xF21E9A77, 0x24594688, 0x6B5C5A9C | |||
}; | |||
/* ciphertext bytes -- right halves */ | |||
static const uint32_t ciphertext_r[NUM_VARIABLE_KEY_TESTS] = { | |||
0x6198DD78, 0xB85ECB8A, 0x613063F2, 0x8B963C9D, 0x2281B096, | |||
0xAFDA1EC7, 0x6198DD78, 0xC6A0A28D, 0xEB05282B, 0x250F09A0, | |||
0x8BEA1DA4, 0xCF2651EB, 0x09CE8F1A, 0x4C379918, 0x8951FC98, | |||
0xD69D1AE5, 0xFFD39C79, 0x3C2DA6E3, 0x5B163969, 0x24D3977B, | |||
0xE4FADA8E, 0xF59B87BD, 0xB49FC019, 0x937E89A3, 0x4986ADB5, | |||
0x658BC778, 0xD13EF201, 0x47B268B2, 0x08EA3CAE, 0x9FAC631D, | |||
0xCDAFF6E4, 0xB71C49BC, 0x5754369A, 0x5D9E0A5A | |||
}; | |||
/* plaintext bytes */ | |||
static const uint8_t plaintext[8] = "BLOWFISH"; | |||
static const uint8_t plaintext2[16] = "BLOWFISHBLOWFISH"; | |||
/* ciphertext bytes */ | |||
static const uint8_t ciphertext[8] = { | |||
0x32, 0x4E, 0xD0, 0xFE, 0xF4, 0x13, 0xA2, 0x03 | |||
}; | |||
static const uint8_t ciphertext2[16] = { | |||
0x53, 0x00, 0x40, 0x06, 0x63, 0xf2, 0x1d, 0x99, | |||
0x3b, 0x9b, 0x27, 0x64, 0x46, 0xfd, 0x20, 0xc1, | |||
}; | |||
#define IV "blowfish" | |||
static void test_blowfish(AVBlowfish *ctx, uint8_t *dst, const uint8_t *src, | |||
const uint8_t *ref, int len, uint8_t *iv, int dir, | |||
const char *test) | |||
{ | |||
av_blowfish_crypt(ctx, dst, src, len, iv, dir); | |||
if (memcmp(dst, ref, 8*len)) { | |||
int i; | |||
printf("%s failed\ngot ", test); | |||
for (i = 0; i < 8*len; i++) | |||
printf("%02x ", dst[i]); | |||
printf("\nexpected "); | |||
for (i = 0; i < 8*len; i++) | |||
printf("%02x ", ref[i]); | |||
printf("\n"); | |||
exit(1); | |||
} | |||
} | |||
int main(void) | |||
{ | |||
AVBlowfish ctx; | |||
uint32_t tmptext_l[NUM_VARIABLE_KEY_TESTS]; | |||
uint32_t tmptext_r[NUM_VARIABLE_KEY_TESTS]; | |||
uint8_t tmp[16], iv[8]; | |||
int i; | |||
av_blowfish_init(&ctx, "abcdefghijklmnopqrstuvwxyz", 26); | |||
test_blowfish(&ctx, tmp, plaintext, ciphertext, 1, NULL, 0, "encryption"); | |||
test_blowfish(&ctx, tmp, ciphertext, plaintext, 1, NULL, 1, "decryption"); | |||
test_blowfish(&ctx, tmp, tmp, ciphertext, 1, NULL, 0, "Inplace encryption"); | |||
test_blowfish(&ctx, tmp, tmp, plaintext, 1, NULL, 1, "Inplace decryption"); | |||
memcpy(iv, IV, 8); | |||
test_blowfish(&ctx, tmp, plaintext2, ciphertext2, 2, iv, 0, "CBC encryption"); | |||
memcpy(iv, IV, 8); | |||
test_blowfish(&ctx, tmp, ciphertext2, plaintext2, 2, iv, 1, "CBC decryption"); | |||
memcpy(iv, IV, 8); | |||
test_blowfish(&ctx, tmp, tmp, ciphertext2, 2, iv, 0, "Inplace CBC encryption"); | |||
memcpy(iv, IV, 8); | |||
test_blowfish(&ctx, tmp, tmp, plaintext2, 2, iv, 1, "Inplace CBC decryption"); | |||
memcpy(tmptext_l, plaintext_l, sizeof(*plaintext_l) * NUM_VARIABLE_KEY_TESTS); | |||
memcpy(tmptext_r, plaintext_r, sizeof(*plaintext_r) * NUM_VARIABLE_KEY_TESTS); | |||
for (i = 0; i < NUM_VARIABLE_KEY_TESTS; i++) { | |||
av_blowfish_init(&ctx, variable_key[i], 8); | |||
av_blowfish_crypt_ecb(&ctx, &tmptext_l[i], &tmptext_r[i], 0); | |||
if (tmptext_l[i] != ciphertext_l[i] || tmptext_r[i] != ciphertext_r[i]) { | |||
printf("Test encryption failed.\n"); | |||
return 1; | |||
} | |||
av_blowfish_crypt_ecb(&ctx, &tmptext_l[i], &tmptext_r[i], 1); | |||
if (tmptext_l[i] != plaintext_l[i] || tmptext_r[i] != plaintext_r[i]) { | |||
printf("Test decryption failed.\n"); | |||
return 1; | |||
} | |||
} | |||
printf("Test encryption/decryption success.\n"); | |||
return 0; | |||
} | |||
@@ -439,173 +439,3 @@ void av_blowfish_crypt(AVBlowfish *ctx, uint8_t *dst, const uint8_t *src, | |||
} | |||
} | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
#define NUM_VARIABLE_KEY_TESTS 34 | |||
/* plaintext bytes -- left halves */ | |||
static const uint32_t plaintext_l[NUM_VARIABLE_KEY_TESTS] = { | |||
0x00000000, 0xFFFFFFFF, 0x10000000, 0x11111111, 0x11111111, | |||
0x01234567, 0x00000000, 0x01234567, 0x01A1D6D0, 0x5CD54CA8, | |||
0x0248D438, 0x51454B58, 0x42FD4430, 0x059B5E08, 0x0756D8E0, | |||
0x762514B8, 0x3BDD1190, 0x26955F68, 0x164D5E40, 0x6B056E18, | |||
0x004BD6EF, 0x480D3900, 0x437540C8, 0x072D43A0, 0x02FE5577, | |||
0x1D9D5C50, 0x30553228, 0x01234567, 0x01234567, 0x01234567, | |||
0xFFFFFFFF, 0x00000000, 0x00000000, 0xFFFFFFFF | |||
}; | |||
/* plaintext bytes -- right halves */ | |||
static const uint32_t plaintext_r[NUM_VARIABLE_KEY_TESTS] = { | |||
0x00000000, 0xFFFFFFFF, 0x00000001, 0x11111111, 0x11111111, | |||
0x89ABCDEF, 0x00000000, 0x89ABCDEF, 0x39776742, 0x3DEF57DA, | |||
0x06F67172, 0x2DDF440A, 0x59577FA2, 0x51CF143A, 0x774761D2, | |||
0x29BF486A, 0x49372802, 0x35AF609A, 0x4F275232, 0x759F5CCA, | |||
0x09176062, 0x6EE762F2, 0x698F3CFA, 0x77075292, 0x8117F12A, | |||
0x18F728C2, 0x6D6F295A, 0x89ABCDEF, 0x89ABCDEF, 0x89ABCDEF, | |||
0xFFFFFFFF, 0x00000000, 0x00000000, 0xFFFFFFFF | |||
}; | |||
/* key bytes for variable key tests */ | |||
static const uint8_t variable_key[NUM_VARIABLE_KEY_TESTS][8] = { | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF }, | |||
{ 0x30, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11 }, | |||
{ 0x01, 0x23, 0x45, 0x67, 0x89, 0xAB, 0xCD, 0xEF }, | |||
{ 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11 }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0xFE, 0xDC, 0xBA, 0x98, 0x76, 0x54, 0x32, 0x10 }, | |||
{ 0x7C, 0xA1, 0x10, 0x45, 0x4A, 0x1A, 0x6E, 0x57 }, | |||
{ 0x01, 0x31, 0xD9, 0x61, 0x9D, 0xC1, 0x37, 0x6E }, | |||
{ 0x07, 0xA1, 0x13, 0x3E, 0x4A, 0x0B, 0x26, 0x86 }, | |||
{ 0x38, 0x49, 0x67, 0x4C, 0x26, 0x02, 0x31, 0x9E }, | |||
{ 0x04, 0xB9, 0x15, 0xBA, 0x43, 0xFE, 0xB5, 0xB6 }, | |||
{ 0x01, 0x13, 0xB9, 0x70, 0xFD, 0x34, 0xF2, 0xCE }, | |||
{ 0x01, 0x70, 0xF1, 0x75, 0x46, 0x8F, 0xB5, 0xE6 }, | |||
{ 0x43, 0x29, 0x7F, 0xAD, 0x38, 0xE3, 0x73, 0xFE }, | |||
{ 0x07, 0xA7, 0x13, 0x70, 0x45, 0xDA, 0x2A, 0x16 }, | |||
{ 0x04, 0x68, 0x91, 0x04, 0xC2, 0xFD, 0x3B, 0x2F }, | |||
{ 0x37, 0xD0, 0x6B, 0xB5, 0x16, 0xCB, 0x75, 0x46 }, | |||
{ 0x1F, 0x08, 0x26, 0x0D, 0x1A, 0xC2, 0x46, 0x5E }, | |||
{ 0x58, 0x40, 0x23, 0x64, 0x1A, 0xBA, 0x61, 0x76 }, | |||
{ 0x02, 0x58, 0x16, 0x16, 0x46, 0x29, 0xB0, 0x07 }, | |||
{ 0x49, 0x79, 0x3E, 0xBC, 0x79, 0xB3, 0x25, 0x8F }, | |||
{ 0x4F, 0xB0, 0x5E, 0x15, 0x15, 0xAB, 0x73, 0xA7 }, | |||
{ 0x49, 0xE9, 0x5D, 0x6D, 0x4C, 0xA2, 0x29, 0xBF }, | |||
{ 0x01, 0x83, 0x10, 0xDC, 0x40, 0x9B, 0x26, 0xD6 }, | |||
{ 0x1C, 0x58, 0x7F, 0x1C, 0x13, 0x92, 0x4F, 0xEF }, | |||
{ 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01 }, | |||
{ 0x1F, 0x1F, 0x1F, 0x1F, 0x0E, 0x0E, 0x0E, 0x0E }, | |||
{ 0xE0, 0xFE, 0xE0, 0xFE, 0xF1, 0xFE, 0xF1, 0xFE }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF }, | |||
{ 0x01, 0x23, 0x45, 0x67, 0x89, 0xAB, 0xCD, 0xEF }, | |||
{ 0xFE, 0xDC, 0xBA, 0x98, 0x76, 0x54, 0x32, 0x10 } | |||
}; | |||
/* ciphertext bytes -- left halves */ | |||
static const uint32_t ciphertext_l[NUM_VARIABLE_KEY_TESTS] = { | |||
0x4EF99745, 0x51866FD5, 0x7D856F9A, 0x2466DD87, 0x61F9C380, | |||
0x7D0CC630, 0x4EF99745, 0x0ACEAB0F, 0x59C68245, 0xB1B8CC0B, | |||
0x1730E577, 0xA25E7856, 0x353882B1, 0x48F4D088, 0x432193B7, | |||
0x13F04154, 0x2EEDDA93, 0xD887E039, 0x5F99D04F, 0x4A057A3B, | |||
0x452031C1, 0x7555AE39, 0x53C55F9C, 0x7A8E7BFA, 0xCF9C5D7A, | |||
0xD1ABB290, 0x55CB3774, 0xFA34EC48, 0xA7907951, 0xC39E072D, | |||
0x014933E0, 0xF21E9A77, 0x24594688, 0x6B5C5A9C | |||
}; | |||
/* ciphertext bytes -- right halves */ | |||
static const uint32_t ciphertext_r[NUM_VARIABLE_KEY_TESTS] = { | |||
0x6198DD78, 0xB85ECB8A, 0x613063F2, 0x8B963C9D, 0x2281B096, | |||
0xAFDA1EC7, 0x6198DD78, 0xC6A0A28D, 0xEB05282B, 0x250F09A0, | |||
0x8BEA1DA4, 0xCF2651EB, 0x09CE8F1A, 0x4C379918, 0x8951FC98, | |||
0xD69D1AE5, 0xFFD39C79, 0x3C2DA6E3, 0x5B163969, 0x24D3977B, | |||
0xE4FADA8E, 0xF59B87BD, 0xB49FC019, 0x937E89A3, 0x4986ADB5, | |||
0x658BC778, 0xD13EF201, 0x47B268B2, 0x08EA3CAE, 0x9FAC631D, | |||
0xCDAFF6E4, 0xB71C49BC, 0x5754369A, 0x5D9E0A5A | |||
}; | |||
/* plaintext bytes */ | |||
static const uint8_t plaintext[8] = "BLOWFISH"; | |||
static const uint8_t plaintext2[16] = "BLOWFISHBLOWFISH"; | |||
/* ciphertext bytes */ | |||
static const uint8_t ciphertext[8] = { | |||
0x32, 0x4E, 0xD0, 0xFE, 0xF4, 0x13, 0xA2, 0x03 | |||
}; | |||
static const uint8_t ciphertext2[16] = { | |||
0x53, 0x00, 0x40, 0x06, 0x63, 0xf2, 0x1d, 0x99, | |||
0x3b, 0x9b, 0x27, 0x64, 0x46, 0xfd, 0x20, 0xc1, | |||
}; | |||
#define IV "blowfish" | |||
static void test_blowfish(AVBlowfish *ctx, uint8_t *dst, const uint8_t *src, | |||
const uint8_t *ref, int len, uint8_t *iv, int dir, | |||
const char *test) | |||
{ | |||
av_blowfish_crypt(ctx, dst, src, len, iv, dir); | |||
if (memcmp(dst, ref, 8*len)) { | |||
int i; | |||
printf("%s failed\ngot ", test); | |||
for (i = 0; i < 8*len; i++) | |||
printf("%02x ", dst[i]); | |||
printf("\nexpected "); | |||
for (i = 0; i < 8*len; i++) | |||
printf("%02x ", ref[i]); | |||
printf("\n"); | |||
exit(1); | |||
} | |||
} | |||
int main(void) | |||
{ | |||
AVBlowfish ctx; | |||
uint32_t tmptext_l[NUM_VARIABLE_KEY_TESTS]; | |||
uint32_t tmptext_r[NUM_VARIABLE_KEY_TESTS]; | |||
uint8_t tmp[16], iv[8]; | |||
int i; | |||
av_blowfish_init(&ctx, "abcdefghijklmnopqrstuvwxyz", 26); | |||
test_blowfish(&ctx, tmp, plaintext, ciphertext, 1, NULL, 0, "encryption"); | |||
test_blowfish(&ctx, tmp, ciphertext, plaintext, 1, NULL, 1, "decryption"); | |||
test_blowfish(&ctx, tmp, tmp, ciphertext, 1, NULL, 0, "Inplace encryption"); | |||
test_blowfish(&ctx, tmp, tmp, plaintext, 1, NULL, 1, "Inplace decryption"); | |||
memcpy(iv, IV, 8); | |||
test_blowfish(&ctx, tmp, plaintext2, ciphertext2, 2, iv, 0, "CBC encryption"); | |||
memcpy(iv, IV, 8); | |||
test_blowfish(&ctx, tmp, ciphertext2, plaintext2, 2, iv, 1, "CBC decryption"); | |||
memcpy(iv, IV, 8); | |||
test_blowfish(&ctx, tmp, tmp, ciphertext2, 2, iv, 0, "Inplace CBC encryption"); | |||
memcpy(iv, IV, 8); | |||
test_blowfish(&ctx, tmp, tmp, plaintext2, 2, iv, 1, "Inplace CBC decryption"); | |||
memcpy(tmptext_l, plaintext_l, sizeof(*plaintext_l) * NUM_VARIABLE_KEY_TESTS); | |||
memcpy(tmptext_r, plaintext_r, sizeof(*plaintext_r) * NUM_VARIABLE_KEY_TESTS); | |||
for (i = 0; i < NUM_VARIABLE_KEY_TESTS; i++) { | |||
av_blowfish_init(&ctx, variable_key[i], 8); | |||
av_blowfish_crypt_ecb(&ctx, &tmptext_l[i], &tmptext_r[i], 0); | |||
if (tmptext_l[i] != ciphertext_l[i] || tmptext_r[i] != ciphertext_r[i]) { | |||
printf("Test encryption failed.\n"); | |||
return 1; | |||
} | |||
av_blowfish_crypt_ecb(&ctx, &tmptext_l[i], &tmptext_r[i], 1); | |||
if (tmptext_l[i] != plaintext_l[i] || tmptext_r[i] != plaintext_r[i]) { | |||
printf("Test decryption failed.\n"); | |||
return 1; | |||
} | |||
} | |||
printf("Test encryption/decryption success.\n"); | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,136 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include "config.h" | |||
#if HAVE_UNISTD_H | |||
#include <unistd.h> | |||
#elif !HAVE_GETOPT | |||
#include "compat/getopt.c" | |||
#endif | |||
#include <stdint.h> | |||
#include <stdio.h> | |||
#include "avstring.h" | |||
#include "common.h" | |||
#include "cpu.h" | |||
static const struct { | |||
int flag; | |||
const char *name; | |||
} cpu_flag_tab[] = { | |||
#if ARCH_AARCH64 | |||
{ AV_CPU_FLAG_ARMV8, "armv8" }, | |||
{ AV_CPU_FLAG_NEON, "neon" }, | |||
{ AV_CPU_FLAG_VFP, "vfp" }, | |||
#elif ARCH_ARM | |||
{ AV_CPU_FLAG_ARMV5TE, "armv5te" }, | |||
{ AV_CPU_FLAG_ARMV6, "armv6" }, | |||
{ AV_CPU_FLAG_ARMV6T2, "armv6t2" }, | |||
{ AV_CPU_FLAG_VFP, "vfp" }, | |||
{ AV_CPU_FLAG_VFP_VM, "vfp_vm" }, | |||
{ AV_CPU_FLAG_VFPV3, "vfpv3" }, | |||
{ AV_CPU_FLAG_NEON, "neon" }, | |||
#elif ARCH_PPC | |||
{ AV_CPU_FLAG_ALTIVEC, "altivec" }, | |||
#elif ARCH_X86 | |||
{ AV_CPU_FLAG_MMX, "mmx" }, | |||
{ AV_CPU_FLAG_MMXEXT, "mmxext" }, | |||
{ AV_CPU_FLAG_SSE, "sse" }, | |||
{ AV_CPU_FLAG_SSE2, "sse2" }, | |||
{ AV_CPU_FLAG_SSE2SLOW, "sse2(slow)" }, | |||
{ AV_CPU_FLAG_SSE3, "sse3" }, | |||
{ AV_CPU_FLAG_SSE3SLOW, "sse3(slow)" }, | |||
{ AV_CPU_FLAG_SSSE3, "ssse3" }, | |||
{ AV_CPU_FLAG_ATOM, "atom" }, | |||
{ AV_CPU_FLAG_SSE4, "sse4.1" }, | |||
{ AV_CPU_FLAG_SSE42, "sse4.2" }, | |||
{ AV_CPU_FLAG_AVX, "avx" }, | |||
{ AV_CPU_FLAG_AVXSLOW, "avxslow" }, | |||
{ AV_CPU_FLAG_XOP, "xop" }, | |||
{ AV_CPU_FLAG_FMA3, "fma3" }, | |||
{ AV_CPU_FLAG_FMA4, "fma4" }, | |||
{ AV_CPU_FLAG_3DNOW, "3dnow" }, | |||
{ AV_CPU_FLAG_3DNOWEXT, "3dnowext" }, | |||
{ AV_CPU_FLAG_CMOV, "cmov" }, | |||
{ AV_CPU_FLAG_AVX2, "avx2" }, | |||
{ AV_CPU_FLAG_BMI1, "bmi1" }, | |||
{ AV_CPU_FLAG_BMI2, "bmi2" }, | |||
#endif | |||
{ 0 } | |||
}; | |||
static void print_cpu_flags(int cpu_flags, const char *type) | |||
{ | |||
int i; | |||
fprintf(stderr, "cpu_flags(%s) = 0x%08X\n", type, cpu_flags); | |||
fprintf(stderr, "cpu_flags_str(%s) =", type); | |||
for (i = 0; cpu_flag_tab[i].flag; i++) | |||
if (cpu_flags & cpu_flag_tab[i].flag) | |||
fprintf(stderr, " %s", cpu_flag_tab[i].name); | |||
fprintf(stderr, "\n"); | |||
} | |||
int main(int argc, char **argv) | |||
{ | |||
int cpu_flags_raw = av_get_cpu_flags(); | |||
int cpu_flags_eff; | |||
int cpu_count = av_cpu_count(); | |||
char threads[5] = "auto"; | |||
if (cpu_flags_raw < 0) | |||
return 1; | |||
for (;;) { | |||
int c = getopt(argc, argv, "c:t:"); | |||
if (c == -1) | |||
break; | |||
switch (c) { | |||
case 'c': | |||
{ | |||
int cpuflags = av_parse_cpu_flags(optarg); | |||
if (cpuflags < 0) | |||
return 2; | |||
av_set_cpu_flags_mask(cpuflags); | |||
break; | |||
} | |||
case 't': | |||
{ | |||
int len = av_strlcpy(threads, optarg, sizeof(threads)); | |||
if (len >= sizeof(threads)) { | |||
fprintf(stderr, "Invalid thread count '%s'\n", optarg); | |||
return 2; | |||
} | |||
} | |||
} | |||
} | |||
cpu_flags_eff = av_get_cpu_flags(); | |||
if (cpu_flags_eff < 0) | |||
return 3; | |||
print_cpu_flags(cpu_flags_raw, "raw"); | |||
print_cpu_flags(cpu_flags_eff, "effective"); | |||
fprintf(stderr, "threads = %s (cpu_count = %d)\n", threads, cpu_count); | |||
return 0; | |||
} |
@@ -178,122 +178,3 @@ int av_cpu_count(void) | |||
return nb_cpus; | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
#if HAVE_UNISTD_H | |||
#include <unistd.h> | |||
#endif | |||
#include "avstring.h" | |||
#if !HAVE_GETOPT | |||
#include "compat/getopt.c" | |||
#endif | |||
static const struct { | |||
int flag; | |||
const char *name; | |||
} cpu_flag_tab[] = { | |||
#if ARCH_AARCH64 | |||
{ AV_CPU_FLAG_ARMV8, "armv8" }, | |||
{ AV_CPU_FLAG_NEON, "neon" }, | |||
{ AV_CPU_FLAG_VFP, "vfp" }, | |||
#elif ARCH_ARM | |||
{ AV_CPU_FLAG_ARMV5TE, "armv5te" }, | |||
{ AV_CPU_FLAG_ARMV6, "armv6" }, | |||
{ AV_CPU_FLAG_ARMV6T2, "armv6t2" }, | |||
{ AV_CPU_FLAG_VFP, "vfp" }, | |||
{ AV_CPU_FLAG_VFP_VM, "vfp_vm" }, | |||
{ AV_CPU_FLAG_VFPV3, "vfpv3" }, | |||
{ AV_CPU_FLAG_NEON, "neon" }, | |||
#elif ARCH_PPC | |||
{ AV_CPU_FLAG_ALTIVEC, "altivec" }, | |||
#elif ARCH_X86 | |||
{ AV_CPU_FLAG_MMX, "mmx" }, | |||
{ AV_CPU_FLAG_MMXEXT, "mmxext" }, | |||
{ AV_CPU_FLAG_SSE, "sse" }, | |||
{ AV_CPU_FLAG_SSE2, "sse2" }, | |||
{ AV_CPU_FLAG_SSE2SLOW, "sse2(slow)" }, | |||
{ AV_CPU_FLAG_SSE3, "sse3" }, | |||
{ AV_CPU_FLAG_SSE3SLOW, "sse3(slow)" }, | |||
{ AV_CPU_FLAG_SSSE3, "ssse3" }, | |||
{ AV_CPU_FLAG_ATOM, "atom" }, | |||
{ AV_CPU_FLAG_SSE4, "sse4.1" }, | |||
{ AV_CPU_FLAG_SSE42, "sse4.2" }, | |||
{ AV_CPU_FLAG_AVX, "avx" }, | |||
{ AV_CPU_FLAG_AVXSLOW, "avxslow" }, | |||
{ AV_CPU_FLAG_XOP, "xop" }, | |||
{ AV_CPU_FLAG_FMA3, "fma3" }, | |||
{ AV_CPU_FLAG_FMA4, "fma4" }, | |||
{ AV_CPU_FLAG_3DNOW, "3dnow" }, | |||
{ AV_CPU_FLAG_3DNOWEXT, "3dnowext" }, | |||
{ AV_CPU_FLAG_CMOV, "cmov" }, | |||
{ AV_CPU_FLAG_AVX2, "avx2" }, | |||
{ AV_CPU_FLAG_BMI1, "bmi1" }, | |||
{ AV_CPU_FLAG_BMI2, "bmi2" }, | |||
#endif | |||
{ 0 } | |||
}; | |||
static void print_cpu_flags(int cpu_flags, const char *type) | |||
{ | |||
int i; | |||
fprintf(stderr, "cpu_flags(%s) = 0x%08X\n", type, cpu_flags); | |||
fprintf(stderr, "cpu_flags_str(%s) =", type); | |||
for (i = 0; cpu_flag_tab[i].flag; i++) | |||
if (cpu_flags & cpu_flag_tab[i].flag) | |||
fprintf(stderr, " %s", cpu_flag_tab[i].name); | |||
fprintf(stderr, "\n"); | |||
} | |||
int main(int argc, char **argv) | |||
{ | |||
int cpu_flags_raw = av_get_cpu_flags(); | |||
int cpu_flags_eff; | |||
int cpu_count = av_cpu_count(); | |||
char threads[5] = "auto"; | |||
if (cpu_flags_raw < 0) | |||
return 1; | |||
for (;;) { | |||
int c = getopt(argc, argv, "c:t:"); | |||
if (c == -1) | |||
break; | |||
switch (c) { | |||
case 'c': | |||
{ | |||
int cpuflags = av_parse_cpu_flags(optarg); | |||
if (cpuflags < 0) | |||
return 2; | |||
av_set_cpu_flags_mask(cpuflags); | |||
break; | |||
} | |||
case 't': | |||
{ | |||
int len = av_strlcpy(threads, optarg, sizeof(threads)); | |||
if (len >= sizeof(threads)) { | |||
fprintf(stderr, "Invalid thread count '%s'\n", optarg); | |||
return 2; | |||
} | |||
} | |||
} | |||
} | |||
cpu_flags_eff = av_get_cpu_flags(); | |||
if (cpu_flags_eff < 0) | |||
return 3; | |||
print_cpu_flags(cpu_flags_raw, "raw"); | |||
print_cpu_flags(cpu_flags_eff, "effective"); | |||
fprintf(stderr, "threads = %s (cpu_count = %d)\n", threads, cpu_count); | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,45 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; 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 "crc.h" | |||
int main(void) | |||
{ | |||
uint8_t buf[1999]; | |||
int i; | |||
int p[5][3] = { | |||
{ AV_CRC_32_IEEE_LE, 0xEDB88320, 0x3D5CDD04 }, | |||
{ AV_CRC_32_IEEE, 0x04C11DB7, 0xC0F5BAE0 }, | |||
{ AV_CRC_16_ANSI_LE, 0xA001, 0xBFD8 }, | |||
{ AV_CRC_16_ANSI, 0x8005, 0x1FBB }, | |||
{ AV_CRC_8_ATM, 0x07, 0xE3 } | |||
}; | |||
const AVCRC *ctx; | |||
for (i = 0; i < sizeof(buf); i++) | |||
buf[i] = i + i * i; | |||
for (i = 0; i < 5; i++) { | |||
ctx = av_crc_get_table(p[i][0]); | |||
printf("crc %08X = %X\n", p[i][1], av_crc(ctx, 0, buf, sizeof(buf))); | |||
} | |||
return 0; | |||
} |
@@ -333,28 +333,3 @@ uint32_t av_crc(const AVCRC *ctx, uint32_t crc, | |||
return crc; | |||
} | |||
#ifdef TEST | |||
int main(void) | |||
{ | |||
uint8_t buf[1999]; | |||
int i; | |||
int p[5][3] = { | |||
{ AV_CRC_32_IEEE_LE, 0xEDB88320, 0x3D5CDD04 }, | |||
{ AV_CRC_32_IEEE, 0x04C11DB7, 0xC0F5BAE0 }, | |||
{ AV_CRC_16_ANSI_LE, 0xA001, 0xBFD8 }, | |||
{ AV_CRC_16_ANSI, 0x8005, 0x1FBB }, | |||
{ AV_CRC_8_ATM, 0x07, 0xE3 } | |||
}; | |||
const AVCRC *ctx; | |||
for (i = 0; i < sizeof(buf); i++) | |||
buf[i] = i + i * i; | |||
for (i = 0; i < 5; i++) { | |||
ctx = av_crc_get_table(p[i][0]); | |||
printf("crc %08X = %X\n", p[i][1], av_crc(ctx, 0, buf, sizeof(buf))); | |||
} | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,128 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include "des.c" | |||
#include <stdint.h> | |||
#include <stdio.h> | |||
#include <stdlib.h> | |||
#include <string.h> | |||
#include "time.h" | |||
static uint64_t rand64(void) | |||
{ | |||
uint64_t r = rand(); | |||
r = (r << 32) | rand(); | |||
return r; | |||
} | |||
static const uint8_t test_key[] = { 0x12, 0x34, 0x56, 0x78, 0x9a, 0xbc, 0xde, 0xf0 }; | |||
static const DECLARE_ALIGNED(8, uint8_t, plain)[] = { 0xfe, 0xdc, 0xba, 0x98, 0x76, 0x54, 0x32, 0x10 }; | |||
static const DECLARE_ALIGNED(8, uint8_t, crypt)[] = { 0x4a, 0xb6, 0x5b, 0x3d, 0x4b, 0x06, 0x15, 0x18 }; | |||
static DECLARE_ALIGNED(8, uint8_t, tmp)[8]; | |||
static DECLARE_ALIGNED(8, uint8_t, large_buffer)[10002][8]; | |||
static const uint8_t cbc_key[] = { | |||
0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, | |||
0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0x01, | |||
0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0x01, 0x23 | |||
}; | |||
static int run_test(int cbc, int decrypt) | |||
{ | |||
AVDES d; | |||
int delay = cbc && !decrypt ? 2 : 1; | |||
uint64_t res; | |||
AV_WB64(large_buffer[0], 0x4e6f772069732074ULL); | |||
AV_WB64(large_buffer[1], 0x1234567890abcdefULL); | |||
AV_WB64(tmp, 0x1234567890abcdefULL); | |||
av_des_init(&d, cbc_key, 192, decrypt); | |||
av_des_crypt(&d, large_buffer[delay], large_buffer[0], 10000, cbc ? tmp : NULL, decrypt); | |||
res = AV_RB64(large_buffer[9999 + delay]); | |||
if (cbc) { | |||
if (decrypt) | |||
return res == 0xc5cecf63ecec514cULL; | |||
else | |||
return res == 0xcb191f85d1ed8439ULL; | |||
} else { | |||
if (decrypt) | |||
return res == 0x8325397644091a0aULL; | |||
else | |||
return res == 0xdd17e8b8b437d232ULL; | |||
} | |||
} | |||
int main(void) | |||
{ | |||
AVDES d; | |||
int i; | |||
uint64_t key[3]; | |||
uint64_t data; | |||
uint64_t ct; | |||
uint64_t roundkeys[16]; | |||
srand(av_gettime()); | |||
key[0] = AV_RB64(test_key); | |||
data = AV_RB64(plain); | |||
gen_roundkeys(roundkeys, key[0]); | |||
if (des_encdec(data, roundkeys, 0) != AV_RB64(crypt)) { | |||
printf("Test 1 failed\n"); | |||
return 1; | |||
} | |||
av_des_init(&d, test_key, 64, 0); | |||
av_des_crypt(&d, tmp, plain, 1, NULL, 0); | |||
if (memcmp(tmp, crypt, sizeof(crypt))) { | |||
printf("Public API decryption failed\n"); | |||
return 1; | |||
} | |||
if (!run_test(0, 0) || !run_test(0, 1) || !run_test(1, 0) || !run_test(1, 1)) { | |||
printf("Partial Monte-Carlo test failed\n"); | |||
return 1; | |||
} | |||
for (i = 0; i < 1000; i++) { | |||
key[0] = rand64(); | |||
key[1] = rand64(); | |||
key[2] = rand64(); | |||
data = rand64(); | |||
av_des_init(&d, key, 192, 0); | |||
av_des_crypt(&d, &ct, &data, 1, NULL, 0); | |||
av_des_init(&d, key, 192, 1); | |||
av_des_crypt(&d, &ct, &ct, 1, NULL, 1); | |||
if (ct != data) { | |||
printf("Test 2 failed\n"); | |||
return 1; | |||
} | |||
} | |||
#ifdef GENTABLES | |||
printf("static const uint32_t S_boxes_P_shuffle[8][64] = {\n"); | |||
for (i = 0; i < 8; i++) { | |||
int j; | |||
printf(" {"); | |||
for (j = 0; j < 64; j++) { | |||
uint32_t v = S_boxes[i][j >> 1]; | |||
v = j & 1 ? v >> 4 : v & 0xf; | |||
v <<= 28 - 4 * i; | |||
v = shuffle(v, P_shuffle, sizeof(P_shuffle)); | |||
printf((j & 7) == 0 ? "\n " : " "); | |||
printf("0x%08X,", v); | |||
} | |||
printf("\n },\n"); | |||
} | |||
printf("};\n"); | |||
#endif | |||
return 0; | |||
} |
@@ -111,7 +111,7 @@ static const uint8_t S_boxes[8][32] = { | |||
#else | |||
/** | |||
* This table contains the results of applying both the S-box and P-shuffle. | |||
* It can be regenerated by compiling this file with -DCONFIG_SMALL -DTEST -DGENTABLES | |||
* It can be regenerated by compiling des-test.c with "-DCONFIG_SMALL -DGENTABLES". | |||
*/ | |||
static const uint32_t S_boxes_P_shuffle[8][64] = { | |||
{ 0x00808200, 0x00000000, 0x00008000, 0x00808202, 0x00808002, 0x00008202, 0x00000002, 0x00008000, | |||
@@ -337,112 +337,3 @@ void av_des_mac(AVDES *d, uint8_t *dst, const uint8_t *src, int count) | |||
{ | |||
av_des_crypt_mac(d, dst, src, count, (uint8_t[8]) { 0 }, 0, 1); | |||
} | |||
#ifdef TEST | |||
#include <stdlib.h> | |||
#include <stdio.h> | |||
#include "time.h" | |||
static uint64_t rand64(void) | |||
{ | |||
uint64_t r = rand(); | |||
r = (r << 32) | rand(); | |||
return r; | |||
} | |||
static const uint8_t test_key[] = { 0x12, 0x34, 0x56, 0x78, 0x9a, 0xbc, 0xde, 0xf0 }; | |||
static const DECLARE_ALIGNED(8, uint8_t, plain)[] = { 0xfe, 0xdc, 0xba, 0x98, 0x76, 0x54, 0x32, 0x10 }; | |||
static const DECLARE_ALIGNED(8, uint8_t, crypt)[] = { 0x4a, 0xb6, 0x5b, 0x3d, 0x4b, 0x06, 0x15, 0x18 }; | |||
static DECLARE_ALIGNED(8, uint8_t, tmp)[8]; | |||
static DECLARE_ALIGNED(8, uint8_t, large_buffer)[10002][8]; | |||
static const uint8_t cbc_key[] = { | |||
0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, | |||
0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0x01, | |||
0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0x01, 0x23 | |||
}; | |||
static int run_test(int cbc, int decrypt) | |||
{ | |||
AVDES d; | |||
int delay = cbc && !decrypt ? 2 : 1; | |||
uint64_t res; | |||
AV_WB64(large_buffer[0], 0x4e6f772069732074ULL); | |||
AV_WB64(large_buffer[1], 0x1234567890abcdefULL); | |||
AV_WB64(tmp, 0x1234567890abcdefULL); | |||
av_des_init(&d, cbc_key, 192, decrypt); | |||
av_des_crypt(&d, large_buffer[delay], large_buffer[0], 10000, cbc ? tmp : NULL, decrypt); | |||
res = AV_RB64(large_buffer[9999 + delay]); | |||
if (cbc) { | |||
if (decrypt) | |||
return res == 0xc5cecf63ecec514cULL; | |||
else | |||
return res == 0xcb191f85d1ed8439ULL; | |||
} else { | |||
if (decrypt) | |||
return res == 0x8325397644091a0aULL; | |||
else | |||
return res == 0xdd17e8b8b437d232ULL; | |||
} | |||
} | |||
int main(void) | |||
{ | |||
AVDES d; | |||
int i; | |||
uint64_t key[3]; | |||
uint64_t data; | |||
uint64_t ct; | |||
uint64_t roundkeys[16]; | |||
srand(av_gettime()); | |||
key[0] = AV_RB64(test_key); | |||
data = AV_RB64(plain); | |||
gen_roundkeys(roundkeys, key[0]); | |||
if (des_encdec(data, roundkeys, 0) != AV_RB64(crypt)) { | |||
printf("Test 1 failed\n"); | |||
return 1; | |||
} | |||
av_des_init(&d, test_key, 64, 0); | |||
av_des_crypt(&d, tmp, plain, 1, NULL, 0); | |||
if (memcmp(tmp, crypt, sizeof(crypt))) { | |||
printf("Public API decryption failed\n"); | |||
return 1; | |||
} | |||
if (!run_test(0, 0) || !run_test(0, 1) || !run_test(1, 0) || !run_test(1, 1)) { | |||
printf("Partial Monte-Carlo test failed\n"); | |||
return 1; | |||
} | |||
for (i = 0; i < 1000; i++) { | |||
key[0] = rand64(); | |||
key[1] = rand64(); | |||
key[2] = rand64(); | |||
data = rand64(); | |||
av_des_init(&d, key, 192, 0); | |||
av_des_crypt(&d, &ct, &data, 1, NULL, 0); | |||
av_des_init(&d, key, 192, 1); | |||
av_des_crypt(&d, &ct, &ct, 1, NULL, 1); | |||
if (ct != data) { | |||
printf("Test 2 failed\n"); | |||
return 1; | |||
} | |||
} | |||
#ifdef GENTABLES | |||
printf("static const uint32_t S_boxes_P_shuffle[8][64] = {\n"); | |||
for (i = 0; i < 8; i++) { | |||
int j; | |||
printf(" {"); | |||
for (j = 0; j < 64; j++) { | |||
uint32_t v = S_boxes[i][j >> 1]; | |||
v = j & 1 ? v >> 4 : v & 0xf; | |||
v <<= 28 - 4 * i; | |||
v = shuffle(v, P_shuffle, sizeof(P_shuffle)); | |||
printf((j & 7) == 0 ? "\n " : " "); | |||
printf("0x%08X,", v); | |||
} | |||
printf("\n },\n"); | |||
} | |||
printf("};\n"); | |||
#endif | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,144 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <math.h> | |||
#include <stdio.h> | |||
#include <string.h> | |||
#include "timer.h" | |||
#include "eval.h" | |||
static const double const_values[] = { | |||
M_PI, | |||
M_E, | |||
0 | |||
}; | |||
static const char *const const_names[] = { | |||
"PI", | |||
"E", | |||
0 | |||
}; | |||
int main(int argc, char **argv) | |||
{ | |||
int i; | |||
double d; | |||
const char *const *expr; | |||
static const char *const exprs[] = { | |||
"", | |||
"1;2", | |||
"-20", | |||
"-PI", | |||
"+PI", | |||
"1+(5-2)^(3-1)+1/2+sin(PI)-max(-2.2,-3.1)", | |||
"80G/80Gi", | |||
"1k", | |||
"1Gi", | |||
"1gi", | |||
"1GiFoo", | |||
"1k+1k", | |||
"1Gi*3foo", | |||
"foo", | |||
"foo(", | |||
"foo()", | |||
"foo)", | |||
"sin", | |||
"sin(", | |||
"sin()", | |||
"sin)", | |||
"sin 10", | |||
"sin(1,2,3)", | |||
"sin(1 )", | |||
"1", | |||
"1foo", | |||
"bar + PI + E + 100f*2 + foo", | |||
"13k + 12f - foo(1, 2)", | |||
"1gi", | |||
"1Gi", | |||
"st(0, 123)", | |||
"st(1, 123); ld(1)", | |||
"lte(0, 1)", | |||
"lte(1, 1)", | |||
"lte(1, 0)", | |||
"lt(0, 1)", | |||
"lt(1, 1)", | |||
"gt(1, 0)", | |||
"gt(2, 7)", | |||
"gte(122, 122)", | |||
/* compute 1+2+...+N */ | |||
"st(0, 1); while(lte(ld(0), 100), st(1, ld(1)+ld(0));st(0, ld(0)+1)); ld(1)", | |||
/* compute Fib(N) */ | |||
"st(1, 1); st(2, 2); st(0, 1); while(lte(ld(0),10), st(3, ld(1)+ld(2)); st(1, ld(2)); st(2, ld(3)); st(0, ld(0)+1)); ld(3)", | |||
"while(0, 10)", | |||
"st(0, 1); while(lte(ld(0),100), st(1, ld(1)+ld(0)); st(0, ld(0)+1))", | |||
"isnan(1)", | |||
"isnan(NAN)", | |||
"isnan(INF)", | |||
"isinf(1)", | |||
"isinf(NAN)", | |||
"isinf(INF)", | |||
"floor(NAN)", | |||
"floor(123.123)", | |||
"floor(-123.123)", | |||
"trunc(123.123)", | |||
"trunc(-123.123)", | |||
"ceil(123.123)", | |||
"ceil(-123.123)", | |||
"sqrt(1764)", | |||
"isnan(sqrt(-1))", | |||
"not(1)", | |||
"not(NAN)", | |||
"not(0)", | |||
"6.0206dB", | |||
"-3.0103dB", | |||
NULL | |||
}; | |||
for (expr = exprs; *expr; expr++) { | |||
printf("Evaluating '%s'\n", *expr); | |||
av_expr_parse_and_eval(&d, *expr, | |||
const_names, const_values, | |||
NULL, NULL, NULL, NULL, NULL, 0, NULL); | |||
if (isnan(d)) | |||
printf("'%s' -> nan\n\n", *expr); | |||
else | |||
printf("'%s' -> %f\n\n", *expr, d); | |||
} | |||
av_expr_parse_and_eval(&d, "1+(5-2)^(3-1)+1/2+sin(PI)-max(-2.2,-3.1)", | |||
const_names, const_values, | |||
NULL, NULL, NULL, NULL, NULL, 0, NULL); | |||
printf("%f == 12.7\n", d); | |||
av_expr_parse_and_eval(&d, "80G/80Gi", | |||
const_names, const_values, | |||
NULL, NULL, NULL, NULL, NULL, 0, NULL); | |||
printf("%f == 0.931322575\n", d); | |||
if (argc > 1 && !strcmp(argv[1], "-t")) { | |||
for (i = 0; i < 1050; i++) { | |||
START_TIMER; | |||
av_expr_parse_and_eval(&d, "1+(5-2)^(3-1)+1/2+sin(PI)-max(-2.2,-3.1)", | |||
const_names, const_values, | |||
NULL, NULL, NULL, NULL, NULL, 0, NULL); | |||
STOP_TIMER("av_expr_parse_and_eval"); | |||
} | |||
} | |||
return 0; | |||
} |
@@ -565,127 +565,3 @@ int av_expr_parse_and_eval(double *d, const char *s, | |||
av_expr_free(e); | |||
return isnan(*d) ? AVERROR(EINVAL) : 0; | |||
} | |||
#ifdef TEST | |||
#include <string.h> | |||
static const double const_values[] = { | |||
M_PI, | |||
M_E, | |||
0 | |||
}; | |||
static const char *const const_names[] = { | |||
"PI", | |||
"E", | |||
0 | |||
}; | |||
int main(int argc, char **argv) | |||
{ | |||
int i; | |||
double d; | |||
const char *const *expr; | |||
static const char *const exprs[] = { | |||
"", | |||
"1;2", | |||
"-20", | |||
"-PI", | |||
"+PI", | |||
"1+(5-2)^(3-1)+1/2+sin(PI)-max(-2.2,-3.1)", | |||
"80G/80Gi", | |||
"1k", | |||
"1Gi", | |||
"1gi", | |||
"1GiFoo", | |||
"1k+1k", | |||
"1Gi*3foo", | |||
"foo", | |||
"foo(", | |||
"foo()", | |||
"foo)", | |||
"sin", | |||
"sin(", | |||
"sin()", | |||
"sin)", | |||
"sin 10", | |||
"sin(1,2,3)", | |||
"sin(1 )", | |||
"1", | |||
"1foo", | |||
"bar + PI + E + 100f*2 + foo", | |||
"13k + 12f - foo(1, 2)", | |||
"1gi", | |||
"1Gi", | |||
"st(0, 123)", | |||
"st(1, 123); ld(1)", | |||
"lte(0, 1)", | |||
"lte(1, 1)", | |||
"lte(1, 0)", | |||
"lt(0, 1)", | |||
"lt(1, 1)", | |||
"gt(1, 0)", | |||
"gt(2, 7)", | |||
"gte(122, 122)", | |||
/* compute 1+2+...+N */ | |||
"st(0, 1); while(lte(ld(0), 100), st(1, ld(1)+ld(0));st(0, ld(0)+1)); ld(1)", | |||
/* compute Fib(N) */ | |||
"st(1, 1); st(2, 2); st(0, 1); while(lte(ld(0),10), st(3, ld(1)+ld(2)); st(1, ld(2)); st(2, ld(3)); st(0, ld(0)+1)); ld(3)", | |||
"while(0, 10)", | |||
"st(0, 1); while(lte(ld(0),100), st(1, ld(1)+ld(0)); st(0, ld(0)+1))", | |||
"isnan(1)", | |||
"isnan(NAN)", | |||
"isnan(INF)", | |||
"isinf(1)", | |||
"isinf(NAN)", | |||
"isinf(INF)", | |||
"floor(NAN)", | |||
"floor(123.123)", | |||
"floor(-123.123)", | |||
"trunc(123.123)", | |||
"trunc(-123.123)", | |||
"ceil(123.123)", | |||
"ceil(-123.123)", | |||
"sqrt(1764)", | |||
"isnan(sqrt(-1))", | |||
"not(1)", | |||
"not(NAN)", | |||
"not(0)", | |||
"6.0206dB", | |||
"-3.0103dB", | |||
NULL | |||
}; | |||
for (expr = exprs; *expr; expr++) { | |||
printf("Evaluating '%s'\n", *expr); | |||
av_expr_parse_and_eval(&d, *expr, | |||
const_names, const_values, | |||
NULL, NULL, NULL, NULL, NULL, 0, NULL); | |||
if (isnan(d)) | |||
printf("'%s' -> nan\n\n", *expr); | |||
else | |||
printf("'%s' -> %f\n\n", *expr, d); | |||
} | |||
av_expr_parse_and_eval(&d, "1+(5-2)^(3-1)+1/2+sin(PI)-max(-2.2,-3.1)", | |||
const_names, const_values, | |||
NULL, NULL, NULL, NULL, NULL, 0, NULL); | |||
printf("%f == 12.7\n", d); | |||
av_expr_parse_and_eval(&d, "80G/80Gi", | |||
const_names, const_values, | |||
NULL, NULL, NULL, NULL, NULL, 0, NULL); | |||
printf("%f == 0.931322575\n", d); | |||
if (argc > 1 && !strcmp(argv[1], "-t")) { | |||
for (i = 0; i < 1050; i++) { | |||
START_TIMER; | |||
av_expr_parse_and_eval(&d, "1+(5-2)^(3-1)+1/2+sin(PI)-max(-2.2,-3.1)", | |||
const_names, const_values, | |||
NULL, NULL, NULL, NULL, NULL, 0, NULL); | |||
STOP_TIMER("av_expr_parse_and_eval"); | |||
} | |||
} | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,51 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <stdio.h> | |||
#include "fifo.h" | |||
int main(void) | |||
{ | |||
/* create a FIFO buffer */ | |||
AVFifoBuffer *fifo = av_fifo_alloc(13 * sizeof(int)); | |||
int i, j, n; | |||
/* fill data */ | |||
for (i = 0; av_fifo_space(fifo) >= sizeof(int); i++) | |||
av_fifo_generic_write(fifo, &i, sizeof(int), NULL); | |||
/* peek at FIFO */ | |||
n = av_fifo_size(fifo) / sizeof(int); | |||
for (i = -n + 1; i < n; i++) { | |||
int *v = (int *)av_fifo_peek2(fifo, i * sizeof(int)); | |||
printf("%d: %d\n", i, *v); | |||
} | |||
printf("\n"); | |||
/* read data */ | |||
for (i = 0; av_fifo_size(fifo) >= sizeof(int); i++) { | |||
av_fifo_generic_read(fifo, &j, sizeof(int), NULL); | |||
printf("%d ", j); | |||
} | |||
printf("\n"); | |||
av_fifo_free(fifo); | |||
return 0; | |||
} |
@@ -131,37 +131,3 @@ void av_fifo_drain(AVFifoBuffer *f, int size) | |||
f->rptr -= f->end - f->buffer; | |||
f->rndx += size; | |||
} | |||
#ifdef TEST | |||
int main(void) | |||
{ | |||
/* create a FIFO buffer */ | |||
AVFifoBuffer *fifo = av_fifo_alloc(13 * sizeof(int)); | |||
int i, j, n; | |||
/* fill data */ | |||
for (i = 0; av_fifo_space(fifo) >= sizeof(int); i++) | |||
av_fifo_generic_write(fifo, &i, sizeof(int), NULL); | |||
/* peek at FIFO */ | |||
n = av_fifo_size(fifo) / sizeof(int); | |||
for (i = -n + 1; i < n; i++) { | |||
int *v = (int *)av_fifo_peek2(fifo, i * sizeof(int)); | |||
printf("%d: %d\n", i, *v); | |||
} | |||
printf("\n"); | |||
/* read data */ | |||
for (i = 0; av_fifo_size(fifo) >= sizeof(int); i++) { | |||
av_fifo_generic_read(fifo, &j, sizeof(int), NULL); | |||
printf("%d ", j); | |||
} | |||
printf("\n"); | |||
av_fifo_free(fifo); | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,296 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <float.h> | |||
#include <stdint.h> | |||
#include <stdlib.h> | |||
#include <string.h> | |||
#include "cpu.h" | |||
#include "internal.h" | |||
#include "lfg.h" | |||
#include "log.h" | |||
#include "random_seed.h" | |||
#include "float_dsp.h" | |||
#define LEN 240 | |||
static void fill_float_array(AVLFG *lfg, float *a, int len) | |||
{ | |||
int i; | |||
double bmg[2], stddev = 10.0, mean = 0.0; | |||
for (i = 0; i < len; i += 2) { | |||
av_bmg_get(lfg, bmg); | |||
a[i] = bmg[0] * stddev + mean; | |||
a[i + 1] = bmg[1] * stddev + mean; | |||
} | |||
} | |||
static int compare_floats(const float *a, const float *b, int len, | |||
float max_diff) | |||
{ | |||
int i; | |||
for (i = 0; i < len; i++) { | |||
if (fabsf(a[i] - b[i]) > max_diff) { | |||
av_log(NULL, AV_LOG_ERROR, "%d: %- .12f - %- .12f = % .12g\n", | |||
i, a[i], b[i], a[i] - b[i]); | |||
return -1; | |||
} | |||
} | |||
return 0; | |||
} | |||
static void fill_double_array(AVLFG *lfg, double *a, int len) | |||
{ | |||
int i; | |||
double bmg[2], stddev = 10.0, mean = 0.0; | |||
for (i = 0; i < len; i += 2) { | |||
av_bmg_get(lfg, bmg); | |||
a[i] = bmg[0] * stddev + mean; | |||
a[i + 1] = bmg[1] * stddev + mean; | |||
} | |||
} | |||
static int compare_doubles(const double *a, const double *b, int len, | |||
double max_diff) | |||
{ | |||
int i; | |||
for (i = 0; i < len; i++) { | |||
if (fabs(a[i] - b[i]) > max_diff) { | |||
av_log(NULL, AV_LOG_ERROR, "%d: %- .12f - %- .12f = % .12g\n", | |||
i, a[i], b[i], a[i] - b[i]); | |||
return -1; | |||
} | |||
} | |||
return 0; | |||
} | |||
static int test_vector_fmul(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul(cdst, v1, v2, LEN); | |||
fdsp->vector_fmul(odst, v1, v2, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, FLT_EPSILON)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul failed\n"); | |||
return ret; | |||
} | |||
#define ARBITRARY_FMAC_SCALAR_CONST 0.005 | |||
static int test_vector_fmac_scalar(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *src0, float scale) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
memcpy(cdst, v1, LEN * sizeof(*v1)); | |||
memcpy(odst, v1, LEN * sizeof(*v1)); | |||
cdsp->vector_fmac_scalar(cdst, src0, scale, LEN); | |||
fdsp->vector_fmac_scalar(odst, src0, scale, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, ARBITRARY_FMAC_SCALAR_CONST)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmac_scalar failed\n"); | |||
return ret; | |||
} | |||
static int test_vector_fmul_scalar(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, float scale) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul_scalar(cdst, v1, scale, LEN); | |||
fdsp->vector_fmul_scalar(odst, v1, scale, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, FLT_EPSILON)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul_scalar failed\n"); | |||
return ret; | |||
} | |||
static int test_vector_dmul_scalar(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const double *v1, double scale) | |||
{ | |||
LOCAL_ALIGNED(32, double, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, double, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_dmul_scalar(cdst, v1, scale, LEN); | |||
fdsp->vector_dmul_scalar(odst, v1, scale, LEN); | |||
if (ret = compare_doubles(cdst, odst, LEN, DBL_EPSILON)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_dmul_scalar failed\n"); | |||
return ret; | |||
} | |||
#define ARBITRARY_FMUL_WINDOW_CONST 0.008 | |||
static int test_vector_fmul_window(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2, const float *v3) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul_window(cdst, v1, v2, v3, LEN / 2); | |||
fdsp->vector_fmul_window(odst, v1, v2, v3, LEN / 2); | |||
if (ret = compare_floats(cdst, odst, LEN, ARBITRARY_FMUL_WINDOW_CONST)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul_window failed\n"); | |||
return ret; | |||
} | |||
#define ARBITRARY_FMUL_ADD_CONST 0.005 | |||
static int test_vector_fmul_add(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2, const float *v3) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul_add(cdst, v1, v2, v3, LEN); | |||
fdsp->vector_fmul_add(odst, v1, v2, v3, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, ARBITRARY_FMUL_ADD_CONST)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul_add failed\n"); | |||
return ret; | |||
} | |||
static int test_vector_fmul_reverse(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul_reverse(cdst, v1, v2, LEN); | |||
fdsp->vector_fmul_reverse(odst, v1, v2, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, FLT_EPSILON)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul_reverse failed\n"); | |||
return ret; | |||
} | |||
static int test_butterflies_float(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2) | |||
{ | |||
LOCAL_ALIGNED(32, float, cv1, [LEN]); | |||
LOCAL_ALIGNED(32, float, cv2, [LEN]); | |||
LOCAL_ALIGNED(32, float, ov1, [LEN]); | |||
LOCAL_ALIGNED(32, float, ov2, [LEN]); | |||
int ret; | |||
memcpy(cv1, v1, LEN * sizeof(*v1)); | |||
memcpy(cv2, v2, LEN * sizeof(*v2)); | |||
memcpy(ov1, v1, LEN * sizeof(*v1)); | |||
memcpy(ov2, v2, LEN * sizeof(*v2)); | |||
cdsp->butterflies_float(cv1, cv2, LEN); | |||
fdsp->butterflies_float(ov1, ov2, LEN); | |||
if ((ret = compare_floats(cv1, ov1, LEN, FLT_EPSILON)) || | |||
(ret = compare_floats(cv2, ov2, LEN, FLT_EPSILON))) | |||
av_log(NULL, AV_LOG_ERROR, "butterflies_float failed\n"); | |||
return ret; | |||
} | |||
#define ARBITRARY_SCALARPRODUCT_CONST 0.2 | |||
static int test_scalarproduct_float(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2) | |||
{ | |||
float cprod, oprod; | |||
int ret; | |||
cprod = cdsp->scalarproduct_float(v1, v2, LEN); | |||
oprod = fdsp->scalarproduct_float(v1, v2, LEN); | |||
if (ret = compare_floats(&cprod, &oprod, 1, ARBITRARY_SCALARPRODUCT_CONST)) | |||
av_log(NULL, AV_LOG_ERROR, "scalarproduct_float failed\n"); | |||
return ret; | |||
} | |||
int main(int argc, char **argv) | |||
{ | |||
int ret = 0; | |||
uint32_t seed; | |||
AVFloatDSPContext fdsp, cdsp; | |||
AVLFG lfg; | |||
LOCAL_ALIGNED(32, float, src0, [LEN]); | |||
LOCAL_ALIGNED(32, float, src1, [LEN]); | |||
LOCAL_ALIGNED(32, float, src2, [LEN]); | |||
LOCAL_ALIGNED(32, double, dbl_src0, [LEN]); | |||
LOCAL_ALIGNED(32, double, dbl_src1, [LEN]); | |||
if (argc > 2 && !strcmp(argv[1], "-s")) | |||
seed = strtoul(argv[2], NULL, 10); | |||
else | |||
seed = av_get_random_seed(); | |||
av_log(NULL, AV_LOG_INFO, "float_dsp-test: random seed %u\n", seed); | |||
av_lfg_init(&lfg, seed); | |||
fill_float_array(&lfg, src0, LEN); | |||
fill_float_array(&lfg, src1, LEN); | |||
fill_float_array(&lfg, src2, LEN); | |||
fill_double_array(&lfg, dbl_src0, LEN); | |||
fill_double_array(&lfg, dbl_src1, LEN); | |||
avpriv_float_dsp_init(&fdsp, 1); | |||
av_set_cpu_flags_mask(0); | |||
avpriv_float_dsp_init(&cdsp, 1); | |||
if (test_vector_fmul(&fdsp, &cdsp, src0, src1)) | |||
ret -= 1 << 0; | |||
if (test_vector_fmac_scalar(&fdsp, &cdsp, src2, src0, src1[0])) | |||
ret -= 1 << 1; | |||
if (test_vector_fmul_scalar(&fdsp, &cdsp, src0, src1[0])) | |||
ret -= 1 << 2; | |||
if (test_vector_fmul_window(&fdsp, &cdsp, src0, src1, src2)) | |||
ret -= 1 << 3; | |||
if (test_vector_fmul_add(&fdsp, &cdsp, src0, src1, src2)) | |||
ret -= 1 << 4; | |||
if (test_vector_fmul_reverse(&fdsp, &cdsp, src0, src1)) | |||
ret -= 1 << 5; | |||
if (test_butterflies_float(&fdsp, &cdsp, src0, src1)) | |||
ret -= 1 << 6; | |||
if (test_scalarproduct_float(&fdsp, &cdsp, src0, src1)) | |||
ret -= 1 << 7; | |||
if (test_vector_dmul_scalar(&fdsp, &cdsp, dbl_src0, dbl_src1[0])) | |||
ret -= 1 << 8; | |||
return ret; | |||
} |
@@ -133,288 +133,3 @@ av_cold void avpriv_float_dsp_init(AVFloatDSPContext *fdsp, int bit_exact) | |||
if (ARCH_X86) | |||
ff_float_dsp_init_x86(fdsp); | |||
} | |||
#ifdef TEST | |||
#include <float.h> | |||
#include <math.h> | |||
#include <stdint.h> | |||
#include <stdlib.h> | |||
#include <string.h> | |||
#include "common.h" | |||
#include "cpu.h" | |||
#include "internal.h" | |||
#include "lfg.h" | |||
#include "log.h" | |||
#include "mem.h" | |||
#include "random_seed.h" | |||
#define LEN 240 | |||
static void fill_float_array(AVLFG *lfg, float *a, int len) | |||
{ | |||
int i; | |||
double bmg[2], stddev = 10.0, mean = 0.0; | |||
for (i = 0; i < len; i += 2) { | |||
av_bmg_get(lfg, bmg); | |||
a[i] = bmg[0] * stddev + mean; | |||
a[i + 1] = bmg[1] * stddev + mean; | |||
} | |||
} | |||
static int compare_floats(const float *a, const float *b, int len, | |||
float max_diff) | |||
{ | |||
int i; | |||
for (i = 0; i < len; i++) { | |||
if (fabsf(a[i] - b[i]) > max_diff) { | |||
av_log(NULL, AV_LOG_ERROR, "%d: %- .12f - %- .12f = % .12g\n", | |||
i, a[i], b[i], a[i] - b[i]); | |||
return -1; | |||
} | |||
} | |||
return 0; | |||
} | |||
static void fill_double_array(AVLFG *lfg, double *a, int len) | |||
{ | |||
int i; | |||
double bmg[2], stddev = 10.0, mean = 0.0; | |||
for (i = 0; i < len; i += 2) { | |||
av_bmg_get(lfg, bmg); | |||
a[i] = bmg[0] * stddev + mean; | |||
a[i + 1] = bmg[1] * stddev + mean; | |||
} | |||
} | |||
static int compare_doubles(const double *a, const double *b, int len, | |||
double max_diff) | |||
{ | |||
int i; | |||
for (i = 0; i < len; i++) { | |||
if (fabs(a[i] - b[i]) > max_diff) { | |||
av_log(NULL, AV_LOG_ERROR, "%d: %- .12f - %- .12f = % .12g\n", | |||
i, a[i], b[i], a[i] - b[i]); | |||
return -1; | |||
} | |||
} | |||
return 0; | |||
} | |||
static int test_vector_fmul(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul(cdst, v1, v2, LEN); | |||
fdsp->vector_fmul(odst, v1, v2, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, FLT_EPSILON)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul failed\n"); | |||
return ret; | |||
} | |||
#define ARBITRARY_FMAC_SCALAR_CONST 0.005 | |||
static int test_vector_fmac_scalar(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *src0, float scale) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
memcpy(cdst, v1, LEN * sizeof(*v1)); | |||
memcpy(odst, v1, LEN * sizeof(*v1)); | |||
cdsp->vector_fmac_scalar(cdst, src0, scale, LEN); | |||
fdsp->vector_fmac_scalar(odst, src0, scale, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, ARBITRARY_FMAC_SCALAR_CONST)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmac_scalar failed\n"); | |||
return ret; | |||
} | |||
static int test_vector_fmul_scalar(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, float scale) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul_scalar(cdst, v1, scale, LEN); | |||
fdsp->vector_fmul_scalar(odst, v1, scale, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, FLT_EPSILON)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul_scalar failed\n"); | |||
return ret; | |||
} | |||
static int test_vector_dmul_scalar(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const double *v1, double scale) | |||
{ | |||
LOCAL_ALIGNED(32, double, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, double, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_dmul_scalar(cdst, v1, scale, LEN); | |||
fdsp->vector_dmul_scalar(odst, v1, scale, LEN); | |||
if (ret = compare_doubles(cdst, odst, LEN, DBL_EPSILON)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_dmul_scalar failed\n"); | |||
return ret; | |||
} | |||
#define ARBITRARY_FMUL_WINDOW_CONST 0.008 | |||
static int test_vector_fmul_window(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2, const float *v3) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul_window(cdst, v1, v2, v3, LEN / 2); | |||
fdsp->vector_fmul_window(odst, v1, v2, v3, LEN / 2); | |||
if (ret = compare_floats(cdst, odst, LEN, ARBITRARY_FMUL_WINDOW_CONST)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul_window failed\n"); | |||
return ret; | |||
} | |||
#define ARBITRARY_FMUL_ADD_CONST 0.005 | |||
static int test_vector_fmul_add(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2, const float *v3) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul_add(cdst, v1, v2, v3, LEN); | |||
fdsp->vector_fmul_add(odst, v1, v2, v3, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, ARBITRARY_FMUL_ADD_CONST)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul_add failed\n"); | |||
return ret; | |||
} | |||
static int test_vector_fmul_reverse(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2) | |||
{ | |||
LOCAL_ALIGNED(32, float, cdst, [LEN]); | |||
LOCAL_ALIGNED(32, float, odst, [LEN]); | |||
int ret; | |||
cdsp->vector_fmul_reverse(cdst, v1, v2, LEN); | |||
fdsp->vector_fmul_reverse(odst, v1, v2, LEN); | |||
if (ret = compare_floats(cdst, odst, LEN, FLT_EPSILON)) | |||
av_log(NULL, AV_LOG_ERROR, "vector_fmul_reverse failed\n"); | |||
return ret; | |||
} | |||
static int test_butterflies_float(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2) | |||
{ | |||
LOCAL_ALIGNED(32, float, cv1, [LEN]); | |||
LOCAL_ALIGNED(32, float, cv2, [LEN]); | |||
LOCAL_ALIGNED(32, float, ov1, [LEN]); | |||
LOCAL_ALIGNED(32, float, ov2, [LEN]); | |||
int ret; | |||
memcpy(cv1, v1, LEN * sizeof(*v1)); | |||
memcpy(cv2, v2, LEN * sizeof(*v2)); | |||
memcpy(ov1, v1, LEN * sizeof(*v1)); | |||
memcpy(ov2, v2, LEN * sizeof(*v2)); | |||
cdsp->butterflies_float(cv1, cv2, LEN); | |||
fdsp->butterflies_float(ov1, ov2, LEN); | |||
if ((ret = compare_floats(cv1, ov1, LEN, FLT_EPSILON)) || | |||
(ret = compare_floats(cv2, ov2, LEN, FLT_EPSILON))) | |||
av_log(NULL, AV_LOG_ERROR, "butterflies_float failed\n"); | |||
return ret; | |||
} | |||
#define ARBITRARY_SCALARPRODUCT_CONST 0.2 | |||
static int test_scalarproduct_float(AVFloatDSPContext *fdsp, AVFloatDSPContext *cdsp, | |||
const float *v1, const float *v2) | |||
{ | |||
float cprod, oprod; | |||
int ret; | |||
cprod = cdsp->scalarproduct_float(v1, v2, LEN); | |||
oprod = fdsp->scalarproduct_float(v1, v2, LEN); | |||
if (ret = compare_floats(&cprod, &oprod, 1, ARBITRARY_SCALARPRODUCT_CONST)) | |||
av_log(NULL, AV_LOG_ERROR, "scalarproduct_float failed\n"); | |||
return ret; | |||
} | |||
int main(int argc, char **argv) | |||
{ | |||
int ret = 0; | |||
uint32_t seed; | |||
AVFloatDSPContext fdsp, cdsp; | |||
AVLFG lfg; | |||
LOCAL_ALIGNED(32, float, src0, [LEN]); | |||
LOCAL_ALIGNED(32, float, src1, [LEN]); | |||
LOCAL_ALIGNED(32, float, src2, [LEN]); | |||
LOCAL_ALIGNED(32, double, dbl_src0, [LEN]); | |||
LOCAL_ALIGNED(32, double, dbl_src1, [LEN]); | |||
if (argc > 2 && !strcmp(argv[1], "-s")) | |||
seed = strtoul(argv[2], NULL, 10); | |||
else | |||
seed = av_get_random_seed(); | |||
av_log(NULL, AV_LOG_INFO, "float_dsp-test: random seed %u\n", seed); | |||
av_lfg_init(&lfg, seed); | |||
fill_float_array(&lfg, src0, LEN); | |||
fill_float_array(&lfg, src1, LEN); | |||
fill_float_array(&lfg, src2, LEN); | |||
fill_double_array(&lfg, dbl_src0, LEN); | |||
fill_double_array(&lfg, dbl_src1, LEN); | |||
avpriv_float_dsp_init(&fdsp, 1); | |||
av_set_cpu_flags_mask(0); | |||
avpriv_float_dsp_init(&cdsp, 1); | |||
if (test_vector_fmul(&fdsp, &cdsp, src0, src1)) | |||
ret -= 1 << 0; | |||
if (test_vector_fmac_scalar(&fdsp, &cdsp, src2, src0, src1[0])) | |||
ret -= 1 << 1; | |||
if (test_vector_fmul_scalar(&fdsp, &cdsp, src0, src1[0])) | |||
ret -= 1 << 2; | |||
if (test_vector_fmul_window(&fdsp, &cdsp, src0, src1, src2)) | |||
ret -= 1 << 3; | |||
if (test_vector_fmul_add(&fdsp, &cdsp, src0, src1, src2)) | |||
ret -= 1 << 4; | |||
if (test_vector_fmul_reverse(&fdsp, &cdsp, src0, src1)) | |||
ret -= 1 << 5; | |||
if (test_butterflies_float(&fdsp, &cdsp, src0, src1)) | |||
ret -= 1 << 6; | |||
if (test_scalarproduct_float(&fdsp, &cdsp, src0, src1)) | |||
ret -= 1 << 7; | |||
if (test_vector_dmul_scalar(&fdsp, &cdsp, dbl_src0, dbl_src1[0])) | |||
ret -= 1 << 8; | |||
return ret; | |||
} | |||
#endif /* TEST */ |
@@ -0,0 +1,92 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include "hmac.c" | |||
#include <stdio.h> | |||
#include <string.h> | |||
static void test(AVHMAC *hmac, const uint8_t *key, int keylen, | |||
const uint8_t *data, int datalen) | |||
{ | |||
uint8_t buf[MAX_HASHLEN]; | |||
int out, i; | |||
// Some of the test vectors are strings, where sizeof() includes the | |||
// trailing null byte - remove that. | |||
if (!key[keylen - 1]) | |||
keylen--; | |||
if (!data[datalen - 1]) | |||
datalen--; | |||
out = av_hmac_calc(hmac, data, datalen, key, keylen, buf, sizeof(buf)); | |||
for (i = 0; i < out; i++) | |||
printf("%02x", buf[i]); | |||
printf("\n"); | |||
} | |||
int main(void) | |||
{ | |||
uint8_t key1[20], key3[131], data3[50]; | |||
enum AVHMACType i = AV_HMAC_SHA224; | |||
const uint8_t key2[] = "Jefe"; | |||
const uint8_t data1[] = "Hi There"; | |||
const uint8_t data2[] = "what do ya want for nothing?"; | |||
const uint8_t data4[] = "Test Using Larger Than Block-Size Key - Hash Key First"; | |||
const uint8_t data5[] = "Test Using Larger Than Block-Size Key and Larger Than One Block-Size Data"; | |||
const uint8_t data6[] = "This is a test using a larger than block-size key and a larger " | |||
"than block-size data. The key needs to be hashed before being used" | |||
" by the HMAC algorithm."; | |||
AVHMAC *hmac = av_hmac_alloc(AV_HMAC_MD5); | |||
if (!hmac) | |||
return 1; | |||
memset(key1, 0x0b, sizeof(key1)); | |||
memset(key3, 0xaa, sizeof(key3)); | |||
memset(data3, 0xdd, sizeof(data3)); | |||
// RFC 2202 test vectors | |||
test(hmac, key1, 16, data1, sizeof(data1)); | |||
test(hmac, key2, sizeof(key2), data2, sizeof(data2)); | |||
test(hmac, key3, 16, data3, sizeof(data3)); | |||
test(hmac, key3, 80, data4, sizeof(data4)); | |||
test(hmac, key3, 80, data5, sizeof(data5)); | |||
av_hmac_free(hmac); | |||
/* SHA-1 */ | |||
hmac = av_hmac_alloc(AV_HMAC_SHA1); | |||
if (!hmac) | |||
return 1; | |||
// RFC 2202 test vectors | |||
test(hmac, key1, sizeof(key1), data1, sizeof(data1)); | |||
test(hmac, key2, sizeof(key2), data2, sizeof(data2)); | |||
test(hmac, key3, 20, data3, sizeof(data3)); | |||
test(hmac, key3, 80, data4, sizeof(data4)); | |||
test(hmac, key3, 80, data5, sizeof(data5)); | |||
av_hmac_free(hmac); | |||
/* SHA-2 */ | |||
while (i <= AV_HMAC_SHA256) { | |||
hmac = av_hmac_alloc(i); | |||
// RFC 4231 test vectors | |||
test(hmac, key1, sizeof(key1), data1, sizeof(data1)); | |||
test(hmac, key2, sizeof(key2), data2, sizeof(data2)); | |||
test(hmac, key3, 20, data3, sizeof(data3)); | |||
test(hmac, key3, sizeof(key3), data4, sizeof(data4)); | |||
test(hmac, key3, sizeof(key3), data6, sizeof(data6)); | |||
av_hmac_free(hmac); | |||
i++; | |||
} | |||
return 0; | |||
} |
@@ -158,77 +158,3 @@ int av_hmac_calc(AVHMAC *c, const uint8_t *data, unsigned int len, | |||
av_hmac_update(c, data, len); | |||
return av_hmac_final(c, out, outlen); | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
static void test(AVHMAC *hmac, const uint8_t *key, int keylen, | |||
const uint8_t *data, int datalen) | |||
{ | |||
uint8_t buf[MAX_HASHLEN]; | |||
int out, i; | |||
// Some of the test vectors are strings, where sizeof() includes the | |||
// trailing null byte - remove that. | |||
if (!key[keylen - 1]) | |||
keylen--; | |||
if (!data[datalen - 1]) | |||
datalen--; | |||
out = av_hmac_calc(hmac, data, datalen, key, keylen, buf, sizeof(buf)); | |||
for (i = 0; i < out; i++) | |||
printf("%02x", buf[i]); | |||
printf("\n"); | |||
} | |||
int main(void) | |||
{ | |||
uint8_t key1[20], key3[131], data3[50]; | |||
enum AVHMACType i = AV_HMAC_SHA224; | |||
const uint8_t key2[] = "Jefe"; | |||
const uint8_t data1[] = "Hi There"; | |||
const uint8_t data2[] = "what do ya want for nothing?"; | |||
const uint8_t data4[] = "Test Using Larger Than Block-Size Key - Hash Key First"; | |||
const uint8_t data5[] = "Test Using Larger Than Block-Size Key and Larger Than One Block-Size Data"; | |||
const uint8_t data6[] = "This is a test using a larger than block-size key and a larger " | |||
"than block-size data. The key needs to be hashed before being used" | |||
" by the HMAC algorithm."; | |||
AVHMAC *hmac = av_hmac_alloc(AV_HMAC_MD5); | |||
if (!hmac) | |||
return 1; | |||
memset(key1, 0x0b, sizeof(key1)); | |||
memset(key3, 0xaa, sizeof(key3)); | |||
memset(data3, 0xdd, sizeof(data3)); | |||
// RFC 2202 test vectors | |||
test(hmac, key1, 16, data1, sizeof(data1)); | |||
test(hmac, key2, sizeof(key2), data2, sizeof(data2)); | |||
test(hmac, key3, 16, data3, sizeof(data3)); | |||
test(hmac, key3, 80, data4, sizeof(data4)); | |||
test(hmac, key3, 80, data5, sizeof(data5)); | |||
av_hmac_free(hmac); | |||
/* SHA-1 */ | |||
hmac = av_hmac_alloc(AV_HMAC_SHA1); | |||
if (!hmac) | |||
return 1; | |||
// RFC 2202 test vectors | |||
test(hmac, key1, sizeof(key1), data1, sizeof(data1)); | |||
test(hmac, key2, sizeof(key2), data2, sizeof(data2)); | |||
test(hmac, key3, 20, data3, sizeof(data3)); | |||
test(hmac, key3, 80, data4, sizeof(data4)); | |||
test(hmac, key3, 80, data5, sizeof(data5)); | |||
av_hmac_free(hmac); | |||
/* SHA-2 */ | |||
while (i <= AV_HMAC_SHA256) { | |||
hmac = av_hmac_alloc(i); | |||
// RFC 4231 test vectors | |||
test(hmac, key1, sizeof(key1), data1, sizeof(data1)); | |||
test(hmac, key2, sizeof(key2), data2, sizeof(data2)); | |||
test(hmac, key3, 20, data3, sizeof(data3)); | |||
test(hmac, key3, sizeof(key3), data4, sizeof(data4)); | |||
test(hmac, key3, sizeof(key3), data6, sizeof(data6)); | |||
av_hmac_free(hmac); | |||
i++; | |||
} | |||
return 0; | |||
} | |||
#endif /* TEST */ |
@@ -0,0 +1,56 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include "log.h" | |||
#include "timer.h" | |||
#include "lfg.h" | |||
int main(void) | |||
{ | |||
int x = 0; | |||
int i, j; | |||
AVLFG state; | |||
av_lfg_init(&state, 0xdeadbeef); | |||
for (j = 0; j < 10000; j++) { | |||
START_TIMER | |||
for (i = 0; i < 624; i++) | |||
x += av_lfg_get(&state); | |||
STOP_TIMER("624 calls of av_lfg_get"); | |||
} | |||
av_log(NULL, AV_LOG_ERROR, "final value:%X\n", x); | |||
/* BMG usage example */ | |||
{ | |||
double mean = 1000; | |||
double stddev = 53; | |||
av_lfg_init(&state, 42); | |||
for (i = 0; i < 1000; i += 2) { | |||
double bmg_out[2]; | |||
av_bmg_get(&state, bmg_out); | |||
av_log(NULL, AV_LOG_INFO, | |||
"%f\n%f\n", | |||
bmg_out[0] * stddev + mean, | |||
bmg_out[1] * stddev + mean); | |||
} | |||
} | |||
return 0; | |||
} |
@@ -58,43 +58,3 @@ void av_bmg_get(AVLFG *lfg, double out[2]) | |||
out[0] = x1 * w; | |||
out[1] = x2 * w; | |||
} | |||
#ifdef TEST | |||
#include "log.h" | |||
#include "timer.h" | |||
int main(void) | |||
{ | |||
int x = 0; | |||
int i, j; | |||
AVLFG state; | |||
av_lfg_init(&state, 0xdeadbeef); | |||
for (j = 0; j < 10000; j++) { | |||
START_TIMER | |||
for (i = 0; i < 624; i++) | |||
x += av_lfg_get(&state); | |||
STOP_TIMER("624 calls of av_lfg_get"); | |||
} | |||
av_log(NULL, AV_LOG_ERROR, "final value:%X\n", x); | |||
/* BMG usage example */ | |||
{ | |||
double mean = 1000; | |||
double stddev = 53; | |||
av_lfg_init(&state, 42); | |||
for (i = 0; i < 1000; i += 2) { | |||
double bmg_out[2]; | |||
av_bmg_get(&state, bmg_out); | |||
av_log(NULL, AV_LOG_INFO, | |||
"%f\n%f\n", | |||
bmg_out[0] * stddev + mean, | |||
bmg_out[1] * stddev + mean); | |||
} | |||
} | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,54 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <limits.h> | |||
#include <stdio.h> | |||
#include "internal.h" | |||
#include "lfg.h" | |||
#include "lls.h" | |||
int main(void) | |||
{ | |||
LLSModel m; | |||
int i, order; | |||
AVLFG lfg; | |||
av_lfg_init(&lfg, 1); | |||
avpriv_init_lls(&m, 3); | |||
for (i = 0; i < 100; i++) { | |||
LOCAL_ALIGNED(32, double, var, [4]); | |||
double eval; | |||
var[0] = (av_lfg_get(&lfg) / (double) UINT_MAX - 0.5) * 2; | |||
var[1] = var[0] + av_lfg_get(&lfg) / (double) UINT_MAX - 0.5; | |||
var[2] = var[1] + av_lfg_get(&lfg) / (double) UINT_MAX - 0.5; | |||
var[3] = var[2] + av_lfg_get(&lfg) / (double) UINT_MAX - 0.5; | |||
m.update_lls(&m, var); | |||
avpriv_solve_lls(&m, 0.001, 0); | |||
for (order = 0; order < 3; order++) { | |||
eval = m.evaluate_lls(&m, var + 1, order); | |||
printf("real:%9f order:%d pred:%9f var:%f coeffs:%f %9f %9f\n", | |||
var[0], order, eval, sqrt(m.variance[order] / (i + 1)), | |||
m.coeff[order][0], m.coeff[order][1], | |||
m.coeff[order][2]); | |||
} | |||
} | |||
return 0; | |||
} |
@@ -121,41 +121,3 @@ av_cold void avpriv_init_lls(LLSModel *m, int indep_count) | |||
if (ARCH_X86) | |||
ff_init_lls_x86(m); | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
#include <limits.h> | |||
#include "lfg.h" | |||
int main(void) | |||
{ | |||
LLSModel m; | |||
int i, order; | |||
AVLFG lfg; | |||
av_lfg_init(&lfg, 1); | |||
avpriv_init_lls(&m, 3); | |||
for (i = 0; i < 100; i++) { | |||
LOCAL_ALIGNED(32, double, var, [4]); | |||
double eval; | |||
var[0] = (av_lfg_get(&lfg) / (double) UINT_MAX - 0.5) * 2; | |||
var[1] = var[0] + av_lfg_get(&lfg) / (double) UINT_MAX - 0.5; | |||
var[2] = var[1] + av_lfg_get(&lfg) / (double) UINT_MAX - 0.5; | |||
var[3] = var[2] + av_lfg_get(&lfg) / (double) UINT_MAX - 0.5; | |||
m.update_lls(&m, var); | |||
avpriv_solve_lls(&m, 0.001, 0); | |||
for (order = 0; order < 3; order++) { | |||
eval = m.evaluate_lls(&m, var + 1, order); | |||
printf("real:%9f order:%d pred:%9f var:%f coeffs:%f %9f %9f\n", | |||
var[0], order, eval, sqrt(m.variance[order] / (i + 1)), | |||
m.coeff[order][0], m.coeff[order][1], | |||
m.coeff[order][2]); | |||
} | |||
} | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,54 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; 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 "md5.h" | |||
static void print_md5(uint8_t *md5) | |||
{ | |||
int i; | |||
for (i = 0; i < 16; i++) | |||
printf("%02x", md5[i]); | |||
printf("\n"); | |||
} | |||
int main(void) | |||
{ | |||
uint8_t md5val[16]; | |||
int i; | |||
uint8_t in[1000]; | |||
for (i = 0; i < 1000; i++) | |||
in[i] = i * i; | |||
av_md5_sum(md5val, in, 1000); | |||
print_md5(md5val); | |||
av_md5_sum(md5val, in, 63); | |||
print_md5(md5val); | |||
av_md5_sum(md5val, in, 64); | |||
print_md5(md5val); | |||
av_md5_sum(md5val, in, 65); | |||
print_md5(md5val); | |||
for (i = 0; i < 1000; i++) | |||
in[i] = i % 127; | |||
av_md5_sum(md5val, in, 999); | |||
print_md5(md5val); | |||
return 0; | |||
} |
@@ -185,39 +185,3 @@ void av_md5_sum(uint8_t *dst, const uint8_t *src, const int len) | |||
av_md5_update(&ctx, src, len); | |||
av_md5_final(&ctx, dst); | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
static void print_md5(uint8_t *md5) | |||
{ | |||
int i; | |||
for (i = 0; i < 16; i++) | |||
printf("%02x", md5[i]); | |||
printf("\n"); | |||
} | |||
int main(void) | |||
{ | |||
uint8_t md5val[16]; | |||
int i; | |||
uint8_t in[1000]; | |||
for (i = 0; i < 1000; i++) | |||
in[i] = i * i; | |||
av_md5_sum(md5val, in, 1000); | |||
print_md5(md5val); | |||
av_md5_sum(md5val, in, 63); | |||
print_md5(md5val); | |||
av_md5_sum(md5val, in, 64); | |||
print_md5(md5val); | |||
av_md5_sum(md5val, in, 65); | |||
print_md5(md5val); | |||
for (i = 0; i < 1000; i++) | |||
in[i] = i % 127; | |||
av_md5_sum(md5val, in, 999); | |||
print_md5(md5val); | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,110 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include <limits.h> | |||
#include <stdio.h> | |||
#include "common.h" | |||
#include "error.h" | |||
#include "log.h" | |||
#include "mem.h" | |||
#include "rational.h" | |||
#include "opt.h" | |||
typedef struct TestContext { | |||
const AVClass *class; | |||
int num; | |||
int toggle; | |||
char *string; | |||
int flags; | |||
AVRational rational; | |||
} TestContext; | |||
#define OFFSET(x) offsetof(TestContext, x) | |||
#define TEST_FLAG_COOL 01 | |||
#define TEST_FLAG_LAME 02 | |||
#define TEST_FLAG_MU 04 | |||
static const AVOption test_options[] = { | |||
{ "num", "set num", OFFSET(num), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 100 }, | |||
{ "toggle", "set toggle", OFFSET(toggle), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1 }, | |||
{ "rational", "set rational", OFFSET(rational), AV_OPT_TYPE_RATIONAL, { .dbl = 0 }, 0, 10 }, | |||
{ "string", "set string", OFFSET(string), AV_OPT_TYPE_STRING, { 0 }, CHAR_MIN, CHAR_MAX }, | |||
{ "flags", "set flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, { .i64 = 0 }, 0, INT_MAX, 0, "flags"}, | |||
{ "cool", "set cool flag ", 0, AV_OPT_TYPE_CONST, { .i64 = TEST_FLAG_COOL }, INT_MIN, INT_MAX, 0, "flags"}, | |||
{ "lame", "set lame flag ", 0, AV_OPT_TYPE_CONST, { .i64 = TEST_FLAG_LAME }, INT_MIN, INT_MAX, 0, "flags"}, | |||
{ "mu", "set mu flag ", 0, AV_OPT_TYPE_CONST, { .i64 = TEST_FLAG_MU }, INT_MIN, INT_MAX, 0, "flags"}, | |||
{ NULL }, | |||
}; | |||
static const char *test_get_name(void *ctx) | |||
{ | |||
return "test"; | |||
} | |||
static const AVClass test_class = { | |||
"TestContext", | |||
test_get_name, | |||
test_options | |||
}; | |||
int main(void) | |||
{ | |||
int i; | |||
TestContext test_ctx = { .class = &test_class }; | |||
const char *options[] = { | |||
"", | |||
":", | |||
"=", | |||
"foo=:", | |||
":=foo", | |||
"=foo", | |||
"foo=", | |||
"foo", | |||
"foo=val", | |||
"foo==val", | |||
"toggle=:", | |||
"string=:", | |||
"toggle=1 : foo", | |||
"toggle=100", | |||
"toggle==1", | |||
"flags=+mu-lame : num=42: toggle=0", | |||
"num=42 : string=blahblah", | |||
"rational=0 : rational=1/2 : rational=1/-1", | |||
"rational=-1/0", | |||
}; | |||
printf("\nTesting av_set_options_string()\n"); | |||
av_opt_set_defaults(&test_ctx); | |||
test_ctx.string = av_strdup("default"); | |||
if (!test_ctx.string) | |||
return AVERROR(ENOMEM); | |||
av_log_set_level(AV_LOG_DEBUG); | |||
for (i = 0; i < FF_ARRAY_ELEMS(options); i++) { | |||
av_log(&test_ctx, AV_LOG_DEBUG, "Setting options string '%s'\n", options[i]); | |||
if (av_set_options_string(&test_ctx, options[i], "=", ":") < 0) | |||
av_log(&test_ctx, AV_LOG_ERROR, "Error setting options string: '%s'\n", options[i]); | |||
printf("\n"); | |||
} | |||
return 0; | |||
} |
@@ -885,90 +885,3 @@ int av_opt_copy(void *dst, const void *src) | |||
} | |||
return ret; | |||
} | |||
#ifdef TEST | |||
typedef struct TestContext { | |||
const AVClass *class; | |||
int num; | |||
int toggle; | |||
char *string; | |||
int flags; | |||
AVRational rational; | |||
} TestContext; | |||
#define OFFSET(x) offsetof(TestContext, x) | |||
#define TEST_FLAG_COOL 01 | |||
#define TEST_FLAG_LAME 02 | |||
#define TEST_FLAG_MU 04 | |||
static const AVOption test_options[] = { | |||
{ "num", "set num", OFFSET(num), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 100 }, | |||
{ "toggle", "set toggle", OFFSET(toggle), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1 }, | |||
{ "rational", "set rational", OFFSET(rational), AV_OPT_TYPE_RATIONAL, { .dbl = 0 }, 0, 10 }, | |||
{ "string", "set string", OFFSET(string), AV_OPT_TYPE_STRING, { 0 }, CHAR_MIN, CHAR_MAX }, | |||
{ "flags", "set flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, { .i64 = 0 }, 0, INT_MAX, 0, "flags"}, | |||
{ "cool", "set cool flag ", 0, AV_OPT_TYPE_CONST, { .i64 = TEST_FLAG_COOL }, INT_MIN, INT_MAX, 0, "flags"}, | |||
{ "lame", "set lame flag ", 0, AV_OPT_TYPE_CONST, { .i64 = TEST_FLAG_LAME }, INT_MIN, INT_MAX, 0, "flags"}, | |||
{ "mu", "set mu flag ", 0, AV_OPT_TYPE_CONST, { .i64 = TEST_FLAG_MU }, INT_MIN, INT_MAX, 0, "flags"}, | |||
{ NULL }, | |||
}; | |||
static const char *test_get_name(void *ctx) | |||
{ | |||
return "test"; | |||
} | |||
static const AVClass test_class = { | |||
"TestContext", | |||
test_get_name, | |||
test_options | |||
}; | |||
int main(void) | |||
{ | |||
int i; | |||
TestContext test_ctx = { .class = &test_class }; | |||
const char *options[] = { | |||
"", | |||
":", | |||
"=", | |||
"foo=:", | |||
":=foo", | |||
"=foo", | |||
"foo=", | |||
"foo", | |||
"foo=val", | |||
"foo==val", | |||
"toggle=:", | |||
"string=:", | |||
"toggle=1 : foo", | |||
"toggle=100", | |||
"toggle==1", | |||
"flags=+mu-lame : num=42: toggle=0", | |||
"num=42 : string=blahblah", | |||
"rational=0 : rational=1/2 : rational=1/-1", | |||
"rational=-1/0", | |||
}; | |||
printf("\nTesting av_set_options_string()\n"); | |||
av_opt_set_defaults(&test_ctx); | |||
test_ctx.string = av_strdup("default"); | |||
if (!test_ctx.string) | |||
return AVERROR(ENOMEM); | |||
av_log_set_level(AV_LOG_DEBUG); | |||
for (i = 0; i < FF_ARRAY_ELEMS(options); i++) { | |||
av_log(&test_ctx, AV_LOG_DEBUG, "Setting options string '%s'\n", options[i]); | |||
if (av_set_options_string(&test_ctx, options[i], "=", ":") < 0) | |||
av_log(&test_ctx, AV_LOG_ERROR, "Error setting options string: '%s'\n", options[i]); | |||
printf("\n"); | |||
} | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,116 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; 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 "common.h" | |||
#include "log.h" | |||
#include "rational.h" | |||
#include "parseutils.h" | |||
int main(void) | |||
{ | |||
int i; | |||
uint8_t rgba[4]; | |||
static const char *const rates[] = { | |||
"-inf", | |||
"inf", | |||
"nan", | |||
"123/0", | |||
"-123 / 0", | |||
"", | |||
"/", | |||
" 123 / 321", | |||
"foo/foo", | |||
"foo/1", | |||
"1/foo", | |||
"0/0", | |||
"/0", | |||
"1/", | |||
"1", | |||
"0", | |||
"-123/123", | |||
"-foo", | |||
"123.23", | |||
".23", | |||
"-.23", | |||
"-0.234", | |||
"-0.0000001", | |||
" 21332.2324 ", | |||
" -21332.2324 ", | |||
}; | |||
static const char *const color_names[] = { | |||
"foo", | |||
"red", | |||
"Red ", | |||
"RED", | |||
"Violet", | |||
"Yellow", | |||
"Red", | |||
"0x000000", | |||
"0x0000000", | |||
"0xff000000", | |||
"0x3e34ff", | |||
"0x3e34ffaa", | |||
"0xffXXee", | |||
"0xfoobar", | |||
"0xffffeeeeeeee", | |||
"#ff0000", | |||
"#ffXX00", | |||
"ff0000", | |||
"ffXX00", | |||
"red@foo", | |||
"random@10", | |||
"0xff0000@1.0", | |||
"red@", | |||
"red@0xfff", | |||
"red@0xf", | |||
"red@2", | |||
"red@0.1", | |||
"red@-1", | |||
"red@0.5", | |||
"red@1.0", | |||
"red@256", | |||
"red@10foo", | |||
"red@-1.0", | |||
"red@-0.0", | |||
}; | |||
printf("Testing av_parse_video_rate()\n"); | |||
for (i = 0; i < FF_ARRAY_ELEMS(rates); i++) { | |||
int ret; | |||
AVRational q = { 0, 0 }; | |||
ret = av_parse_video_rate(&q, rates[i]); | |||
printf("'%s' -> %d/%d %s\n", | |||
rates[i], q.num, q.den, ret ? "ERROR" : "OK"); | |||
} | |||
printf("\nTesting av_parse_color()\n"); | |||
av_log_set_level(AV_LOG_DEBUG); | |||
for (i = 0; i < FF_ARRAY_ELEMS(color_names); i++) { | |||
if (av_parse_color(rgba, color_names[i], -1, NULL) >= 0) | |||
printf("%s -> R(%d) G(%d) B(%d) A(%d)\n", | |||
color_names[i], rgba[0], rgba[1], rgba[2], rgba[3]); | |||
} | |||
return 0; | |||
} |
@@ -654,98 +654,3 @@ int av_find_info_tag(char *arg, int arg_size, const char *tag1, const char *info | |||
} | |||
return 0; | |||
} | |||
#ifdef TEST | |||
int main(void) | |||
{ | |||
int i; | |||
uint8_t rgba[4]; | |||
static const char *const rates[] = { | |||
"-inf", | |||
"inf", | |||
"nan", | |||
"123/0", | |||
"-123 / 0", | |||
"", | |||
"/", | |||
" 123 / 321", | |||
"foo/foo", | |||
"foo/1", | |||
"1/foo", | |||
"0/0", | |||
"/0", | |||
"1/", | |||
"1", | |||
"0", | |||
"-123/123", | |||
"-foo", | |||
"123.23", | |||
".23", | |||
"-.23", | |||
"-0.234", | |||
"-0.0000001", | |||
" 21332.2324 ", | |||
" -21332.2324 ", | |||
}; | |||
static const char *const color_names[] = { | |||
"foo", | |||
"red", | |||
"Red ", | |||
"RED", | |||
"Violet", | |||
"Yellow", | |||
"Red", | |||
"0x000000", | |||
"0x0000000", | |||
"0xff000000", | |||
"0x3e34ff", | |||
"0x3e34ffaa", | |||
"0xffXXee", | |||
"0xfoobar", | |||
"0xffffeeeeeeee", | |||
"#ff0000", | |||
"#ffXX00", | |||
"ff0000", | |||
"ffXX00", | |||
"red@foo", | |||
"random@10", | |||
"0xff0000@1.0", | |||
"red@", | |||
"red@0xfff", | |||
"red@0xf", | |||
"red@2", | |||
"red@0.1", | |||
"red@-1", | |||
"red@0.5", | |||
"red@1.0", | |||
"red@256", | |||
"red@10foo", | |||
"red@-1.0", | |||
"red@-0.0", | |||
}; | |||
printf("Testing av_parse_video_rate()\n"); | |||
for (i = 0; i < FF_ARRAY_ELEMS(rates); i++) { | |||
int ret; | |||
AVRational q = { 0, 0 }; | |||
ret = av_parse_video_rate(&q, rates[i]); | |||
printf("'%s' -> %d/%d %s\n", | |||
rates[i], q.num, q.den, ret ? "ERROR" : "OK"); | |||
} | |||
printf("\nTesting av_parse_color()\n"); | |||
av_log_set_level(AV_LOG_DEBUG); | |||
for (i = 0; i < FF_ARRAY_ELEMS(color_names); i++) { | |||
if (av_parse_color(rgba, color_names[i], -1, NULL) >= 0) | |||
printf("%s -> R(%d) G(%d) B(%d) A(%d)\n", | |||
color_names[i], rgba[0], rgba[1], rgba[2], rgba[3]); | |||
} | |||
return 0; | |||
} | |||
#endif /* TEST */ |
@@ -0,0 +1,69 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include "sha.c" | |||
#include <stdio.h> | |||
int main(void) | |||
{ | |||
int i, j, k; | |||
AVSHA ctx; | |||
unsigned char digest[32]; | |||
const int lengths[3] = { 160, 224, 256 }; | |||
for (j = 0; j < 3; j++) { | |||
printf("Testing SHA-%d\n", lengths[j]); | |||
for (k = 0; k < 3; k++) { | |||
av_sha_init(&ctx, lengths[j]); | |||
if (k == 0) | |||
av_sha_update(&ctx, "abc", 3); | |||
else if (k == 1) | |||
av_sha_update(&ctx, "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq", 56); | |||
else | |||
for (i = 0; i < 1000*1000; i++) | |||
av_sha_update(&ctx, "a", 1); | |||
av_sha_final(&ctx, digest); | |||
for (i = 0; i < lengths[j] >> 3; i++) | |||
printf("%02X", digest[i]); | |||
putchar('\n'); | |||
} | |||
switch (j) { | |||
case 0: | |||
//test vectors (from FIPS PUB 180-1) | |||
printf("A9993E36 4706816A BA3E2571 7850C26C 9CD0D89D\n" | |||
"84983E44 1C3BD26E BAAE4AA1 F95129E5 E54670F1\n" | |||
"34AA973C D4C4DAA4 F61EEB2B DBAD2731 6534016F\n"); | |||
break; | |||
case 1: | |||
//test vectors (from FIPS PUB 180-2 Appendix A) | |||
printf("23097d22 3405d822 8642a477 bda255b3 2aadbce4 bda0b3f7 e36c9da7\n" | |||
"75388b16 512776cc 5dba5da1 fd890150 b0c6455c b4f58b19 52522525\n" | |||
"20794655 980c91d8 bbb4c1ea 97618a4b f03f4258 1948b2ee 4ee7ad67\n"); | |||
break; | |||
case 2: | |||
//test vectors (from FIPS PUB 180-2) | |||
printf("ba7816bf 8f01cfea 414140de 5dae2223 b00361a3 96177a9c b410ff61 f20015ad\n" | |||
"248d6a61 d20638b8 e5c02693 0c3e6039 a33ce459 64ff2167 f6ecedd4 19db06c1\n" | |||
"cdc76e5c 9914fb92 81a1c7e2 84d73e67 f1809a48 a497200e 046d39cc c7112cd0\n"); | |||
break; | |||
} | |||
} | |||
return 0; | |||
} |
@@ -329,55 +329,3 @@ void av_sha_final(AVSHA* ctx, uint8_t *digest) | |||
for (i = 0; i < ctx->digest_len; i++) | |||
AV_WB32(digest + i*4, ctx->state[i]); | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
int main(void) | |||
{ | |||
int i, j, k; | |||
AVSHA ctx; | |||
unsigned char digest[32]; | |||
const int lengths[3] = { 160, 224, 256 }; | |||
for (j = 0; j < 3; j++) { | |||
printf("Testing SHA-%d\n", lengths[j]); | |||
for (k = 0; k < 3; k++) { | |||
av_sha_init(&ctx, lengths[j]); | |||
if (k == 0) | |||
av_sha_update(&ctx, "abc", 3); | |||
else if (k == 1) | |||
av_sha_update(&ctx, "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq", 56); | |||
else | |||
for (i = 0; i < 1000*1000; i++) | |||
av_sha_update(&ctx, "a", 1); | |||
av_sha_final(&ctx, digest); | |||
for (i = 0; i < lengths[j] >> 3; i++) | |||
printf("%02X", digest[i]); | |||
putchar('\n'); | |||
} | |||
switch (j) { | |||
case 0: | |||
//test vectors (from FIPS PUB 180-1) | |||
printf("A9993E36 4706816A BA3E2571 7850C26C 9CD0D89D\n" | |||
"84983E44 1C3BD26E BAAE4AA1 F95129E5 E54670F1\n" | |||
"34AA973C D4C4DAA4 F61EEB2B DBAD2731 6534016F\n"); | |||
break; | |||
case 1: | |||
//test vectors (from FIPS PUB 180-2 Appendix A) | |||
printf("23097d22 3405d822 8642a477 bda255b3 2aadbce4 bda0b3f7 e36c9da7\n" | |||
"75388b16 512776cc 5dba5da1 fd890150 b0c6455c b4f58b19 52522525\n" | |||
"20794655 980c91d8 bbb4c1ea 97618a4b f03f4258 1948b2ee 4ee7ad67\n"); | |||
break; | |||
case 2: | |||
//test vectors (from FIPS PUB 180-2) | |||
printf("ba7816bf 8f01cfea 414140de 5dae2223 b00361a3 96177a9c b410ff61 f20015ad\n" | |||
"248d6a61 d20638b8 e5c02693 0c3e6039 a33ce459 64ff2167 f6ecedd4 19db06c1\n" | |||
"cdc76e5c 9914fb92 81a1c7e2 84d73e67 f1809a48 a497200e 046d39cc c7112cd0\n"); | |||
break; | |||
} | |||
} | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,110 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; if not, write to the Free Software | |||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |||
*/ | |||
#include "tree.c" | |||
#include <stdint.h> | |||
#include "common.h" | |||
#include "lfg.h" | |||
#include "log.h" | |||
static int check(AVTreeNode *t) | |||
{ | |||
if (t) { | |||
int left = check(t->child[0]); | |||
int right = check(t->child[1]); | |||
if (left > 999 || right > 999) | |||
return 1000; | |||
if (right - left != t->state) | |||
return 1000; | |||
if (t->state > 1 || t->state < -1) | |||
return 1000; | |||
return FFMAX(left, right) + 1; | |||
} | |||
return 0; | |||
} | |||
static void print(AVTreeNode *t, int depth) | |||
{ | |||
int i; | |||
for (i = 0; i < depth * 4; i++) | |||
av_log(NULL, AV_LOG_ERROR, " "); | |||
if (t) { | |||
av_log(NULL, AV_LOG_ERROR, "Node %p %2d %p\n", t, t->state, t->elem); | |||
print(t->child[0], depth + 1); | |||
print(t->child[1], depth + 1); | |||
} else | |||
av_log(NULL, AV_LOG_ERROR, "NULL\n"); | |||
} | |||
static int cmp(void *a, const void *b) | |||
{ | |||
return (uint8_t *) a - (const uint8_t *) b; | |||
} | |||
int main(void) | |||
{ | |||
int i; | |||
AVTreeNode *root = NULL, *node = NULL; | |||
AVLFG prng; | |||
av_lfg_init(&prng, 1); | |||
for (i = 0; i < 10000; i++) { | |||
AVTreeNode *node2 = NULL; | |||
intptr_t j = av_lfg_get(&prng) % 86294; | |||
void *ret, *jj = (void *)(j + 1); | |||
while (ret = av_tree_find(root, jj, cmp, NULL)) { | |||
j = av_lfg_get(&prng) % 86294; | |||
jj = (void *)(j + 1); | |||
} | |||
if (check(root) > 999) { | |||
av_log(NULL, AV_LOG_ERROR, "FATAL error %d\n", i); | |||
print(root, 0); | |||
return 1; | |||
} | |||
if (!node) | |||
node = av_tree_node_alloc(); | |||
if (!node) { | |||
av_log(NULL, AV_LOG_ERROR, "Memory allocation failure.\n"); | |||
return 1; | |||
} | |||
av_tree_insert(&root, jj, cmp, &node); | |||
while (ret = av_tree_find(root, jj, cmp, NULL)) { | |||
j = av_lfg_get(&prng) % 86294; | |||
jj = (void *)(j + 1); | |||
} | |||
ret = av_tree_insert(&root, jj, cmp, &node2); | |||
if (ret != jj) | |||
av_tree_destroy(node2); | |||
ret = av_tree_find(root, jj, cmp, NULL); | |||
if (ret) | |||
av_log(NULL, AV_LOG_ERROR, "removal failure %d\n", i); | |||
} | |||
av_tree_destroy(root); | |||
return 0; | |||
} |
@@ -19,7 +19,6 @@ | |||
*/ | |||
#include "error.h" | |||
#include "log.h" | |||
#include "mem.h" | |||
#include "tree.h" | |||
@@ -164,94 +163,3 @@ void av_tree_enumerate(AVTreeNode *t, void *opaque, | |||
av_tree_enumerate(t->child[1], opaque, cmp, enu); | |||
} | |||
} | |||
#ifdef TEST | |||
#include "common.h" | |||
#include "lfg.h" | |||
static int check(AVTreeNode *t) | |||
{ | |||
if (t) { | |||
int left = check(t->child[0]); | |||
int right = check(t->child[1]); | |||
if (left > 999 || right > 999) | |||
return 1000; | |||
if (right - left != t->state) | |||
return 1000; | |||
if (t->state > 1 || t->state < -1) | |||
return 1000; | |||
return FFMAX(left, right) + 1; | |||
} | |||
return 0; | |||
} | |||
static void print(AVTreeNode *t, int depth) | |||
{ | |||
int i; | |||
for (i = 0; i < depth * 4; i++) | |||
av_log(NULL, AV_LOG_ERROR, " "); | |||
if (t) { | |||
av_log(NULL, AV_LOG_ERROR, "Node %p %2d %p\n", t, t->state, t->elem); | |||
print(t->child[0], depth + 1); | |||
print(t->child[1], depth + 1); | |||
} else | |||
av_log(NULL, AV_LOG_ERROR, "NULL\n"); | |||
} | |||
static int cmp(void *a, const void *b) | |||
{ | |||
return (uint8_t *) a - (const uint8_t *) b; | |||
} | |||
int main(void) | |||
{ | |||
int i; | |||
AVTreeNode *root = NULL, *node = NULL; | |||
AVLFG prng; | |||
av_lfg_init(&prng, 1); | |||
for (i = 0; i < 10000; i++) { | |||
AVTreeNode *node2 = NULL; | |||
intptr_t j = av_lfg_get(&prng) % 86294; | |||
void *ret, *jj = (void *)(j + 1); | |||
while (ret = av_tree_find(root, jj, cmp, NULL)) { | |||
j = av_lfg_get(&prng) % 86294; | |||
jj = (void *)(j + 1); | |||
} | |||
if (check(root) > 999) { | |||
av_log(NULL, AV_LOG_ERROR, "FATAL error %d\n", i); | |||
print(root, 0); | |||
return 1; | |||
} | |||
if (!node) | |||
node = av_tree_node_alloc(); | |||
if (!node) { | |||
av_log(NULL, AV_LOG_ERROR, "Memory allocation failure.\n"); | |||
return 1; | |||
} | |||
av_tree_insert(&root, jj, cmp, &node); | |||
while (ret = av_tree_find(root, jj, cmp, NULL)) { | |||
j = av_lfg_get(&prng) % 86294; | |||
jj = (void *)(j + 1); | |||
} | |||
ret = av_tree_insert(&root, jj, cmp, &node2); | |||
if (ret != jj) | |||
av_tree_destroy(node2); | |||
ret = av_tree_find(root, jj, cmp, NULL); | |||
if (ret) | |||
av_log(NULL, AV_LOG_ERROR, "removal failure %d\n", i); | |||
} | |||
av_tree_destroy(root); | |||
return 0; | |||
} | |||
#endif |
@@ -0,0 +1,120 @@ | |||
/* | |||
* This file is part of Libav. | |||
* | |||
* Libav 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. | |||
* | |||
* Libav 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 Libav; 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 <stdlib.h> | |||
#include <string.h> | |||
#include "intreadwrite.h" | |||
#include "xtea.h" | |||
#define XTEA_NUM_TESTS 6 | |||
static const uint8_t xtea_test_key[XTEA_NUM_TESTS][16] = { | |||
{ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, | |||
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f }, | |||
{ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, | |||
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f }, | |||
{ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, | |||
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 } | |||
}; | |||
static const uint8_t xtea_test_pt[XTEA_NUM_TESTS][8] = { | |||
{ 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48 }, | |||
{ 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 }, | |||
{ 0x5a, 0x5b, 0x6e, 0x27, 0x89, 0x48, 0xd7, 0x7f }, | |||
{ 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48 }, | |||
{ 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 }, | |||
{ 0x70, 0xe1, 0x22, 0x5d, 0x6e, 0x4e, 0x76, 0x55 } | |||
}; | |||
static const uint8_t xtea_test_ct[XTEA_NUM_TESTS][8] = { | |||
{ 0x49, 0x7d, 0xf3, 0xd0, 0x72, 0x61, 0x2c, 0xb5 }, | |||
{ 0xe7, 0x8f, 0x2d, 0x13, 0x74, 0x43, 0x41, 0xd8 }, | |||
{ 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 }, | |||
{ 0xa0, 0x39, 0x05, 0x89, 0xf8, 0xb8, 0xef, 0xa5 }, | |||
{ 0xed, 0x23, 0x37, 0x5a, 0x82, 0x1a, 0x8c, 0x2d }, | |||
{ 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 } | |||
}; | |||
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)) | |||
{ | |||
crypt(ctx, dst, src, len, iv, dir); | |||
if (memcmp(dst, ref, 8*len)) { | |||
int i; | |||
printf("%s failed\ngot ", test); | |||
for (i = 0; i < 8*len; i++) | |||
printf("%02x ", dst[i]); | |||
printf("\nexpected "); | |||
for (i = 0; i < 8*len; i++) | |||
printf("%02x ", ref[i]); | |||
printf("\n"); | |||
exit(1); | |||
} | |||
} | |||
int main(void) | |||
{ | |||
AVXTEA ctx; | |||
uint8_t buf[16], iv[8]; | |||
int i, j; | |||
const uint8_t src[32] = "HelloWorldHelloWorldHelloWorld"; | |||
uint8_t ct[32]; | |||
uint8_t pl[32]; | |||
for (i = 0; i < XTEA_NUM_TESTS; i++) { | |||
av_xtea_init(&ctx, xtea_test_key[i]); | |||
test_xtea(&ctx, buf, xtea_test_pt[i], xtea_test_ct[i], 1, NULL, 0, "encryption", av_xtea_crypt); | |||
test_xtea(&ctx, buf, xtea_test_ct[i], xtea_test_pt[i], 1, NULL, 1, "decryption", av_xtea_crypt); | |||
for (j = 0; j < 4; j++) | |||
AV_WL32(&buf[4*j], AV_RB32(&xtea_test_key[i][4*j])); | |||
av_xtea_le_init(&ctx, buf); | |||
for (j = 0; j < 2; j++) { | |||
AV_WL32(&ct[4*j], AV_RB32(&xtea_test_ct[i][4*j])); | |||
AV_WL32(&pl[4*j], AV_RB32(&xtea_test_pt[i][4*j])); | |||
} | |||
test_xtea(&ctx, buf, pl, ct, 1, NULL, 0, "encryption", av_xtea_le_crypt); | |||
test_xtea(&ctx, buf, ct, pl, 1, NULL, 1, "decryption", av_xtea_le_crypt); | |||
/* encrypt */ | |||
memcpy(iv, "HALLO123", 8); | |||
av_xtea_crypt(&ctx, ct, src, 4, iv, 0); | |||
/* decrypt into pl */ | |||
memcpy(iv, "HALLO123", 8); | |||
test_xtea(&ctx, pl, ct, src, 4, iv, 1, "CBC decryption", av_xtea_crypt); | |||
memcpy(iv, "HALLO123", 8); | |||
test_xtea(&ctx, ct, ct, src, 4, iv, 1, "CBC inplace decryption", av_xtea_crypt); | |||
} | |||
printf("Test encryption/decryption success.\n"); | |||
return 0; | |||
} |
@@ -173,103 +173,3 @@ void av_xtea_le_crypt(AVXTEA *ctx, uint8_t *dst, const uint8_t *src, int count, | |||
{ | |||
xtea_crypt(ctx, dst, src, count, iv, decrypt, xtea_le_crypt_ecb); | |||
} | |||
#ifdef TEST | |||
#include <stdio.h> | |||
#define XTEA_NUM_TESTS 6 | |||
static const uint8_t xtea_test_key[XTEA_NUM_TESTS][16] = { | |||
{ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, | |||
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f }, | |||
{ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, | |||
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f }, | |||
{ 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, | |||
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, | |||
{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, | |||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 } | |||
}; | |||
static const uint8_t xtea_test_pt[XTEA_NUM_TESTS][8] = { | |||
{ 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48 }, | |||
{ 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 }, | |||
{ 0x5a, 0x5b, 0x6e, 0x27, 0x89, 0x48, 0xd7, 0x7f }, | |||
{ 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48 }, | |||
{ 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 }, | |||
{ 0x70, 0xe1, 0x22, 0x5d, 0x6e, 0x4e, 0x76, 0x55 } | |||
}; | |||
static const uint8_t xtea_test_ct[XTEA_NUM_TESTS][8] = { | |||
{ 0x49, 0x7d, 0xf3, 0xd0, 0x72, 0x61, 0x2c, 0xb5 }, | |||
{ 0xe7, 0x8f, 0x2d, 0x13, 0x74, 0x43, 0x41, 0xd8 }, | |||
{ 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 }, | |||
{ 0xa0, 0x39, 0x05, 0x89, 0xf8, 0xb8, 0xef, 0xa5 }, | |||
{ 0xed, 0x23, 0x37, 0x5a, 0x82, 0x1a, 0x8c, 0x2d }, | |||
{ 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41 } | |||
}; | |||
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)) | |||
{ | |||
crypt(ctx, dst, src, len, iv, dir); | |||
if (memcmp(dst, ref, 8*len)) { | |||
int i; | |||
printf("%s failed\ngot ", test); | |||
for (i = 0; i < 8*len; i++) | |||
printf("%02x ", dst[i]); | |||
printf("\nexpected "); | |||
for (i = 0; i < 8*len; i++) | |||
printf("%02x ", ref[i]); | |||
printf("\n"); | |||
exit(1); | |||
} | |||
} | |||
int main(void) | |||
{ | |||
AVXTEA ctx; | |||
uint8_t buf[16], iv[8]; | |||
int i, j; | |||
const uint8_t src[32] = "HelloWorldHelloWorldHelloWorld"; | |||
uint8_t ct[32]; | |||
uint8_t pl[32]; | |||
for (i = 0; i < XTEA_NUM_TESTS; i++) { | |||
av_xtea_init(&ctx, xtea_test_key[i]); | |||
test_xtea(&ctx, buf, xtea_test_pt[i], xtea_test_ct[i], 1, NULL, 0, "encryption", av_xtea_crypt); | |||
test_xtea(&ctx, buf, xtea_test_ct[i], xtea_test_pt[i], 1, NULL, 1, "decryption", av_xtea_crypt); | |||
for (j = 0; j < 4; j++) | |||
AV_WL32(&buf[4*j], AV_RB32(&xtea_test_key[i][4*j])); | |||
av_xtea_le_init(&ctx, buf); | |||
for (j = 0; j < 2; j++) { | |||
AV_WL32(&ct[4*j], AV_RB32(&xtea_test_ct[i][4*j])); | |||
AV_WL32(&pl[4*j], AV_RB32(&xtea_test_pt[i][4*j])); | |||
} | |||
test_xtea(&ctx, buf, pl, ct, 1, NULL, 0, "encryption", av_xtea_le_crypt); | |||
test_xtea(&ctx, buf, ct, pl, 1, NULL, 1, "decryption", av_xtea_le_crypt); | |||
/* encrypt */ | |||
memcpy(iv, "HALLO123", 8); | |||
av_xtea_crypt(&ctx, ct, src, 4, iv, 0); | |||
/* decrypt into pl */ | |||
memcpy(iv, "HALLO123", 8); | |||
test_xtea(&ctx, pl, ct, src, 4, iv, 1, "CBC decryption", av_xtea_crypt); | |||
memcpy(iv, "HALLO123", 8); | |||
test_xtea(&ctx, ct, ct, src, 4, iv, 1, "CBC inplace decryption", av_xtea_crypt); | |||
} | |||
printf("Test encryption/decryption success.\n"); | |||
return 0; | |||
} | |||
#endif |
@@ -10,18 +10,6 @@ INSTHEADERS := $(INSTHEADERS) $(HEADERS:%=$(SUBDIR)%) | |||
all-$(CONFIG_STATIC): $(SUBDIR)$(LIBNAME) | |||
all-$(CONFIG_SHARED): $(SUBDIR)$(SLIBNAME) | |||
$(SUBDIR)%-test.o: $(SUBDIR)%-test.c | |||
$(COMPILE_C) | |||
$(SUBDIR)%-test.o: $(SUBDIR)%.c | |||
$(COMPILE_C) | |||
$(SUBDIR)%-test.i: $(SUBDIR)%-test.c | |||
$(CC) $(CCFLAGS) $(CC_E) $< | |||
$(SUBDIR)%-test.i: $(SUBDIR)%.c | |||
$(CC) $(CCFLAGS) $(CC_E) $< | |||
$(SUBDIR)x86/%.o: $(SUBDIR)x86/%.asm | |||
$(DEPYASM) $(YASMFLAGS) -I $(<D)/ -M -o $@ $< > $(@:.o=.d) | |||
$(YASM) $(YASMFLAGS) -I $(<D)/ -o $@ $< | |||
@@ -29,7 +17,6 @@ $(SUBDIR)x86/%.o: $(SUBDIR)x86/%.asm | |||
LIBOBJS := $(OBJS) $(SUBDIR)%.h.o $(TESTOBJS) | |||
$(LIBOBJS) $(LIBOBJS:.o=.i): CPPFLAGS += -DHAVE_AV_CONFIG_H | |||
$(TESTOBJS) $(TESTOBJS:.o=.i): CPPFLAGS += -DTEST | |||
$(SUBDIR)$(LIBNAME): $(OBJS) | |||
$(RM) $@ | |||