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.
		
		
		
		
		
			
		
			
				
					
					
						
							305 lines
						
					
					
						
							7.9 KiB
						
					
					
				
			
		
		
	
	
							305 lines
						
					
					
						
							7.9 KiB
						
					
					
				// 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_
 | 
						|
 |