SemaPseudoObject.cpp revision d10099e5c8238fa0327f03921cf2e3c8975c881e
13c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//===--- SemaPseudoObject.cpp - Semantic Analysis for Pseudo-Objects ------===//
23c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//
33c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//                     The LLVM Compiler Infrastructure
43c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//
53c3b7f90a863af43fa63043d396553ecf205351cJohn McCall// This file is distributed under the University of Illinois Open Source
63c3b7f90a863af43fa63043d396553ecf205351cJohn McCall// License. See LICENSE.TXT for details.
73c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//
83c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//===----------------------------------------------------------------------===//
93c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//
103c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  This file implements semantic analysis for expressions involving
113c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  pseudo-object references.  Pseudo-objects are conceptual objects
123c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  whose storage is entirely abstract and all accesses to which are
133c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  translated through some sort of abstraction barrier.
143c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//
153c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  For example, Objective-C objects can have "properties", either
163c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  declared or undeclared.  A property may be accessed by writing
173c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//    expr.prop
183c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  where 'expr' is an r-value of Objective-C pointer type and 'prop'
193c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  is the name of the property.  If this expression is used in a context
203c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  needing an r-value, it is treated as if it were a message-send
213c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  of the associated 'getter' selector, typically:
223c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//    [expr prop]
233c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  If it is used as the LHS of a simple assignment, it is treated
243c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  as a message-send of the associated 'setter' selector, typically:
253c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//    [expr setProp: RHS]
263c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  If it is used as the LHS of a compound assignment, or the operand
273c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  of a unary increment or decrement, both are required;  for example,
283c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//  'expr.prop *= 100' would be translated to:
293c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//    [expr setProp: [expr prop] * 100]
303c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//
313c3b7f90a863af43fa63043d396553ecf205351cJohn McCall//===----------------------------------------------------------------------===//
323c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
333c3b7f90a863af43fa63043d396553ecf205351cJohn McCall#include "clang/Sema/SemaInternal.h"
343c3b7f90a863af43fa63043d396553ecf205351cJohn McCall#include "clang/Sema/Initialization.h"
353c3b7f90a863af43fa63043d396553ecf205351cJohn McCall#include "clang/AST/ExprObjC.h"
363c3b7f90a863af43fa63043d396553ecf205351cJohn McCall#include "clang/Lex/Preprocessor.h"
373c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
383c3b7f90a863af43fa63043d396553ecf205351cJohn McCallusing namespace clang;
393c3b7f90a863af43fa63043d396553ecf205351cJohn McCallusing namespace sema;
403c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
414b9c2d235fb9449e249d74f48ecfec601650de93John McCallnamespace {
424b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Basically just a very focused copy of TreeTransform.
434b9c2d235fb9449e249d74f48ecfec601650de93John McCall  template <class T> struct Rebuilder {
444b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Sema &S;
454b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Rebuilder(Sema &S) : S(S) {}
464b9c2d235fb9449e249d74f48ecfec601650de93John McCall
474b9c2d235fb9449e249d74f48ecfec601650de93John McCall    T &getDerived() { return static_cast<T&>(*this); }
484b9c2d235fb9449e249d74f48ecfec601650de93John McCall
494b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Expr *rebuild(Expr *e) {
504b9c2d235fb9449e249d74f48ecfec601650de93John McCall      // Fast path: nothing to look through.
514b9c2d235fb9449e249d74f48ecfec601650de93John McCall      if (typename T::specific_type *specific
524b9c2d235fb9449e249d74f48ecfec601650de93John McCall            = dyn_cast<typename T::specific_type>(e))
534b9c2d235fb9449e249d74f48ecfec601650de93John McCall        return getDerived().rebuildSpecific(specific);
544b9c2d235fb9449e249d74f48ecfec601650de93John McCall
554b9c2d235fb9449e249d74f48ecfec601650de93John McCall      // Otherwise, we should look through and rebuild anything that
564b9c2d235fb9449e249d74f48ecfec601650de93John McCall      // IgnoreParens would.
574b9c2d235fb9449e249d74f48ecfec601650de93John McCall
584b9c2d235fb9449e249d74f48ecfec601650de93John McCall      if (ParenExpr *parens = dyn_cast<ParenExpr>(e)) {
594b9c2d235fb9449e249d74f48ecfec601650de93John McCall        e = rebuild(parens->getSubExpr());
604b9c2d235fb9449e249d74f48ecfec601650de93John McCall        return new (S.Context) ParenExpr(parens->getLParen(),
614b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                         parens->getRParen(),
624b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                         e);
634b9c2d235fb9449e249d74f48ecfec601650de93John McCall      }
644b9c2d235fb9449e249d74f48ecfec601650de93John McCall
654b9c2d235fb9449e249d74f48ecfec601650de93John McCall      if (UnaryOperator *uop = dyn_cast<UnaryOperator>(e)) {
664b9c2d235fb9449e249d74f48ecfec601650de93John McCall        assert(uop->getOpcode() == UO_Extension);
674b9c2d235fb9449e249d74f48ecfec601650de93John McCall        e = rebuild(uop->getSubExpr());
684b9c2d235fb9449e249d74f48ecfec601650de93John McCall        return new (S.Context) UnaryOperator(e, uop->getOpcode(),
694b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             uop->getType(),
704b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             uop->getValueKind(),
714b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             uop->getObjectKind(),
724b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             uop->getOperatorLoc());
734b9c2d235fb9449e249d74f48ecfec601650de93John McCall      }
744b9c2d235fb9449e249d74f48ecfec601650de93John McCall
754b9c2d235fb9449e249d74f48ecfec601650de93John McCall      if (GenericSelectionExpr *gse = dyn_cast<GenericSelectionExpr>(e)) {
764b9c2d235fb9449e249d74f48ecfec601650de93John McCall        assert(!gse->isResultDependent());
774b9c2d235fb9449e249d74f48ecfec601650de93John McCall        unsigned resultIndex = gse->getResultIndex();
784b9c2d235fb9449e249d74f48ecfec601650de93John McCall        unsigned numAssocs = gse->getNumAssocs();
794b9c2d235fb9449e249d74f48ecfec601650de93John McCall
804b9c2d235fb9449e249d74f48ecfec601650de93John McCall        SmallVector<Expr*, 8> assocs(numAssocs);
814b9c2d235fb9449e249d74f48ecfec601650de93John McCall        SmallVector<TypeSourceInfo*, 8> assocTypes(numAssocs);
824b9c2d235fb9449e249d74f48ecfec601650de93John McCall
834b9c2d235fb9449e249d74f48ecfec601650de93John McCall        for (unsigned i = 0; i != numAssocs; ++i) {
844b9c2d235fb9449e249d74f48ecfec601650de93John McCall          Expr *assoc = gse->getAssocExpr(i);
854b9c2d235fb9449e249d74f48ecfec601650de93John McCall          if (i == resultIndex) assoc = rebuild(assoc);
864b9c2d235fb9449e249d74f48ecfec601650de93John McCall          assocs[i] = assoc;
874b9c2d235fb9449e249d74f48ecfec601650de93John McCall          assocTypes[i] = gse->getAssocTypeSourceInfo(i);
884b9c2d235fb9449e249d74f48ecfec601650de93John McCall        }
894b9c2d235fb9449e249d74f48ecfec601650de93John McCall
904b9c2d235fb9449e249d74f48ecfec601650de93John McCall        return new (S.Context) GenericSelectionExpr(S.Context,
914b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                    gse->getGenericLoc(),
924b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                    gse->getControllingExpr(),
934b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                    assocTypes.data(),
944b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                    assocs.data(),
954b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                    numAssocs,
964b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                    gse->getDefaultLoc(),
974b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                    gse->getRParenLoc(),
984b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                      gse->containsUnexpandedParameterPack(),
994b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                    resultIndex);
1004b9c2d235fb9449e249d74f48ecfec601650de93John McCall      }
1014b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1024b9c2d235fb9449e249d74f48ecfec601650de93John McCall      llvm_unreachable("bad expression to rebuild!");
1034b9c2d235fb9449e249d74f48ecfec601650de93John McCall    }
1044b9c2d235fb9449e249d74f48ecfec601650de93John McCall  };
1054b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1064b9c2d235fb9449e249d74f48ecfec601650de93John McCall  struct ObjCPropertyRefRebuilder : Rebuilder<ObjCPropertyRefRebuilder> {
1074b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Expr *NewBase;
1084b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ObjCPropertyRefRebuilder(Sema &S, Expr *newBase)
109acf9e82fd34c0a125bf2aa8be1ab868082a7f06aBenjamin Kramer      : Rebuilder<ObjCPropertyRefRebuilder>(S), NewBase(newBase) {}
1104b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1114b9c2d235fb9449e249d74f48ecfec601650de93John McCall    typedef ObjCPropertyRefExpr specific_type;
1124b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Expr *rebuildSpecific(ObjCPropertyRefExpr *refExpr) {
1134b9c2d235fb9449e249d74f48ecfec601650de93John McCall      // Fortunately, the constraint that we're rebuilding something
1144b9c2d235fb9449e249d74f48ecfec601650de93John McCall      // with a base limits the number of cases here.
1154b9c2d235fb9449e249d74f48ecfec601650de93John McCall      assert(refExpr->getBase());
1164b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1174b9c2d235fb9449e249d74f48ecfec601650de93John McCall      if (refExpr->isExplicitProperty()) {
1184b9c2d235fb9449e249d74f48ecfec601650de93John McCall        return new (S.Context)
1194b9c2d235fb9449e249d74f48ecfec601650de93John McCall          ObjCPropertyRefExpr(refExpr->getExplicitProperty(),
1204b9c2d235fb9449e249d74f48ecfec601650de93John McCall                              refExpr->getType(), refExpr->getValueKind(),
1214b9c2d235fb9449e249d74f48ecfec601650de93John McCall                              refExpr->getObjectKind(), refExpr->getLocation(),
1224b9c2d235fb9449e249d74f48ecfec601650de93John McCall                              NewBase);
1234b9c2d235fb9449e249d74f48ecfec601650de93John McCall      }
1244b9c2d235fb9449e249d74f48ecfec601650de93John McCall      return new (S.Context)
1254b9c2d235fb9449e249d74f48ecfec601650de93John McCall        ObjCPropertyRefExpr(refExpr->getImplicitPropertyGetter(),
1264b9c2d235fb9449e249d74f48ecfec601650de93John McCall                            refExpr->getImplicitPropertySetter(),
1274b9c2d235fb9449e249d74f48ecfec601650de93John McCall                            refExpr->getType(), refExpr->getValueKind(),
1284b9c2d235fb9449e249d74f48ecfec601650de93John McCall                            refExpr->getObjectKind(),refExpr->getLocation(),
1294b9c2d235fb9449e249d74f48ecfec601650de93John McCall                            NewBase);
1304b9c2d235fb9449e249d74f48ecfec601650de93John McCall    }
1314b9c2d235fb9449e249d74f48ecfec601650de93John McCall  };
1324b9c2d235fb9449e249d74f48ecfec601650de93John McCall
133ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  struct ObjCSubscriptRefRebuilder : Rebuilder<ObjCSubscriptRefRebuilder> {
134ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    Expr *NewBase;
135ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    Expr *NewKeyExpr;
136ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ObjCSubscriptRefRebuilder(Sema &S, Expr *newBase, Expr *newKeyExpr)
137ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    : Rebuilder<ObjCSubscriptRefRebuilder>(S),
138ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      NewBase(newBase), NewKeyExpr(newKeyExpr) {}
139ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
140ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    typedef ObjCSubscriptRefExpr specific_type;
141ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    Expr *rebuildSpecific(ObjCSubscriptRefExpr *refExpr) {
142ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      assert(refExpr->getBaseExpr());
143ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      assert(refExpr->getKeyExpr());
144ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
145ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      return new (S.Context)
146ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        ObjCSubscriptRefExpr(NewBase,
147ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                             NewKeyExpr,
148ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                             refExpr->getType(), refExpr->getValueKind(),
149ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                             refExpr->getObjectKind(),refExpr->getAtIndexMethodDecl(),
150ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                             refExpr->setAtIndexMethodDecl(),
151ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                             refExpr->getRBracket());
152ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    }
153ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  };
154ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1554b9c2d235fb9449e249d74f48ecfec601650de93John McCall  class PseudoOpBuilder {
1564b9c2d235fb9449e249d74f48ecfec601650de93John McCall  public:
1574b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Sema &S;
1584b9c2d235fb9449e249d74f48ecfec601650de93John McCall    unsigned ResultIndex;
1594b9c2d235fb9449e249d74f48ecfec601650de93John McCall    SourceLocation GenericLoc;
1604b9c2d235fb9449e249d74f48ecfec601650de93John McCall    SmallVector<Expr *, 4> Semantics;
1614b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1624b9c2d235fb9449e249d74f48ecfec601650de93John McCall    PseudoOpBuilder(Sema &S, SourceLocation genericLoc)
1634b9c2d235fb9449e249d74f48ecfec601650de93John McCall      : S(S), ResultIndex(PseudoObjectExpr::NoResult),
1644b9c2d235fb9449e249d74f48ecfec601650de93John McCall        GenericLoc(genericLoc) {}
1654b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1661aa1721f14f2162811f5393dbd84e01eadc2c963Matt Beaumont-Gay    virtual ~PseudoOpBuilder() {}
1671aa1721f14f2162811f5393dbd84e01eadc2c963Matt Beaumont-Gay
1684b9c2d235fb9449e249d74f48ecfec601650de93John McCall    /// Add a normal semantic expression.
1694b9c2d235fb9449e249d74f48ecfec601650de93John McCall    void addSemanticExpr(Expr *semantic) {
1704b9c2d235fb9449e249d74f48ecfec601650de93John McCall      Semantics.push_back(semantic);
1714b9c2d235fb9449e249d74f48ecfec601650de93John McCall    }
1724b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1734b9c2d235fb9449e249d74f48ecfec601650de93John McCall    /// Add the 'result' semantic expression.
1744b9c2d235fb9449e249d74f48ecfec601650de93John McCall    void addResultSemanticExpr(Expr *resultExpr) {
1754b9c2d235fb9449e249d74f48ecfec601650de93John McCall      assert(ResultIndex == PseudoObjectExpr::NoResult);
1764b9c2d235fb9449e249d74f48ecfec601650de93John McCall      ResultIndex = Semantics.size();
1774b9c2d235fb9449e249d74f48ecfec601650de93John McCall      Semantics.push_back(resultExpr);
1784b9c2d235fb9449e249d74f48ecfec601650de93John McCall    }
1794b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1804b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult buildRValueOperation(Expr *op);
1814b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult buildAssignmentOperation(Scope *Sc,
1824b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                        SourceLocation opLoc,
1834b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                        BinaryOperatorKind opcode,
1844b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                        Expr *LHS, Expr *RHS);
1854b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult buildIncDecOperation(Scope *Sc, SourceLocation opLoc,
1864b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                    UnaryOperatorKind opcode,
1874b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                    Expr *op);
1884b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1894b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult complete(Expr *syntacticForm);
1904b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1914b9c2d235fb9449e249d74f48ecfec601650de93John McCall    OpaqueValueExpr *capture(Expr *op);
1924b9c2d235fb9449e249d74f48ecfec601650de93John McCall    OpaqueValueExpr *captureValueAsResult(Expr *op);
1934b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1944b9c2d235fb9449e249d74f48ecfec601650de93John McCall    void setResultToLastSemantic() {
1954b9c2d235fb9449e249d74f48ecfec601650de93John McCall      assert(ResultIndex == PseudoObjectExpr::NoResult);
1964b9c2d235fb9449e249d74f48ecfec601650de93John McCall      ResultIndex = Semantics.size() - 1;
1974b9c2d235fb9449e249d74f48ecfec601650de93John McCall    }
1984b9c2d235fb9449e249d74f48ecfec601650de93John McCall
1994b9c2d235fb9449e249d74f48ecfec601650de93John McCall    /// Return true if assignments have a non-void result.
2004b9c2d235fb9449e249d74f48ecfec601650de93John McCall    virtual bool assignmentsHaveResult() { return true; }
2014b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2024b9c2d235fb9449e249d74f48ecfec601650de93John McCall    virtual Expr *rebuildAndCaptureObject(Expr *) = 0;
2034b9c2d235fb9449e249d74f48ecfec601650de93John McCall    virtual ExprResult buildGet() = 0;
2044b9c2d235fb9449e249d74f48ecfec601650de93John McCall    virtual ExprResult buildSet(Expr *, SourceLocation,
2054b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                bool captureSetValueAsResult) = 0;
2064b9c2d235fb9449e249d74f48ecfec601650de93John McCall  };
2074b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2084b9c2d235fb9449e249d74f48ecfec601650de93John McCall  /// A PseudoOpBuilder for Objective-C @properties.
2094b9c2d235fb9449e249d74f48ecfec601650de93John McCall  class ObjCPropertyOpBuilder : public PseudoOpBuilder {
2104b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ObjCPropertyRefExpr *RefExpr;
211b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    ObjCPropertyRefExpr *SyntacticRefExpr;
2124b9c2d235fb9449e249d74f48ecfec601650de93John McCall    OpaqueValueExpr *InstanceReceiver;
2134b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ObjCMethodDecl *Getter;
2144b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2154b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ObjCMethodDecl *Setter;
2164b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Selector SetterSelector;
217a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian    Selector GetterSelector;
2184b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2194b9c2d235fb9449e249d74f48ecfec601650de93John McCall  public:
2204b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ObjCPropertyOpBuilder(Sema &S, ObjCPropertyRefExpr *refExpr) :
2214b9c2d235fb9449e249d74f48ecfec601650de93John McCall      PseudoOpBuilder(S, refExpr->getLocation()), RefExpr(refExpr),
222b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis      SyntacticRefExpr(0), InstanceReceiver(0), Getter(0), Setter(0) {
2234b9c2d235fb9449e249d74f48ecfec601650de93John McCall    }
2244b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2254b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult buildRValueOperation(Expr *op);
2264b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult buildAssignmentOperation(Scope *Sc,
2274b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                        SourceLocation opLoc,
2284b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                        BinaryOperatorKind opcode,
2294b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                        Expr *LHS, Expr *RHS);
2304b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult buildIncDecOperation(Scope *Sc, SourceLocation opLoc,
2314b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                    UnaryOperatorKind opcode,
2324b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                    Expr *op);
2334b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2344b9c2d235fb9449e249d74f48ecfec601650de93John McCall    bool tryBuildGetOfReference(Expr *op, ExprResult &result);
2354b9c2d235fb9449e249d74f48ecfec601650de93John McCall    bool findSetter();
2364b9c2d235fb9449e249d74f48ecfec601650de93John McCall    bool findGetter();
2374b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2384b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Expr *rebuildAndCaptureObject(Expr *syntacticBase);
2394b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult buildGet();
2404b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult buildSet(Expr *op, SourceLocation, bool);
2414b9c2d235fb9449e249d74f48ecfec601650de93John McCall  };
242ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
243ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek /// A PseudoOpBuilder for Objective-C array/dictionary indexing.
244ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek class ObjCSubscriptOpBuilder : public PseudoOpBuilder {
245ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   ObjCSubscriptRefExpr *RefExpr;
246ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   OpaqueValueExpr *InstanceBase;
247ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   OpaqueValueExpr *InstanceKey;
248ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   ObjCMethodDecl *AtIndexGetter;
249ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   Selector AtIndexGetterSelector;
250ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
251ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   ObjCMethodDecl *AtIndexSetter;
252ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   Selector AtIndexSetterSelector;
253ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
254ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek public:
255ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ObjCSubscriptOpBuilder(Sema &S, ObjCSubscriptRefExpr *refExpr) :
256ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      PseudoOpBuilder(S, refExpr->getSourceRange().getBegin()),
257ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      RefExpr(refExpr),
258ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    InstanceBase(0), InstanceKey(0),
259ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexGetter(0), AtIndexSetter(0) { }
260ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
261ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   ExprResult buildRValueOperation(Expr *op);
262ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   ExprResult buildAssignmentOperation(Scope *Sc,
263ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                       SourceLocation opLoc,
264ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                       BinaryOperatorKind opcode,
265ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                       Expr *LHS, Expr *RHS);
266ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   Expr *rebuildAndCaptureObject(Expr *syntacticBase);
267ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
268ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   bool findAtIndexGetter();
269ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   bool findAtIndexSetter();
270ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
271ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   ExprResult buildGet();
272ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek   ExprResult buildSet(Expr *op, SourceLocation, bool);
273ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek };
274ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
2754b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
2764b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2774b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// Capture the given expression in an OpaqueValueExpr.
2784b9c2d235fb9449e249d74f48ecfec601650de93John McCallOpaqueValueExpr *PseudoOpBuilder::capture(Expr *e) {
2794b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Make a new OVE whose source is the given expression.
2804b9c2d235fb9449e249d74f48ecfec601650de93John McCall  OpaqueValueExpr *captured =
2814b9c2d235fb9449e249d74f48ecfec601650de93John McCall    new (S.Context) OpaqueValueExpr(GenericLoc, e->getType(),
28297df54e0c075bc8d6a869597e771dad0c11a2180Douglas Gregor                                    e->getValueKind(), e->getObjectKind(),
28397df54e0c075bc8d6a869597e771dad0c11a2180Douglas Gregor                                    e);
2844b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2854b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Make sure we bind that in the semantics.
2864b9c2d235fb9449e249d74f48ecfec601650de93John McCall  addSemanticExpr(captured);
2874b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return captured;
2884b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
2894b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2904b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// Capture the given expression as the result of this pseudo-object
2914b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// operation.  This routine is safe against expressions which may
2924b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// already be captured.
2934b9c2d235fb9449e249d74f48ecfec601650de93John McCall///
2944b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// \param Returns the captured expression, which will be the
2954b9c2d235fb9449e249d74f48ecfec601650de93John McCall///   same as the input if the input was already captured
2964b9c2d235fb9449e249d74f48ecfec601650de93John McCallOpaqueValueExpr *PseudoOpBuilder::captureValueAsResult(Expr *e) {
2974b9c2d235fb9449e249d74f48ecfec601650de93John McCall  assert(ResultIndex == PseudoObjectExpr::NoResult);
2984b9c2d235fb9449e249d74f48ecfec601650de93John McCall
2994b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // If the expression hasn't already been captured, just capture it
3004b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // and set the new semantic
3014b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (!isa<OpaqueValueExpr>(e)) {
3024b9c2d235fb9449e249d74f48ecfec601650de93John McCall    OpaqueValueExpr *cap = capture(e);
3034b9c2d235fb9449e249d74f48ecfec601650de93John McCall    setResultToLastSemantic();
3044b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return cap;
3054b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
3064b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3074b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Otherwise, it must already be one of our semantic expressions;
3084b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // set ResultIndex to its index.
3094b9c2d235fb9449e249d74f48ecfec601650de93John McCall  unsigned index = 0;
3104b9c2d235fb9449e249d74f48ecfec601650de93John McCall  for (;; ++index) {
3114b9c2d235fb9449e249d74f48ecfec601650de93John McCall    assert(index < Semantics.size() &&
3124b9c2d235fb9449e249d74f48ecfec601650de93John McCall           "captured expression not found in semantics!");
3134b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (e == Semantics[index]) break;
3144b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
3154b9c2d235fb9449e249d74f48ecfec601650de93John McCall  ResultIndex = index;
3164b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return cast<OpaqueValueExpr>(e);
3174b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
3184b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3194b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// The routine which creates the final PseudoObjectExpr.
3204b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult PseudoOpBuilder::complete(Expr *syntactic) {
3214b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return PseudoObjectExpr::Create(S.Context, syntactic,
3224b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                  Semantics, ResultIndex);
3234b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
3244b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3254b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// The main skeleton for building an r-value operation.
3264b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult PseudoOpBuilder::buildRValueOperation(Expr *op) {
3274b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Expr *syntacticBase = rebuildAndCaptureObject(op);
3284b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3294b9c2d235fb9449e249d74f48ecfec601650de93John McCall  ExprResult getExpr = buildGet();
3304b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (getExpr.isInvalid()) return ExprError();
3314b9c2d235fb9449e249d74f48ecfec601650de93John McCall  addResultSemanticExpr(getExpr.take());
3324b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3334b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return complete(syntacticBase);
3344b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
3354b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3364b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// The basic skeleton for building a simple or compound
3374b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// assignment operation.
3384b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult
3394b9c2d235fb9449e249d74f48ecfec601650de93John McCallPseudoOpBuilder::buildAssignmentOperation(Scope *Sc, SourceLocation opcLoc,
3404b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                          BinaryOperatorKind opcode,
3414b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                          Expr *LHS, Expr *RHS) {
3424b9c2d235fb9449e249d74f48ecfec601650de93John McCall  assert(BinaryOperator::isAssignmentOp(opcode));
3434b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3444b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Expr *syntacticLHS = rebuildAndCaptureObject(LHS);
3454b9c2d235fb9449e249d74f48ecfec601650de93John McCall  OpaqueValueExpr *capturedRHS = capture(RHS);
3464b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3474b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Expr *syntactic;
3484b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3494b9c2d235fb9449e249d74f48ecfec601650de93John McCall  ExprResult result;
3504b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (opcode == BO_Assign) {
3514b9c2d235fb9449e249d74f48ecfec601650de93John McCall    result = capturedRHS;
3524b9c2d235fb9449e249d74f48ecfec601650de93John McCall    syntactic = new (S.Context) BinaryOperator(syntacticLHS, capturedRHS,
3534b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                               opcode, capturedRHS->getType(),
3544b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                               capturedRHS->getValueKind(),
3554b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                               OK_Ordinary, opcLoc);
3564b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else {
3574b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult opLHS = buildGet();
3584b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (opLHS.isInvalid()) return ExprError();
3594b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3604b9c2d235fb9449e249d74f48ecfec601650de93John McCall    // Build an ordinary, non-compound operation.
3614b9c2d235fb9449e249d74f48ecfec601650de93John McCall    BinaryOperatorKind nonCompound =
3624b9c2d235fb9449e249d74f48ecfec601650de93John McCall      BinaryOperator::getOpForCompoundAssignment(opcode);
3634b9c2d235fb9449e249d74f48ecfec601650de93John McCall    result = S.BuildBinOp(Sc, opcLoc, nonCompound,
3644b9c2d235fb9449e249d74f48ecfec601650de93John McCall                          opLHS.take(), capturedRHS);
3654b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (result.isInvalid()) return ExprError();
3664b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3674b9c2d235fb9449e249d74f48ecfec601650de93John McCall    syntactic =
3684b9c2d235fb9449e249d74f48ecfec601650de93John McCall      new (S.Context) CompoundAssignOperator(syntacticLHS, capturedRHS, opcode,
3694b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             result.get()->getType(),
3704b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             result.get()->getValueKind(),
3714b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             OK_Ordinary,
3724b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             opLHS.get()->getType(),
3734b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             result.get()->getType(),
3744b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             opcLoc);
3754b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
3764b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3774b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // The result of the assignment, if not void, is the value set into
3784b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // the l-value.
3794b9c2d235fb9449e249d74f48ecfec601650de93John McCall  result = buildSet(result.take(), opcLoc, assignmentsHaveResult());
3804b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (result.isInvalid()) return ExprError();
3814b9c2d235fb9449e249d74f48ecfec601650de93John McCall  addSemanticExpr(result.take());
3824b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3834b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return complete(syntactic);
3844b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
3854b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3864b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// The basic skeleton for building an increment or decrement
3874b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// operation.
3884b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult
3894b9c2d235fb9449e249d74f48ecfec601650de93John McCallPseudoOpBuilder::buildIncDecOperation(Scope *Sc, SourceLocation opcLoc,
3904b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                      UnaryOperatorKind opcode,
3914b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                      Expr *op) {
3924b9c2d235fb9449e249d74f48ecfec601650de93John McCall  assert(UnaryOperator::isIncrementDecrementOp(opcode));
3934b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3944b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Expr *syntacticOp = rebuildAndCaptureObject(op);
3954b9c2d235fb9449e249d74f48ecfec601650de93John McCall
3964b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Load the value.
3974b9c2d235fb9449e249d74f48ecfec601650de93John McCall  ExprResult result = buildGet();
3984b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (result.isInvalid()) return ExprError();
3994b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4004b9c2d235fb9449e249d74f48ecfec601650de93John McCall  QualType resultType = result.get()->getType();
4014b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4024b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // That's the postfix result.
4034b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (UnaryOperator::isPostfix(opcode) && assignmentsHaveResult()) {
4044b9c2d235fb9449e249d74f48ecfec601650de93John McCall    result = capture(result.take());
4054b9c2d235fb9449e249d74f48ecfec601650de93John McCall    setResultToLastSemantic();
4064b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
4074b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4084b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Add or subtract a literal 1.
4094b9c2d235fb9449e249d74f48ecfec601650de93John McCall  llvm::APInt oneV(S.Context.getTypeSize(S.Context.IntTy), 1);
4104b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Expr *one = IntegerLiteral::Create(S.Context, oneV, S.Context.IntTy,
4114b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                     GenericLoc);
4124b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4134b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (UnaryOperator::isIncrementOp(opcode)) {
4144b9c2d235fb9449e249d74f48ecfec601650de93John McCall    result = S.BuildBinOp(Sc, opcLoc, BO_Add, result.take(), one);
4154b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else {
4164b9c2d235fb9449e249d74f48ecfec601650de93John McCall    result = S.BuildBinOp(Sc, opcLoc, BO_Sub, result.take(), one);
4174b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
4184b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (result.isInvalid()) return ExprError();
4194b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4204b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Store that back into the result.  The value stored is the result
4214b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // of a prefix operation.
4224b9c2d235fb9449e249d74f48ecfec601650de93John McCall  result = buildSet(result.take(), opcLoc,
4234b9c2d235fb9449e249d74f48ecfec601650de93John McCall             UnaryOperator::isPrefix(opcode) && assignmentsHaveResult());
4244b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (result.isInvalid()) return ExprError();
4254b9c2d235fb9449e249d74f48ecfec601650de93John McCall  addSemanticExpr(result.take());
4264b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4274b9c2d235fb9449e249d74f48ecfec601650de93John McCall  UnaryOperator *syntactic =
4284b9c2d235fb9449e249d74f48ecfec601650de93John McCall    new (S.Context) UnaryOperator(syntacticOp, opcode, resultType,
4294b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                  VK_LValue, OK_Ordinary, opcLoc);
4304b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return complete(syntactic);
4314b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
4324b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4334b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4344b9c2d235fb9449e249d74f48ecfec601650de93John McCall//===----------------------------------------------------------------------===//
4354b9c2d235fb9449e249d74f48ecfec601650de93John McCall//  Objective-C @property and implicit property references
4364b9c2d235fb9449e249d74f48ecfec601650de93John McCall//===----------------------------------------------------------------------===//
4374b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4384b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// Look up a method in the receiver type of an Objective-C property
4394b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// reference.
4403c3b7f90a863af43fa63043d396553ecf205351cJohn McCallstatic ObjCMethodDecl *LookupMethodInReceiverType(Sema &S, Selector sel,
4413c3b7f90a863af43fa63043d396553ecf205351cJohn McCall                                            const ObjCPropertyRefExpr *PRE) {
4423c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  if (PRE->isObjectReceiver()) {
443aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer    const ObjCObjectPointerType *PT =
444aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer      PRE->getBase()->getType()->castAs<ObjCObjectPointerType>();
4454b9c2d235fb9449e249d74f48ecfec601650de93John McCall
4464b9c2d235fb9449e249d74f48ecfec601650de93John McCall    // Special case for 'self' in class method implementations.
4474b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (PT->isObjCClassType() &&
4484b9c2d235fb9449e249d74f48ecfec601650de93John McCall        S.isSelfExpr(const_cast<Expr*>(PRE->getBase()))) {
4494b9c2d235fb9449e249d74f48ecfec601650de93John McCall      // This cast is safe because isSelfExpr is only true within
4504b9c2d235fb9449e249d74f48ecfec601650de93John McCall      // methods.
4514b9c2d235fb9449e249d74f48ecfec601650de93John McCall      ObjCMethodDecl *method =
4524b9c2d235fb9449e249d74f48ecfec601650de93John McCall        cast<ObjCMethodDecl>(S.CurContext->getNonClosureAncestor());
4534b9c2d235fb9449e249d74f48ecfec601650de93John McCall      return S.LookupMethodInObjectType(sel,
4544b9c2d235fb9449e249d74f48ecfec601650de93John McCall                 S.Context.getObjCInterfaceType(method->getClassInterface()),
4554b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                        /*instance*/ false);
4564b9c2d235fb9449e249d74f48ecfec601650de93John McCall    }
4574b9c2d235fb9449e249d74f48ecfec601650de93John McCall
458aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer    return S.LookupMethodInObjectType(sel, PT->getPointeeType(), true);
4593c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  }
4603c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
461aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer  if (PRE->isSuperReceiver()) {
462aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer    if (const ObjCObjectPointerType *PT =
463aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer        PRE->getSuperReceiverType()->getAs<ObjCObjectPointerType>())
464aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer      return S.LookupMethodInObjectType(sel, PT->getPointeeType(), true);
465aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer
466aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer    return S.LookupMethodInObjectType(sel, PRE->getSuperReceiverType(), false);
467aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer  }
468aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer
469aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer  assert(PRE->isClassReceiver() && "Invalid expression");
470aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer  QualType IT = S.Context.getObjCInterfaceType(PRE->getClassReceiver());
471aa9807a85959ffbdc5d9f649d7b24b9b2056d2cdBenjamin Kramer  return S.LookupMethodInObjectType(sel, IT, false);
4723c3b7f90a863af43fa63043d396553ecf205351cJohn McCall}
4733c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
4744b9c2d235fb9449e249d74f48ecfec601650de93John McCallbool ObjCPropertyOpBuilder::findGetter() {
4754b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (Getter) return true;
4764b9c2d235fb9449e249d74f48ecfec601650de93John McCall
477dc4df51d32dea56fbec79037570832dffbcc14e6John McCall  // For implicit properties, just trust the lookup we already did.
478dc4df51d32dea56fbec79037570832dffbcc14e6John McCall  if (RefExpr->isImplicitProperty()) {
479a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian    if ((Getter = RefExpr->getImplicitPropertyGetter())) {
480a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      GetterSelector = Getter->getSelector();
481a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      return true;
482a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian    }
483a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian    else {
484a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      // Must build the getter selector the hard way.
485a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      ObjCMethodDecl *setter = RefExpr->getImplicitPropertySetter();
486a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      assert(setter && "both setter and getter are null - cannot happen");
487a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      IdentifierInfo *setterName =
488a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian        setter->getSelector().getIdentifierInfoForSlot(0);
489a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      const char *compStr = setterName->getNameStart();
490a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      compStr += 3;
491a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      IdentifierInfo *getterName = &S.Context.Idents.get(compStr);
492a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      GetterSelector =
493a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian        S.PP.getSelectorTable().getNullarySelector(getterName);
494a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      return false;
495a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian
496a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian    }
497dc4df51d32dea56fbec79037570832dffbcc14e6John McCall  }
498dc4df51d32dea56fbec79037570832dffbcc14e6John McCall
499dc4df51d32dea56fbec79037570832dffbcc14e6John McCall  ObjCPropertyDecl *prop = RefExpr->getExplicitProperty();
500dc4df51d32dea56fbec79037570832dffbcc14e6John McCall  Getter = LookupMethodInReceiverType(S, prop->getGetterName(), RefExpr);
5014b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return (Getter != 0);
5024b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
5034b9c2d235fb9449e249d74f48ecfec601650de93John McCall
5044b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// Try to find the most accurate setter declaration for the property
5054b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// reference.
5064b9c2d235fb9449e249d74f48ecfec601650de93John McCall///
5074b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// \return true if a setter was found, in which case Setter
5084b9c2d235fb9449e249d74f48ecfec601650de93John McCallbool ObjCPropertyOpBuilder::findSetter() {
5094b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // For implicit properties, just trust the lookup we already did.
5104b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (RefExpr->isImplicitProperty()) {
5114b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (ObjCMethodDecl *setter = RefExpr->getImplicitPropertySetter()) {
5124b9c2d235fb9449e249d74f48ecfec601650de93John McCall      Setter = setter;
5134b9c2d235fb9449e249d74f48ecfec601650de93John McCall      SetterSelector = setter->getSelector();
5144b9c2d235fb9449e249d74f48ecfec601650de93John McCall      return true;
5153c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    } else {
5164b9c2d235fb9449e249d74f48ecfec601650de93John McCall      IdentifierInfo *getterName =
5174b9c2d235fb9449e249d74f48ecfec601650de93John McCall        RefExpr->getImplicitPropertyGetter()->getSelector()
5184b9c2d235fb9449e249d74f48ecfec601650de93John McCall          .getIdentifierInfoForSlot(0);
5194b9c2d235fb9449e249d74f48ecfec601650de93John McCall      SetterSelector =
5204b9c2d235fb9449e249d74f48ecfec601650de93John McCall        SelectorTable::constructSetterName(S.PP.getIdentifierTable(),
5214b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                           S.PP.getSelectorTable(),
5224b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                           getterName);
5234b9c2d235fb9449e249d74f48ecfec601650de93John McCall      return false;
5243c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    }
5253c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  }
5263c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
5274b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // For explicit properties, this is more involved.
5284b9c2d235fb9449e249d74f48ecfec601650de93John McCall  ObjCPropertyDecl *prop = RefExpr->getExplicitProperty();
5294b9c2d235fb9449e249d74f48ecfec601650de93John McCall  SetterSelector = prop->getSetterName();
5304b9c2d235fb9449e249d74f48ecfec601650de93John McCall
5314b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Do a normal method lookup first.
5324b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (ObjCMethodDecl *setter =
5334b9c2d235fb9449e249d74f48ecfec601650de93John McCall        LookupMethodInReceiverType(S, SetterSelector, RefExpr)) {
5344b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Setter = setter;
5354b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return true;
5364b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
5374b9c2d235fb9449e249d74f48ecfec601650de93John McCall
5384b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // That can fail in the somewhat crazy situation that we're
5394b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // type-checking a message send within the @interface declaration
5404b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // that declared the @property.  But it's not clear that that's
5414b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // valuable to support.
5424b9c2d235fb9449e249d74f48ecfec601650de93John McCall
5434b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return false;
5444b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
5454b9c2d235fb9449e249d74f48ecfec601650de93John McCall
5464b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// Capture the base object of an Objective-C property expression.
5474b9c2d235fb9449e249d74f48ecfec601650de93John McCallExpr *ObjCPropertyOpBuilder::rebuildAndCaptureObject(Expr *syntacticBase) {
5484b9c2d235fb9449e249d74f48ecfec601650de93John McCall  assert(InstanceReceiver == 0);
5494b9c2d235fb9449e249d74f48ecfec601650de93John McCall
5504b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // If we have a base, capture it in an OVE and rebuild the syntactic
5514b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // form to use the OVE as its base.
5524b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (RefExpr->isObjectReceiver()) {
5534b9c2d235fb9449e249d74f48ecfec601650de93John McCall    InstanceReceiver = capture(RefExpr->getBase());
5544b9c2d235fb9449e249d74f48ecfec601650de93John McCall
5554b9c2d235fb9449e249d74f48ecfec601650de93John McCall    syntacticBase =
5564b9c2d235fb9449e249d74f48ecfec601650de93John McCall      ObjCPropertyRefRebuilder(S, InstanceReceiver).rebuild(syntacticBase);
5574b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
5584b9c2d235fb9449e249d74f48ecfec601650de93John McCall
559b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis  if (ObjCPropertyRefExpr *
560b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis        refE = dyn_cast<ObjCPropertyRefExpr>(syntacticBase->IgnoreParens()))
561b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    SyntacticRefExpr = refE;
562b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis
5634b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return syntacticBase;
5644b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
5654b9c2d235fb9449e249d74f48ecfec601650de93John McCall
5664b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// Load from an Objective-C property reference.
5674b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult ObjCPropertyOpBuilder::buildGet() {
5684b9c2d235fb9449e249d74f48ecfec601650de93John McCall  findGetter();
5694b9c2d235fb9449e249d74f48ecfec601650de93John McCall  assert(Getter);
570b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis
571b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis  if (SyntacticRefExpr)
572b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    SyntacticRefExpr->setIsMessagingGetter();
573b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis
5744b9c2d235fb9449e249d74f48ecfec601650de93John McCall  QualType receiverType;
5754b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (RefExpr->isClassReceiver()) {
5764b9c2d235fb9449e249d74f48ecfec601650de93John McCall    receiverType = S.Context.getObjCInterfaceType(RefExpr->getClassReceiver());
5774b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else if (RefExpr->isSuperReceiver()) {
5784b9c2d235fb9449e249d74f48ecfec601650de93John McCall    receiverType = RefExpr->getSuperReceiverType();
5794b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else {
5804b9c2d235fb9449e249d74f48ecfec601650de93John McCall    assert(InstanceReceiver);
5814b9c2d235fb9449e249d74f48ecfec601650de93John McCall    receiverType = InstanceReceiver->getType();
5824b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
5833c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
5844b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Build a message-send.
5854b9c2d235fb9449e249d74f48ecfec601650de93John McCall  ExprResult msg;
5864b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (Getter->isInstanceMethod() || RefExpr->isObjectReceiver()) {
5874b9c2d235fb9449e249d74f48ecfec601650de93John McCall    assert(InstanceReceiver || RefExpr->isSuperReceiver());
588746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis    msg = S.BuildInstanceMessageImplicit(InstanceReceiver, receiverType,
589746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                         GenericLoc, Getter->getSelector(),
590746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                         Getter, MultiExprArg());
5914b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else {
592746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis    msg = S.BuildClassMessageImplicit(receiverType, RefExpr->isSuperReceiver(),
593746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                      GenericLoc,
594746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                      Getter->getSelector(), Getter,
595746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                      MultiExprArg());
5964b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
5974b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return msg;
5983c3b7f90a863af43fa63043d396553ecf205351cJohn McCall}
5993c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
6004b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// Store to an Objective-C property reference.
6014b9c2d235fb9449e249d74f48ecfec601650de93John McCall///
6024b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// \param bindSetValueAsResult - If true, capture the actual
6034b9c2d235fb9449e249d74f48ecfec601650de93John McCall///   value being set as the value of the property operation.
6044b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult ObjCPropertyOpBuilder::buildSet(Expr *op, SourceLocation opcLoc,
6054b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                           bool captureSetValueAsResult) {
6064b9c2d235fb9449e249d74f48ecfec601650de93John McCall  bool hasSetter = findSetter();
6074b9c2d235fb9449e249d74f48ecfec601650de93John McCall  assert(hasSetter); (void) hasSetter;
6084b9c2d235fb9449e249d74f48ecfec601650de93John McCall
609b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis  if (SyntacticRefExpr)
610b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    SyntacticRefExpr->setIsMessagingSetter();
611b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis
6124b9c2d235fb9449e249d74f48ecfec601650de93John McCall  QualType receiverType;
6134b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (RefExpr->isClassReceiver()) {
6144b9c2d235fb9449e249d74f48ecfec601650de93John McCall    receiverType = S.Context.getObjCInterfaceType(RefExpr->getClassReceiver());
6154b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else if (RefExpr->isSuperReceiver()) {
6164b9c2d235fb9449e249d74f48ecfec601650de93John McCall    receiverType = RefExpr->getSuperReceiverType();
6174b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else {
6184b9c2d235fb9449e249d74f48ecfec601650de93John McCall    assert(InstanceReceiver);
6194b9c2d235fb9449e249d74f48ecfec601650de93John McCall    receiverType = InstanceReceiver->getType();
6204b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
6213c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
6224b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Use assignment constraints when possible; they give us better
6234b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // diagnostics.  "When possible" basically means anything except a
6244b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // C++ class type.
6254e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie  if (!S.getLangOpts().CPlusPlus || !op->getType()->isRecordType()) {
6264b9c2d235fb9449e249d74f48ecfec601650de93John McCall    QualType paramType = (*Setter->param_begin())->getType();
6274e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie    if (!S.getLangOpts().CPlusPlus || !paramType->isRecordType()) {
6284b9c2d235fb9449e249d74f48ecfec601650de93John McCall      ExprResult opResult = op;
6294b9c2d235fb9449e249d74f48ecfec601650de93John McCall      Sema::AssignConvertType assignResult
6304b9c2d235fb9449e249d74f48ecfec601650de93John McCall        = S.CheckSingleAssignmentConstraints(paramType, opResult);
6314b9c2d235fb9449e249d74f48ecfec601650de93John McCall      if (S.DiagnoseAssignmentResult(assignResult, opcLoc, paramType,
6324b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                     op->getType(), opResult.get(),
6334b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                     Sema::AA_Assigning))
6344b9c2d235fb9449e249d74f48ecfec601650de93John McCall        return ExprError();
6354b9c2d235fb9449e249d74f48ecfec601650de93John McCall
6364b9c2d235fb9449e249d74f48ecfec601650de93John McCall      op = opResult.take();
6374b9c2d235fb9449e249d74f48ecfec601650de93John McCall      assert(op && "successful assignment left argument invalid?");
6383c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    }
6394b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
6403c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
6414b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Arguments.
6424b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Expr *args[] = { op };
6433c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
6444b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Build a message-send.
6454b9c2d235fb9449e249d74f48ecfec601650de93John McCall  ExprResult msg;
6464b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (Setter->isInstanceMethod() || RefExpr->isObjectReceiver()) {
647746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis    msg = S.BuildInstanceMessageImplicit(InstanceReceiver, receiverType,
648746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                         GenericLoc, SetterSelector, Setter,
649746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                         MultiExprArg(args, 1));
6504b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else {
651746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis    msg = S.BuildClassMessageImplicit(receiverType, RefExpr->isSuperReceiver(),
652746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                      GenericLoc,
653746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                      SetterSelector, Setter,
654746f5bcbfde5b25269169c63c66492311673b67dArgyrios Kyrtzidis                                      MultiExprArg(args, 1));
6554b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
6563c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
6574b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (!msg.isInvalid() && captureSetValueAsResult) {
6584b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ObjCMessageExpr *msgExpr =
6594b9c2d235fb9449e249d74f48ecfec601650de93John McCall      cast<ObjCMessageExpr>(msg.get()->IgnoreImplicit());
6604b9c2d235fb9449e249d74f48ecfec601650de93John McCall    Expr *arg = msgExpr->getArg(0);
6614b9c2d235fb9449e249d74f48ecfec601650de93John McCall    msgExpr->setArg(0, captureValueAsResult(arg));
6624b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
6633c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
6644b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return msg;
6654b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
6664b9c2d235fb9449e249d74f48ecfec601650de93John McCall
6674b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// @property-specific behavior for doing lvalue-to-rvalue conversion.
6684b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult ObjCPropertyOpBuilder::buildRValueOperation(Expr *op) {
6694b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Explicit properties always have getters, but implicit ones don't.
6704b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Check that before proceeding.
6714b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (RefExpr->isImplicitProperty() &&
6724b9c2d235fb9449e249d74f48ecfec601650de93John McCall      !RefExpr->getImplicitPropertyGetter()) {
6734b9c2d235fb9449e249d74f48ecfec601650de93John McCall    S.Diag(RefExpr->getLocation(), diag::err_getter_not_found)
6744b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << RefExpr->getBase()->getType();
6754b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return ExprError();
6764b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
6774b9c2d235fb9449e249d74f48ecfec601650de93John McCall
6784b9c2d235fb9449e249d74f48ecfec601650de93John McCall  ExprResult result = PseudoOpBuilder::buildRValueOperation(op);
6794b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (result.isInvalid()) return ExprError();
6804b9c2d235fb9449e249d74f48ecfec601650de93John McCall
6814b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (RefExpr->isExplicitProperty() && !Getter->hasRelatedResultType())
6824b9c2d235fb9449e249d74f48ecfec601650de93John McCall    S.DiagnosePropertyAccessorMismatch(RefExpr->getExplicitProperty(),
6834b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                       Getter, RefExpr->getLocation());
6844b9c2d235fb9449e249d74f48ecfec601650de93John McCall
6854b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // As a special case, if the method returns 'id', try to get
6864b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // a better type from the property.
6874b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (RefExpr->isExplicitProperty() && result.get()->isRValue() &&
6884b9c2d235fb9449e249d74f48ecfec601650de93John McCall      result.get()->getType()->isObjCIdType()) {
6894b9c2d235fb9449e249d74f48ecfec601650de93John McCall    QualType propType = RefExpr->getExplicitProperty()->getType();
6904b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (const ObjCObjectPointerType *ptr
6914b9c2d235fb9449e249d74f48ecfec601650de93John McCall          = propType->getAs<ObjCObjectPointerType>()) {
6924b9c2d235fb9449e249d74f48ecfec601650de93John McCall      if (!ptr->isObjCIdType())
6934b9c2d235fb9449e249d74f48ecfec601650de93John McCall        result = S.ImpCastExprToType(result.get(), propType, CK_BitCast);
6943c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    }
6954b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
6964b9c2d235fb9449e249d74f48ecfec601650de93John McCall
6974b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return result;
6983c3b7f90a863af43fa63043d396553ecf205351cJohn McCall}
6993c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7004b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// Try to build this as a call to a getter that returns a reference.
7014b9c2d235fb9449e249d74f48ecfec601650de93John McCall///
7024b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// \return true if it was possible, whether or not it actually
7034b9c2d235fb9449e249d74f48ecfec601650de93John McCall///   succeeded
7044b9c2d235fb9449e249d74f48ecfec601650de93John McCallbool ObjCPropertyOpBuilder::tryBuildGetOfReference(Expr *op,
7054b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                   ExprResult &result) {
7064e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie  if (!S.getLangOpts().CPlusPlus) return false;
7074b9c2d235fb9449e249d74f48ecfec601650de93John McCall
7084b9c2d235fb9449e249d74f48ecfec601650de93John McCall  findGetter();
7094b9c2d235fb9449e249d74f48ecfec601650de93John McCall  assert(Getter && "property has no setter and no getter!");
7104b9c2d235fb9449e249d74f48ecfec601650de93John McCall
7114b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Only do this if the getter returns an l-value reference type.
7124b9c2d235fb9449e249d74f48ecfec601650de93John McCall  QualType resultType = Getter->getResultType();
7134b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (!resultType->isLValueReferenceType()) return false;
7144b9c2d235fb9449e249d74f48ecfec601650de93John McCall
7154b9c2d235fb9449e249d74f48ecfec601650de93John McCall  result = buildRValueOperation(op);
7164b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return true;
7174b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
7184b9c2d235fb9449e249d74f48ecfec601650de93John McCall
7194b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// @property-specific behavior for doing assignments.
7204b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult
7214b9c2d235fb9449e249d74f48ecfec601650de93John McCallObjCPropertyOpBuilder::buildAssignmentOperation(Scope *Sc,
7224b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                SourceLocation opcLoc,
7234b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                BinaryOperatorKind opcode,
7244b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                                Expr *LHS, Expr *RHS) {
7254b9c2d235fb9449e249d74f48ecfec601650de93John McCall  assert(BinaryOperator::isAssignmentOp(opcode));
7263c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7273c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  // If there's no setter, we have no choice but to try to assign to
7283c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  // the result of the getter.
7294b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (!findSetter()) {
7304b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult result;
7314b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (tryBuildGetOfReference(LHS, result)) {
7324b9c2d235fb9449e249d74f48ecfec601650de93John McCall      if (result.isInvalid()) return ExprError();
7334b9c2d235fb9449e249d74f48ecfec601650de93John McCall      return S.BuildBinOp(Sc, opcLoc, opcode, result.take(), RHS);
7343c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    }
7353c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7363c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    // Otherwise, it's an error.
7374b9c2d235fb9449e249d74f48ecfec601650de93John McCall    S.Diag(opcLoc, diag::err_nosetter_property_assignment)
7384b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << unsigned(RefExpr->isImplicitProperty())
7394b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << SetterSelector
7404b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << LHS->getSourceRange() << RHS->getSourceRange();
7413c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    return ExprError();
7423c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  }
7433c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7444b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // If there is a setter, we definitely want to use it.
7454b9c2d235fb9449e249d74f48ecfec601650de93John McCall
7464b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Verify that we can do a compound assignment.
7474b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (opcode != BO_Assign && !findGetter()) {
7484b9c2d235fb9449e249d74f48ecfec601650de93John McCall    S.Diag(opcLoc, diag::err_nogetter_property_compound_assignment)
7494b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << LHS->getSourceRange() << RHS->getSourceRange();
7503c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    return ExprError();
7513c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  }
7523c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7534b9c2d235fb9449e249d74f48ecfec601650de93John McCall  ExprResult result =
7544b9c2d235fb9449e249d74f48ecfec601650de93John McCall    PseudoOpBuilder::buildAssignmentOperation(Sc, opcLoc, opcode, LHS, RHS);
7553c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  if (result.isInvalid()) return ExprError();
7563c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7574b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Various warnings about property assignments in ARC.
7584e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie  if (S.getLangOpts().ObjCAutoRefCount && InstanceReceiver) {
7594b9c2d235fb9449e249d74f48ecfec601650de93John McCall    S.checkRetainCycles(InstanceReceiver->getSourceExpr(), RHS);
7604b9c2d235fb9449e249d74f48ecfec601650de93John McCall    S.checkUnsafeExprAssigns(opcLoc, LHS, RHS);
7614b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
7624b9c2d235fb9449e249d74f48ecfec601650de93John McCall
7633c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  return result;
7643c3b7f90a863af43fa63043d396553ecf205351cJohn McCall}
7653c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7664b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// @property-specific behavior for doing increments and decrements.
7674b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult
7684b9c2d235fb9449e249d74f48ecfec601650de93John McCallObjCPropertyOpBuilder::buildIncDecOperation(Scope *Sc, SourceLocation opcLoc,
7694b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                            UnaryOperatorKind opcode,
7704b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                            Expr *op) {
7713c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  // If there's no setter, we have no choice but to try to assign to
7723c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  // the result of the getter.
7734b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (!findSetter()) {
7744b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ExprResult result;
7754b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (tryBuildGetOfReference(op, result)) {
7764b9c2d235fb9449e249d74f48ecfec601650de93John McCall      if (result.isInvalid()) return ExprError();
7774b9c2d235fb9449e249d74f48ecfec601650de93John McCall      return S.BuildUnaryOp(Sc, opcLoc, opcode, result.take());
7783c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    }
7793c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7803c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    // Otherwise, it's an error.
7814b9c2d235fb9449e249d74f48ecfec601650de93John McCall    S.Diag(opcLoc, diag::err_nosetter_property_incdec)
7824b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << unsigned(RefExpr->isImplicitProperty())
7834b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << unsigned(UnaryOperator::isDecrementOp(opcode))
7844b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << SetterSelector
7854b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << op->getSourceRange();
7863c3b7f90a863af43fa63043d396553ecf205351cJohn McCall    return ExprError();
7873c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  }
7883c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7893c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  // If there is a setter, we definitely want to use it.
7903c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
7914b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // We also need a getter.
7924b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (!findGetter()) {
7934b9c2d235fb9449e249d74f48ecfec601650de93John McCall    assert(RefExpr->isImplicitProperty());
7944b9c2d235fb9449e249d74f48ecfec601650de93John McCall    S.Diag(opcLoc, diag::err_nogetter_property_incdec)
7954b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << unsigned(UnaryOperator::isDecrementOp(opcode))
796a2c91e75c944e3599fb108db600f91c47c8b4342Fariborz Jahanian      << GetterSelector
7974b9c2d235fb9449e249d74f48ecfec601650de93John McCall      << op->getSourceRange();
7984b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return ExprError();
7994b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
8003c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
8014b9c2d235fb9449e249d74f48ecfec601650de93John McCall  return PseudoOpBuilder::buildIncDecOperation(Sc, opcLoc, opcode, op);
8024b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
8033c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
804ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek// ObjCSubscript build stuff.
805ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek//
806ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
807ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// objective-c subscripting-specific behavior for doing lvalue-to-rvalue
808ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// conversion.
809ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// FIXME. Remove this routine if it is proven that no additional
810ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// specifity is needed.
811ebcb57a8d298862c65043e88b2429591ab3c58d3Ted KremenekExprResult ObjCSubscriptOpBuilder::buildRValueOperation(Expr *op) {
812ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  ExprResult result = PseudoOpBuilder::buildRValueOperation(op);
813ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (result.isInvalid()) return ExprError();
814ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  return result;
815ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek}
816ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
817ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// objective-c subscripting-specific  behavior for doing assignments.
818ebcb57a8d298862c65043e88b2429591ab3c58d3Ted KremenekExprResult
819ebcb57a8d298862c65043e88b2429591ab3c58d3Ted KremenekObjCSubscriptOpBuilder::buildAssignmentOperation(Scope *Sc,
820ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SourceLocation opcLoc,
821ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                BinaryOperatorKind opcode,
822ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                Expr *LHS, Expr *RHS) {
823ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  assert(BinaryOperator::isAssignmentOp(opcode));
824ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // There must be a method to do the Index'ed assignment.
825ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (!findAtIndexSetter())
826ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return ExprError();
827ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
828ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // Verify that we can do a compound assignment.
829ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (opcode != BO_Assign && !findAtIndexGetter())
830ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return ExprError();
831ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
832ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  ExprResult result =
833ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  PseudoOpBuilder::buildAssignmentOperation(Sc, opcLoc, opcode, LHS, RHS);
834ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (result.isInvalid()) return ExprError();
835ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
836ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // Various warnings about objc Index'ed assignments in ARC.
8374e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie  if (S.getLangOpts().ObjCAutoRefCount && InstanceBase) {
838ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    S.checkRetainCycles(InstanceBase->getSourceExpr(), RHS);
839ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    S.checkUnsafeExprAssigns(opcLoc, LHS, RHS);
840ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
841ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
842ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  return result;
843ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek}
844ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
845ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// Capture the base object of an Objective-C Index'ed expression.
846ebcb57a8d298862c65043e88b2429591ab3c58d3Ted KremenekExpr *ObjCSubscriptOpBuilder::rebuildAndCaptureObject(Expr *syntacticBase) {
847ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  assert(InstanceBase == 0);
848ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
849ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // Capture base expression in an OVE and rebuild the syntactic
850ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // form to use the OVE as its base expression.
851ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  InstanceBase = capture(RefExpr->getBaseExpr());
852ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  InstanceKey = capture(RefExpr->getKeyExpr());
853ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
854ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  syntacticBase =
855ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ObjCSubscriptRefRebuilder(S, InstanceBase,
856ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                              InstanceKey).rebuild(syntacticBase);
857ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
858ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  return syntacticBase;
859ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek}
860ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
861ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// CheckSubscriptingKind - This routine decide what type
862ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// of indexing represented by "FromE" is being done.
863ebcb57a8d298862c65043e88b2429591ab3c58d3Ted KremenekSema::ObjCSubscriptKind
864ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Sema::CheckSubscriptingKind(Expr *FromE) {
865ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // If the expression already has integral or enumeration type, we're golden.
866ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  QualType T = FromE->getType();
867ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (T->isIntegralOrEnumerationType())
868ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return OS_Array;
869ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
870ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // If we don't have a class type in C++, there's no way we can get an
871ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // expression of integral or enumeration type.
872ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  const RecordType *RecordTy = T->getAs<RecordType>();
873a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian  if (!RecordTy && T->isObjCObjectPointerType())
874ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // All other scalar cases are assumed to be dictionary indexing which
875ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // caller handles, with diagnostics if needed.
876ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return OS_Dictionary;
877a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian  if (!getLangOpts().CPlusPlus ||
878a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian      !RecordTy || RecordTy->isIncompleteType()) {
879ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // No indexing can be done. Issue diagnostics and quit.
880a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian    const Expr *IndexExpr = FromE->IgnoreParenImpCasts();
881a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian    if (isa<StringLiteral>(IndexExpr))
882a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian      Diag(FromE->getExprLoc(), diag::err_objc_subscript_pointer)
883a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian        << T << FixItHint::CreateInsertion(FromE->getExprLoc(), "@");
884a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian    else
885a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian      Diag(FromE->getExprLoc(), diag::err_objc_subscript_type_conversion)
886a78eca20429e55b041bffcea1938cd0df07a6ebdFariborz Jahanian        << T;
887ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return OS_Error;
888ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
889ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
890ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // We must have a complete class type.
891ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (RequireCompleteType(FromE->getExprLoc(), T,
892d10099e5c8238fa0327f03921cf2e3c8975c881eDouglas Gregor                          diag::err_objc_index_incomplete_class_type, FromE))
893ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return OS_Error;
894ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
895ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // Look for a conversion to an integral, enumeration type, or
896ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // objective-C pointer type.
897ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  UnresolvedSet<4> ViableConversions;
898ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  UnresolvedSet<4> ExplicitConversions;
899ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  const UnresolvedSetImpl *Conversions
900ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    = cast<CXXRecordDecl>(RecordTy->getDecl())->getVisibleConversionFunctions();
901ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
902ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  int NoIntegrals=0, NoObjCIdPointers=0;
903ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  SmallVector<CXXConversionDecl *, 4> ConversionDecls;
904ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
905ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  for (UnresolvedSetImpl::iterator I = Conversions->begin(),
906ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek       E = Conversions->end();
907ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek       I != E;
908ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek       ++I) {
909ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    if (CXXConversionDecl *Conversion
910ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        = dyn_cast<CXXConversionDecl>((*I)->getUnderlyingDecl())) {
911ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      QualType CT = Conversion->getConversionType().getNonReferenceType();
912ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      if (CT->isIntegralOrEnumerationType()) {
913ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        ++NoIntegrals;
914ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        ConversionDecls.push_back(Conversion);
915ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      }
916ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      else if (CT->isObjCIdType() ||CT->isBlockPointerType()) {
917ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        ++NoObjCIdPointers;
918ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        ConversionDecls.push_back(Conversion);
919ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      }
920ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    }
921ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
922ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (NoIntegrals ==1 && NoObjCIdPointers == 0)
923ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return OS_Array;
924ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (NoIntegrals == 0 && NoObjCIdPointers == 1)
925ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return OS_Dictionary;
926ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (NoIntegrals == 0 && NoObjCIdPointers == 0) {
927ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // No conversion function was found. Issue diagnostic and return.
928ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    Diag(FromE->getExprLoc(), diag::err_objc_subscript_type_conversion)
929ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      << FromE->getType();
930ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return OS_Error;
931ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
932ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Diag(FromE->getExprLoc(), diag::err_objc_multiple_subscript_type_conversion)
933ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      << FromE->getType();
934ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  for (unsigned int i = 0; i < ConversionDecls.size(); i++)
935ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    Diag(ConversionDecls[i]->getLocation(), diag::not_conv_function_declared_at);
936ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
937ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  return OS_Error;
938ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek}
939ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
940ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenekbool ObjCSubscriptOpBuilder::findAtIndexGetter() {
941ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (AtIndexGetter)
942ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return true;
943ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
944ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Expr *BaseExpr = RefExpr->getBaseExpr();
945ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  QualType BaseT = BaseExpr->getType();
946ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
947ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  QualType ResultType;
948ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (const ObjCObjectPointerType *PTy =
949ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      BaseT->getAs<ObjCObjectPointerType>()) {
950ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ResultType = PTy->getPointeeType();
951ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    if (const ObjCObjectType *iQFaceTy =
952ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        ResultType->getAsObjCQualifiedInterfaceType())
953ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      ResultType = iQFaceTy->getBaseType();
954ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
955ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Sema::ObjCSubscriptKind Res =
956ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    S.CheckSubscriptingKind(RefExpr->getKeyExpr());
957ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (Res == Sema::OS_Error)
958ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return false;
959ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  bool arrayRef = (Res == Sema::OS_Array);
960ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
961ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (ResultType.isNull()) {
962ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    S.Diag(BaseExpr->getExprLoc(), diag::err_objc_subscript_base_type)
963ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      << BaseExpr->getType() << arrayRef;
964ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return false;
965ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
966ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (!arrayRef) {
967ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // dictionary subscripting.
968ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // - (id)objectForKeyedSubscript:(id)key;
969ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    IdentifierInfo *KeyIdents[] = {
970ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      &S.Context.Idents.get("objectForKeyedSubscript")
971ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    };
972ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexGetterSelector = S.Context.Selectors.getSelector(1, KeyIdents);
973ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
974ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  else {
975ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // - (id)objectAtIndexedSubscript:(size_t)index;
976ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    IdentifierInfo *KeyIdents[] = {
977ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      &S.Context.Idents.get("objectAtIndexedSubscript")
978ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    };
979ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
980ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexGetterSelector = S.Context.Selectors.getSelector(1, KeyIdents);
981ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
982ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
983ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  AtIndexGetter = S.LookupMethodInObjectType(AtIndexGetterSelector, ResultType,
984ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                             true /*instance*/);
985ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  bool receiverIdType = (BaseT->isObjCIdType() ||
986ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                         BaseT->isObjCQualifiedIdType());
987ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
9884e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie  if (!AtIndexGetter && S.getLangOpts().DebuggerObjCLiteral) {
989ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexGetter = ObjCMethodDecl::Create(S.Context, SourceLocation(),
990ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           SourceLocation(), AtIndexGetterSelector,
991ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           S.Context.getObjCIdType() /*ReturnType*/,
992ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           0 /*TypeSourceInfo */,
993ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           S.Context.getTranslationUnitDecl(),
994ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           true /*Instance*/, false/*isVariadic*/,
995ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           /*isSynthesized=*/false,
996ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           /*isImplicitlyDeclared=*/true, /*isDefined=*/false,
997ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           ObjCMethodDecl::Required,
998ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           false);
999ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ParmVarDecl *Argument = ParmVarDecl::Create(S.Context, AtIndexGetter,
1000ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SourceLocation(), SourceLocation(),
1001ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                arrayRef ? &S.Context.Idents.get("index")
1002ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                         : &S.Context.Idents.get("key"),
1003ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                arrayRef ? S.Context.UnsignedLongTy
1004ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                         : S.Context.getObjCIdType(),
1005ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                /*TInfo=*/0,
1006ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SC_None,
1007ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SC_None,
1008ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                0);
1009ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexGetter->setMethodParams(S.Context, Argument,
1010ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                   ArrayRef<SourceLocation>());
1011ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1012ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1013ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (!AtIndexGetter) {
1014ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    if (!receiverIdType) {
1015ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(BaseExpr->getExprLoc(), diag::err_objc_subscript_method_not_found)
1016ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      << BaseExpr->getType() << 0 << arrayRef;
1017ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      return false;
1018ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    }
1019ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexGetter =
1020ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.LookupInstanceMethodInGlobalPool(AtIndexGetterSelector,
1021ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                         RefExpr->getSourceRange(),
1022ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                         true, false);
1023ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1024ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1025ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (AtIndexGetter) {
1026ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    QualType T = AtIndexGetter->param_begin()[0]->getType();
1027ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    if ((arrayRef && !T->isIntegralOrEnumerationType()) ||
1028ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        (!arrayRef && !T->isObjCObjectPointerType())) {
1029ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(RefExpr->getKeyExpr()->getExprLoc(),
1030ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek             arrayRef ? diag::err_objc_subscript_index_type
1031ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                      : diag::err_objc_subscript_key_type) << T;
1032ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(AtIndexGetter->param_begin()[0]->getLocation(),
1033ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek             diag::note_parameter_type) << T;
1034ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      return false;
1035ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    }
1036ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    QualType R = AtIndexGetter->getResultType();
1037ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    if (!R->isObjCObjectPointerType()) {
1038ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(RefExpr->getKeyExpr()->getExprLoc(),
1039ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek             diag::err_objc_indexing_method_result_type) << R << arrayRef;
1040ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(AtIndexGetter->getLocation(), diag::note_method_declared_at) <<
1041ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        AtIndexGetter->getDeclName();
1042ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    }
1043ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1044ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  return true;
1045ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek}
1046ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1047ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenekbool ObjCSubscriptOpBuilder::findAtIndexSetter() {
1048ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (AtIndexSetter)
1049ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return true;
1050ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1051ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Expr *BaseExpr = RefExpr->getBaseExpr();
1052ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  QualType BaseT = BaseExpr->getType();
1053ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1054ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  QualType ResultType;
1055ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (const ObjCObjectPointerType *PTy =
1056ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      BaseT->getAs<ObjCObjectPointerType>()) {
1057ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ResultType = PTy->getPointeeType();
1058ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    if (const ObjCObjectType *iQFaceTy =
1059ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        ResultType->getAsObjCQualifiedInterfaceType())
1060ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      ResultType = iQFaceTy->getBaseType();
1061ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1062ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1063ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Sema::ObjCSubscriptKind Res =
1064ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    S.CheckSubscriptingKind(RefExpr->getKeyExpr());
1065ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (Res == Sema::OS_Error)
1066ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return false;
1067ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  bool arrayRef = (Res == Sema::OS_Array);
1068ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1069ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (ResultType.isNull()) {
1070ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    S.Diag(BaseExpr->getExprLoc(), diag::err_objc_subscript_base_type)
1071ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      << BaseExpr->getType() << arrayRef;
1072ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return false;
1073ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1074ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1075ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (!arrayRef) {
1076ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // dictionary subscripting.
1077ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // - (void)setObject:(id)object forKeyedSubscript:(id)key;
1078ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    IdentifierInfo *KeyIdents[] = {
1079ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      &S.Context.Idents.get("setObject"),
1080ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      &S.Context.Idents.get("forKeyedSubscript")
1081ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    };
1082ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexSetterSelector = S.Context.Selectors.getSelector(2, KeyIdents);
1083ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1084ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  else {
1085ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    // - (void)setObject:(id)object atIndexedSubscript:(NSInteger)index;
1086ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    IdentifierInfo *KeyIdents[] = {
1087ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      &S.Context.Idents.get("setObject"),
1088ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      &S.Context.Idents.get("atIndexedSubscript")
1089ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    };
1090ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexSetterSelector = S.Context.Selectors.getSelector(2, KeyIdents);
1091ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1092ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  AtIndexSetter = S.LookupMethodInObjectType(AtIndexSetterSelector, ResultType,
1093ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                             true /*instance*/);
1094ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1095ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  bool receiverIdType = (BaseT->isObjCIdType() ||
1096ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                         BaseT->isObjCQualifiedIdType());
1097ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
10984e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie  if (!AtIndexSetter && S.getLangOpts().DebuggerObjCLiteral) {
1099ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    TypeSourceInfo *ResultTInfo = 0;
1100ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    QualType ReturnType = S.Context.VoidTy;
1101ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexSetter = ObjCMethodDecl::Create(S.Context, SourceLocation(),
1102ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           SourceLocation(), AtIndexSetterSelector,
1103ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           ReturnType,
1104ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           ResultTInfo,
1105ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           S.Context.getTranslationUnitDecl(),
1106ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           true /*Instance*/, false/*isVariadic*/,
1107ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           /*isSynthesized=*/false,
1108ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           /*isImplicitlyDeclared=*/true, /*isDefined=*/false,
1109ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           ObjCMethodDecl::Required,
1110ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                           false);
1111ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    SmallVector<ParmVarDecl *, 2> Params;
1112ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ParmVarDecl *object = ParmVarDecl::Create(S.Context, AtIndexSetter,
1113ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SourceLocation(), SourceLocation(),
1114ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                &S.Context.Idents.get("object"),
1115ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                S.Context.getObjCIdType(),
1116ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                /*TInfo=*/0,
1117ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SC_None,
1118ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SC_None,
1119ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                0);
1120ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    Params.push_back(object);
1121ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ParmVarDecl *key = ParmVarDecl::Create(S.Context, AtIndexSetter,
1122ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SourceLocation(), SourceLocation(),
1123ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                arrayRef ?  &S.Context.Idents.get("index")
1124ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                         :  &S.Context.Idents.get("key"),
1125ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                arrayRef ? S.Context.UnsignedLongTy
1126ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                         : S.Context.getObjCIdType(),
1127ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                /*TInfo=*/0,
1128ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SC_None,
1129ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                SC_None,
1130ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                0);
1131ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    Params.push_back(key);
1132ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexSetter->setMethodParams(S.Context, Params, ArrayRef<SourceLocation>());
1133ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1134ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1135ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (!AtIndexSetter) {
1136ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    if (!receiverIdType) {
1137ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(BaseExpr->getExprLoc(),
1138ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek             diag::err_objc_subscript_method_not_found)
1139ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      << BaseExpr->getType() << 1 << arrayRef;
1140ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      return false;
1141ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    }
1142ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    AtIndexSetter =
1143ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.LookupInstanceMethodInGlobalPool(AtIndexSetterSelector,
1144ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                         RefExpr->getSourceRange(),
1145ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                         true, false);
1146ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1147ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1148ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  bool err = false;
1149ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (AtIndexSetter && arrayRef) {
1150ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    QualType T = AtIndexSetter->param_begin()[1]->getType();
1151ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    if (!T->isIntegralOrEnumerationType()) {
1152ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(RefExpr->getKeyExpr()->getExprLoc(),
1153ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek             diag::err_objc_subscript_index_type) << T;
1154ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(AtIndexSetter->param_begin()[1]->getLocation(),
1155ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek             diag::note_parameter_type) << T;
1156ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      err = true;
1157ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    }
1158ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    T = AtIndexSetter->param_begin()[0]->getType();
1159ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    if (!T->isObjCObjectPointerType()) {
1160ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(RefExpr->getBaseExpr()->getExprLoc(),
1161ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek             diag::err_objc_subscript_object_type) << T << arrayRef;
1162ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      S.Diag(AtIndexSetter->param_begin()[0]->getLocation(),
1163ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek             diag::note_parameter_type) << T;
1164ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      err = true;
1165ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    }
1166ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1167ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  else if (AtIndexSetter && !arrayRef)
1168ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    for (unsigned i=0; i <2; i++) {
1169ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      QualType T = AtIndexSetter->param_begin()[i]->getType();
1170ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      if (!T->isObjCObjectPointerType()) {
1171ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        if (i == 1)
1172ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek          S.Diag(RefExpr->getKeyExpr()->getExprLoc(),
1173ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                 diag::err_objc_subscript_key_type) << T;
1174ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        else
1175ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek          S.Diag(RefExpr->getBaseExpr()->getExprLoc(),
1176ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                 diag::err_objc_subscript_dic_object_type) << T;
1177ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        S.Diag(AtIndexSetter->param_begin()[i]->getLocation(),
1178ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek               diag::note_parameter_type) << T;
1179ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek        err = true;
1180ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      }
1181ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    }
1182ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1183ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  return !err;
1184ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek}
1185ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1186ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek// Get the object at "Index" position in the container.
1187ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek// [BaseExpr objectAtIndexedSubscript : IndexExpr];
1188ebcb57a8d298862c65043e88b2429591ab3c58d3Ted KremenekExprResult ObjCSubscriptOpBuilder::buildGet() {
1189ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (!findAtIndexGetter())
1190ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return ExprError();
1191ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1192ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  QualType receiverType = InstanceBase->getType();
1193ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1194ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // Build a message-send.
1195ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  ExprResult msg;
1196ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Expr *Index = InstanceKey;
1197ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1198ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // Arguments.
1199ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Expr *args[] = { Index };
1200ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  assert(InstanceBase);
1201ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  msg = S.BuildInstanceMessageImplicit(InstanceBase, receiverType,
1202ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                       GenericLoc,
1203ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                       AtIndexGetterSelector, AtIndexGetter,
1204ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                       MultiExprArg(args, 1));
1205ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  return msg;
1206ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek}
1207ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1208ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// Store into the container the "op" object at "Index"'ed location
1209ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// by building this messaging expression:
1210ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// - (void)setObject:(id)object atIndexedSubscript:(NSInteger)index;
1211ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek/// \param bindSetValueAsResult - If true, capture the actual
1212ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek///   value being set as the value of the property operation.
1213ebcb57a8d298862c65043e88b2429591ab3c58d3Ted KremenekExprResult ObjCSubscriptOpBuilder::buildSet(Expr *op, SourceLocation opcLoc,
1214ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                           bool captureSetValueAsResult) {
1215ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (!findAtIndexSetter())
1216ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return ExprError();
1217ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1218ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  QualType receiverType = InstanceBase->getType();
1219ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Expr *Index = InstanceKey;
1220ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1221ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // Arguments.
1222ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  Expr *args[] = { op, Index };
1223ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1224ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  // Build a message-send.
1225ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  ExprResult msg = S.BuildInstanceMessageImplicit(InstanceBase, receiverType,
1226ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                  GenericLoc,
1227ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                  AtIndexSetterSelector,
1228ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                  AtIndexSetter,
1229ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                                  MultiExprArg(args, 2));
1230ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1231ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  if (!msg.isInvalid() && captureSetValueAsResult) {
1232ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ObjCMessageExpr *msgExpr =
1233ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek      cast<ObjCMessageExpr>(msg.get()->IgnoreImplicit());
1234ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    Expr *arg = msgExpr->getArg(0);
1235ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    msgExpr->setArg(0, captureValueAsResult(arg));
1236ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1237ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
1238ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  return msg;
1239ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek}
1240ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek
12414b9c2d235fb9449e249d74f48ecfec601650de93John McCall//===----------------------------------------------------------------------===//
12424b9c2d235fb9449e249d74f48ecfec601650de93John McCall//  General Sema routines.
12434b9c2d235fb9449e249d74f48ecfec601650de93John McCall//===----------------------------------------------------------------------===//
12443c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
12454b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult Sema::checkPseudoObjectRValue(Expr *E) {
12464b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Expr *opaqueRef = E->IgnoreParens();
12474b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (ObjCPropertyRefExpr *refExpr
12484b9c2d235fb9449e249d74f48ecfec601650de93John McCall        = dyn_cast<ObjCPropertyRefExpr>(opaqueRef)) {
12494b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ObjCPropertyOpBuilder builder(*this, refExpr);
12504b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return builder.buildRValueOperation(E);
1251ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  }
1252ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  else if (ObjCSubscriptRefExpr *refExpr
1253ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek           = dyn_cast<ObjCSubscriptRefExpr>(opaqueRef)) {
1254ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ObjCSubscriptOpBuilder builder(*this, refExpr);
1255ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return builder.buildRValueOperation(E);
12564b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else {
12574b9c2d235fb9449e249d74f48ecfec601650de93John McCall    llvm_unreachable("unknown pseudo-object kind!");
12583c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  }
12594b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
12603c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
12614b9c2d235fb9449e249d74f48ecfec601650de93John McCall/// Check an increment or decrement of a pseudo-object expression.
12624b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult Sema::checkPseudoObjectIncDec(Scope *Sc, SourceLocation opcLoc,
12634b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                         UnaryOperatorKind opcode, Expr *op) {
12644b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Do nothing if the operand is dependent.
12654b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (op->isTypeDependent())
12664b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return new (Context) UnaryOperator(op, opcode, Context.DependentTy,
12674b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                       VK_RValue, OK_Ordinary, opcLoc);
12684b9c2d235fb9449e249d74f48ecfec601650de93John McCall
12694b9c2d235fb9449e249d74f48ecfec601650de93John McCall  assert(UnaryOperator::isIncrementDecrementOp(opcode));
12704b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Expr *opaqueRef = op->IgnoreParens();
12714b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (ObjCPropertyRefExpr *refExpr
12724b9c2d235fb9449e249d74f48ecfec601650de93John McCall        = dyn_cast<ObjCPropertyRefExpr>(opaqueRef)) {
12734b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ObjCPropertyOpBuilder builder(*this, refExpr);
12744b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return builder.buildIncDecOperation(Sc, opcLoc, opcode, op);
1275ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  } else if (isa<ObjCSubscriptRefExpr>(opaqueRef)) {
1276ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    Diag(opcLoc, diag::err_illegal_container_subscripting_op);
1277ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return ExprError();
12784b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else {
12794b9c2d235fb9449e249d74f48ecfec601650de93John McCall    llvm_unreachable("unknown pseudo-object kind!");
12803c3b7f90a863af43fa63043d396553ecf205351cJohn McCall  }
12814b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
12823c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
12834b9c2d235fb9449e249d74f48ecfec601650de93John McCallExprResult Sema::checkPseudoObjectAssignment(Scope *S, SourceLocation opcLoc,
12844b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             BinaryOperatorKind opcode,
12854b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                             Expr *LHS, Expr *RHS) {
12864b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Do nothing if either argument is dependent.
12874b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (LHS->isTypeDependent() || RHS->isTypeDependent())
12884b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return new (Context) BinaryOperator(LHS, RHS, opcode, Context.DependentTy,
12894b9c2d235fb9449e249d74f48ecfec601650de93John McCall                                        VK_RValue, OK_Ordinary, opcLoc);
12904b9c2d235fb9449e249d74f48ecfec601650de93John McCall
12914b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Filter out non-overload placeholder types in the RHS.
129232509f1e60451d86e9fbc473b6e853ba10b5fd1eJohn McCall  if (RHS->getType()->isNonOverloadPlaceholderType()) {
129332509f1e60451d86e9fbc473b6e853ba10b5fd1eJohn McCall    ExprResult result = CheckPlaceholderExpr(RHS);
129432509f1e60451d86e9fbc473b6e853ba10b5fd1eJohn McCall    if (result.isInvalid()) return ExprError();
129532509f1e60451d86e9fbc473b6e853ba10b5fd1eJohn McCall    RHS = result.take();
12964b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
12973c3b7f90a863af43fa63043d396553ecf205351cJohn McCall
12984b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Expr *opaqueRef = LHS->IgnoreParens();
12994b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (ObjCPropertyRefExpr *refExpr
13004b9c2d235fb9449e249d74f48ecfec601650de93John McCall        = dyn_cast<ObjCPropertyRefExpr>(opaqueRef)) {
13014b9c2d235fb9449e249d74f48ecfec601650de93John McCall    ObjCPropertyOpBuilder builder(*this, refExpr);
13024b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return builder.buildAssignmentOperation(S, opcLoc, opcode, LHS, RHS);
1303ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  } else if (ObjCSubscriptRefExpr *refExpr
1304ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek             = dyn_cast<ObjCSubscriptRefExpr>(opaqueRef)) {
1305ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    ObjCSubscriptOpBuilder builder(*this, refExpr);
1306ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return builder.buildAssignmentOperation(S, opcLoc, opcode, LHS, RHS);
13074b9c2d235fb9449e249d74f48ecfec601650de93John McCall  } else {
13084b9c2d235fb9449e249d74f48ecfec601650de93John McCall    llvm_unreachable("unknown pseudo-object kind!");
13094b9c2d235fb9449e249d74f48ecfec601650de93John McCall  }
13103c3b7f90a863af43fa63043d396553ecf205351cJohn McCall}
131101e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall
131201e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall/// Given a pseudo-object reference, rebuild it without the opaque
131301e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall/// values.  Basically, undo the behavior of rebuildAndCaptureObject.
131401e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall/// This should never operate in-place.
131501e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCallstatic Expr *stripOpaqueValuesFromPseudoObjectRef(Sema &S, Expr *E) {
131601e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  Expr *opaqueRef = E->IgnoreParens();
131701e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  if (ObjCPropertyRefExpr *refExpr
131801e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall        = dyn_cast<ObjCPropertyRefExpr>(opaqueRef)) {
131988507ddf6f83b929454c04470924cf56b4b8147aDouglas Gregor    // Class and super property references don't have opaque values in them.
132088507ddf6f83b929454c04470924cf56b4b8147aDouglas Gregor    if (refExpr->isClassReceiver() || refExpr->isSuperReceiver())
132188507ddf6f83b929454c04470924cf56b4b8147aDouglas Gregor      return E;
132288507ddf6f83b929454c04470924cf56b4b8147aDouglas Gregor
132388507ddf6f83b929454c04470924cf56b4b8147aDouglas Gregor    assert(refExpr->isObjectReceiver() && "Unknown receiver kind?");
132488507ddf6f83b929454c04470924cf56b4b8147aDouglas Gregor    OpaqueValueExpr *baseOVE = cast<OpaqueValueExpr>(refExpr->getBase());
132588507ddf6f83b929454c04470924cf56b4b8147aDouglas Gregor    return ObjCPropertyRefRebuilder(S, baseOVE->getSourceExpr()).rebuild(E);
1326ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek  } else if (ObjCSubscriptRefExpr *refExpr
1327ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek               = dyn_cast<ObjCSubscriptRefExpr>(opaqueRef)) {
1328ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    OpaqueValueExpr *baseOVE = cast<OpaqueValueExpr>(refExpr->getBaseExpr());
1329ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    OpaqueValueExpr *keyOVE = cast<OpaqueValueExpr>(refExpr->getKeyExpr());
1330ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek    return ObjCSubscriptRefRebuilder(S, baseOVE->getSourceExpr(),
1331ebcb57a8d298862c65043e88b2429591ab3c58d3Ted Kremenek                                     keyOVE->getSourceExpr()).rebuild(E);
133201e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  } else {
133301e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    llvm_unreachable("unknown pseudo-object kind!");
133401e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  }
133501e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall}
133601e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall
133701e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall/// Given a pseudo-object expression, recreate what it looks like
133801e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall/// syntactically without the attendant OpaqueValueExprs.
133901e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall///
134001e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall/// This is a hack which should be removed when TreeTransform is
134101e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall/// capable of rebuilding a tree without stripping implicit
134201e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall/// operations.
134301e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCallExpr *Sema::recreateSyntacticForm(PseudoObjectExpr *E) {
134401e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  Expr *syntax = E->getSyntacticForm();
134501e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  if (UnaryOperator *uop = dyn_cast<UnaryOperator>(syntax)) {
134601e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    Expr *op = stripOpaqueValuesFromPseudoObjectRef(*this, uop->getSubExpr());
134701e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    return new (Context) UnaryOperator(op, uop->getOpcode(), uop->getType(),
134801e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                       uop->getValueKind(), uop->getObjectKind(),
134901e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                       uop->getOperatorLoc());
135001e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  } else if (CompoundAssignOperator *cop
135101e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall               = dyn_cast<CompoundAssignOperator>(syntax)) {
135201e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    Expr *lhs = stripOpaqueValuesFromPseudoObjectRef(*this, cop->getLHS());
135301e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    Expr *rhs = cast<OpaqueValueExpr>(cop->getRHS())->getSourceExpr();
135401e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    return new (Context) CompoundAssignOperator(lhs, rhs, cop->getOpcode(),
135501e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                                cop->getType(),
135601e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                                cop->getValueKind(),
135701e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                                cop->getObjectKind(),
135801e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                                cop->getComputationLHSType(),
135901e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                                cop->getComputationResultType(),
136001e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                                cop->getOperatorLoc());
136101e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  } else if (BinaryOperator *bop = dyn_cast<BinaryOperator>(syntax)) {
136201e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    Expr *lhs = stripOpaqueValuesFromPseudoObjectRef(*this, bop->getLHS());
136301e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    Expr *rhs = cast<OpaqueValueExpr>(bop->getRHS())->getSourceExpr();
136401e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    return new (Context) BinaryOperator(lhs, rhs, bop->getOpcode(),
136501e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                        bop->getType(), bop->getValueKind(),
136601e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                        bop->getObjectKind(),
136701e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall                                        bop->getOperatorLoc());
136801e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  } else {
136901e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    assert(syntax->hasPlaceholderType(BuiltinType::PseudoObject));
137001e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall    return stripOpaqueValuesFromPseudoObjectRef(*this, syntax);
137101e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall  }
137201e19be69a37bc4ab7746c454cfaa6aec7bb7c6aJohn McCall}
1373