Plugins/IPMonitor/dnsAgent.m

   1 /*
   2  * Copyright (c) 2015, 2016 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 "configAgent.h"
  25
  26 @interface DNSAgent:ConfigAgent
  27
  28 @property (nonatomic) AgentType internalAgentType;
  29 @property (nonatomic) NSString *internalAgentName;
  30 @property (nonatomic) AgentSubType internalAgentSubType;
  31
  32 @end
  33
  34 @implementation DNSAgent
  35
  36 @synthesize agentUUID;
  37 @synthesize agentDescription;
  38
  39 + (NSString *)agentType
  40 {
  41         return @kConfigAgentTypeDNS;
  42 }
  43
  44 - (instancetype)initWithParameters:(NSDictionary *)parameters
  45 {
  46         self = [super initWithParameters:parameters];
  47         if (self) {
  48                 NSString *intf = [parameters valueForKey:@kEntityName];
  49                 NSNumber *subType = [parameters valueForKey:@kAgentSubType];
  50                 NSString *type = [[self class] agentType];
  51
  52                 if ([subType unsignedIntegerValue] == kAgentSubTypeMulticast) {
  53                         type = @kConfigAgentTypeDNSMulticast;
  54                 } else if ([subType unsignedIntegerValue] == kAgentSubTypePrivate) {
  55                         type = @kConfigAgentTypeDNSPrivate;
  56                 }
  57
  58                 _internalAgentName = [NSString stringWithFormat:@"%@-%@", type, intf];
  59                 _internalAgentSubType = [subType unsignedIntegerValue];
  60                 _internalAgentType = kAgentTypeDNS;
  61
  62                 agentDescription = _internalAgentName;
  63                 agentUUID = [super createUUIDForName:agentDescription];
  64                 if (agentUUID == nil) {
  65                         agentUUID = [NSUUID UUID];
  66                 }
  67         }
  68
  69         return self;
  70 }
  71
  72 - (AgentType)getAgentType
  73 {
  74         return _internalAgentType;
  75 }
  76
  77 - (NSString *)getAgentName
  78 {
  79         return _internalAgentName;
  80 }
  81
  82 - (AgentSubType)getAgentSubType
  83 {
  84         return _internalAgentSubType;
  85 }
  86
  87 - (NSUUID *)getAgentUUID
  88 {
  89         return agentUUID;
  90 }
  91
  92 @end