[PATCH 3 of 3] Stream: ngx_stream_pass_module

Sergey Kandaurov pluknet at nginx.com
Wed Dec 13 15:45:10 UTC 2023


> On 10 Nov 2023, at 14:07, Roman Arutyunyan <arut at nginx.com> wrote:
> 
> # HG changeset patch
> # User Roman Arutyunyan <arut at nginx.com>
> # Date 1699543504 -14400
> #      Thu Nov 09 19:25:04 2023 +0400
> # Node ID 3cab85fe55272835674b7f1c296796955256d019
> # Parent  1d3464283405a4d8ac54caae9bf1815c723f04c5
> Stream: ngx_stream_pass_module.
> 
> The module allows to pass connections from Stream to other modules such as HTTP
> or Mail, as well as back to Stream.  Previously, this was only possible with
> proxying.  Connections with preread buffer read out from socket cannot be
> passed.
> 
> The module allows to terminate SSL selectively based on SNI.
> 
>    stream {
>        server {
>            listen 8000 default_server;
>            ssl_preread on;
>            ...
>        }
> 
>        server {
>            listen 8000;
>            server_name foo.example.com;
>            pass 8001; # to HTTP
>        }
> 
>        server {
>            listen 8000;
>            server_name bar.example.com;
>            ...
>        }
>    }
> 
>    http {
>        server {
>            listen 8001 ssl;
>            ...
> 
>            location / {
>                root html;
>            }
>        }
>    }
> 
> diff --git a/auto/modules b/auto/modules
> --- a/auto/modules
> +++ b/auto/modules
> @@ -1166,6 +1166,16 @@ if [ $STREAM != NO ]; then
>         . auto/module
>     fi
> 
> +    if [ $STREAM_PASS = YES ]; then
> +        ngx_module_name=ngx_stream_pass_module
> +        ngx_module_deps=
> +        ngx_module_srcs=src/stream/ngx_stream_pass_module.c
> +        ngx_module_libs=
> +        ngx_module_link=$STREAM_PASS
> +
> +        . auto/module
> +    fi
> +
>     if [ $STREAM_SET = YES ]; then
>         ngx_module_name=ngx_stream_set_module
>         ngx_module_deps=
> diff --git a/auto/options b/auto/options
> --- a/auto/options
> +++ b/auto/options
> @@ -127,6 +127,7 @@ STREAM_GEOIP=NO
> STREAM_MAP=YES
> STREAM_SPLIT_CLIENTS=YES
> STREAM_RETURN=YES
> +STREAM_PASS=YES
> STREAM_SET=YES
> STREAM_UPSTREAM_HASH=YES
> STREAM_UPSTREAM_LEAST_CONN=YES
> @@ -337,6 +338,7 @@ use the \"--with-mail_ssl_module\" optio
>         --without-stream_split_clients_module)
>                                          STREAM_SPLIT_CLIENTS=NO    ;;
>         --without-stream_return_module)  STREAM_RETURN=NO           ;;
> +        --without-stream_pass_module)    STREAM_PASS=NO             ;;
>         --without-stream_set_module)     STREAM_SET=NO              ;;
>         --without-stream_upstream_hash_module)
>                                          STREAM_UPSTREAM_HASH=NO    ;;
> @@ -556,6 +558,7 @@ cat << END
>   --without-stream_split_clients_module
>                                      disable ngx_stream_split_clients_module
>   --without-stream_return_module     disable ngx_stream_return_module
> +  --without-stream_pass_module       disable ngx_stream_pass_module
>   --without-stream_set_module        disable ngx_stream_set_module
>   --without-stream_upstream_hash_module
>                                      disable ngx_stream_upstream_hash_module
> diff --git a/src/stream/ngx_stream_pass_module.c b/src/stream/ngx_stream_pass_module.c
> new file mode 100644
> --- /dev/null
> +++ b/src/stream/ngx_stream_pass_module.c
> @@ -0,0 +1,245 @@
> +
> +/*
> + * Copyright (C) Roman Arutyunyan
> + * Copyright (C) Nginx, Inc.
> + */
> +
> +
> +#include <ngx_config.h>
> +#include <ngx_core.h>
> +#include <ngx_stream.h>
> +
> +
> +typedef struct {
> +    ngx_addr_t                  *addr;
> +    ngx_stream_complex_value_t  *addr_value;
> +} ngx_stream_pass_srv_conf_t;
> +
> +
> +static void ngx_stream_pass_handler(ngx_stream_session_t *s);
> +static void *ngx_stream_pass_create_srv_conf(ngx_conf_t *cf);
> +static char *ngx_stream_pass(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
> +
> +
> +static ngx_command_t  ngx_stream_pass_commands[] = {
> +
> +    { ngx_string("pass"),
> +      NGX_STREAM_SRV_CONF|NGX_CONF_TAKE1,
> +      ngx_stream_pass,
> +      NGX_STREAM_SRV_CONF_OFFSET,
> +      0,
> +      NULL },
> +
> +      ngx_null_command
> +};
> +
> +
> +static ngx_stream_module_t  ngx_stream_pass_module_ctx = {
> +    NULL,                                  /* preconfiguration */
> +    NULL,                                  /* postconfiguration */
> +
> +    NULL,                                  /* create main configuration */
> +    NULL,                                  /* init main configuration */
> +
> +    ngx_stream_pass_create_srv_conf,       /* create server configuration */
> +    NULL                                   /* merge server configuration */
> +};
> +
> +
> +ngx_module_t  ngx_stream_pass_module = {
> +    NGX_MODULE_V1,
> +    &ngx_stream_pass_module_ctx,           /* module conaddr */

typo: s/conaddr/context/

Otherwise, looks good.

> +    ngx_stream_pass_commands,              /* module directives */
> +    NGX_STREAM_MODULE,                     /* module type */
> +    NULL,                                  /* init master */
> +    NULL,                                  /* init module */
> +    NULL,                                  /* init process */
> +    NULL,                                  /* init thread */
> +    NULL,                                  /* exit thread */
> +    NULL,                                  /* exit process */
> +    NULL,                                  /* exit master */
> +    NGX_MODULE_V1_PADDING
> +};
> +
> +
> +static void
> +ngx_stream_pass_handler(ngx_stream_session_t *s)
> +{
> +    ngx_url_t                    u;
> +    ngx_str_t                    url;
> +    ngx_addr_t                  *addr;
> +    ngx_uint_t                   i;
> +    ngx_listening_t             *ls;
> +    ngx_connection_t            *c;
> +    ngx_stream_pass_srv_conf_t  *pscf;
> +
> +    c = s->connection;
> +
> +    c->log->action = "passing connection to another module";
> +
> +    if (c->buffer && c->buffer->pos != c->buffer->last) {
> +        ngx_log_error(NGX_LOG_ERR, s->connection->log, 0,
> +                      "cannot pass connection with preread data");
> +        goto failed;
> +    }
> +
> +    pscf = ngx_stream_get_module_srv_conf(s, ngx_stream_pass_module);
> +
> +    addr = pscf->addr;
> +
> +    if (addr == NULL) {
> +        if (ngx_stream_complex_value(s, pscf->addr_value, &url) != NGX_OK) {
> +            goto failed;
> +        }
> +
> +        ngx_memzero(&u, sizeof(ngx_url_t));
> +
> +        u.url = url;
> +        u.listen = 1;
> +        u.no_resolve = 1;
> +
> +        if (ngx_parse_url(s->connection->pool, &u) != NGX_OK) {
> +            if (u.err) {
> +                ngx_log_error(NGX_LOG_ERR, s->connection->log, 0,
> +                              "%s in pass \"%V\"", u.err, &u.url);
> +            }
> +
> +            goto failed;
> +        }
> +
> +        if (u.naddrs == 0) {
> +            ngx_log_error(NGX_LOG_ERR, s->connection->log, 0,
> +                          "no addresses in pass \"%V\"", &u.url);
> +            goto failed;
> +        }
> +
> +        addr = &u.addrs[0];
> +    }
> +
> +    ngx_log_debug1(NGX_LOG_DEBUG_STREAM, c->log, 0,
> +                   "stream pass addr: \"%V\"", &addr->name);
> +
> +    ls = ngx_cycle->listening.elts;
> +
> +    for (i = 0; i < ngx_cycle->listening.nelts; i++) {
> +        if (ngx_cmp_sockaddr(ls[i].sockaddr, ls[i].socklen,
> +                             addr->sockaddr, addr->socklen, 1)
> +            == NGX_OK)
> +        {
> +            c->listening = &ls[i];
> +
> +            c->data = NULL;
> +            c->buffer = NULL;
> +
> +            *c->log = c->listening->log;
> +            c->log->handler = NULL;
> +            c->log->data = NULL;
> +
> +            c->listening->handler(c);
> +
> +            return;
> +        }
> +    }
> +
> +    ngx_log_error(NGX_LOG_ERR, c->log, 0,
> +                  "listen not found for \"%V\"", &addr->name);
> +
> +    ngx_stream_finalize_session(s, NGX_STREAM_OK);
> +
> +    return;
> +
> +failed:
> +
> +    ngx_stream_finalize_session(s, NGX_STREAM_INTERNAL_SERVER_ERROR);
> +}
> +
> +
> +static void *
> +ngx_stream_pass_create_srv_conf(ngx_conf_t *cf)
> +{
> +    ngx_stream_pass_srv_conf_t  *conf;
> +
> +    conf = ngx_pcalloc(cf->pool, sizeof(ngx_stream_pass_srv_conf_t));
> +    if (conf == NULL) {
> +        return NULL;
> +    }
> +
> +    /*
> +     * set by ngx_pcalloc():
> +     *
> +     *     conf->addr = NULL;
> +     *     conf->addr_value = NULL;
> +     */
> +
> +    return conf;
> +}
> +
> +
> +static char *
> +ngx_stream_pass(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
> +{
> +    ngx_stream_pass_srv_conf_t *pscf = conf;
> +
> +    ngx_url_t                            u;
> +    ngx_str_t                           *value, *url;
> +    ngx_stream_complex_value_t           cv;
> +    ngx_stream_core_srv_conf_t          *cscf;
> +    ngx_stream_compile_complex_value_t   ccv;
> +
> +    if (pscf->addr || pscf->addr_value) {
> +        return "is duplicate";
> +    }
> +
> +    cscf = ngx_stream_conf_get_module_srv_conf(cf, ngx_stream_core_module);
> +
> +    cscf->handler = ngx_stream_pass_handler;
> +
> +    value = cf->args->elts;
> +
> +    url = &value[1];
> +
> +    ngx_memzero(&ccv, sizeof(ngx_stream_compile_complex_value_t));
> +
> +    ccv.cf = cf;
> +    ccv.value = url;
> +    ccv.complex_value = &cv;
> +
> +    if (ngx_stream_compile_complex_value(&ccv) != NGX_OK) {
> +        return NGX_CONF_ERROR;
> +    }
> +
> +    if (cv.lengths) {
> +        pscf->addr_value = ngx_palloc(cf->pool,
> +                                      sizeof(ngx_stream_complex_value_t));
> +        if (pscf->addr_value == NULL) {
> +            return NGX_CONF_ERROR;
> +        }
> +
> +        *pscf->addr_value = cv;
> +
> +        return NGX_CONF_OK;
> +    }
> +
> +    ngx_memzero(&u, sizeof(ngx_url_t));
> +
> +    u.url = *url;
> +    u.listen = 1;
> +
> +    if (ngx_parse_url(cf->pool, &u) != NGX_OK) {
> +        if (u.err) {
> +            ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
> +                               "%s in \"%V\" of the \"pass\" directive",
> +                               u.err, &u.url);
> +        }
> +
> +        return NGX_CONF_ERROR;
> +    }
> +
> +    if (u.naddrs == 0) {
> +        return "has no addresses";
> +    }
> +
> +    pscf->addr = &u.addrs[0];
> +
> +    return NGX_CONF_OK;
> +}

-- 
Sergey Kandaurov


More information about the nginx-devel mailing list