1 files changed, 171 insertions, 0 deletions
diff --git a/web/server/h2o/libh2o/t/00unit/lib/common/socket.c b/web/server/h2o/libh2o/t/00unit/lib/common/socket.c
new file mode 100644
index 00000000..89aae515
--- /dev/null
+++ b/web/server/h2o/libh2o/t/00unit/lib/common/socket.c
@@ -0,0 +1,171 @@
+/*
+ * Copyright (c) 2015 DeNA Co., Ltd.
+ *
+ * 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 <stdlib.h>
+#include "../../test.h"
+#include "../../../../lib/common/socket.c"
+
+static void test_on_alpn_select(void)
+{
+    static const h2o_iovec_t protocols[] = {{H2O_STRLIT("h2")}, {H2O_STRLIT("h2-16")}, {H2O_STRLIT("h2-14")}, {NULL}};
+    const unsigned char *out;
+    unsigned char outlen;
+    int ret;
+
+    ret = on_alpn_select(NULL, &out, &outlen, (const unsigned char *)H2O_STRLIT("\3foo"), (void *)protocols);
+    ok(ret == SSL_TLSEXT_ERR_NOACK);
+
+    ret = on_alpn_select(NULL, &out, &outlen, (const unsigned char *)H2O_STRLIT("\2h2"), (void *)protocols);
+    ok(ret == SSL_TLSEXT_ERR_OK);
+    ok(h2o_memis(out, outlen, H2O_STRLIT("h2")));
+
+    ret = on_alpn_select(NULL, &out, &outlen, (const unsigned char *)H2O_STRLIT("\5h2-14\5h2-16\2h2"), (void *)protocols);
+    ok(ret == SSL_TLSEXT_ERR_OK);
+    ok(h2o_memis(out, outlen, H2O_STRLIT("h2")));
+
+    ret = on_alpn_select(NULL, &out, &outlen, (const unsigned char *)H2O_STRLIT("\5h2-14\5h2-16"), (void *)protocols);
+    ok(ret == SSL_TLSEXT_ERR_OK);
+    ok(h2o_memis(out, outlen, H2O_STRLIT("h2-16")));
+}
+
+static void test_sliding_counter(void)
+{
+    h2o_sliding_counter_t counter = {0};
+    size_t i;
+
+    h2o_sliding_counter_start(&counter, 100);
+    h2o_sliding_counter_stop(&counter, 80 + 100);
+    ok(counter.average == 10);
+
+    for (i = 0; i != 7; ++i) {
+        h2o_sliding_counter_start(&counter, 1);
+        h2o_sliding_counter_stop(&counter, 81);
+    }
+    ok(counter.average == 80);
+
+    h2o_sliding_counter_start(&counter, 1000);
+    h2o_sliding_counter_stop(&counter, 1000 + 1000 * 8 - 80 * 7);
+    ok(counter.average == 1000);
+
+    for (i = 0; i != 8; ++i) {
+        h2o_sliding_counter_start(&counter, 1);
+        h2o_sliding_counter_stop(&counter, 11);
+    }
+    ok(counter.average == 10);
+}
+
+static struct {
+    struct {
+        int ret;
+    } fetch_tcp_info;
+    struct {
+        int ret;
+        unsigned cur;
+        size_t call_cnt;
+    } minimize_notsent_lowat;
+    struct {
+        unsigned long ret;
+    } get_cipher;
+} cb_ret_vars;
+
+static int test_adjust_notsent_lowat(h2o_socket_t *sock, unsigned notsent_lowat)
+{
+    cb_ret_vars.minimize_notsent_lowat.cur = notsent_lowat;
+    ++cb_ret_vars.minimize_notsent_lowat.call_cnt;
+    return cb_ret_vars.minimize_notsent_lowat.ret;
+}
+
+static void test_prepare_for_latency_optimization(void)
+{
+    struct st_h2o_socket_ssl_t sock_ssl = {NULL, NULL, NULL, 5 + 8 + 16 /* GCM overhead */};
+    h2o_socket_t sock = {NULL, &sock_ssl};
+    h2o_socket_latency_optimization_conditions_t cond = {UINT_MAX, 10, 65535};
+
+    /* option disabled, or if rtt is too small */
+    memset(&sock._latency_optimization, 0, sizeof(sock._latency_optimization));
+    memset(&cb_ret_vars, 0, sizeof(cb_ret_vars));
+    cb_ret_vars.fetch_tcp_info.ret = 0;
+    cb_ret_vars.get_cipher.ret = TLS1_CK_RSA_WITH_AES_128_GCM_SHA256;
+    prepare_for_latency_optimized_write(&sock, &cond, 50000 /* rtt */, 1400 /* mss */, 10 /* cwnd_size */, 6 /* cwnd_avail */, 4,
+                                        test_adjust_notsent_lowat);
+    ok(sock._latency_optimization.state == H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_DISABLED);
+    ok(sock._latency_optimization.suggested_tls_payload_size == 16384);
+    ok(sock._latency_optimization.suggested_write_size == SIZE_MAX);
+    ok(cb_ret_vars.minimize_notsent_lowat.call_cnt == 0);
+
+    /* trigger optimiziation */
+    memset(&sock._latency_optimization, 0, sizeof(sock._latency_optimization));
+    cond.min_rtt = 25; /* 25 ms */
+    prepare_for_latency_optimized_write(&sock, &cond, 50000 /* rtt */, 1400 /* mss */, 10 /* cwnd_size */, 6 /* cwnd_avail */, 4,
+                                        test_adjust_notsent_lowat);
+    ok(sock._latency_optimization.state == H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_DETERMINED);
+    ok(sock._latency_optimization.suggested_tls_payload_size == 1400 - (5 + 8 + 16));
+    ok(sock._latency_optimization.suggested_write_size == (1400 - (5 + 8 + 16)) * (10 - 5 + 1));
+    ok(cb_ret_vars.minimize_notsent_lowat.call_cnt == 1);
+    ok(cb_ret_vars.minimize_notsent_lowat.cur == 1);
+
+    /* recalculate with an updated cwnd,unacked */
+    sock._latency_optimization.state = H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_NEEDS_UPDATE;
+    prepare_for_latency_optimized_write(&sock, &cond, 50000 /* rtt */, 1400 /* mss */, 14 /* cwnd_size */, 12 /* cwnd_avail */, 4,
+                                        test_adjust_notsent_lowat);
+    ok(sock._latency_optimization.state == H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_DETERMINED);
+    ok(sock._latency_optimization.suggested_tls_payload_size == 1400 - (5 + 8 + 16));
+    ok(sock._latency_optimization.suggested_write_size == (1400 - (5 + 8 + 16)) * (14 - 3 + 1));
+    ok(cb_ret_vars.minimize_notsent_lowat.call_cnt == 1);
+    ok(cb_ret_vars.minimize_notsent_lowat.cur == 1);
+
+    /* switches to B/W optimization when CWND becomes greater */
+    sock._latency_optimization.state = H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_NEEDS_UPDATE;
+    prepare_for_latency_optimized_write(&sock, &cond, 50000 /* rtt */, 1400 /* mss */, (65535 / 1400) + 1 /* cwnd_size */,
+                                        (65535 / 1400) + 1 /* cwnd_avail */, 4, test_adjust_notsent_lowat);
+    ok(sock._latency_optimization.state == H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_DETERMINED);
+    ok(sock._latency_optimization.suggested_tls_payload_size == 16384);
+    ok(sock._latency_optimization.suggested_write_size == SIZE_MAX);
+    ok(cb_ret_vars.minimize_notsent_lowat.call_cnt == 2);
+    ok(cb_ret_vars.minimize_notsent_lowat.cur == 0);
+
+    /* switches back to latency optimization when CWND becomes small */
+    sock._latency_optimization.state = H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_NEEDS_UPDATE;
+    prepare_for_latency_optimized_write(&sock, &cond, 50000 /* rtt */, 1400 /* mss */, 8 /* cwnd_size */, 6 /* cwnd_avail */, 4,
+                                        test_adjust_notsent_lowat);
+    ok(sock._latency_optimization.state == H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_DETERMINED);
+    ok(sock._latency_optimization.suggested_tls_payload_size == 1400 - (5 + 8 + 16));
+    ok(sock._latency_optimization.suggested_write_size == (1400 - (5 + 8 + 16)) * (8 - 3 + 1));
+    ok(cb_ret_vars.minimize_notsent_lowat.call_cnt == 3);
+    ok(cb_ret_vars.minimize_notsent_lowat.cur == 1);
+
+    /* switches back to B/W optimization when loop time becomes greater than threshold */
+    sock._latency_optimization.state = H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_NEEDS_UPDATE;
+    prepare_for_latency_optimized_write(&sock, &cond, 50000 /* rtt */, 1400 /* mss */, 8 /* cwnd_size */, 6 /* cwnd_avail */, 6,
+                                        test_adjust_notsent_lowat);
+    ok(sock._latency_optimization.state == H2O_SOCKET_LATENCY_OPTIMIZATION_STATE_DISABLED);
+    ok(sock._latency_optimization.suggested_tls_payload_size == 16384);
+    ok(sock._latency_optimization.suggested_write_size == SIZE_MAX);
+    ok(cb_ret_vars.minimize_notsent_lowat.call_cnt == 4);
+    ok(cb_ret_vars.minimize_notsent_lowat.cur == 0);
+}
+
+void test_lib__common__socket_c(void)
+{
+    subtest("on_alpn_select", test_on_alpn_select);
+    subtest("sliding_counter", test_sliding_counter);
+    subtest("prepare_for_latency_optimization", test_prepare_for_latency_optimization);
+}