|
4 | 4 | package cmd |
5 | 5 |
|
6 | 6 | import ( |
| 7 | + "fmt" |
7 | 8 | "log/syslog" |
| 9 | + "net" |
| 10 | + "os" |
| 11 | + "strings" |
| 12 | + "sync" |
| 13 | + "time" |
8 | 14 |
|
9 | 15 | "github.com/semaphoreui/semaphore/util" |
10 | 16 | log "github.com/sirupsen/logrus" |
11 | 17 | lSyslog "github.com/sirupsen/logrus/hooks/syslog" |
12 | 18 | ) |
13 | 19 |
|
| 20 | +var localSyslogPaths = []string{"/dev/log", "/var/run/syslog", "/var/run/log"} |
| 21 | + |
14 | 22 | func initSyslog(conf *util.SyslogConfig) { |
15 | | - if conf.Enabled { |
| 23 | + if !conf.Enabled { |
| 24 | + return |
| 25 | + } |
| 26 | + |
| 27 | + switch conf.Format { |
| 28 | + case util.SyslogRFC5424: |
| 29 | + hook, err := newRFC5424Hook(conf.Network, conf.Address, conf.Tag) |
| 30 | + if err != nil { |
| 31 | + log.WithError(err).Fatal("Failed to create syslog hook") |
| 32 | + return |
| 33 | + } |
| 34 | + log.AddHook(hook) |
| 35 | + log.Info("Syslog logging enabled (RFC 5424)") |
| 36 | + default: |
16 | 37 | hook, err := lSyslog.NewSyslogHook(conf.Network, conf.Address, syslog.LOG_DEBUG, conf.Tag) |
17 | | - if err == nil { |
18 | | - log.AddHook(hook) |
19 | | - log.Info("Syslog logging enabled") |
20 | | - } else { |
| 38 | + if err != nil { |
21 | 39 | log.WithError(err).Fatal("Failed to create syslog hook") |
| 40 | + return |
22 | 41 | } |
| 42 | + log.AddHook(hook) |
| 43 | + log.Info("Syslog logging enabled") |
23 | 44 | } |
24 | 45 | } |
| 46 | + |
| 47 | +type rfc5424Hook struct { |
| 48 | + conn net.Conn |
| 49 | + tag string |
| 50 | + hostname string |
| 51 | + mu sync.Mutex |
| 52 | +} |
| 53 | + |
| 54 | +func newRFC5424Hook(network, address, tag string) (*rfc5424Hook, error) { |
| 55 | + var conn net.Conn |
| 56 | + var err error |
| 57 | + |
| 58 | + if network != "" && address != "" { |
| 59 | + conn, err = net.Dial(network, address) |
| 60 | + } else { |
| 61 | + for _, path := range localSyslogPaths { |
| 62 | + conn, err = net.Dial("unixgram", path) |
| 63 | + if err == nil { |
| 64 | + break |
| 65 | + } |
| 66 | + } |
| 67 | + } |
| 68 | + if err != nil { |
| 69 | + return nil, err |
| 70 | + } |
| 71 | + |
| 72 | + hostname, _ := os.Hostname() |
| 73 | + |
| 74 | + return &rfc5424Hook{ |
| 75 | + conn: conn, |
| 76 | + tag: tag, |
| 77 | + hostname: hostname, |
| 78 | + }, nil |
| 79 | +} |
| 80 | + |
| 81 | +var levelToSeverity = map[log.Level]syslog.Priority{ |
| 82 | + log.PanicLevel: syslog.LOG_CRIT, |
| 83 | + log.FatalLevel: syslog.LOG_CRIT, |
| 84 | + log.ErrorLevel: syslog.LOG_ERR, |
| 85 | + log.WarnLevel: syslog.LOG_WARNING, |
| 86 | + log.InfoLevel: syslog.LOG_INFO, |
| 87 | + log.DebugLevel: syslog.LOG_DEBUG, |
| 88 | + log.TraceLevel: syslog.LOG_DEBUG, |
| 89 | +} |
| 90 | + |
| 91 | +func (h *rfc5424Hook) Levels() []log.Level { |
| 92 | + return log.AllLevels |
| 93 | +} |
| 94 | + |
| 95 | +func (h *rfc5424Hook) Fire(entry *log.Entry) error { |
| 96 | + severity, ok := levelToSeverity[entry.Level] |
| 97 | + if !ok { |
| 98 | + severity = syslog.LOG_INFO |
| 99 | + } |
| 100 | + pri := syslog.LOG_USER | severity |
| 101 | + |
| 102 | + sd := "-" |
| 103 | + if len(entry.Data) > 0 { |
| 104 | + var pairs []string |
| 105 | + for k, v := range entry.Data { |
| 106 | + pairs = append(pairs, fmt.Sprintf(`%s="%s"`, k, escapeSDValue(fmt.Sprintf("%v", v)))) |
| 107 | + } |
| 108 | + sd = fmt.Sprintf("[%s@0 %s]", h.tag, strings.Join(pairs, " ")) |
| 109 | + } |
| 110 | + |
| 111 | + // RFC 5424: <PRI>VERSION SP TIMESTAMP SP HOSTNAME SP APP-NAME SP PROCID SP MSGID SP STRUCTURED-DATA [SP MSG] |
| 112 | + msg := fmt.Sprintf("<%d>1 %s %s %s %d - %s %s", |
| 113 | + pri, |
| 114 | + entry.Time.Format(time.RFC3339), |
| 115 | + h.hostname, |
| 116 | + h.tag, |
| 117 | + os.Getpid(), |
| 118 | + sd, |
| 119 | + entry.Message, |
| 120 | + ) |
| 121 | + |
| 122 | + h.mu.Lock() |
| 123 | + defer h.mu.Unlock() |
| 124 | + _, err := fmt.Fprintln(h.conn, msg) |
| 125 | + return err |
| 126 | +} |
| 127 | + |
| 128 | +func escapeSDValue(v string) string { |
| 129 | + v = strings.ReplaceAll(v, `\`, `\\`) |
| 130 | + v = strings.ReplaceAll(v, `"`, `\"`) |
| 131 | + v = strings.ReplaceAll(v, `]`, `\]`) |
| 132 | + return v |
| 133 | +} |
0 commit comments