Initial commit

Source/santad/SNTDatabaseController.m

@@ -0,0 +1,82 @@
+/// 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