1 files changed, 138 insertions, 0 deletions
diff --git a/web/server/h2o/libh2o/lib/handler/compress/brotli.cc b/web/server/h2o/libh2o/lib/handler/compress/brotli.cc
new file mode 100644
index 00000000..d4d06e9c
--- /dev/null
+++ b/web/server/h2o/libh2o/lib/handler/compress/brotli.cc
@@ -0,0 +1,138 @@
+/*
+ * Copyright (c) 2016 DeNA Co., Ltd., Kazuho Oku
+ *
+ * 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 <cassert>
+#include <cstdlib>
+#include <vector>
+#include <zlib.h>
+#include "h2o.h"
+#include "encode.h"
+
+namespace {
+    class brotli_context : public h2o_compress_context_t {
+    protected:
+        brotli::BrotliCompressor *brotli_;
+        brotli::BrotliParams params_;
+        std::vector<h2o_iovec_t> bufs_; // all bufs_[nnn].base must be free(3)ed
+    public:
+        brotli_context(int quality, size_t estimated_content_length) : brotli_(NULL) {
+            name = h2o_iovec_init(H2O_STRLIT("br"));
+            transform = _compress;
+            params_.quality = quality;
+            if (estimated_content_length != std::numeric_limits<size_t>::max())
+                _update_lgwin(params_, estimated_content_length);
+        }
+        ~brotli_context() {
+            _clear_bufs();
+            delete brotli_;
+        }
+        static void dispose(void *_self) {
+            brotli_context *self = static_cast<brotli_context *>(_self);
+            self->~brotli_context();
+        }
+    private:
+        void _clear_bufs();
+        void _emit(bool is_last, bool force_flush);
+        void _compress(h2o_iovec_t *inbufs, size_t inbufcnt, int is_final, h2o_iovec_t **outbufs, size_t *outbufcnt);
+        static void _compress(h2o_compress_context_t *self, h2o_iovec_t *inbufs, size_t inbufcnt, h2o_send_state_t state,
+                              h2o_iovec_t **outbufs, size_t *outbufcnt) {
+            static_cast<brotli_context*>(self)->_compress(inbufs, inbufcnt, !h2o_send_state_is_in_progress(state), outbufs, outbufcnt);
+        }
+        static void _update_lgwin(brotli::BrotliParams &params, size_t estimated_content_length);
+    };
+}
+
+void brotli_context::_clear_bufs()
+{
+    for (std::vector<h2o_iovec_t>::iterator i = bufs_.begin(); i != bufs_.end(); ++i)
+        free(i->base);
+    bufs_.clear();
+}
+
+void brotli_context::_emit(bool is_last, bool force_flush)
+{
+    uint8_t *output;
+    size_t out_size;
+    bool ret = brotli_->WriteBrotliData(is_last, force_flush, &out_size, &output);
+    assert(ret);
+    (void)ret;
+    if (out_size != 0)
+        bufs_.push_back(h2o_strdup(NULL, reinterpret_cast<const char *>(output), out_size));
+}
+
+void brotli_context::_compress(h2o_iovec_t *inbufs, size_t inbufcnt, int is_final, h2o_iovec_t **outbufs, size_t *outbufcnt)
+{
+    if (brotli_ == NULL) {
+        if (is_final) {
+            uint64_t len = 0;
+            for (size_t i = 0; i != inbufcnt; ++i)
+                len += inbufs[i].len;
+            if (len < std::numeric_limits<size_t>::max())
+                _update_lgwin(params_, len);
+        }
+        brotli_ = new brotli::BrotliCompressor(params_);
+    }
+
+    _clear_bufs();
+
+    if (inbufcnt != 0) {
+        size_t inbufindex = 0, offset = 0, block_space = brotli_->input_block_size();
+        while (inbufindex != inbufcnt) {
+            size_t copy_len = std::min(block_space, inbufs[inbufindex].len - offset);
+            brotli_->CopyInputToRingBuffer(copy_len, reinterpret_cast<const uint8_t *>(inbufs[inbufindex].base) + offset);
+            offset += copy_len;
+            if (inbufs[inbufindex].len == offset) {
+                if (++inbufindex == inbufcnt)
+                    break;
+                offset = 0;
+            }
+            if (block_space == 0) {
+                _emit(false, false);
+                block_space = brotli_->input_block_size();
+            }
+        }
+        _emit(is_final, !is_final);
+    } else {
+        if (is_final)
+            _emit(true, false);
+    }
+
+    if (is_final) {
+        delete brotli_;
+        brotli_ = NULL;
+    }
+
+    *outbufs = &bufs_.front();
+    *outbufcnt = bufs_.size();
+}
+
+void brotli_context::_update_lgwin(brotli::BrotliParams &params, size_t estimated_content_length)
+{
+    int bits = estimated_content_length > 1 ? sizeof(unsigned long long) * 8 - __builtin_clzll(estimated_content_length - 1) : 1;
+    if (bits < params.lgwin)
+        params.lgwin = std::max(bits, brotli::kMinWindowBits);
+}
+
+h2o_compress_context_t *h2o_compress_brotli_open(h2o_mem_pool_t *pool, int quality, size_t estimated_content_length)
+{
+    brotli_context *ctx = static_cast<brotli_context *>(h2o_mem_alloc_shared(pool, sizeof(*ctx), brotli_context::dispose));
+    return new (ctx) brotli_context(quality, estimated_content_length);
+}