1 //===--- ForbiddenSubclassingCheck.cpp - clang-tidy -----------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "ForbiddenSubclassingCheck.h"
10 #include "clang/AST/ASTContext.h"
11 #include "clang/ASTMatchers/ASTMatchFinder.h"
12 #include "llvm/ADT/Hashing.h"
13 #include "llvm/ADT/SmallVector.h"
14 #include "../utils/OptionsUtils.h"
15
16 using namespace clang::ast_matchers;
17
18 namespace clang {
19 namespace tidy {
20 namespace objc {
21
22 namespace {
23
24 constexpr char DefaultForbiddenSuperClassNames[] =
25 "ABNewPersonViewController;"
26 "ABPeoplePickerNavigationController;"
27 "ABPersonViewController;"
28 "ABUnknownPersonViewController;"
29 "NSHashTable;"
30 "NSMapTable;"
31 "NSPointerArray;"
32 "NSPointerFunctions;"
33 "NSTimer;"
34 "UIActionSheet;"
35 "UIAlertView;"
36 "UIImagePickerController;"
37 "UITextInputMode;"
38 "UIWebView";
39
40 } // namespace
41
ForbiddenSubclassingCheck(StringRef Name,ClangTidyContext * Context)42 ForbiddenSubclassingCheck::ForbiddenSubclassingCheck(
43 StringRef Name,
44 ClangTidyContext *Context)
45 : ClangTidyCheck(Name, Context),
46 ForbiddenSuperClassNames(
47 utils::options::parseStringList(
48 Options.get("ClassNames", DefaultForbiddenSuperClassNames))) {
49 }
50
registerMatchers(MatchFinder * Finder)51 void ForbiddenSubclassingCheck::registerMatchers(MatchFinder *Finder) {
52 Finder->addMatcher(
53 objcInterfaceDecl(
54 isDerivedFrom(
55 objcInterfaceDecl(
56 hasAnyName(
57 std::vector<StringRef>(
58 ForbiddenSuperClassNames.begin(),
59 ForbiddenSuperClassNames.end())))
60 .bind("superclass")))
61 .bind("subclass"),
62 this);
63 }
64
check(const MatchFinder::MatchResult & Result)65 void ForbiddenSubclassingCheck::check(
66 const MatchFinder::MatchResult &Result) {
67 const auto *SubClass = Result.Nodes.getNodeAs<ObjCInterfaceDecl>(
68 "subclass");
69 assert(SubClass != nullptr);
70 const auto *SuperClass = Result.Nodes.getNodeAs<ObjCInterfaceDecl>(
71 "superclass");
72 assert(SuperClass != nullptr);
73 diag(SubClass->getLocation(),
74 "Objective-C interface %0 subclasses %1, which is not "
75 "intended to be subclassed")
76 << SubClass
77 << SuperClass;
78 }
79
storeOptions(ClangTidyOptions::OptionMap & Opts)80 void ForbiddenSubclassingCheck::storeOptions(
81 ClangTidyOptions::OptionMap &Opts) {
82 Options.store(
83 Opts,
84 "ForbiddenSuperClassNames",
85 utils::options::serializeStringList(ForbiddenSuperClassNames));
86 }
87
88 } // namespace objc
89 } // namespace tidy
90 } // namespace clang
91