summaryrefslogtreecommitdiffstats
path: root/staging_vespalib/src/tests/crc/crc_test.cpp
blob: 8692cd8f6d50832a9c9cc69bea21acb9787e142b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include <vespa/vespalib/testkit/testapp.h>
#include <vespa/vespalib/util/crc.h>
#include <boost/crc.hpp>
#include <vector>

using namespace vespalib;

class Test : public TestApp
{
public:
    int Main() override;
    void testCorrectNess();
    void testBenchmark(bool our, size_t bufSz, size_t numRep);
};

int
Test::Main()
{
    TEST_INIT("crc_test");
    testCorrectNess();
    if (_argc >= 2) {
        testBenchmark(false, 1024, 1000*1000);
    } else {
        testBenchmark(true, 1024, 1000*1000);
    }
    TEST_DONE();
}

void Test::testCorrectNess()
{
    const char *a[7] = { "", "a", "ab", "abc", "abcd", "abcde", "doc:crawler:http://www.ntnu.no/" };
    for (size_t i(0); i < sizeof(a)/sizeof(a[0]); i++) {
        uint32_t vespaCrc32 = crc_32_type::crc(a[i], strlen(a[i]));
        boost::crc_32_type calculator;
        calculator.process_bytes(a[i], strlen(a[i]));
        EXPECT_EQUAL(vespaCrc32, calculator.checksum());
        vespalib::crc_32_type calculator2;
        calculator2.process_bytes(a[i], strlen(a[i]));
        EXPECT_EQUAL(vespaCrc32, calculator2.checksum());
        EXPECT_EQUAL(calculator.checksum(), calculator2.checksum());
    }
    vespalib::crc_32_type calculator2;
    boost::crc_32_type calculator;
    for (size_t i(0); i < sizeof(a)/sizeof(a[0]); i++) {
        calculator.process_bytes(a[i], strlen(a[i]));
        calculator2.process_bytes(a[i], strlen(a[i]));
        EXPECT_EQUAL(calculator.checksum(), calculator2.checksum());
    }
    EXPECT_EQUAL(calculator.checksum(), calculator2.checksum());
}

void Test::testBenchmark(bool our, size_t bufSz, size_t numRep)
{
    std::vector<char> a(numRep+bufSz);
    for(size_t i(0), m(a.size()); i < m; i++) {
        a[i] = i&0xff;
    }
    uint32_t sum(0);
    if (our) {
        for (size_t i(0); i < (numRep); i++) {
            //sum ^= crc_32_type::crc(&a[i], bufSz);
            vespalib::crc_32_type calculator;
            calculator.process_bytes(&a[i], bufSz);
            sum ^=calculator.checksum();
        }
    } else {
        for (size_t i(0); i < (numRep); i++) {
            boost::crc_32_type calculator;
            calculator.process_bytes(&a[i], bufSz);
            sum ^=calculator.checksum();
        }
    }
    printf("sum = %x\n", sum);
}

TEST_APPHOOK(Test)