Analytics/SQLite/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 #import "SFObjCType.h"
  25
  26
  27 static NSArray<SFObjCType *> *_SFObjCTypesByCode = nil;
  28
  29
  30 #define OBJC_TYPE(_code, _encoding, _name, _size, _flags) \
  31     [[SFObjCType alloc] initWithCode:SFObjCType##_code encoding:_encoding name:_name className:nil size:_size flags:SFObjCTypeFlag##_flags]
  32
  33 @implementation SFObjCType
  34
  35 @synthesize code = _code;
  36 @synthesize encoding = _encoding;
  37 @synthesize name = _name;
  38 @synthesize className = _className;
  39 @synthesize size = _size;
  40 @synthesize flags = _flags;
  41
  42 + (SFObjCType *)typeForEncoding:(const char *)encodingUTF8 {
  43     NSString *encoding = @(encodingUTF8);
  44
  45     static dispatch_once_t once;
  46     dispatch_once(&once, ^{
  47         _SFObjCTypesByCode = @[
  48             OBJC_TYPE(Char,             @"c",   @"char",                sizeof(char),               IntegerNumber),
  49             OBJC_TYPE(Short,            @"s",   @"short",               sizeof(short),              IntegerNumber),
  50             OBJC_TYPE(Int,              @"i",   @"int",                 sizeof(int),                IntegerNumber),
  51             OBJC_TYPE(Long,             @"l",   @"long",                sizeof(long),               IntegerNumber),
  52             OBJC_TYPE(LongLong,         @"q",   @"long long",           sizeof(long long),          IntegerNumber),
  53             OBJC_TYPE(UnsignedChar,     @"C",   @"unsigned char",       sizeof(unsigned char),      IntegerNumber),
  54             OBJC_TYPE(UnsignedShort,    @"S",   @"unsigned short",      sizeof(unsigned short),     IntegerNumber),
  55             OBJC_TYPE(UnsignedInt,      @"I",   @"unsigned int",        sizeof(unsigned int),       IntegerNumber),
  56             OBJC_TYPE(UnsignedLong,     @"L",   @"unsigned long",       sizeof(unsigned long),      IntegerNumber),
  57             OBJC_TYPE(UnsignedLongLong, @"Q",   @"unsigned long long",  sizeof(unsigned long long), IntegerNumber),
  58             OBJC_TYPE(Float,            @"f",   @"float",               sizeof(float),              FloatingPointNumber),
  59             OBJC_TYPE(Double,           @"d",   @"double",              sizeof(double),             FloatingPointNumber),
  60             OBJC_TYPE(Bool,             @"B",   @"bool",                sizeof(bool),               IntegerNumber),
  61             OBJC_TYPE(Void,             @"v",   @"void",                sizeof(void),               None),
  62             OBJC_TYPE(CharPointer,      @"*",   @"char*",               sizeof(char*),              None),
  63             OBJC_TYPE(Object,           @"@",   @"id",                  sizeof(id),                 None),
  64             OBJC_TYPE(Class,            @"#",   @"Class",               sizeof(Class),              None),
  65         ];
  66     });
  67
  68     switch (encodingUTF8[0]) {
  69         case 'c': return _SFObjCTypesByCode[SFObjCTypeChar];
  70         case 's': return _SFObjCTypesByCode[SFObjCTypeShort];
  71         case 'i': return _SFObjCTypesByCode[SFObjCTypeInt];
  72         case 'l': return _SFObjCTypesByCode[SFObjCTypeLong];
  73         case 'q': return _SFObjCTypesByCode[SFObjCTypeLongLong];
  74         case 'C': return _SFObjCTypesByCode[SFObjCTypeUnsignedChar];
  75         case 'S': return _SFObjCTypesByCode[SFObjCTypeUnsignedShort];
  76         case 'I': return _SFObjCTypesByCode[SFObjCTypeUnsignedInt];
  77         case 'L': return _SFObjCTypesByCode[SFObjCTypeUnsignedLong];
  78         case 'Q': return _SFObjCTypesByCode[SFObjCTypeUnsignedLongLong];
  79         case 'f': return _SFObjCTypesByCode[SFObjCTypeFloat];
  80         case 'd': return _SFObjCTypesByCode[SFObjCTypeDouble];
  81         case 'B': return _SFObjCTypesByCode[SFObjCTypeBool];
  82         case 'v': return _SFObjCTypesByCode[SFObjCTypeVoid];
  83         case '*': return _SFObjCTypesByCode[SFObjCTypeCharPointer];
  84         case '@': {
  85             if (encoding.length > 3 && [encoding characterAtIndex:1] == '"' && [encoding characterAtIndex:encoding.length-1] == '"') {
  86                 NSString *className = [encoding substringWithRange:NSMakeRange(2, encoding.length-3)];
  87                 NSString *name = [className stringByAppendingString:@"*"];
  88                 return [[SFObjCType alloc] initWithCode:SFObjCTypeObject encoding:encoding name:name className:className size:sizeof(id) flags:SFObjCTypeFlagNone];
  89             } else {
  90                 return _SFObjCTypesByCode[SFObjCTypeObject];
  91             }
  92         }
  93         case '#': return _SFObjCTypesByCode[SFObjCTypeClass];
  94         case ':': return _SFObjCTypesByCode[SFObjCTypeSelector];
  95         case '[': return OBJC_TYPE(Array,       encoding,   @"array",       0,              None);
  96         case '{': return OBJC_TYPE(Structure,   encoding,   @"structure",   0,              None);
  97         case '(': return OBJC_TYPE(Union,       encoding,   @"union",       0,              None);
  98         case 'b': return OBJC_TYPE(Bitfield,    encoding,   @"bitfield",    0,              None);
  99         case '^': return OBJC_TYPE(Pointer,     encoding,   @"pointer",     sizeof(void*),  None);
 100         case '?':
 101         default:
 102             return [[SFObjCType alloc] initWithCode:SFObjCTypeUnknown encoding:encoding name:@"unknown" className:nil size:0 flags:0];
 103     }
 104 }
 105
 106 + (SFObjCType *)typeForValue:(NSValue *)value {
 107     NS_VALID_UNTIL_END_OF_SCOPE NSValue *arcSafeValue = value;
 108     return [SFObjCType typeForEncoding:[arcSafeValue objCType]];
 109 }
 110
 111 - (id)initWithCode:(SFObjCTypeCode)code encoding:(NSString *)encoding name:(NSString *)name className:(NSString *)className size:(NSUInteger)size flags:(NSUInteger)flags {
 112     if ((self = [super init])) {
 113         _code = code;
 114         _encoding = encoding;
 115         _name = name;
 116         _className = className;
 117         _size = size;
 118         _flags = flags;
 119     }
 120     return self;
 121 }
 122
 123 - (BOOL)isNumber {
 124     return (_flags & SFObjCTypeFlagNumberMask) != 0;
 125 }
 126
 127 - (BOOL)isIntegerNumber {
 128     return (_flags & SFObjCTypeFlagIntegerNumber) != 0;
 129 }
 130
 131 - (BOOL)isFloatingPointNumber {
 132     return (_flags & SFObjCTypeFlagFloatingPointNumber) != 0;
 133 }
 134
 135 - (BOOL)isObject {
 136     return _code == SFObjCTypeObject;
 137 }
 138
 139 - (id)objectWithBytes:(const void *)bytes {
 140     switch (_code) {
 141         case SFObjCTypeChar:              return [NSNumber numberWithChar:*((const char *)bytes)];
 142         case SFObjCTypeShort:             return [NSNumber numberWithShort:*((const short *)bytes)];
 143         case SFObjCTypeInt:               return [NSNumber numberWithInt:*((const int *)bytes)];
 144         case SFObjCTypeLong:              return [NSNumber numberWithLong:*((const long *)bytes)];
 145         case SFObjCTypeLongLong:          return [NSNumber numberWithLongLong:*((const long long *)bytes)];
 146         case SFObjCTypeUnsignedChar:      return [NSNumber numberWithUnsignedChar:*((const unsigned char *)bytes)];
 147         case SFObjCTypeUnsignedShort:     return [NSNumber numberWithUnsignedShort:*((const unsigned short *)bytes)];
 148         case SFObjCTypeUnsignedInt:       return [NSNumber numberWithUnsignedInt:*((const unsigned int *)bytes)];
 149         case SFObjCTypeUnsignedLong:      return [NSNumber numberWithUnsignedLong:*((const unsigned long *)bytes)];
 150         case SFObjCTypeUnsignedLongLong:  return [NSNumber numberWithUnsignedLongLong:*((const unsigned long long *)bytes)];
 151         case SFObjCTypeFloat:             return [NSNumber numberWithFloat:*((const float *)bytes)];
 152         case SFObjCTypeDouble:            return [NSNumber numberWithDouble:*((const double *)bytes)];
 153         case SFObjCTypeBool:              return [NSNumber numberWithBool:(BOOL)*((const _Bool *)bytes)];
 154         case SFObjCTypeObject:            return (__bridge id)((const void *)(*((uintptr_t *)bytes)));
 155         default:
 156             [NSException raise:NSInternalInconsistencyException format:@"For class %@, Unsupported boxing type: %@", _className, _name];
 157             return nil;
 158     }
 159 }
 160
 161 - (void)getBytes:(void *)bytes forObject:(id)object {
 162     if ([object isKindOfClass:[NSValue class]]) {
 163         [object getValue:bytes];
 164     } else {
 165         [NSException raise:NSInternalInconsistencyException format:@"Unsupported unboxing type: %@", _name];
 166     }
 167 }
 168
 169 @end