## Nim-Codex ## Copyright (c) 2021 Status Research & Development GmbH ## Licensed under either of ## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE)) ## * MIT license ([LICENSE-MIT](LICENSE-MIT)) ## at your option. ## This file may not be copied, modified, or distributed except according to ## those terms. import pkg/chronos import pkg/questionable import pkg/confutils import pkg/confutils/defs import pkg/confutils/std/net import pkg/confutils/toml/defs as confTomlDefs import pkg/confutils/toml/std/net as confTomlNet import pkg/confutils/toml/std/uri as confTomlUri import pkg/toml_serialization import pkg/libp2p import ./codex/conf import ./codex/codex import ./codex/logutils import ./codex/units import ./codex/utils/keyutils import ./codex/codextypes export codex, conf, libp2p, chronos, logutils when isMainModule: import std/sequtils import std/os import pkg/confutils/defs import ./codex/utils/fileutils logScope: topics = "codex" when defined(posix): import system/ansi_c type CodexStatus {.pure.} = enum Stopped, Stopping, Running let config = CodexConf.load( version = codexFullVersion, envVarsPrefix = "codex", secondarySources = proc (config: CodexConf, sources: auto) = if configFile =? config.configFile: sources.addConfigFile(Toml, configFile) ) config.setupLogging() config.setupMetrics() if config.nat == ValidIpAddress.init(IPv4_any()): error "`--nat` cannot be set to the any (`0.0.0.0`) address" quit QuitFailure if config.nat == ValidIpAddress.init("127.0.0.1"): warn "`--nat` is set to loopback, your node wont properly announce over the DHT" if not(checkAndCreateDataDir((config.dataDir).string)): # We are unable to access/create data folder or data folder's # permissions are insecure. quit QuitFailure if config.prover() and not(checkAndCreateDataDir((config.circuitDir).string)): quit QuitFailure trace "Data dir initialized", dir = $config.dataDir if not(checkAndCreateDataDir((config.dataDir / "repo"))): # We are unable to access/create data folder or data folder's # permissions are insecure. quit QuitFailure trace "Repo dir initialized", dir = config.dataDir / "repo" var state: CodexStatus shutdown: Future[void] let keyPath = if isAbsolute(config.netPrivKeyFile): config.netPrivKeyFile else: config.dataDir / config.netPrivKeyFile privateKey = setupKey(keyPath).expect("Should setup private key!") server = try: CodexServer.new(config, privateKey) except Exception as exc: error "Failed to start Codex", msg = exc.msg quit QuitFailure ## Ctrl+C handling proc doShutdown() = shutdown = server.stop() state = CodexStatus.Stopping notice "Stopping Codex" proc controlCHandler() {.noconv.} = when defined(windows): # workaround for https://github.com/nim-lang/Nim/issues/4057 try: setupForeignThreadGc() except Exception as exc: raiseAssert exc.msg # shouldn't happen notice "Shutting down after having received SIGINT" doShutdown() try: setControlCHook(controlCHandler) except Exception as exc: # TODO Exception warn "Cannot set ctrl-c handler", msg = exc.msg # equivalent SIGTERM handler when defined(posix): proc SIGTERMHandler(signal: cint) {.noconv.} = notice "Shutting down after having received SIGTERM" doShutdown() c_signal(ansi_c.SIGTERM, SIGTERMHandler) try: waitFor server.start() except CatchableError as error: error "Codex failed to start", error = error.msg # XXX ideally we'd like to issue a stop instead of quitting cold turkey, # but this would mean we'd have to fix the implementation of all # services so they won't crash if we attempt to stop them before they # had a chance to start (currently you'll get a SISGSEV if you try to). quit QuitFailure state = CodexStatus.Running while state == CodexStatus.Running: try: # poll chronos chronos.poll() except Exception as exc: error "Unhandled exception in async proc, aborting", msg = exc.msg quit QuitFailure try: # signal handlers guarantee that the shutdown Future will # be assigned before state switches to Stopping waitFor shutdown except CatchableError as error: error "Codex didn't shutdown correctly", error = error.msg quit QuitFailure notice "Exited codex"