package main import ( "fmt" "io" "os" "path/filepath" "github.com/btcsuite/btcd/connmgr" "github.com/btcsuite/btclog" "github.com/jrick/logrotate/rotator" "github.com/lightninglabs/neutrino" "github.com/lightningnetwork/lightning-onion" "github.com/lightningnetwork/lnd/autopilot" "github.com/lightningnetwork/lnd/build" "github.com/lightningnetwork/lnd/chainntnfs" "github.com/lightningnetwork/lnd/channeldb" "github.com/lightningnetwork/lnd/contractcourt" "github.com/lightningnetwork/lnd/discovery" "github.com/lightningnetwork/lnd/htlcswitch" "github.com/lightningnetwork/lnd/lnrpc/autopilotrpc" "github.com/lightningnetwork/lnd/lnrpc/signrpc" "github.com/lightningnetwork/lnd/lnrpc/walletrpc" "github.com/lightningnetwork/lnd/lnwallet" "github.com/lightningnetwork/lnd/routing" "github.com/lightningnetwork/lnd/signal" "github.com/lightningnetwork/lnd/sweep" ) // Loggers per subsystem. A single backend logger is created and all subsystem // loggers created from it will write to the backend. When adding new // subsystems, add the subsystem logger variable here and to the // subsystemLoggers map. // // Loggers can not be used before the log rotator has been initialized with a // log file. This must be performed early during application startup by // calling initLogRotator. var ( logWriter = &build.LogWriter{} // backendLog is the logging backend used to create all subsystem // loggers. The backend must not be used before the log rotator has // been initialized, or data races and/or nil pointer dereferences will // occur. backendLog = btclog.NewBackend(logWriter) // logRotator is one of the logging outputs. It should be closed on // application shutdown. logRotator *rotator.Rotator ltndLog = build.NewSubLogger("LTND", backendLog.Logger) lnwlLog = build.NewSubLogger("LNWL", backendLog.Logger) peerLog = build.NewSubLogger("PEER", backendLog.Logger) discLog = build.NewSubLogger("DISC", backendLog.Logger) rpcsLog = build.NewSubLogger("RPCS", backendLog.Logger) srvrLog = build.NewSubLogger("SRVR", backendLog.Logger) ntfnLog = build.NewSubLogger("NTFN", backendLog.Logger) chdbLog = build.NewSubLogger("CHDB", backendLog.Logger) fndgLog = build.NewSubLogger("FNDG", backendLog.Logger) hswcLog = build.NewSubLogger("HSWC", backendLog.Logger) utxnLog = build.NewSubLogger("UTXN", backendLog.Logger) brarLog = build.NewSubLogger("BRAR", backendLog.Logger) cmgrLog = build.NewSubLogger("CMGR", backendLog.Logger) crtrLog = build.NewSubLogger("CRTR", backendLog.Logger) btcnLog = build.NewSubLogger("BTCN", backendLog.Logger) atplLog = build.NewSubLogger("ATPL", backendLog.Logger) cnctLog = build.NewSubLogger("CNCT", backendLog.Logger) sphxLog = build.NewSubLogger("SPHX", backendLog.Logger) swprLog = build.NewSubLogger("SWPR", backendLog.Logger) sgnrLog = build.NewSubLogger("SGNR", backendLog.Logger) wlktLog = build.NewSubLogger("WLKT", backendLog.Logger) arpcLog = build.NewSubLogger("ARPC", backendLog.Logger) ) // Initialize package-global logger variables. func init() { lnwallet.UseLogger(lnwlLog) discovery.UseLogger(discLog) chainntnfs.UseLogger(ntfnLog) channeldb.UseLogger(chdbLog) htlcswitch.UseLogger(hswcLog) connmgr.UseLogger(cmgrLog) routing.UseLogger(crtrLog) neutrino.UseLogger(btcnLog) autopilot.UseLogger(atplLog) contractcourt.UseLogger(cnctLog) sphinx.UseLogger(sphxLog) signal.UseLogger(ltndLog) sweep.UseLogger(swprLog) signrpc.UseLogger(sgnrLog) walletrpc.UseLogger(wlktLog) autopilotrpc.UseLogger(arpcLog) } // subsystemLoggers maps each subsystem identifier to its associated logger. var subsystemLoggers = map[string]btclog.Logger{ "LTND": ltndLog, "LNWL": lnwlLog, "PEER": peerLog, "DISC": discLog, "RPCS": rpcsLog, "SRVR": srvrLog, "NTFN": ntfnLog, "CHDB": chdbLog, "FNDG": fndgLog, "HSWC": hswcLog, "UTXN": utxnLog, "BRAR": brarLog, "CMGR": cmgrLog, "CRTR": crtrLog, "BTCN": btcnLog, "ATPL": atplLog, "CNCT": cnctLog, "SPHX": sphxLog, "SWPR": swprLog, "SGNR": sgnrLog, "WLKT": wlktLog, "ARPC": arpcLog, } // initLogRotator initializes the logging rotator to write logs to logFile and // create roll files in the same directory. It must be called before the // package-global log rotator variables are used. func initLogRotator(logFile string, MaxLogFileSize int, MaxLogFiles int) { logDir, _ := filepath.Split(logFile) err := os.MkdirAll(logDir, 0700) if err != nil { fmt.Fprintf(os.Stderr, "failed to create log directory: %v\n", err) os.Exit(1) } r, err := rotator.New(logFile, int64(MaxLogFileSize*1024), false, MaxLogFiles) if err != nil { fmt.Fprintf(os.Stderr, "failed to create file rotator: %v\n", err) os.Exit(1) } pr, pw := io.Pipe() go r.Run(pr) logWriter.RotatorPipe = pw logRotator = r } // setLogLevel sets the logging level for provided subsystem. Invalid // subsystems are ignored. Uninitialized subsystems are dynamically created as // needed. func setLogLevel(subsystemID string, logLevel string) { // Ignore invalid subsystems. logger, ok := subsystemLoggers[subsystemID] if !ok { return } // Defaults to info if the log level is invalid. level, _ := btclog.LevelFromString(logLevel) logger.SetLevel(level) } // setLogLevels sets the log level for all subsystem loggers to the passed // level. It also dynamically creates the subsystem loggers as needed, so it // can be used to initialize the logging system. func setLogLevels(logLevel string) { // Configure all sub-systems with the new logging level. Dynamically // create loggers as needed. for subsystemID := range subsystemLoggers { setLogLevel(subsystemID, logLevel) } } // logClosure is used to provide a closure over expensive logging operations so // don't have to be performed when the logging level doesn't warrant it. type logClosure func() string // String invokes the underlying function and returns the result. func (c logClosure) String() string { return c() } // newLogClosure returns a new closure over a function that returns a string // which itself provides a Stringer interface so that it can be used with the // logging system. func newLogClosure(c func() string) logClosure { return logClosure(c) }