149 lines
4.0 KiB
Nim
149 lines
4.0 KiB
Nim
## 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/chronicles
|
|
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/units
|
|
import ./codex/utils/keyutils
|
|
import ./codex/loopmeasure
|
|
|
|
export codex, conf, libp2p, chronos, chronicles
|
|
|
|
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()
|
|
|
|
case config.cmd:
|
|
of StartUpCommand.noCommand:
|
|
|
|
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
|
|
|
|
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
|
|
pendingFuts: seq[Future[void]]
|
|
|
|
let
|
|
keyPath =
|
|
if isAbsolute(config.netPrivKeyFile):
|
|
config.netPrivKeyFile
|
|
else:
|
|
config.dataDir / config.netPrivKeyFile
|
|
|
|
privateKey = setupKey(keyPath).expect("Should setup private key!")
|
|
loopMeasure = LoopMeasure.new()
|
|
server = CodexServer.new(config, privateKey, loopMeasure)
|
|
|
|
## Ctrl+C handling
|
|
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"
|
|
|
|
pendingFuts.add(server.stop())
|
|
state = CodexStatus.Stopping
|
|
|
|
notice "Stopping Codex"
|
|
|
|
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"
|
|
|
|
pendingFuts.add(server.stop())
|
|
state = CodexStatus.Stopping
|
|
|
|
notice "Stopping Codex"
|
|
|
|
c_signal(ansi_c.SIGTERM, SIGTERMHandler)
|
|
|
|
pendingFuts.add(server.start())
|
|
|
|
state = CodexStatus.Running
|
|
while state == CodexStatus.Running:
|
|
# poll chronos
|
|
loopMeasure.startMeasure()
|
|
chronos.poll()
|
|
loopMeasure.stopMeasure()
|
|
|
|
# wait fot futures to finish
|
|
let res = waitFor allFinished(pendingFuts)
|
|
state = CodexStatus.Stopped
|
|
|
|
if res.anyIt( it.failed ):
|
|
error "Codex didn't shutdown correctly"
|
|
quit QuitFailure
|
|
|
|
notice "Exited codex"
|
|
|
|
of StartUpCommand.initNode:
|
|
discard
|