Browse Source

zlib: don't use C++ templates

Reduces the executable's size by a few kilobytes.
v0.9.1-release
ssuda 13 years ago
committed by Ben Noordhuis
parent
commit
44eb279ef8
  1. 18
      lib/zlib.js
  2. 88
      src/node_zlib.cc

18
lib/zlib.js

@ -169,12 +169,12 @@ function zlibBuffer(engine, buffer, callback) {
// minimal 2-byte header // minimal 2-byte header
function Deflate(opts) { function Deflate(opts) {
if (!(this instanceof Deflate)) return new Deflate(opts); if (!(this instanceof Deflate)) return new Deflate(opts);
Zlib.call(this, opts, binding.Deflate); Zlib.call(this, opts, binding.DEFLATE);
} }
function Inflate(opts) { function Inflate(opts) {
if (!(this instanceof Inflate)) return new Inflate(opts); if (!(this instanceof Inflate)) return new Inflate(opts);
Zlib.call(this, opts, binding.Inflate); Zlib.call(this, opts, binding.INFLATE);
} }
@ -182,12 +182,12 @@ function Inflate(opts) {
// gzip - bigger header, same deflate compression // gzip - bigger header, same deflate compression
function Gzip(opts) { function Gzip(opts) {
if (!(this instanceof Gzip)) return new Gzip(opts); if (!(this instanceof Gzip)) return new Gzip(opts);
Zlib.call(this, opts, binding.Gzip); Zlib.call(this, opts, binding.GZIP);
} }
function Gunzip(opts) { function Gunzip(opts) {
if (!(this instanceof Gunzip)) return new Gunzip(opts); if (!(this instanceof Gunzip)) return new Gunzip(opts);
Zlib.call(this, opts, binding.Gunzip); Zlib.call(this, opts, binding.GUNZIP);
} }
@ -195,19 +195,19 @@ function Gunzip(opts) {
// raw - no header // raw - no header
function DeflateRaw(opts) { function DeflateRaw(opts) {
if (!(this instanceof DeflateRaw)) return new DeflateRaw(opts); if (!(this instanceof DeflateRaw)) return new DeflateRaw(opts);
Zlib.call(this, opts, binding.DeflateRaw); Zlib.call(this, opts, binding.DEFLATERAW);
} }
function InflateRaw(opts) { function InflateRaw(opts) {
if (!(this instanceof InflateRaw)) return new InflateRaw(opts); if (!(this instanceof InflateRaw)) return new InflateRaw(opts);
Zlib.call(this, opts, binding.InflateRaw); Zlib.call(this, opts, binding.INFLATERAW);
} }
// auto-detect header. // auto-detect header.
function Unzip(opts) { function Unzip(opts) {
if (!(this instanceof Unzip)) return new Unzip(opts); if (!(this instanceof Unzip)) return new Unzip(opts);
Zlib.call(this, opts, binding.Unzip); Zlib.call(this, opts, binding.UNZIP);
} }
@ -216,7 +216,7 @@ function Unzip(opts) {
// true or false if there is anything in the queue when // true or false if there is anything in the queue when
// you call the .write() method. // you call the .write() method.
function Zlib(opts, Binding) { function Zlib(opts, mode) {
this._opts = opts = opts || {}; this._opts = opts = opts || {};
this._queue = []; this._queue = [];
this._processing = false; this._processing = false;
@ -269,7 +269,7 @@ function Zlib(opts, Binding) {
} }
} }
this._binding = new Binding(); this._binding = new binding.Zlib(mode);
this._binding.init(opts.windowBits || exports.Z_DEFAULT_WINDOWBITS, this._binding.init(opts.windowBits || exports.Z_DEFAULT_WINDOWBITS,
opts.level || exports.Z_DEFAULT_COMPRESSION, opts.level || exports.Z_DEFAULT_COMPRESSION,
opts.memLevel || exports.Z_DEFAULT_MEMLEVEL, opts.memLevel || exports.Z_DEFAULT_MEMLEVEL,

88
src/node_zlib.cc

@ -47,8 +47,6 @@ enum node_zlib_mode {
UNZIP UNZIP
}; };
template <node_zlib_mode mode> class ZCtx;
void InitZlib(v8::Handle<v8::Object> target); void InitZlib(v8::Handle<v8::Object> target);
@ -56,17 +54,15 @@ void InitZlib(v8::Handle<v8::Object> target);
/** /**
* Deflate/Inflate * Deflate/Inflate
*/ */
template <node_zlib_mode mode> class ZCtx : public ObjectWrap { class ZCtx : public ObjectWrap {
public: public:
ZCtx() : ObjectWrap() { ZCtx(node_zlib_mode mode) : ObjectWrap(), dictionary_(NULL), mode_(mode) {}
dictionary_ = NULL;
}
~ZCtx() { ~ZCtx() {
if (mode == DEFLATE || mode == GZIP || mode == DEFLATERAW) { if (mode_ == DEFLATE || mode_ == GZIP || mode_ == DEFLATERAW) {
(void)deflateEnd(&strm_); (void)deflateEnd(&strm_);
} else if (mode == INFLATE || mode == GUNZIP || mode == INFLATERAW) { } else if (mode_ == INFLATE || mode_ == GUNZIP || mode_ == INFLATERAW) {
(void)inflateEnd(&strm_); (void)inflateEnd(&strm_);
} }
@ -78,7 +74,7 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
HandleScope scope; HandleScope scope;
assert(args.Length() == 7); assert(args.Length() == 7);
ZCtx<mode> *ctx = ObjectWrap::Unwrap< ZCtx<mode> >(args.This()); ZCtx *ctx = ObjectWrap::Unwrap<ZCtx>(args.This());
assert(ctx->init_done_ && "write before init"); assert(ctx->init_done_ && "write before init");
assert(!ctx->write_in_progress_ && "write already in progress"); assert(!ctx->write_in_progress_ && "write already in progress");
@ -127,8 +123,8 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
uv_queue_work(uv_default_loop(), uv_queue_work(uv_default_loop(),
work_req, work_req,
ZCtx<mode>::Process, ZCtx::Process,
ZCtx<mode>::After); ZCtx::After);
ctx->Ref(); ctx->Ref();
@ -141,13 +137,13 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
// for a single write() call, until all of the input bytes have // for a single write() call, until all of the input bytes have
// been consumed. // been consumed.
static void Process(uv_work_t* work_req) { static void Process(uv_work_t* work_req) {
ZCtx<mode> *ctx = container_of(work_req, ZCtx<mode>, work_req_); ZCtx *ctx = container_of(work_req, ZCtx, work_req_);
// If the avail_out is left at 0, then it means that it ran out // If the avail_out is left at 0, then it means that it ran out
// of room. If there was avail_out left over, then it means // of room. If there was avail_out left over, then it means
// that all of the input was consumed. // that all of the input was consumed.
int err; int err;
switch (mode) { switch (ctx->mode_) {
case DEFLATE: case DEFLATE:
case GZIP: case GZIP:
case DEFLATERAW: case DEFLATERAW:
@ -186,7 +182,7 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
// v8 land! // v8 land!
static void After(uv_work_t* work_req) { static void After(uv_work_t* work_req) {
HandleScope scope; HandleScope scope;
ZCtx<mode> *ctx = container_of(work_req, ZCtx<mode>, work_req_); ZCtx *ctx = container_of(work_req, ZCtx, work_req_);
Local<Integer> avail_out = Integer::New(ctx->strm_.avail_out); Local<Integer> avail_out = Integer::New(ctx->strm_.avail_out);
Local<Integer> avail_in = Integer::New(ctx->strm_.avail_in); Local<Integer> avail_in = Integer::New(ctx->strm_.avail_in);
@ -204,7 +200,16 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
static Handle<Value> New(const Arguments& args) { static Handle<Value> New(const Arguments& args) {
HandleScope scope; HandleScope scope;
ZCtx<mode> *ctx = new ZCtx<mode>(); if (args.Length() < 1 || !args[0]->IsInt32()) {
return ThrowException(Exception::TypeError(String::New("Bad argument")));
}
node_zlib_mode mode = (node_zlib_mode) args[0]->Int32Value();
if (mode < DEFLATE || mode > UNZIP) {
return ThrowException(Exception::TypeError(String::New("Bad argument")));
}
ZCtx *ctx = new ZCtx(mode);
ctx->Wrap(args.This()); ctx->Wrap(args.This());
return args.This(); return args.This();
} }
@ -216,7 +221,7 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
assert((args.Length() == 4 || args.Length() == 5) && assert((args.Length() == 4 || args.Length() == 5) &&
"init(windowBits, level, memLevel, strategy, [dictionary])"); "init(windowBits, level, memLevel, strategy, [dictionary])");
ZCtx<mode> *ctx = ObjectWrap::Unwrap< ZCtx<mode> >(args.This()); ZCtx *ctx = ObjectWrap::Unwrap<ZCtx>(args.This());
int windowBits = args[0]->Uint32Value(); int windowBits = args[0]->Uint32Value();
assert((windowBits >= 8 && windowBits <= 15) && "invalid windowBits"); assert((windowBits >= 8 && windowBits <= 15) && "invalid windowBits");
@ -254,7 +259,7 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
static Handle<Value> Reset(const Arguments &args) { static Handle<Value> Reset(const Arguments &args) {
HandleScope scope; HandleScope scope;
ZCtx<mode> *ctx = ObjectWrap::Unwrap< ZCtx<mode> >(args.This()); ZCtx *ctx = ObjectWrap::Unwrap<ZCtx>(args.This());
Reset(ctx); Reset(ctx);
SetDictionary(ctx); SetDictionary(ctx);
@ -274,20 +279,20 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
ctx->flush_ = Z_NO_FLUSH; ctx->flush_ = Z_NO_FLUSH;
if (mode == GZIP || mode == GUNZIP) { if (ctx->mode_ == GZIP || ctx->mode_ == GUNZIP) {
ctx->windowBits_ += 16; ctx->windowBits_ += 16;
} }
if (mode == UNZIP) { if (ctx->mode_ == UNZIP) {
ctx->windowBits_ += 32; ctx->windowBits_ += 32;
} }
if (mode == DEFLATERAW || mode == INFLATERAW) { if (ctx->mode_ == DEFLATERAW || ctx->mode_ == INFLATERAW) {
ctx->windowBits_ *= -1; ctx->windowBits_ *= -1;
} }
int err; int err;
switch (mode) { switch (ctx->mode_) {
case DEFLATE: case DEFLATE:
case GZIP: case GZIP:
case DEFLATERAW: case DEFLATERAW:
@ -322,7 +327,7 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
int err = Z_OK; int err = Z_OK;
switch (mode) { switch (ctx->mode_) {
case DEFLATE: case DEFLATE:
case DEFLATERAW: case DEFLATERAW:
err = deflateSetDictionary(&ctx->strm_, err = deflateSetDictionary(&ctx->strm_,
@ -339,7 +344,7 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
static void Reset(ZCtx* ctx) { static void Reset(ZCtx* ctx) {
int err = Z_OK; int err = Z_OK;
switch (mode) { switch (ctx->mode_) {
case DEFLATE: case DEFLATE:
case DEFLATERAW: case DEFLATERAW:
err = deflateReset(&ctx->strm_); err = deflateReset(&ctx->strm_);
@ -375,30 +380,23 @@ template <node_zlib_mode mode> class ZCtx : public ObjectWrap {
bool write_in_progress_; bool write_in_progress_;
uv_work_t work_req_; uv_work_t work_req_;
node_zlib_mode mode_;
}; };
#define NODE_ZLIB_CLASS(mode, name) \
{ \
Local<FunctionTemplate> z = FunctionTemplate::New(ZCtx<mode>::New); \
z->InstanceTemplate()->SetInternalFieldCount(1); \
NODE_SET_PROTOTYPE_METHOD(z, "write", ZCtx<mode>::Write); \
NODE_SET_PROTOTYPE_METHOD(z, "init", ZCtx<mode>::Init); \
NODE_SET_PROTOTYPE_METHOD(z, "reset", ZCtx<mode>::Reset); \
z->SetClassName(String::NewSymbol(name)); \
target->Set(String::NewSymbol(name), z->GetFunction()); \
}
void InitZlib(Handle<Object> target) { void InitZlib(Handle<Object> target) {
HandleScope scope; HandleScope scope;
NODE_ZLIB_CLASS(INFLATE, "Inflate") Local<FunctionTemplate> z = FunctionTemplate::New(ZCtx::New);
NODE_ZLIB_CLASS(DEFLATE, "Deflate")
NODE_ZLIB_CLASS(INFLATERAW, "InflateRaw") z->InstanceTemplate()->SetInternalFieldCount(1);
NODE_ZLIB_CLASS(DEFLATERAW, "DeflateRaw")
NODE_ZLIB_CLASS(GZIP, "Gzip") NODE_SET_PROTOTYPE_METHOD(z, "write", ZCtx::Write);
NODE_ZLIB_CLASS(GUNZIP, "Gunzip") NODE_SET_PROTOTYPE_METHOD(z, "init", ZCtx::Init);
NODE_ZLIB_CLASS(UNZIP, "Unzip") NODE_SET_PROTOTYPE_METHOD(z, "reset", ZCtx::Reset);
z->SetClassName(String::NewSymbol("Zlib"));
target->Set(String::NewSymbol("Zlib"), z->GetFunction());
callback_sym = NODE_PSYMBOL("callback"); callback_sym = NODE_PSYMBOL("callback");
@ -428,6 +426,14 @@ void InitZlib(Handle<Object> target) {
NODE_DEFINE_CONSTANT(target, Z_DEFAULT_STRATEGY); NODE_DEFINE_CONSTANT(target, Z_DEFAULT_STRATEGY);
NODE_DEFINE_CONSTANT(target, ZLIB_VERNUM); NODE_DEFINE_CONSTANT(target, ZLIB_VERNUM);
NODE_DEFINE_CONSTANT(target, DEFLATE);
NODE_DEFINE_CONSTANT(target, INFLATE);
NODE_DEFINE_CONSTANT(target, GZIP);
NODE_DEFINE_CONSTANT(target, GUNZIP);
NODE_DEFINE_CONSTANT(target, DEFLATERAW);
NODE_DEFINE_CONSTANT(target, INFLATERAW);
NODE_DEFINE_CONSTANT(target, UNZIP);
target->Set(String::NewSymbol("ZLIB_VERSION"), String::New(ZLIB_VERSION)); target->Set(String::NewSymbol("ZLIB_VERSION"), String::New(ZLIB_VERSION));
} }

Loading…
Cancel
Save