mirror of
https://github.com/FrankerFaceZ/FrankerFaceZ.git
synced 2025-07-04 10:08:31 +00:00
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package main // import "bitbucket.org/stendec/frankerfacez/socketserver/cmd/ffzsocketserver"
|
|
|
|
import (
|
|
"../../server"
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
)
|
|
|
|
import _ "net/http/pprof"
|
|
|
|
var configFilename = flag.String("config", "config.json", "Configuration file, including the keypairs for the NaCl crypto library, for communicating with the backend.")
|
|
var flagGenerateKeys = flag.Bool("genkeys", false, "Generate NaCl keys instead of serving requests.\nArguments: [int serverId] [base64 backendPublic]\nThe backend public key can either be specified in base64 on the command line, or put in the json file later.")
|
|
|
|
var BuildTime string = "build not stamped"
|
|
var BuildHash string = "build not stamped"
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
if *flagGenerateKeys {
|
|
generateKeys(*configFilename)
|
|
return
|
|
}
|
|
|
|
confFile, err := os.Open(*configFilename)
|
|
if os.IsNotExist(err) {
|
|
fmt.Println("Error: No config file. Run with -genkeys and edit config.json")
|
|
os.Exit(3)
|
|
}
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
conf := &server.ConfigFile{}
|
|
confBytes, err := ioutil.ReadAll(confFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
err = json.Unmarshal(confBytes, &conf)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// logFile, err := os.OpenFile("output.log", os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
|
|
// if err != nil {
|
|
// log.Fatal("Could not create logfile: ", err)
|
|
// }
|
|
|
|
server.SetupServerAndHandle(conf, http.DefaultServeMux)
|
|
server.SetBuildStamp(BuildTime, BuildHash)
|
|
|
|
go commandLineConsole()
|
|
|
|
if conf.UseSSL {
|
|
go func() {
|
|
if err := http.ListenAndServeTLS(conf.SSLListenAddr, conf.SSLCertificateFile, conf.SSLKeyFile, http.DefaultServeMux); err != nil {
|
|
log.Fatal("ListenAndServeTLS: ", err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
if err = http.ListenAndServe(conf.ListenAddr, http.DefaultServeMux); err != nil {
|
|
log.Fatal("ListenAndServe: ", err)
|
|
}
|
|
}
|
|
|
|
func generateKeys(outputFile string) {
|
|
if flag.NArg() < 1 {
|
|
fmt.Println("Specify a numeric server ID after -genkeys")
|
|
os.Exit(2)
|
|
}
|
|
if flag.NArg() >= 2 {
|
|
server.GenerateKeys(outputFile, flag.Arg(0), flag.Arg(1))
|
|
} else {
|
|
server.GenerateKeys(outputFile, flag.Arg(0), "")
|
|
}
|
|
fmt.Println("Keys generated. Now edit config.json")
|
|
}
|