Add an NPN callback helper that finds the correct version of spdy

.gitignore

@@ -26,3 +26,4 @@ stamp-h1
 .deps/
 INSTALL
 .DS_STORE
+tests/main

examples/spdylay_ssl.cc

@@ -328,23 +328,15 @@ int select_next_proto_cb(SSL* ssl,
                          const unsigned char *in, unsigned int inlen,
                          void *arg)
 {
-  *out = (unsigned char*)in+1;
+  if (spdylay_select_next_protocol(out, outlen, in, inlen) == 0) {
-  *outlen = in[0];
+    std::cerr << "Invalid protocol: "
+              << std::string((const char*)*out, (size_t)*outlen) << std::endl;
+    abort();
+  }
   if(ssl_debug) {
     print_timer();
-    std::cout << " NPN select next protocol: the remote server offers:"
+    std::cout << " NPN selected the protocol: "
-              << std::endl;
+              << std::string((const char*)*out, (size_t)*outlen) << std::endl;
-  }
-  for(unsigned int i = 0; i < inlen; i += in[i]+1) {
-    if(ssl_debug) {
-      std::cout << "  * ";
-      std::cout.write(reinterpret_cast<const char*>(&in[i+1]), in[i]);
-      std::cout << std::endl;
-    }
-    if(in[i] == 6 && memcmp(&in[i+1], "spdy/2", in[i]) == 0) {
-      *out = (unsigned char*)in+i+1;
-      *outlen = in[i];
-    }
   }
   return SSL_TLSEXT_ERR_OK;
 }

lib/Makefile.am

@@ -33,7 +33,7 @@ lib_LTLIBRARIES = libspdylay.la
 
 OBJECTS = spdylay_pq.c spdylay_map.c spdylay_queue.c \
 	spdylay_buffer.c spdylay_frame.c spdylay_zlib.c \
-	spdylay_session.c spdylay_helper.c spdylay_stream.c
+	spdylay_session.c spdylay_helper.c spdylay_stream.c spdylay_npn.c
 
 HFILES = spdylay_pq.h spdylay_int.h spdylay_map.h spdylay_queue.h \
 	spdylay_buffer.h spdylay_frame.h spdylay_zlib.h \

lib/includes/spdylay/spdylay.h

@@ -406,6 +406,24 @@ int spdylay_submit_ping(spdylay_session *session);
  */
 int spdylay_submit_goaway(spdylay_session *session);
 
+/* A helper function for dealing with NPN.  Returns the version of spdy that
+ * was negotiated.  To use this method you should do something like:
+ * static int select_next_proto_cb(SSL* ssl,
+ *                                 unsigned char **out, unsigned char *outlen,
+ *                                 const unsigned char *in, unsigned int inlen,
+ *                                 void *arg)
+ * {
+ *   if (spdylay_select_next_protocol(out, outlen, in, inlen) > 0) {
+ *     ((MyType*)arg)->spdy = 1;
+ *   }
+ *   return SSL_TLSEXT_ERR_OK;
+ * }
+ * ...
+ * SSL_CTX_set_next_proto_select_cb(ssl_ctx, select_next_proto_cb, my_obj);
+ */
+int spdylay_select_next_protocol(unsigned char **out, unsigned char *outlen,
+                                 const unsigned char *in, unsigned int inlen);
+
 #ifdef __cplusplus
 }
 #endif

lib/spdylay_npn.c

@@ -0,0 +1,42 @@
+/*
+ * 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 <spdylay/spdylay.h>
+#include <string.h>
+
+int spdylay_select_next_protocol(unsigned char **out, unsigned char *outlen,
+                                 const unsigned char *in, unsigned int inlen)
+{
+  unsigned int i = 0;
+  for(; i < inlen; i += in[i]+1) {
+    /* Always assign to *out so that the last one is picked. */
+    *out = (unsigned char*)in+i+1;
+    *outlen = in[i];
+    if(in[i] == 6 && memcmp(&in[i+1], "spdy/2", in[i]) == 0) {
+      return 2;
+    }
+  }
+  return 0;
+}

tests/Makefile.am

@@ -27,17 +27,17 @@ check_PROGRAMS = main
 
 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_frame_test.c
+	spdylay_frame_test.c spdylay_npn_test.c
 
 HFILES = spdylay_pq_test.h spdylay_map_test.h spdylay_queue_test.h \
 	spdylay_buffer_test.h spdylay_zlib_test.h spdylay_session_test.h \
-	spdylay_frame_test.h
+	spdylay_frame_test.h spdylay_npn_test.h
 
 main_SOURCES = $(HFILES) $(OBJECTS)
 
-main_LDADD = ${top_builddir}/lib/libspdylay.la -lcunit
+main_LDADD = ${top_builddir}/lib/libspdylay.la
-main_LDFLAGS = -static
+main_LDFLAGS = -static `pkg-config --libs cunit` -lncurses
-AM_CFLAGS = -Wall -g -O2 -I${top_srcdir}/lib -I${top_srcdir}/lib/includes
+AM_CFLAGS = -Wall -g -O2 -I${top_srcdir}/lib -I${top_srcdir}/lib/includes `pkg-config --cflags cunit`
 TESTS = main
 
 endif # HAVE_CUNIT

tests/main.c

@@ -33,6 +33,7 @@
 #include "spdylay_zlib_test.h"
 #include "spdylay_session_test.h"
 #include "spdylay_frame_test.h"
+#include "spdylay_npn_test.h"
 
 int init_suite1(void)
 {
@@ -45,7 +46,7 @@ int clean_suite1(void)
 }
 
 
-int main()
+int main(int argc, char* argv[])
 {
    CU_pSuite pSuite = NULL;
 
@@ -66,6 +67,7 @@ int main()
       !CU_add_test(pSuite, "queue", test_spdylay_queue) ||
       !CU_add_test(pSuite, "buffer", test_spdylay_buffer) ||
       !CU_add_test(pSuite, "zlib", test_spdylay_zlib) ||
+      !CU_add_test(pSuite, "npn", test_spdylay_npn) ||
       !CU_add_test(pSuite, "session_recv", test_spdylay_session_recv) ||
       !CU_add_test(pSuite, "session_recv_invalid_stream_id",
                    test_spdylay_session_recv_invalid_stream_id) ||

tests/spdylay_npn_test.c

@@ -0,0 +1,65 @@
+/*
+ * Spdylay - SPDY Library
+ *
+ * Copyright (c) 2012 Twist Inc.
+ *
+ * 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 <CUnit/CUnit.h>
+#include <spdylay/spdylay.h>
+#include <string.h>
+
+static void spdy2()
+{
+  const unsigned char spdy[] = {
+    8, 'h', 't', 't', 'p', '/', '1', '.', '0',
+    6, 's', 'p', 'd', 'y', '/', '2',
+    6, 's', 'p', 'd', 'y', '/', '3'
+  };
+  unsigned char outlen;
+  unsigned char* out;
+  CU_ASSERT(2 == spdylay_select_next_protocol(&out, &outlen,
+                                              spdy, sizeof(spdy)));
+  CU_ASSERT(6 == outlen);
+  CU_ASSERT(memcmp("spdy/2", out, 6) == 0);
+}
+
+static void spdy4()
+{
+  const unsigned char spdy[] = {
+    6, 's', 'p', 'd', 'y', '/', '4',
+    8, 's', 'p', 'd', 'y', '/', '2', '.', '1',
+    8, 'h', 't', 't', 'p', '/', '1', '.', '0',
+  };
+  unsigned char outlen;
+  unsigned char* out;
+  CU_ASSERT(0 == spdylay_select_next_protocol(&out, &outlen,
+                                              spdy, sizeof(spdy)));
+  CU_ASSERT(8 == outlen);
+  CU_ASSERT(memcmp("http/1.0", out, outlen) == 0);
+}
+
+void test_spdylay_npn()
+{
+  spdy2();
+  spdy4();
+}  
+

tests/spdylay_npn_test.h

@@ -0,0 +1,30 @@
+/*
+ * Spdylay - SPDY Library
+ *
+ * Copyright (c) 2012 Twist Inc.
+ *
+ * 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 SPDYLAY_NPN_TEST_H
+#define SPDYLAY_NPN_TEST_H
+
+void test_spdylay_npn();
+
+#endif // SPDYLAY_NPN_TEST_H