1 /* 2 * Copyright (C) 2010 Adam Barth. 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 6 * are met: 7 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY 14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR 17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26 #include "config.h" 27 #include "HTMLSourceTracker.h" 28 29 namespace WebCore { 30 31 HTMLSourceTracker::HTMLSourceTracker() 32 { 33 } 34 35 void HTMLSourceTracker::start(const HTMLInputStream& input, HTMLToken& token) 36 { 37 m_sourceFromPreviousSegments = token.type() == HTMLToken::Uninitialized ? String() : m_sourceFromPreviousSegments + m_source.toString(); 38 m_source = input.current(); 39 token.setBaseOffset(input.current().numberOfCharactersConsumed() - m_sourceFromPreviousSegments.length()); 40 } 41 42 void HTMLSourceTracker::end(const HTMLInputStream& input, HTMLToken& token) 43 { 44 m_cachedSourceForToken = String(); 45 // FIXME: This work should really be done by the HTMLTokenizer. 46 token.end(input.current().numberOfCharactersConsumed()); 47 } 48 49 String HTMLSourceTracker::sourceForToken(const HTMLToken& token) 50 { 51 if (token.type() == HTMLToken::EndOfFile) 52 return String(); // Hides the null character we use to mark the end of file. 53 54 if (!m_cachedSourceForToken.isEmpty()) 55 return m_cachedSourceForToken; 56 57 ASSERT(!token.startIndex()); 58 UChar* data = 0; 59 int length = token.endIndex() - token.startIndex() - m_sourceFromPreviousSegments.length(); 60 String source = String::createUninitialized(length, data); 61 for (int i = 0; i < length; ++i) { 62 data[i] = *m_source; 63 m_source.advance(); 64 } 65 m_cachedSourceForToken = m_sourceFromPreviousSegments + source; 66 return m_cachedSourceForToken; 67 } 68 69 } 70
