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