1/* 2 * Copyright (C) 2011 Google Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions are 6 * met: 7 * 8 * * Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * * Redistributions in binary form must reproduce the above 11 * copyright notice, this list of conditions and the following disclaimer 12 * in the documentation and/or other materials provided with the 13 * distribution. 14 * * Neither the name of Google Inc. nor the names of its 15 * contributors may be used to endorse or promote products derived from 16 * this software without specific prior written permission. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29 */ 30 31#include "config.h" 32#include "platform/SharedBufferChunkReader.h" 33 34#include "platform/SharedBuffer.h" 35 36namespace WebCore { 37 38SharedBufferChunkReader::SharedBufferChunkReader(SharedBuffer* buffer, const Vector<char>& separator) 39 : m_buffer(buffer) 40 , m_bufferPosition(0) 41 , m_segment(0) 42 , m_segmentLength(0) 43 , m_segmentIndex(0) 44 , m_reachedEndOfFile(false) 45 , m_separator(separator) 46 , m_separatorIndex(0) 47{ 48} 49 50SharedBufferChunkReader::SharedBufferChunkReader(SharedBuffer* buffer, const char* separator) 51 : m_buffer(buffer) 52 , m_bufferPosition(0) 53 , m_segment(0) 54 , m_segmentLength(0) 55 , m_segmentIndex(0) 56 , m_reachedEndOfFile(false) 57 , m_separatorIndex(0) 58{ 59 setSeparator(separator); 60} 61 62void SharedBufferChunkReader::setSeparator(const Vector<char>& separator) 63{ 64 m_separator = separator; 65} 66 67void SharedBufferChunkReader::setSeparator(const char* separator) 68{ 69 m_separator.clear(); 70 m_separator.append(separator, strlen(separator)); 71} 72 73bool SharedBufferChunkReader::nextChunk(Vector<char>& chunk, bool includeSeparator) 74{ 75 if (m_reachedEndOfFile) 76 return false; 77 78 chunk.clear(); 79 while (true) { 80 while (m_segmentIndex < m_segmentLength) { 81 char currentCharacter = m_segment[m_segmentIndex++]; 82 if (currentCharacter != m_separator[m_separatorIndex]) { 83 if (m_separatorIndex > 0) { 84 ASSERT_WITH_SECURITY_IMPLICATION(m_separatorIndex <= m_separator.size()); 85 chunk.append(m_separator.data(), m_separatorIndex); 86 m_separatorIndex = 0; 87 } 88 chunk.append(currentCharacter); 89 continue; 90 } 91 m_separatorIndex++; 92 if (m_separatorIndex == m_separator.size()) { 93 if (includeSeparator) 94 chunk.append(m_separator); 95 m_separatorIndex = 0; 96 return true; 97 } 98 } 99 100 // Read the next segment. 101 m_segmentIndex = 0; 102 m_bufferPosition += m_segmentLength; 103 m_segmentLength = m_buffer->getSomeData(m_segment, m_bufferPosition); 104 if (!m_segmentLength) { 105 m_reachedEndOfFile = true; 106 if (m_separatorIndex > 0) 107 chunk.append(m_separator.data(), m_separatorIndex); 108 return !chunk.isEmpty(); 109 } 110 } 111 ASSERT_NOT_REACHED(); 112 return false; 113} 114 115String SharedBufferChunkReader::nextChunkAsUTF8StringWithLatin1Fallback(bool includeSeparator) 116{ 117 Vector<char> data; 118 if (!nextChunk(data, includeSeparator)) 119 return String(); 120 121 return data.size() ? String::fromUTF8WithLatin1Fallback(data.data(), data.size()) : emptyString(); 122} 123 124size_t SharedBufferChunkReader::peek(Vector<char>& data, size_t requestedSize) 125{ 126 data.clear(); 127 if (requestedSize <= m_segmentLength - m_segmentIndex) { 128 data.append(m_segment + m_segmentIndex, requestedSize); 129 return requestedSize; 130 } 131 132 size_t readBytesCount = m_segmentLength - m_segmentIndex; 133 data.append(m_segment + m_segmentIndex, readBytesCount); 134 135 size_t bufferPosition = m_bufferPosition + m_segmentLength; 136 const char* segment = 0; 137 while (size_t segmentLength = m_buffer->getSomeData(segment, bufferPosition)) { 138 if (requestedSize <= readBytesCount + segmentLength) { 139 data.append(segment, requestedSize - readBytesCount); 140 readBytesCount += (requestedSize - readBytesCount); 141 break; 142 } 143 data.append(segment, segmentLength); 144 readBytesCount += segmentLength; 145 bufferPosition += segmentLength; 146 } 147 return readBytesCount; 148} 149 150} 151