mirror of
https://github.com/google/santa.git
synced 2026-01-15 01:08:12 -05:00
dc692c8256
Add helper to make declaring unit tests easier Add unit_tests test_suite containing all unit tests Fix reload rule Update to workspace-relative header locations that were missed before
119 lines
3.5 KiB
Objective-C
119 lines
3.5 KiB
Objective-C
/// Copyright 2015 Google Inc. All rights reserved.
|
|
///
|
|
/// Licensed under the Apache License, Version 2.0 (the "License");
|
|
/// you may not use this file except in compliance with the License.
|
|
/// You may obtain a copy of the License at
|
|
///
|
|
/// http://www.apache.org/licenses/LICENSE-2.0
|
|
///
|
|
/// Unless required by applicable law or agreed to in writing, software
|
|
/// distributed under the License is distributed on an "AS IS" BASIS,
|
|
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
/// See the License for the specific language governing permissions and
|
|
/// limitations under the License.
|
|
|
|
#import "Source/common/SNTRule.h"
|
|
|
|
@interface SNTRule()
|
|
@property(readwrite) NSUInteger timestamp;
|
|
@end
|
|
|
|
@implementation SNTRule
|
|
|
|
- (instancetype)initWithShasum:(NSString *)shasum
|
|
state:(SNTRuleState)state
|
|
type:(SNTRuleType)type
|
|
customMsg:(NSString *)customMsg
|
|
timestamp:(NSUInteger)timestamp {
|
|
self = [super init];
|
|
if (self) {
|
|
_shasum = shasum;
|
|
_state = state;
|
|
_type = type;
|
|
_customMsg = customMsg;
|
|
_timestamp = timestamp;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithShasum:(NSString *)shasum
|
|
state:(SNTRuleState)state
|
|
type:(SNTRuleType)type
|
|
customMsg:(NSString *)customMsg {
|
|
self = [self initWithShasum:shasum
|
|
state:state
|
|
type:type
|
|
customMsg:customMsg
|
|
timestamp:0];
|
|
// Initialize timestamp to current time if rule is transitive.
|
|
if (self && state == SNTRuleStateWhitelistTransitive) {
|
|
[self resetTimestamp];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
|
|
#pragma mark NSSecureCoding
|
|
|
|
#pragma clang diagnostic push
|
|
#pragma clang diagnostic ignored "-Wobjc-literal-conversion"
|
|
#define ENCODE(obj, key) if (obj) [coder encodeObject:obj forKey:key]
|
|
#define DECODE(cls, key) [decoder decodeObjectOfClass:[cls class] forKey:key]
|
|
|
|
+ (BOOL)supportsSecureCoding {
|
|
return YES;
|
|
}
|
|
|
|
- (void)encodeWithCoder:(NSCoder *)coder {
|
|
ENCODE(self.shasum, @"shasum");
|
|
ENCODE(@(self.state), @"state");
|
|
ENCODE(@(self.type), @"type");
|
|
ENCODE(self.customMsg, @"custommsg");
|
|
ENCODE(@(self.timestamp), @"timestamp");
|
|
}
|
|
|
|
- (instancetype)initWithCoder:(NSCoder *)decoder {
|
|
self = [super init];
|
|
if (self) {
|
|
_shasum = DECODE(NSString, @"shasum");
|
|
_state = [DECODE(NSNumber, @"state") intValue];
|
|
_type = [DECODE(NSNumber, @"type") intValue];
|
|
_customMsg = DECODE(NSString, @"custommsg");
|
|
_timestamp = [DECODE(NSNumber, @"timestamp") unsignedIntegerValue];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
#undef DECODE
|
|
#undef ENCODE
|
|
#pragma clang diagnostic pop
|
|
|
|
- (BOOL)isEqual:(id)other {
|
|
if (other == self) return YES;
|
|
if (![other isKindOfClass:[SNTRule class]]) return NO;
|
|
SNTRule *o = other;
|
|
return ([self.shasum isEqual:o.shasum] && self.state == o.state && self.type == o.type);
|
|
}
|
|
|
|
- (NSUInteger)hash {
|
|
NSUInteger prime = 31;
|
|
NSUInteger result = 1;
|
|
result = prime * result + [self.shasum hash];
|
|
result = prime * result + self.state;
|
|
result = prime * result + self.type;
|
|
return result;
|
|
}
|
|
|
|
- (NSString *)description {
|
|
return [NSString stringWithFormat:@"SNTRule: SHA-256: %@, State: %ld, Type: %ld, Timestamp: %lu",
|
|
self.shasum, self.state, self.type, (unsigned long)self.timestamp];
|
|
}
|
|
|
|
# pragma mark Last-access Timestamp
|
|
|
|
- (void)resetTimestamp {
|
|
self.timestamp = (NSUInteger)[[NSDate date] timeIntervalSinceReferenceDate];
|
|
}
|
|
|
|
@end
|