src/server/test/rewrite.c

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2025 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 "rewrite.h"

#include "../daemon/rewrite.h"
#include "../daemon/config.h"
#include <cx/mempool.h>
#include <cx/linked_list.h>
#include <string.h>

CX_TEST(test_rewrite_rule_create) {
    CxMempool *mp = cxMempoolCreate(100, CX_MEMPOOL_TYPE_ADVANCED);
    const CxAllocator *a = mp->allocator;
    
    ServerConfiguration cfg = { NULL };
    cfg.a = a;
    cfg.destr = cxLinkedListCreate(a, NULL, sizeof(ScfgDestr));
    
    CX_TEST_DO {
        RewriteRule *rule = rewrite_rule_create(&cfg, cx_mutstr(NULL), cx_mutstr("/rewrite"));
        CX_TEST_ASSERT(rule != NULL);
        CX_TEST_ASSERT(!rule->has_regex);
        CX_TEST_ASSERT(rule->url != NULL);
        
        cxmutstr newuri = string_template_build_string(rule->url, a, NULL, NULL);
        CX_TEST_ASSERT(!cx_strcmp(newuri, "/rewrite"));
        
        RewriteRule *rule2 = rewrite_rule_create(&cfg, cx_mutstr("/path/(.*)/files/"), cx_mutstr("/rewrite/$1/"));
        CX_TEST_ASSERT(rule2 != NULL);
        CX_TEST_ASSERT(rule2->has_regex);
        CX_TEST_ASSERT(rule2->url != NULL);
    }
    
    server_config_destroy(&cfg);
    cxMempoolFree(mp);
}

CX_TEST(test_rewrite_url_no_regex) {
    CxMempool *mp = cxMempoolCreate(100, CX_MEMPOOL_TYPE_ADVANCED);
    const CxAllocator *a = mp->allocator;
    
    ServerConfiguration cfg = { NULL };
    cfg.a = a;
    cfg.destr = cxLinkedListCreate(a, NULL, sizeof(ScfgDestr));
    
    RewriteRule rule;
    memset(&rule, 0, sizeof(RewriteRule));
    rule.url = string_template_compile(a, cx_str("/static/"));
    
    RewriteRule rule2;
    memset(&rule2, 0, sizeof(RewriteRule));
    rule2.url = string_template_compile(a, cx_str("/dynamic/$0"));
    
    CX_TEST_DO {
        char *new_url = NULL;
        int r = rewrite_url(&rule, NULL, 0, a, "/previous/url", &new_url);
        
        CX_TEST_ASSERT(r == 0);
        CX_TEST_ASSERT(new_url != NULL);
        CX_TEST_ASSERT(!strcmp(new_url, "/static/"));
        
        new_url = NULL;
        regmatch_t m;
        m.rm_so = 6;
        m.rm_eo = 9;
        r = rewrite_url(&rule2, &m, 1, a, "/test/url", &new_url);
        
        CX_TEST_ASSERT(r == 0);
        CX_TEST_ASSERT(new_url != NULL);
        CX_TEST_ASSERT(!strcmp(new_url, "/dynamic/url"));
    }
    
    server_config_destroy(&cfg);
    cxMempoolFree(mp);
}

CX_TEST(test_rewrite_url) {
    CxMempool *mp = cxMempoolCreate(100, CX_MEMPOOL_TYPE_ADVANCED);
    const CxAllocator *a = mp->allocator;
    
    ServerConfiguration cfg = { NULL };
    cfg.a = a;
    cfg.destr = cxLinkedListCreate(a, NULL, sizeof(ScfgDestr));
    
    RewriteRule *rule0 = rewrite_rule_create(&cfg, cx_mutstr("test.*"), cx_mutstr("/rewrite"));
    RewriteRule *rule1 = rewrite_rule_create(&cfg, cx_mutstr("/abc/(.*)\\.html"), cx_mutstr("/newpath/$1.md"));
    RewriteRule *rule2 = rewrite_rule_create(&cfg, cx_mutstr("/abc/(.*)/dir/(.*)"), cx_mutstr("/new/$1/files/$2"));
    
    CX_TEST_DO {
        char *new_url = NULL;
        int r = rewrite_url(rule0, NULL, 0, a, "nomatch", &new_url);
        
        CX_TEST_ASSERT(r == 0);
        CX_TEST_ASSERT(new_url == NULL);
        
        r = rewrite_url(rule0, NULL, 0, a, "test_match", &new_url);
        
        CX_TEST_ASSERT(r == 0);
        CX_TEST_ASSERT(new_url != NULL);
        CX_TEST_ASSERT(!strcmp(new_url, "/rewrite"));
        
        new_url = NULL;
        r = rewrite_url(rule1, NULL, 0, a, "/abc/index.html", &new_url);
        
        CX_TEST_ASSERT(r == 0);
        CX_TEST_ASSERT(new_url != NULL);
        CX_TEST_ASSERT(!strcmp(new_url, "/newpath/index.md"));
        
        new_url = NULL;
        r = rewrite_url(rule2, NULL, 0, a, "/abc/sub/dir/index.html", &new_url);
        
        CX_TEST_ASSERT(r == 0);
        CX_TEST_ASSERT(new_url != NULL);
        CX_TEST_ASSERT(!strcmp(new_url, "/new/sub/files/index.html"));
    }
    
    server_config_destroy(&cfg);
    cxMempoolFree(mp);
}