Support for mixed null/smi equality: do not deoptimize, emit same optimized code as if that was smi…

runtime/vm/intermediate_language_ia32.cc

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "vm/globals.h"  // Needed here to get TARGET_ARCH_IA32.
 #if defined(TARGET_ARCH_IA32)
 
 #include "vm/intermediate_language.h"
 
 #include "lib/error.h"
@@ skipping 293 matching lines @@
   if (HasICData() && (ic_data()->NumberOfChecks() > 0)) {
     const intptr_t kNumTemps = 1;
     LocationSummary* locs =
         new LocationSummary(kNumInputs, kNumTemps, LocationSummary::kCall);
     locs->set_in(0, Location::RegisterLocation(ECX));
     locs->set_in(1, Location::RegisterLocation(EDX));
     locs->set_temp(0, Location::RegisterLocation(EBX));
     locs->set_out(Location::RegisterLocation(EAX));
     return locs;
   }
-  const intptr_t kNumTemps = 0;
+  const intptr_t kNumTemps = 1;
   LocationSummary* locs =
       new LocationSummary(kNumInputs, kNumTemps, LocationSummary::kCall);
-  locs->set_in(0, Location::RegisterLocation(ECX));
+  locs->set_in(0, Location::RegisterLocation(EBX));
   locs->set_in(1, Location::RegisterLocation(EDX));
+  locs->set_temp(0, Location::RegisterLocation(ECX));
   locs->set_out(Location::RegisterLocation(EAX));
   return locs;
 }
 
 
 static void EmitEqualityAsInstanceCall(FlowGraphCompiler* compiler,
                                        intptr_t deopt_id,
                                        intptr_t token_pos,
                                        Token::Kind kind,
                                        LocationSummary* locs) {
   if (!compiler->is_optimizing()) {
     compiler->AddCurrentDescriptor(PcDescriptors::kDeoptBefore,
                                    deopt_id,
                                    token_pos);
   }
   const String& operator_name = String::ZoneHandle(Symbols::New("=="));
   const int kNumberOfArguments = 2;
   const Array& kNoArgumentNames = Array::Handle();
   const int kNumArgumentsChecked = 2;
 
-  Label done, false_label, true_label;
-  Register left = locs->in(0).reg();
-  Register right = locs->in(1).reg();
-  __ popl(right);
-  __ popl(left);
   const Immediate raw_null =
       Immediate(reinterpret_cast<intptr_t>(Object::null()));
-  Label check_identity, instance_call;
-  __ cmpl(right, raw_null);
+  Label check_identity;
+  __ cmpl(Address(ESP, 0 * kWordSize), raw_null);
   __ j(EQUAL, &check_identity, Assembler::kNearJump);
-  __ cmpl(left, raw_null);
-  __ j(NOT_EQUAL, &instance_call, Assembler::kNearJump);
+  __ cmpl(Address(ESP, 1 * kWordSize), raw_null);
+  __ j(EQUAL, &check_identity, Assembler::kNearJump);
+
+  const ICData& ic_data = compiler->GenerateInstanceCall(deopt_id,
+                                                         token_pos,
+                                                         operator_name,
+                                                         kNumberOfArguments,
+                                                         kNoArgumentNames,
+                                                         kNumArgumentsChecked,
+                                                         locs);
+  Label check_ne;
+  __ jmp(&check_ne);
 
   __ Bind(&check_identity);
-  __ cmpl(left, right);
-  __ j(EQUAL, &true_label);
-  if (kind == Token::kEQ) {
-    __ LoadObject(EAX, compiler->bool_false());
-    __ jmp(&done);
-    __ Bind(&true_label);
-    __ LoadObject(EAX, compiler->bool_true());
-    __ jmp(&done);
-  } else {
-    ASSERT(kind == Token::kNE);
-    __ jmp(&false_label);
-  }
-
-  __ Bind(&instance_call);
-  __ pushl(left);
-  __ pushl(right);
-  compiler->GenerateInstanceCall(deopt_id,
-                                 token_pos,
-                                 operator_name,
-                                 kNumberOfArguments,
-                                 kNoArgumentNames,
-                                 kNumArgumentsChecked,
-                                 locs);
+  // Call stub, load IC data in register. The stub will update ICData if
+  // necessary.
+  Register ic_data_reg = locs->temp(0).reg();
+  ASSERT(ic_data_reg == ECX);  // Stub depends on it.
+  __ LoadObject(ic_data_reg, ic_data);
+  compiler->GenerateCall(token_pos,
+                         &StubCode::EqualityWithNullArgLabel(),
+                         PcDescriptors::kOther,
+                         locs);
+  __ Drop(2);
+  __ Bind(&check_ne);
   if (kind == Token::kNE) {
+    Label false_label, true_label, done;
     // Negate the condition: true label returns false and vice versa.
     __ CompareObject(EAX, compiler->bool_true());
     __ j(EQUAL, &true_label, Assembler::kNearJump);
     __ Bind(&false_label);
     __ LoadObject(EAX, compiler->bool_true());
     __ jmp(&done, Assembler::kNearJump);
     __ Bind(&true_label);
     __ LoadObject(EAX, compiler->bool_false());
+    __ Bind(&done);
   }
-  __ Bind(&done);
 }
 
 
 static void EmitEqualityAsPolymorphicCall(FlowGraphCompiler* compiler,
                                           const ICData& orig_ic_data,
                                           LocationSummary* locs,
                                           BranchInstr* branch,
                                           Token::Kind kind,
                                           intptr_t deopt_id,
                                           intptr_t token_pos) {
@@ skipping 180 matching lines @@
 static void EmitSmiComparisonOp(FlowGraphCompiler* compiler,
                                 const LocationSummary& locs,
                                 Token::Kind kind,
                                 BranchInstr* branch) {
   Location left = locs.in(0);
   Location right = locs.in(1);
 
   Condition true_condition = TokenKindToSmiCondition(kind);
 
   if (left.IsConstant() && right.IsConstant()) {
+    bool result = false;
+    // One of them could be NULL (for equality only).
+    if (left.constant().IsNull() || right.constant().IsNull()) {
+      ASSERT((kind == Token::kEQ) || (kind == Token::kNE));
+      result = left.constant().IsNull() && right.constant().IsNull();
+      if (kind == Token::kNE) {
+        result = !result;
+      }
+    } else {
     // TODO(vegorov): should be eliminated earlier by constant propagation.
-    const bool result = FlowGraphCompiler::EvaluateCondition(
-        true_condition,
-        Smi::Cast(left.constant()).Value(),
-        Smi::Cast(right.constant()).Value());
+      result = FlowGraphCompiler::EvaluateCondition(
+          true_condition,
+          Smi::Cast(left.constant()).Value(),
+          Smi::Cast(right.constant()).Value());
+    }
 
     if (branch != NULL) {
       branch->EmitBranchOnValue(compiler, result);
     } else {
       __ LoadObject(locs.out().reg(), result ? compiler->bool_true()
                                              : compiler->bool_false());
     }
 
     return;
   }
@@ skipping 1471 matching lines @@
   summary->set_temp(0, Location::RequiresRegister());
   return summary;
 }
 
 
 void CheckClassInstr::EmitNativeCode(FlowGraphCompiler* compiler) {
   Register value = locs()->in(0).reg();
   Register temp = locs()->temp(0).reg();
   Label* deopt = compiler->AddDeoptStub(deopt_id(),
                                         kDeoptCheckClass);
-  ASSERT(unary_checks().GetReceiverClassIdAt(0) != kSmiCid);
-  __ testl(value, Immediate(kSmiTagMask));
-  __ j(ZERO, deopt);
+  Label is_ok;
+  intptr_t cix = 0;
+  if (unary_checks().GetReceiverClassIdAt(cix) == kSmiCid) {
+    __ testl(value, Immediate(kSmiTagMask));
+    __ j(ZERO, &is_ok);
+    cix++;  // Skip first check.
+  } else {
+    __ testl(value, Immediate(kSmiTagMask));
+    __ j(ZERO, deopt);
+  }
   __ LoadClassId(temp, value);
-  Label is_ok;
   const intptr_t num_checks = unary_checks().NumberOfChecks();
   const bool use_near_jump = num_checks < 5;
-  for (intptr_t i = 0; i < num_checks; i++) {
+  for (intptr_t i = cix; i < num_checks; i++) {
+    ASSERT(unary_checks().GetReceiverClassIdAt(i) != kSmiCid);
     __ cmpl(temp, Immediate(unary_checks().GetReceiverClassIdAt(i)));
     if (i == (num_checks - 1)) {
       __ j(NOT_EQUAL, deopt);
     } else {
       if (use_near_jump) {
         __ j(EQUAL, &is_ok, Assembler::kNearJump);
       } else {
         __ j(EQUAL, &is_ok);
       }
     }
@@ skipping 298 matching lines @@
   __ pcmpeqq(XMM0, XMM0);  // Generate all 1's.
   __ pxor(value, XMM0);
 }
 
 
 }  // namespace dart
 
 #undef __
 
 #endif  // defined TARGET_ARCH_X64