check_buffer_sizes.c 2.62 KB
Newer Older
1 2 3 4 5 6 7 8 9
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "check_aec.h"

#define BUF_SIZE 1024 * 3

int check_block_sizes(struct test_state *state)
{
Mathis Rosenhauer's avatar
Mathis Rosenhauer committed
10 11
    for (int bs = 8; bs <= 64; bs *= 2) {
        int status;
12
        state->strm->block_size = bs;
13 14
        state->strm->rsi = (int)(state->buf_len
                                 / (bs * state->bytes_per_sample));
15

16
        status = encode_decode_large(state);
17 18 19 20 21 22 23 24
        if (status)
            return status;
    }
    return 0;
}

int check_block_sizes_short(struct test_state *state)
{
Mathis Rosenhauer's avatar
Mathis Rosenhauer committed
25 26 27
    size_t tmp = state->ibuf_len;
    for (int bs = 8; bs <= 64; bs *= 2) {
        int status;
28
        state->strm->block_size = bs;
29 30
        state->strm->rsi = (int)(state->buf_len
                                 / (bs * state->bytes_per_sample));
31
        state->ibuf_len = state->buf_len - 2 * bs + 4;
32
        status = encode_decode_large(state);
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
        if (status)
            return status;
        if (state->strm->total_out != state->buf_len) {
            printf("FAIL: Unexpected buffer length. Got %i expected %i\n",
                   (int)state->strm->total_out,
                   (int)state->buf_len);
            return 99;
        }
    }
    state->ibuf_len = tmp;
    return 0;
}

int check_rsi(struct test_state *state)
{
Mathis Rosenhauer's avatar
Mathis Rosenhauer committed
48 49
    int status;
    int size = state->bytes_per_sample;
50

Mathis Rosenhauer's avatar
Mathis Rosenhauer committed
51
    for (unsigned char *tmp = state->ubuf;
52
         tmp < state->ubuf + state->buf_len;
Mathis Rosenhauer's avatar
Mathis Rosenhauer committed
53
         tmp += 2 * state->bytes_per_sample) {
54 55 56 57 58 59 60 61 62
        state->out(tmp, state->xmax, size);
        state->out(tmp + size, state->xmin, size);
    }

    printf("Checking full rsi ... ");
    status = check_block_sizes(state);
    if (status)
        return status;

Mathis Rosenhauer's avatar
Mathis Rosenhauer committed
63
    printf ("%s\n", CHECK_PASS);
64 65 66 67 68 69

    printf("Checking short rsi ... ");
    status = check_block_sizes_short(state);
    if (status)
        return status;

Mathis Rosenhauer's avatar
Mathis Rosenhauer committed
70
    printf ("%s\n", CHECK_PASS);
71 72 73 74 75 76 77 78 79
    return 0;
}

int main (void)
{
    int status;
    struct aec_stream strm;
    struct test_state state;

80
    state.dump = 0;
81 82 83 84 85 86 87 88 89 90 91 92 93 94
    state.buf_len = state.ibuf_len = BUF_SIZE;
    state.cbuf_len = 2 * BUF_SIZE;

    state.ubuf = (unsigned char *)malloc(state.buf_len);
    state.cbuf = (unsigned char *)malloc(state.cbuf_len);
    state.obuf = (unsigned char *)malloc(state.buf_len);

    if (!state.ubuf || !state.cbuf || !state.obuf) {
        printf("Not enough memory.\n");
        return 99;
    }

    strm.flags = AEC_DATA_PREPROCESS;
    state.strm = &strm;
Mathis Rosenhauer's avatar
Mathis Rosenhauer committed
95
    strm.bits_per_sample = 32;
96 97 98 99 100 101 102 103 104 105 106 107 108
    update_state(&state);

    status = check_rsi(&state);
    if (status)
        goto DESTRUCT;

DESTRUCT:
    free(state.ubuf);
    free(state.cbuf);
    free(state.obuf);

    return status;
}