Sun, 15 Sep 2024 09:47:36 +0200
improve sessionhandler trace logging
/* * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. * * Copyright 2019 Olaf Wintermann. All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * POSSIBILITY OF SUCH DAMAGE. */ #include <stdio.h> #include <stdlib.h> #include <limits.h> #include <cx/string.h> #include <cx/utils.h> #include <cx/printf.h> #include "../util/pblock.h" #include "../util/io.h" #include "testutils.h" Session* testutil_session(void) { pool_handle_t *pool = pool_create(); NSAPISession *sn = nsapisession_create(pool); sn->connection = testutil_dummy_connection(pool); return &sn->sn; } Request* testutil_request(pool_handle_t *pool, const char *method, const char *uri) { NSAPIRequest *rq = pool_malloc(pool, sizeof(NSAPIRequest)); ZERO(rq, sizeof(NSAPIRequest)); HTTPRequest httprequest; ZERO(&httprequest, sizeof(HTTPRequest)); request_initialize(pool, &httprequest, rq); cxmutstr clf = cx_asprintf("%s %s HTTP/1.1", method, uri); pblock_kvinsert( pb_key_clf_request, clf.ptr, clf.length, rq->rq.reqpb); free(clf.ptr); pblock_nvinsert( "method", method, rq->rq.reqpb); pblock_nvinsert( "protocol", "HTTP/1.1", rq->rq.reqpb); pblock_nvinsert("uri", uri, rq->rq.reqpb); return &rq->rq; } static int dummyconn_read(Connection *conn, void *buf, int len) { return len; } static int dummyconn_write(Connection *conn, const void *buf, int len) { return len; } static void dummyconn_close(Connection *conn) { } Connection* testutil_dummy_connection(pool_handle_t *pool) { Connection *conn = pool_malloc(pool, sizeof(Connection)); ZERO(conn, sizeof(Connection)); conn->read = dummyconn_read; conn->write = dummyconn_write; conn->close = dummyconn_close; return conn; } void testutil_request_body(Session *sn, Request *rq, const char *body, size_t len) { cxmutstr cl = cx_asprintf("%d", (int)len); pblock_nvreplace("content-length", cl.ptr, rq->headers); free(cl.ptr); netbuf *inbuf = pool_malloc(sn->pool, sizeof(netbuf)); inbuf->sd = NULL; inbuf->inbuf = pool_malloc(sn->pool, len); inbuf->pos = 0; inbuf->maxsize = len; inbuf->cursize = len; sn->inbuf = inbuf; memcpy(inbuf->inbuf, body, len); } void testutil_destroy_session(Session *sn) { pool_destroy(sn->pool); } static ssize_t test_io_write(IOStream *io, const void *buf, size_t size) { TestIOStream *st = (TestIOStream*)io; if(size > st->max_write) size = st->max_write; return cxBufferWrite(buf, 1, size, st->buf); } static ssize_t test_io_writev(IOStream *io, struct iovec *iovec, int iovctn) { TestIOStream *st = (TestIOStream*)io; ssize_t wv = 0; for(int i=0;i<iovctn;i++) { ssize_t available = st->max_write - wv; size_t len = iovec[i].iov_len; if(len > available) { len = available; } ssize_t w = test_io_write(io, iovec[i].iov_base, len); if(w <= 0) { break; } wv += w; if(wv >= st->max_write) { break; } } return wv; } static ssize_t test_io_read(IOStream *io, void *buf, size_t size) { return -1; } static void test_io_close(IOStream *io) { } static void test_io_finish(IOStream *io) { } static void test_io_setmode(IOStream *io, int mode) { } static int test_io_poll(IOStream *io, EventHandler *ev, int events , Event *event) { return 1; } TestIOStream* testutil_iostream(size_t size, int autoextend) { TestIOStream *stream = calloc(1, sizeof(TestIOStream)); int flags = 0; if(autoextend) { flags = CX_BUFFER_AUTO_EXTEND|CX_BUFFER_FREE_CONTENTS; } stream->buf = malloc(sizeof(CxBuffer)); stream->max_write = INT_MAX; cxBufferInit(stream->buf, NULL, size, cxDefaultAllocator, flags); stream->io.st.write = test_io_write; stream->io.st.writev = test_io_writev; stream->io.st.close = test_io_close; stream->io.st.finish = test_io_finish; stream->io.st.type = IO_STREAM_TYPE_HTTP; return stream; } void testutil_iostream_destroy(TestIOStream *stream) { cxBufferDestroy(stream->buf); free(stream->buf); free(stream); }