mirror of
https://github.com/google/santa.git
synced 2026-01-14 08:47:57 -05:00
1515929752
* Allow per-policy and per-rule FAA URL and button text * Add format string support to the custom URL. Added SNTBlockMessageTest. * Add event URL to TTY message. * Allow rule specific policy to "clear" global to remove buttons for the rule * Remove extra beta label for FAA
96 lines
3.5 KiB
Objective-C
96 lines
3.5 KiB
Objective-C
/// Copyright 2023 Google LLC
|
|
///
|
|
/// 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
|
|
///
|
|
/// https://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 <OCMock/OCMock.h>
|
|
#import <XCTest/XCTest.h>
|
|
|
|
#import "Source/common/SNTBlockMessage.h"
|
|
#import "Source/common/SNTConfigurator.h"
|
|
#include "Source/common/SNTFileAccessEvent.h"
|
|
#include "Source/common/SNTStoredEvent.h"
|
|
#import "Source/common/SNTSystemInfo.h"
|
|
|
|
@interface SNTBlockMessageTest : XCTestCase
|
|
@property id mockConfigurator;
|
|
@property id mockSystemInfo;
|
|
@end
|
|
|
|
@implementation SNTBlockMessageTest
|
|
|
|
- (void)setUp {
|
|
self.mockConfigurator = OCMClassMock([SNTConfigurator class]);
|
|
OCMStub([self.mockConfigurator configurator]).andReturn(self.mockConfigurator);
|
|
OCMStub([self.mockConfigurator machineID]).andReturn(@"my_mid");
|
|
|
|
self.mockSystemInfo = OCMClassMock([SNTSystemInfo class]);
|
|
OCMStub([self.mockSystemInfo longHostname]).andReturn(@"my_hn");
|
|
OCMStub([self.mockSystemInfo hardwareUUID]).andReturn(@"my_u");
|
|
OCMStub([self.mockSystemInfo serialNumber]).andReturn(@"my_s");
|
|
}
|
|
|
|
- (void)testEventDetailURLForEvent {
|
|
SNTStoredEvent *se = [[SNTStoredEvent alloc] init];
|
|
|
|
se.fileSHA256 = @"my_fi";
|
|
se.executingUser = @"my_un";
|
|
|
|
NSString *url = @"http://"
|
|
@"localhost?fs=%file_sha%&fi=%file_identifier%&bfi=%bundle_or_file_identifier%&"
|
|
@"un=%username%&mid=%machine_id%&hn=%hostname%&u=%uuid%&s=%serial%";
|
|
NSString *wantUrl =
|
|
@"http://"
|
|
@"localhost?fs=my_fi&fi=my_fi&bfi=my_fi&bfi=my_fi&un=my_un&mid=my_mid&hn=my_hn&u=my_u&s=my_s";
|
|
|
|
NSURL *gotUrl = [SNTBlockMessage eventDetailURLForEvent:se customURL:url];
|
|
|
|
// Set fileBundleHash and test again for newly expected values
|
|
se.fileBundleHash = @"my_fbh";
|
|
|
|
wantUrl = @"http://"
|
|
@"localhost?fs=my_fbh&fi=my_fi&bfi=my_fbh&un=my_un&mid=my_mid&hn=my_hn&u=my_u&s=my_s";
|
|
|
|
gotUrl = [SNTBlockMessage eventDetailURLForEvent:se customURL:url];
|
|
|
|
XCTAssertEqualObjects(gotUrl.absoluteString, wantUrl);
|
|
|
|
XCTAssertNil([SNTBlockMessage eventDetailURLForEvent:se customURL:nil]);
|
|
XCTAssertNil([SNTBlockMessage eventDetailURLForEvent:se customURL:@"null"]);
|
|
}
|
|
|
|
- (void)testEventDetailURLForFileAccessEvent {
|
|
SNTFileAccessEvent *fae = [[SNTFileAccessEvent alloc] init];
|
|
|
|
fae.ruleVersion = @"my_rv";
|
|
fae.ruleName = @"my_rn";
|
|
fae.fileSHA256 = @"my_fi";
|
|
fae.accessedPath = @"my_ap";
|
|
fae.executingUser = @"my_un";
|
|
|
|
NSString *url = @"http://"
|
|
@"localhost?rv=%rule_version%&rn=%rule_name%&fi=%file_identifier%&ap=%accessed_"
|
|
@"path%&un=%username%&mid=%machine_id%&hn=%hostname%&u=%uuid%&s=%serial%";
|
|
NSString *wantUrl =
|
|
@"http://"
|
|
@"localhost?rv=my_rv&rn=my_rn&fi=my_fi&ap=my_ap&un=my_un&mid=my_mid&hn=my_hn&u=my_u&s=my_s";
|
|
|
|
NSURL *gotUrl = [SNTBlockMessage eventDetailURLForFileAccessEvent:fae customURL:url];
|
|
|
|
XCTAssertEqualObjects(gotUrl.absoluteString, wantUrl);
|
|
|
|
XCTAssertNil([SNTBlockMessage eventDetailURLForFileAccessEvent:fae customURL:nil]);
|
|
XCTAssertNil([SNTBlockMessage eventDetailURLForFileAccessEvent:fae customURL:@"null"]);
|
|
}
|
|
|
|
@end
|