/*
 * Copyright (c) 2018 Apple Inc. All rights reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 *
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this
 * file.
 *
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 *
 * @APPLE_LICENSE_HEADER_END@
 */

#import <Foundation/Foundation.h>
#import <EventFactory/EventFactory.h>

#import "IPMonitorParser.h"

#define TokenInterfaceNameV6 "interfaceNameV6"
#define TokenAddressAction "addressAction"
#define TokenAddressAction6 "addressAction6"
#define TokenAddress "address"
#define TokenAddress6 "address6"
#define TokenServiceID "serviceID"
#define TokenV4Changes "v4Changes"
#define TokenV6Changes "v6Changes"

@interface IPMonitorParser ()
@property (readonly, nonatomic) NSRegularExpression *netChangeRegex;
@end

@implementation IPMonitorParser

- (instancetype)init
{
	NSError *regexError = nil;
	_netChangeRegex = [[NSRegularExpression alloc] initWithPattern:@"(?<"TokenInterfaceName">\\w+)(?<"TokenAddressAction">[\\+\\-\\!\\/\\\\]?)(:(?<"TokenAddress">[:.0-9a-f]+))?" options:0 error:&regexError];
	if (_netChangeRegex == nil) {
		specs_log_err("Failed to create the network change regex: %@", regexError);
		return nil;
	}

	NSArray<EFLogEventMatch *> *matches = @[
		[[EFLogEventMatch alloc] initWithPattern:@"\\d+. (?<"TokenInterfaceName">\\w+) serviceID=(?<"TokenServiceID">[-\\w]+) addr=(?<"TokenAddress">) rank=\\w+"
					 newEventHandler:
		 ^EFEvent *(NSTextCheckingResult *matchResult, EFLogEvent *logEvent, BOOL *isComplete) {
			 *isComplete = YES;
			 EFNetworkControlPathEvent *newEvent = nil;
			 NSString *serviceID = [logEvent substringForCaptureGroup:@TokenServiceID inMatchResult:matchResult];
			 NSString *addressString = [logEvent substringForCaptureGroup:@TokenAddress inMatchResult:matchResult];
			 if (serviceID != nil && addressString != nil) {
				 newEvent = [self createInterfaceEventWithLogEvent:logEvent matchResult:matchResult];
				 if (newEvent != nil) {
					 newEvent.serviceID = serviceID;
					 [self addAddress:addressString toInterfaceEvent:newEvent];
				 }
			 }
			 return newEvent;
		 }],
		[[EFLogEventMatch alloc] initWithPattern:@"network changed:( v4\\((?<"TokenV4Changes">[^\\)]+)\\))?( v6\\((?<"TokenV6Changes">[^\\)]+)\\))?"
				 multipleNewEventHandler:
		 ^NSArray<EFEvent *> *(NSTextCheckingResult *matchResult, EFLogEvent *logEvent) {
			 NSMutableDictionary<NSString *, EFNetworkControlPathEvent *> *newEventsMap = nil;
			 NSArray<NSString *> *tokens = @[ @TokenV4Changes, @TokenV6Changes ];
			 for (NSString *token in tokens) {
				 BOOL isIPv4 = [token isEqualToString:@TokenV4Changes];
				 NSString *changes = [logEvent substringForCaptureGroup:token inMatchResult:matchResult];
				 if (changes == nil) {
					 continue;
				 }
				 NSArray<NSTextCheckingResult *> *matches = [self.netChangeRegex matchesInString:changes options:0 range:NSMakeRange(0, changes.length)];
				 BOOL isPrimary = YES;
				 for (NSTextCheckingResult *match in matches) {
					 NSString *interfaceName = [self substringOfString:changes forCaptureGroup:@TokenInterfaceName inMatchResult:match];
					 NSString *addressAction = [self substringOfString:changes forCaptureGroup:@TokenAddressAction inMatchResult:match];
					 NSString *address = [self substringOfString:changes forCaptureGroup:@TokenAddress inMatchResult:match];

					 if (interfaceName == nil || address.length == 0) {
						 continue;
					 }

					 EFNetworkControlPathEvent *event = newEventsMap[interfaceName];
					 if (event == nil) {
						 event = [self createInterfaceEventWithLogEvent:logEvent interfaceName:interfaceName];
						 if (newEventsMap == nil) {
							 newEventsMap = [[NSMutableDictionary alloc] init];
						 }
						 newEventsMap[interfaceName] = event;
					 }

					 if (addressAction.length > 0 && [addressAction isEqualToString:@"-"]) {
						 if (isPrimary) {
							 if (isIPv4) {
								 event.primaryStateIPv4 = EFPrimaryStateNotPrimary;
							 } else {
								 event.primaryStateIPv6 = EFPrimaryStateNotPrimary;
							 }
						 }
						 [self removeAddress:address fromInterfaceEvent:event];
					 } else {
						 if (isPrimary) {
							 if (isIPv4) {
								 event.primaryStateIPv4 = EFPrimaryStatePrimary;
							 } else {
								 event.primaryStateIPv6 = EFPrimaryStatePrimary;
							 }
							 for (NSString *otherInterfaceName in SCLogParser.interfaceMap) {
								 if ([otherInterfaceName isEqualToString:interfaceName]) {
									 continue;
								 }
								 EFNetworkControlPathEvent *otherEvent = newEventsMap[otherInterfaceName];
								 if (otherEvent == nil) {
									 otherEvent = [self createInterfaceEventWithLogEvent:logEvent interfaceName:otherInterfaceName];
									 if (newEventsMap == nil) {
										 newEventsMap = [[NSMutableDictionary alloc] init];
									 }
									 newEventsMap[otherInterfaceName] = otherEvent;
								 }
								 if (isIPv4) {
									 otherEvent.primaryStateIPv4 = EFPrimaryStateNotPrimary;
								 } else {
									 otherEvent.primaryStateIPv6 = EFPrimaryStateNotPrimary;
								 }
							 }
							 isPrimary = NO;
						 }
						 [self addAddress:address toInterfaceEvent:event];
					 }
				 }
			 }
			 return newEventsMap.allValues;
		 }],
	];
	EFLogEventParser *parser = [[EFLogEventParser alloc] initWithMatches:matches];
	return [super initWithCategory:@"IPMonitor" eventParser:parser];
}

@end