The Tor Browser: netwerk/streamconv/converters/nsTXTToHTMLConv.cpp@deefc01c0e14 (annotated)

netwerk/streamconv/converters/nsTXTToHTMLConv.cpp@deefc01c0e14 (annotated)

netwerk/streamconv/converters/nsTXTToHTMLConv.cpp

Thu, 15 Jan 2015 21:03:48 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Thu, 15 Jan 2015 21:03:48 +0100
branch: TOR_BUG_9701
changeset 11: deefc01c0e14
permissions: -rw-r--r--

Integrate friendly tips from Tor colleagues to make (or not) 4.5 alpha 3;
This includes removal of overloaded (but unused) methods, and addition of
a overlooked call to DataStruct::SetData(nsISupports, uint32_t, bool.)

 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
 /* This Source Code Form is subject to the terms of the Mozilla Public
  * License, v. 2.0. If a copy of the MPL was not distributed with this
  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
 #include "nsTXTToHTMLConv.h"
 #include "nsEscape.h"
 #include "nsStringStream.h"
 #include "nsAutoPtr.h"
 #include "nsIChannel.h"
 #include <algorithm>
 #define TOKEN_DELIMITERS MOZ_UTF16("\t\r\n ")
 // nsISupports methods
 NS_IMPL_ISUPPORTS(nsTXTToHTMLConv,
                   nsIStreamConverter,
                   nsITXTToHTMLConv,
                   nsIRequestObserver,
                   nsIStreamListener)
 // nsIStreamConverter methods
 NS_IMETHODIMP
 nsTXTToHTMLConv::Convert(nsIInputStream *aFromStream,
                          const char *aFromType, const char *aToType,
                          nsISupports *aCtxt, nsIInputStream * *_retval)
 {
     return NS_ERROR_NOT_IMPLEMENTED;
 }
 NS_IMETHODIMP
 nsTXTToHTMLConv::AsyncConvertData(const char *aFromType,
                                   const char *aToType,
                                   nsIStreamListener *aListener,
                                   nsISupports *aCtxt)
 {
     NS_ASSERTION(aListener, "null pointer");
     mListener = aListener;
     return NS_OK;
 }
 // nsIRequestObserver methods
 NS_IMETHODIMP
 nsTXTToHTMLConv::OnStartRequest(nsIRequest* request, nsISupports *aContext)
 {
     mBuffer.AssignLiteral("<html>\n<head><title>");
     mBuffer.Append(mPageTitle);
     mBuffer.AppendLiteral("</title></head>\n<body>\n");
     if (mPreFormatHTML) {     // Use <pre> tags
         mBuffer.AppendLiteral("<pre>\n");
     }
     // Push mBuffer to the listener now, so the initial HTML will not
     // be parsed in OnDataAvailable().
     nsCOMPtr<nsIChannel> channel = do_QueryInterface(request);
     if (channel)
         channel->SetContentType(NS_LITERAL_CSTRING("text/html"));
     // else, assume there is a channel somewhere that knows what it is doing!
     nsresult rv = mListener->OnStartRequest(request, aContext);
     if (NS_FAILED(rv)) return rv;
     // The request may have been canceled, and if that happens, we want to
     // suppress calls to OnDataAvailable.
     request->GetStatus(&rv);
     if (NS_FAILED(rv)) return rv;
     nsCOMPtr<nsIInputStream> inputData;
     rv = NS_NewStringInputStream(getter_AddRefs(inputData), mBuffer);
     if (NS_FAILED(rv)) return rv;
     rv = mListener->OnDataAvailable(request, aContext,
                                     inputData, 0, mBuffer.Length());
     if (NS_FAILED(rv)) return rv;
     mBuffer.Truncate();
     return rv;
 }
 NS_IMETHODIMP
 nsTXTToHTMLConv::OnStopRequest(nsIRequest* request, nsISupports *aContext,
                                nsresult aStatus)
 {
     nsresult rv = NS_OK;
     if (mToken) {
         // we still have an outstanding token
         NS_ASSERTION(mToken->prepend,
                      "Non prepending tokens should be handled in "
                      "OnDataAvailable. There should only be a single "
                      "prepending token left to be processed.");
         (void)CatHTML(0, mBuffer.Length());
     }
     if (mPreFormatHTML) {
         mBuffer.AppendLiteral("</pre>\n");
     }
     mBuffer.AppendLiteral("\n</body></html>");
     nsCOMPtr<nsIInputStream> inputData;
     rv = NS_NewStringInputStream(getter_AddRefs(inputData), mBuffer);
     if (NS_FAILED(rv)) return rv;
     rv = mListener->OnDataAvailable(request, aContext,
                                     inputData, 0, mBuffer.Length());
     if (NS_FAILED(rv)) return rv;
     return mListener->OnStopRequest(request, aContext, aStatus);
 }
 // nsITXTToHTMLConv methods
 NS_IMETHODIMP
 nsTXTToHTMLConv::SetTitle(const char16_t *aTitle)
 {
     mPageTitle.Assign(aTitle);
     return NS_OK;
 }
 NS_IMETHODIMP
 nsTXTToHTMLConv::PreFormatHTML(bool value)
 {
     mPreFormatHTML = value;
     return NS_OK;
 }
 // nsIStreamListener method
 NS_IMETHODIMP
 nsTXTToHTMLConv::OnDataAvailable(nsIRequest* request, nsISupports *aContext,
                                  nsIInputStream *aInStream,
                                  uint64_t aOffset, uint32_t aCount)
 {
     nsresult rv = NS_OK;
     nsString pushBuffer;
     uint32_t amtRead = 0;
     nsAutoArrayPtr<char> buffer(new char[aCount+1]);
     if (!buffer) return NS_ERROR_OUT_OF_MEMORY;
     do {
         uint32_t read = 0;
         // XXX readSegments, to avoid the first copy?
         rv = aInStream->Read(buffer, aCount-amtRead, &read);
         if (NS_FAILED(rv)) return rv;
         buffer[read] = '\0';
         // XXX charsets?? non-latin1 characters?? utf-16??
         AppendASCIItoUTF16(buffer, mBuffer);
         amtRead += read;
         int32_t front = -1, back = -1, tokenLoc = -1, cursor = 0;
         while ( (tokenLoc = FindToken(cursor, &mToken)) > -1) {
             if (mToken->prepend) {
                 front = mBuffer.RFindCharInSet(TOKEN_DELIMITERS, tokenLoc);
                 front++;
                 back = mBuffer.FindCharInSet(TOKEN_DELIMITERS, tokenLoc);
             } else {
                 front = tokenLoc;
                 back = front + mToken->token.Length();
             }
             if (back == -1) {
                 // didn't find an ending, buffer up.
                 mBuffer.Left(pushBuffer, front);
                 cursor = front;
                 break;
             }
             // found the end of the token.
             cursor = CatHTML(front, back);
         }
         int32_t end = mBuffer.RFind(TOKEN_DELIMITERS, mBuffer.Length());
         mBuffer.Left(pushBuffer, std::max(cursor, end));
         mBuffer.Cut(0, std::max(cursor, end));
         cursor = 0;
         if (!pushBuffer.IsEmpty()) {
             nsCOMPtr<nsIInputStream> inputData;
             rv = NS_NewStringInputStream(getter_AddRefs(inputData), pushBuffer);
             if (NS_FAILED(rv))
                 return rv;
             rv = mListener->OnDataAvailable(request, aContext,
                                             inputData, 0, pushBuffer.Length());
             if (NS_FAILED(rv))
                 return rv;
         }
     } while (amtRead < aCount);
     return rv;
 }
 // nsTXTToHTMLConv methods
 nsTXTToHTMLConv::nsTXTToHTMLConv()
 {
     mToken = nullptr;
     mPreFormatHTML = false;
 }
 nsTXTToHTMLConv::~nsTXTToHTMLConv()
 {
     mTokens.Clear();
 }
 nsresult
 nsTXTToHTMLConv::Init()
 {
     nsresult rv = NS_OK;
     // build up the list of tokens to handle
     convToken *token = new convToken;
     if (!token) return NS_ERROR_OUT_OF_MEMORY;
     token->prepend = false;
     token->token.Assign(char16_t('<'));
     token->modText.AssignLiteral("&lt;");
     mTokens.AppendElement(token);
     token = new convToken;
     if (!token) return NS_ERROR_OUT_OF_MEMORY;
     token->prepend = false;
     token->token.Assign(char16_t('>'));
     token->modText.AssignLiteral("&gt;");
     mTokens.AppendElement(token);
     token = new convToken;
     if (!token) return NS_ERROR_OUT_OF_MEMORY;
     token->prepend = false;
     token->token.Assign(char16_t('&'));
     token->modText.AssignLiteral("&amp;");
     mTokens.AppendElement(token);
     token = new convToken;
     if (!token) return NS_ERROR_OUT_OF_MEMORY;
     token->prepend = true;
     token->token.AssignLiteral("http://"); // XXX need to iterate through all protos
     mTokens.AppendElement(token);
     token = new convToken;
     if (!token) return NS_ERROR_OUT_OF_MEMORY;
     token->prepend = true;
     token->token.Assign(char16_t('@'));
     token->modText.AssignLiteral("mailto:");
     mTokens.AppendElement(token);
     return rv;
 }
 int32_t
 nsTXTToHTMLConv::FindToken(int32_t cursor, convToken* *_retval)
 {
     int32_t loc = -1, firstToken = mBuffer.Length();
     int8_t token = -1;
     for (uint8_t i=0; i < mTokens.Length(); i++) {
         loc = mBuffer.Find(mTokens[i]->token, cursor);
         if (loc != -1)
             if (loc < firstToken) {
                 firstToken = loc;
                 token = i;
             }
     }
     if (token == -1)
         return -1;
     *_retval = mTokens[token];
     return firstToken;
 }
 int32_t
 nsTXTToHTMLConv::CatHTML(int32_t front, int32_t back)
 {
     int32_t cursor = 0;
     int32_t modLen = mToken->modText.Length();
     if (!mToken->prepend) {
         // replace the entire token (from delimiter to delimiter)
         mBuffer.Cut(front, back - front);
         mBuffer.Insert(mToken->modText, front);
         cursor = front+modLen;
     } else {
         nsString linkText;
         // href is implied
         mBuffer.Mid(linkText, front, back-front);
         mBuffer.Insert(NS_LITERAL_STRING("<a href=\""), front);
         cursor += front+9;
         if (modLen) {
             mBuffer.Insert(mToken->modText, cursor);
             cursor += modLen;
         }
         NS_ConvertUTF16toUTF8 linkTextUTF8(linkText);
         nsCString escaped;
         if (NS_EscapeURL(linkTextUTF8.Data(), linkTextUTF8.Length(), esc_Minimal, escaped)) {
             mBuffer.Cut(cursor, back - front);
             CopyUTF8toUTF16(escaped, linkText);
             mBuffer.Insert(linkText, cursor);
             back = front + linkText.Length();
         }
         cursor += back-front;
         mBuffer.Insert(NS_LITERAL_STRING("\">"), cursor);
         cursor += 2;
         mBuffer.Insert(linkText, cursor);
         cursor += linkText.Length();
         mBuffer.Insert(NS_LITERAL_STRING("</a>"), cursor);
         cursor += 4;
     }
     mToken = nullptr; // indicates completeness
     return cursor;
 }

The Tor Browser / annotate

netwerk/streamconv/converters/nsTXTToHTMLConv.cpp@deefc01c0e14 (annotated)

netwerk/streamconv/converters/nsTXTToHTMLConv.cpp