boulder/cmd/boulder-publisher/main.go

115 lines
3.2 KiB
Go

package main
import (
"flag"
"os"
"runtime"
ct "github.com/google/certificate-transparency-go"
"github.com/honeycombio/beeline-go"
"google.golang.org/grpc/health"
healthpb "google.golang.org/grpc/health/grpc_health_v1"
"github.com/letsencrypt/boulder/cmd"
"github.com/letsencrypt/boulder/features"
bgrpc "github.com/letsencrypt/boulder/grpc"
"github.com/letsencrypt/boulder/issuance"
"github.com/letsencrypt/boulder/publisher"
pubpb "github.com/letsencrypt/boulder/publisher/proto"
)
type config struct {
Publisher struct {
cmd.ServiceConfig
Features map[string]bool
// If this is non-zero, profile blocking events such that one even is
// sampled every N nanoseconds.
// https://golang.org/pkg/runtime/#SetBlockProfileRate
BlockProfileRate int
UserAgent string
// Chains is a list of lists of certificate filenames. Each inner list is
// a chain, starting with the issuing intermediate, followed by one or
// more additional certificates, up to and including a root.
Chains [][]string
}
Syslog cmd.SyslogConfig
Beeline cmd.BeelineConfig
}
func main() {
grpcAddr := flag.String("addr", "", "gRPC listen address override")
debugAddr := flag.String("debug-addr", "", "Debug server address override")
configFile := flag.String("config", "", "File path to the configuration file for this service")
flag.Parse()
if *configFile == "" {
flag.Usage()
os.Exit(1)
}
var c config
err := cmd.ReadConfigFile(*configFile, &c)
cmd.FailOnError(err, "Reading JSON config file into config structure")
err = features.Set(c.Publisher.Features)
cmd.FailOnError(err, "Failed to set feature flags")
runtime.SetBlockProfileRate(c.Publisher.BlockProfileRate)
if *grpcAddr != "" {
c.Publisher.GRPC.Address = *grpcAddr
}
if *debugAddr != "" {
c.Publisher.DebugAddr = *debugAddr
}
if c.Publisher.UserAgent == "" {
c.Publisher.UserAgent = "certificate-transparency-go/1.0"
}
bc, err := c.Beeline.Load()
cmd.FailOnError(err, "Failed to load Beeline config")
beeline.Init(bc)
defer beeline.Close()
scope, logger := cmd.StatsAndLogging(c.Syslog, c.Publisher.DebugAddr)
defer logger.AuditPanic()
logger.Info(cmd.VersionString())
if c.Publisher.Chains == nil {
logger.AuditErr("No chain files provided")
os.Exit(1)
}
bundles := make(map[issuance.IssuerNameID][]ct.ASN1Cert)
for _, files := range c.Publisher.Chains {
chain, err := issuance.LoadChain(files)
cmd.FailOnError(err, "failed to load chain.")
issuer := chain[0]
id := issuer.NameID()
bundles[id] = publisher.GetCTBundleForChain(chain)
}
tlsConfig, err := c.Publisher.TLS.Load()
cmd.FailOnError(err, "TLS config")
clk := cmd.Clock()
pubi := publisher.New(bundles, c.Publisher.UserAgent, logger, scope)
serverMetrics := bgrpc.NewServerMetrics(scope)
grpcSrv, l, err := bgrpc.NewServer(c.Publisher.GRPC, tlsConfig, serverMetrics, clk)
cmd.FailOnError(err, "Unable to setup Publisher gRPC server")
pubpb.RegisterPublisherServer(grpcSrv, pubi)
hs := health.NewServer()
healthpb.RegisterHealthServer(grpcSrv, hs)
go cmd.CatchSignals(logger, func() {
hs.Shutdown()
grpcSrv.GracefulStop()
})
err = cmd.FilterShutdownErrors(grpcSrv.Serve(l))
cmd.FailOnError(err, "Publisher gRPC service failed")
}