File: checksums.cpp

package info (click to toggle)
mkvtoolnix 97.0-1
  • links: PTS
  • area: main
  • in suites: forky, sid
  • size: 60,284 kB
  • sloc: cpp: 217,034; ruby: 11,453; xml: 8,125; ansic: 6,885; sh: 5,274; python: 1,041; perl: 191; makefile: 113; awk: 16; javascript: 4
file content (139 lines) | stat: -rw-r--r-- 7,036 bytes parent folder | download | duplicates (3)
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#include "common/common_pch.h"

#include "common/checksums/base.h"
#include "common/mm_file_io.h"
#include "common/mm_proxy_io.h"
#include "common/mm_text_io.h"

#include "tests/unit/init.h"
#include "tests/unit/util.h"

namespace {

class ChecksumTest: public ::testing::Test {
public:
  memory_cptr m_data, m_data_md5, m_onetwothree_md5;
  std::string m_onetwothree;

  ChecksumTest()
    : m_data{mm_file_io_c::slurp("tests/unit/data/text/chapters-with-ebmlvoid.xml")}
    , m_onetwothree{"123456789"}
  {
    unsigned char data_md5_raw[16]        = { 0x43, 0x3a, 0x9a, 0x3e, 0x62, 0xe4, 0x93, 0xb5, 0x15, 0x9d, 0xf9, 0x1f, 0x58, 0x5b, 0x26, 0x91 };
    m_data_md5                            = memory_c::clone(data_md5_raw, 16);

    unsigned char onetwothree_md5_raw[16] = { 0x25, 0xf9, 0xe7, 0x94, 0x32, 0x3b, 0x45, 0x38, 0x85, 0xf5, 0x18, 0x1f, 0x1b, 0x62, 0x4d, 0x0b };
    m_onetwothree_md5                     = memory_c::clone(onetwothree_md5_raw, 16);
  }

  uint32_t
  calculate_int(mtx::checksum::algorithm_e algorithm,
                uint32_t initial_value,
                size_t chunk_size) {
    auto ptr       = m_data->get_buffer();
    auto remaining = m_data->get_size();
    auto worker    = mtx::checksum::for_algorithm(algorithm, initial_value);

    while (remaining) {
      auto to_handle = std::min<size_t>(remaining, chunk_size);

      worker->add(ptr, to_handle);

      ptr       += to_handle;
      remaining -= to_handle;
    }

    worker->finish();

    return dynamic_cast<mtx::checksum::uint_result_c &>(*worker).get_result_as_uint();
  }

  memory_cptr
  calculate_bin(mtx::checksum::algorithm_e algorithm,
                size_t chunk_size) {
    auto ptr       = m_data->get_buffer();
    auto remaining = m_data->get_size();
    auto worker    = mtx::checksum::for_algorithm(algorithm);

    while (remaining) {
      auto to_handle = std::min<size_t>(remaining, chunk_size);

      worker->add(ptr, to_handle);

      ptr       += to_handle;
      remaining -= to_handle;
    }

    worker->finish();

    return worker->get_result();
  }
};

TEST_F(ChecksumTest, OneTwoThree) {
  auto ptr  = reinterpret_cast<unsigned char const *>(m_onetwothree.c_str());

  EXPECT_EQ(0xf4,       mtx::checksum::calculate_as_uint(mtx::checksum::algorithm_e::crc8_atm,      ptr, m_onetwothree.length(),          0));
  EXPECT_EQ(0xe8fe,     mtx::checksum::calculate_as_uint(mtx::checksum::algorithm_e::crc16_ansi,    ptr, m_onetwothree.length(),          0));
  EXPECT_EQ(0xc331,     mtx::checksum::calculate_as_uint(mtx::checksum::algorithm_e::crc16_ccitt,   ptr, m_onetwothree.length(),          0));
  EXPECT_EQ(0xe7e67603, mtx::checksum::calculate_as_uint(mtx::checksum::algorithm_e::crc32_ieee,    ptr, m_onetwothree.length(), 0xffffffff));
  EXPECT_EQ(0x340bc6d9, mtx::checksum::calculate_as_uint(mtx::checksum::algorithm_e::crc32_ieee_le, ptr, m_onetwothree.length(), 0xffffffff));

  EXPECT_EQ(*m_onetwothree_md5, *mtx::checksum::calculate(mtx::checksum::algorithm_e::md5, ptr, m_onetwothree.length()));
}

TEST_F(ChecksumTest, FileAllInOne) {
  EXPECT_EQ(0xab,         calculate_int(mtx::checksum::algorithm_e::crc8_atm,               0, m_data->get_size()));
  EXPECT_EQ(0x18fe,       calculate_int(mtx::checksum::algorithm_e::crc16_ansi,             0, m_data->get_size()));
  EXPECT_EQ(0x218f,       calculate_int(mtx::checksum::algorithm_e::crc16_ccitt,            0, m_data->get_size()));
  EXPECT_EQ(0x5a0a3951,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee,    0xffffffff, m_data->get_size()));
  EXPECT_EQ(0x88c5b46f,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee_le, 0xffffffff, m_data->get_size()));
  EXPECT_EQ(*m_data_md5, *calculate_bin(mtx::checksum::algorithm_e::md5,                       m_data->get_size()));
}

TEST_F(ChecksumTest, FileChunked37) {
  EXPECT_EQ(0xab,         calculate_int(mtx::checksum::algorithm_e::crc8_atm,               0, 37));
  EXPECT_EQ(0x18fe,       calculate_int(mtx::checksum::algorithm_e::crc16_ansi,             0, 37));
  EXPECT_EQ(0x218f,       calculate_int(mtx::checksum::algorithm_e::crc16_ccitt,            0, 37));
  EXPECT_EQ(0x5a0a3951,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee,    0xffffffff, 37));
  EXPECT_EQ(0x88c5b46f,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee_le, 0xffffffff, 37));
  EXPECT_EQ(*m_data_md5, *calculate_bin(mtx::checksum::algorithm_e::md5,                       37));
}

TEST_F(ChecksumTest, FileChunked63) {
  EXPECT_EQ(0xab,         calculate_int(mtx::checksum::algorithm_e::crc8_atm,               0, 63));
  EXPECT_EQ(0x18fe,       calculate_int(mtx::checksum::algorithm_e::crc16_ansi,             0, 63));
  EXPECT_EQ(0x218f,       calculate_int(mtx::checksum::algorithm_e::crc16_ccitt,            0, 63));
  EXPECT_EQ(0x5a0a3951,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee,    0xffffffff, 63));
  EXPECT_EQ(0x88c5b46f,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee_le, 0xffffffff, 63));
  EXPECT_EQ(*m_data_md5, *calculate_bin(mtx::checksum::algorithm_e::md5,                       63));
}

TEST_F(ChecksumTest, FileChunked64) {
  EXPECT_EQ(0xab,         calculate_int(mtx::checksum::algorithm_e::crc8_atm,               0, 64));
  EXPECT_EQ(0x18fe,       calculate_int(mtx::checksum::algorithm_e::crc16_ansi,             0, 64));
  EXPECT_EQ(0x218f,       calculate_int(mtx::checksum::algorithm_e::crc16_ccitt,            0, 64));
  EXPECT_EQ(0x5a0a3951,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee,    0xffffffff, 64));
  EXPECT_EQ(0x88c5b46f,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee_le, 0xffffffff, 64));
  EXPECT_EQ(*m_data_md5, *calculate_bin(mtx::checksum::algorithm_e::md5,                       64));
}

TEST_F(ChecksumTest, FileChunked65) {
  EXPECT_EQ(0xab,         calculate_int(mtx::checksum::algorithm_e::crc8_atm,               0, 65));
  EXPECT_EQ(0x18fe,       calculate_int(mtx::checksum::algorithm_e::crc16_ansi,             0, 65));
  EXPECT_EQ(0x218f,       calculate_int(mtx::checksum::algorithm_e::crc16_ccitt,            0, 65));
  EXPECT_EQ(0x5a0a3951,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee,    0xffffffff, 65));
  EXPECT_EQ(0x88c5b46f,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee_le, 0xffffffff, 65));
  EXPECT_EQ(*m_data_md5, *calculate_bin(mtx::checksum::algorithm_e::md5,                       65));
}

TEST_F(ChecksumTest, FileChunked1000) {
  EXPECT_EQ(0xab,         calculate_int(mtx::checksum::algorithm_e::crc8_atm,               0, 1000));
  EXPECT_EQ(0x18fe,       calculate_int(mtx::checksum::algorithm_e::crc16_ansi,             0, 1000));
  EXPECT_EQ(0x218f,       calculate_int(mtx::checksum::algorithm_e::crc16_ccitt,            0, 1000));
  EXPECT_EQ(0x5a0a3951,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee,    0xffffffff, 1000));
  EXPECT_EQ(0x88c5b46f,   calculate_int(mtx::checksum::algorithm_e::crc32_ieee_le, 0xffffffff, 1000));
  EXPECT_EQ(*m_data_md5, *calculate_bin(mtx::checksum::algorithm_e::md5,                       1000));
}

}