10d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//=- CheckObjCInstMethodRetTy.cpp - Check ObjC method signatures -*- C++ -*-==//
20d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//
30d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//                     The LLVM Compiler Infrastructure
40d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//
50d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek// This file is distributed under the University of Illinois Open Source
60d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek// License. See LICENSE.TXT for details.
70d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//
80d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//===----------------------------------------------------------------------===//
90d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//
100d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//  This file defines a CheckObjCInstMethSignature, a flow-insenstive check
110d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//  that determines if an Objective-C class interface incorrectly redefines
120d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//  the method signature in a subclass.
130d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//
140d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek//===----------------------------------------------------------------------===//
150d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
167dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis#include "ClangSACheckers.h"
1755fc873017f10f6f566b182b70f6fc22aefa3464Chandler Carruth#include "clang/AST/ASTContext.h"
180d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek#include "clang/AST/DeclObjC.h"
190d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek#include "clang/AST/Type.h"
2055fc873017f10f6f566b182b70f6fc22aefa3464Chandler Carruth#include "clang/StaticAnalyzer/Core/BugReporter/BugReporter.h"
2155fc873017f10f6f566b182b70f6fc22aefa3464Chandler Carruth#include "clang/StaticAnalyzer/Core/BugReporter/PathDiagnostic.h"
2255fc873017f10f6f566b182b70f6fc22aefa3464Chandler Carruth#include "clang/StaticAnalyzer/Core/Checker.h"
230d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek#include "llvm/ADT/DenseMap.h"
2413493ea1583f39d62a66e2b2a0802f08d8ec32caTed Kremenek#include "llvm/Support/raw_ostream.h"
250d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
260d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenekusing namespace clang;
279ef6537a894c33003359b1f9b9676e9178e028b7Ted Kremenekusing namespace ento;
280d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
290d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenekstatic bool AreTypesCompatible(QualType Derived, QualType Ancestor,
309c378f705405d37f49795d5e915989de774fe11fTed Kremenek                               ASTContext &C) {
310d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
320d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  // Right now don't compare the compatibility of pointers.  That involves
330d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  // looking at subtyping relationships.  FIXME: Future patch.
3458f9f2c884af6b72d036b746a016d8031d31cb7aSteve Naroff  if (Derived->isAnyPointerType() &&  Ancestor->isAnyPointerType())
350d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek    return true;
360d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
370d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  return C.typesAreCompatible(Derived, Ancestor);
380d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek}
390d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
402376002038c8b904acd20be754aedd1a7471be71Ted Kremenekstatic void CompareReturnTypes(const ObjCMethodDecl *MethDerived,
412376002038c8b904acd20be754aedd1a7471be71Ted Kremenek                               const ObjCMethodDecl *MethAncestor,
422376002038c8b904acd20be754aedd1a7471be71Ted Kremenek                               BugReporter &BR, ASTContext &Ctx,
432376002038c8b904acd20be754aedd1a7471be71Ted Kremenek                               const ObjCImplementationDecl *ID) {
441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
450d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  QualType ResDerived  = MethDerived->getResultType();
461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  QualType ResAncestor = MethAncestor->getResultType();
471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
480d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  if (!AreTypesCompatible(ResDerived, ResAncestor, Ctx)) {
4913493ea1583f39d62a66e2b2a0802f08d8ec32caTed Kremenek    std::string sbuf;
5013493ea1583f39d62a66e2b2a0802f08d8ec32caTed Kremenek    llvm::raw_string_ostream os(sbuf);
511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
520d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek    os << "The Objective-C class '"
53b8989f27f116ff2400e92a52c067a69846119eb5Benjamin Kramer       << *MethDerived->getClassInterface()
540d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek       << "', which is derived from class '"
55b8989f27f116ff2400e92a52c067a69846119eb5Benjamin Kramer       << *MethAncestor->getClassInterface()
560d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek       << "', defines the instance method '"
57077bf5e2f48acfa9e7d69429b6e4ba86ea14896dChris Lattner       << MethDerived->getSelector().getAsString()
580d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek       << "' whose return type is '"
590d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek       << ResDerived.getAsString()
60e4773eb6398acb03ad692050cb84b53ca1750b5bTed Kremenek       << "'.  A method with the same name (same selector) is also defined in "
61e4773eb6398acb03ad692050cb84b53ca1750b5bTed Kremenek          "class '"
62b8989f27f116ff2400e92a52c067a69846119eb5Benjamin Kramer       << *MethAncestor->getClassInterface()
63e4773eb6398acb03ad692050cb84b53ca1750b5bTed Kremenek       << "' and has a return type of '"
640d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek       << ResAncestor.getAsString()
650d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek       << "'.  These two types are incompatible, and may result in undefined "
660d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek          "behavior for clients of these classes.";
671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
68590dd8e0959d8df5621827768987c4792b74fc06Anna Zaks    PathDiagnosticLocation MethDLoc =
69590dd8e0959d8df5621827768987c4792b74fc06Anna Zaks      PathDiagnosticLocation::createBegin(MethDerived,
70590dd8e0959d8df5621827768987c4792b74fc06Anna Zaks                                          BR.getSourceManager());
71590dd8e0959d8df5621827768987c4792b74fc06Anna Zaks
7207189521a15d9c088216b943649cb9fe231cbb57Ted Kremenek    BR.EmitBasicReport(MethDerived,
7307189521a15d9c088216b943649cb9fe231cbb57Ted Kremenek                       "Incompatible instance method return type",
746fd4505ad67a186da8cc26fdb493c93fe4937555Ted Kremenek                       categories::CoreFoundationObjectiveC,
75590dd8e0959d8df5621827768987c4792b74fc06Anna Zaks                       os.str(), MethDLoc);
760d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  }
770d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek}
780d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
799c378f705405d37f49795d5e915989de774fe11fTed Kremenekstatic void CheckObjCInstMethSignature(const ObjCImplementationDecl *ID,
807dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis                                       BugReporter& BR) {
811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
829c378f705405d37f49795d5e915989de774fe11fTed Kremenek  const ObjCInterfaceDecl *D = ID->getClassInterface();
839c378f705405d37f49795d5e915989de774fe11fTed Kremenek  const ObjCInterfaceDecl *C = D->getSuperClass();
840d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
850d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  if (!C)
860d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek    return;
871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
889c378f705405d37f49795d5e915989de774fe11fTed Kremenek  ASTContext &Ctx = BR.getContext();
891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
900d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  // Build a DenseMap of the methods for quick querying.
910d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  typedef llvm::DenseMap<Selector,ObjCMethodDecl*> MapTy;
920d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  MapTy IMeths;
930d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  unsigned NumMethods = 0;
941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9517945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  for (ObjCImplementationDecl::instmeth_iterator I=ID->instmeth_begin(),
961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump       E=ID->instmeth_end(); I!=E; ++I) {
971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
989c378f705405d37f49795d5e915989de774fe11fTed Kremenek    ObjCMethodDecl *M = *I;
990d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek    IMeths[M->getSelector()] = M;
1000d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek    ++NumMethods;
1010d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  }
1020d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
1030d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  // Now recurse the class hierarchy chain looking for methods with the
1040d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  // same signatures.
1050d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  while (C && NumMethods) {
10617945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis    for (ObjCInterfaceDecl::instmeth_iterator I=C->instmeth_begin(),
10717945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis         E=C->instmeth_end(); I!=E; ++I) {
1080d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
1099c378f705405d37f49795d5e915989de774fe11fTed Kremenek      ObjCMethodDecl *M = *I;
1100d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek      Selector S = M->getSelector();
1111eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1120d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek      MapTy::iterator MI = IMeths.find(S);
1130d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek
1140d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek      if (MI == IMeths.end() || MI->second == 0)
1150d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek        continue;
1161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1170d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek      --NumMethods;
1189c378f705405d37f49795d5e915989de774fe11fTed Kremenek      ObjCMethodDecl *MethDerived = MI->second;
1190d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek      MI->second = 0;
1201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1210d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek      CompareReturnTypes(MethDerived, M, BR, Ctx, ID);
1220d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek    }
1231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1240d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek    C = C->getSuperClass();
1250d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek  }
1260d8019e55c0f465bafc11b04aed691de95b9131dTed Kremenek}
1277dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis
1287dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis//===----------------------------------------------------------------------===//
1297dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis// ObjCMethSigsChecker
1307dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis//===----------------------------------------------------------------------===//
1317dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis
1327dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidisnamespace {
133ec8605f1d7ec846dbf51047bfd5c56d32d1ff91cArgyrios Kyrtzidisclass ObjCMethSigsChecker : public Checker<
1347dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis                                      check::ASTDecl<ObjCImplementationDecl> > {
1357dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidispublic:
1367dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis  void checkASTDecl(const ObjCImplementationDecl *D, AnalysisManager& mgr,
1377dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis                    BugReporter &BR) const {
1387dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis    CheckObjCInstMethSignature(D, BR);
1397dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis  }
1407dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis};
1417dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis}
1427dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis
1437dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidisvoid ento::registerObjCMethSigsChecker(CheckerManager &mgr) {
1447dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis  mgr.registerChecker<ObjCMethSigsChecker>();
1457dd445ec20e704846cfbdb132e56539280d71311Argyrios Kyrtzidis}
146