/*

    Copyright (C) 2016 Apple Inc. All Rights Reserved.

    See LICENSE.txt for this sample’s licensing information

    Abstract:

    This file contains the main code for the SimpleTunnel server.

*/

import Foundation

/// Dispatch source to catch and handle SIGINT

let interruptSignalSource = dispatch_source_create(DISPATCH_SOURCE_TYPE_SIGNAL, UInt(SIGINT), 0, dispatch_get_main_queue())

/// Dispatch source to catch and handle SIGTERM

let termSignalSource = dispatch_source_create(DISPATCH_SOURCE_TYPE_SIGNAL, UInt(SIGTERM), 0, dispatch_get_main_queue())

/// Basic sanity check of the parameters.

if Process.arguments.count < 3 {

    print("Usage: \(Process.arguments[0]) <port> <config-file>")

    exit(1)

}

func ignore(_: Int32)  {

}

signal(SIGTERM, ignore)

signal(SIGINT, ignore)

let portString = Process.arguments[1]

let configurationPath = Process.arguments[2]

let networkService: NSNetService

// Initialize the server.

if !ServerTunnel.initializeWithConfigurationFile(configurationPath) {

    exit(1)

}

if let portNumber = Int(portString)  {

    networkService = ServerTunnel.startListeningOnPort(Int32(portNumber))

}

else {

    print("Invalid port: \(portString)")

    exit(1)

}

// Set up signal handling.

dispatch_source_set_event_handler(interruptSignalSource) {

    networkService.stop()

    return

}

dispatch_resume(interruptSignalSource)

dispatch_source_set_event_handler(termSignalSource) {

    networkService.stop()

    return

}

dispatch_resume(termSignalSource)

NSRunLoop.mainRunLoop().run()