diff options
| -rw-r--r-- | src/nxt_brotli.c | 78 | ||||
| -rw-r--r-- | src/nxt_http_compression.c | 12 | ||||
| -rw-r--r-- | src/nxt_http_compression.h | 16 |
3 files changed, 106 insertions, 0 deletions
diff --git a/src/nxt_brotli.c b/src/nxt_brotli.c new file mode 100644 index 00000000..773b8b6f --- /dev/null +++ b/src/nxt_brotli.c @@ -0,0 +1,78 @@ +/* + * Copyright (C) Andrew Clayton + * Copyright (C) F5, Inc. + */ + +#include <stddef.h> +#include <stdint.h> +#include <stdbool.h> + +#include <brotli/encode.h> + +#include "nxt_http_compression.h" + + +static int +nxt_brotli_init(nxt_http_comp_compressor_ctx_t *ctx) +{ + BrotliEncoderState **brotli = &ctx->brotli_ctx; + + *brotli = BrotliEncoderCreateInstance(NULL, NULL, NULL); + if (*brotli == NULL) { + return -1; + } + BrotliEncoderSetParameter(*brotli, BROTLI_PARAM_QUALITY, ctx->level); + + return 0; +} + + +static size_t +nxt_brotli_bound(const nxt_http_comp_compressor_ctx_t *ctx, size_t in_len) +{ + return BrotliEncoderMaxCompressedSize(in_len); +} + + +static ssize_t +nxt_brotli_compress(nxt_http_comp_compressor_ctx_t *ctx, const uint8_t *in_buf, + size_t in_len, uint8_t *out_buf, size_t out_len, bool last) +{ + bool ok; + size_t out_bytes = out_len; + BrotliEncoderState *brotli = ctx->brotli_ctx; + + ok = BrotliEncoderCompressStream(brotli, BROTLI_OPERATION_PROCESS, + &in_len, &in_buf, &out_bytes, &out_buf, + NULL); + if (!ok) { + return -1; + } + + ok = BrotliEncoderCompressStream(brotli, BROTLI_OPERATION_FLUSH, + &in_len, &in_buf, &out_bytes, &out_buf, + NULL); + if (!ok) { + return -1; + } + + if (last) { + ok = BrotliEncoderCompressStream(brotli, BROTLI_OPERATION_FINISH, + &in_len, &in_buf, &out_bytes, + &out_buf, NULL); + if (!ok) { + return -1; + } + + BrotliEncoderDestroyInstance(brotli); + } + + return out_len - out_bytes; +} + + +const nxt_http_comp_operations_t nxt_http_comp_brotli_ops = { + .init = nxt_brotli_init, + .bound = nxt_brotli_bound, + .deflate = nxt_brotli_compress, +}; diff --git a/src/nxt_http_compression.c b/src/nxt_http_compression.c index dfd4a13a..e6083ec3 100644 --- a/src/nxt_http_compression.c +++ b/src/nxt_http_compression.c @@ -34,6 +34,9 @@ enum nxt_http_comp_scheme_e { #if NXT_HAVE_ZSTD NXT_HTTP_COMP_SCHEME_ZSTD, #endif +#if NXT_HAVE_BROTLI + NXT_HTTP_COMP_SCHEME_BROTLI, +#endif /* keep last */ NXT_HTTP_COMP_SCHEME_UNKNOWN @@ -121,6 +124,15 @@ static const nxt_http_comp_type_t nxt_http_comp_compressors[] = { .comp_max = NXT_HTTP_COMP_ZSTD_COMP_MAX, .cops = &nxt_http_comp_zstd_ops, #endif +#if NXT_HAVE_BROTLI + }, { + .token = nxt_string("br"), + .scheme = NXT_HTTP_COMP_SCHEME_BROTLI, + .def_compr = NXT_HTTP_COMP_BROTLI_DEFAULT_LEVEL, + .comp_min = NXT_HTTP_COMP_BROTLI_COMP_MIN, + .comp_max = NXT_HTTP_COMP_BROTLI_COMP_MAX, + .cops = &nxt_http_comp_brotli_ops, +#endif }, }; diff --git a/src/nxt_http_compression.h b/src/nxt_http_compression.h index 7c731a86..c2007f8e 100644 --- a/src/nxt_http_compression.h +++ b/src/nxt_http_compression.h @@ -20,6 +20,10 @@ #include <zstd.h> #endif +#if NXT_HAVE_BROTLI +#include <brotli/encode.h> +#endif + #include <nxt_main.h> #include <nxt_router.h> #include <nxt_string.h> @@ -36,6 +40,11 @@ #define NXT_HTTP_COMP_ZSTD_COMP_MIN (-7) #define NXT_HTTP_COMP_ZSTD_COMP_MAX 22 #endif +#if NXT_HAVE_BROTLI +#define NXT_HTTP_COMP_BROTLI_DEFAULT_LEVEL BROTLI_DEFAULT_QUALITY +#define NXT_HTTP_COMP_BROTLI_COMP_MIN BROTLI_MIN_QUALITY +#define NXT_HTTP_COMP_BROTLI_COMP_MAX BROTLI_MAX_QUALITY +#endif typedef struct nxt_http_comp_compressor_ctx_s nxt_http_comp_compressor_ctx_t; @@ -51,6 +60,9 @@ struct nxt_http_comp_compressor_ctx_s { #if NXT_HAVE_ZSTD ZSTD_CStream *zstd_ctx; #endif +#if NXT_HAVE_BROTLI + BrotliEncoderState *brotli_ctx; +#endif }; }; @@ -73,6 +85,10 @@ extern const nxt_http_comp_operations_t nxt_http_comp_gzip_ops; extern const nxt_http_comp_operations_t nxt_http_comp_zstd_ops; #endif +#if NXT_HAVE_BROTLI +extern const nxt_http_comp_operations_t nxt_http_comp_brotli_ops; +#endif + extern bool nxt_http_comp_wants_compression(void); extern bool nxt_http_comp_compressor_is_valid(const nxt_str_t *token); |