| // Copyright 2017 The PDFium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| |
| #include "core/fxcrt/cfx_bitstream.h" |
| |
| #include <limits> |
| |
| #include "core/fxcrt/fx_memory.h" |
| #include "core/fxcrt/fx_system.h" |
| #include "third_party/base/check_op.h" |
| |
| CFX_BitStream::CFX_BitStream(pdfium::span<const uint8_t> pData) |
| : m_BitSize(pData.size() * 8), m_pData(pData) { |
| CHECK_LE(m_pData.size(), std::numeric_limits<size_t>::max() / 8); |
| } |
| |
| CFX_BitStream::~CFX_BitStream() = default; |
| |
| void CFX_BitStream::ByteAlign() { |
| m_BitPos = FxAlignToBoundary<8>(m_BitPos); |
| } |
| |
| uint32_t CFX_BitStream::GetBits(uint32_t nBits) { |
| DCHECK(nBits > 0); |
| DCHECK(nBits <= 32); |
| if (nBits > m_BitSize || m_BitPos > m_BitSize - nBits) |
| return 0; |
| |
| const uint32_t bit_pos = m_BitPos % 8; |
| size_t byte_pos = m_BitPos / 8; |
| uint8_t current_byte = m_pData[byte_pos]; |
| |
| if (nBits == 1) { |
| uint32_t bit = (current_byte & (1 << (7 - bit_pos))) ? 1 : 0; |
| m_BitPos++; |
| return bit; |
| } |
| |
| uint32_t bit_left = nBits; |
| uint32_t result = 0; |
| if (bit_pos) { |
| uint32_t bits_readable = 8 - bit_pos; |
| if (bits_readable >= bit_left) { |
| result = (current_byte & (0xff >> bit_pos)) >> (bits_readable - bit_left); |
| m_BitPos += bit_left; |
| return result; |
| } |
| bit_left -= bits_readable; |
| result = (current_byte & ((1 << bits_readable) - 1)) << bit_left; |
| ++byte_pos; |
| } |
| while (bit_left >= 8) { |
| bit_left -= 8; |
| result |= m_pData[byte_pos++] << bit_left; |
| } |
| if (bit_left) |
| result |= m_pData[byte_pos] >> (8 - bit_left); |
| m_BitPos += nBits; |
| return result; |
| } |