| Index: core/fxcodec/codec/fx_codec_rle_unittest.cpp
|
| diff --git a/core/fxcodec/codec/fx_codec_rle_unittest.cpp b/core/fxcodec/codec/fx_codec_rle_unittest.cpp
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..3513e1309c2564ea07460ef619ffd4885e6f2604
|
| --- /dev/null
|
| +++ b/core/fxcodec/codec/fx_codec_rle_unittest.cpp
|
| @@ -0,0 +1,195 @@
|
| +// Copyright 2016 PDFium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include <stdint.h>
|
| +
|
| +#include <limits>
|
| +
|
| +#include "core/fpdfapi/parser/fpdf_parser_decode.h"
|
| +#include "core/fxcodec/codec/ccodec_basicmodule.h"
|
| +#include "core/fxcodec/fx_codec.h"
|
| +#include "testing/fx_string_testhelpers.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +
|
| +TEST(fxcodec, RLETestBadInputs) {
|
| + uint8_t src_buf[1] = {1};
|
| + uint8_t* dest_buf = nullptr;
|
| + uint32_t src_size = 4;
|
| + uint32_t dest_size = 0;
|
| +
|
| + CCodec_BasicModule* pEncoders = CCodec_ModuleMgr().GetBasicModule();
|
| + EXPECT_TRUE(pEncoders);
|
| +
|
| + // Error codes, not segvs, should callers pass us a nullptr pointer.
|
| + EXPECT_FALSE(
|
| + pEncoders->RunLengthEncode(src_buf, src_size, &dest_buf, nullptr));
|
| + EXPECT_FALSE(
|
| + pEncoders->RunLengthEncode(src_buf, src_size, nullptr, &dest_size));
|
| + EXPECT_FALSE(pEncoders->RunLengthEncode(src_buf, 0, &dest_buf, &dest_size));
|
| + EXPECT_FALSE(
|
| + pEncoders->RunLengthEncode(nullptr, src_size, &dest_buf, &dest_size));
|
| +}
|
| +
|
| +// Check length 1 input works. Check terminating character is applied.
|
| +TEST(fxcodec, RLETestShortInput) {
|
| + uint8_t src_buf[1] = {1};
|
| + uint8_t* dest_buf = nullptr;
|
| + uint32_t src_size = 1;
|
| + uint32_t dest_size = 0;
|
| +
|
| + CCodec_BasicModule* pEncoders = CCodec_ModuleMgr().GetBasicModule();
|
| + EXPECT_TRUE(pEncoders);
|
| +
|
| + EXPECT_TRUE(
|
| + pEncoders->RunLengthEncode(src_buf, src_size, &dest_buf, &dest_size));
|
| + EXPECT_EQ(3u, dest_size);
|
| + EXPECT_EQ(dest_buf[0], 0);
|
| + EXPECT_EQ(dest_buf[1], 1);
|
| + EXPECT_EQ(dest_buf[2], 128);
|
| +
|
| + FX_Free(dest_buf);
|
| +}
|
| +
|
| +// Check a few basic cases (2 matching runs in a row, matching run followed
|
| +// by a nonmatching run, and nonmatching run followed by a matching run).
|
| +TEST(fxcodec, RLETestNormalInputs) {
|
| + // Match, match
|
| + uint8_t src_buf_1[10] = {2, 2, 2, 2, 4, 4, 4, 4, 4, 4};
|
| +
|
| + // Match, nonmatch
|
| + uint8_t src_buf_2[10] = {2, 2, 2, 2, 1, 2, 3, 4, 5, 6};
|
| +
|
| + // Nonmatch, match
|
| + uint8_t src_buf_3[10] = {1, 2, 3, 4, 5, 3, 3, 3, 3, 3};
|
| +
|
| + uint32_t src_size = 10;
|
| + uint32_t dest_size = 0;
|
| + uint8_t* dest_buf = nullptr;
|
| +
|
| + CCodec_BasicModule* pEncoders = CCodec_ModuleMgr().GetBasicModule();
|
| + EXPECT_TRUE(pEncoders);
|
| +
|
| + // Case 1:
|
| + EXPECT_TRUE(
|
| + pEncoders->RunLengthEncode(src_buf_1, src_size, &dest_buf, &dest_size));
|
| + uint8_t* decoded_buf = nullptr;
|
| + uint32_t decoded_size = 0;
|
| + RunLengthDecode(dest_buf, dest_size, decoded_buf, decoded_size);
|
| + EXPECT_EQ(decoded_size, src_size);
|
| + for (uint32_t i = 0; i < src_size; i++)
|
| + EXPECT_EQ(decoded_buf[i], src_buf_1[i]) << " at " << i;
|
| + FX_Free(dest_buf);
|
| + FX_Free(decoded_buf);
|
| +
|
| + // Case 2:
|
| + dest_buf = nullptr;
|
| + dest_size = 0;
|
| + EXPECT_TRUE(
|
| + pEncoders->RunLengthEncode(src_buf_2, src_size, &dest_buf, &dest_size));
|
| + decoded_buf = nullptr;
|
| + decoded_size = 0;
|
| + RunLengthDecode(dest_buf, dest_size, decoded_buf, decoded_size);
|
| + EXPECT_EQ(decoded_size, src_size);
|
| + for (uint32_t i = 0; i < src_size; i++)
|
| + EXPECT_EQ(decoded_buf[i], src_buf_2[i]) << " at " << i;
|
| + FX_Free(dest_buf);
|
| + FX_Free(decoded_buf);
|
| +
|
| + // Case 3:
|
| + dest_buf = nullptr;
|
| + dest_size = 0;
|
| + EXPECT_TRUE(
|
| + pEncoders->RunLengthEncode(src_buf_3, src_size, &dest_buf, &dest_size));
|
| + decoded_buf = nullptr;
|
| + decoded_size = 0;
|
| + RunLengthDecode(dest_buf, dest_size, decoded_buf, decoded_size);
|
| + EXPECT_EQ(decoded_size, src_size);
|
| + for (uint32_t i = 0; i < src_size; i++)
|
| + EXPECT_EQ(decoded_buf[i], src_buf_3[i]) << " at " << i;
|
| + FX_Free(dest_buf);
|
| + FX_Free(decoded_buf);
|
| +}
|
| +
|
| +// Check that runs longer than 128 are broken up properly, both matched and
|
| +// nonmatched.
|
| +TEST(fxcodec, RLETestFullLengthInputs) {
|
| + // Match, match
|
| + uint8_t src_buf_1[260] = {1};
|
| +
|
| + // Match, nonmatch
|
| + uint8_t src_buf_2[260] = {2};
|
| + for (uint16_t i = 128; i < 260; i++)
|
| + src_buf_2[i] = (uint8_t)(i - 125);
|
| +
|
| + // Nonmatch, match
|
| + uint8_t src_buf_3[260] = {3};
|
| + for (uint8_t i = 0; i < 128; i++)
|
| + src_buf_3[i] = i;
|
| +
|
| + // Nonmatch, nonmatch
|
| + uint8_t src_buf_4[260];
|
| + for (uint16_t i = 0; i < 260; i++)
|
| + src_buf_4[i] = (uint8_t)(i);
|
| +
|
| + uint32_t src_size = 260;
|
| + uint32_t dest_size = 0;
|
| + uint8_t* dest_buf = nullptr;
|
| +
|
| + CCodec_BasicModule* pEncoders = CCodec_ModuleMgr().GetBasicModule();
|
| + EXPECT_TRUE(pEncoders);
|
| +
|
| + // Case 1:
|
| + EXPECT_TRUE(
|
| + pEncoders->RunLengthEncode(src_buf_1, src_size, &dest_buf, &dest_size));
|
| + uint8_t* decoded_buf = nullptr;
|
| + uint32_t decoded_size = 0;
|
| + RunLengthDecode(dest_buf, dest_size, decoded_buf, decoded_size);
|
| + EXPECT_EQ(decoded_size, src_size);
|
| + for (uint32_t i = 0; i < src_size; i++)
|
| + EXPECT_EQ(decoded_buf[i], src_buf_1[i]) << " at " << i;
|
| + FX_Free(dest_buf);
|
| + FX_Free(decoded_buf);
|
| +
|
| + // Case 2:
|
| + dest_buf = nullptr;
|
| + dest_size = 0;
|
| + EXPECT_TRUE(
|
| + pEncoders->RunLengthEncode(src_buf_2, src_size, &dest_buf, &dest_size));
|
| + decoded_buf = nullptr;
|
| + decoded_size = 0;
|
| + RunLengthDecode(dest_buf, dest_size, decoded_buf, decoded_size);
|
| + EXPECT_EQ(decoded_size, src_size);
|
| + for (uint32_t i = 0; i < src_size; i++)
|
| + EXPECT_EQ(decoded_buf[i], src_buf_2[i]) << " at " << i;
|
| + FX_Free(dest_buf);
|
| + FX_Free(decoded_buf);
|
| +
|
| + // Case 3:
|
| + dest_buf = nullptr;
|
| + dest_size = 0;
|
| + EXPECT_TRUE(
|
| + pEncoders->RunLengthEncode(src_buf_3, src_size, &dest_buf, &dest_size));
|
| + decoded_buf = nullptr;
|
| + decoded_size = 0;
|
| + RunLengthDecode(dest_buf, dest_size, decoded_buf, decoded_size);
|
| + EXPECT_EQ(decoded_size, src_size);
|
| + for (uint32_t i = 0; i < src_size; i++)
|
| + EXPECT_EQ(decoded_buf[i], src_buf_3[i]) << " at " << i;
|
| + FX_Free(dest_buf);
|
| + FX_Free(decoded_buf);
|
| +
|
| + // Case 4:
|
| + dest_buf = nullptr;
|
| + dest_size = 0;
|
| + EXPECT_TRUE(
|
| + pEncoders->RunLengthEncode(src_buf_4, src_size, &dest_buf, &dest_size));
|
| + decoded_buf = nullptr;
|
| + decoded_size = 0;
|
| + RunLengthDecode(dest_buf, dest_size, decoded_buf, decoded_size);
|
| + EXPECT_EQ(decoded_size, src_size);
|
| + for (uint32_t i = 0; i < src_size; i++)
|
| + EXPECT_EQ(decoded_buf[i], src_buf_4[i]) << " at " << i;
|
| + FX_Free(dest_buf);
|
| + FX_Free(decoded_buf);
|
| +}
|
|
|