update 2026-09-26 17:00:06

liep/luci-app-nginx-pingos/modules/nginx-client-module/t/ngx_client_test_module.c

@@ -0,0 +1,188 @@
+#include <ngx_config.h>
+#include <ngx_core.h>
+#include <ngx_http.h>
+#include "ngx_client.h"
+
+
+static char *ngx_client_test(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
+
+
+static ngx_command_t  ngx_client_test_commands[] = {
+
+    { ngx_string("client_test"),
+      NGX_HTTP_LOC_CONF|NGX_CONF_NOARGS,
+      ngx_client_test,
+      0,
+      0,
+      NULL },
+
+      ngx_null_command
+};
+
+
+static ngx_http_module_t  ngx_client_test_module_ctx = {
+    NULL,                                   /* preconfiguration */
+    NULL,                                   /* postconfiguration */
+
+    NULL,                                   /* create main configuration */
+    NULL,                                   /* init main configuration */
+
+    NULL,                                   /* create server configuration */
+    NULL,                                   /* merge server configuration */
+
+    NULL,                                   /* create location configuration */
+    NULL                                    /* merge location configuration */
+};
+
+
+ngx_module_t  ngx_client_test_module = {
+    NGX_MODULE_V1,
+    &ngx_client_test_module_ctx,            /* module context */
+    ngx_client_test_commands,               /* module directives */
+    NGX_HTTP_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_client_test_connected(ngx_client_session_t *s)
+{
+    ngx_buf_t                  *b;
+    size_t                      len;
+    ngx_chain_t                 out;
+    ngx_http_request_t         *r;
+    ngx_event_t                *wev;
+
+    ngx_log_error(NGX_LOG_ERR, &s->log, 0, "client connected");
+
+    r = s->data;
+    wev = s->peer.connection->write;
+
+    len = sizeof("nginx client test\n") - 1;
+    b = ngx_create_temp_buf(s->pool, len);
+
+    if (b == NULL) {
+        ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR);
+        return;
+    }
+
+    b->last = ngx_copy(b->last, "nginx client test\n", len);
+    b->last_buf = 1;
+
+    out.buf = b;
+    out.next = NULL;
+
+    ngx_client_write(s, &out);
+
+    ngx_handle_write_event(wev, 0);
+}
+
+static void
+ngx_client_test_recv(ngx_client_session_t *s)
+{
+    ngx_buf_t                  *b;
+    ngx_int_t                   n;
+    ngx_connection_t           *c;
+    ngx_str_t                   recv;
+    ngx_http_request_t         *r;
+
+    c = s->peer.connection;
+    r = s->data;
+
+    b = ngx_create_temp_buf(s->pool, 4096);
+    if (b == NULL) {
+        ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR);
+        return;
+    }
+
+    n = c->recv(c, b->pos, b->end - b->last);
+    if (n == NGX_AGAIN) {
+        ngx_log_error(NGX_LOG_ERR, &s->log, 0, "client recv NGX_AGAIN");
+        return;
+    }
+
+    if (n == NGX_ERROR || n == 0) {
+        ngx_log_error(NGX_LOG_ERR, &s->log, 0, "client recv NGX_ERROR");
+        ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR);
+        ngx_client_close(s);
+        return;
+    }
+
+    b->last += n;
+
+    recv.data = b->pos;
+    recv.len = b->last - b->pos;
+
+    ngx_log_error(NGX_LOG_ERR, &s->log, 0, "client recv %d: %V, %z",
+            n, &recv, recv.len);
+
+    ngx_http_finalize_request(r, NGX_HTTP_FORBIDDEN);
+    ngx_client_set_keepalive(s);
+    return;
+}
+
+static void
+ngx_client_test_send(ngx_client_session_t *s)
+{
+    ngx_log_error(NGX_LOG_ERR, &s->log, 0, "client send");
+}
+
+static void
+ngx_client_test_closed(ngx_client_session_t *s)
+{
+    ngx_log_error(NGX_LOG_ERR, &s->log, 0, "client closed");
+}
+
+static ngx_int_t
+ngx_client_test_handler(ngx_http_request_t *r)
+{
+    ngx_client_session_t           *s;
+    ngx_str_t                       echo;
+
+    ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "client test handler");
+
+    if (ngx_http_arg(r, (u_char *) "echo", sizeof("echo") - 1, &echo)
+            != NGX_OK)
+    {
+        return NGX_HTTP_BAD_REQUEST;
+    }
+
+    s = ngx_client_create(&echo, NULL, 0, r->connection->log);
+    if (s == NULL) {
+        return NGX_HTTP_INTERNAL_SERVER_ERROR;
+    }
+
+    //ci->dynamic_resolver = 0;
+    //ci->recvbuf = 4096;
+
+    s->client_connected = ngx_client_test_connected;
+    s->client_recv = ngx_client_test_recv;
+    s->client_send = ngx_client_test_send;
+    s->client_closed = ngx_client_test_closed;
+    s->data = r;
+
+    ngx_client_connect(s);
+
+    ++r->count;
+
+    return NGX_DONE;
+}
+
+
+static char *
+ngx_client_test(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
+{
+    ngx_http_core_loc_conf_t  *clcf;
+
+    clcf = ngx_http_conf_get_module_loc_conf(cf, ngx_http_core_module);
+    clcf->handler = ngx_client_test_handler;
+
+    return NGX_CONF_OK;
+}