mirror of
https://github.com/google/santa.git
synced 2026-01-15 01:08:12 -05:00
83 lines
2.7 KiB
Objective-C
83 lines
2.7 KiB
Objective-C
/// Copyright 2014 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 "SNTDatabaseController.h"
|
|
|
|
#import "SNTEventTable.h"
|
|
#import "SNTLogging.h"
|
|
#import "SNTRuleTable.h"
|
|
|
|
@implementation SNTDatabaseController
|
|
|
|
static NSString * const kDatabasePath = @"/var/db/santa";
|
|
static NSString * const kRulesDatabaseName = @"rules.db";
|
|
static NSString * const kEventsDatabaseName = @"events.db";
|
|
|
|
+ (SNTEventTable *)eventTable {
|
|
static FMDatabaseQueue *eventDatabaseQueue = nil;
|
|
static dispatch_once_t eventDatabaseToken;
|
|
dispatch_once(&eventDatabaseToken, ^{
|
|
[self createDatabasePath];
|
|
NSString *fullPath = [kDatabasePath stringByAppendingPathComponent:kEventsDatabaseName];
|
|
eventDatabaseQueue = [[FMDatabaseQueue alloc] initWithPath:fullPath];
|
|
|
|
#ifndef DEBUG
|
|
[eventDatabaseQueue inDatabase:^(FMDatabase *db) {
|
|
db.logsErrors = NO;
|
|
}];
|
|
#endif
|
|
});
|
|
|
|
return [[SNTEventTable alloc] initWithDatabaseQueue:eventDatabaseQueue];
|
|
}
|
|
|
|
+ (SNTRuleTable *)ruleTable {
|
|
static FMDatabaseQueue *ruleDatabaseQueue = nil;
|
|
static dispatch_once_t ruleDatabaseToken;
|
|
dispatch_once(&ruleDatabaseToken, ^{
|
|
[self createDatabasePath];
|
|
NSString *fullPath = [kDatabasePath stringByAppendingPathComponent:kRulesDatabaseName];
|
|
ruleDatabaseQueue = [[FMDatabaseQueue alloc] initWithPath:fullPath];
|
|
|
|
#ifndef DEBUG
|
|
[ruleDatabaseQueue inDatabase:^(FMDatabase *db) {
|
|
db.logsErrors = NO;
|
|
}];
|
|
#endif
|
|
});
|
|
return [[SNTRuleTable alloc] initWithDatabaseQueue:ruleDatabaseQueue];
|
|
}
|
|
|
|
#pragma mark - Private
|
|
|
|
/// Create the folder that contains the databases
|
|
+ (void)createDatabasePath {
|
|
NSFileManager *fm = [NSFileManager defaultManager];
|
|
|
|
NSDictionary *attrs = @{ NSFileOwnerAccountName: @"root",
|
|
NSFileGroupOwnerAccountName: @"wheel",
|
|
NSFilePosixPermissions: @0755 };
|
|
|
|
if (![fm fileExistsAtPath:kDatabasePath]) {
|
|
[fm createDirectoryAtPath:kDatabasePath
|
|
withIntermediateDirectories:YES
|
|
attributes:attrs
|
|
error:nil];
|
|
} else {
|
|
[fm setAttributes:attrs ofItemAtPath:kDatabasePath error:nil];
|
|
}
|
|
}
|
|
|
|
@end
|