// Copyright Joyent, Inc. and other Node contributors. // // Permission is hereby granted, free of charge, to any person obtaining a // copy of this software and associated documentation files (the // "Software"), to deal in the Software without restriction, including // without limitation the rights to use, copy, modify, merge, publish, // distribute, sublicense, and/or sell copies of the Software, and to permit // persons to whom the Software is furnished to do so, subject to the // following conditions: // // The above copyright notice and this permission notice shall be included // in all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN // NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR // OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE // USE OR OTHER DEALINGS IN THE SOFTWARE. #include #include #include #include #include // malloc, free #include // memcpy #ifdef __MINGW32__ # include "platform.h" #endif #ifdef __POSIX__ # include // htons, htonl #endif #define MIN(a,b) ((a) < (b) ? (a) : (b)) namespace node { using namespace v8; #define SLICE_ARGS(start_arg, end_arg) \ if (!start_arg->IsInt32() || !end_arg->IsInt32()) { \ return ThrowException(Exception::TypeError( \ String::New("Bad argument."))); \ } \ int32_t start = start_arg->Int32Value(); \ int32_t end = end_arg->Int32Value(); \ if (start < 0 || end < 0) { \ return ThrowException(Exception::TypeError( \ String::New("Bad argument."))); \ } \ if (!(start <= end)) { \ return ThrowException(Exception::Error( \ String::New("Must have start <= end"))); \ } \ if ((size_t)end > parent->length_) { \ return ThrowException(Exception::Error( \ String::New("end cannot be longer than parent.length"))); \ } static Persistent length_symbol; static Persistent chars_written_sym; static Persistent write_sym; Persistent Buffer::constructor_template; static inline size_t base64_decoded_size(const char *src, size_t size) { const char *const end = src + size; const int remainder = size % 4; size = (size / 4) * 3; if (remainder) { if (size == 0 && remainder == 1) { // special case: 1-byte input cannot be decoded size = 0; } else { // non-padded input, add 1 or 2 extra bytes size += 1 + (remainder == 3); } } // check for trailing padding (1 or 2 bytes) if (size > 0) { if (end[-1] == '=') size--; if (end[-2] == '=') size--; } return size; } static size_t ByteLength (Handle string, enum encoding enc) { HandleScope scope; if (enc == UTF8) { return string->Utf8Length(); } else if (enc == BASE64) { String::Utf8Value v(string); return base64_decoded_size(*v, v.length()); } else if (enc == UCS2) { return string->Length() * 2; } else if (enc == HEX) { return string->Length() / 2; } else { return string->Length(); } } Handle Buffer::New(Handle string) { HandleScope scope; // get Buffer from global scope. Local global = v8::Context::GetCurrent()->Global(); Local bv = global->Get(String::NewSymbol("Buffer")); assert(bv->IsFunction()); Local b = Local::Cast(bv); Local argv[1] = { Local::New(string) }; Local instance = b->NewInstance(1, argv); return scope.Close(instance); } Buffer* Buffer::New(size_t length) { HandleScope scope; Local arg = Integer::NewFromUnsigned(length); Local b = constructor_template->GetFunction()->NewInstance(1, &arg); if (b.IsEmpty()) return NULL; return ObjectWrap::Unwrap(b); } Buffer* Buffer::New(char* data, size_t length) { HandleScope scope; Local arg = Integer::NewFromUnsigned(0); Local obj = constructor_template->GetFunction()->NewInstance(1, &arg); Buffer *buffer = ObjectWrap::Unwrap(obj); buffer->Replace(data, length, NULL, NULL); return buffer; } Buffer* Buffer::New(char *data, size_t length, free_callback callback, void *hint) { HandleScope scope; Local arg = Integer::NewFromUnsigned(0); Local obj = constructor_template->GetFunction()->NewInstance(1, &arg); Buffer *buffer = ObjectWrap::Unwrap(obj); buffer->Replace(data, length, callback, hint); return buffer; } Handle Buffer::New(const Arguments &args) { if (!args.IsConstructCall()) { return FromConstructorTemplate(constructor_template, args); } HandleScope scope; Buffer *buffer; if (args[0]->IsInt32()) { // var buffer = new Buffer(1024); size_t length = args[0]->Uint32Value(); buffer = new Buffer(args.This(), length); } else { return ThrowException(Exception::TypeError(String::New("Bad argument"))); } return args.This(); } Buffer::Buffer(Handle wrapper, size_t length) : ObjectWrap() { Wrap(wrapper); length_ = 0; callback_ = NULL; Replace(NULL, length, NULL, NULL); } Buffer::~Buffer() { Replace(NULL, 0, NULL, NULL); } void Buffer::Replace(char *data, size_t length, free_callback callback, void *hint) { HandleScope scope; if (callback_) { callback_(data_, callback_hint_); } else if (length_) { delete [] data_; V8::AdjustAmountOfExternalAllocatedMemory(-(sizeof(Buffer) + length_)); } length_ = length; callback_ = callback; callback_hint_ = hint; if (callback_) { data_ = data; } else if (length_) { data_ = new char[length_]; if (data) memcpy(data_, data, length_); V8::AdjustAmountOfExternalAllocatedMemory(sizeof(Buffer) + length_); } else { data_ = NULL; } handle_->SetIndexedPropertiesToExternalArrayData(data_, kExternalUnsignedByteArray, length_); handle_->Set(length_symbol, Integer::NewFromUnsigned(length_)); } Handle Buffer::BinarySlice(const Arguments &args) { HandleScope scope; Buffer *parent = ObjectWrap::Unwrap(args.This()); SLICE_ARGS(args[0], args[1]) char *data = parent->data_ + start; //Local string = String::New(data, end - start); Local b = Encode(data, end - start, BINARY); return scope.Close(b); } Handle Buffer::AsciiSlice(const Arguments &args) { HandleScope scope; Buffer *parent = ObjectWrap::Unwrap(args.This()); SLICE_ARGS(args[0], args[1]) char* data = parent->data_ + start; Local string = String::New(data, end - start); return scope.Close(string); } Handle Buffer::Utf8Slice(const Arguments &args) { HandleScope scope; Buffer *parent = ObjectWrap::Unwrap(args.This()); SLICE_ARGS(args[0], args[1]) char *data = parent->data_ + start; Local string = String::New(data, end - start); return scope.Close(string); } Handle Buffer::Ucs2Slice(const Arguments &args) { HandleScope scope; Buffer *parent = ObjectWrap::Unwrap(args.This()); SLICE_ARGS(args[0], args[1]) uint16_t *data = (uint16_t*)(parent->data_ + start); Local string = String::New(data, (end - start) / 2); return scope.Close(string); } static const char *base64_table = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" "abcdefghijklmnopqrstuvwxyz" "0123456789+/"; static const int unbase64_table[] = {-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-2,-1,-1,-2,-1,-1 ,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1 ,-2,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,62,-1,-1,-1,63 ,52,53,54,55,56,57,58,59,60,61,-1,-1,-1,-1,-1,-1 ,-1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14 ,15,16,17,18,19,20,21,22,23,24,25,-1,-1,-1,-1,-1 ,-1,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40 ,41,42,43,44,45,46,47,48,49,50,51,-1,-1,-1,-1,-1 ,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1 ,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1 ,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1 ,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1 ,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1 ,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1 ,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1 ,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1 }; #define unbase64(x) unbase64_table[(uint8_t)(x)] Handle Buffer::Base64Slice(const Arguments &args) { HandleScope scope; Buffer *parent = ObjectWrap::Unwrap(args.This()); SLICE_ARGS(args[0], args[1]) int n = end - start; int out_len = (n + 2 - ((n + 2) % 3)) / 3 * 4; char *out = new char[out_len]; uint8_t bitbuf[3]; int i = start; // data() index int j = 0; // out index char c; bool b1_oob, b2_oob; while (i < end) { bitbuf[0] = parent->data_[i++]; if (i < end) { bitbuf[1] = parent->data_[i]; b1_oob = false; } else { bitbuf[1] = 0; b1_oob = true; } i++; if (i < end) { bitbuf[2] = parent->data_[i]; b2_oob = false; } else { bitbuf[2] = 0; b2_oob = true; } i++; c = bitbuf[0] >> 2; assert(c < 64); out[j++] = base64_table[(int)c]; assert(j < out_len); c = ((bitbuf[0] & 0x03) << 4) | (bitbuf[1] >> 4); assert(c < 64); out[j++] = base64_table[(int)c]; assert(j < out_len); if (b1_oob) { out[j++] = '='; } else { c = ((bitbuf[1] & 0x0F) << 2) | (bitbuf[2] >> 6); assert(c < 64); out[j++] = base64_table[(int)c]; } assert(j < out_len); if (b2_oob) { out[j++] = '='; } else { c = bitbuf[2] & 0x3F; assert(c < 64); out[j++] = base64_table[(int)c]; } assert(j <= out_len); } Local string = String::New(out, out_len); delete [] out; return scope.Close(string); } // buffer.fill(value, start, end); Handle Buffer::Fill(const Arguments &args) { HandleScope scope; if (!args[0]->IsInt32()) { return ThrowException(Exception::Error(String::New( "value is not a number"))); } int value = (char)args[0]->Int32Value(); Buffer *parent = ObjectWrap::Unwrap(args.This()); SLICE_ARGS(args[1], args[2]) memset( (void*)(parent->data_ + start), value, end - start); return Undefined(); } // var bytesCopied = buffer.copy(target, targetStart, sourceStart, sourceEnd); Handle Buffer::Copy(const Arguments &args) { HandleScope scope; Buffer *source = ObjectWrap::Unwrap(args.This()); if (!Buffer::HasInstance(args[0])) { return ThrowException(Exception::TypeError(String::New( "First arg should be a Buffer"))); } Local target = args[0]->ToObject(); char *target_data = Buffer::Data(target); ssize_t target_length = Buffer::Length(target); ssize_t target_start = args[1]->Int32Value(); ssize_t source_start = args[2]->Int32Value(); ssize_t source_end = args[3]->IsInt32() ? args[3]->Int32Value() : source->length_; if (source_end < source_start) { return ThrowException(Exception::Error(String::New( "sourceEnd < sourceStart"))); } // Copy 0 bytes; we're done if (source_end == source_start) { return scope.Close(Integer::New(0)); } if (target_start < 0 || target_start >= target_length) { return ThrowException(Exception::Error(String::New( "targetStart out of bounds"))); } if (source_start < 0 || source_start >= source->length_) { return ThrowException(Exception::Error(String::New( "sourceStart out of bounds"))); } if (source_end < 0 || source_end > source->length_) { return ThrowException(Exception::Error(String::New( "sourceEnd out of bounds"))); } ssize_t to_copy = MIN(MIN(source_end - source_start, target_length - target_start), source->length_ - source_start); // need to use slightly slower memmove is the ranges might overlap memmove((void *)(target_data + target_start), (const void*)(source->data_ + source_start), to_copy); return scope.Close(Integer::New(to_copy)); } // var charsWritten = buffer.utf8Write(string, offset, [maxLength]); Handle Buffer::Utf8Write(const Arguments &args) { HandleScope scope; Buffer *buffer = ObjectWrap::Unwrap(args.This()); if (!args[0]->IsString()) { return ThrowException(Exception::TypeError(String::New( "Argument must be a string"))); } Local s = args[0]->ToString(); size_t offset = args[1]->Uint32Value(); int length = s->Length(); if (length == 0) { constructor_template->GetFunction()->Set(chars_written_sym, Integer::New(0)); return scope.Close(Integer::New(0)); } if (length > 0 && offset >= buffer->length_) { return ThrowException(Exception::TypeError(String::New( "Offset is out of bounds"))); } size_t max_length = args[2]->IsUndefined() ? buffer->length_ - offset : args[2]->Uint32Value(); max_length = MIN(buffer->length_ - offset, max_length); char* p = buffer->data_ + offset; int char_written; int written = s->WriteUtf8(p, max_length, &char_written, (String::HINT_MANY_WRITES_EXPECTED | String::NO_NULL_TERMINATION)); constructor_template->GetFunction()->Set(chars_written_sym, Integer::New(char_written)); return scope.Close(Integer::New(written)); } // var charsWritten = buffer.ucs2Write(string, offset, [maxLength]); Handle Buffer::Ucs2Write(const Arguments &args) { HandleScope scope; Buffer *buffer = ObjectWrap::Unwrap(args.This()); if (!args[0]->IsString()) { return ThrowException(Exception::TypeError(String::New( "Argument must be a string"))); } Local s = args[0]->ToString(); size_t offset = args[1]->Uint32Value(); if (s->Length() > 0 && offset >= buffer->length_) { return ThrowException(Exception::TypeError(String::New( "Offset is out of bounds"))); } size_t max_length = args[2]->IsUndefined() ? buffer->length_ - offset : args[2]->Uint32Value(); max_length = MIN(buffer->length_ - offset, max_length) / 2; uint16_t* p = (uint16_t*)(buffer->data_ + offset); int written = s->Write(p, 0, max_length, (String::HINT_MANY_WRITES_EXPECTED | String::NO_NULL_TERMINATION)); constructor_template->GetFunction()->Set(chars_written_sym, Integer::New(written)); return scope.Close(Integer::New(written * 2)); } // var charsWritten = buffer.asciiWrite(string, offset); Handle Buffer::AsciiWrite(const Arguments &args) { HandleScope scope; Buffer *buffer = ObjectWrap::Unwrap(args.This()); if (!args[0]->IsString()) { return ThrowException(Exception::TypeError(String::New( "Argument must be a string"))); } Local s = args[0]->ToString(); size_t offset = args[1]->Int32Value(); if (s->Length() > 0 && offset >= buffer->length_) { return ThrowException(Exception::TypeError(String::New( "Offset is out of bounds"))); } size_t max_length = args[2]->IsUndefined() ? buffer->length_ - offset : args[2]->Uint32Value(); max_length = MIN(s->Length(), MIN(buffer->length_ - offset, max_length)); char *p = buffer->data_ + offset; int written = s->WriteAscii(p, 0, max_length, (String::HINT_MANY_WRITES_EXPECTED | String::NO_NULL_TERMINATION)); constructor_template->GetFunction()->Set(chars_written_sym, Integer::New(written)); return scope.Close(Integer::New(written)); } // var bytesWritten = buffer.base64Write(string, offset, [maxLength]); Handle Buffer::Base64Write(const Arguments &args) { HandleScope scope; assert(unbase64('/') == 63); assert(unbase64('+') == 62); assert(unbase64('T') == 19); assert(unbase64('Z') == 25); assert(unbase64('t') == 45); assert(unbase64('z') == 51); assert(unbase64(' ') == -2); assert(unbase64('\n') == -2); assert(unbase64('\r') == -2); Buffer *buffer = ObjectWrap::Unwrap(args.This()); if (!args[0]->IsString()) { return ThrowException(Exception::TypeError(String::New( "Argument must be a string"))); } String::AsciiValue s(args[0]->ToString()); size_t offset = args[1]->Int32Value(); // handle zero-length buffers graciously if (offset == 0 && buffer->length_ == 0) { return scope.Close(Integer::New(0)); } if (offset >= buffer->length_) { return ThrowException(Exception::TypeError(String::New( "Offset is out of bounds"))); } const size_t size = base64_decoded_size(*s, s.length()); if (size > buffer->length_ - offset) { // throw exception, don't silently truncate return ThrowException(Exception::TypeError(String::New( "Buffer too small"))); } char a, b, c, d; char* start = buffer->data_ + offset; char* dst = start; const char *src = *s; const char *const srcEnd = src + s.length(); while (src < srcEnd) { int remaining = srcEnd - src; while (unbase64(*src) < 0 && src < srcEnd) { src++; remaining--; } if (remaining == 0 || *src == '=') break; a = unbase64(*src++); while (unbase64(*src) < 0 && src < srcEnd) { src++; remaining--; } if (remaining <= 1 || *src == '=') break; b = unbase64(*src++); *dst++ = (a << 2) | ((b & 0x30) >> 4); while (unbase64(*src) < 0 && src < srcEnd) { src++; remaining--; } if (remaining <= 2 || *src == '=') break; c = unbase64(*src++); *dst++ = ((b & 0x0F) << 4) | ((c & 0x3C) >> 2); while (unbase64(*src) < 0 && src < srcEnd) { src++; remaining--; } if (remaining <= 3 || *src == '=') break; d = unbase64(*src++); *dst++ = ((c & 0x03) << 6) | (d & 0x3F); } constructor_template->GetFunction()->Set(chars_written_sym, Integer::New(s.length())); return scope.Close(Integer::New(dst - start)); } Handle Buffer::BinaryWrite(const Arguments &args) { HandleScope scope; Buffer *buffer = ObjectWrap::Unwrap(args.This()); if (!args[0]->IsString()) { return ThrowException(Exception::TypeError(String::New( "Argument must be a string"))); } Local s = args[0]->ToString(); size_t offset = args[1]->Int32Value(); if (s->Length() > 0 && offset >= buffer->length_) { return ThrowException(Exception::TypeError(String::New( "Offset is out of bounds"))); } char *p = (char*)buffer->data_ + offset; size_t max_length = args[2]->IsUndefined() ? buffer->length_ - offset : args[2]->Uint32Value(); max_length = MIN(s->Length(), MIN(buffer->length_ - offset, max_length)); int written = DecodeWrite(p, max_length, s, BINARY); constructor_template->GetFunction()->Set(chars_written_sym, Integer::New(written)); return scope.Close(Integer::New(written)); } // var nbytes = Buffer.byteLength("string", "utf8") Handle Buffer::ByteLength(const Arguments &args) { HandleScope scope; if (!args[0]->IsString()) { return ThrowException(Exception::TypeError(String::New( "Argument must be a string"))); } Local s = args[0]->ToString(); enum encoding e = ParseEncoding(args[1], UTF8); return scope.Close(Integer::New(node::ByteLength(s, e))); } Handle Buffer::MakeFastBuffer(const Arguments &args) { HandleScope scope; if (!Buffer::HasInstance(args[0])) { return ThrowException(Exception::TypeError(String::New( "First argument must be a Buffer"))); } Buffer *buffer = ObjectWrap::Unwrap(args[0]->ToObject()); Local fast_buffer = args[1]->ToObject();; uint32_t offset = args[2]->Uint32Value(); uint32_t length = args[3]->Uint32Value(); fast_buffer->SetIndexedPropertiesToExternalArrayData(buffer->data_ + offset, kExternalUnsignedByteArray, length); return Undefined(); } bool Buffer::HasInstance(v8::Handle val) { if (!val->IsObject()) return false; v8::Local obj = val->ToObject(); if (obj->GetIndexedPropertiesExternalArrayDataType() == kExternalUnsignedByteArray) return true; // Also check for SlowBuffers that are empty. if (constructor_template->HasInstance(obj)) return true; return false; } void Buffer::Initialize(Handle target) { HandleScope scope; length_symbol = Persistent::New(String::NewSymbol("length")); chars_written_sym = Persistent::New(String::NewSymbol("_charsWritten")); Local t = FunctionTemplate::New(Buffer::New); constructor_template = Persistent::New(t); constructor_template->InstanceTemplate()->SetInternalFieldCount(1); constructor_template->SetClassName(String::NewSymbol("SlowBuffer")); // copy free NODE_SET_PROTOTYPE_METHOD(constructor_template, "binarySlice", Buffer::BinarySlice); NODE_SET_PROTOTYPE_METHOD(constructor_template, "asciiSlice", Buffer::AsciiSlice); NODE_SET_PROTOTYPE_METHOD(constructor_template, "base64Slice", Buffer::Base64Slice); NODE_SET_PROTOTYPE_METHOD(constructor_template, "ucs2Slice", Buffer::Ucs2Slice); // TODO NODE_SET_PROTOTYPE_METHOD(t, "utf16Slice", Utf16Slice); // copy NODE_SET_PROTOTYPE_METHOD(constructor_template, "utf8Slice", Buffer::Utf8Slice); NODE_SET_PROTOTYPE_METHOD(constructor_template, "utf8Write", Buffer::Utf8Write); NODE_SET_PROTOTYPE_METHOD(constructor_template, "asciiWrite", Buffer::AsciiWrite); NODE_SET_PROTOTYPE_METHOD(constructor_template, "binaryWrite", Buffer::BinaryWrite); NODE_SET_PROTOTYPE_METHOD(constructor_template, "base64Write", Buffer::Base64Write); NODE_SET_PROTOTYPE_METHOD(constructor_template, "ucs2Write", Buffer::Ucs2Write); NODE_SET_PROTOTYPE_METHOD(constructor_template, "fill", Buffer::Fill); NODE_SET_PROTOTYPE_METHOD(constructor_template, "copy", Buffer::Copy); NODE_SET_METHOD(constructor_template->GetFunction(), "byteLength", Buffer::ByteLength); NODE_SET_METHOD(constructor_template->GetFunction(), "makeFastBuffer", Buffer::MakeFastBuffer); target->Set(String::NewSymbol("SlowBuffer"), constructor_template->GetFunction()); } } // namespace node NODE_MODULE(node_buffer, node::Buffer::Initialize)