build/clang-plugin/ParamTraitsEnumChecker.cpp
author Andi-Bogdan Postelnicu <bpostelnicu@mozilla.com>
Mon, 17 Feb 2020 17:23:16 +0000
changeset 514326 517de066cc6f8b6823b64b694204ee423511613b
parent 444370 538a16d495142178a73e0bdc30f100b43d2fd62b
permissions -rw-r--r--
Bug 1615826 - Reformat `clang-plugin` to LLVM standard. r=froydnj Differential Revision: https://phabricator.services.mozilla.com/D63000

/* 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 "ParamTraitsEnumChecker.h"
#include "CustomMatchers.h"

void ParamTraitsEnumChecker::registerMatchers(MatchFinder *AstMatcher) {
  AstMatcher->addMatcher(
      classTemplateSpecializationDecl(hasName("ParamTraits")).bind("decl"),
      this);
}

void ParamTraitsEnumChecker::check(const MatchFinder::MatchResult &Result) {
  const ClassTemplateSpecializationDecl *Decl =
      Result.Nodes.getNodeAs<ClassTemplateSpecializationDecl>("decl");

  for (auto &Inner : Decl->decls()) {
    if (auto *Def = dyn_cast<TypedefDecl>(Inner)) {
      QualType UnderlyingType = Def->getUnderlyingType();
      QualType CanonicalType = UnderlyingType.getCanonicalType();

      const clang::Type *TypePtr = CanonicalType.getTypePtrOrNull();
      if (!TypePtr) {
        return;
      }

      if (TypePtr->isEnumeralType()) {
        diag(Decl->getBeginLoc(),
             "Custom ParamTraits implementation for an enum type",
             DiagnosticIDs::Error);
        diag(Decl->getBeginLoc(),
             "Please use a helper class for example ContiguousEnumSerializer",
             DiagnosticIDs::Note);
      }
    }
  }
}