Added tests for malloc failures

This commit is contained in:
Tatsuhiro Tsujikawa 2012-04-24 22:51:06 +09:00
parent f21c0ee3d9
commit c91a4ec091
7 changed files with 843 additions and 2 deletions

1
.gitignore vendored
View File

@ -27,3 +27,4 @@ stamp-h1
INSTALL INSTALL
.DS_STORE .DS_STORE
tests/main tests/main
tests/failmalloc

View File

@ -23,7 +23,7 @@
if HAVE_CUNIT if HAVE_CUNIT
check_PROGRAMS = main check_PROGRAMS = main failmalloc
OBJECTS = main.c spdylay_pq_test.c spdylay_map_test.c spdylay_queue_test.c \ OBJECTS = main.c spdylay_pq_test.c spdylay_map_test.c spdylay_queue_test.c \
spdylay_buffer_test.c spdylay_zlib_test.c spdylay_session_test.c \ spdylay_buffer_test.c spdylay_zlib_test.c spdylay_session_test.c \
@ -39,10 +39,16 @@ main_SOURCES = $(HFILES) $(OBJECTS)
main_LDADD = ${top_builddir}/lib/libspdylay.la main_LDADD = ${top_builddir}/lib/libspdylay.la
main_LDFLAGS = -static @CUNIT_LIBS@ main_LDFLAGS = -static @CUNIT_LIBS@
failmalloc_SOURCES = failmalloc.c failmalloc_test.c failmalloc_test.h \
malloc_wrapper.c malloc_wrapper.h
failmalloc_LDADD = $(main_LDADD) -ldl
failmalloc_LDFLAGS = $(main_LDFLAGS)
AM_CFLAGS = -Wall -g -O2 -I${top_srcdir}/lib -I${top_srcdir}/lib/includes -I${top_builddir}/lib/includes \ AM_CFLAGS = -Wall -g -O2 -I${top_srcdir}/lib -I${top_srcdir}/lib/includes -I${top_builddir}/lib/includes \
@CUNIT_CFLAGS@ @DEFS@ @CUNIT_CFLAGS@ @DEFS@
TESTS = main TESTS = main failmalloc
if ENABLE_EXAMPLES if ENABLE_EXAMPLES

71
tests/failmalloc.c Normal file
View File

@ -0,0 +1,71 @@
/*
* Spdylay - SPDY Library
*
* Copyright (c) 2012 Tatsuhiro Tsujikawa
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#include <stdio.h>
#include <string.h>
#include <CUnit/Basic.h>
/* include test cases' include files here */
#include "failmalloc_test.h"
static int init_suite1(void)
{
return 0;
}
static int clean_suite1(void)
{
return 0;
}
int main(int argc, char* argv[])
{
CU_pSuite pSuite = NULL;
/* initialize the CUnit test registry */
if (CUE_SUCCESS != CU_initialize_registry())
return CU_get_error();
/* add a suite to the registry */
pSuite = CU_add_suite("libspdylay_TestSuite", init_suite1, clean_suite1);
if (NULL == pSuite) {
CU_cleanup_registry();
return CU_get_error();
}
/* add the tests to the suite */
if(!CU_add_test(pSuite, "failmalloc_session_send",
test_spdylay_session_send) ||
!CU_add_test(pSuite, "failmalloc_session_recv",
test_spdylay_session_recv) ||
!CU_add_test(pSuite, "failmalloc_frame", test_spdylay_frame)) {
CU_cleanup_registry();
return CU_get_error();
}
/* Run all tests using the CUnit Basic interface */
CU_basic_set_mode(CU_BRM_VERBOSE);
CU_basic_run_tests();
CU_cleanup_registry();
return CU_get_error();
}

604
tests/failmalloc_test.c Normal file
View File

@ -0,0 +1,604 @@
/*
* Spdylay - SPDY Library
*
* Copyright (c) 2012 Tatsuhiro Tsujikawa
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#include "failmalloc_test.h"
#include <CUnit/CUnit.h>
#include <stdio.h>
#include <assert.h>
#include "spdylay_session.h"
#include "spdylay_stream.h"
#include "spdylay_frame.h"
#include "spdylay_helper.h"
#include "malloc_wrapper.h"
static char* strcopy(const char* s)
{
size_t len = strlen(s);
char *dest = malloc(len+1);
if(dest == NULL) {
return NULL;
}
memcpy(dest, s, len);
dest[len] = '\0';
return dest;
}
typedef struct {
uint8_t data[8192];
uint8_t *datamark, *datalimit;
} data_feed;
typedef struct {
data_feed *df;
size_t data_source_length;
} my_user_data;
static void data_feed_init(data_feed *df, uint8_t *data, size_t data_length)
{
assert(data_length <= sizeof(df->data));
memcpy(df->data, data, data_length);
df->datamark = df->data;
df->datalimit = df->data+data_length;
}
static ssize_t null_send_callback(spdylay_session *session,
const uint8_t* data, size_t len, int flags,
void *user_data)
{
return len;
}
static ssize_t data_feed_recv_callback(spdylay_session *session,
uint8_t* data, size_t len, int flags,
void *user_data)
{
data_feed *df = ((my_user_data*)user_data)->df;
size_t avail = df->datalimit - df->datamark;
size_t wlen = spdylay_min(avail, len);
memcpy(data, df->datamark, wlen);
df->datamark += wlen;
return wlen;
}
static ssize_t fixed_length_data_source_read_callback
(spdylay_session *session, int32_t stream_id,
uint8_t *buf, size_t len, int *eof,
spdylay_data_source *source, void *user_data)
{
my_user_data *ud = (my_user_data*)user_data;
size_t wlen;
if(len < ud->data_source_length) {
wlen = len;
} else {
wlen = ud->data_source_length;
}
ud->data_source_length -= wlen;
if(ud->data_source_length == 0) {
*eof = 1;
}
return wlen;
}
static ssize_t get_credential_ncerts(spdylay_session *session,
const spdylay_origin *origin,
void *user_data)
{
if(strcmp("example.org", origin->host) == 0 &&
strcmp("https", origin->scheme) == 0 &&
443 == origin->port) {
return 2;
} else {
return 0;
}
}
static ssize_t get_credential_cert(spdylay_session *session,
const spdylay_origin *origin,
size_t index,
uint8_t *cert, size_t certlen,
void *user_data)
{
size_t len = strlen(origin->host);
if(certlen == 0) {
return len;
} else {
assert(certlen == len);
memcpy(cert, origin->host, len);
return 0;
}
}
static ssize_t get_credential_proof(spdylay_session *session,
const spdylay_origin *origin,
uint8_t *proof, size_t prooflen,
void *uer_data)
{
size_t len = strlen(origin->scheme);
if(prooflen == 0) {
return len;
} else {
assert(prooflen == len);
memcpy(proof, origin->scheme, len);
return 0;
}
}
#define TEST_FAILMALLOC_RUN(FUN) \
size_t nmalloc, i; \
\
spdylay_nmalloc = 0; \
FUN(); \
nmalloc = spdylay_nmalloc; \
\
spdylay_failmalloc = 1; \
for(i = 0; i < nmalloc; ++i) { \
spdylay_nmalloc = 0; \
spdylay_failstart = i; \
/* printf("i=%zu\n", i); */ \
FUN(); \
/* printf("nmalloc=%d\n", spdylay_nmalloc); */ \
}
static void run_spdylay_session_send(void)
{
spdylay_session *session;
spdylay_session_callbacks callbacks;
const char *nv[] = { ":host", "example.org",
":scheme", "https",
NULL };
spdylay_data_provider data_prd;
spdylay_settings_entry iv[2];
my_user_data ud;
int rv;
memset(&callbacks, 0, sizeof(spdylay_session_callbacks));
callbacks.send_callback = null_send_callback;
callbacks.get_credential_ncerts = get_credential_ncerts;
callbacks.get_credential_cert = get_credential_cert;
callbacks.get_credential_proof = get_credential_proof;
data_prd.read_callback = fixed_length_data_source_read_callback;
ud.data_source_length = 64*1024;
iv[0].settings_id = SPDYLAY_SETTINGS_UPLOAD_BANDWIDTH;
iv[0].flags = SPDYLAY_ID_FLAG_SETTINGS_PERSIST_VALUE;
iv[0].value = 256;
iv[1].settings_id = SPDYLAY_SETTINGS_MAX_CONCURRENT_STREAMS;
iv[1].flags = SPDYLAY_ID_FLAG_SETTINGS_NONE;
iv[1].value = 100;
rv = spdylay_session_client_new(&session, SPDYLAY_PROTO_SPDY3,
&callbacks, &ud);
if(rv != 0) {
goto client_new_fail;
}
rv = spdylay_submit_request(session, 3, nv, &data_prd, NULL);
if(rv != 0) {
goto fail;
}
rv = spdylay_submit_syn_stream(session, SPDYLAY_CTRL_FLAG_NONE,
0, 3, nv, NULL);
if(rv != 0) {
goto fail;
}
rv = spdylay_session_send(session);
if(rv != 0) {
goto fail;
}
/* The SYN_STREAM submitted by the previous
spdylay_submit_syn_stream will have stream ID 3. Send HEADERS to
that stream. */
rv = spdylay_submit_headers(session, SPDYLAY_CTRL_FLAG_NONE, 3, nv);
if(rv != 0) {
goto fail;
}
rv = spdylay_submit_data(session, 3, SPDYLAY_DATA_FLAG_FIN, &data_prd);
if(rv != 0) {
goto fail;
}
rv = spdylay_session_send(session);
if(rv != 0) {
goto fail;
}
rv = spdylay_submit_rst_stream(session, 3, SPDYLAY_CANCEL);
if(rv != 0) {
goto fail;
}
rv = spdylay_session_send(session);
if(rv != 0) {
goto fail;
}
/* Sending against half-closed stream */
rv = spdylay_submit_headers(session, SPDYLAY_CTRL_FLAG_NONE, 3, nv);
if(rv != 0) {
goto fail;
}
rv = spdylay_submit_data(session, 3, SPDYLAY_DATA_FLAG_FIN, &data_prd);
if(rv != 0) {
goto fail;
}
rv = spdylay_submit_ping(session);
if(rv != 0) {
goto fail;
}
rv = spdylay_submit_settings(session, SPDYLAY_FLAG_SETTINGS_NONE, iv, 2);
if(rv != 0) {
goto fail;
}
rv = spdylay_session_send(session);
if(rv != 0) {
goto fail;
}
rv = spdylay_submit_goaway(session, SPDYLAY_GOAWAY_OK);
if(rv != 0) {
goto fail;
}
rv = spdylay_session_send(session);
if(rv != 0) {
goto fail;
}
fail:
spdylay_session_del(session);
client_new_fail:
;
}
void test_spdylay_session_send(void)
{
TEST_FAILMALLOC_RUN(run_spdylay_session_send);
}
static void run_spdylay_session_recv(void)
{
spdylay_session *session;
spdylay_session_callbacks callbacks;
spdylay_zlib deflater;
spdylay_frame frame;
uint8_t *buf = NULL, *nvbuf = NULL;
size_t buflen = 0, nvbuflen = 0;
ssize_t framelen;
const char *nv[] = { ":host", "example.org",
":scheme", "https",
NULL };
spdylay_settings_entry iv[2];
spdylay_mem_chunk proof;
spdylay_mem_chunk *certs;
size_t ncerts;
my_user_data ud;
data_feed df;
int rv;
memset(&callbacks, 0, sizeof(spdylay_session_callbacks));
callbacks.recv_callback = data_feed_recv_callback;
ud.df = &df;
spdylay_failmalloc_pause();
spdylay_zlib_deflate_hd_init(&deflater, SPDYLAY_PROTO_SPDY3);
spdylay_session_server_new(&session, SPDYLAY_PROTO_SPDY3, &callbacks, &ud);
spdylay_failmalloc_unpause();
/* SYN_STREAM */
spdylay_failmalloc_pause();
spdylay_frame_syn_stream_init(&frame.syn_stream, SPDYLAY_PROTO_SPDY3,
SPDYLAY_CTRL_FLAG_FIN, 1, 0, 2,
spdylay_frame_nv_copy(nv));
framelen = spdylay_frame_pack_syn_stream(&buf, &buflen,
&nvbuf, &nvbuflen,
&frame.syn_stream, &deflater);
spdylay_frame_syn_stream_free(&frame.syn_stream);
data_feed_init(&df, buf, framelen);
spdylay_failmalloc_unpause();
rv = spdylay_session_recv(session);
if(rv != 0) {
goto fail;
}
/* PING */
spdylay_failmalloc_pause();
spdylay_frame_ping_init(&frame.ping, SPDYLAY_PROTO_SPDY3, 1);
framelen = spdylay_frame_pack_ping(&buf, &buflen, &frame.ping);
spdylay_frame_ping_free(&frame.ping);
data_feed_init(&df, buf, framelen);
spdylay_failmalloc_unpause();
rv = spdylay_session_recv(session);
if(rv != 0) {
goto fail;
}
/* RST_STREAM */
spdylay_failmalloc_pause();
spdylay_frame_rst_stream_init(&frame.rst_stream, SPDYLAY_PROTO_SPDY3, 1,
SPDYLAY_PROTOCOL_ERROR);
framelen = spdylay_frame_pack_rst_stream(&buf, &buflen, &frame.rst_stream);
spdylay_frame_rst_stream_free(&frame.rst_stream);
spdylay_failmalloc_unpause();
rv = spdylay_session_recv(session);
if(rv != 0) {
goto fail;
}
/* SETTINGS */
spdylay_failmalloc_pause();
iv[0].settings_id = SPDYLAY_SETTINGS_UPLOAD_BANDWIDTH;
iv[0].flags = SPDYLAY_ID_FLAG_SETTINGS_PERSIST_VALUE;
iv[0].value = 256;
iv[1].settings_id = SPDYLAY_SETTINGS_MAX_CONCURRENT_STREAMS;
iv[1].flags = SPDYLAY_ID_FLAG_SETTINGS_NONE;
iv[1].value = 100;
spdylay_frame_settings_init(&frame.settings, SPDYLAY_PROTO_SPDY3,
SPDYLAY_FLAG_SETTINGS_CLEAR_SETTINGS,
spdylay_frame_iv_copy(iv, 2), 2);
framelen = spdylay_frame_pack_settings(&buf, &buflen, &frame.settings);
spdylay_frame_settings_free(&frame.settings);
spdylay_failmalloc_unpause();
rv = spdylay_session_recv(session);
if(rv != 0) {
goto fail;
}
/* CREDENTIAL */
spdylay_failmalloc_pause();
proof.data = (uint8_t*)strcopy("PROOF");
proof.length = strlen("PROOF");
ncerts = 2;
certs = malloc(sizeof(spdylay_mem_chunk)*ncerts);
certs[0].data = (uint8_t*)strcopy("CERT0");
certs[0].length = strlen("CERT0");
certs[1].data = (uint8_t*)strcopy("CERT1");
certs[1].length = strlen("CERT1");
spdylay_frame_credential_init(&frame.credential, SPDYLAY_PROTO_SPDY3,
1, &proof, certs, ncerts);
framelen = spdylay_frame_pack_credential(&buf, &buflen, &frame.credential);
spdylay_frame_credential_free(&frame.credential);
spdylay_failmalloc_unpause();
rv = spdylay_session_recv(session);
if(rv != 0) {
goto fail;
}
fail:
free(buf);
free(nvbuf);
spdylay_session_del(session);
spdylay_zlib_deflate_free(&deflater);
}
void test_spdylay_session_recv(void)
{
TEST_FAILMALLOC_RUN(run_spdylay_session_recv);
}
static void run_spdylay_frame_pack_syn_stream(void)
{
spdylay_zlib deflater, inflater;
spdylay_frame frame, oframe;
uint8_t *buf = NULL, *nvbuf = NULL;
size_t buflen = 0, nvbuflen = 0;
spdylay_buffer inflatebuf;
ssize_t framelen;
const char *nv[] = { ":host", "example.org",
":scheme", "https",
NULL };
char **nv_copy;
int rv;
spdylay_buffer_init(&inflatebuf, 4096);
rv = spdylay_zlib_deflate_hd_init(&deflater, SPDYLAY_PROTO_SPDY3);
if(rv != 0) {
goto deflate_init_fail;
}
rv = spdylay_zlib_inflate_hd_init(&inflater, SPDYLAY_PROTO_SPDY3);
if(rv != 0) {
goto inflate_init_fail;
}
nv_copy = spdylay_frame_nv_copy(nv);
if(nv_copy == NULL) {
goto nv_copy_fail;
}
spdylay_frame_syn_stream_init(&frame.syn_stream, SPDYLAY_PROTO_SPDY3,
SPDYLAY_CTRL_FLAG_FIN, 1, 0, 2, nv_copy);
framelen = spdylay_frame_pack_syn_stream(&buf, &buflen,
&nvbuf, &nvbuflen,
&frame.syn_stream, &deflater);
if(framelen < 0) {
goto fail;
}
rv = spdylay_frame_unpack_syn_stream(&oframe.syn_stream,
&inflatebuf,
&nvbuf, &nvbuflen,
&buf[0], SPDYLAY_FRAME_HEAD_LENGTH,
&buf[SPDYLAY_FRAME_HEAD_LENGTH],
framelen-SPDYLAY_FRAME_HEAD_LENGTH,
&inflater);
if(rv != 0) {
goto fail;
}
spdylay_frame_syn_stream_free(&oframe.syn_stream);
fail:
free(buf);
free(nvbuf);
spdylay_frame_syn_stream_free(&frame.syn_stream);
nv_copy_fail:
spdylay_zlib_inflate_free(&inflater);
inflate_init_fail:
spdylay_zlib_deflate_free(&deflater);
deflate_init_fail:
spdylay_buffer_free(&inflatebuf);
}
static void run_spdylay_frame_pack_ping(void)
{
spdylay_frame frame;
uint8_t *buf = NULL;
size_t buflen = 0;
spdylay_frame_ping_init(&frame.ping, SPDYLAY_PROTO_SPDY3, 1);
spdylay_frame_pack_ping(&buf, &buflen, &frame.ping);
free(buf);
spdylay_frame_ping_free(&frame.ping);
}
static void run_spdylay_frame_pack_goaway(void)
{
spdylay_frame frame;
uint8_t *buf = NULL;
size_t buflen = 0;
spdylay_frame_goaway_init(&frame.goaway, SPDYLAY_PROTO_SPDY3, 1000000007,
SPDYLAY_GOAWAY_PROTOCOL_ERROR);
spdylay_frame_pack_goaway(&buf, &buflen, &frame.goaway);
free(buf);
spdylay_frame_goaway_free(&frame.goaway);
}
static void run_spdylay_frame_pack_rst_stream(void)
{
spdylay_frame frame;
uint8_t *buf = NULL;
size_t buflen = 0;
spdylay_frame_rst_stream_init(&frame.rst_stream, SPDYLAY_PROTO_SPDY3, 1,
SPDYLAY_PROTOCOL_ERROR);
spdylay_frame_pack_rst_stream(&buf, &buflen, &frame.rst_stream);
free(buf);
spdylay_frame_rst_stream_free(&frame.rst_stream);
}
static void run_spdylay_frame_pack_window_update(void)
{
spdylay_frame frame;
uint8_t *buf = NULL;
size_t buflen = 0;
spdylay_frame_window_update_init(&frame.window_update, SPDYLAY_PROTO_SPDY3,
1000000007, 4096);
spdylay_frame_pack_window_update(&buf, &buflen,
&frame.window_update);
free(buf);
spdylay_frame_window_update_free(&frame.window_update);
}
static void run_spdylay_frame_pack_settings(void)
{
spdylay_frame frame, oframe;
uint8_t *buf = NULL;
size_t buflen = 0;
ssize_t framelen;
spdylay_settings_entry iv[2], *iv_copy;
int rv;
iv[0].settings_id = SPDYLAY_SETTINGS_UPLOAD_BANDWIDTH;
iv[0].flags = SPDYLAY_ID_FLAG_SETTINGS_PERSIST_VALUE;
iv[0].value = 256;
iv[1].settings_id = SPDYLAY_SETTINGS_MAX_CONCURRENT_STREAMS;
iv[1].flags = SPDYLAY_ID_FLAG_SETTINGS_NONE;
iv[1].value = 100;
iv_copy = spdylay_frame_iv_copy(iv, 2);
if(iv_copy == NULL) {
goto iv_copy_fail;
}
spdylay_frame_settings_init(&frame.settings, SPDYLAY_PROTO_SPDY3,
SPDYLAY_FLAG_SETTINGS_CLEAR_SETTINGS,
iv_copy, 2);
framelen = spdylay_frame_pack_settings(&buf, &buflen, &frame.settings);
if(framelen < 0) {
goto fail;
}
rv = spdylay_frame_unpack_settings(&oframe.settings,
&buf[0], SPDYLAY_FRAME_HEAD_LENGTH,
&buf[SPDYLAY_FRAME_HEAD_LENGTH],
framelen-SPDYLAY_FRAME_HEAD_LENGTH);
if(rv != 0) {
goto fail;
}
spdylay_frame_settings_free(&oframe.settings);
fail:
free(buf);
spdylay_frame_settings_free(&frame.settings);
iv_copy_fail:
;
}
static void run_spdylay_frame_pack_credential(void)
{
spdylay_frame frame, oframe;
uint8_t *buf = NULL;
size_t buflen = 0;
ssize_t framelen;
spdylay_mem_chunk proof;
spdylay_mem_chunk *certs;
size_t ncerts;
int rv;
spdylay_failmalloc_pause();
proof.data = (uint8_t*)strcopy("PROOF");
proof.length = strlen("PROOF");
ncerts = 2;
certs = malloc(sizeof(spdylay_mem_chunk)*ncerts);
certs[0].data = (uint8_t*)strcopy("CERT0");
certs[0].length = strlen("CERT0");
certs[1].data = (uint8_t*)strcopy("CERT1");
certs[1].length = strlen("CERT1");
spdylay_failmalloc_unpause();
spdylay_frame_credential_init(&frame.credential, SPDYLAY_PROTO_SPDY3,
1, &proof, certs, ncerts);
framelen = spdylay_frame_pack_credential(&buf, &buflen, &frame.credential);
if(framelen < 0) {
goto fail;
}
rv = spdylay_frame_unpack_credential(&oframe.credential,
&buf[0], SPDYLAY_FRAME_HEAD_LENGTH,
&buf[SPDYLAY_FRAME_HEAD_LENGTH],
framelen-SPDYLAY_FRAME_HEAD_LENGTH);
if(rv != 0) {
goto fail;
}
spdylay_frame_credential_free(&oframe.credential);
fail:
free(buf);
spdylay_frame_credential_free(&frame.credential);
}
static void run_spdylay_frame(void)
{
run_spdylay_frame_pack_syn_stream();
run_spdylay_frame_pack_ping();
run_spdylay_frame_pack_goaway();
run_spdylay_frame_pack_rst_stream();
run_spdylay_frame_pack_window_update();
run_spdylay_frame_pack_settings();
run_spdylay_frame_pack_credential();
}
void test_spdylay_frame(void)
{
TEST_FAILMALLOC_RUN(run_spdylay_frame);
}

32
tests/failmalloc_test.h Normal file
View File

@ -0,0 +1,32 @@
/*
* Spdylay - SPDY Library
*
* Copyright (c) 2012 Tatsuhiro Tsujikawa
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#ifndef FAILMALLOC_TEST_H
#define FAILMALLOC_TEST_H
void test_spdylay_session_send(void);
void test_spdylay_session_recv(void);
void test_spdylay_frame(void);
#endif /* FAILMALLOC_TEST_H */

71
tests/malloc_wrapper.c Normal file
View File

@ -0,0 +1,71 @@
/*
* Spdylay - SPDY Library
*
* Copyright (c) 2012 Tatsuhiro Tsujikawa
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#include "malloc_wrapper.h"
#define __USE_GNU
#include <dlfcn.h>
int spdylay_failmalloc = 0;
int spdylay_failstart = 0;
int spdylay_countmalloc = 1;
int spdylay_nmalloc = 0;
static void* (*real_malloc)(size_t) = NULL;
static void init(void)
{
real_malloc = dlsym(RTLD_NEXT, "malloc");
}
void* malloc(size_t size)
{
if(real_malloc == NULL) {
init();
}
if(spdylay_failmalloc && spdylay_nmalloc >= spdylay_failstart) {
return NULL;
} else {
if(spdylay_countmalloc) {
++spdylay_nmalloc;
}
return real_malloc(size);
}
}
static int failmalloc_bk, countmalloc_bk;
void spdylay_failmalloc_pause(void)
{
failmalloc_bk = spdylay_failmalloc;
countmalloc_bk = spdylay_countmalloc;
spdylay_failmalloc = 0;
spdylay_countmalloc = 0;
}
void spdylay_failmalloc_unpause(void)
{
spdylay_failmalloc = failmalloc_bk;
spdylay_countmalloc = countmalloc_bk;
}

56
tests/malloc_wrapper.h Normal file
View File

@ -0,0 +1,56 @@
/*
* Spdylay - SPDY Library
*
* Copyright (c) 2012 Tatsuhiro Tsujikawa
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#ifndef MALLOC_WRAPPER_H
#define MALLOC_WRAPPER_H
#include <stdlib.h>
/* Global variables to control the behavior of malloc() */
/* If nonzero, malloc failure mode is on */
extern int spdylay_failmalloc;
/* If spdylay_failstart <= spdylay_nmalloc and spdylay_failmalloc is
nonzero, malloc() fails. */
extern int spdylay_failstart;
/* If nonzero, spdylay_nmalloc is incremented if malloc() succeeds. */
extern int spdylay_countmalloc;
/* The number of successful invocation of malloc(). This value is only
incremented if spdylay_nmalloc is nonzero. */
extern int spdylay_nmalloc;
void* malloc(size_t size);
/* Copies spdylay_failmalloc and spdylay_countmalloc to statically
allocated space and sets 0 to them. This will effectively make
malloc() work like normal malloc(). This is useful when you want to
disable malloc() failure mode temporarily. */
void spdylay_failmalloc_pause(void);
/* Restores the values of spdylay_failmalloc and spdylay_countmalloc
with the values saved by the previous
spdylay_failmalloc_pause(). */
void spdylay_failmalloc_unpause(void);
#endif /* MALLOC_WRAPPER_H */