You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
			
				
					306 lines
				
				7.9 KiB
			
		
		
			
		
	
	
					306 lines
				
				7.9 KiB
			| 
								 
											4 years ago
										 
									 | 
							
								// 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.
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#ifndef NAN_STRING_BYTES_H_
							 | 
						||
| 
								 | 
							
								#define NAN_STRING_BYTES_H_
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								// Decodes a v8::Local<v8::String> or Buffer to a raw char*
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								namespace imp {
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								using v8::Local;
							 | 
						||
| 
								 | 
							
								using v8::Object;
							 | 
						||
| 
								 | 
							
								using v8::String;
							 | 
						||
| 
								 | 
							
								using v8::Value;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								//// Base 64 ////
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define base64_encoded_size(size) ((size + 2 - ((size + 2) % 3)) / 3 * 4)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								//// HEX ////
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								static bool contains_non_ascii_slow(const char* buf, size_t len) {
							 | 
						||
| 
								 | 
							
								  for (size_t i = 0; i < len; ++i) {
							 | 
						||
| 
								 | 
							
								    if (buf[i] & 0x80) return true;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								  return false;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								static bool contains_non_ascii(const char* src, size_t len) {
							 | 
						||
| 
								 | 
							
								  if (len < 16) {
							 | 
						||
| 
								 | 
							
								    return contains_non_ascii_slow(src, len);
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const unsigned bytes_per_word = sizeof(void*);
							 | 
						||
| 
								 | 
							
								  const unsigned align_mask = bytes_per_word - 1;
							 | 
						||
| 
								 | 
							
								  const unsigned unaligned = reinterpret_cast<uintptr_t>(src) & align_mask;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (unaligned > 0) {
							 | 
						||
| 
								 | 
							
								    const unsigned n = bytes_per_word - unaligned;
							 | 
						||
| 
								 | 
							
								    if (contains_non_ascii_slow(src, n)) return true;
							 | 
						||
| 
								 | 
							
								    src += n;
							 | 
						||
| 
								 | 
							
								    len -= n;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#if defined(__x86_64__) || defined(_WIN64)
							 | 
						||
| 
								 | 
							
								  const uintptr_t mask = 0x8080808080808080ll;
							 | 
						||
| 
								 | 
							
								#else
							 | 
						||
| 
								 | 
							
								  const uintptr_t mask = 0x80808080l;
							 | 
						||
| 
								 | 
							
								#endif
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const uintptr_t* srcw = reinterpret_cast<const uintptr_t*>(src);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  for (size_t i = 0, n = len / bytes_per_word; i < n; ++i) {
							 | 
						||
| 
								 | 
							
								    if (srcw[i] & mask) return true;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const unsigned remainder = len & align_mask;
							 | 
						||
| 
								 | 
							
								  if (remainder > 0) {
							 | 
						||
| 
								 | 
							
								    const size_t offset = len - remainder;
							 | 
						||
| 
								 | 
							
								    if (contains_non_ascii_slow(src + offset, remainder)) return true;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return false;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								static void force_ascii_slow(const char* src, char* dst, size_t len) {
							 | 
						||
| 
								 | 
							
								  for (size_t i = 0; i < len; ++i) {
							 | 
						||
| 
								 | 
							
								    dst[i] = src[i] & 0x7f;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								static void force_ascii(const char* src, char* dst, size_t len) {
							 | 
						||
| 
								 | 
							
								  if (len < 16) {
							 | 
						||
| 
								 | 
							
								    force_ascii_slow(src, dst, len);
							 | 
						||
| 
								 | 
							
								    return;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const unsigned bytes_per_word = sizeof(void*);
							 | 
						||
| 
								 | 
							
								  const unsigned align_mask = bytes_per_word - 1;
							 | 
						||
| 
								 | 
							
								  const unsigned src_unalign = reinterpret_cast<uintptr_t>(src) & align_mask;
							 | 
						||
| 
								 | 
							
								  const unsigned dst_unalign = reinterpret_cast<uintptr_t>(dst) & align_mask;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (src_unalign > 0) {
							 | 
						||
| 
								 | 
							
								    if (src_unalign == dst_unalign) {
							 | 
						||
| 
								 | 
							
								      const unsigned unalign = bytes_per_word - src_unalign;
							 | 
						||
| 
								 | 
							
								      force_ascii_slow(src, dst, unalign);
							 | 
						||
| 
								 | 
							
								      src += unalign;
							 | 
						||
| 
								 | 
							
								      dst += unalign;
							 | 
						||
| 
								 | 
							
								      len -= src_unalign;
							 | 
						||
| 
								 | 
							
								    } else {
							 | 
						||
| 
								 | 
							
								      force_ascii_slow(src, dst, len);
							 | 
						||
| 
								 | 
							
								      return;
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#if defined(__x86_64__) || defined(_WIN64)
							 | 
						||
| 
								 | 
							
								  const uintptr_t mask = ~0x8080808080808080ll;
							 | 
						||
| 
								 | 
							
								#else
							 | 
						||
| 
								 | 
							
								  const uintptr_t mask = ~0x80808080l;
							 | 
						||
| 
								 | 
							
								#endif
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const uintptr_t* srcw = reinterpret_cast<const uintptr_t*>(src);
							 | 
						||
| 
								 | 
							
								  uintptr_t* dstw = reinterpret_cast<uintptr_t*>(dst);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  for (size_t i = 0, n = len / bytes_per_word; i < n; ++i) {
							 | 
						||
| 
								 | 
							
								    dstw[i] = srcw[i] & mask;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const unsigned remainder = len & align_mask;
							 | 
						||
| 
								 | 
							
								  if (remainder > 0) {
							 | 
						||
| 
								 | 
							
								    const size_t offset = len - remainder;
							 | 
						||
| 
								 | 
							
								    force_ascii_slow(src + offset, dst + offset, remainder);
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								static size_t base64_encode(const char* src,
							 | 
						||
| 
								 | 
							
								                            size_t slen,
							 | 
						||
| 
								 | 
							
								                            char* dst,
							 | 
						||
| 
								 | 
							
								                            size_t dlen) {
							 | 
						||
| 
								 | 
							
								  // We know how much we'll write, just make sure that there's space.
							 | 
						||
| 
								 | 
							
								  assert(dlen >= base64_encoded_size(slen) &&
							 | 
						||
| 
								 | 
							
								      "not enough space provided for base64 encode");
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  dlen = base64_encoded_size(slen);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  unsigned a;
							 | 
						||
| 
								 | 
							
								  unsigned b;
							 | 
						||
| 
								 | 
							
								  unsigned c;
							 | 
						||
| 
								 | 
							
								  unsigned i;
							 | 
						||
| 
								 | 
							
								  unsigned k;
							 | 
						||
| 
								 | 
							
								  unsigned n;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  static const char table[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
							 | 
						||
| 
								 | 
							
								                              "abcdefghijklmnopqrstuvwxyz"
							 | 
						||
| 
								 | 
							
								                              "0123456789+/";
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  i = 0;
							 | 
						||
| 
								 | 
							
								  k = 0;
							 | 
						||
| 
								 | 
							
								  n = slen / 3 * 3;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  while (i < n) {
							 | 
						||
| 
								 | 
							
								    a = src[i + 0] & 0xff;
							 | 
						||
| 
								 | 
							
								    b = src[i + 1] & 0xff;
							 | 
						||
| 
								 | 
							
								    c = src[i + 2] & 0xff;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    dst[k + 0] = table[a >> 2];
							 | 
						||
| 
								 | 
							
								    dst[k + 1] = table[((a & 3) << 4) | (b >> 4)];
							 | 
						||
| 
								 | 
							
								    dst[k + 2] = table[((b & 0x0f) << 2) | (c >> 6)];
							 | 
						||
| 
								 | 
							
								    dst[k + 3] = table[c & 0x3f];
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    i += 3;
							 | 
						||
| 
								 | 
							
								    k += 4;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (n != slen) {
							 | 
						||
| 
								 | 
							
								    switch (slen - n) {
							 | 
						||
| 
								 | 
							
								      case 1:
							 | 
						||
| 
								 | 
							
								        a = src[i + 0] & 0xff;
							 | 
						||
| 
								 | 
							
								        dst[k + 0] = table[a >> 2];
							 | 
						||
| 
								 | 
							
								        dst[k + 1] = table[(a & 3) << 4];
							 | 
						||
| 
								 | 
							
								        dst[k + 2] = '=';
							 | 
						||
| 
								 | 
							
								        dst[k + 3] = '=';
							 | 
						||
| 
								 | 
							
								        break;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      case 2:
							 | 
						||
| 
								 | 
							
								        a = src[i + 0] & 0xff;
							 | 
						||
| 
								 | 
							
								        b = src[i + 1] & 0xff;
							 | 
						||
| 
								 | 
							
								        dst[k + 0] = table[a >> 2];
							 | 
						||
| 
								 | 
							
								        dst[k + 1] = table[((a & 3) << 4) | (b >> 4)];
							 | 
						||
| 
								 | 
							
								        dst[k + 2] = table[(b & 0x0f) << 2];
							 | 
						||
| 
								 | 
							
								        dst[k + 3] = '=';
							 | 
						||
| 
								 | 
							
								        break;
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return dlen;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								static size_t hex_encode(const char* src, size_t slen, char* dst, size_t dlen) {
							 | 
						||
| 
								 | 
							
								  // We know how much we'll write, just make sure that there's space.
							 | 
						||
| 
								 | 
							
								  assert(dlen >= slen * 2 &&
							 | 
						||
| 
								 | 
							
								      "not enough space provided for hex encode");
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  dlen = slen * 2;
							 | 
						||
| 
								 | 
							
								  for (uint32_t i = 0, k = 0; k < dlen; i += 1, k += 2) {
							 | 
						||
| 
								 | 
							
								    static const char hex[] = "0123456789abcdef";
							 | 
						||
| 
								 | 
							
								    uint8_t val = static_cast<uint8_t>(src[i]);
							 | 
						||
| 
								 | 
							
								    dst[k + 0] = hex[val >> 4];
							 | 
						||
| 
								 | 
							
								    dst[k + 1] = hex[val & 15];
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return dlen;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								static Local<Value> Encode(const char* buf,
							 | 
						||
| 
								 | 
							
								                           size_t buflen,
							 | 
						||
| 
								 | 
							
								                           enum Encoding encoding) {
							 | 
						||
| 
								 | 
							
								  assert(buflen <= node::Buffer::kMaxLength);
							 | 
						||
| 
								 | 
							
								  if (!buflen && encoding != BUFFER)
							 | 
						||
| 
								 | 
							
								    return New("").ToLocalChecked();
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  Local<String> val;
							 | 
						||
| 
								 | 
							
								  switch (encoding) {
							 | 
						||
| 
								 | 
							
								    case BUFFER:
							 | 
						||
| 
								 | 
							
								      return CopyBuffer(buf, buflen).ToLocalChecked();
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    case ASCII:
							 | 
						||
| 
								 | 
							
								      if (contains_non_ascii(buf, buflen)) {
							 | 
						||
| 
								 | 
							
								        char* out = new char[buflen];
							 | 
						||
| 
								 | 
							
								        force_ascii(buf, out, buflen);
							 | 
						||
| 
								 | 
							
								        val = New<String>(out, buflen).ToLocalChecked();
							 | 
						||
| 
								 | 
							
								        delete[] out;
							 | 
						||
| 
								 | 
							
								      } else {
							 | 
						||
| 
								 | 
							
								        val = New<String>(buf, buflen).ToLocalChecked();
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								      break;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    case UTF8:
							 | 
						||
| 
								 | 
							
								      val = New<String>(buf, buflen).ToLocalChecked();
							 | 
						||
| 
								 | 
							
								      break;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    case BINARY: {
							 | 
						||
| 
								 | 
							
								      // TODO(isaacs) use ExternalTwoByteString?
							 | 
						||
| 
								 | 
							
								      const unsigned char *cbuf = reinterpret_cast<const unsigned char*>(buf);
							 | 
						||
| 
								 | 
							
								      uint16_t * twobytebuf = new uint16_t[buflen];
							 | 
						||
| 
								 | 
							
								      for (size_t i = 0; i < buflen; i++) {
							 | 
						||
| 
								 | 
							
								        // XXX is the following line platform independent?
							 | 
						||
| 
								 | 
							
								        twobytebuf[i] = cbuf[i];
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								      val = New<String>(twobytebuf, buflen).ToLocalChecked();
							 | 
						||
| 
								 | 
							
								      delete[] twobytebuf;
							 | 
						||
| 
								 | 
							
								      break;
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    case BASE64: {
							 | 
						||
| 
								 | 
							
								      size_t dlen = base64_encoded_size(buflen);
							 | 
						||
| 
								 | 
							
								      char* dst = new char[dlen];
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      size_t written = base64_encode(buf, buflen, dst, dlen);
							 | 
						||
| 
								 | 
							
								      assert(written == dlen);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      val = New<String>(dst, dlen).ToLocalChecked();
							 | 
						||
| 
								 | 
							
								      delete[] dst;
							 | 
						||
| 
								 | 
							
								      break;
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    case UCS2: {
							 | 
						||
| 
								 | 
							
								      const uint16_t* data = reinterpret_cast<const uint16_t*>(buf);
							 | 
						||
| 
								 | 
							
								      val = New<String>(data, buflen / 2).ToLocalChecked();
							 | 
						||
| 
								 | 
							
								      break;
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    case HEX: {
							 | 
						||
| 
								 | 
							
								      size_t dlen = buflen * 2;
							 | 
						||
| 
								 | 
							
								      char* dst = new char[dlen];
							 | 
						||
| 
								 | 
							
								      size_t written = hex_encode(buf, buflen, dst, dlen);
							 | 
						||
| 
								 | 
							
								      assert(written == dlen);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      val = New<String>(dst, dlen).ToLocalChecked();
							 | 
						||
| 
								 | 
							
								      delete[] dst;
							 | 
						||
| 
								 | 
							
								      break;
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    default:
							 | 
						||
| 
								 | 
							
								      assert(0 && "unknown encoding");
							 | 
						||
| 
								 | 
							
								      break;
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return val;
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#undef base64_encoded_size
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								}  // end of namespace imp
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#endif  // NAN_STRING_BYTES_H_
							 |