-/* Cycript - Remote Execution Server and Disassembler
+/* Cycript - Inlining/Optimizing JavaScript Compiler
* Copyright (C) 2009 Jay Freeman (saurik)
*/
#include <string>
#include <vector>
+#include <map>
+#include <set>
#include <cstdlib>
#include "location.hh"
#include "Pooling.hpp"
+#include "Options.hpp"
+
+class CYContext;
template <typename Type_>
struct CYNext {
};
struct CYThing {
+ virtual ~CYThing() {
+ }
+
virtual void Output(struct CYOutput &out) const = 0;
};
struct CYOutput {
std::ostream &out_;
+ CYOptions &options_;
bool pretty_;
unsigned indent_;
+ bool right_;
enum {
NoMode,
Terminated
} mode_;
- CYOutput(std::ostream &out) :
+ CYOutput(std::ostream &out, CYOptions &options) :
out_(out),
+ options_(options),
pretty_(false),
indent_(0),
+ right_(false),
mode_(NoMode)
{
}
struct CYPropertyName {
virtual void PropertyName(CYOutput &out) const = 0;
+
+ virtual ~CYPropertyName() {
+ }
};
struct CYExpression;
CYNoCall = (1 << 3),
CYNoRightHand = (1 << 4),
CYNoDangle = (1 << 5),
+ CYNoInteger = (1 << 6),
CYNoBF = (CYNoBrace | CYNoFunction),
};
-struct CYContext {
- apr_pool_t *pool_;
-
- CYContext(apr_pool_t *pool) :
- pool_(pool)
- {
- }
-
- template <typename Type_>
- void Replace(Type_ *&value) {
- if (value != NULL)
- while (Type_ *replace = value->Replace(*this))
- value = replace;
- }
-};
-
struct CYStatement :
CYNext<CYStatement>
{
+ virtual ~CYStatement() {
+ }
+
void Single(CYOutput &out, CYFlags flags) const;
void Multiple(CYOutput &out, CYFlags flags = CYNoFlags) const;
CYStatement *ReplaceAll(CYContext &context);
+ virtual CYStatement *Collapse(CYContext &context);
virtual CYStatement *Replace(CYContext &context) = 0;
};
struct CYClassName {
+ virtual ~CYClassName() {
+ }
+
virtual CYExpression *ClassName(CYContext &context, bool object) = 0;
virtual void ClassName(CYOutput &out, bool object) const = 0;
};
{
}
- const char *Value() const {
- return word_;
+ void Set(const char *value) {
+ word_ = value;
}
+ virtual const char *Word() const;
virtual void Output(CYOutput &out) const;
virtual CYExpression *ClassName(CYContext &context, bool object);
};
_finline std::ostream &operator <<(std::ostream &lhs, const CYWord &rhs) {
- return lhs << rhs.Value();
+ lhs << &rhs << '=';
+ return lhs << rhs.Word();
}
struct CYIdentifier :
+ CYNext<CYIdentifier>,
CYWord
{
+ CYIdentifier *replace_;
+ size_t offset_;
+
CYIdentifier(const char *word) :
- CYWord(word)
+ CYWord(word),
+ replace_(NULL),
+ offset_(0)
+ {
+ }
+
+ virtual const char *Word() const;
+ CYIdentifier *Replace(CYContext &context);
+};
+
+struct CYComment :
+ CYStatement
+{
+ const char *value_;
+
+ CYComment(const char *value) :
+ value_(value)
{
}
+
+ virtual CYStatement *Replace(CYContext &context);
+ virtual void Output(CYOutput &out, CYFlags flags) const;
};
struct CYLabel :
virtual void Output(CYOutput &out, CYFlags flags) const;
};
+struct CYCStringLess :
+ std::binary_function<const char *, const char *, bool>
+{
+ _finline bool operator ()(const char *lhs, const char *rhs) const {
+ return strcmp(lhs, rhs) < 0;
+ }
+};
+
+struct CYIdentifierValueLess :
+ std::binary_function<CYIdentifier *, CYIdentifier *, bool>
+{
+ _finline bool operator ()(CYIdentifier *lhs, CYIdentifier *rhs) const {
+ return CYCStringLess()(lhs->Word(), rhs->Word());
+ }
+};
+
+enum CYIdentifierFlags {
+ CYIdentifierArgument,
+ CYIdentifierVariable,
+ CYIdentifierOther,
+ CYIdentifierMagic,
+};
+
+typedef std::set<const char *, CYCStringLess> CYCStringSet;
+typedef std::set<CYIdentifier *, CYIdentifierValueLess> CYIdentifierValueSet;
+typedef std::vector<CYIdentifier *> CYIdentifierAddressVector;
+typedef std::map<CYIdentifier *, CYIdentifierFlags> CYIdentifierAddressFlagsMap;
+
+struct CYScope {
+ CYScope *parent_;
+ CYIdentifierAddressFlagsMap internal_;
+
+ CYIdentifierValueSet identifiers_;
+
+ CYScope() :
+ parent_(NULL)
+ {
+ }
+
+ virtual ~CYScope() {
+ }
+
+ void Declare(CYContext &context, CYIdentifier *identifier, CYIdentifierFlags flags);
+ virtual CYIdentifier *Lookup(CYContext &context, CYIdentifier *identifier);
+ void Merge(CYContext &context, CYIdentifier *identifier);
+ void Scope(CYContext &context, CYStatement *&statements);
+};
+
struct CYProgram :
+ CYScope,
CYThing
{
CYStatement *statements_;
+ CYIdentifierAddressVector rename_;
CYProgram(CYStatement *statements) :
statements_(statements)
}
virtual void Replace(CYContext &context);
-
virtual void Output(CYOutput &out) const;
};
+struct CYContext :
+ CYScope
+{
+ apr_pool_t *pool_;
+ CYOptions &options_;
+ CYScope *scope_;
+ CYProgram *program_;
+
+ CYContext(apr_pool_t *pool, CYOptions &options) :
+ pool_(pool),
+ options_(options),
+ scope_(this),
+ program_(NULL)
+ {
+ }
+
+ virtual ~CYContext() {
+ }
+
+ template <typename Type_>
+ void Replace(Type_ *&value) {
+ for (;;) if (value == NULL)
+ break;
+ else {
+ Type_ *replace(value->Replace(*this));
+ if (replace != value)
+ value = replace;
+ else break;
+ }
+ }
+};
+
struct CYBlock :
CYStatement,
CYThing
{
CYStatement *statements_;
+ CYScope *scope_;
- CYBlock(CYStatement *statements) :
- statements_(statements)
+ CYBlock(CYStatement *statements, CYScope *scope = NULL) :
+ statements_(statements),
+ scope_(scope)
{
}
return statements_;
}
+ void AddPrev(CYStatement *statement) {
+ CYStatement *last(statement);
+ while (last->next_ != NULL)
+ last = last->next_;
+ last->SetNext(statements_);
+ statements_ = statement;
+ }
+
virtual CYStatement *Replace(CYContext &context);
virtual void Output(CYOutput &out) const;
enum Condition {
RegExpCondition,
+ XMLContentCondition,
+ XMLTagCondition,
};
std::string filename_;
CYDriver(const std::string &filename);
~CYDriver();
- void BeginCondition(Condition condition);
+ Condition GetCondition();
+ void SetCondition(Condition condition);
+
+ void PushCondition(Condition condition);
+ void PopCondition();
void Warning(const cy::location &location, const char *message);
};
struct CYForInitialiser {
+ virtual ~CYForInitialiser() {
+ }
+
virtual void For(CYOutput &out) const = 0;
+ virtual CYExpression *Replace(CYContext &context) = 0;
};
struct CYForInInitialiser {
+ virtual ~CYForInInitialiser() {
+ }
+
virtual void ForIn(CYOutput &out, CYFlags flags) const = 0;
virtual const char *ForEachIn() const = 0;
virtual CYExpression *ForEachIn(CYContext &out) = 0;
+ virtual CYExpression *Replace(CYContext &context) = 0;
};
struct CYNumber;
{
CYExpression *expressions_;
- CYCompound(CYExpression *expressions) :
+ CYCompound(CYExpression *expressions = NULL) :
expressions_(expressions)
{
}
{
}
+ void Replace(CYContext &context);
virtual void Output(CYOutput &out) const;
};
}
virtual const char *Name() const {
- return name_->Value();
+ return name_->Word();
}
virtual CYFunctionParameter *Parameter(CYContext &context) const;
}
virtual const char *Name() const {
- return name_->Value();
+ return name_->Word();
}
virtual CYFunctionParameter *Parameter(CYContext &context) const;
}
CYString(const CYWord *word) :
- value_(word->Value()),
+ value_(word->Word()),
size_(strlen(value_))
{
}
virtual CYNumber *Number(CYContext &context);
virtual CYString *String(CYContext &context);
- virtual CYString *Concat(CYContext &out, CYString *rhs) const;
+ CYString *Concat(CYContext &out, CYString *rhs) const;
virtual void Output(CYOutput &out, CYFlags flags) const;
virtual void PropertyName(CYOutput &out) const;
};
virtual void Output(CYOutput &out, CYFlags flags) const;
};
+struct CYProperty :
+ CYNext<CYProperty>,
+ CYThing
+{
+ CYPropertyName *name_;
+ CYExpression *value_;
+
+ CYProperty(CYPropertyName *name, CYExpression *value, CYProperty *next = NULL) :
+ CYNext<CYProperty>(next),
+ name_(name),
+ value_(value)
+ {
+ }
+
+ void Replace(CYContext &context);
+ virtual void Output(CYOutput &out) const;
+};
+
struct CYDeclaration :
CYForInInitialiser
{
CYIdentifier *identifier_;
CYExpression *initialiser_;
- CYDeclaration(CYIdentifier *identifier, CYExpression *initialiser) :
+ CYDeclaration(CYIdentifier *identifier, CYExpression *initialiser = NULL) :
identifier_(identifier),
initialiser_(initialiser)
{
virtual const char *ForEachIn() const;
virtual CYExpression *ForEachIn(CYContext &out);
- void Replace(CYContext &context);
+ virtual CYExpression *Replace(CYContext &context);
+ virtual CYAssignment *Assignment(CYContext &context);
virtual void Output(CYOutput &out, CYFlags flags) const;
};
struct CYDeclarations :
CYNext<CYDeclarations>,
- CYForInitialiser,
- CYThing
+ CYThing,
+ CYForInitialiser
{
CYDeclaration *declaration_;
virtual void For(CYOutput &out) const;
- void Replace(CYContext &context);
+ virtual CYCompound *Replace(CYContext &context);
+ CYProperty *Property(CYContext &context);
virtual void Output(CYOutput &out) const;
virtual void Output(CYOutput &out, CYFlags flags) const;
virtual void Output(CYOutput &out, CYFlags flags) const;
};
-struct CYProperty :
- CYNext<CYProperty>,
- CYThing
-{
- CYPropertyName *name_;
- CYExpression *value_;
-
- CYProperty(CYPropertyName *name, CYExpression *value, CYProperty *next = NULL) :
- CYNext<CYProperty>(next),
- name_(name),
- value_(value)
- {
- }
-
- void Replace(CYContext &context);
- virtual void Output(CYOutput &out) const;
-};
-
struct CYObject :
CYLiteral
{
void Output(CYOutput &out, CYFlags flags) const;
};
-struct CYCatch :
- CYThing
-{
- CYIdentifier *name_;
- CYBlock code_;
-
- CYCatch(CYIdentifier *name, CYStatement *statements) :
- name_(name),
- code_(statements)
- {
- }
-
- void Replace(CYContext &context);
- virtual void Output(CYOutput &out) const;
-};
-
struct CYMember :
CYExpression
{
virtual void Output(CYOutput &out, CYFlags flags) const;
};
-struct CYFunction {
+struct CYFunction :
+ CYScope
+{
CYIdentifier *name_;
CYFunctionParameter *parameters_;
CYBlock code_;
CYFunction(CYIdentifier *name, CYFunctionParameter *parameters, CYStatement *statements) :
name_(name),
parameters_(parameters),
- code_(statements)
+ code_(statements, this)
{
}
- virtual void Replace_(CYContext &context);
+ virtual ~CYFunction() {
+ }
+
+ void Inject(CYContext &context);
+ virtual void Replace_(CYContext &context, bool outer);
virtual void Output(CYOutput &out, CYFlags flags) const;
};
CYExpress(CYExpression *expression) :
expression_(expression)
{
+ if (expression == NULL)
+ throw;
}
+ virtual CYStatement *Collapse(CYContext &context);
virtual CYStatement *Replace(CYContext &context);
virtual void Output(CYOutput &out, CYFlags flags) const;
};
struct CYEmpty :
CYStatement
{
+ virtual CYStatement *Collapse(CYContext &context);
virtual CYStatement *Replace(CYContext &context);
virtual void Output(CYOutput &out, CYFlags flags) const;
};
virtual void Output(CYOutput &out) const;
};
-struct CYTry :
+namespace cy {
+namespace Syntax {
+
+struct Catch :
+ CYThing
+{
+ CYIdentifier *name_;
+ CYBlock code_;
+
+ Catch(CYIdentifier *name, CYStatement *statements) :
+ name_(name),
+ code_(statements)
+ {
+ }
+
+ void Replace(CYContext &context);
+ virtual void Output(CYOutput &out) const;
+};
+
+struct Try :
CYStatement
{
CYBlock code_;
- CYCatch *catch_;
+ Catch *catch_;
CYFinally *finally_;
- CYTry(CYStatement *statements, CYCatch *_catch, CYFinally *finally) :
+ Try(CYStatement *statements, Catch *_catch, CYFinally *finally) :
code_(statements),
catch_(_catch),
finally_(finally)
virtual void Output(CYOutput &out, CYFlags flags) const;
};
-struct CYThrow :
+struct Throw :
CYStatement
{
CYExpression *value_;
- CYThrow(CYExpression *value) :
+ Throw(CYExpression *value) :
value_(value)
{
}
virtual void Output(CYOutput &out, CYFlags flags) const;
};
+} }
+
struct CYWith :
CYStatement
{