/* BEGIN_HEADER */
#include "mbedtls/arc4.h"
/* END_HEADER */

/* BEGIN_DEPENDENCIES
 * depends_on:MBEDTLS_ARC4_C
 * END_DEPENDENCIES
 */

/* BEGIN_CASE */
void mbedtls_arc4_crypt( data_t * src_str, data_t * key_str, data_t * dst )
{
    unsigned char dst_str[1000];
    mbedtls_arc4_context ctx;

    memset(dst_str, 0x00, 1000);
    mbedtls_arc4_init( &ctx );


    mbedtls_arc4_setup(&ctx, key_str->x, key_str->len);
    TEST_ASSERT( mbedtls_arc4_crypt(&ctx, src_str->len,
                                    src_str->x, dst_str ) == 0 );

    TEST_ASSERT( mbedtls_test_hexcmp( dst_str, dst->x,
                                      src_str->len, dst->len ) == 0 );

exit:
    mbedtls_arc4_free( &ctx );
}
/* END_CASE */

/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
void arc4_selftest(  )
{
    TEST_ASSERT( mbedtls_arc4_self_test( 1 ) == 0 );
}
/* END_CASE */