123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204 |
- // Copyright (c) 2012-2014 Jeremy Latt
- // Copyright (c) 2014-2015 Edmund Huber
- // Copyright (c) 2016-2017 Daniel Oaks <daniel@danieloaks.net>
- // released under the MIT license
-
- package main
-
- import (
- "bufio"
- "fmt"
- "log"
- "os"
- "strings"
- "syscall"
-
- "github.com/docopt/docopt-go"
- "github.com/oragono/oragono/irc"
- "github.com/oragono/oragono/irc/logger"
- "github.com/oragono/oragono/irc/mkcerts"
- "github.com/oragono/oragono/irc/utils"
- "golang.org/x/crypto/bcrypt"
- "golang.org/x/crypto/ssh/terminal"
- )
-
- var commit = ""
-
- // get a password from stdin from the user
- func getPassword() string {
- fd := int(os.Stdin.Fd())
- if terminal.IsTerminal(fd) {
- bytePassword, err := terminal.ReadPassword(int(syscall.Stdin))
- if err != nil {
- log.Fatal("Error reading password:", err.Error())
- }
- return string(bytePassword)
- }
- reader := bufio.NewReader(os.Stdin)
- text, _ := reader.ReadString('\n')
- return strings.TrimSpace(text)
- }
-
- func fileDoesNotExist(file string) bool {
- if _, err := os.Stat(file); os.IsNotExist(err) {
- return true
- }
- return false
- }
-
- // implements the `oragono mkcerts` command
- func doMkcerts(configFile string, quiet bool) {
- config, err := irc.LoadRawConfig(configFile)
- if err != nil {
- log.Fatal(err)
- }
- if !quiet {
- log.Println("making self-signed certificates")
- }
-
- certToKey := make(map[string]string)
- for name, conf := range config.Server.Listeners {
- if conf.TLS.Cert == "" {
- continue
- }
- existingKey, ok := certToKey[conf.TLS.Cert]
- if ok {
- if existingKey == conf.TLS.Key {
- continue
- } else {
- log.Fatal("Conflicting TLS key files for ", conf.TLS.Cert)
- }
- }
- if !quiet {
- log.Printf(" making cert for %s listener\n", name)
- }
- host := config.Server.Name
- cert, key := conf.TLS.Cert, conf.TLS.Key
- if !(fileDoesNotExist(cert) && fileDoesNotExist(key)) {
- log.Fatalf("Preexisting TLS cert and/or key files: %s %s", cert, key)
- }
- err := mkcerts.CreateCert("Oragono", host, cert, key)
- if err == nil {
- if !quiet {
- log.Printf(" Certificate created at %s : %s\n", cert, key)
- }
- certToKey[cert] = key
- } else {
- log.Fatal(" Could not create certificate:", err.Error())
- }
- }
- }
-
- func main() {
- version := irc.SemVer
- usage := `oragono.
- Usage:
- oragono initdb [--conf <filename>] [--quiet]
- oragono upgradedb [--conf <filename>] [--quiet]
- oragono genpasswd [--conf <filename>] [--quiet]
- oragono mkcerts [--conf <filename>] [--quiet]
- oragono mksecret [--conf <filename>] [--quiet]
- oragono run [--conf <filename>] [--quiet] [--smoke]
- oragono -h | --help
- oragono --version
- Options:
- --conf <filename> Configuration file to use [default: ircd.yaml].
- --quiet Don't show startup/shutdown lines.
- -h --help Show this screen.
- --version Show version.`
-
- arguments, _ := docopt.ParseArgs(usage, nil, version)
-
- // don't require a config file for genpasswd or mksecret
- if arguments["genpasswd"].(bool) {
- var password string
- fd := int(os.Stdin.Fd())
- if terminal.IsTerminal(fd) {
- fmt.Print("Enter Password: ")
- password = getPassword()
- fmt.Print("\n")
- fmt.Print("Reenter Password: ")
- confirm := getPassword()
- fmt.Print("\n")
- if confirm != password {
- log.Fatal("passwords do not match")
- }
- } else {
- password = getPassword()
- }
- hash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.MinCost)
- if err != nil {
- log.Fatal("encoding error:", err.Error())
- }
- fmt.Print(string(hash))
- if terminal.IsTerminal(fd) {
- fmt.Println()
- }
- return
- } else if arguments["mksecret"].(bool) {
- fmt.Println(utils.GenerateSecretKey())
- return
- } else if arguments["mkcerts"].(bool) {
- doMkcerts(arguments["--conf"].(string), arguments["--quiet"].(bool))
- return
- }
-
- configfile := arguments["--conf"].(string)
- config, err := irc.LoadConfig(configfile)
- if err != nil && !(err == irc.ErrInvalidCertKeyPair && arguments["mkcerts"].(bool)) {
- log.Fatal("Config file did not load successfully: ", err.Error())
- }
-
- logman, err := logger.NewManager(config.Logging)
- if err != nil {
- log.Fatal("Logger did not load successfully:", err.Error())
- }
-
- if arguments["initdb"].(bool) {
- irc.InitDB(config.Datastore.Path)
- if !arguments["--quiet"].(bool) {
- log.Println("database initialized: ", config.Datastore.Path)
- }
- } else if arguments["upgradedb"].(bool) {
- err = irc.UpgradeDB(config)
- if err != nil {
- log.Fatal("Error while upgrading db:", err.Error())
- }
- if !arguments["--quiet"].(bool) {
- log.Println("database upgraded: ", config.Datastore.Path)
- }
- } else if arguments["run"].(bool) {
- if !arguments["--quiet"].(bool) {
- logman.Info("server", fmt.Sprintf("Oragono v%s starting", irc.SemVer))
- if commit == "" {
- logman.Debug("server", fmt.Sprintf("Could not get current commit"))
- } else {
- logman.Info("server", fmt.Sprintf("Running commit %s", commit))
- }
- }
-
- // set current git commit
- irc.Commit = commit
- if commit != "" {
- irc.Ver = fmt.Sprintf("%s-%s", irc.Ver, commit)
- }
-
- // warning if running a non-final version
- if strings.Contains(irc.SemVer, "unreleased") {
- logman.Warning("server", "You are currently running an unreleased beta version of Oragono that may be unstable and could corrupt your database.\nIf you are running a production network, please download the latest build from https://oragono.io/downloads.html and run that instead.")
- }
-
- server, err := irc.NewServer(config, logman)
- if err != nil {
- logman.Error("server", fmt.Sprintf("Could not load server: %s", err.Error()))
- os.Exit(1)
- }
- if !arguments["--quiet"].(bool) {
- logman.Info("server", "Server running")
- defer logman.Info("server", fmt.Sprintf("Oragono v%s exiting", irc.SemVer))
- }
- if !arguments["--smoke"].(bool) {
- server.Run()
- }
- }
- }
|