Added tests for malloc failures

.gitignore

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

tests/Makefile.am

@@ -23,7 +23,7 @@
 
 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 \
 	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_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 \
 	@CUNIT_CFLAGS@ @DEFS@
 
-TESTS = main
+TESTS = main failmalloc
 
 if ENABLE_EXAMPLES
 

tests/failmalloc.c

@@ -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();
+}

tests/failmalloc_test.c

@@ -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);
+}

tests/failmalloc_test.h

@@ -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 */

tests/malloc_wrapper.c

@@ -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;
+}

tests/malloc_wrapper.h

@@ -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 */