func main() { flag.Usage = usage flag.Parse() if debug { go debugServer(DebugAddress) } // when the server starts print the version for debugging and issue logs later logrus.Infof("Version: %s, Git commit: %s", version.NotaryVersion, version.GitCommit) ctx := context.Background() filename := filepath.Base(configFile) ext := filepath.Ext(configFile) configPath := filepath.Dir(configFile) viper.SetConfigType(strings.TrimPrefix(ext, ".")) viper.SetConfigName(strings.TrimSuffix(filename, ext)) viper.AddConfigPath(configPath) // Automatically accept configuration options from the environment viper.SetEnvPrefix("NOTARY_SERVER") viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) viper.AutomaticEnv() err := viper.ReadInConfig() if err != nil { logrus.Error("Viper Error: ", err.Error()) logrus.Error("Could not read config at ", configFile) os.Exit(1) } lvl, err := logrus.ParseLevel(viper.GetString("logging.level")) if err != nil { lvl = logrus.ErrorLevel logrus.Error("Could not parse log level from config. Defaulting to ErrorLevel") } logrus.SetLevel(lvl) // set up bugsnag and attach to logrus bugs := viper.GetString("reporting.bugsnag") if bugs != "" { apiKey := viper.GetString("reporting.bugsnag_api_key") releaseStage := viper.GetString("reporting.bugsnag_release_stage") bugsnag.Configure(bugsnag.Configuration{ APIKey: apiKey, ReleaseStage: releaseStage, }) hook, err := bugsnag_hook.NewBugsnagHook() if err != nil { logrus.Error("Could not attach bugsnag to logrus: ", err.Error()) } else { logrus.AddHook(hook) } } keyAlgo := viper.GetString("trust_service.key_algorithm") if keyAlgo == "" { logrus.Fatal("no key algorithm configured.") os.Exit(1) } ctx = context.WithValue(ctx, "keyAlgorithm", keyAlgo) var trust signed.CryptoService if viper.GetString("trust_service.type") == "remote" { logrus.Info("Using remote signing service") trust = signer.NewNotarySigner( viper.GetString("trust_service.hostname"), viper.GetString("trust_service.port"), viper.GetString("trust_service.tls_ca_file"), ) } else { logrus.Info("Using local signing service") trust = signed.NewEd25519() } if viper.GetString("storage.backend") == "mysql" { logrus.Info("Using mysql backend") dbURL := viper.GetString("storage.db_url") db, err := sql.Open("mysql", dbURL) if err != nil { logrus.Fatal("Error starting DB driver: ", err.Error()) return // not strictly needed but let's be explicit } ctx = context.WithValue(ctx, "metaStore", storage.NewMySQLStorage(db)) } else { logrus.Debug("Using memory backend") ctx = context.WithValue(ctx, "metaStore", storage.NewMemStorage()) } logrus.Info("Starting Server") err = server.Run( ctx, viper.GetString("server.addr"), viper.GetString("server.tls_cert_file"), viper.GetString("server.tls_key_file"), trust, viper.GetString("auth.type"), viper.Get("auth.options"), ) logrus.Error(err.Error()) return }
func main() { flag.Usage = usage flag.Parse() if debug { go debugServer(DebugAddress) } ctx := context.Background() filename := filepath.Base(configFile) ext := filepath.Ext(configFile) configPath := filepath.Dir(configFile) viper.SetConfigType(strings.TrimPrefix(ext, ".")) viper.SetConfigName(strings.TrimSuffix(filename, ext)) viper.AddConfigPath(configPath) err := viper.ReadInConfig() if err != nil { logrus.Error("Viper Error: ", err.Error()) logrus.Error("Could not read config at ", configFile) os.Exit(1) } logrus.SetLevel(logrus.Level(viper.GetInt("logging.level"))) sigHup := make(chan os.Signal) sigTerm := make(chan os.Signal) signal.Notify(sigHup, syscall.SIGHUP) signal.Notify(sigTerm, syscall.SIGTERM) var trust signed.CryptoService if viper.GetString("trust_service.type") == "remote" { logrus.Info("[Notary Server] : Using remote signing service") trust = signer.NewRufusSigner( viper.GetString("trust_service.hostname"), viper.GetString("trust_service.port"), viper.GetString("trust_service.tls_ca_file"), ) } else { logrus.Info("[Notary Server] : Using local signing service") trust = signed.NewEd25519() } if viper.GetString("store.backend") == "mysql" { dbURL := viper.GetString("storage.db_url") db, err := sql.Open("mysql", dbURL) if err != nil { logrus.Fatal("[Notary Server] Error starting DB driver: ", err.Error()) return // not strictly needed but let's be explicit } ctx = context.WithValue(ctx, "metaStore", storage.NewMySQLStorage(db)) } else { ctx = context.WithValue(ctx, "metaStore", storage.NewMemStorage()) } logrus.Info("[Notary Server] Starting Server") err = server.Run( ctx, viper.GetString("server.addr"), viper.GetString("server.tls_cert_file"), viper.GetString("server.tls_key_file"), trust, ) logrus.Error("[Notary Server]", err.Error()) return }
func main() { flag.Usage = usage flag.Parse() if DebugAddress != "" { go debugServer(DebugAddress) } ctx := context.Background() conf, err := parseConfig(configFile) if err != nil { logrus.Fatal("Error parsing config: ", err.Error()) return // not strictly needed but let's be explicit } if conf.Logging.Level > 0 { logrus.SetLevel(logrus.Level(conf.Logging.Level)) } sigHup := make(chan os.Signal) sigTerm := make(chan os.Signal) signal.Notify(sigHup, syscall.SIGHUP) signal.Notify(sigTerm, syscall.SIGTERM) var trust signed.CryptoService if conf.TrustService.Type == "remote" { logrus.Info("[Notary Server] : Using remote signing service") trust = signer.NewRufusSigner(conf.TrustService.Hostname, conf.TrustService.Port, conf.TrustService.TLSCAFile) } else { logrus.Info("[Notary Server] : Using local signing service") trust = signed.NewEd25519() } db, err := sql.Open("mysql", "dockercondemo:dockercondemo@tcp(notarymysql:3306)/dockercondemo") if err != nil { logrus.Fatal("Error starting DB driver: ", err.Error()) return // not strictly needed but let's be explicit } ctx = context.WithValue(ctx, "versionStore", storage.NewMySQLStorage(db)) for { logrus.Info("[Notary Server] Starting Server") childCtx, cancel := context.WithCancel(ctx) go server.Run(childCtx, conf.Server, trust) for { select { // On a sighup we cancel and restart a new server // with updated config case <-sigHup: logrus.Infof("[Notary Server] Server restart requested. Attempting to parse config at %s", configFile) conf, err = parseConfig(configFile) if err != nil { logrus.Infof("[Notary Server] Unable to parse config. Old configuration will keep running. Parse Err: %s", err.Error()) continue } else { cancel() logrus.Info("[Notary Server] Stopping server for restart") break } // On sigkill we cancel and shutdown case <-sigTerm: cancel() logrus.Info("[Notary Server] Shutting Down Hard") os.Exit(0) } } } }