575 lines
22 KiB
C++
575 lines
22 KiB
C++
/*
|
|
* Copyright (c) 2016 MariaDB Corporation Ab
|
|
*
|
|
* Use of this software is governed by the Business Source License included
|
|
* in the LICENSE.TXT file and at www.mariadb.com/bsl11.
|
|
*
|
|
* Change Date: 2023-01-01
|
|
*
|
|
* On the date above, in accordance with the Business Source License, use
|
|
* of this software will be governed by version 2 or later of the General
|
|
* Public License.
|
|
*/
|
|
|
|
// To ensure that ss_info_assert asserts also when building in non-debug mode.
|
|
#if !defined (SS_DEBUG)
|
|
#define SS_DEBUG
|
|
#endif
|
|
#if defined (NDEBUG)
|
|
#undef NDEBUG
|
|
#endif
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include <maxbase/assert.h>
|
|
#include <maxbase/log.hh>
|
|
#include <maxbase/alloc.h>
|
|
#include <maxscale/buffer.hh>
|
|
#include <maxscale/hint.h>
|
|
|
|
/*< Return the byte at offset byte from the start of the unconsumed portion of the buffer */
|
|
#define GWBUF_DATA_CHAR(b, byte) (GWBUF_LENGTH(b) < ((byte) + 1) ? -1 : *(((char*)(b)->start) + 4))
|
|
|
|
/*< Check that the data in a buffer has the SQL marker*/
|
|
#define GWBUF_IS_SQL(b) (0x03 == GWBUF_DATA_CHAR(b, 4))
|
|
|
|
/**
|
|
* Generate predefined test data
|
|
*
|
|
* @param count Number of bytes to generate
|
|
* @return Pointer to @c count bytes of data
|
|
*/
|
|
uint8_t* generate_data(size_t count)
|
|
{
|
|
uint8_t* data = (uint8_t*)MXS_MALLOC(count);
|
|
MXS_ABORT_IF_NULL(data);
|
|
|
|
srand(0);
|
|
|
|
for (size_t i = 0; i < count; i++)
|
|
{
|
|
data[i] = rand() % 256;
|
|
}
|
|
|
|
return data;
|
|
}
|
|
|
|
size_t buffers[] =
|
|
{
|
|
2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67,
|
|
71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 131, 137, 139, 149
|
|
};
|
|
|
|
const int n_buffers = sizeof(buffers) / sizeof(size_t);
|
|
|
|
GWBUF* create_test_buffer()
|
|
{
|
|
GWBUF* head = NULL;
|
|
size_t total = 0;
|
|
|
|
for (int i = 0; i < n_buffers; i++)
|
|
{
|
|
total += buffers[i];
|
|
}
|
|
|
|
uint8_t* data = generate_data(total);
|
|
total = 0;
|
|
|
|
for (size_t i = 0; i < sizeof(buffers) / sizeof(size_t); i++)
|
|
{
|
|
head = gwbuf_append(head, gwbuf_alloc_and_load(buffers[i], data + total));
|
|
total += buffers[i];
|
|
}
|
|
|
|
MXS_FREE(data);
|
|
|
|
return head;
|
|
}
|
|
|
|
int get_length_at(int n)
|
|
{
|
|
int total = 0;
|
|
|
|
for (int i = 0; i < n_buffers && i <= n; i++)
|
|
{
|
|
total += buffers[i];
|
|
}
|
|
|
|
return total;
|
|
}
|
|
|
|
void split_buffer(int n, int offset)
|
|
{
|
|
size_t cutoff = get_length_at(n) + offset;
|
|
GWBUF* buffer = create_test_buffer();
|
|
int len = gwbuf_length(buffer);
|
|
GWBUF* newbuf = gwbuf_split(&buffer, cutoff);
|
|
|
|
mxb_assert_message(buffer && newbuf, "Both buffers should be non-NULL");
|
|
mxb_assert_message(gwbuf_length(newbuf) == cutoff, "New buffer should be have correct length");
|
|
mxb_assert_message(gwbuf_length(buffer) == len - cutoff, "Old buffer should be have correct length");
|
|
gwbuf_free(buffer);
|
|
gwbuf_free(newbuf);
|
|
}
|
|
|
|
|
|
void consume_buffer(int n, int offset)
|
|
{
|
|
size_t cutoff = get_length_at(n) + offset;
|
|
GWBUF* buffer = create_test_buffer();
|
|
int len = gwbuf_length(buffer);
|
|
buffer = gwbuf_consume(buffer, cutoff);
|
|
|
|
mxb_assert_message(buffer, "Buffer should be non-NULL");
|
|
mxb_assert_message(gwbuf_length(buffer) == len - cutoff, "Buffer should be have correct length");
|
|
gwbuf_free(buffer);
|
|
}
|
|
|
|
void copy_buffer(int n, int offset)
|
|
{
|
|
size_t cutoff = get_length_at(n) + offset;
|
|
uint8_t* data = generate_data(cutoff);
|
|
GWBUF* buffer = create_test_buffer();
|
|
int len = gwbuf_length(buffer);
|
|
uint8_t dest[cutoff];
|
|
|
|
memset(dest, 0, sizeof(dest));
|
|
mxb_assert_message(gwbuf_copy_data(buffer, 0, cutoff, dest) == cutoff, "All bytes should be read");
|
|
mxb_assert_message(memcmp(data, dest, sizeof(dest)) == 0, "Data should be OK");
|
|
gwbuf_free(buffer);
|
|
MXS_FREE(data);
|
|
}
|
|
|
|
/** gwbuf_split test - These tests assume allocation will always succeed */
|
|
void test_split()
|
|
{
|
|
size_t headsize = 10;
|
|
size_t tailsize = 20;
|
|
|
|
GWBUF* oldchain = gwbuf_append(gwbuf_alloc(headsize), gwbuf_alloc(tailsize));
|
|
mxb_assert_message(gwbuf_length(oldchain) == headsize + tailsize, "Allocated buffer should be 30 bytes");
|
|
GWBUF* newchain = gwbuf_split(&oldchain, headsize + 5);
|
|
mxb_assert_message(newchain && oldchain, "Both chains should be non-NULL");
|
|
mxb_assert_message(gwbuf_length(newchain) == headsize + 5, "New chain should be 15 bytes long");
|
|
mxb_assert_message(GWBUF_LENGTH(newchain) == headsize && GWBUF_LENGTH(newchain->next) == 5,
|
|
"The new chain should have a 10 byte buffer and a 5 byte buffer");
|
|
mxb_assert_message(gwbuf_length(oldchain) == tailsize - 5, "Old chain should be 15 bytes long");
|
|
mxb_assert_message(GWBUF_LENGTH(oldchain) == tailsize - 5 && oldchain->next == NULL,
|
|
"The old chain should have a 15 byte buffer and no next buffer");
|
|
gwbuf_free(oldchain);
|
|
gwbuf_free(newchain);
|
|
|
|
oldchain = gwbuf_append(gwbuf_alloc(headsize), gwbuf_alloc(tailsize));
|
|
newchain = gwbuf_split(&oldchain, headsize);
|
|
mxb_assert_message(gwbuf_length(newchain) == headsize, "New chain should be 10 bytes long");
|
|
mxb_assert_message(gwbuf_length(oldchain) == tailsize, "Old chain should be 20 bytes long");
|
|
mxb_assert_message(oldchain->tail == oldchain, "Old chain tail should point to old chain");
|
|
mxb_assert_message(oldchain->next == NULL, "Old chain should not have next buffer");
|
|
mxb_assert_message(newchain->tail == newchain, "Old chain tail should point to old chain");
|
|
mxb_assert_message(newchain->next == NULL, "new chain should not have next buffer");
|
|
gwbuf_free(oldchain);
|
|
gwbuf_free(newchain);
|
|
|
|
oldchain = gwbuf_append(gwbuf_alloc(headsize), gwbuf_alloc(tailsize));
|
|
newchain = gwbuf_split(&oldchain, headsize + tailsize);
|
|
mxb_assert_message(newchain, "New chain should be non-NULL");
|
|
mxb_assert_message(gwbuf_length(newchain) == headsize + tailsize, "New chain should be 30 bytes long");
|
|
mxb_assert_message(oldchain == NULL, "Old chain should be NULL");
|
|
gwbuf_free(newchain);
|
|
|
|
/** Splitting of contiguous memory */
|
|
GWBUF* buffer = gwbuf_alloc(10);
|
|
GWBUF* newbuf = gwbuf_split(&buffer, 5);
|
|
mxb_assert_message(buffer != newbuf, "gwbuf_split should return different pointers");
|
|
mxb_assert_message(gwbuf_length(buffer) == 5 && GWBUF_LENGTH(buffer) == 5,
|
|
"Old buffer should be 5 bytes");
|
|
mxb_assert_message(gwbuf_length(newbuf) == 5 && GWBUF_LENGTH(newbuf) == 5,
|
|
"New buffer should be 5 bytes");
|
|
mxb_assert_message(buffer->tail == buffer, "Old buffer's tail should point to itself");
|
|
mxb_assert_message(newbuf->tail == newbuf, "New buffer's tail should point to itself");
|
|
mxb_assert_message(buffer->next == NULL, "Old buffer's next pointer should be NULL");
|
|
mxb_assert_message(newbuf->next == NULL, "New buffer's next pointer should be NULL");
|
|
gwbuf_free(buffer);
|
|
gwbuf_free(newbuf);
|
|
|
|
/** Bad parameter tests */
|
|
buffer = gwbuf_alloc(10);
|
|
mxb_assert_message(gwbuf_split(&buffer, 0) == NULL, "gwbuf_split with length of 0 should return NULL");
|
|
mxb_assert_message(gwbuf_length(buffer) == 10, "Buffer should be 10 bytes");
|
|
gwbuf_free(buffer);
|
|
|
|
/** Splitting near buffer boudaries */
|
|
for (int i = 0; i < n_buffers - 1; i++)
|
|
{
|
|
split_buffer(i, -1);
|
|
split_buffer(i, 0);
|
|
split_buffer(i, 1);
|
|
}
|
|
|
|
/** Split near last buffer's end */
|
|
split_buffer(n_buffers - 1, -1);
|
|
}
|
|
|
|
/** gwbuf_alloc_and_load and gwbuf_copy_data tests */
|
|
void test_load_and_copy()
|
|
{
|
|
uint8_t data[] = {1, 2, 3, 4, 5, 6, 7, 8};
|
|
uint8_t dest[8];
|
|
GWBUF* head = gwbuf_alloc_and_load(4, data);
|
|
GWBUF* tail = gwbuf_alloc_and_load(4, data + 4);
|
|
|
|
mxb_assert_message(memcmp(GWBUF_DATA(head), data, 4) == 0, "Loading 4 bytes should succeed");
|
|
mxb_assert_message(memcmp(GWBUF_DATA(tail), data + 4, 4) == 0, "Loading 4 bytes should succeed");
|
|
|
|
memset(dest, 0, sizeof(dest));
|
|
mxb_assert_message(gwbuf_copy_data(head, 0, 4, dest) == 4, "Copying 4 bytes should succeed");
|
|
mxb_assert_message(memcmp(dest, data, 4) == 0, "Copied data should be from 1 to 4");
|
|
|
|
memset(dest, 0, sizeof(dest));
|
|
mxb_assert_message(gwbuf_copy_data(tail, 0, 4, dest) == 4, "Copying 4 bytes should succeed");
|
|
mxb_assert_message(memcmp(dest, data + 4, 4) == 0, "Copied data should be from 5 to 8");
|
|
head = gwbuf_append(head, tail);
|
|
|
|
memset(dest, 0, sizeof(dest));
|
|
mxb_assert_message(gwbuf_copy_data(head, 0, 8, dest) == 8, "Copying 8 bytes should succeed");
|
|
mxb_assert_message(memcmp(dest, data, 8) == 0, "Copied data should be from 1 to 8");
|
|
|
|
memset(dest, 0, sizeof(dest));
|
|
mxb_assert_message(gwbuf_copy_data(head, 4, 4, dest) == 4, "Copying 4 bytes at offset 4 should succeed");
|
|
mxb_assert_message(memcmp(dest, data + 4, 4) == 0, "Copied data should be from 5 to 8");
|
|
|
|
memset(dest, 0, sizeof(dest));
|
|
mxb_assert_message(gwbuf_copy_data(head, 2, 4, dest) == 4, "Copying 4 bytes at offset 2 should succeed");
|
|
mxb_assert_message(memcmp(dest, data + 2, 4) == 0, "Copied data should be from 5 to 8");
|
|
|
|
memset(dest, 0, sizeof(dest));
|
|
mxb_assert_message(gwbuf_copy_data(head, 0, 10, dest) == 8, "Copying 10 bytes should only copy 8 bytes");
|
|
mxb_assert_message(memcmp(dest, data, 8) == 0, "Copied data should be from 1 to 8");
|
|
|
|
memset(dest, 0, sizeof(dest));
|
|
mxb_assert_message(gwbuf_copy_data(head, 0, 0, dest) == 0, "Copying 0 bytes should not copy any bytes");
|
|
|
|
memset(dest, 0, sizeof(dest));
|
|
mxb_assert_message(gwbuf_copy_data(head, 0, -1, dest) == sizeof(data),
|
|
"Copying -1 bytes should copy all available data (cast to unsigned)");
|
|
mxb_assert_message(memcmp(dest, data, 8) == 0, "Copied data should be from 1 to 8");
|
|
|
|
mxb_assert_message(gwbuf_copy_data(head, -1, -1, dest) == 0,
|
|
"Copying -1 bytes at an offset of -1 should not copy any bytes");
|
|
mxb_assert_message(gwbuf_copy_data(head, -1, 0, dest) == 0,
|
|
"Copying 0 bytes at an offset of -1 should not copy any bytes");
|
|
gwbuf_free(head);
|
|
|
|
/** Copying near buffer boudaries */
|
|
for (int i = 0; i < n_buffers - 1; i++)
|
|
{
|
|
copy_buffer(i, -1);
|
|
copy_buffer(i, 0);
|
|
copy_buffer(i, 1);
|
|
}
|
|
|
|
/** Copy near last buffer's end */
|
|
copy_buffer(n_buffers - 1, -1);
|
|
}
|
|
|
|
void test_consume()
|
|
{
|
|
uint8_t data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
|
|
GWBUF* buffer = gwbuf_append(gwbuf_alloc_and_load(5, data),
|
|
gwbuf_alloc_and_load(5, data + 5));
|
|
|
|
mxb_assert_message(gwbuf_consume(buffer, 0) == buffer,
|
|
"Consuming 0 bytes from a buffer should return original buffer");
|
|
mxb_assert_message(gwbuf_length(buffer) == 10, "Buffer should be 10 bytes after consuming 0 bytes");
|
|
|
|
buffer = gwbuf_consume(buffer, 1);
|
|
mxb_assert_message(GWBUF_LENGTH(buffer) == 4, "First buffer should be 4 bytes long");
|
|
mxb_assert_message(buffer->next, "Buffer should have next pointer set");
|
|
mxb_assert_message(GWBUF_LENGTH(buffer->next) == 5, "Next buffer should be 5 bytes long");
|
|
mxb_assert_message(gwbuf_length(buffer) == 9, "Buffer should be 9 bytes after consuming 1 bytes");
|
|
mxb_assert_message(*((uint8_t*)buffer->start) == 2, "First byte should be 2");
|
|
|
|
buffer = gwbuf_consume(buffer, 5);
|
|
mxb_assert_message(buffer->next == NULL, "Buffer should not have the next pointer set");
|
|
mxb_assert_message(GWBUF_LENGTH(buffer) == 4, "Buffer should be 4 bytes after consuming 6 bytes");
|
|
mxb_assert_message(gwbuf_length(buffer) == 4, "Buffer should be 4 bytes after consuming 6 bytes");
|
|
mxb_assert_message(*((uint8_t*)buffer->start) == 7, "First byte should be 7");
|
|
mxb_assert_message(gwbuf_consume(buffer, 4) == NULL, "Consuming all bytes should return NULL");
|
|
|
|
buffer = gwbuf_append(gwbuf_alloc_and_load(5, data),
|
|
gwbuf_alloc_and_load(5, data + 5));
|
|
mxb_assert_message(gwbuf_consume(buffer, 100) == NULL,
|
|
"Consuming more bytes than are available should return NULL");
|
|
|
|
|
|
/** Consuming near buffer boudaries */
|
|
for (int i = 0; i < n_buffers - 1; i++)
|
|
{
|
|
consume_buffer(i, -1);
|
|
consume_buffer(i, 0);
|
|
consume_buffer(i, 1);
|
|
}
|
|
|
|
/** Consume near last buffer's end */
|
|
consume_buffer(n_buffers - 1, -1);
|
|
}
|
|
|
|
void test_compare()
|
|
{
|
|
static const uint8_t data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
|
|
|
|
fprintf(stderr, "testbuffer : testing GWBUF comparisons\n");
|
|
|
|
GWBUF* lhs = gwbuf_alloc_and_load(10, data);
|
|
|
|
// The same array
|
|
mxb_assert(gwbuf_compare(lhs, lhs) == 0);
|
|
|
|
// Identical array
|
|
GWBUF* rhs = gwbuf_alloc_and_load(10, data);
|
|
mxb_assert(gwbuf_compare(lhs, rhs) == 0);
|
|
|
|
// One shorter
|
|
gwbuf_free(rhs);
|
|
rhs = gwbuf_alloc_and_load(9, data + 1);
|
|
mxb_assert(gwbuf_compare(lhs, rhs) > 0);
|
|
mxb_assert(gwbuf_compare(rhs, lhs) < 0);
|
|
|
|
// One segmented, but otherwise identical.
|
|
gwbuf_free(rhs);
|
|
rhs = NULL;
|
|
rhs = gwbuf_append(rhs, gwbuf_alloc_and_load(3, data));
|
|
rhs = gwbuf_append(rhs, gwbuf_alloc_and_load(3, data + 3));
|
|
rhs = gwbuf_append(rhs, gwbuf_alloc_and_load(4, data + 3 + 3));
|
|
|
|
mxb_assert(gwbuf_compare(lhs, rhs) == 0);
|
|
mxb_assert(gwbuf_compare(rhs, rhs) == 0);
|
|
|
|
// Both segmented, but otherwise identical.
|
|
gwbuf_free(lhs);
|
|
lhs = NULL;
|
|
lhs = gwbuf_append(lhs, gwbuf_alloc_and_load(5, data));
|
|
lhs = gwbuf_append(lhs, gwbuf_alloc_and_load(5, data + 5));
|
|
|
|
mxb_assert(gwbuf_compare(lhs, rhs) == 0);
|
|
mxb_assert(gwbuf_compare(rhs, lhs) == 0);
|
|
|
|
// Both segmented and of same length, but different.
|
|
gwbuf_free(lhs);
|
|
lhs = NULL;
|
|
lhs = gwbuf_append(lhs, gwbuf_alloc_and_load(5, data + 5)); // Values in different order
|
|
lhs = gwbuf_append(lhs, gwbuf_alloc_and_load(5, data));
|
|
|
|
mxb_assert(gwbuf_compare(lhs, rhs) > 0); // 5 > 1
|
|
mxb_assert(gwbuf_compare(rhs, lhs) < 0); // 5 > 1
|
|
|
|
// Identical, but one containing empty segments.
|
|
gwbuf_free(rhs);
|
|
rhs = NULL;
|
|
rhs = gwbuf_append(rhs, gwbuf_alloc_and_load(0, data));
|
|
rhs = gwbuf_append(rhs, gwbuf_alloc_and_load(5, data + 5));
|
|
rhs = gwbuf_append(rhs, gwbuf_alloc_and_load(0, data));
|
|
rhs = gwbuf_append(rhs, gwbuf_alloc_and_load(5, data));
|
|
rhs = gwbuf_append(rhs, gwbuf_alloc_and_load(0, data));
|
|
|
|
mxb_assert(gwbuf_compare(lhs, rhs) == 0);
|
|
mxb_assert(gwbuf_compare(rhs, lhs) == 0);
|
|
|
|
gwbuf_free(lhs);
|
|
gwbuf_free(rhs);
|
|
}
|
|
|
|
|
|
|
|
void test_clone()
|
|
{
|
|
GWBUF* original = gwbuf_alloc_and_load(1, "1");
|
|
|
|
original = gwbuf_append(original, gwbuf_alloc_and_load(1, "1"));
|
|
original = gwbuf_append(original, gwbuf_alloc_and_load(2, "12"));
|
|
original = gwbuf_append(original, gwbuf_alloc_and_load(3, "123"));
|
|
original = gwbuf_append(original, gwbuf_alloc_and_load(5, "12345"));
|
|
original = gwbuf_append(original, gwbuf_alloc_and_load(8, "12345678"));
|
|
original = gwbuf_append(original, gwbuf_alloc_and_load(13, "1234567890123"));
|
|
original = gwbuf_append(original, gwbuf_alloc_and_load(21, "123456789012345678901"));
|
|
|
|
GWBUF* clone = gwbuf_clone(original);
|
|
|
|
GWBUF* o = original;
|
|
GWBUF* c = clone;
|
|
|
|
mxb_assert(gwbuf_length(o) == gwbuf_length(c));
|
|
|
|
while (o)
|
|
{
|
|
mxb_assert(c);
|
|
mxb_assert(GWBUF_LENGTH(o) == GWBUF_LENGTH(c));
|
|
|
|
const char* i = (char*)GWBUF_DATA(o);
|
|
const char* end = i + GWBUF_LENGTH(o);
|
|
const char* j = (char*)GWBUF_DATA(c);
|
|
|
|
while (i != end)
|
|
{
|
|
mxb_assert(*i == *j);
|
|
++i;
|
|
++j;
|
|
}
|
|
|
|
o = o->next;
|
|
c = c->next;
|
|
}
|
|
|
|
mxb_assert(c == NULL);
|
|
|
|
gwbuf_free(clone);
|
|
gwbuf_free(original);
|
|
|
|
original = nullptr;
|
|
original = gwbuf_append(original, gwbuf_alloc_and_load(1, "1"));
|
|
original = gwbuf_append(original, gwbuf_alloc_and_load(2, "12"));
|
|
|
|
clone = gwbuf_clone(original);
|
|
clone = gwbuf_append(clone, gwbuf_alloc_and_load(3, "123"));
|
|
|
|
mxb_assert(gwbuf_length(clone) == 1 + 2 + 3);
|
|
|
|
gwbuf_free(clone);
|
|
gwbuf_free(original);
|
|
}
|
|
|
|
/**
|
|
* test1 Allocate a buffer and do lots of things
|
|
*
|
|
*/
|
|
static int test1()
|
|
{
|
|
GWBUF* buffer, * extra, * clone, * partclone;
|
|
HINT* hint;
|
|
size_t size = 100;
|
|
size_t bite1 = 35;
|
|
size_t bite2 = 60;
|
|
size_t bite3 = 10;
|
|
size_t buflen;
|
|
|
|
/* Single buffer tests */
|
|
fprintf(stderr,
|
|
"testbuffer : creating buffer with data size %lu bytes",
|
|
size);
|
|
buffer = gwbuf_alloc(size);
|
|
fprintf(stderr, "\t..done\nAllocated buffer of size %lu.", size);
|
|
buflen = GWBUF_LENGTH(buffer);
|
|
fprintf(stderr, "\nBuffer length is now %lu", buflen);
|
|
mxb_assert_message(size == buflen, "Incorrect buffer size");
|
|
mxb_assert_message(0 == GWBUF_EMPTY(buffer), "Buffer should not be empty");
|
|
mxb_assert_message(GWBUF_IS_TYPE_UNDEFINED(buffer), "Buffer type should be undefined");
|
|
fprintf(stderr, "\t..done\nSet a property for the buffer");
|
|
gwbuf_add_property(buffer, (char*)"name", (char*)"value");
|
|
mxb_assert_message(0 == strcmp("value", gwbuf_get_property(buffer, (char*)"name")),
|
|
"Should now have correct property");
|
|
strcpy((char*)GWBUF_DATA(buffer), "The quick brown fox jumps over the lazy dog");
|
|
fprintf(stderr, "\t..done\nLoad some data into the buffer");
|
|
mxb_assert_message('q' == GWBUF_DATA_CHAR(buffer, 4), "Fourth character of buffer must be 'q'");
|
|
mxb_assert_message(-1 == GWBUF_DATA_CHAR(buffer, 105),
|
|
"Hundred and fifth character of buffer must return -1");
|
|
mxb_assert_message(0 == GWBUF_IS_SQL(buffer), "Must say buffer is not SQL, as it does not have marker");
|
|
strcpy((char*)GWBUF_DATA(buffer), "1234\x03SELECT * FROM sometable");
|
|
fprintf(stderr, "\t..done\nLoad SQL data into the buffer");
|
|
mxb_assert_message(1 == GWBUF_IS_SQL(buffer), "Must say buffer is SQL, as it does have marker");
|
|
clone = gwbuf_clone(buffer);
|
|
fprintf(stderr, "\t..done\nCloned buffer");
|
|
buflen = GWBUF_LENGTH(clone);
|
|
fprintf(stderr, "\nCloned buffer length is now %lu", buflen);
|
|
mxb_assert_message(size == buflen, "Incorrect buffer size");
|
|
mxb_assert_message(0 == GWBUF_EMPTY(clone), "Cloned buffer should not be empty");
|
|
fprintf(stderr, "\t..done\n");
|
|
gwbuf_free(clone);
|
|
fprintf(stderr, "Freed cloned buffer");
|
|
fprintf(stderr, "\t..done\n");
|
|
buffer = gwbuf_consume(buffer, bite1);
|
|
mxb_assert_message(NULL != buffer, "Buffer should not be null");
|
|
buflen = GWBUF_LENGTH(buffer);
|
|
fprintf(stderr, "Consumed %lu bytes, now have %lu, should have %lu", bite1, buflen, size - bite1);
|
|
mxb_assert_message((size - bite1) == buflen, "Incorrect buffer size");
|
|
mxb_assert_message(0 == GWBUF_EMPTY(buffer), "Buffer should not be empty");
|
|
fprintf(stderr, "\t..done\n");
|
|
buffer = gwbuf_consume(buffer, bite2);
|
|
mxb_assert_message(NULL != buffer, "Buffer should not be null");
|
|
buflen = GWBUF_LENGTH(buffer);
|
|
fprintf(stderr, "Consumed %lu bytes, now have %lu, should have %lu", bite2, buflen, size - bite1 - bite2);
|
|
mxb_assert_message((size - bite1 - bite2) == buflen, "Incorrect buffer size");
|
|
mxb_assert_message(0 == GWBUF_EMPTY(buffer), "Buffer should not be empty");
|
|
fprintf(stderr, "\t..done\n");
|
|
buffer = gwbuf_consume(buffer, bite3);
|
|
fprintf(stderr, "Consumed %lu bytes, should have null buffer", bite3);
|
|
mxb_assert_message(NULL == buffer, "Buffer should be null");
|
|
|
|
/* Buffer list tests */
|
|
size = 100000;
|
|
buffer = gwbuf_alloc(size);
|
|
fprintf(stderr, "\t..done\nAllocated buffer of size %lu.", size);
|
|
buflen = GWBUF_LENGTH(buffer);
|
|
fprintf(stderr, "\nBuffer length is now %lu", buflen);
|
|
mxb_assert_message(size == buflen, "Incorrect buffer size");
|
|
mxb_assert_message(0 == GWBUF_EMPTY(buffer), "Buffer should not be empty");
|
|
mxb_assert_message(GWBUF_IS_TYPE_UNDEFINED(buffer), "Buffer type should be undefined");
|
|
extra = gwbuf_alloc(size);
|
|
buflen = GWBUF_LENGTH(buffer);
|
|
fprintf(stderr, "\t..done\nAllocated extra buffer of size %lu.", size);
|
|
mxb_assert_message(size == buflen, "Incorrect buffer size");
|
|
buffer = gwbuf_append(buffer, extra);
|
|
buflen = gwbuf_length(buffer);
|
|
fprintf(stderr, "\t..done\nAppended extra buffer to original buffer to create list of size %lu", buflen);
|
|
mxb_assert_message((size * 2) == gwbuf_length(buffer), "Incorrect size for set of buffers");
|
|
buffer = gwbuf_rtrim(buffer, 60000);
|
|
buflen = GWBUF_LENGTH(buffer);
|
|
fprintf(stderr, "\t..done\nTrimmed 60 bytes from buffer, now size is %lu.", buflen);
|
|
mxb_assert_message((size - 60000) == buflen, "Incorrect buffer size");
|
|
buffer = gwbuf_rtrim(buffer, 60000);
|
|
buflen = GWBUF_LENGTH(buffer);
|
|
fprintf(stderr, "\t..done\nTrimmed another 60 bytes from buffer, now size is %lu.", buflen);
|
|
mxb_assert_message(100000 == buflen, "Incorrect buffer size");
|
|
mxb_assert_message(buffer == extra, "The buffer pointer should now point to the extra buffer");
|
|
fprintf(stderr, "\t..done\n");
|
|
gwbuf_free(buffer);
|
|
/** gwbuf_clone_all test */
|
|
size_t headsize = 10;
|
|
GWBUF* head = gwbuf_alloc(headsize);
|
|
size_t tailsize = 20;
|
|
GWBUF* tail = gwbuf_alloc(tailsize);
|
|
|
|
mxb_assert_message(head && tail, "Head and tail buffers should both be non-NULL");
|
|
GWBUF* append = gwbuf_append(head, tail);
|
|
mxb_assert_message(append == head, "gwbuf_append should return head");
|
|
mxb_assert_message(append->next == tail, "After append tail should be in the next pointer of head");
|
|
mxb_assert_message(append->tail == tail, "After append tail should be in the tail pointer of head");
|
|
GWBUF* all_clones = gwbuf_clone(head);
|
|
mxb_assert_message(all_clones && all_clones->next, "Cloning all should work");
|
|
mxb_assert_message(GWBUF_LENGTH(all_clones) == headsize, "First buffer should be 10 bytes");
|
|
mxb_assert_message(GWBUF_LENGTH(all_clones->next) == tailsize, "Second buffer should be 20 bytes");
|
|
mxb_assert_message(gwbuf_length(all_clones) == headsize + tailsize,
|
|
"Total buffer length should be 30 bytes");
|
|
gwbuf_free(all_clones);
|
|
gwbuf_free(head);
|
|
|
|
test_split();
|
|
test_load_and_copy();
|
|
test_consume();
|
|
test_compare();
|
|
test_clone();
|
|
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
mxb::Log log;
|
|
|
|
int result = 0;
|
|
|
|
result += test1();
|
|
|
|
return result;
|
|
}
|