Inside Bitcoin

1// Copyright (c) 2018 The LevelDB Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file. See the AUTHORS file for names of contributors.
4
5#include <limits>
6#include <string>
7
8#include "leveldb/slice.h"
9#include "util/logging.h"
10#include "util/testharness.h"
11
12namespace leveldb {
13
14class Logging {};
15
16TEST(Logging, NumberToString) {
17  ASSERT_EQ("0", NumberToString(0));
18  ASSERT_EQ("1", NumberToString(1));
19  ASSERT_EQ("9", NumberToString(9));
20
21  ASSERT_EQ("10", NumberToString(10));
22  ASSERT_EQ("11", NumberToString(11));
23  ASSERT_EQ("19", NumberToString(19));
24  ASSERT_EQ("99", NumberToString(99));
25
26  ASSERT_EQ("100", NumberToString(100));
27  ASSERT_EQ("109", NumberToString(109));
28  ASSERT_EQ("190", NumberToString(190));
29  ASSERT_EQ("123", NumberToString(123));
30  ASSERT_EQ("12345678", NumberToString(12345678));
31
32  static_assert(std::numeric_limits<uint64_t>::max() == 18446744073709551615U,
33                "Test consistency check");
34  ASSERT_EQ("18446744073709551000", NumberToString(18446744073709551000U));
35  ASSERT_EQ("18446744073709551600", NumberToString(18446744073709551600U));
36  ASSERT_EQ("18446744073709551610", NumberToString(18446744073709551610U));
37  ASSERT_EQ("18446744073709551614", NumberToString(18446744073709551614U));
38  ASSERT_EQ("18446744073709551615", NumberToString(18446744073709551615U));
39}
40
41void ConsumeDecimalNumberRoundtripTest(uint64_t number,
42                                       const std::string& padding = "") {
43  std::string decimal_number = NumberToString(number);
44  std::string input_string = decimal_number + padding;
45  Slice input(input_string);
46  Slice output = input;
47  uint64_t result;
48  ASSERT_TRUE(ConsumeDecimalNumber(&output, &result));
49  ASSERT_EQ(number, result);
50  ASSERT_EQ(decimal_number.size(), output.data() - input.data());
51  ASSERT_EQ(padding.size(), output.size());
52}
53
54TEST(Logging, ConsumeDecimalNumberRoundtrip) {
55  ConsumeDecimalNumberRoundtripTest(0);
56  ConsumeDecimalNumberRoundtripTest(1);
57  ConsumeDecimalNumberRoundtripTest(9);
58
59  ConsumeDecimalNumberRoundtripTest(10);
60  ConsumeDecimalNumberRoundtripTest(11);
61  ConsumeDecimalNumberRoundtripTest(19);
62  ConsumeDecimalNumberRoundtripTest(99);
63
64  ConsumeDecimalNumberRoundtripTest(100);
65  ConsumeDecimalNumberRoundtripTest(109);
66  ConsumeDecimalNumberRoundtripTest(190);
67  ConsumeDecimalNumberRoundtripTest(123);
68  ASSERT_EQ("12345678", NumberToString(12345678));
69
70  for (uint64_t i = 0; i < 100; ++i) {
71    uint64_t large_number = std::numeric_limits<uint64_t>::max() - i;
72    ConsumeDecimalNumberRoundtripTest(large_number);
73  }
74}
75
76TEST(Logging, ConsumeDecimalNumberRoundtripWithPadding) {
77  ConsumeDecimalNumberRoundtripTest(0, " ");
78  ConsumeDecimalNumberRoundtripTest(1, "abc");
79  ConsumeDecimalNumberRoundtripTest(9, "x");
80
81  ConsumeDecimalNumberRoundtripTest(10, "_");
82  ConsumeDecimalNumberRoundtripTest(11, std::string("\0\0\0", 3));
83  ConsumeDecimalNumberRoundtripTest(19, "abc");
84  ConsumeDecimalNumberRoundtripTest(99, "padding");
85
86  ConsumeDecimalNumberRoundtripTest(100, " ");
87
88  for (uint64_t i = 0; i < 100; ++i) {
89    uint64_t large_number = std::numeric_limits<uint64_t>::max() - i;
90    ConsumeDecimalNumberRoundtripTest(large_number, "pad");
91  }
92}
93
94void ConsumeDecimalNumberOverflowTest(const std::string& input_string) {
95  Slice input(input_string);
96  Slice output = input;
97  uint64_t result;
98  ASSERT_EQ(false, ConsumeDecimalNumber(&output, &result));
99}
100
101TEST(Logging, ConsumeDecimalNumberOverflow) {
102  static_assert(std::numeric_limits<uint64_t>::max() == 18446744073709551615U,
103                "Test consistency check");
104  ConsumeDecimalNumberOverflowTest("18446744073709551616");
105  ConsumeDecimalNumberOverflowTest("18446744073709551617");
106  ConsumeDecimalNumberOverflowTest("18446744073709551618");
107  ConsumeDecimalNumberOverflowTest("18446744073709551619");
108  ConsumeDecimalNumberOverflowTest("18446744073709551620");
109  ConsumeDecimalNumberOverflowTest("18446744073709551621");
110  ConsumeDecimalNumberOverflowTest("18446744073709551622");
111  ConsumeDecimalNumberOverflowTest("18446744073709551623");
112  ConsumeDecimalNumberOverflowTest("18446744073709551624");
113  ConsumeDecimalNumberOverflowTest("18446744073709551625");
114  ConsumeDecimalNumberOverflowTest("18446744073709551626");
115
116  ConsumeDecimalNumberOverflowTest("18446744073709551700");
117
118  ConsumeDecimalNumberOverflowTest("99999999999999999999");
119}
120
121void ConsumeDecimalNumberNoDigitsTest(const std::string& input_string) {
122  Slice input(input_string);
123  Slice output = input;
124  uint64_t result;
125  ASSERT_EQ(false, ConsumeDecimalNumber(&output, &result));
126  ASSERT_EQ(input.data(), output.data());
127  ASSERT_EQ(input.size(), output.size());
128}
129
130TEST(Logging, ConsumeDecimalNumberNoDigits) {
131  ConsumeDecimalNumberNoDigitsTest("");
132  ConsumeDecimalNumberNoDigitsTest(" ");
133  ConsumeDecimalNumberNoDigitsTest("a");
134  ConsumeDecimalNumberNoDigitsTest(" 123");
135  ConsumeDecimalNumberNoDigitsTest("a123");
136  ConsumeDecimalNumberNoDigitsTest(std::string("\000123", 4));
137  ConsumeDecimalNumberNoDigitsTest(std::string("\177123", 4));
138  ConsumeDecimalNumberNoDigitsTest(std::string("\377123", 4));
139}
140
141}  // namespace leveldb
142
143int main(int argc, char** argv) { return leveldb::test::RunAllTests(); }
144