test_http_range.cc

Go to the documentation of this file.

/*
 * Copyright (C) 1996-2023 The Squid Software Foundation and contributors
 *
 * Squid software is distributed under GPLv2+ license and includes
 * contributions from numerous individuals and organizations.
 * Please see the COPYING and CONTRIBUTORS files for details.
 */
 
#include "squid.h"
#include "fatal.h"
#include "HttpHeader.h"
#include "HttpHeaderRange.h"
#include "HttpHeaderTools.h"
 
// TODO: refactor as cppunit test
 
static void
testRangeParser(char const *rangestring)
{
    String aString (rangestring);
    HttpHdrRange *range = HttpHdrRange::ParseCreate (&aString);
 
    if (!range)
        exit(EXIT_FAILURE);
 
    HttpHdrRange copy(*range);
 
    assert (copy.specs.size() == range->specs.size());
 
    HttpHdrRange::iterator pos = range->begin();
 
    assert (*pos);
 
    delete range;
}
 
static HttpHdrRange *
rangeFromString(char const *rangestring)
{
    String aString (rangestring);
    HttpHdrRange *range = HttpHdrRange::ParseCreate (&aString);
 
    if (!range)
        exit(EXIT_FAILURE);
 
    return range;
}
 
static void
testRangeIter ()
{
    HttpHdrRange *range=rangeFromString("bytes=0-3, 1-, -2");
    assert (range->specs.size() == 3);
    size_t counter = 0;
    HttpHdrRange::iterator i = range->begin();
 
    while (i != range->end()) {
        ++counter;
        ++i;
    }
 
    assert (counter == 3);
    i = range->begin();
    assert (i - range->begin() == 0);
    ++i;
    assert (i - range->begin() == 1);
    assert (i - range->end() == -2);
}
 
static void
testRangeCanonization()
{
    HttpHdrRange *range=rangeFromString("bytes=0-3, 1-, -2");
    assert (range->specs.size() == 3);
 
    /* 0-3 needs a content length of 4 */
    /* This passes in the extant code - but should it? */
 
    if (!range->canonize(3))
        exit(EXIT_FAILURE);
 
    assert (range->specs.size() == 3);
 
    delete range;
 
    range=rangeFromString("bytes=0-3, 1-, -2");
 
    assert (range->specs.size() == 3);
 
    /* 0-3 needs a content length of 4 */
    if (!range->canonize(4))
        exit(EXIT_FAILURE);
 
    delete range;
 
    range=rangeFromString("bytes=3-6");
 
    assert (range->specs.size() == 1);
 
    /* 3-6 needs a content length of 4 or more */
    if (range->canonize(3))
        exit(EXIT_FAILURE);
 
    delete range;
 
    range=rangeFromString("bytes=3-6");
 
    assert (range->specs.size() == 1);
 
    /* 3-6 needs a content length of 4 or more */
    if (!range->canonize(4))
        exit(EXIT_FAILURE);
 
    delete range;
 
    range=rangeFromString("bytes=1-1,2-3");
 
    assert (range->specs.size()== 2);
 
    if (!range->canonize(4))
        exit(EXIT_FAILURE);
 
    assert (range->specs.size() == 2);
 
    delete range;
}
 
int
main(int, char **)
{
    try {
        Mem::Init();
        /* enable for debugging to console */
        // Debug::debugOptions = xstrdup("ALL,1 64,9");
        // Debug::BanCacheLogUse();
        testRangeParser("bytes=0-3");
        testRangeParser("bytes=-3");
        testRangeParser("bytes=1-");
        testRangeParser("bytes=0-3, 1-, -2");
        testRangeIter();
        testRangeCanonization();
    } catch (const std::exception &e) {
        printf("Error: dying from an unhandled exception: %s\n", e.what());
        return EXIT_FAILURE;
    } catch (...) {
        printf("Error: dying from an unhandled exception.\n");
        return EXIT_FAILURE;
    }
    return EXIT_SUCCESS;
}