-/* Cycript - Remove Execution Server and Disassembler
+/* Cycript - Inlining/Optimizing JavaScript Compiler
* Copyright (C) 2009 Jay Freeman (saurik)
*/
#include <string>
#include <vector>
+#include <map>
+#include <set>
+#include <cstdio>
#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_;
+ size_t usage_;
+
CYIdentifier(const char *word) :
- CYWord(word)
+ CYWord(word),
+ replace_(NULL),
+ offset_(0),
+ usage_(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::map<CYIdentifier *, CYIdentifierFlags> CYIdentifierAddressFlagsMap;
+
+struct CYIdentifierUsage {
+ CYIdentifier *identifier_;
+ size_t usage_;
+};
+
+typedef std::vector<CYIdentifierUsage> CYIdentifierUsageVector;
+
+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 :
CYThing
{
}
virtual void Replace(CYContext &context);
-
virtual void Output(CYOutput &out) const;
};
+struct CYContext {
+ apr_pool_t *pool_;
+ CYOptions &options_;
+ CYScope *scope_;
+ CYIdentifierUsageVector rename_;
+
+ CYContext(apr_pool_t *pool, CYOptions &options) :
+ pool_(pool),
+ options_(options),
+ scope_(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
{
}
+ operator CYStatement *() const {
+ 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;
bool strict_;
enum Condition {
- RegExStart,
- RegExRest
+ RegExpCondition,
+ XMLContentCondition,
+ XMLTagCondition,
};
std::string filename_;
CYDriver(const std::string &filename);
~CYDriver();
+ 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 void ForEachIn(CYOutput &out) const = 0;
virtual CYExpression *ForEachIn(CYContext &out) = 0;
+ virtual CYExpression *Replace(CYContext &context) = 0;
};
+struct CYNumber;
+struct CYString;
+
struct CYExpression :
CYNext<CYExpression>,
CYForInitialiser,
virtual void ForIn(CYOutput &out, CYFlags flags) const;
virtual const char *ForEachIn() const;
- virtual void ForEachIn(CYOutput &out) const;
virtual CYExpression *ForEachIn(CYContext &out);
virtual void Output(CYOutput &out) const;
virtual CYExpression *Replace(CYContext &context) = 0;
+ virtual CYExpression *Primitive(CYContext &context) {
+ return this;
+ }
+
+ virtual CYNumber *Number(CYContext &context) {
+ return NULL;
+ }
+
+ virtual CYString *String(CYContext &context) {
+ return NULL;
+ }
+
virtual const char *Word() const {
return NULL;
}
{
CYExpression *expressions_;
- CYCompound(CYExpression *expressions) :
+ CYCompound(CYExpression *expressions = NULL) :
expressions_(expressions)
{
}
{
}
+ void Replace(CYContext &context);
virtual void Output(CYOutput &out) const;
};
CYNext<CYComprehension>,
CYThing
{
- void Output(CYOutput &out) const;
virtual const char *Name() const = 0;
- virtual void Begin_(CYOutput &out) const = 0;
-
- virtual void End_(CYOutput &out) const {
- }
-
virtual CYFunctionParameter *Parameter(CYContext &context) const = 0;
CYFunctionParameter *Parameters(CYContext &context) const;
virtual CYStatement *Replace(CYContext &context, CYStatement *statement) const;
+ virtual void Output(CYOutput &out) const = 0;
};
struct CYForInComprehension :
}
virtual const char *Name() const {
- return name_->Value();
+ return name_->Word();
}
- virtual void Begin_(CYOutput &out) const;
-
virtual CYFunctionParameter *Parameter(CYContext &context) const;
virtual CYStatement *Replace(CYContext &context, CYStatement *statement) const;
+ virtual void Output(CYOutput &out) const;
};
struct CYForEachInComprehension :
}
virtual const char *Name() const {
- return name_->Value();
+ return name_->Word();
}
- virtual void Begin_(CYOutput &out) const;
- virtual void End_(CYOutput &out) const;
-
virtual CYFunctionParameter *Parameter(CYContext &context) const;
virtual CYStatement *Replace(CYContext &context, CYStatement *statement) const;
+ virtual void Output(CYOutput &out) const;
};
struct CYIfComprehension :
return NULL;
}
- virtual void Begin_(CYOutput &out) const;
-
virtual CYFunctionParameter *Parameter(CYContext &context) const;
virtual CYStatement *Replace(CYContext &context, CYStatement *statement) const;
+ virtual void Output(CYOutput &out) const;
};
struct CYArrayComprehension :
}
CYString(const CYWord *word) :
- value_(word->Value()),
+ value_(word->Word()),
size_(strlen(value_))
{
}
virtual const char *Word() const;
+ virtual CYNumber *Number(CYContext &context);
+ virtual CYString *String(CYContext &context);
+
+ CYString *Concat(CYContext &out, CYString *rhs) const;
virtual void Output(CYOutput &out, CYFlags flags) const;
virtual void PropertyName(CYOutput &out) const;
};
return value_;
}
+ virtual CYNumber *Number(CYContext &context);
+ virtual CYString *String(CYContext &context);
+
virtual void Output(CYOutput &out, CYFlags flags) const;
virtual void PropertyName(CYOutput &out) const;
};
{
}
+ virtual CYNumber *Number(CYContext &context);
+ virtual CYString *String(CYContext &context);
+
virtual void Output(CYOutput &out, CYFlags flags) const;
};
virtual bool Value() const {
return false;
}
+
+ virtual CYNumber *Number(CYContext &context);
+ virtual CYString *String(CYContext &context);
};
struct CYTrue :
virtual bool Value() const {
return true;
}
+
+ virtual CYNumber *Number(CYContext &context);
+ virtual CYString *String(CYContext &context);
};
struct CYVariable :
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 void ForIn(CYOutput &out, CYFlags flags) const;
virtual const char *ForEachIn() const;
- virtual void ForEachIn(CYOutput &out) 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_;
- CYDeclarations(CYDeclaration *declaration, CYDeclarations *next) :
+ CYDeclarations(CYDeclaration *declaration, CYDeclarations *next = NULL) :
CYNext<CYDeclarations>(next),
declaration_(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 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
{
virtual CYExpression *Replace(CYContext &context);
};
-#define CYPostfix_(op, name) \
+#define CYReplace \
+ virtual CYExpression *Replace(CYContext &context);
+
+#define CYPostfix_(op, name, args...) \
struct CY ## name : \
CYPostfix \
- { \
+ { args \
CY ## name(CYExpression *lhs) : \
CYPostfix(lhs) \
{ \
} \
};
-#define CYPrefix_(alphabetic, op, name) \
+#define CYPrefix_(alphabetic, op, name, args...) \
struct CY ## name : \
CYPrefix \
- { \
+ { args \
CY ## name(CYExpression *rhs) : \
CYPrefix(rhs) \
{ \
} \
};
-#define CYInfix_(alphabetic, precedence, op, name) \
+#define CYInfix_(alphabetic, precedence, op, name, args...) \
struct CY ## name : \
CYInfix \
- { \
+ { args \
CY ## name(CYExpression *lhs, CYExpression *rhs) : \
CYInfix(lhs, rhs) \
{ \
} \
};
-#define CYAssignment_(op, name) \
+#define CYAssignment_(op, name, args...) \
struct CY ## name ## Assign : \
CYAssignment \
- { \
+ { args \
CY ## name ## Assign(CYExpression *lhs, CYExpression *rhs) : \
CYAssignment(lhs, rhs) \
{ \
CYInfix_(false, 5, "*", Multiply)
CYInfix_(false, 5, "/", Divide)
CYInfix_(false, 5, "%", Modulus)
-CYInfix_(false, 6, "+", Add)
+CYInfix_(false, 6, "+", Add, CYReplace)
CYInfix_(false, 6, "-", Subtract)
CYInfix_(false, 7, "<<", ShiftLeft)
CYInfix_(false, 7, ">>", ShiftRightSigned)
projects / cycript.git / blobdiff