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::Replace_(CYContext &context) {
 335     parent_ = context.scope_;
 336     context.scope_ = this;
 337
 338     parameters_->Replace(context);
 339     code_.Replace(context);
 340
 341     context.scope_ = parent_;
 342     Scope(context, code_.statements_);
 343 }
 344
 345 CYExpression *CYFunctionExpression::Replace(CYContext &context) {
 346     Replace_(context);
 347     return this;
 348 }
 349
 350 void CYFunctionParameter::Replace(CYContext &context) { $T()
 351     name_ = name_->Replace(context);
 352     context.scope_->internal_.insert(CYIdentifierAddressFlagsMap::value_type(name_, CYIdentifierArgument));
 353     next_->Replace(context);
 354 }
 355
 356 CYStatement *CYFunctionStatement::Replace(CYContext &context) {
 357     Replace_(context);
 358     return this;
 359 }
 360
 361 CYIdentifier *CYIdentifier::Replace(CYContext &context) {
 362     if (replace_ != NULL)
 363         return replace_;
 364
 365     CYIdentifierValueSet &identifiers(context.scope_->identifiers_);
 366     std::pair<CYIdentifierValueSet::iterator, bool> insert(identifiers.insert(this));
 367     if (!insert.second)
 368         return *insert.first;
 369
 370     replace_ = this;
 371     return this;
 372 }
 373
 374 CYStatement *CYIf::Replace(CYContext &context) {
 375     context.Replace(test_);
 376     context.Replace(true_);
 377     context.Replace(false_);
 378     return this;
 379 }
 380
 381 CYFunctionParameter *CYIfComprehension::Parameter(CYContext &context) const {
 382     return NULL;
 383 }
 384
 385 CYStatement *CYIfComprehension::Replace(CYContext &context, CYStatement *statement) const {
 386     return $ CYIf(test_, CYComprehension::Replace(context, statement));
 387 }
 388
 389 CYExpression *CYIndirect::Replace(CYContext &context) {
 390     CYPrefix::Replace(context);
 391     return $M(rhs_, $S("$cyi"));
 392 }
 393
 394 CYExpression *CYIndirectMember::Replace(CYContext &context) {
 395     Replace_(context);
 396     return $M($ CYIndirect(object_), property_);
 397 }
 398
 399 CYExpression *CYInfix::Replace(CYContext &context) {
 400     context.Replace(lhs_);
 401     context.Replace(rhs_);
 402     return this;
 403 }
 404
 405 CYStatement *CYLabel::Replace(CYContext &context) {
 406     context.Replace(statement_);
 407     return this;
 408 }
 409
 410 CYStatement *CYLet::Replace(CYContext &context) {
 411     return $ CYWith($ CYObject(declarations_->Property(context)), &code_);
 412 }
 413
 414 void CYMember::Replace_(CYContext &context) {
 415     context.Replace(object_);
 416     context.Replace(property_);
 417 }
 418
 419 CYExpression *CYNew::Replace(CYContext &context) {
 420     context.Replace(constructor_);
 421     arguments_->Replace(context);
 422     return this;
 423 }
 424
 425 CYNumber *CYNull::Number(CYContext &context) {
 426     return $D(0);
 427 }
 428
 429 CYString *CYNull::String(CYContext &context) {
 430     return $S("null");
 431 }
 432
 433 CYNumber *CYNumber::Number(CYContext &context) {
 434     return this;
 435 }
 436
 437 CYString *CYNumber::String(CYContext &context) {
 438     // XXX: there is a precise algorithm for this
 439     return $S(apr_psprintf(context.pool_, "%.17g", Value()));
 440 }
 441
 442 CYExpression *CYObject::Replace(CYContext &context) {
 443     properties_->Replace(context);
 444     return this;
 445 }
 446
 447 CYExpression *CYPostfix::Replace(CYContext &context) {
 448     context.Replace(lhs_);
 449     return this;
 450 }
 451
 452 CYExpression *CYPrefix::Replace(CYContext &context) {
 453     context.Replace(rhs_);
 454     return this;
 455 }
 456
 457 void CYProgram::Replace(CYContext &context) {
 458     parent_ = context.scope_;
 459     context.scope_ = this;
 460     statements_ = statements_->ReplaceAll(context);
 461     context.scope_ = parent_;
 462     Scope(context, statements_);
 463 }
 464
 465 void CYProperty::Replace(CYContext &context) { $T()
 466     context.Replace(value_);
 467     next_->Replace(context);
 468 }
 469
 470 CYStatement *CYReturn::Replace(CYContext &context) {
 471     context.Replace(value_);
 472     return this;
 473 }
 474
 475 void CYScope::Add(CYContext &context, CYIdentifierAddressSet &external) {
 476     for (CYIdentifierAddressSet::const_iterator i(external.begin()); i != external.end(); ++i) {
 477         std::pair<CYIdentifierAddressSet::iterator, bool> insert(identifiers_.insert(*i));
 478         if (!insert.second)
 479             (*i)->replace_ = *insert.first;
 480     }
 481 }
 482
 483 void CYScope::Scope(CYContext &context, CYStatement *&statements) {
 484     CYIdentifierAddressSet external;
 485
 486     if (context.options_.verbose_)
 487         std::cout << this << ':';
 488
 489     CYDeclarations *last(NULL), *curr(NULL);
 490
 491     for (CYIdentifierValueSet::const_iterator i(identifiers_.begin()); i != identifiers_.end(); ++i)
 492         if (internal_.find(*i) == internal_.end()) {
 493             if (context.options_.verbose_)
 494                 std::cout << ' ' << (*i)->Word() << '@' << static_cast<const CYWord *>(*i);
 495             external.insert(*i);
 496         } else {
 497             if (context.options_.verbose_) {
 498                 std::cout << ' ' << offset_ << ':' << (*i)->Word() << '@' << static_cast<const CYWord *>(*i);
 499                 (*i)->Set(apr_psprintf(context.pool_, "$%u", offset_++));
 500             } else {
 501                 (*i)->Set(apr_psprintf(context.pool_, "$%u", offset_++));
 502             }
 503
 504             CYDeclarations *next($ CYDeclarations($ CYDeclaration(*i)));
 505             if (last == NULL)
 506                 last = next;
 507             if (curr != NULL)
 508                 curr->SetNext(next);
 509             curr = next;
 510         }
 511
 512     if (context.options_.verbose_)
 513         std::cout << " ->" << parent_ << std::endl;
 514
 515     if (last != NULL) {
 516         CYVar *var($ CYVar(last));
 517         var->SetNext(statements);
 518         statements = var;
 519     }
 520
 521     if (parent_ != NULL) {
 522         if (parent_->offset_ < offset_)
 523             parent_->offset_ = offset_;
 524         parent_->Add(context, external);
 525     }
 526 }
 527
 528 CYStatement *CYStatement::Collapse(CYContext &context) {
 529     return this;
 530 }
 531
 532 CYStatement *CYStatement::ReplaceAll(CYContext &context) { $T(NULL)
 533     CYStatement *replace(this);
 534     context.Replace(replace);
 535     replace->SetNext(next_->ReplaceAll(context));
 536     return replace->Collapse(context);
 537 }
 538
 539 CYString *CYString::Concat(CYContext &context, CYString *rhs) const {
 540     size_t size(size_ + rhs->size_);
 541     char *value(new(context.pool_) char[size + 1]);
 542     memcpy(value, value_, size_);
 543     memcpy(value + size_, rhs->value_, rhs->size_);
 544     value[size] = '\0';
 545     return $S(value, size);
 546 }
 547
 548 CYNumber *CYString::Number(CYContext &context) {
 549     // XXX: there is a precise algorithm for this
 550     return NULL;
 551 }
 552
 553 CYString *CYString::String(CYContext &context) {
 554     return this;
 555 }
 556
 557 CYStatement *CYSwitch::Replace(CYContext &context) {
 558     context.Replace(value_);
 559     clauses_->Replace(context);
 560     return this;
 561 }
 562
 563 CYExpression *CYThis::Replace(CYContext &context) {
 564     return this;
 565 }
 566
 567 namespace cy {
 568 namespace Syntax {
 569
 570 CYStatement *Throw::Replace(CYContext &context) {
 571     context.Replace(value_);
 572     return this;
 573 }
 574
 575 } }
 576
 577 CYExpression *CYTrivial::Replace(CYContext &context) {
 578     return this;
 579 }
 580
 581 CYNumber *CYTrue::Number(CYContext &context) {
 582     return $D(1);
 583 }
 584
 585 CYString *CYTrue::String(CYContext &context) {
 586     return $S("true");
 587 }
 588
 589 namespace cy {
 590 namespace Syntax {
 591
 592 CYStatement *Try::Replace(CYContext &context) {
 593     code_.Replace(context);
 594     catch_->Replace(context);
 595     finally_->Replace(context);
 596     return this;
 597 }
 598
 599 } }
 600
 601 CYStatement *CYVar::Replace(CYContext &context) {
 602     return $E(declarations_->Replace(context));
 603 }
 604
 605 CYExpression *CYVariable::Replace(CYContext &context) {
 606     name_ = name_->Replace(context);
 607     return this;
 608 }
 609
 610 CYStatement *CYWhile::Replace(CYContext &context) {
 611     context.Replace(test_);
 612     context.Replace(code_);
 613     return this;
 614 }
 615
 616 CYStatement *CYWith::Replace(CYContext &context) {
 617     context.Replace(scope_);
 618     context.Replace(code_);
 619     return this;
 620 }
 621
 622 CYExpression *CYWord::ClassName(CYContext &context, bool object) {
 623     CYString *name($S(this));
 624     if (object)
 625         return $C1($V("objc_getClass"), name);
 626     else
 627         return name;
 628 }