src/3rdparty/v8/src/x64/regexp-macro-assembler-x64.cc

   1 // Copyright 2011 the V8 project authors. All rights reserved.
   2 // Redistribution and use in source and binary forms, with or without
   3 // modification, are permitted provided that the following conditions are
   4 // met:
   5 //
   6 //     * Redistributions of source code must retain the above copyright
   7 //       notice, this list of conditions and the following disclaimer.
   8 //     * Redistributions in binary form must reproduce the above
   9 //       copyright notice, this list of conditions and the following
  10 //       disclaimer in the documentation and/or other materials provided
  11 //       with the distribution.
  12 //     * Neither the name of Google Inc. nor the names of its
  13 //       contributors may be used to endorse or promote products derived
  14 //       from this software without specific prior written permission.
  15 //
  16 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  17 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  18 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  19 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  20 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  21 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  22 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  23 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  24 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  26 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27
  28 #include "v8.h"
  29
  30 #if defined(V8_TARGET_ARCH_X64)
  31
  32 #include "serialize.h"
  33 #include "unicode.h"
  34 #include "log.h"
  35 #include "regexp-stack.h"
  36 #include "macro-assembler.h"
  37 #include "regexp-macro-assembler.h"
  38 #include "x64/regexp-macro-assembler-x64.h"
  39
  40 namespace v8 {
  41 namespace internal {
  42
  43 #ifndef V8_INTERPRETED_REGEXP
  44
  45 /*
  46  * This assembler uses the following register assignment convention
  47  * - rdx : currently loaded character(s) as ASCII or UC16. Must be loaded using
  48  *         LoadCurrentCharacter before using any of the dispatch methods.
  49  * - rdi : current position in input, as negative offset from end of string.
  50  *         Please notice that this is the byte offset, not the character
  51  *         offset! Is always a 32-bit signed (negative) offset, but must be
  52  *         maintained sign-extended to 64 bits, since it is used as index.
  53  * - rsi : end of input (points to byte after last character in input),
  54  *         so that rsi+rdi points to the current character.
  55  * - rbp : frame pointer. Used to access arguments, local variables and
  56  *         RegExp registers.
  57  * - rsp : points to tip of C stack.
  58  * - rcx : points to tip of backtrack stack. The backtrack stack contains
  59  *         only 32-bit values. Most are offsets from some base (e.g., character
  60  *         positions from end of string or code location from Code* pointer).
  61  * - r8  : code object pointer. Used to convert between absolute and
  62  *         code-object-relative addresses.
  63  *
  64  * The registers rax, rbx, r9 and r11 are free to use for computations.
  65  * If changed to use r12+, they should be saved as callee-save registers.
  66  * The macro assembler special registers r12 and r13 (kSmiConstantRegister,
  67  * kRootRegister) aren't special during execution of RegExp code (they don't
  68  * hold the values assumed when creating JS code), so no Smi or Root related
  69  * macro operations can be used.
  70  *
  71  * Each call to a C++ method should retain these registers.
  72  *
  73  * The stack will have the following content, in some order, indexable from the
  74  * frame pointer (see, e.g., kStackHighEnd):
  75  *    - Isolate* isolate     (Address of the current isolate)
  76  *    - direct_call          (if 1, direct call from JavaScript code, if 0 call
  77  *                            through the runtime system)
  78  *    - stack_area_base      (High end of the memory area to use as
  79  *                            backtracking stack)
  80  *    - int* capture_array   (int[num_saved_registers_], for output).
  81  *    - end of input         (Address of end of string)
  82  *    - start of input       (Address of first character in string)
  83  *    - start index          (character index of start)
  84  *    - String* input_string (input string)
  85  *    - return address
  86  *    - backup of callee save registers (rbx, possibly rsi and rdi).
  87  *    - Offset of location before start of input (effectively character
  88  *      position -1). Used to initialize capture registers to a non-position.
  89  *    - At start of string (if 1, we are starting at the start of the
  90  *      string, otherwise 0)
  91  *    - register 0  rbp[-n]   (Only positions must be stored in the first
  92  *    - register 1  rbp[-n-8]  num_saved_registers_ registers)
  93  *    - ...
  94  *
  95  * The first num_saved_registers_ registers are initialized to point to
  96  * "character -1" in the string (i.e., char_size() bytes before the first
  97  * character of the string). The remaining registers starts out uninitialized.
  98  *
  99  * The first seven values must be provided by the calling code by
 100  * calling the code's entry address cast to a function pointer with the
 101  * following signature:
 102  * int (*match)(String* input_string,
 103  *              int start_index,
 104  *              Address start,
 105  *              Address end,
 106  *              int* capture_output_array,
 107  *              bool at_start,
 108  *              byte* stack_area_base,
 109  *              bool direct_call)
 110  */
 111
 112 #define __ ACCESS_MASM((&masm_))
 113
 114 RegExpMacroAssemblerX64::RegExpMacroAssemblerX64(
 115     Mode mode,
 116     int registers_to_save)
 117     : masm_(Isolate::Current(), NULL, kRegExpCodeSize),
 118       no_root_array_scope_(&masm_),
 119       code_relative_fixup_positions_(4),
 120       mode_(mode),
 121       num_registers_(registers_to_save),
 122       num_saved_registers_(registers_to_save),
 123       entry_label_(),
 124       start_label_(),
 125       success_label_(),
 126       backtrack_label_(),
 127       exit_label_() {
 128   ASSERT_EQ(0, registers_to_save % 2);
 129   __ jmp(&entry_label_);   // We'll write the entry code when we know more.
 130   __ bind(&start_label_);  // And then continue from here.
 131 }
 132
 133
 134 RegExpMacroAssemblerX64::~RegExpMacroAssemblerX64() {
 135   // Unuse labels in case we throw away the assembler without calling GetCode.
 136   entry_label_.Unuse();
 137   start_label_.Unuse();
 138   success_label_.Unuse();
 139   backtrack_label_.Unuse();
 140   exit_label_.Unuse();
 141   check_preempt_label_.Unuse();
 142   stack_overflow_label_.Unuse();
 143 }
 144
 145
 146 int RegExpMacroAssemblerX64::stack_limit_slack()  {
 147   return RegExpStack::kStackLimitSlack;
 148 }
 149
 150
 151 void RegExpMacroAssemblerX64::AdvanceCurrentPosition(int by) {
 152   if (by != 0) {
 153     __ addq(rdi, Immediate(by * char_size()));
 154   }
 155 }
 156
 157
 158 void RegExpMacroAssemblerX64::AdvanceRegister(int reg, int by) {
 159   ASSERT(reg >= 0);
 160   ASSERT(reg < num_registers_);
 161   if (by != 0) {
 162     __ addq(register_location(reg), Immediate(by));
 163   }
 164 }
 165
 166
 167 void RegExpMacroAssemblerX64::Backtrack() {
 168   CheckPreemption();
 169   // Pop Code* offset from backtrack stack, add Code* and jump to location.
 170   Pop(rbx);
 171   __ addq(rbx, code_object_pointer());
 172   __ jmp(rbx);
 173 }
 174
 175
 176 void RegExpMacroAssemblerX64::Bind(Label* label) {
 177   __ bind(label);
 178 }
 179
 180
 181 void RegExpMacroAssemblerX64::CheckCharacter(uint32_t c, Label* on_equal) {
 182   __ cmpl(current_character(), Immediate(c));
 183   BranchOrBacktrack(equal, on_equal);
 184 }
 185
 186
 187 void RegExpMacroAssemblerX64::CheckCharacterGT(uc16 limit, Label* on_greater) {
 188   __ cmpl(current_character(), Immediate(limit));
 189   BranchOrBacktrack(greater, on_greater);
 190 }
 191
 192
 193 void RegExpMacroAssemblerX64::CheckAtStart(Label* on_at_start) {
 194   Label not_at_start;
 195   // Did we start the match at the start of the string at all?
 196   __ cmpl(Operand(rbp, kStartIndex), Immediate(0));
 197   BranchOrBacktrack(not_equal, &not_at_start);
 198   // If we did, are we still at the start of the input?
 199   __ lea(rax, Operand(rsi, rdi, times_1, 0));
 200   __ cmpq(rax, Operand(rbp, kInputStart));
 201   BranchOrBacktrack(equal, on_at_start);
 202   __ bind(&not_at_start);
 203 }
 204
 205
 206 void RegExpMacroAssemblerX64::CheckNotAtStart(Label* on_not_at_start) {
 207   // Did we start the match at the start of the string at all?
 208   __ cmpl(Operand(rbp, kStartIndex), Immediate(0));
 209   BranchOrBacktrack(not_equal, on_not_at_start);
 210   // If we did, are we still at the start of the input?
 211   __ lea(rax, Operand(rsi, rdi, times_1, 0));
 212   __ cmpq(rax, Operand(rbp, kInputStart));
 213   BranchOrBacktrack(not_equal, on_not_at_start);
 214 }
 215
 216
 217 void RegExpMacroAssemblerX64::CheckCharacterLT(uc16 limit, Label* on_less) {
 218   __ cmpl(current_character(), Immediate(limit));
 219   BranchOrBacktrack(less, on_less);
 220 }
 221
 222
 223 void RegExpMacroAssemblerX64::CheckCharacters(Vector<const uc16> str,
 224                                               int cp_offset,
 225                                               Label* on_failure,
 226                                               bool check_end_of_string) {
 227 #ifdef DEBUG
 228   // If input is ASCII, don't even bother calling here if the string to
 229   // match contains a non-ASCII character.
 230   if (mode_ == ASCII) {
 231     ASSERT(String::IsAscii(str.start(), str.length()));
 232   }
 233 #endif
 234   int byte_length = str.length() * char_size();
 235   int byte_offset = cp_offset * char_size();
 236   if (check_end_of_string) {
 237     // Check that there are at least str.length() characters left in the input.
 238     __ cmpl(rdi, Immediate(-(byte_offset + byte_length)));
 239     BranchOrBacktrack(greater, on_failure);
 240   }
 241
 242   if (on_failure == NULL) {
 243     // Instead of inlining a backtrack, (re)use the global backtrack target.
 244     on_failure = &backtrack_label_;
 245   }
 246
 247   // Do one character test first to minimize loading for the case that
 248   // we don't match at all (loading more than one character introduces that
 249   // chance of reading unaligned and reading across cache boundaries).
 250   // If the first character matches, expect a larger chance of matching the
 251   // string, and start loading more characters at a time.
 252   if (mode_ == ASCII) {
 253     __ cmpb(Operand(rsi, rdi, times_1, byte_offset),
 254             Immediate(static_cast<int8_t>(str[0])));
 255   } else {
 256     // Don't use 16-bit immediate. The size changing prefix throws off
 257     // pre-decoding.
 258     __ movzxwl(rax,
 259                Operand(rsi, rdi, times_1, byte_offset));
 260     __ cmpl(rax, Immediate(static_cast<int32_t>(str[0])));
 261   }
 262   BranchOrBacktrack(not_equal, on_failure);
 263
 264   __ lea(rbx, Operand(rsi, rdi, times_1, 0));
 265   for (int i = 1, n = str.length(); i < n; ) {
 266     if (mode_ == ASCII) {
 267       if (i + 8 <= n) {
 268         uint64_t combined_chars =
 269             (static_cast<uint64_t>(str[i + 0]) << 0) ||
 270             (static_cast<uint64_t>(str[i + 1]) << 8) ||
 271             (static_cast<uint64_t>(str[i + 2]) << 16) ||
 272             (static_cast<uint64_t>(str[i + 3]) << 24) ||
 273             (static_cast<uint64_t>(str[i + 4]) << 32) ||
 274             (static_cast<uint64_t>(str[i + 5]) << 40) ||
 275             (static_cast<uint64_t>(str[i + 6]) << 48) ||
 276             (static_cast<uint64_t>(str[i + 7]) << 56);
 277         __ movq(rax, combined_chars, RelocInfo::NONE);
 278         __ cmpq(rax, Operand(rbx, byte_offset + i));
 279         i += 8;
 280       } else if (i + 4 <= n) {
 281         uint32_t combined_chars =
 282             (static_cast<uint32_t>(str[i + 0]) << 0) ||
 283             (static_cast<uint32_t>(str[i + 1]) << 8) ||
 284             (static_cast<uint32_t>(str[i + 2]) << 16) ||
 285             (static_cast<uint32_t>(str[i + 3]) << 24);
 286         __ cmpl(Operand(rbx, byte_offset + i), Immediate(combined_chars));
 287         i += 4;
 288       } else {
 289         __ cmpb(Operand(rbx, byte_offset + i),
 290                 Immediate(static_cast<int8_t>(str[i])));
 291         i++;
 292       }
 293     } else {
 294       ASSERT(mode_ == UC16);
 295       if (i + 4 <= n) {
 296         uint64_t combined_chars = *reinterpret_cast<const uint64_t*>(&str[i]);
 297         __ movq(rax, combined_chars, RelocInfo::NONE);
 298         __ cmpq(rax,
 299                 Operand(rsi, rdi, times_1, byte_offset + i * sizeof(uc16)));
 300         i += 4;
 301       } else if (i + 2 <= n) {
 302         uint32_t combined_chars = *reinterpret_cast<const uint32_t*>(&str[i]);
 303         __ cmpl(Operand(rsi, rdi, times_1, byte_offset + i * sizeof(uc16)),
 304                 Immediate(combined_chars));
 305         i += 2;
 306       } else {
 307         __ movzxwl(rax,
 308                    Operand(rsi, rdi, times_1, byte_offset + i * sizeof(uc16)));
 309         __ cmpl(rax, Immediate(str[i]));
 310         i++;
 311       }
 312     }
 313     BranchOrBacktrack(not_equal, on_failure);
 314   }
 315 }
 316
 317
 318 void RegExpMacroAssemblerX64::CheckGreedyLoop(Label* on_equal) {
 319   Label fallthrough;
 320   __ cmpl(rdi, Operand(backtrack_stackpointer(), 0));
 321   __ j(not_equal, &fallthrough);
 322   Drop();
 323   BranchOrBacktrack(no_condition, on_equal);
 324   __ bind(&fallthrough);
 325 }
 326
 327
 328 void RegExpMacroAssemblerX64::CheckNotBackReferenceIgnoreCase(
 329     int start_reg,
 330     Label* on_no_match) {
 331   Label fallthrough;
 332   __ movq(rdx, register_location(start_reg));  // Offset of start of capture
 333   __ movq(rbx, register_location(start_reg + 1));  // Offset of end of capture
 334   __ subq(rbx, rdx);  // Length of capture.
 335
 336   // -----------------------
 337   // rdx  = Start offset of capture.
 338   // rbx = Length of capture
 339
 340   // If length is negative, this code will fail (it's a symptom of a partial or
 341   // illegal capture where start of capture after end of capture).
 342   // This must not happen (no back-reference can reference a capture that wasn't
 343   // closed before in the reg-exp, and we must not generate code that can cause
 344   // this condition).
 345
 346   // If length is zero, either the capture is empty or it is nonparticipating.
 347   // In either case succeed immediately.
 348   __ j(equal, &fallthrough);
 349
 350   if (mode_ == ASCII) {
 351     Label loop_increment;
 352     if (on_no_match == NULL) {
 353       on_no_match = &backtrack_label_;
 354     }
 355
 356     __ lea(r9, Operand(rsi, rdx, times_1, 0));
 357     __ lea(r11, Operand(rsi, rdi, times_1, 0));
 358     __ addq(rbx, r9);  // End of capture
 359     // ---------------------
 360     // r11 - current input character address
 361     // r9 - current capture character address
 362     // rbx - end of capture
 363
 364     Label loop;
 365     __ bind(&loop);
 366     __ movzxbl(rdx, Operand(r9, 0));
 367     __ movzxbl(rax, Operand(r11, 0));
 368     // al - input character
 369     // dl - capture character
 370     __ cmpb(rax, rdx);
 371     __ j(equal, &loop_increment);
 372
 373     // Mismatch, try case-insensitive match (converting letters to lower-case).
 374     // I.e., if or-ing with 0x20 makes values equal and in range 'a'-'z', it's
 375     // a match.
 376     __ or_(rax, Immediate(0x20));  // Convert match character to lower-case.
 377     __ or_(rdx, Immediate(0x20));  // Convert capture character to lower-case.
 378     __ cmpb(rax, rdx);
 379     __ j(not_equal, on_no_match);  // Definitely not equal.
 380     __ subb(rax, Immediate('a'));
 381     __ cmpb(rax, Immediate('z' - 'a'));
 382     __ j(above, on_no_match);  // Weren't letters anyway.
 383
 384     __ bind(&loop_increment);
 385     // Increment pointers into match and capture strings.
 386     __ addq(r11, Immediate(1));
 387     __ addq(r9, Immediate(1));
 388     // Compare to end of capture, and loop if not done.
 389     __ cmpq(r9, rbx);
 390     __ j(below, &loop);
 391
 392     // Compute new value of character position after the matched part.
 393     __ movq(rdi, r11);
 394     __ subq(rdi, rsi);
 395   } else {
 396     ASSERT(mode_ == UC16);
 397     // Save important/volatile registers before calling C function.
 398 #ifndef _WIN64
 399     // Caller save on Linux and callee save in Windows.
 400     __ push(rsi);
 401     __ push(rdi);
 402 #endif
 403     __ push(backtrack_stackpointer());
 404
 405     static const int num_arguments = 4;
 406     __ PrepareCallCFunction(num_arguments);
 407
 408     // Put arguments into parameter registers. Parameters are
 409     //   Address byte_offset1 - Address captured substring's start.
 410     //   Address byte_offset2 - Address of current character position.
 411     //   size_t byte_length - length of capture in bytes(!)
 412     //   Isolate* isolate
 413 #ifdef _WIN64
 414     // Compute and set byte_offset1 (start of capture).
 415     __ lea(rcx, Operand(rsi, rdx, times_1, 0));
 416     // Set byte_offset2.
 417     __ lea(rdx, Operand(rsi, rdi, times_1, 0));
 418     // Set byte_length.
 419     __ movq(r8, rbx);
 420     // Isolate.
 421     __ LoadAddress(r9, ExternalReference::isolate_address());
 422 #else  // AMD64 calling convention
 423     // Compute byte_offset2 (current position = rsi+rdi).
 424     __ lea(rax, Operand(rsi, rdi, times_1, 0));
 425     // Compute and set byte_offset1 (start of capture).
 426     __ lea(rdi, Operand(rsi, rdx, times_1, 0));
 427     // Set byte_offset2.
 428     __ movq(rsi, rax);
 429     // Set byte_length.
 430     __ movq(rdx, rbx);
 431     // Isolate.
 432     __ LoadAddress(rcx, ExternalReference::isolate_address());
 433 #endif
 434
 435     { // NOLINT: Can't find a way to open this scope without confusing the
 436       // linter.
 437       AllowExternalCallThatCantCauseGC scope(&masm_);
 438       ExternalReference compare =
 439           ExternalReference::re_case_insensitive_compare_uc16(masm_.isolate());
 440       __ CallCFunction(compare, num_arguments);
 441     }
 442
 443     // Restore original values before reacting on result value.
 444     __ Move(code_object_pointer(), masm_.CodeObject());
 445     __ pop(backtrack_stackpointer());
 446 #ifndef _WIN64
 447     __ pop(rdi);
 448     __ pop(rsi);
 449 #endif
 450
 451     // Check if function returned non-zero for success or zero for failure.
 452     __ testq(rax, rax);
 453     BranchOrBacktrack(zero, on_no_match);
 454     // On success, increment position by length of capture.
 455     // Requires that rbx is callee save (true for both Win64 and AMD64 ABIs).
 456     __ addq(rdi, rbx);
 457   }
 458   __ bind(&fallthrough);
 459 }
 460
 461
 462 void RegExpMacroAssemblerX64::CheckNotBackReference(
 463     int start_reg,
 464     Label* on_no_match) {
 465   Label fallthrough;
 466
 467   // Find length of back-referenced capture.
 468   __ movq(rdx, register_location(start_reg));
 469   __ movq(rax, register_location(start_reg + 1));
 470   __ subq(rax, rdx);  // Length to check.
 471
 472   // Fail on partial or illegal capture (start of capture after end of capture).
 473   // This must not happen (no back-reference can reference a capture that wasn't
 474   // closed before in the reg-exp).
 475   __ Check(greater_equal, "Invalid capture referenced");
 476
 477   // Succeed on empty capture (including non-participating capture)
 478   __ j(equal, &fallthrough);
 479
 480   // -----------------------
 481   // rdx - Start of capture
 482   // rax - length of capture
 483
 484   // Check that there are sufficient characters left in the input.
 485   __ movl(rbx, rdi);
 486   __ addl(rbx, rax);
 487   BranchOrBacktrack(greater, on_no_match);
 488
 489   // Compute pointers to match string and capture string
 490   __ lea(rbx, Operand(rsi, rdi, times_1, 0));  // Start of match.
 491   __ addq(rdx, rsi);  // Start of capture.
 492   __ lea(r9, Operand(rdx, rax, times_1, 0));  // End of capture
 493
 494   // -----------------------
 495   // rbx - current capture character address.
 496   // rbx - current input character address .
 497   // r9 - end of input to match (capture length after rbx).
 498
 499   Label loop;
 500   __ bind(&loop);
 501   if (mode_ == ASCII) {
 502     __ movzxbl(rax, Operand(rdx, 0));
 503     __ cmpb(rax, Operand(rbx, 0));
 504   } else {
 505     ASSERT(mode_ == UC16);
 506     __ movzxwl(rax, Operand(rdx, 0));
 507     __ cmpw(rax, Operand(rbx, 0));
 508   }
 509   BranchOrBacktrack(not_equal, on_no_match);
 510   // Increment pointers into capture and match string.
 511   __ addq(rbx, Immediate(char_size()));
 512   __ addq(rdx, Immediate(char_size()));
 513   // Check if we have reached end of match area.
 514   __ cmpq(rdx, r9);
 515   __ j(below, &loop);
 516
 517   // Success.
 518   // Set current character position to position after match.
 519   __ movq(rdi, rbx);
 520   __ subq(rdi, rsi);
 521
 522   __ bind(&fallthrough);
 523 }
 524
 525
 526 void RegExpMacroAssemblerX64::CheckNotRegistersEqual(int reg1,
 527                                                      int reg2,
 528                                                      Label* on_not_equal) {
 529   __ movq(rax, register_location(reg1));
 530   __ cmpq(rax, register_location(reg2));
 531   BranchOrBacktrack(not_equal, on_not_equal);
 532 }
 533
 534
 535 void RegExpMacroAssemblerX64::CheckNotCharacter(uint32_t c,
 536                                                 Label* on_not_equal) {
 537   __ cmpl(current_character(), Immediate(c));
 538   BranchOrBacktrack(not_equal, on_not_equal);
 539 }
 540
 541
 542 void RegExpMacroAssemblerX64::CheckCharacterAfterAnd(uint32_t c,
 543                                                      uint32_t mask,
 544                                                      Label* on_equal) {
 545   if (c == 0) {
 546     __ testl(current_character(), Immediate(mask));
 547   } else {
 548     __ movl(rax, Immediate(mask));
 549     __ and_(rax, current_character());
 550     __ cmpl(rax, Immediate(c));
 551   }
 552   BranchOrBacktrack(equal, on_equal);
 553 }
 554
 555
 556 void RegExpMacroAssemblerX64::CheckNotCharacterAfterAnd(uint32_t c,
 557                                                         uint32_t mask,
 558                                                         Label* on_not_equal) {
 559   if (c == 0) {
 560     __ testl(current_character(), Immediate(mask));
 561   } else {
 562     __ movl(rax, Immediate(mask));
 563     __ and_(rax, current_character());
 564     __ cmpl(rax, Immediate(c));
 565   }
 566   BranchOrBacktrack(not_equal, on_not_equal);
 567 }
 568
 569
 570 void RegExpMacroAssemblerX64::CheckNotCharacterAfterMinusAnd(
 571     uc16 c,
 572     uc16 minus,
 573     uc16 mask,
 574     Label* on_not_equal) {
 575   ASSERT(minus < String::kMaxUtf16CodeUnit);
 576   __ lea(rax, Operand(current_character(), -minus));
 577   __ and_(rax, Immediate(mask));
 578   __ cmpl(rax, Immediate(c));
 579   BranchOrBacktrack(not_equal, on_not_equal);
 580 }
 581
 582
 583 void RegExpMacroAssemblerX64::CheckCharacterInRange(
 584     uc16 from,
 585     uc16 to,
 586     Label* on_in_range) {
 587   __ leal(rax, Operand(current_character(), -from));
 588   __ cmpl(rax, Immediate(to - from));
 589   BranchOrBacktrack(below_equal, on_in_range);
 590 }
 591
 592
 593 void RegExpMacroAssemblerX64::CheckCharacterNotInRange(
 594     uc16 from,
 595     uc16 to,
 596     Label* on_not_in_range) {
 597   __ leal(rax, Operand(current_character(), -from));
 598   __ cmpl(rax, Immediate(to - from));
 599   BranchOrBacktrack(above, on_not_in_range);
 600 }
 601
 602
 603 void RegExpMacroAssemblerX64::CheckBitInTable(
 604     Handle<ByteArray> table,
 605     Label* on_bit_set) {
 606   __ Move(rax, table);
 607   Register index = current_character();
 608   if (mode_ != ASCII || kTableMask != String::kMaxAsciiCharCode) {
 609     __ movq(rbx, current_character());
 610     __ and_(rbx, Immediate(kTableMask));
 611     index = rbx;
 612   }
 613   __ cmpb(FieldOperand(rax, index, times_1, ByteArray::kHeaderSize),
 614           Immediate(0));
 615   BranchOrBacktrack(not_equal, on_bit_set);
 616 }
 617
 618
 619 bool RegExpMacroAssemblerX64::CheckSpecialCharacterClass(uc16 type,
 620                                                          Label* on_no_match) {
 621   // Range checks (c in min..max) are generally implemented by an unsigned
 622   // (c - min) <= (max - min) check, using the sequence:
 623   //   lea(rax, Operand(current_character(), -min)) or sub(rax, Immediate(min))
 624   //   cmp(rax, Immediate(max - min))
 625   switch (type) {
 626   case 's':
 627     // Match space-characters
 628     if (mode_ == ASCII) {
 629       // ASCII space characters are '\t'..'\r' and ' '.
 630       Label success;
 631       __ cmpl(current_character(), Immediate(' '));
 632       __ j(equal, &success);
 633       // Check range 0x09..0x0d
 634       __ lea(rax, Operand(current_character(), -'\t'));
 635       __ cmpl(rax, Immediate('\r' - '\t'));
 636       BranchOrBacktrack(above, on_no_match);
 637       __ bind(&success);
 638       return true;
 639     }
 640     return false;
 641   case 'S':
 642     // Match non-space characters.
 643     if (mode_ == ASCII) {
 644       // ASCII space characters are '\t'..'\r' and ' '.
 645       __ cmpl(current_character(), Immediate(' '));
 646       BranchOrBacktrack(equal, on_no_match);
 647       __ lea(rax, Operand(current_character(), -'\t'));
 648       __ cmpl(rax, Immediate('\r' - '\t'));
 649       BranchOrBacktrack(below_equal, on_no_match);
 650       return true;
 651     }
 652     return false;
 653   case 'd':
 654     // Match ASCII digits ('0'..'9')
 655     __ lea(rax, Operand(current_character(), -'0'));
 656     __ cmpl(rax, Immediate('9' - '0'));
 657     BranchOrBacktrack(above, on_no_match);
 658     return true;
 659   case 'D':
 660     // Match non ASCII-digits
 661     __ lea(rax, Operand(current_character(), -'0'));
 662     __ cmpl(rax, Immediate('9' - '0'));
 663     BranchOrBacktrack(below_equal, on_no_match);
 664     return true;
 665   case '.': {
 666     // Match non-newlines (not 0x0a('\n'), 0x0d('\r'), 0x2028 and 0x2029)
 667     __ movl(rax, current_character());
 668     __ xor_(rax, Immediate(0x01));
 669     // See if current character is '\n'^1 or '\r'^1, i.e., 0x0b or 0x0c
 670     __ subl(rax, Immediate(0x0b));
 671     __ cmpl(rax, Immediate(0x0c - 0x0b));
 672     BranchOrBacktrack(below_equal, on_no_match);
 673     if (mode_ == UC16) {
 674       // Compare original value to 0x2028 and 0x2029, using the already
 675       // computed (current_char ^ 0x01 - 0x0b). I.e., check for
 676       // 0x201d (0x2028 - 0x0b) or 0x201e.
 677       __ subl(rax, Immediate(0x2028 - 0x0b));
 678       __ cmpl(rax, Immediate(0x2029 - 0x2028));
 679       BranchOrBacktrack(below_equal, on_no_match);
 680     }
 681     return true;
 682   }
 683   case 'n': {
 684     // Match newlines (0x0a('\n'), 0x0d('\r'), 0x2028 and 0x2029)
 685     __ movl(rax, current_character());
 686     __ xor_(rax, Immediate(0x01));
 687     // See if current character is '\n'^1 or '\r'^1, i.e., 0x0b or 0x0c
 688     __ subl(rax, Immediate(0x0b));
 689     __ cmpl(rax, Immediate(0x0c - 0x0b));
 690     if (mode_ == ASCII) {
 691       BranchOrBacktrack(above, on_no_match);
 692     } else {
 693       Label done;
 694       BranchOrBacktrack(below_equal, &done);
 695       // Compare original value to 0x2028 and 0x2029, using the already
 696       // computed (current_char ^ 0x01 - 0x0b). I.e., check for
 697       // 0x201d (0x2028 - 0x0b) or 0x201e.
 698       __ subl(rax, Immediate(0x2028 - 0x0b));
 699       __ cmpl(rax, Immediate(0x2029 - 0x2028));
 700       BranchOrBacktrack(above, on_no_match);
 701       __ bind(&done);
 702     }
 703     return true;
 704   }
 705   case 'w': {
 706     if (mode_ != ASCII) {
 707       // Table is 128 entries, so all ASCII characters can be tested.
 708       __ cmpl(current_character(), Immediate('z'));
 709       BranchOrBacktrack(above, on_no_match);
 710     }
 711     __ movq(rbx, ExternalReference::re_word_character_map());
 712     ASSERT_EQ(0, word_character_map[0]);  // Character '\0' is not a word char.
 713     __ testb(Operand(rbx, current_character(), times_1, 0),
 714              current_character());
 715     BranchOrBacktrack(zero, on_no_match);
 716     return true;
 717   }
 718   case 'W': {
 719     Label done;
 720     if (mode_ != ASCII) {
 721       // Table is 128 entries, so all ASCII characters can be tested.
 722       __ cmpl(current_character(), Immediate('z'));
 723       __ j(above, &done);
 724     }
 725     __ movq(rbx, ExternalReference::re_word_character_map());
 726     ASSERT_EQ(0, word_character_map[0]);  // Character '\0' is not a word char.
 727     __ testb(Operand(rbx, current_character(), times_1, 0),
 728              current_character());
 729     BranchOrBacktrack(not_zero, on_no_match);
 730     if (mode_ != ASCII) {
 731       __ bind(&done);
 732     }
 733     return true;
 734   }
 735
 736   case '*':
 737     // Match any character.
 738     return true;
 739   // No custom implementation (yet): s(UC16), S(UC16).
 740   default:
 741     return false;
 742   }
 743 }
 744
 745
 746 void RegExpMacroAssemblerX64::Fail() {
 747   ASSERT(FAILURE == 0);  // Return value for failure is zero.
 748   __ Set(rax, 0);
 749   __ jmp(&exit_label_);
 750 }
 751
 752
 753 Handle<HeapObject> RegExpMacroAssemblerX64::GetCode(Handle<String> source) {
 754   // Finalize code - write the entry point code now we know how many
 755   // registers we need.
 756   // Entry code:
 757   __ bind(&entry_label_);
 758
 759   // Tell the system that we have a stack frame.  Because the type is MANUAL, no
 760   // is generated.
 761   FrameScope scope(&masm_, StackFrame::MANUAL);
 762
 763   // Actually emit code to start a new stack frame.
 764   __ push(rbp);
 765   __ movq(rbp, rsp);
 766   // Save parameters and callee-save registers. Order here should correspond
 767   //  to order of kBackup_ebx etc.
 768 #ifdef _WIN64
 769   // MSVC passes arguments in rcx, rdx, r8, r9, with backing stack slots.
 770   // Store register parameters in pre-allocated stack slots,
 771   __ movq(Operand(rbp, kInputString), rcx);
 772   __ movq(Operand(rbp, kStartIndex), rdx);  // Passed as int32 in edx.
 773   __ movq(Operand(rbp, kInputStart), r8);
 774   __ movq(Operand(rbp, kInputEnd), r9);
 775   // Callee-save on Win64.
 776   __ push(rsi);
 777   __ push(rdi);
 778   __ push(rbx);
 779 #else
 780   // GCC passes arguments in rdi, rsi, rdx, rcx, r8, r9 (and then on stack).
 781   // Push register parameters on stack for reference.
 782   ASSERT_EQ(kInputString, -1 * kPointerSize);
 783   ASSERT_EQ(kStartIndex, -2 * kPointerSize);
 784   ASSERT_EQ(kInputStart, -3 * kPointerSize);
 785   ASSERT_EQ(kInputEnd, -4 * kPointerSize);
 786   ASSERT_EQ(kRegisterOutput, -5 * kPointerSize);
 787   ASSERT_EQ(kStackHighEnd, -6 * kPointerSize);
 788   __ push(rdi);
 789   __ push(rsi);
 790   __ push(rdx);
 791   __ push(rcx);
 792   __ push(r8);
 793   __ push(r9);
 794
 795   __ push(rbx);  // Callee-save
 796 #endif
 797
 798   __ push(Immediate(0));  // Make room for "at start" constant.
 799
 800   // Check if we have space on the stack for registers.
 801   Label stack_limit_hit;
 802   Label stack_ok;
 803
 804   ExternalReference stack_limit =
 805       ExternalReference::address_of_stack_limit(masm_.isolate());
 806   __ movq(rcx, rsp);
 807   __ movq(kScratchRegister, stack_limit);
 808   __ subq(rcx, Operand(kScratchRegister, 0));
 809   // Handle it if the stack pointer is already below the stack limit.
 810   __ j(below_equal, &stack_limit_hit);
 811   // Check if there is room for the variable number of registers above
 812   // the stack limit.
 813   __ cmpq(rcx, Immediate(num_registers_ * kPointerSize));
 814   __ j(above_equal, &stack_ok);
 815   // Exit with OutOfMemory exception. There is not enough space on the stack
 816   // for our working registers.
 817   __ Set(rax, EXCEPTION);
 818   __ jmp(&exit_label_);
 819
 820   __ bind(&stack_limit_hit);
 821   __ Move(code_object_pointer(), masm_.CodeObject());
 822   CallCheckStackGuardState();  // Preserves no registers beside rbp and rsp.
 823   __ testq(rax, rax);
 824   // If returned value is non-zero, we exit with the returned value as result.
 825   __ j(not_zero, &exit_label_);
 826
 827   __ bind(&stack_ok);
 828
 829   // Allocate space on stack for registers.
 830   __ subq(rsp, Immediate(num_registers_ * kPointerSize));
 831   // Load string length.
 832   __ movq(rsi, Operand(rbp, kInputEnd));
 833   // Load input position.
 834   __ movq(rdi, Operand(rbp, kInputStart));
 835   // Set up rdi to be negative offset from string end.
 836   __ subq(rdi, rsi);
 837   // Set rax to address of char before start of the string
 838   // (effectively string position -1).
 839   __ movq(rbx, Operand(rbp, kStartIndex));
 840   __ neg(rbx);
 841   if (mode_ == UC16) {
 842     __ lea(rax, Operand(rdi, rbx, times_2, -char_size()));
 843   } else {
 844     __ lea(rax, Operand(rdi, rbx, times_1, -char_size()));
 845   }
 846   // Store this value in a local variable, for use when clearing
 847   // position registers.
 848   __ movq(Operand(rbp, kInputStartMinusOne), rax);
 849
 850   if (num_saved_registers_ > 0) {
 851     // Fill saved registers with initial value = start offset - 1
 852     // Fill in stack push order, to avoid accessing across an unwritten
 853     // page (a problem on Windows).
 854     __ Set(rcx, kRegisterZero);
 855     Label init_loop;
 856     __ bind(&init_loop);
 857     __ movq(Operand(rbp, rcx, times_1, 0), rax);
 858     __ subq(rcx, Immediate(kPointerSize));
 859     __ cmpq(rcx,
 860             Immediate(kRegisterZero - num_saved_registers_ * kPointerSize));
 861     __ j(greater, &init_loop);
 862   }
 863   // Ensure that we have written to each stack page, in order. Skipping a page
 864   // on Windows can cause segmentation faults. Assuming page size is 4k.
 865   const int kPageSize = 4096;
 866   const int kRegistersPerPage = kPageSize / kPointerSize;
 867   for (int i = num_saved_registers_ + kRegistersPerPage - 1;
 868       i < num_registers_;
 869       i += kRegistersPerPage) {
 870     __ movq(register_location(i), rax);  // One write every page.
 871   }
 872
 873   // Initialize backtrack stack pointer.
 874   __ movq(backtrack_stackpointer(), Operand(rbp, kStackHighEnd));
 875   // Initialize code object pointer.
 876   __ Move(code_object_pointer(), masm_.CodeObject());
 877   // Load previous char as initial value of current-character.
 878   Label at_start;
 879   __ cmpb(Operand(rbp, kStartIndex), Immediate(0));
 880   __ j(equal, &at_start);
 881   LoadCurrentCharacterUnchecked(-1, 1);  // Load previous char.
 882   __ jmp(&start_label_);
 883   __ bind(&at_start);
 884   __ Set(current_character(), '\n');
 885   __ jmp(&start_label_);
 886
 887
 888   // Exit code:
 889   if (success_label_.is_linked()) {
 890     // Save captures when successful.
 891     __ bind(&success_label_);
 892     if (num_saved_registers_ > 0) {
 893       // copy captures to output
 894       __ movq(rdx, Operand(rbp, kStartIndex));
 895       __ movq(rbx, Operand(rbp, kRegisterOutput));
 896       __ movq(rcx, Operand(rbp, kInputEnd));
 897       __ subq(rcx, Operand(rbp, kInputStart));
 898       if (mode_ == UC16) {
 899         __ lea(rcx, Operand(rcx, rdx, times_2, 0));
 900       } else {
 901         __ addq(rcx, rdx);
 902       }
 903       for (int i = 0; i < num_saved_registers_; i++) {
 904         __ movq(rax, register_location(i));
 905         __ addq(rax, rcx);  // Convert to index from start, not end.
 906         if (mode_ == UC16) {
 907           __ sar(rax, Immediate(1));  // Convert byte index to character index.
 908         }
 909         __ movl(Operand(rbx, i * kIntSize), rax);
 910       }
 911     }
 912     __ Set(rax, SUCCESS);
 913   }
 914
 915   // Exit and return rax
 916   __ bind(&exit_label_);
 917
 918 #ifdef _WIN64
 919   // Restore callee save registers.
 920   __ lea(rsp, Operand(rbp, kLastCalleeSaveRegister));
 921   __ pop(rbx);
 922   __ pop(rdi);
 923   __ pop(rsi);
 924   // Stack now at rbp.
 925 #else
 926   // Restore callee save register.
 927   __ movq(rbx, Operand(rbp, kBackup_rbx));
 928   // Skip rsp to rbp.
 929   __ movq(rsp, rbp);
 930 #endif
 931   // Exit function frame, restore previous one.
 932   __ pop(rbp);
 933   __ ret(0);
 934
 935   // Backtrack code (branch target for conditional backtracks).
 936   if (backtrack_label_.is_linked()) {
 937     __ bind(&backtrack_label_);
 938     Backtrack();
 939   }
 940
 941   Label exit_with_exception;
 942
 943   // Preempt-code
 944   if (check_preempt_label_.is_linked()) {
 945     SafeCallTarget(&check_preempt_label_);
 946
 947     __ push(backtrack_stackpointer());
 948     __ push(rdi);
 949
 950     CallCheckStackGuardState();
 951     __ testq(rax, rax);
 952     // If returning non-zero, we should end execution with the given
 953     // result as return value.
 954     __ j(not_zero, &exit_label_);
 955
 956     // Restore registers.
 957     __ Move(code_object_pointer(), masm_.CodeObject());
 958     __ pop(rdi);
 959     __ pop(backtrack_stackpointer());
 960     // String might have moved: Reload esi from frame.
 961     __ movq(rsi, Operand(rbp, kInputEnd));
 962     SafeReturn();
 963   }
 964
 965   // Backtrack stack overflow code.
 966   if (stack_overflow_label_.is_linked()) {
 967     SafeCallTarget(&stack_overflow_label_);
 968     // Reached if the backtrack-stack limit has been hit.
 969
 970     Label grow_failed;
 971     // Save registers before calling C function
 972 #ifndef _WIN64
 973     // Callee-save in Microsoft 64-bit ABI, but not in AMD64 ABI.
 974     __ push(rsi);
 975     __ push(rdi);
 976 #endif
 977
 978     // Call GrowStack(backtrack_stackpointer())
 979     static const int num_arguments = 3;
 980     __ PrepareCallCFunction(num_arguments);
 981 #ifdef _WIN64
 982     // Microsoft passes parameters in rcx, rdx, r8.
 983     // First argument, backtrack stackpointer, is already in rcx.
 984     __ lea(rdx, Operand(rbp, kStackHighEnd));  // Second argument
 985     __ LoadAddress(r8, ExternalReference::isolate_address());
 986 #else
 987     // AMD64 ABI passes parameters in rdi, rsi, rdx.
 988     __ movq(rdi, backtrack_stackpointer());   // First argument.
 989     __ lea(rsi, Operand(rbp, kStackHighEnd));  // Second argument.
 990     __ LoadAddress(rdx, ExternalReference::isolate_address());
 991 #endif
 992     ExternalReference grow_stack =
 993         ExternalReference::re_grow_stack(masm_.isolate());
 994     __ CallCFunction(grow_stack, num_arguments);
 995     // If return NULL, we have failed to grow the stack, and
 996     // must exit with a stack-overflow exception.
 997     __ testq(rax, rax);
 998     __ j(equal, &exit_with_exception);
 999     // Otherwise use return value as new stack pointer.
1000     __ movq(backtrack_stackpointer(), rax);
1001     // Restore saved registers and continue.
1002     __ Move(code_object_pointer(), masm_.CodeObject());
1003 #ifndef _WIN64
1004     __ pop(rdi);
1005     __ pop(rsi);
1006 #endif
1007     SafeReturn();
1008   }
1009
1010   if (exit_with_exception.is_linked()) {
1011     // If any of the code above needed to exit with an exception.
1012     __ bind(&exit_with_exception);
1013     // Exit with Result EXCEPTION(-1) to signal thrown exception.
1014     __ Set(rax, EXCEPTION);
1015     __ jmp(&exit_label_);
1016   }
1017
1018   FixupCodeRelativePositions();
1019
1020   CodeDesc code_desc;
1021   masm_.GetCode(&code_desc);
1022   Isolate* isolate = ISOLATE;
1023   Handle<Code> code = isolate->factory()->NewCode(
1024       code_desc, Code::ComputeFlags(Code::REGEXP),
1025       masm_.CodeObject());
1026   PROFILE(isolate, RegExpCodeCreateEvent(*code, *source));
1027   return Handle<HeapObject>::cast(code);
1028 }
1029
1030
1031 void RegExpMacroAssemblerX64::GoTo(Label* to) {
1032   BranchOrBacktrack(no_condition, to);
1033 }
1034
1035
1036 void RegExpMacroAssemblerX64::IfRegisterGE(int reg,
1037                                            int comparand,
1038                                            Label* if_ge) {
1039   __ cmpq(register_location(reg), Immediate(comparand));
1040   BranchOrBacktrack(greater_equal, if_ge);
1041 }
1042
1043
1044 void RegExpMacroAssemblerX64::IfRegisterLT(int reg,
1045                                            int comparand,
1046                                            Label* if_lt) {
1047   __ cmpq(register_location(reg), Immediate(comparand));
1048   BranchOrBacktrack(less, if_lt);
1049 }
1050
1051
1052 void RegExpMacroAssemblerX64::IfRegisterEqPos(int reg,
1053                                               Label* if_eq) {
1054   __ cmpq(rdi, register_location(reg));
1055   BranchOrBacktrack(equal, if_eq);
1056 }
1057
1058
1059 RegExpMacroAssembler::IrregexpImplementation
1060     RegExpMacroAssemblerX64::Implementation() {
1061   return kX64Implementation;
1062 }
1063
1064
1065 void RegExpMacroAssemblerX64::LoadCurrentCharacter(int cp_offset,
1066                                                    Label* on_end_of_input,
1067                                                    bool check_bounds,
1068                                                    int characters) {
1069   ASSERT(cp_offset >= -1);      // ^ and \b can look behind one character.
1070   ASSERT(cp_offset < (1<<30));  // Be sane! (And ensure negation works)
1071   if (check_bounds) {
1072     CheckPosition(cp_offset + characters - 1, on_end_of_input);
1073   }
1074   LoadCurrentCharacterUnchecked(cp_offset, characters);
1075 }
1076
1077
1078 void RegExpMacroAssemblerX64::PopCurrentPosition() {
1079   Pop(rdi);
1080 }
1081
1082
1083 void RegExpMacroAssemblerX64::PopRegister(int register_index) {
1084   Pop(rax);
1085   __ movq(register_location(register_index), rax);
1086 }
1087
1088
1089 void RegExpMacroAssemblerX64::PushBacktrack(Label* label) {
1090   Push(label);
1091   CheckStackLimit();
1092 }
1093
1094
1095 void RegExpMacroAssemblerX64::PushCurrentPosition() {
1096   Push(rdi);
1097 }
1098
1099
1100 void RegExpMacroAssemblerX64::PushRegister(int register_index,
1101                                            StackCheckFlag check_stack_limit) {
1102   __ movq(rax, register_location(register_index));
1103   Push(rax);
1104   if (check_stack_limit) CheckStackLimit();
1105 }
1106
1107
1108 void RegExpMacroAssemblerX64::ReadCurrentPositionFromRegister(int reg) {
1109   __ movq(rdi, register_location(reg));
1110 }
1111
1112
1113 void RegExpMacroAssemblerX64::ReadStackPointerFromRegister(int reg) {
1114   __ movq(backtrack_stackpointer(), register_location(reg));
1115   __ addq(backtrack_stackpointer(), Operand(rbp, kStackHighEnd));
1116 }
1117
1118
1119 void RegExpMacroAssemblerX64::SetCurrentPositionFromEnd(int by) {
1120   Label after_position;
1121   __ cmpq(rdi, Immediate(-by * char_size()));
1122   __ j(greater_equal, &after_position, Label::kNear);
1123   __ movq(rdi, Immediate(-by * char_size()));
1124   // On RegExp code entry (where this operation is used), the character before
1125   // the current position is expected to be already loaded.
1126   // We have advanced the position, so it's safe to read backwards.
1127   LoadCurrentCharacterUnchecked(-1, 1);
1128   __ bind(&after_position);
1129 }
1130
1131
1132 void RegExpMacroAssemblerX64::SetRegister(int register_index, int to) {
1133   ASSERT(register_index >= num_saved_registers_);  // Reserved for positions!
1134   __ movq(register_location(register_index), Immediate(to));
1135 }
1136
1137
1138 void RegExpMacroAssemblerX64::Succeed() {
1139   __ jmp(&success_label_);
1140 }
1141
1142
1143 void RegExpMacroAssemblerX64::WriteCurrentPositionToRegister(int reg,
1144                                                              int cp_offset) {
1145   if (cp_offset == 0) {
1146     __ movq(register_location(reg), rdi);
1147   } else {
1148     __ lea(rax, Operand(rdi, cp_offset * char_size()));
1149     __ movq(register_location(reg), rax);
1150   }
1151 }
1152
1153
1154 void RegExpMacroAssemblerX64::ClearRegisters(int reg_from, int reg_to) {
1155   ASSERT(reg_from <= reg_to);
1156   __ movq(rax, Operand(rbp, kInputStartMinusOne));
1157   for (int reg = reg_from; reg <= reg_to; reg++) {
1158     __ movq(register_location(reg), rax);
1159   }
1160 }
1161
1162
1163 void RegExpMacroAssemblerX64::WriteStackPointerToRegister(int reg) {
1164   __ movq(rax, backtrack_stackpointer());
1165   __ subq(rax, Operand(rbp, kStackHighEnd));
1166   __ movq(register_location(reg), rax);
1167 }
1168
1169
1170 // Private methods:
1171
1172 void RegExpMacroAssemblerX64::CallCheckStackGuardState() {
1173   // This function call preserves no register values. Caller should
1174   // store anything volatile in a C call or overwritten by this function.
1175   static const int num_arguments = 3;
1176   __ PrepareCallCFunction(num_arguments);
1177 #ifdef _WIN64
1178   // Second argument: Code* of self. (Do this before overwriting r8).
1179   __ movq(rdx, code_object_pointer());
1180   // Third argument: RegExp code frame pointer.
1181   __ movq(r8, rbp);
1182   // First argument: Next address on the stack (will be address of
1183   // return address).
1184   __ lea(rcx, Operand(rsp, -kPointerSize));
1185 #else
1186   // Third argument: RegExp code frame pointer.
1187   __ movq(rdx, rbp);
1188   // Second argument: Code* of self.
1189   __ movq(rsi, code_object_pointer());
1190   // First argument: Next address on the stack (will be address of
1191   // return address).
1192   __ lea(rdi, Operand(rsp, -kPointerSize));
1193 #endif
1194   ExternalReference stack_check =
1195       ExternalReference::re_check_stack_guard_state(masm_.isolate());
1196   __ CallCFunction(stack_check, num_arguments);
1197 }
1198
1199
1200 // Helper function for reading a value out of a stack frame.
1201 template <typename T>
1202 static T& frame_entry(Address re_frame, int frame_offset) {
1203   return reinterpret_cast<T&>(Memory::int32_at(re_frame + frame_offset));
1204 }
1205
1206
1207 int RegExpMacroAssemblerX64::CheckStackGuardState(Address* return_address,
1208                                                   Code* re_code,
1209                                                   Address re_frame) {
1210   Isolate* isolate = frame_entry<Isolate*>(re_frame, kIsolate);
1211   ASSERT(isolate == Isolate::Current());
1212   if (isolate->stack_guard()->IsStackOverflow()) {
1213     isolate->StackOverflow();
1214     return EXCEPTION;
1215   }
1216
1217   // If not real stack overflow the stack guard was used to interrupt
1218   // execution for another purpose.
1219
1220   // If this is a direct call from JavaScript retry the RegExp forcing the call
1221   // through the runtime system. Currently the direct call cannot handle a GC.
1222   if (frame_entry<int>(re_frame, kDirectCall) == 1) {
1223     return RETRY;
1224   }
1225
1226   // Prepare for possible GC.
1227   HandleScope handles(isolate);
1228   Handle<Code> code_handle(re_code);
1229
1230   Handle<String> subject(frame_entry<String*>(re_frame, kInputString));
1231
1232   // Current string.
1233   bool is_ascii = subject->IsAsciiRepresentationUnderneath();
1234
1235   ASSERT(re_code->instruction_start() <= *return_address);
1236   ASSERT(*return_address <=
1237       re_code->instruction_start() + re_code->instruction_size());
1238
1239   MaybeObject* result = Execution::HandleStackGuardInterrupt(isolate);
1240
1241   if (*code_handle != re_code) {  // Return address no longer valid
1242     intptr_t delta = code_handle->address() - re_code->address();
1243     // Overwrite the return address on the stack.
1244     *return_address += delta;
1245   }
1246
1247   if (result->IsException()) {
1248     return EXCEPTION;
1249   }
1250
1251   Handle<String> subject_tmp = subject;
1252   int slice_offset = 0;
1253
1254   // Extract the underlying string and the slice offset.
1255   if (StringShape(*subject_tmp).IsCons()) {
1256     subject_tmp = Handle<String>(ConsString::cast(*subject_tmp)->first());
1257   } else if (StringShape(*subject_tmp).IsSliced()) {
1258     SlicedString* slice = SlicedString::cast(*subject_tmp);
1259     subject_tmp = Handle<String>(slice->parent());
1260     slice_offset = slice->offset();
1261   }
1262
1263   // String might have changed.
1264   if (subject_tmp->IsAsciiRepresentation() != is_ascii) {
1265     // If we changed between an ASCII and an UC16 string, the specialized
1266     // code cannot be used, and we need to restart regexp matching from
1267     // scratch (including, potentially, compiling a new version of the code).
1268     return RETRY;
1269   }
1270
1271   // Otherwise, the content of the string might have moved. It must still
1272   // be a sequential or external string with the same content.
1273   // Update the start and end pointers in the stack frame to the current
1274   // location (whether it has actually moved or not).
1275   ASSERT(StringShape(*subject_tmp).IsSequential() ||
1276       StringShape(*subject_tmp).IsExternal());
1277
1278   // The original start address of the characters to match.
1279   const byte* start_address = frame_entry<const byte*>(re_frame, kInputStart);
1280
1281   // Find the current start address of the same character at the current string
1282   // position.
1283   int start_index = frame_entry<int>(re_frame, kStartIndex);
1284   const byte* new_address = StringCharacterPosition(*subject_tmp,
1285                                                     start_index + slice_offset);
1286
1287   if (start_address != new_address) {
1288     // If there is a difference, update the object pointer and start and end
1289     // addresses in the RegExp stack frame to match the new value.
1290     const byte* end_address = frame_entry<const byte* >(re_frame, kInputEnd);
1291     int byte_length = static_cast<int>(end_address - start_address);
1292     frame_entry<const String*>(re_frame, kInputString) = *subject;
1293     frame_entry<const byte*>(re_frame, kInputStart) = new_address;
1294     frame_entry<const byte*>(re_frame, kInputEnd) = new_address + byte_length;
1295   } else if (frame_entry<const String*>(re_frame, kInputString) != *subject) {
1296     // Subject string might have been a ConsString that underwent
1297     // short-circuiting during GC. That will not change start_address but
1298     // will change pointer inside the subject handle.
1299     frame_entry<const String*>(re_frame, kInputString) = *subject;
1300   }
1301
1302   return 0;
1303 }
1304
1305
1306 Operand RegExpMacroAssemblerX64::register_location(int register_index) {
1307   ASSERT(register_index < (1<<30));
1308   if (num_registers_ <= register_index) {
1309     num_registers_ = register_index + 1;
1310   }
1311   return Operand(rbp, kRegisterZero - register_index * kPointerSize);
1312 }
1313
1314
1315 void RegExpMacroAssemblerX64::CheckPosition(int cp_offset,
1316                                             Label* on_outside_input) {
1317   __ cmpl(rdi, Immediate(-cp_offset * char_size()));
1318   BranchOrBacktrack(greater_equal, on_outside_input);
1319 }
1320
1321
1322 void RegExpMacroAssemblerX64::BranchOrBacktrack(Condition condition,
1323                                                 Label* to) {
1324   if (condition < 0) {  // No condition
1325     if (to == NULL) {
1326       Backtrack();
1327       return;
1328     }
1329     __ jmp(to);
1330     return;
1331   }
1332   if (to == NULL) {
1333     __ j(condition, &backtrack_label_);
1334     return;
1335   }
1336   __ j(condition, to);
1337 }
1338
1339
1340 void RegExpMacroAssemblerX64::SafeCall(Label* to) {
1341   __ call(to);
1342 }
1343
1344
1345 void RegExpMacroAssemblerX64::SafeCallTarget(Label* label) {
1346   __ bind(label);
1347   __ subq(Operand(rsp, 0), code_object_pointer());
1348 }
1349
1350
1351 void RegExpMacroAssemblerX64::SafeReturn() {
1352   __ addq(Operand(rsp, 0), code_object_pointer());
1353   __ ret(0);
1354 }
1355
1356
1357 void RegExpMacroAssemblerX64::Push(Register source) {
1358   ASSERT(!source.is(backtrack_stackpointer()));
1359   // Notice: This updates flags, unlike normal Push.
1360   __ subq(backtrack_stackpointer(), Immediate(kIntSize));
1361   __ movl(Operand(backtrack_stackpointer(), 0), source);
1362 }
1363
1364
1365 void RegExpMacroAssemblerX64::Push(Immediate value) {
1366   // Notice: This updates flags, unlike normal Push.
1367   __ subq(backtrack_stackpointer(), Immediate(kIntSize));
1368   __ movl(Operand(backtrack_stackpointer(), 0), value);
1369 }
1370
1371
1372 void RegExpMacroAssemblerX64::FixupCodeRelativePositions() {
1373   for (int i = 0, n = code_relative_fixup_positions_.length(); i < n; i++) {
1374     int position = code_relative_fixup_positions_[i];
1375     // The position succeeds a relative label offset from position.
1376     // Patch the relative offset to be relative to the Code object pointer
1377     // instead.
1378     int patch_position = position - kIntSize;
1379     int offset = masm_.long_at(patch_position);
1380     masm_.long_at_put(patch_position,
1381                        offset
1382                        + position
1383                        + Code::kHeaderSize
1384                        - kHeapObjectTag);
1385   }
1386   code_relative_fixup_positions_.Clear();
1387 }
1388
1389
1390 void RegExpMacroAssemblerX64::Push(Label* backtrack_target) {
1391   __ subq(backtrack_stackpointer(), Immediate(kIntSize));
1392   __ movl(Operand(backtrack_stackpointer(), 0), backtrack_target);
1393   MarkPositionForCodeRelativeFixup();
1394 }
1395
1396
1397 void RegExpMacroAssemblerX64::Pop(Register target) {
1398   ASSERT(!target.is(backtrack_stackpointer()));
1399   __ movsxlq(target, Operand(backtrack_stackpointer(), 0));
1400   // Notice: This updates flags, unlike normal Pop.
1401   __ addq(backtrack_stackpointer(), Immediate(kIntSize));
1402 }
1403
1404
1405 void RegExpMacroAssemblerX64::Drop() {
1406   __ addq(backtrack_stackpointer(), Immediate(kIntSize));
1407 }
1408
1409
1410 void RegExpMacroAssemblerX64::CheckPreemption() {
1411   // Check for preemption.
1412   Label no_preempt;
1413   ExternalReference stack_limit =
1414       ExternalReference::address_of_stack_limit(masm_.isolate());
1415   __ load_rax(stack_limit);
1416   __ cmpq(rsp, rax);
1417   __ j(above, &no_preempt);
1418
1419   SafeCall(&check_preempt_label_);
1420
1421   __ bind(&no_preempt);
1422 }
1423
1424
1425 void RegExpMacroAssemblerX64::CheckStackLimit() {
1426   Label no_stack_overflow;
1427   ExternalReference stack_limit =
1428       ExternalReference::address_of_regexp_stack_limit(masm_.isolate());
1429   __ load_rax(stack_limit);
1430   __ cmpq(backtrack_stackpointer(), rax);
1431   __ j(above, &no_stack_overflow);
1432
1433   SafeCall(&stack_overflow_label_);
1434
1435   __ bind(&no_stack_overflow);
1436 }
1437
1438
1439 void RegExpMacroAssemblerX64::LoadCurrentCharacterUnchecked(int cp_offset,
1440                                                             int characters) {
1441   if (mode_ == ASCII) {
1442     if (characters == 4) {
1443       __ movl(current_character(), Operand(rsi, rdi, times_1, cp_offset));
1444     } else if (characters == 2) {
1445       __ movzxwl(current_character(), Operand(rsi, rdi, times_1, cp_offset));
1446     } else {
1447       ASSERT(characters == 1);
1448       __ movzxbl(current_character(), Operand(rsi, rdi, times_1, cp_offset));
1449     }
1450   } else {
1451     ASSERT(mode_ == UC16);
1452     if (characters == 2) {
1453       __ movl(current_character(),
1454               Operand(rsi, rdi, times_1, cp_offset * sizeof(uc16)));
1455     } else {
1456       ASSERT(characters == 1);
1457       __ movzxwl(current_character(),
1458                  Operand(rsi, rdi, times_1, cp_offset * sizeof(uc16)));
1459     }
1460   }
1461 }
1462
1463 #undef __
1464
1465 #endif  // V8_INTERPRETED_REGEXP
1466
1467 }}  // namespace v8::internal
1468
1469 #endif  // V8_TARGET_ARCH_X64