123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797 |
- // Protocol Buffers - Google's data interchange format
- // Copyright 2008 Google Inc. All rights reserved.
- // https://developers.google.com/protocol-buffers/
- //
- // Redistribution and use in source and binary forms, with or without
- // modification, are permitted provided that the following conditions are
- // met:
- //
- // * Redistributions of source code must retain the above copyright
- // notice, this list of conditions and the following disclaimer.
- // * Redistributions in binary form must reproduce the above
- // copyright notice, this list of conditions and the following disclaimer
- // in the documentation and/or other materials provided with the
- // distribution.
- // * Neither the name of Google Inc. nor the names of its
- // contributors may be used to endorse or promote products derived from
- // this software without specific prior written permission.
- //
- // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- #import "GPBCodedInputStream_PackagePrivate.h"
- #import "GPBDictionary_PackagePrivate.h"
- #import "GPBMessage_PackagePrivate.h"
- #import "GPBUnknownFieldSet_PackagePrivate.h"
- #import "GPBUtilities_PackagePrivate.h"
- #import "GPBWireFormat.h"
- static const NSUInteger kDefaultRecursionLimit = 64;
- static void CheckSize(GPBCodedInputStreamState *state, size_t size) {
- size_t newSize = state->bufferPos + size;
- if (newSize > state->bufferSize) {
- [NSException raise:NSParseErrorException format:@""];
- }
- if (newSize > state->currentLimit) {
- // Fast forward to end of currentLimit;
- state->bufferPos = state->currentLimit;
- [NSException raise:NSParseErrorException format:@""];
- }
- }
- static int8_t ReadRawByte(GPBCodedInputStreamState *state) {
- CheckSize(state, sizeof(int8_t));
- return ((int8_t *)state->bytes)[state->bufferPos++];
- }
- static int32_t ReadRawLittleEndian32(GPBCodedInputStreamState *state) {
- CheckSize(state, sizeof(int32_t));
- int32_t value = OSReadLittleInt32(state->bytes, state->bufferPos);
- state->bufferPos += sizeof(int32_t);
- return value;
- }
- static int64_t ReadRawLittleEndian64(GPBCodedInputStreamState *state) {
- CheckSize(state, sizeof(int64_t));
- int64_t value = OSReadLittleInt64(state->bytes, state->bufferPos);
- state->bufferPos += sizeof(int64_t);
- return value;
- }
- static int32_t ReadRawVarint32(GPBCodedInputStreamState *state) {
- int8_t tmp = ReadRawByte(state);
- if (tmp >= 0) {
- return tmp;
- }
- int32_t result = tmp & 0x7f;
- if ((tmp = ReadRawByte(state)) >= 0) {
- result |= tmp << 7;
- } else {
- result |= (tmp & 0x7f) << 7;
- if ((tmp = ReadRawByte(state)) >= 0) {
- result |= tmp << 14;
- } else {
- result |= (tmp & 0x7f) << 14;
- if ((tmp = ReadRawByte(state)) >= 0) {
- result |= tmp << 21;
- } else {
- result |= (tmp & 0x7f) << 21;
- result |= (tmp = ReadRawByte(state)) << 28;
- if (tmp < 0) {
- // Discard upper 32 bits.
- for (int i = 0; i < 5; i++) {
- if (ReadRawByte(state) >= 0) {
- return result;
- }
- }
- [NSException raise:NSParseErrorException
- format:@"Unable to read varint32"];
- }
- }
- }
- }
- return result;
- }
- static int64_t ReadRawVarint64(GPBCodedInputStreamState *state) {
- int32_t shift = 0;
- int64_t result = 0;
- while (shift < 64) {
- int8_t b = ReadRawByte(state);
- result |= (int64_t)(b & 0x7F) << shift;
- if ((b & 0x80) == 0) {
- return result;
- }
- shift += 7;
- }
- [NSException raise:NSParseErrorException format:@"Unable to read varint64"];
- return 0;
- }
- static void SkipRawData(GPBCodedInputStreamState *state, size_t size) {
- CheckSize(state, size);
- state->bufferPos += size;
- }
- double GPBCodedInputStreamReadDouble(GPBCodedInputStreamState *state) {
- int64_t value = ReadRawLittleEndian64(state);
- return GPBConvertInt64ToDouble(value);
- }
- float GPBCodedInputStreamReadFloat(GPBCodedInputStreamState *state) {
- int32_t value = ReadRawLittleEndian32(state);
- return GPBConvertInt32ToFloat(value);
- }
- uint64_t GPBCodedInputStreamReadUInt64(GPBCodedInputStreamState *state) {
- uint64_t value = ReadRawVarint64(state);
- return value;
- }
- uint32_t GPBCodedInputStreamReadUInt32(GPBCodedInputStreamState *state) {
- uint32_t value = ReadRawVarint32(state);
- return value;
- }
- int64_t GPBCodedInputStreamReadInt64(GPBCodedInputStreamState *state) {
- int64_t value = ReadRawVarint64(state);
- return value;
- }
- int32_t GPBCodedInputStreamReadInt32(GPBCodedInputStreamState *state) {
- int32_t value = ReadRawVarint32(state);
- return value;
- }
- uint64_t GPBCodedInputStreamReadFixed64(GPBCodedInputStreamState *state) {
- uint64_t value = ReadRawLittleEndian64(state);
- return value;
- }
- uint32_t GPBCodedInputStreamReadFixed32(GPBCodedInputStreamState *state) {
- uint32_t value = ReadRawLittleEndian32(state);
- return value;
- }
- int32_t GPBCodedInputStreamReadEnum(GPBCodedInputStreamState *state) {
- int32_t value = ReadRawVarint32(state);
- return value;
- }
- int32_t GPBCodedInputStreamReadSFixed32(GPBCodedInputStreamState *state) {
- int32_t value = ReadRawLittleEndian32(state);
- return value;
- }
- int64_t GPBCodedInputStreamReadSFixed64(GPBCodedInputStreamState *state) {
- int64_t value = ReadRawLittleEndian64(state);
- return value;
- }
- int32_t GPBCodedInputStreamReadSInt32(GPBCodedInputStreamState *state) {
- int32_t value = GPBDecodeZigZag32(ReadRawVarint32(state));
- return value;
- }
- int64_t GPBCodedInputStreamReadSInt64(GPBCodedInputStreamState *state) {
- int64_t value = GPBDecodeZigZag64(ReadRawVarint64(state));
- return value;
- }
- BOOL GPBCodedInputStreamReadBool(GPBCodedInputStreamState *state) {
- return ReadRawVarint32(state) != 0;
- }
- int32_t GPBCodedInputStreamReadTag(GPBCodedInputStreamState *state) {
- if (GPBCodedInputStreamIsAtEnd(state)) {
- state->lastTag = 0;
- return 0;
- }
- state->lastTag = ReadRawVarint32(state);
- if (state->lastTag == 0) {
- // If we actually read zero, that's not a valid tag.
- [NSException raise:NSParseErrorException
- format:@"Invalid last tag %d", state->lastTag];
- }
- return state->lastTag;
- }
- NSString *GPBCodedInputStreamReadRetainedString(
- GPBCodedInputStreamState *state) {
- int32_t size = ReadRawVarint32(state);
- NSString *result;
- if (size == 0) {
- result = @"";
- } else {
- CheckSize(state, size);
- result = GPBCreateGPBStringWithUTF8(&state->bytes[state->bufferPos], size);
- state->bufferPos += size;
- }
- return result;
- }
- NSData *GPBCodedInputStreamReadRetainedBytes(GPBCodedInputStreamState *state) {
- int32_t size = ReadRawVarint32(state);
- if (size < 0) return nil;
- CheckSize(state, size);
- NSData *result = [[NSData alloc] initWithBytes:state->bytes + state->bufferPos
- length:size];
- state->bufferPos += size;
- return result;
- }
- NSData *GPBCodedInputStreamReadRetainedBytesNoCopy(
- GPBCodedInputStreamState *state) {
- int32_t size = ReadRawVarint32(state);
- if (size < 0) return nil;
- CheckSize(state, size);
- // Cast is safe because freeWhenDone is NO.
- NSData *result = [[NSData alloc]
- initWithBytesNoCopy:(void *)(state->bytes + state->bufferPos)
- length:size
- freeWhenDone:NO];
- state->bufferPos += size;
- return result;
- }
- size_t GPBCodedInputStreamPushLimit(GPBCodedInputStreamState *state,
- size_t byteLimit) {
- byteLimit += state->bufferPos;
- size_t oldLimit = state->currentLimit;
- if (byteLimit > oldLimit) {
- [NSException raise:NSInvalidArgumentException
- format:@"byteLimit > oldLimit: %tu > %tu", byteLimit, oldLimit];
- }
- state->currentLimit = byteLimit;
- return oldLimit;
- }
- void GPBCodedInputStreamPopLimit(GPBCodedInputStreamState *state,
- size_t oldLimit) {
- state->currentLimit = oldLimit;
- }
- size_t GPBCodedInputStreamBytesUntilLimit(GPBCodedInputStreamState *state) {
- return state->currentLimit - state->bufferPos;
- }
- BOOL GPBCodedInputStreamIsAtEnd(GPBCodedInputStreamState *state) {
- return (state->bufferPos == state->bufferSize) ||
- (state->bufferPos == state->currentLimit);
- }
- void GPBCodedInputStreamCheckLastTagWas(GPBCodedInputStreamState *state,
- int32_t value) {
- if (state->lastTag != value) {
- [NSException raise:NSParseErrorException
- format:@"Last tag: %d should be %d", state->lastTag, value];
- }
- }
- @implementation GPBCodedInputStream
- + (instancetype)streamWithData:(NSData *)data {
- return [[[self alloc] initWithData:data] autorelease];
- }
- - (instancetype)initWithData:(NSData *)data {
- if ((self = [super init])) {
- #ifdef DEBUG
- NSCAssert([self class] == [GPBCodedInputStream class],
- @"Subclassing of GPBCodedInputStream is not allowed.");
- #endif
- buffer_ = [data retain];
- state_.bytes = (const uint8_t *)[data bytes];
- state_.bufferSize = [data length];
- state_.currentLimit = state_.bufferSize;
- }
- return self;
- }
- - (void)dealloc {
- [buffer_ release];
- [super dealloc];
- }
- - (int32_t)readTag {
- return GPBCodedInputStreamReadTag(&state_);
- }
- - (void)checkLastTagWas:(int32_t)value {
- GPBCodedInputStreamCheckLastTagWas(&state_, value);
- }
- - (BOOL)skipField:(int32_t)tag {
- switch (GPBWireFormatGetTagWireType(tag)) {
- case GPBWireFormatVarint:
- GPBCodedInputStreamReadInt32(&state_);
- return YES;
- case GPBWireFormatFixed64:
- SkipRawData(&state_, sizeof(int64_t));
- return YES;
- case GPBWireFormatLengthDelimited:
- SkipRawData(&state_, ReadRawVarint32(&state_));
- return YES;
- case GPBWireFormatStartGroup:
- [self skipMessage];
- GPBCodedInputStreamCheckLastTagWas(
- &state_, GPBWireFormatMakeTag(GPBWireFormatGetTagFieldNumber(tag),
- GPBWireFormatEndGroup));
- return YES;
- case GPBWireFormatEndGroup:
- return NO;
- case GPBWireFormatFixed32:
- SkipRawData(&state_, sizeof(int32_t));
- return YES;
- }
- [NSException raise:NSParseErrorException format:@"Invalid tag %d", tag];
- return NO;
- }
- - (void)skipMessage {
- while (YES) {
- int32_t tag = GPBCodedInputStreamReadTag(&state_);
- if (tag == 0 || ![self skipField:tag]) {
- return;
- }
- }
- }
- - (double)readDouble {
- return GPBCodedInputStreamReadDouble(&state_);
- }
- - (float)readFloat {
- return GPBCodedInputStreamReadFloat(&state_);
- }
- - (uint64_t)readUInt64 {
- return GPBCodedInputStreamReadUInt64(&state_);
- }
- - (int64_t)readInt64 {
- return GPBCodedInputStreamReadInt64(&state_);
- }
- - (int32_t)readInt32 {
- return GPBCodedInputStreamReadInt32(&state_);
- }
- - (uint64_t)readFixed64 {
- return GPBCodedInputStreamReadFixed64(&state_);
- }
- - (uint32_t)readFixed32 {
- return GPBCodedInputStreamReadFixed32(&state_);
- }
- - (BOOL)readBool {
- return GPBCodedInputStreamReadBool(&state_);
- }
- - (NSString *)readString {
- return [GPBCodedInputStreamReadRetainedString(&state_) autorelease];
- }
- - (void)readGroup:(int32_t)fieldNumber
- message:(GPBMessage *)message
- extensionRegistry:(GPBExtensionRegistry *)extensionRegistry {
- if (state_.recursionDepth >= kDefaultRecursionLimit) {
- [NSException raise:NSParseErrorException
- format:@"recursionDepth(%tu) >= %tu", state_.recursionDepth,
- kDefaultRecursionLimit];
- }
- ++state_.recursionDepth;
- [message mergeFromCodedInputStream:self extensionRegistry:extensionRegistry];
- GPBCodedInputStreamCheckLastTagWas(
- &state_, GPBWireFormatMakeTag(fieldNumber, GPBWireFormatEndGroup));
- --state_.recursionDepth;
- }
- - (void)readUnknownGroup:(int32_t)fieldNumber
- message:(GPBUnknownFieldSet *)message {
- if (state_.recursionDepth >= kDefaultRecursionLimit) {
- [NSException raise:NSParseErrorException
- format:@"recursionDepth(%tu) >= %tu", state_.recursionDepth,
- kDefaultRecursionLimit];
- }
- ++state_.recursionDepth;
- [message mergeFromCodedInputStream:self];
- GPBCodedInputStreamCheckLastTagWas(
- &state_, GPBWireFormatMakeTag(fieldNumber, GPBWireFormatEndGroup));
- --state_.recursionDepth;
- }
- - (void)readMessage:(GPBMessage *)message
- extensionRegistry:(GPBExtensionRegistry *)extensionRegistry {
- int32_t length = ReadRawVarint32(&state_);
- if (state_.recursionDepth >= kDefaultRecursionLimit) {
- [NSException raise:NSParseErrorException
- format:@"recursionDepth(%tu) >= %tu", state_.recursionDepth,
- kDefaultRecursionLimit];
- }
- size_t oldLimit = GPBCodedInputStreamPushLimit(&state_, length);
- ++state_.recursionDepth;
- [message mergeFromCodedInputStream:self extensionRegistry:extensionRegistry];
- GPBCodedInputStreamCheckLastTagWas(&state_, 0);
- --state_.recursionDepth;
- GPBCodedInputStreamPopLimit(&state_, oldLimit);
- }
- - (void)readMapEntry:(id)mapDictionary
- extensionRegistry:(GPBExtensionRegistry *)extensionRegistry
- field:(GPBFieldDescriptor *)field
- parentMessage:(GPBMessage *)parentMessage {
- int32_t length = ReadRawVarint32(&state_);
- if (state_.recursionDepth >= kDefaultRecursionLimit) {
- [NSException raise:NSParseErrorException
- format:@"recursionDepth(%tu) >= %tu", state_.recursionDepth,
- kDefaultRecursionLimit];
- }
- size_t oldLimit = GPBCodedInputStreamPushLimit(&state_, length);
- ++state_.recursionDepth;
- GPBDictionaryReadEntry(mapDictionary, self, extensionRegistry, field,
- parentMessage);
- GPBCodedInputStreamCheckLastTagWas(&state_, 0);
- --state_.recursionDepth;
- GPBCodedInputStreamPopLimit(&state_, oldLimit);
- }
- - (NSData *)readBytes {
- return [GPBCodedInputStreamReadRetainedBytes(&state_) autorelease];
- }
- - (uint32_t)readUInt32 {
- return GPBCodedInputStreamReadUInt32(&state_);
- }
- - (int32_t)readEnum {
- return GPBCodedInputStreamReadEnum(&state_);
- }
- - (int32_t)readSFixed32 {
- return GPBCodedInputStreamReadSFixed32(&state_);
- }
- - (int64_t)readSFixed64 {
- return GPBCodedInputStreamReadSFixed64(&state_);
- }
- - (int32_t)readSInt32 {
- return GPBCodedInputStreamReadSInt32(&state_);
- }
- - (int64_t)readSInt64 {
- return GPBCodedInputStreamReadSInt64(&state_);
- }
- @end
- @implementation GPBString {
- @package
- CFStringRef string_;
- unsigned char *utf8_;
- NSUInteger utf8Len_;
- // This lock is used to gate access to utf8_. Once GPBStringInitStringValue()
- // has been called, string_ will be filled in, and utf8_ will be NULL.
- OSSpinLock lock_;
- BOOL hasBOM_;
- BOOL is7BitAscii_;
- }
- // Returns true if the passed in bytes are 7 bit ascii.
- // This routine needs to be fast.
- static bool AreBytesIn7BitASCII(const uint8_t *bytes, NSUInteger len) {
- // In the loops below, it's more efficient to collect rather than do
- // conditional at every step.
- #if __LP64__
- // Align bytes. This is especially important in case of 3 byte BOM.
- while (len > 0 && ((size_t)bytes & 0x07)) {
- if (*bytes++ & 0x80) return false;
- len--;
- }
- while (len >= 32) {
- uint64_t val = *(const uint64_t *)bytes;
- uint64_t hiBits = (val & 0x8080808080808080ULL);
- bytes += 8;
- val = *(const uint64_t *)bytes;
- hiBits |= (val & 0x8080808080808080ULL);
- bytes += 8;
- val = *(const uint64_t *)bytes;
- hiBits |= (val & 0x8080808080808080ULL);
- bytes += 8;
- val = *(const uint64_t *)bytes;
- if (hiBits | (val & 0x8080808080808080ULL)) return false;
- bytes += 8;
- len -= 32;
- }
- while (len >= 16) {
- uint64_t val = *(const uint64_t *)bytes;
- uint64_t hiBits = (val & 0x8080808080808080ULL);
- bytes += 8;
- val = *(const uint64_t *)bytes;
- if (hiBits | (val & 0x8080808080808080ULL)) return false;
- bytes += 8;
- len -= 16;
- }
- while (len >= 8) {
- uint64_t val = *(const uint64_t *)bytes;
- if (val & 0x8080808080808080ULL) return false;
- bytes += 8;
- len -= 8;
- }
- #else // __LP64__
- // Align bytes. This is especially important in case of 3 byte BOM.
- while (len > 0 && ((size_t)bytes & 0x03)) {
- if (*bytes++ & 0x80) return false;
- len--;
- }
- while (len >= 16) {
- uint32_t val = *(const uint32_t *)bytes;
- uint32_t hiBits = (val & 0x80808080U);
- bytes += 4;
- val = *(const uint32_t *)bytes;
- hiBits |= (val & 0x80808080U);
- bytes += 4;
- val = *(const uint32_t *)bytes;
- hiBits |= (val & 0x80808080U);
- bytes += 4;
- val = *(const uint32_t *)bytes;
- if (hiBits | (val & 0x80808080U)) return false;
- bytes += 4;
- len -= 16;
- }
- while (len >= 8) {
- uint32_t val = *(const uint32_t *)bytes;
- uint32_t hiBits = (val & 0x80808080U);
- bytes += 4;
- val = *(const uint32_t *)bytes;
- if (hiBits | (val & 0x80808080U)) return false;
- bytes += 4;
- len -= 8;
- }
- #endif // __LP64__
- while (len >= 4) {
- uint32_t val = *(const uint32_t *)bytes;
- if (val & 0x80808080U) return false;
- bytes += 4;
- len -= 4;
- }
- while (len--) {
- if (*bytes++ & 0x80) return false;
- }
- return true;
- }
- static void GPBStringInitStringValue(GPBString *string) {
- OSSpinLockLock(&string->lock_);
- GPBStringInitStringValueAlreadyLocked(string);
- OSSpinLockUnlock(&string->lock_);
- }
- static void GPBStringInitStringValueAlreadyLocked(GPBString *string) {
- if (string->string_ == NULL && string->utf8_ != NULL) {
- // Using kCFAllocatorMalloc for contentsDeallocator, as buffer in
- // string->utf8_ is being handed off.
- string->string_ = CFStringCreateWithBytesNoCopy(
- NULL, string->utf8_, string->utf8Len_, kCFStringEncodingUTF8, false,
- kCFAllocatorMalloc);
- if (!string->string_) {
- #ifdef DEBUG
- // https://developers.google.com/protocol-buffers/docs/proto#scalar
- NSLog(@"UTF8 failure, is some field type 'string' when it should be "
- @"'bytes'?");
- #endif
- string->string_ = CFSTR("");
- string->utf8Len_ = 0;
- // On failure, we have to clean up the buffer.
- free(string->utf8_);
- }
- string->utf8_ = NULL;
- }
- }
- GPBString *GPBCreateGPBStringWithUTF8(const void *bytes, NSUInteger length) {
- GPBString *result = [[GPBString alloc] initWithBytes:bytes length:length];
- return result;
- }
- - (instancetype)initWithBytes:(const void *)bytes length:(NSUInteger)length {
- self = [super init];
- if (self) {
- utf8_ = malloc(length);
- memcpy(utf8_, bytes, length);
- utf8Len_ = length;
- lock_ = OS_SPINLOCK_INIT;
- is7BitAscii_ = AreBytesIn7BitASCII(bytes, length);
- if (length >= 3 && memcmp(utf8_, "\xef\xbb\xbf", 3) == 0) {
- // We can't just remove the BOM from the string here, because in the case
- // where we have > 1 BOM at the beginning of the string, we will remove one,
- // and the internal NSString we create will remove the next one, and we will
- // end up with a GPBString != NSString issue.
- // We also just can't remove all the BOMs because then we would end up with
- // potential cases where a GPBString and an NSString made with the same
- // UTF8 buffer would in fact be different.
- // We record the fact we have a BOM, and use it as necessary to simulate
- // what NSString would return for various calls.
- hasBOM_ = YES;
- #if DEBUG
- // Sending BOMs across the line is just wasting bits.
- NSLog(@"Bad data? String should not have BOM!");
- #endif // DEBUG
- }
- }
- return self;
- }
- - (void)dealloc {
- if (string_ != NULL) {
- CFRelease(string_);
- }
- if (utf8_ != NULL) {
- free(utf8_);
- }
- [super dealloc];
- }
- // Required NSString overrides.
- - (NSUInteger)length {
- if (is7BitAscii_) {
- return utf8Len_;
- } else {
- GPBStringInitStringValue(self);
- return CFStringGetLength(string_);
- }
- }
- - (unichar)characterAtIndex:(NSUInteger)anIndex {
- OSSpinLockLock(&lock_);
- if (is7BitAscii_ && utf8_) {
- unichar result = utf8_[anIndex];
- OSSpinLockUnlock(&lock_);
- return result;
- } else {
- GPBStringInitStringValueAlreadyLocked(self);
- OSSpinLockUnlock(&lock_);
- return CFStringGetCharacterAtIndex(string_, anIndex);
- }
- }
- // Override a couple of methods that typically want high performance.
- - (id)copyWithZone:(NSZone *)zone {
- GPBStringInitStringValue(self);
- return [(NSString *)string_ copyWithZone:zone];
- }
- - (id)mutableCopyWithZone:(NSZone *)zone {
- GPBStringInitStringValue(self);
- return [(NSString *)string_ mutableCopyWithZone:zone];
- }
- - (NSUInteger)hash {
- // Must convert to string here to make sure that the hash is always
- // consistent no matter what state the GPBString is in.
- GPBStringInitStringValue(self);
- return CFHash(string_);
- }
- - (BOOL)isEqual:(id)object {
- if (self == object) {
- return YES;
- }
- if ([object isKindOfClass:[NSString class]]) {
- GPBStringInitStringValue(self);
- return CFStringCompare(string_, (CFStringRef)object, 0) ==
- kCFCompareEqualTo;
- }
- return NO;
- }
- - (void)getCharacters:(unichar *)buffer range:(NSRange)aRange {
- OSSpinLockLock(&lock_);
- if (is7BitAscii_ && utf8_) {
- unsigned char *bytes = &(utf8_[aRange.location]);
- for (NSUInteger i = 0; i < aRange.length; ++i) {
- buffer[i] = bytes[i];
- }
- OSSpinLockUnlock(&lock_);
- } else {
- GPBStringInitStringValueAlreadyLocked(self);
- OSSpinLockUnlock(&lock_);
- CFStringGetCharacters(string_, CFRangeMake(aRange.location, aRange.length),
- buffer);
- }
- }
- - (NSUInteger)lengthOfBytesUsingEncoding:(NSStringEncoding)encoding {
- if ((encoding == NSUTF8StringEncoding) ||
- (encoding == NSASCIIStringEncoding && is7BitAscii_)) {
- return utf8Len_ - (hasBOM_ ? 3 : 0);
- } else {
- GPBStringInitStringValue(self);
- return [(NSString *)string_ lengthOfBytesUsingEncoding:encoding];
- }
- }
- - (BOOL)getBytes:(void *)buffer
- maxLength:(NSUInteger)maxLength
- usedLength:(NSUInteger *)usedLength
- encoding:(NSStringEncoding)encoding
- options:(NSStringEncodingConversionOptions)options
- range:(NSRange)range
- remainingRange:(NSRangePointer)remainingRange {
- // [NSString getBytes:maxLength:usedLength:encoding:options:range:remainingRange]
- // does not return reliable results if the maxLength argument is 0
- // (Radar 16385183). Therefore we have special cased it as a slow case so
- // that it behaves however Apple intends it to behave. It should be a rare
- // case.
- //
- // [NSString getBytes:maxLength:usedLength:encoding:options:range:remainingRange]
- // does not return reliable results if the range is outside of the strings
- // length (Radar 16396177). Therefore we have special cased it as a slow
- // case so that it behaves however Apple intends it to behave. It should
- // be a rare case.
- //
- // We can optimize the UTF8StringEncoding and NSASCIIStringEncoding with no
- // options cases.
- if ((options == 0) &&
- (encoding == NSUTF8StringEncoding || encoding == NSASCIIStringEncoding) &&
- (maxLength != 0) &&
- (NSMaxRange(range) <= utf8Len_)) {
- // Might be able to optimize it.
- OSSpinLockLock(&lock_);
- if (is7BitAscii_ && utf8_) {
- NSUInteger length = range.length;
- length = (length < maxLength) ? length : maxLength;
- memcpy(buffer, utf8_ + range.location, length);
- if (usedLength) {
- *usedLength = length;
- }
- if (remainingRange) {
- remainingRange->location = range.location + length;
- remainingRange->length = range.length - length;
- }
- OSSpinLockUnlock(&lock_);
- if (length > 0) {
- return YES;
- } else {
- return NO;
- }
- } else {
- GPBStringInitStringValueAlreadyLocked(self);
- OSSpinLockUnlock(&lock_);
- }
- } else {
- GPBStringInitStringValue(self);
- }
- return [(NSString *)string_ getBytes:buffer
- maxLength:maxLength
- usedLength:usedLength
- encoding:encoding
- options:options
- range:range
- remainingRange:remainingRange];
- }
- @end
|