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

   1 //===--- CloexecAcceptCheck.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 "CloexecAcceptCheck.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 CloexecAcceptCheck::registerMatchers(MatchFinder *Finder) {
  18   auto SockAddrPointerType =
  19       hasType(pointsTo(recordDecl(isStruct(), hasName("sockaddr"))));
  20   auto SockLenPointerType = hasType(pointsTo(namedDecl(hasName("socklen_t"))));
  21
  22   registerMatchersImpl(Finder,
  23                        functionDecl(returns(isInteger()), hasName("accept"),
  24                                     hasParameter(0, hasType(isInteger())),
  25                                     hasParameter(1, SockAddrPointerType),
  26                                     hasParameter(2, SockLenPointerType)));
  27 }
  28
  29 void CloexecAcceptCheck::check(const MatchFinder::MatchResult &Result) {
  30   std::string ReplacementText =
  31       (Twine("accept4(") + getSpellingArg(Result, 0) + ", " +
  32        getSpellingArg(Result, 1) + ", " + getSpellingArg(Result, 2) +
  33        ", SOCK_CLOEXEC)")
  34           .str();
  35
  36   replaceFunc(
  37       Result,
  38       "prefer accept4() to accept() because accept4() allows SOCK_CLOEXEC",
  39       ReplacementText);
  40 }
  41
  42 } // namespace clang::tidy::android