aboutsummaryrefslogtreecommitdiff
path: root/tests/unittests/torture_buffer.c
blob: a64c5a71dcd067799a7b0cd6e64e170959d5e5ca (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#define LIBSSH_STATIC

#include "torture.h"
#include "libssh/buffer.h"

ssh_buffer buffer;
#define LIMIT (8*1024*1024)

static void setup(void) {
    buffer=ssh_buffer_new();
}

static void teardown(void) {
  ssh_buffer_free(buffer);
}

/* test if the continuously growing buffer size never exceeds 2 time its
 * real capacity
 */
START_TEST (torture_growing_buffer)
{
  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){
        ck_assert(buffer_get_rest_len(buffer) * 2 >= buffer->allocated);
      }
    }
  }
}
END_TEST

/* test if the continuously growing buffer size never exceeds 2 time its
 * real capacity, when we remove 1 byte after each call (sliding window)
 */
START_TEST (torture_growing_buffer_shifting)
{
  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){
        ck_assert(buffer_get_rest_len(buffer) * 4 >= buffer->allocated);
        return;
      }
    }
  }
}
END_TEST

Suite *torture_make_suite(void) {
  Suite *s = suite_create("libssh_buffer");

  torture_create_case_fixture(s, "torture_growing_buffer",
            torture_growing_buffer, setup, teardown);
  torture_create_case_fixture(s, "torture_growing_buffer_shifting",
            torture_growing_buffer_shifting, setup, teardown);

  return s;
}