Replace.cpp

   1 /* Cycript - Inlining/Optimizing JavaScript Compiler
   2  * Copyright (C) 2009  Jay Freeman (saurik)
   3 */
   4
   5 /* Modified BSD License {{{ */
   6 /*
   7  *        Redistribution and use in source and binary
   8  * forms, with or without modification, are permitted
   9  * provided that the following conditions are met:
  10  *
  11  * 1. Redistributions of source code must retain the
  12  *    above copyright notice, this list of conditions
  13  *    and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the
  15  *    above copyright notice, this list of conditions
  16  *    and the following disclaimer in the documentation
  17  *    and/or other materials provided with the
  18  *    distribution.
  19  * 3. The name of the author may not be used to endorse
  20  *    or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
  24  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
  25  * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  26  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
  28  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  29  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  30  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  31  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  32  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  33  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
  34  * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  35  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
  36  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  37 */
  38 /* }}} */
  39
  40 #include "Parser.hpp"
  41 #include "Replace.hpp"
  42
  43 #include <iomanip>
  44
  45 CYExpression *CYAdd::Replace(CYContext &context) {
  46     CYInfix::Replace(context);
  47
  48     CYExpression *lhp(lhs_->Primitive(context));
  49     CYExpression *rhp(rhs_->Primitive(context));
  50
  51     CYString *lhs(dynamic_cast<CYString *>(lhp));
  52     CYString *rhs(dynamic_cast<CYString *>(rhp));
  53
  54     if (lhs != NULL || rhs != NULL) {
  55         if (lhs == NULL) {
  56             lhs = lhp->String(context);
  57             if (lhs == NULL)
  58                 return this;
  59         } else if (rhs == NULL) {
  60             rhs = rhp->String(context);
  61             if (rhs == NULL)
  62                 return this;
  63         }
  64
  65         return lhs->Concat(context, rhs);
  66     }
  67
  68     if (CYNumber *lhn = lhp->Number(context))
  69         if (CYNumber *rhn = rhp->Number(context))
  70             return $D(lhn->Value() + rhn->Value());
  71
  72     return this;
  73 }
  74
  75 CYExpression *CYAddressOf::Replace(CYContext &context) {
  76     CYPrefix::Replace(context);
  77     return $C0($M(rhs_, $S("$cya")));
  78 }
  79
  80 void CYArgument::Replace(CYContext &context) { $T()
  81     context.Replace(value_);
  82     next_->Replace(context);
  83 }
  84
  85 CYExpression *CYArray::Replace(CYContext &context) {
  86     elements_->Replace(context);
  87     return this;
  88 }
  89
  90 CYExpression *CYArrayComprehension::Replace(CYContext &context) {
  91     CYVariable *cyv($V("$cyv"));
  92
  93     return $C0($F(NULL, $P1("$cyv", comprehensions_->Parameters(context)), $$->*
  94         $E($ CYAssign(cyv, $ CYArray()))->*
  95         comprehensions_->Replace(context, $E($C1($M(cyv, $S("push")), expression_)))->*
  96         $ CYReturn(cyv)
  97     ));
  98 }
  99
 100 CYExpression *CYAssignment::Replace(CYContext &context) {
 101     context.Replace(lhs_);
 102     context.Replace(rhs_);
 103     return this;
 104 }
 105
 106 CYStatement *CYBlock::Replace(CYContext &context) {
 107     statements_ = statements_->ReplaceAll(context);
 108     if (statements_ == NULL)
 109         return $ CYEmpty();
 110     return this;
 111 }
 112
 113 CYStatement *CYBreak::Replace(CYContext &context) {
 114     return this;
 115 }
 116
 117 CYExpression *CYCall::Replace(CYContext &context) {
 118     context.Replace(function_);
 119     arguments_->Replace(context);
 120     return this;
 121 }
 122
 123 namespace cy {
 124 namespace Syntax {
 125
 126 void Catch::Replace(CYContext &context) { $T()
 127     code_.Replace(context);
 128 }
 129
 130 } }
 131
 132 void CYClause::Replace(CYContext &context) { $T()
 133     context.Replace(case_);
 134     statements_ = statements_->ReplaceAll(context);
 135     next_->Replace(context);
 136 }
 137
 138 CYStatement *CYComment::Replace(CYContext &context) {
 139     return this;
 140 }
 141
 142 CYExpression *CYCompound::Replace(CYContext &context) {
 143     expressions_ = expressions_->ReplaceAll(context);
 144     return expressions_ == NULL ? NULL : this;
 145 }
 146
 147 CYFunctionParameter *CYComprehension::Parameters(CYContext &context) const { $T(NULL)
 148     CYFunctionParameter *next(next_->Parameters(context));
 149     if (CYFunctionParameter *parameter = Parameter(context)) {
 150         parameter->SetNext(next);
 151         return parameter;
 152     } else
 153         return next;
 154 }
 155
 156 CYStatement *CYComprehension::Replace(CYContext &context, CYStatement *statement) const {
 157     return next_ == NULL ? statement : next_->Replace(context, statement);
 158 }
 159
 160 CYExpression *CYCondition::Replace(CYContext &context) {
 161     context.Replace(test_);
 162     context.Replace(true_);
 163     context.Replace(false_);
 164     return this;
 165 }
 166
 167 CYStatement *CYContinue::Replace(CYContext &context) {
 168     return this;
 169 }
 170
 171 CYAssignment *CYDeclaration::Assignment(CYContext &context) {
 172     CYExpression *variable(Replace(context));
 173     return initialiser_ == NULL ? NULL : $ CYAssign(variable, initialiser_);
 174 }
 175
 176 CYExpression *CYDeclaration::ForEachIn(CYContext &context) {
 177     return $ CYVariable(identifier_);
 178 }
 179
 180 CYExpression *CYDeclaration::Replace(CYContext &context) {
 181     context.Replace(identifier_);
 182     context.scope_->Declare(context, identifier_, CYIdentifierVariable);
 183     return $ CYVariable(identifier_);
 184 }
 185
 186 CYProperty *CYDeclarations::Property(CYContext &context) { $T(NULL)
 187     return $ CYProperty(declaration_->identifier_, declaration_->initialiser_ ?: $U, next_->Property(context));
 188 }
 189
 190 CYCompound *CYDeclarations::Replace(CYContext &context) {
 191     CYCompound *compound;
 192     if (next_ == NULL) compound:
 193         compound = $ CYCompound();
 194     else {
 195         compound = next_->Replace(context);
 196         if (compound == NULL)
 197             goto compound;
 198     }
 199
 200     if (CYAssignment *assignment = declaration_->Assignment(context))
 201         compound->AddPrev(assignment);
 202     return compound;
 203 }
 204
 205 CYExpression *CYDirectMember::Replace(CYContext &context) {
 206     Replace_(context);
 207     return this;
 208 }
 209
 210 CYStatement *CYDoWhile::Replace(CYContext &context) {
 211     context.Replace(test_);
 212     context.Replace(code_);
 213     return this;
 214 }
 215
 216 void CYElement::Replace(CYContext &context) { $T()
 217     context.Replace(value_);
 218     next_->Replace(context);
 219 }
 220
 221 CYStatement *CYEmpty::Collapse(CYContext &context) {
 222     return next_;
 223 }
 224
 225 CYStatement *CYEmpty::Replace(CYContext &context) {
 226     return this;
 227 }
 228
 229 CYStatement *CYExpress::Collapse(CYContext &context) {
 230     if (CYExpress *express = dynamic_cast<CYExpress *>(next_)) {
 231         CYCompound *next(dynamic_cast<CYCompound *>(express->expression_));
 232         if (next == NULL)
 233             next = $ CYCompound(express->expression_);
 234         next->AddPrev(expression_);
 235         expression_ = next;
 236         SetNext(express->next_);
 237     }
 238
 239     return this;
 240 }
 241
 242 CYStatement *CYExpress::Replace(CYContext &context) {
 243     context.Replace(expression_);
 244     if (expression_ == NULL)
 245         return $ CYEmpty();
 246     return this;
 247 }
 248
 249 CYExpression *CYExpression::ClassName(CYContext &context, bool object) {
 250     return this;
 251 }
 252
 253 CYExpression *CYExpression::ForEachIn(CYContext &context) {
 254     return this;
 255 }
 256
 257 CYExpression *CYExpression::ReplaceAll(CYContext &context) { $T(NULL)
 258     CYExpression *replace(this);
 259     context.Replace(replace);
 260
 261     if (CYExpression *next = next_->ReplaceAll(context))
 262         replace->SetNext(next);
 263     else
 264         replace->SetNext(next_);
 265
 266     return replace;
 267 }
 268
 269 CYNumber *CYFalse::Number(CYContext &context) {
 270     return $D(0);
 271 }
 272
 273 CYString *CYFalse::String(CYContext &context) {
 274     return $S("false");
 275 }
 276
 277 void CYFinally::Replace(CYContext &context) { $T()
 278     code_.Replace(context);
 279 }
 280
 281 CYStatement *CYFor::Replace(CYContext &context) {
 282     context.Replace(initialiser_);
 283     context.Replace(test_);
 284     context.Replace(increment_);
 285     context.Replace(code_);
 286     return this;
 287 }
 288
 289 CYStatement *CYForIn::Replace(CYContext &context) {
 290     // XXX: this actually might need a prefix statement
 291     context.Replace(initialiser_);
 292     context.Replace(set_);
 293     context.Replace(code_);
 294     return this;
 295 }
 296
 297 CYFunctionParameter *CYForInComprehension::Parameter(CYContext &context) const {
 298     return $ CYFunctionParameter(name_);
 299 }
 300
 301 CYStatement *CYForInComprehension::Replace(CYContext &context, CYStatement *statement) const {
 302     return $ CYForIn($ CYVariable(name_), set_, CYComprehension::Replace(context, statement));
 303 }
 304
 305 CYStatement *CYForEachIn::Replace(CYContext &context) {
 306     CYVariable *cys($V("$cys")), *cyt($V("$cyt"));
 307
 308     return $ CYLet($L2($L($I("$cys"), set_), $L($I("$cyt"))), $$->*
 309         $ CYForIn(cyt, cys, $ CYBlock($$->*
 310             $E($ CYAssign(initialiser_->ForEachIn(context), $M(cys, cyt)))->*
 311             code_
 312         ))
 313     );
 314 }
 315
 316 CYFunctionParameter *CYForEachInComprehension::Parameter(CYContext &context) const {
 317     return $ CYFunctionParameter(name_);
 318 }
 319
 320 CYStatement *CYForEachInComprehension::Replace(CYContext &context, CYStatement *statement) const {
 321     CYVariable *cys($V("$cys")), *name($ CYVariable(name_));
 322
 323     return $E($C0($F(NULL, $P1("$cys"), $$->*
 324         $E($ CYAssign(cys, set_))->*
 325         $ CYForIn(name, cys, $ CYBlock($$->*
 326             $E($ CYAssign(name, $M(cys, name)))->*
 327             CYComprehension::Replace(context, statement)
 328         ))
 329     )));
 330 }
 331
 332 void CYFunction::Inject(CYContext &context) {
 333     context.Replace(name_);
 334     context.scope_->Declare(context, name_, CYIdentifierOther);
 335 }
 336
 337 void CYFunction::Replace_(CYContext &context, bool outer) {
 338     if (outer)
 339         Inject(context);
 340
 341     parent_ = context.scope_;
 342     context.scope_ = this;
 343
 344     if (!outer && name_ != NULL)
 345         Inject(context);
 346
 347     parameters_->Replace(context);
 348     code_.Replace(context);
 349
 350     context.scope_ = parent_;
 351     Scope(context, code_.statements_);
 352 }
 353
 354 CYExpression *CYFunctionExpression::Replace(CYContext &context) {
 355     Replace_(context, false);
 356     return this;
 357 }
 358
 359 void CYFunctionParameter::Replace(CYContext &context) { $T()
 360     name_ = name_->Replace(context);
 361     context.scope_->Declare(context, name_, CYIdentifierArgument);
 362     next_->Replace(context);
 363 }
 364
 365 CYStatement *CYFunctionStatement::Replace(CYContext &context) {
 366     Replace_(context, true);
 367     return this;
 368 }
 369
 370 CYIdentifier *CYIdentifier::Replace(CYContext &context) {
 371     if (replace_ == NULL)
 372         replace_ = context.scope_->Lookup(context, this);
 373     return replace_;
 374 }
 375
 376 CYStatement *CYIf::Replace(CYContext &context) {
 377     context.Replace(test_);
 378     context.Replace(true_);
 379     context.Replace(false_);
 380     return this;
 381 }
 382
 383 CYFunctionParameter *CYIfComprehension::Parameter(CYContext &context) const {
 384     return NULL;
 385 }
 386
 387 CYStatement *CYIfComprehension::Replace(CYContext &context, CYStatement *statement) const {
 388     return $ CYIf(test_, CYComprehension::Replace(context, statement));
 389 }
 390
 391 CYExpression *CYIndirect::Replace(CYContext &context) {
 392     CYPrefix::Replace(context);
 393     return $M(rhs_, $S("$cyi"));
 394 }
 395
 396 CYExpression *CYIndirectMember::Replace(CYContext &context) {
 397     Replace_(context);
 398     return $M($ CYIndirect(object_), property_);
 399 }
 400
 401 CYExpression *CYInfix::Replace(CYContext &context) {
 402     context.Replace(lhs_);
 403     context.Replace(rhs_);
 404     return this;
 405 }
 406
 407 CYStatement *CYLabel::Replace(CYContext &context) {
 408     context.Replace(statement_);
 409     return this;
 410 }
 411
 412 CYStatement *CYLet::Replace(CYContext &context) {
 413     return $ CYWith($ CYObject(declarations_->Property(context)), &code_);
 414 }
 415
 416 void CYMember::Replace_(CYContext &context) {
 417     context.Replace(object_);
 418     context.Replace(property_);
 419 }
 420
 421 CYExpression *CYNew::Replace(CYContext &context) {
 422     context.Replace(constructor_);
 423     arguments_->Replace(context);
 424     return this;
 425 }
 426
 427 CYNumber *CYNull::Number(CYContext &context) {
 428     return $D(0);
 429 }
 430
 431 CYString *CYNull::String(CYContext &context) {
 432     return $S("null");
 433 }
 434
 435 CYNumber *CYNumber::Number(CYContext &context) {
 436     return this;
 437 }
 438
 439 CYString *CYNumber::String(CYContext &context) {
 440     // XXX: there is a precise algorithm for this
 441     return $S(apr_psprintf(context.pool_, "%.17g", Value()));
 442 }
 443
 444 CYExpression *CYObject::Replace(CYContext &context) {
 445     properties_->Replace(context);
 446     return this;
 447 }
 448
 449 CYExpression *CYPostfix::Replace(CYContext &context) {
 450     context.Replace(lhs_);
 451     return this;
 452 }
 453
 454 CYExpression *CYPrefix::Replace(CYContext &context) {
 455     context.Replace(rhs_);
 456     return this;
 457 }
 458
 459 void CYProgram::Replace(CYContext &context) {
 460     parent_ = context.scope_;
 461     CYProgram *program(context.program_);
 462
 463     context.scope_ = this;
 464     context.program_ = this;
 465
 466     statements_ = statements_->ReplaceAll(context);
 467
 468     context.scope_ = parent_;
 469     context.program_ = program;
 470     Scope(context, statements_);
 471
 472     size_t offset(0);
 473
 474     CYCStringSet external;
 475     for (CYIdentifierValueSet::const_iterator i(identifiers_.begin()); i != identifiers_.end(); ++i)
 476         external.insert((*i)->Word());
 477
 478     // XXX: totalling the probable occurrences and sorting by them would improve the result
 479     for (CYIdentifierAddressVector::const_iterator i(rename_.begin()); i != rename_.end(); ++i, ++offset) {
 480         const char *name;
 481
 482         if (context.options_.verbose_)
 483             name = apr_psprintf(context.pool_, "$%"APR_SIZE_T_FMT"", offset);
 484         else {
 485             char id[8];
 486             id[7] = '\0';
 487
 488           id:
 489             unsigned position(7), local(offset + 1);
 490
 491             do {
 492                 unsigned index(local % 53);
 493                 local /= 53;
 494                 id[--position] = index == 0 ? '0' : index < 27 ? index - 1 + 'a' : index - 27 + 'A';
 495             } while (local != 0);
 496
 497             if (external.find(id + position) != external.end()) {
 498                 ++offset;
 499                 goto id;
 500             }
 501
 502             name = apr_pstrmemdup(context.pool_, id + position, 7 - position);
 503             // XXX: at some point, this could become a keyword
 504         }
 505
 506         for (CYIdentifier *identifier(*i); identifier != NULL; identifier = identifier->next_)
 507             identifier->Set(name);
 508     }
 509 }
 510
 511 void CYProperty::Replace(CYContext &context) { $T()
 512     context.Replace(value_);
 513     next_->Replace(context);
 514 }
 515
 516 CYStatement *CYReturn::Replace(CYContext &context) {
 517     context.Replace(value_);
 518     return this;
 519 }
 520
 521 void CYScope::Declare(CYContext &context, CYIdentifier *identifier, CYIdentifierFlags flags) {
 522     internal_.insert(CYIdentifierAddressFlagsMap::value_type(identifier, flags));
 523 }
 524
 525 CYIdentifier *CYScope::Lookup(CYContext &context, CYIdentifier *identifier) {
 526     std::pair<CYIdentifierValueSet::iterator, bool> insert(identifiers_.insert(identifier));
 527     return *insert.first;
 528 }
 529
 530 void CYScope::Merge(CYContext &context, CYIdentifier *identifier) {
 531     std::pair<CYIdentifierAddressSet::iterator, bool> insert(identifiers_.insert(identifier));
 532     if (!insert.second) {
 533         if ((*insert.first)->offset_ < identifier->offset_)
 534             (*insert.first)->offset_ = identifier->offset_;
 535         identifier->replace_ = *insert.first;
 536     }
 537 }
 538
 539 void CYScope::Scope(CYContext &context, CYStatement *&statements) {
 540     CYDeclarations *last(NULL), *curr(NULL);
 541     CYProgram *program(context.program_);
 542
 543     typedef std::multimap<size_t, CYIdentifier *> IdentifierOffsetMap;
 544     IdentifierOffsetMap offsetted;
 545
 546     // XXX: we don't want to do this in order, we want to sort it by probable occurrence
 547     for (CYIdentifierAddressFlagsMap::const_iterator i(internal_.begin()); i != internal_.end(); ++i) {
 548         if (program != NULL && i->second != CYIdentifierMagic)
 549             offsetted.insert(IdentifierOffsetMap::value_type(i->first->offset_, i->first));
 550         if (i->second == CYIdentifierVariable) {
 551             CYDeclarations *next($ CYDeclarations($ CYDeclaration(i->first)));
 552             if (last == NULL)
 553                 last = next;
 554             if (curr != NULL)
 555                 curr->SetNext(next);
 556             curr = next;
 557         }
 558     }
 559
 560     size_t offset(0);
 561
 562     for (IdentifierOffsetMap::const_iterator i(offsetted.begin()); i != offsetted.end(); ++i) {
 563         if (offset < i->first)
 564             offset = i->first;
 565         if (program->rename_.size() <= offset)
 566             program->rename_.resize(offset + 1);
 567         CYIdentifier *&identifier(program->rename_[offset++]);
 568         i->second->SetNext(identifier);
 569         identifier = i->second;
 570     }
 571
 572     if (last != NULL) {
 573         CYVar *var($ CYVar(last));
 574         var->SetNext(statements);
 575         statements = var;
 576     }
 577
 578     for (CYIdentifierValueSet::const_iterator i(identifiers_.begin()); i != identifiers_.end(); ++i)
 579         if (internal_.find(*i) == internal_.end()) {
 580             if ((*i)->offset_ < offset)
 581                 (*i)->offset_ = offset;
 582             parent_->Merge(context, *i);
 583         }
 584 }
 585
 586 CYStatement *CYStatement::Collapse(CYContext &context) {
 587     return this;
 588 }
 589
 590 CYStatement *CYStatement::ReplaceAll(CYContext &context) { $T(NULL)
 591     CYStatement *replace(this);
 592     context.Replace(replace);
 593     replace->SetNext(next_->ReplaceAll(context));
 594     return replace->Collapse(context);
 595 }
 596
 597 CYString *CYString::Concat(CYContext &context, CYString *rhs) const {
 598     size_t size(size_ + rhs->size_);
 599     char *value(new(context.pool_) char[size + 1]);
 600     memcpy(value, value_, size_);
 601     memcpy(value + size_, rhs->value_, rhs->size_);
 602     value[size] = '\0';
 603     return $S(value, size);
 604 }
 605
 606 CYNumber *CYString::Number(CYContext &context) {
 607     // XXX: there is a precise algorithm for this
 608     return NULL;
 609 }
 610
 611 CYString *CYString::String(CYContext &context) {
 612     return this;
 613 }
 614
 615 CYStatement *CYSwitch::Replace(CYContext &context) {
 616     context.Replace(value_);
 617     clauses_->Replace(context);
 618     return this;
 619 }
 620
 621 CYExpression *CYThis::Replace(CYContext &context) {
 622     return this;
 623 }
 624
 625 namespace cy {
 626 namespace Syntax {
 627
 628 CYStatement *Throw::Replace(CYContext &context) {
 629     context.Replace(value_);
 630     return this;
 631 }
 632
 633 } }
 634
 635 CYExpression *CYTrivial::Replace(CYContext &context) {
 636     return this;
 637 }
 638
 639 CYNumber *CYTrue::Number(CYContext &context) {
 640     return $D(1);
 641 }
 642
 643 CYString *CYTrue::String(CYContext &context) {
 644     return $S("true");
 645 }
 646
 647 namespace cy {
 648 namespace Syntax {
 649
 650 CYStatement *Try::Replace(CYContext &context) {
 651     code_.Replace(context);
 652     catch_->Replace(context);
 653     finally_->Replace(context);
 654     return this;
 655 }
 656
 657 } }
 658
 659 CYStatement *CYVar::Replace(CYContext &context) {
 660     return $E(declarations_->Replace(context));
 661 }
 662
 663 CYExpression *CYVariable::Replace(CYContext &context) {
 664     name_ = name_->Replace(context);
 665     return this;
 666 }
 667
 668 CYStatement *CYWhile::Replace(CYContext &context) {
 669     context.Replace(test_);
 670     context.Replace(code_);
 671     return this;
 672 }
 673
 674 CYStatement *CYWith::Replace(CYContext &context) {
 675     context.Replace(scope_);
 676     context.Replace(code_);
 677     return this;
 678 }
 679
 680 CYExpression *CYWord::ClassName(CYContext &context, bool object) {
 681     CYString *name($S(this));
 682     if (object)
 683         return $C1($V("objc_getClass"), name);
 684     else
 685         return name;
 686 }