tor-browser

The Tor Browser
git clone https://git.dasho.dev/tor-browser.git
Log | Files | Refs | README | LICENSE

PolicyTokenizer.cpp (2289B)


      1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
      2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
      3 /* This Source Code Form is subject to the terms of the Mozilla Public
      4 * License, v. 2.0. If a copy of the MPL was not distributed with this
      5 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
      6 
      7 #include "PolicyTokenizer.h"
      8 
      9 #include "mozilla/Logging.h"
     10 
     11 static mozilla::LogModule* GetPolicyTokenizerLog() {
     12  static mozilla::LazyLogModule gPolicyTokenizerPRLog("PolicyTokenizer");
     13  return gPolicyTokenizerPRLog;
     14 }
     15 
     16 #define POLICYTOKENIZERLOG(args) \
     17  MOZ_LOG(GetPolicyTokenizerLog(), mozilla::LogLevel::Debug, args)
     18 
     19 static const char16_t SEMICOL = ';';
     20 
     21 PolicyTokenizer::PolicyTokenizer(const char16_t* aStart, const char16_t* aEnd)
     22    : mCurChar(aStart), mEndChar(aEnd) {
     23  POLICYTOKENIZERLOG(("PolicyTokenizer::PolicyTokenizer"));
     24 }
     25 
     26 PolicyTokenizer::~PolicyTokenizer() {
     27  POLICYTOKENIZERLOG(("PolicyTokenizer::~PolicyTokenizer"));
     28 }
     29 
     30 void PolicyTokenizer::generateNextToken() {
     31  skipWhiteSpaceAndSemicolon();
     32  MOZ_ASSERT(mCurToken.Length() == 0);
     33  const char16_t* const start = mCurChar;
     34  while (!atEnd() && !nsContentUtils::IsHTMLWhitespace(*mCurChar) &&
     35         *mCurChar != SEMICOL) {
     36    mCurChar++;
     37  }
     38  if (start != mCurChar) {
     39    mCurToken.Append(start, mCurChar - start);
     40  }
     41  POLICYTOKENIZERLOG(("PolicyTokenizer::generateNextToken: %s",
     42                      NS_ConvertUTF16toUTF8(mCurToken).get()));
     43 }
     44 
     45 void PolicyTokenizer::generateTokens(policyTokens& outTokens) {
     46  POLICYTOKENIZERLOG(("PolicyTokenizer::generateTokens"));
     47 
     48  // dirAndSrcs holds one set of [ name, src, src, src, ... ]
     49  nsTArray<nsString> dirAndSrcs;
     50 
     51  while (!atEnd()) {
     52    generateNextToken();
     53    dirAndSrcs.AppendElement(mCurToken);
     54    skipWhiteSpace();
     55    if (atEnd() || accept(SEMICOL)) {
     56      outTokens.AppendElement(std::move(dirAndSrcs));
     57      dirAndSrcs.ClearAndRetainStorage();
     58    }
     59  }
     60 }
     61 
     62 void PolicyTokenizer::tokenizePolicy(const nsAString& aPolicyString,
     63                                     policyTokens& outTokens) {
     64  POLICYTOKENIZERLOG(("PolicyTokenizer::tokenizePolicy"));
     65 
     66  PolicyTokenizer tokenizer(aPolicyString.BeginReading(),
     67                            aPolicyString.EndReading());
     68 
     69  tokenizer.generateTokens(outTokens);
     70 }