96 строки
2.7 KiB
C
96 строки
2.7 KiB
C
#define LIBSSH_STATIC
|
|
|
|
#include "torture.h"
|
|
#define DEBUG_BUFFER
|
|
#include "buffer.c"
|
|
|
|
#define LIMIT (8*1024*1024)
|
|
|
|
static void setup(void **state) {
|
|
ssh_buffer buffer;
|
|
buffer = ssh_buffer_new();
|
|
*state = (void *) buffer;
|
|
}
|
|
|
|
static void teardown(void **state) {
|
|
ssh_buffer_free(*state);
|
|
}
|
|
|
|
/*
|
|
* Test if the continuously growing buffer size never exceeds 2 time its
|
|
* real capacity
|
|
*/
|
|
static void torture_growing_buffer(void **state) {
|
|
ssh_buffer buffer = *state;
|
|
int i;
|
|
|
|
for(i=0;i<LIMIT;++i){
|
|
buffer_add_data(buffer,"A",1);
|
|
if(buffer->used >= 128){
|
|
if(buffer_get_rest_len(buffer) * 2 < buffer->allocated){
|
|
assert_true(buffer_get_rest_len(buffer) * 2 >= buffer->allocated);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Test if the continuously growing buffer size never exceeds 2 time its
|
|
* real capacity, when we remove 1 byte after each call (sliding window)
|
|
*/
|
|
static void torture_growing_buffer_shifting(void **state) {
|
|
ssh_buffer buffer = *state;
|
|
int i;
|
|
unsigned char c;
|
|
for(i=0; i<1024;++i){
|
|
buffer_add_data(buffer,"S",1);
|
|
}
|
|
for(i=0;i<LIMIT;++i){
|
|
buffer_get_u8(buffer,&c);
|
|
buffer_add_data(buffer,"A",1);
|
|
if(buffer->used >= 128){
|
|
if(buffer_get_rest_len(buffer) * 4 < buffer->allocated){
|
|
assert_true(buffer_get_rest_len(buffer) * 4 >= buffer->allocated);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Test the behavior of buffer_prepend_data
|
|
*/
|
|
static void torture_buffer_prepend(void **state) {
|
|
ssh_buffer buffer = *state;
|
|
uint32_t v;
|
|
buffer_add_data(buffer,"abcdef",6);
|
|
buffer_prepend_data(buffer,"xyz",3);
|
|
assert_int_equal(buffer_get_rest_len(buffer),9);
|
|
assert_int_equal(memcmp(buffer_get_rest(buffer), "xyzabcdef", 9), 0);
|
|
// Now remove 4 bytes and see if we can replace them
|
|
buffer_get_u32(buffer,&v);
|
|
assert_int_equal(buffer_get_rest_len(buffer),5);
|
|
assert_int_equal(memcmp(buffer_get_rest(buffer), "bcdef", 5), 0);
|
|
buffer_prepend_data(buffer,"aris",4);
|
|
assert_int_equal(buffer_get_rest_len(buffer),9);
|
|
assert_int_equal(memcmp(buffer_get_rest(buffer), "arisbcdef", 9), 0);
|
|
/* same thing but we add 5 bytes now */
|
|
buffer_get_u32(buffer,&v);
|
|
assert_int_equal(buffer_get_rest_len(buffer),5);
|
|
assert_int_equal(memcmp(buffer_get_rest(buffer), "bcdef", 5), 0);
|
|
buffer_prepend_data(buffer,"12345",5);
|
|
assert_int_equal(buffer_get_rest_len(buffer),10);
|
|
assert_int_equal(memcmp(buffer_get_rest(buffer), "12345bcdef", 10), 0);
|
|
|
|
}
|
|
|
|
int torture_run_tests(void) {
|
|
const UnitTest tests[] = {
|
|
unit_test_setup_teardown(torture_growing_buffer, setup, teardown),
|
|
unit_test_setup_teardown(torture_growing_buffer_shifting, setup, teardown),
|
|
unit_test_setup_teardown(torture_buffer_prepend, setup, teardown),
|
|
};
|
|
|
|
return run_tests(tests);
|
|
}
|