2020-02-13 08:42:07 -07:00
|
|
|
package util
|
2019-02-05 04:09:05 -07:00
|
|
|
|
|
|
|
import (
|
|
|
|
"log"
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
"golang.org/x/sys/windows"
|
|
|
|
"golang.org/x/sys/windows/svc/eventlog"
|
|
|
|
)
|
|
|
|
|
|
|
|
type eventLogWriter struct {
|
|
|
|
el *eventlog.Log
|
|
|
|
}
|
|
|
|
|
|
|
|
// Write sends a log message to the Event Log.
|
|
|
|
func (w *eventLogWriter) Write(b []byte) (int, error) {
|
|
|
|
return len(b), w.el.Info(1, string(b))
|
|
|
|
}
|
|
|
|
|
2020-02-13 08:42:07 -07:00
|
|
|
func ConfigureSyslog(serviceName string) error {
|
2019-02-05 04:21:07 -07:00
|
|
|
// Note that the eventlog src is the same as the service name
|
|
|
|
// Otherwise, we will get "the description for event id cannot be found" warning in every log record
|
|
|
|
|
2019-02-05 04:09:05 -07:00
|
|
|
// Continue if we receive "registry key already exists" or if we get
|
|
|
|
// ERROR_ACCESS_DENIED so that we can log without administrative permissions
|
|
|
|
// for pre-existing eventlog sources.
|
2019-02-05 04:21:07 -07:00
|
|
|
if err := eventlog.InstallAsEventCreate(serviceName, eventlog.Info|eventlog.Warning|eventlog.Error); err != nil {
|
2019-02-05 04:09:05 -07:00
|
|
|
if !strings.Contains(err.Error(), "registry key already exists") && err != windows.ERROR_ACCESS_DENIED {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
2019-02-05 04:21:07 -07:00
|
|
|
el, err := eventlog.Open(serviceName)
|
2019-02-05 04:09:05 -07:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
log.SetOutput(&eventLogWriter{el: el})
|
|
|
|
return nil
|
|
|
|
}
|