libkdd/kcdata/KCDStructTypeDescription.m

   1 /*
   2  * Copyright (c) 2015 Apple Inc. All rights reserved.
   3  *
   4  * @APPLE_OSREFERENCE_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. The rights granted to you under the License
  10  * may not be used to create, or enable the creation or redistribution of,
  11  * unlawful or unlicensed copies of an Apple operating system, or to
  12  * circumvent, violate, or enable the circumvention or violation of, any
  13  * terms of an Apple operating system software license agreement.
  14  *
  15  * Please obtain a copy of the License at
  16  * http://www.opensource.apple.com/apsl/ and read it before using this file.
  17  *
  18  * The Original Code and all software distributed under the License are
  19  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  20  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  21  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  22  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  23  * Please see the License for the specific language governing rights and
  24  * limitations under the License.
  25  *
  26  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  27  */
  28
  29 #import "KCDStructTypeDescription.h"
  30
  31 #ifndef KCDATA_TYPE_MAX_WITH_DESC
  32 #define KCDATA_TYPE_MAX_WITH_DESC 0x6
  33 #endif
  34
  35 @interface
  36 KCDStructTypeDescription () {
  37         int _typeID;
  38         NSString * _name;
  39         NSMutableArray * _fields;
  40         BOOL _needDescriptionAsKey;
  41 }
  42
  43 @end
  44
  45 @implementation KCDStructTypeDescription
  46
  47 - (id)initWithType:(int)typeID withName:(NSString *)name
  48 {
  49         if ((self = [super init])) {
  50                 _typeID = typeID;
  51                 _name = name;
  52                 _needDescriptionAsKey = NO;
  53                 if (typeID >= 0x1 && typeID <= KCDATA_TYPE_MAX_WITH_DESC)
  54                         _needDescriptionAsKey = YES;
  55
  56                 _fields = [[NSMutableArray alloc] init];
  57                 return self;
  58         }
  59         return NULL;
  60 }
  61
  62 - (void)addFieldBasicType:(KCDBasicTypeDescription *)fieldType
  63 {
  64         [_fields addObject:fieldType];
  65 }
  66
  67 - (NSMutableDictionary *)parseData:(void *)dataBuffer ofLength:(uint32_t)length
  68 {
  69         NSMutableDictionary * retval = [[NSMutableDictionary alloc] init];
  70         for (KCDataType * fi in _fields) {
  71                 NSMutableDictionary * _d = [fi parseData:dataBuffer ofLength:length];
  72                 for (NSString * k in [_d keyEnumerator]) {
  73                         retval[k] = _d[k];
  74                 }
  75         }
  76         if (_needDescriptionAsKey) {
  77                 NSString * desc = retval[@"desc"];
  78                 NSObject * obj = retval[@"data"];
  79                 retval[desc] = obj;
  80                 [retval removeObjectForKey:@"desc"];
  81                 [retval removeObjectForKey:@"data"];
  82         }
  83         return retval;
  84 }
  85
  86 - (NSString *)description
  87 {
  88         return [NSString stringWithFormat:@"type: %d => \"%@\" ", _typeID, _name];
  89 }
  90
  91 - (NSString *)name
  92 {
  93         return _name;
  94 }
  95
  96 - (uint32_t)count
  97 {
  98         return (uint32_t)[_fields count];
  99 }
 100
 101 - (int)typeID
 102 {
 103         return _typeID;
 104 }
 105
 106 @end