/ Analytics / SQLite / SFObjCType.m
SFObjCType.m
  1  /*
  2   * Copyright (c) 2017 Apple Inc. All Rights Reserved.
  3   *
  4   * @APPLE_LICENSE_HEADER_START@
  5   *
  6   * This file contains Original Code and/or Modifications of Original Code
  7   * as defined in and that are subject to the Apple Public Source License
  8   * Version 2.0 (the 'License'). You may not use this file except in
  9   * compliance with the License. Please obtain a copy of the License at
 10   * http://www.opensource.apple.com/apsl/ and read it before using this
 11   * file.
 12   *
 13   * The Original Code and all software distributed under the License are
 14   * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 15   * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 16   * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 17   * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 18   * Please see the License for the specific language governing rights and
 19   * limitations under the License.
 20   *
 21   * @APPLE_LICENSE_HEADER_END@
 22   */
 23  
 24  #if __OBJC2__
 25  
 26  #import "SFObjCType.h"
 27  
 28  static NSArray<SFObjCType *> *_SFObjCTypesByCode = nil;
 29  
 30  
 31  #define OBJC_TYPE(_code, _encoding, _name, _size, _flags) \
 32      [[SFObjCType alloc] initWithCode:SFObjCType##_code encoding:_encoding name:_name className:nil size:_size flags:SFObjCTypeFlag##_flags]
 33  
 34  @implementation SFObjCType
 35  
 36  @synthesize code = _code;
 37  @synthesize encoding = _encoding;
 38  @synthesize name = _name;
 39  @synthesize className = _className;
 40  @synthesize size = _size;
 41  @synthesize flags = _flags;
 42  
 43  + (SFObjCType *)typeForEncoding:(const char *)encodingUTF8 {
 44      NSString *encoding = @(encodingUTF8);
 45  
 46      static dispatch_once_t once;
 47      dispatch_once(&once, ^{
 48          _SFObjCTypesByCode = @[
 49              OBJC_TYPE(Char,             @"c",   @"char",                sizeof(char),               IntegerNumber),
 50              OBJC_TYPE(Short,            @"s",   @"short",               sizeof(short),              IntegerNumber),
 51              OBJC_TYPE(Int,              @"i",   @"int",                 sizeof(int),                IntegerNumber),
 52              OBJC_TYPE(Long,             @"l",   @"long",                sizeof(long),               IntegerNumber),
 53              OBJC_TYPE(LongLong,         @"q",   @"long long",           sizeof(long long),          IntegerNumber),
 54              OBJC_TYPE(UnsignedChar,     @"C",   @"unsigned char",       sizeof(unsigned char),      IntegerNumber),
 55              OBJC_TYPE(UnsignedShort,    @"S",   @"unsigned short",      sizeof(unsigned short),     IntegerNumber),
 56              OBJC_TYPE(UnsignedInt,      @"I",   @"unsigned int",        sizeof(unsigned int),       IntegerNumber),
 57              OBJC_TYPE(UnsignedLong,     @"L",   @"unsigned long",       sizeof(unsigned long),      IntegerNumber),
 58              OBJC_TYPE(UnsignedLongLong, @"Q",   @"unsigned long long",  sizeof(unsigned long long), IntegerNumber),
 59              OBJC_TYPE(Float,            @"f",   @"float",               sizeof(float),              FloatingPointNumber),
 60              OBJC_TYPE(Double,           @"d",   @"double",              sizeof(double),             FloatingPointNumber),
 61              OBJC_TYPE(Bool,             @"B",   @"bool",                sizeof(bool),               IntegerNumber),
 62              OBJC_TYPE(Void,             @"v",   @"void",                sizeof(void),               None),
 63              OBJC_TYPE(CharPointer,      @"*",   @"char*",               sizeof(char*),              None),
 64              OBJC_TYPE(Object,           @"@",   @"id",                  sizeof(id),                 None),
 65              OBJC_TYPE(Class,            @"#",   @"Class",               sizeof(Class),              None),
 66          ];
 67      });
 68      
 69      switch (encodingUTF8[0]) {
 70          case 'c': return _SFObjCTypesByCode[SFObjCTypeChar];
 71          case 's': return _SFObjCTypesByCode[SFObjCTypeShort];
 72          case 'i': return _SFObjCTypesByCode[SFObjCTypeInt];
 73          case 'l': return _SFObjCTypesByCode[SFObjCTypeLong];
 74          case 'q': return _SFObjCTypesByCode[SFObjCTypeLongLong];
 75          case 'C': return _SFObjCTypesByCode[SFObjCTypeUnsignedChar];
 76          case 'S': return _SFObjCTypesByCode[SFObjCTypeUnsignedShort];
 77          case 'I': return _SFObjCTypesByCode[SFObjCTypeUnsignedInt];
 78          case 'L': return _SFObjCTypesByCode[SFObjCTypeUnsignedLong];
 79          case 'Q': return _SFObjCTypesByCode[SFObjCTypeUnsignedLongLong];
 80          case 'f': return _SFObjCTypesByCode[SFObjCTypeFloat];
 81          case 'd': return _SFObjCTypesByCode[SFObjCTypeDouble];
 82          case 'B': return _SFObjCTypesByCode[SFObjCTypeBool];
 83          case 'v': return _SFObjCTypesByCode[SFObjCTypeVoid];
 84          case '*': return _SFObjCTypesByCode[SFObjCTypeCharPointer];
 85          case '@': {
 86              if (encoding.length > 3 && [encoding characterAtIndex:1] == '"' && [encoding characterAtIndex:encoding.length-1] == '"') {
 87                  NSString *className = [encoding substringWithRange:NSMakeRange(2, encoding.length-3)];
 88                  NSString *name = [className stringByAppendingString:@"*"];
 89                  return [[SFObjCType alloc] initWithCode:SFObjCTypeObject encoding:encoding name:name className:className size:sizeof(id) flags:SFObjCTypeFlagNone];
 90              } else {
 91                  return _SFObjCTypesByCode[SFObjCTypeObject];
 92              }
 93          }
 94          case '#': return _SFObjCTypesByCode[SFObjCTypeClass];
 95          case ':': return _SFObjCTypesByCode[SFObjCTypeSelector];
 96          case '[': return OBJC_TYPE(Array,       encoding,   @"array",       0,              None);
 97          case '{': return OBJC_TYPE(Structure,   encoding,   @"structure",   0,              None);
 98          case '(': return OBJC_TYPE(Union,       encoding,   @"union",       0,              None);
 99          case 'b': return OBJC_TYPE(Bitfield,    encoding,   @"bitfield",    0,              None);
100          case '^': return OBJC_TYPE(Pointer,     encoding,   @"pointer",     sizeof(void*),  None);
101          case '?':
102          default:
103              return [[SFObjCType alloc] initWithCode:SFObjCTypeUnknown encoding:encoding name:@"unknown" className:nil size:0 flags:0];
104      }
105  }
106  
107  + (SFObjCType *)typeForValue:(NSValue *)value {
108      NS_VALID_UNTIL_END_OF_SCOPE NSValue *arcSafeValue = value;
109      return [SFObjCType typeForEncoding:[arcSafeValue objCType]];
110  }
111  
112  - (id)initWithCode:(SFObjCTypeCode)code encoding:(NSString *)encoding name:(NSString *)name className:(NSString *)className size:(NSUInteger)size flags:(NSUInteger)flags {
113      if ((self = [super init])) {
114          _code = code;
115          _encoding = encoding;
116          _name = name;
117          _className = className;
118          _size = size;
119          _flags = flags;
120      }
121      return self;
122  }
123  
124  - (BOOL)isNumber {
125      return (_flags & SFObjCTypeFlagNumberMask) != 0;
126  }
127  
128  - (BOOL)isIntegerNumber {
129      return (_flags & SFObjCTypeFlagIntegerNumber) != 0;
130  }
131  
132  - (BOOL)isFloatingPointNumber {
133      return (_flags & SFObjCTypeFlagFloatingPointNumber) != 0;
134  }
135  
136  - (BOOL)isObject {
137      return _code == SFObjCTypeObject;
138  }
139  
140  - (id)objectWithBytes:(const void *)bytes {
141      switch (_code) {
142          case SFObjCTypeChar:              return [NSNumber numberWithChar:*((const char *)bytes)];
143          case SFObjCTypeShort:             return [NSNumber numberWithShort:*((const short *)bytes)];
144          case SFObjCTypeInt:               return [NSNumber numberWithInt:*((const int *)bytes)];
145          case SFObjCTypeLong:              return [NSNumber numberWithLong:*((const long *)bytes)];
146          case SFObjCTypeLongLong:          return [NSNumber numberWithLongLong:*((const long long *)bytes)];
147          case SFObjCTypeUnsignedChar:      return [NSNumber numberWithUnsignedChar:*((const unsigned char *)bytes)];
148          case SFObjCTypeUnsignedShort:     return [NSNumber numberWithUnsignedShort:*((const unsigned short *)bytes)];
149          case SFObjCTypeUnsignedInt:       return [NSNumber numberWithUnsignedInt:*((const unsigned int *)bytes)];
150          case SFObjCTypeUnsignedLong:      return [NSNumber numberWithUnsignedLong:*((const unsigned long *)bytes)];
151          case SFObjCTypeUnsignedLongLong:  return [NSNumber numberWithUnsignedLongLong:*((const unsigned long long *)bytes)];
152          case SFObjCTypeFloat:             return [NSNumber numberWithFloat:*((const float *)bytes)];
153          case SFObjCTypeDouble:            return [NSNumber numberWithDouble:*((const double *)bytes)];
154          case SFObjCTypeBool:              return [NSNumber numberWithBool:(BOOL)*((const _Bool *)bytes)];
155          case SFObjCTypeObject:            return (__bridge id)((const void *)(*((uintptr_t *)bytes)));
156          default:
157              [NSException raise:NSInternalInconsistencyException format:@"For class %@, Unsupported boxing type: %@", _className, _name];
158              return nil;
159      }
160  }
161  
162  - (void)getBytes:(void *)bytes forObject:(id)object {
163      if ([object isKindOfClass:[NSValue class]]) {
164          [object getValue:bytes];
165      } else {
166          [NSException raise:NSInternalInconsistencyException format:@"Unsupported unboxing type: %@", _name];
167      }
168  }
169  
170  @end
171  
172  #endif