keychain/TrustedPeersHelper/main.swift

   1 /*
   2  * Copyright (c) 2018 Apple Inc. All Rights Reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23
  24 import Foundation
  25 import os.log
  26
  27 let containerMap = ContainerMap(invocableCreator: CKCodeCuttlefishInvocableCreator())
  28
  29 class ServiceDelegate: NSObject, NSXPCListenerDelegate {
  30     func listener(_ listener: NSXPCListener, shouldAcceptNewConnection newConnection: NSXPCConnection) -> Bool {
  31         let tphEntitlement = "com.apple.private.trustedpeershelper.client"
  32
  33         os_log("Received a new client: %@", log: tplogDebug, type: .default, newConnection)
  34         switch newConnection.value(forEntitlement: tphEntitlement) {
  35         case 1 as Int:
  36             os_log("client has entitlement '%@'", log: tplogDebug, type: .default, tphEntitlement)
  37         case true as Bool:
  38             os_log("client has entitlement '%@'", log: tplogDebug, type: .default, tphEntitlement)
  39
  40         case let someInt as Int:
  41             os_log("client(%@) has wrong integer value for '%@' (%d), rejecting", log: tplogDebug, type: .default, newConnection, tphEntitlement, someInt)
  42             return false
  43
  44         case let someBool as Bool:
  45             os_log("client(%@) has wrong boolean value for '%@' (%d), rejecting", log: tplogDebug, type: .default, newConnection, tphEntitlement, someBool)
  46             return false
  47
  48         default:
  49             os_log("client(%@) is missing entitlement '%@', rejecting", log: tplogDebug, type: .default, newConnection, tphEntitlement)
  50             return false
  51         }
  52
  53         newConnection.exportedInterface = TrustedPeersHelperSetupProtocol(NSXPCInterface(with: TrustedPeersHelperProtocol.self))
  54         let exportedObject = Client(endpoint: newConnection.endpoint, containerMap: containerMap)
  55         newConnection.exportedObject = exportedObject
  56         newConnection.resume()
  57
  58         return true
  59     }
  60 }
  61
  62 os_log("Starting up", log: tplogDebug, type: .default)
  63
  64 ValueTransformer.setValueTransformer(SetValueTransformer(), forName: SetValueTransformer.name)
  65
  66 let delegate = ServiceDelegate()
  67 let listener = NSXPCListener.service()
  68 listener.delegate = delegate
  69 listener.resume()