clang-tools-extra/clang-tidy/android/CloexecCreatCheck.cpp

   1 //===--- CloexecCreatCheck.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 "CloexecCreatCheck.h"
  10 #include "clang/AST/ASTContext.h"
  11 #include "clang/ASTMatchers/ASTMatchFinder.h"
  12
  13 using namespace clang::ast_matchers;
  14
  15 namespace clang::tidy::android {
  16
  17 void CloexecCreatCheck::registerMatchers(MatchFinder *Finder) {
  18   auto CharPointerType = hasType(pointerType(pointee(isAnyCharacter())));
  19   auto MODETType = hasType(namedDecl(hasName("mode_t")));
  20   registerMatchersImpl(Finder,
  21                        functionDecl(isExternC(), returns(isInteger()),
  22                                     hasName("creat"),
  23                                     hasParameter(0, CharPointerType),
  24                                     hasParameter(1, MODETType)));
  25 }
  26
  27 void CloexecCreatCheck::check(const MatchFinder::MatchResult &Result) {
  28   const std::string &ReplacementText =
  29       (Twine("open (") + getSpellingArg(Result, 0) +
  30        ", O_WRONLY | O_CREAT | O_TRUNC | O_CLOEXEC, " +
  31        getSpellingArg(Result, 1) + ")")
  32           .str();
  33   replaceFunc(Result,
  34               "prefer open() to creat() because open() allows O_CLOEXEC",
  35               ReplacementText);
  36 }
  37
  38 } // namespace clang::tidy::android