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