/* Copyright (c) 2015 Brian R. Bondy. Distributed under the MPL2 license.
 * 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/. */

#ifndef FILTER_H_
#define FILTER_H_
#include <stdint.h>
#include <string.h>
#include "./base.h"
#include "./context_domain.h"

class BloomFilter;
template <typename T> class HashSet;

enum FilterType {
  FTNoFilterType = 0,
  FTRegex = 01,
  FTElementHiding = 02,
  FTElementHidingException = 04,
  FTHostAnchored = 010,
  FTLeftAnchored = 020,
  FTRightAnchored = 040,
  FTComment = 0100,
  FTException = 0200,
  FTEmpty = 0400,
  FTHostOnly = 01000,
  // E.g. example.org$$script[data-src="banner"] to delete
  // script element with matching attr
  FTHTMLFiltering = 02000,
  FTListTypesMask = FTException|FTElementHiding|
    FTElementHidingException|FTEmpty|FTComment|
    FTHTMLFiltering,
};

enum FilterOption {
  FONoFilterOption = 0,
  FOScript = 01,
  FOImage = 02,
  FOStylesheet = 04,
  FOObject = 010,
  FOXmlHttpRequest = 020,
  FOObjectSubrequest = 040,
  FOSubdocument = 0100,
  FODocument = 0200,
  FOOther = 0400,
  FOXBL = 01000,
  FOCollapse = 02000,
  FODoNotTrack = 04000,
  FOElemHide = 010000,
  // Used internally only, do not use
  FOThirdParty = 020000,
  // Used internally only, do not use
  FONotThirdParty = 040000,
  // Not supported, but we will ignore these rules
  FOPing = 0100000,
  // Not supported, but we will ignore these rules
  FOPopup = 0200000,
  // This is only used by uBlock and currently all instances are 1x1
  // transparent gif which we already do for images
  FORedirect = 0400000,
  // Parse CSPs but consider them unsupported
  FOCSP = 01000000,
  FOFont = 02000000,
  FOMedia = 04000000,
  FOWebRTC = 010000000,
  FOGenericHide = 020000000,
  FOGenericBlock = 040000000,
  // Used by Adguard, purpose unknown, ignore
  FOEmpty = 0100000000,
  FOWebsocket = 0200000000,
  // important means to ignore all exception filters (those prefixed with @@).
  FOImportant = 0400000000,
  // Cancel the request instead of using a 200 OK response
  FOExplicitCancel = 01000000000,

  FOUnknown = 04000000000,
  FOResourcesOnly = FOScript|FOImage|FOStylesheet|FOObject|FOXmlHttpRequest|
    FOObjectSubrequest|FOSubdocument|FODocument|FOOther|FOXBL|FOFont|FOMedia|
    FOWebRTC|FOWebsocket|FOPing,
  FOUnsupportedSoSkipCheck = FOPopup|FOCSP|FOElemHide|FOGenericHide|
    FOGenericBlock|FOEmpty|FOUnknown,
  // Non matching related filters, alters behavior
  BehavioralFilterOnly = FORedirect|FOImportant|FOExplicitCancel|
    FOThirdParty|FONotThirdParty
};

class Filter {
friend class AdBlockClient;
 public:
  Filter();
  Filter(const Filter &other);
  Filter(const char * data, int dataLen, char *domainList = nullptr,
      const char * host = nullptr, int hostLen = -1,
      char *tag = nullptr, int tagLen = 0);

  Filter(FilterType filterType, FilterOption filterOption,
         FilterOption antiFilterOption,
         const char * data, int dataLen,
         char *domainList = nullptr,
         const char * host = nullptr, int hostLen = -1,
         char *tag = nullptr, int tagLen = 0);

  ~Filter();

  // Swaps the data members for 'this' and the passed in filter
  void swapData(Filter *f);

  // Checks to see if any filter matches the input but does not match
  // any exception rule You may want to call the first overload to be
  // slighly more efficient
  bool matches(const char *input, int inputLen,
      FilterOption contextOption = FONoFilterOption,
      const char *contextDomain = nullptr,
      BloomFilter *inputBloomFilter = nullptr,
      const char *inputHost = nullptr, int inputHostLen = 0);
  bool matches(const char *input, FilterOption contextOption = FONoFilterOption,
      const char *contextDomain = nullptr,
      BloomFilter *inputBloomFilter = nullptr,
      const char *inputHost = nullptr, int inputHostLen = 0);

  // Nothing needs to be updated when a filter is added multiple times
  void Update(const Filter &) {}
  bool hasUnsupportedOptions() const;

  // Checks to see if the filter options match for the passed in data
  bool matchesOptions(const char *input, FilterOption contextOption,
      const char *contextDomain = nullptr);

  void parseOptions(const char *input);

  // Checks to see if the specified context domain is in the
  // domain (or antiDmomain) list.
  bool containsDomain(const char* contextDomain, size_t contextDomainLen,
      bool anti = false) const;
  // Returns true if the filter is composed of only domains and no anti domains
  // Note that the set of all domain and anti-domain rules are not mutually
  // exclusive.  One xapmle is:
  // domain=example.com|~foo.example.com restricts the filter to the example.com
  // domain with the exception of "foo.example.com" subdomain.
  bool isDomainOnlyFilter();
  // Returns true if the filter is composed of only anti-domains and no domains
  bool isAntiDomainOnlyFilter();
  uint32_t getDomainCount(bool anti = false);

  uint64_t hash() const;
  uint64_t GetHash() const {
    return hash();
  }

  bool operator==(const Filter &rhs) const {
    /*
     if (filterType != rhs.filterType || filterOption != rhs.filterOption ||
         antiFilterOption != rhs.antiFilterOption) {
      return false;
    }
    */

    int hostLen = 0;
    if (host) {
      hostLen = this->hostLen == -1 ?
        static_cast<int>(strlen(host)) : this->hostLen;
    }
    int rhsHostLen = 0;
    if (rhs.host) {
      rhsHostLen = rhs.hostLen == -1 ?
        static_cast<int>(strlen(rhs.host)) : rhs.hostLen;
    }

    if (hostLen != rhsHostLen) {
      return false;
    }

    return !memcmp(host, rhs.host, hostLen);
  }

  bool operator!=(const Filter &rhs) const {
    return !(*this == rhs);
  }

  uint32_t Serialize(char *buffer);
  uint32_t Deserialize(char *buffer, uint32_t bufferSize);

  // Holds true if the filter should not free memory because for example it
  // was loaded from a large buffer somewhere else via the serialize and
  // deserialize functions.
  bool borrowed_data;

  FilterType filterType;
  FilterOption filterOption;
  FilterOption antiFilterOption;

  // The text of the filter list rule, as it appeared before being parsed.
  char *ruleDefinition;

  char *data;
  int dataLen;
  char *domainList;
  // A filter tag is used for identifying and tagally including
  // certain filters in Brave.
  char *tag;
  int tagLen;
  char *host;
  int hostLen;
  HashSet<ContextDomain>* domains;
  HashSet<ContextDomain>* antiDomains;
  bool domainsParsed;

 protected:
  // Fills |domains| and |antiDomains| sets
  void parseDomains(const char *domainList);
  bool contextDomainMatchesFilter(const char *contextDomain);

  // Parses a single option
  void parseOption(const char *input, int len);
};

bool isThirdPartyHost(const char *baseContextHost,
    int baseContextHostLen,
    const char *testHost,
    int testHostLen);

static inline bool isEndOfLine(char c) {
  return c == '\r' || c == '\n';
}

#endif  // FILTER_H_