2016-09-04 21:03:36 +00:00
|
|
|
// +build darwin dragonfly freebsd linux netbsd openbsd solaris windows
|
2016-03-17 09:53:31 +00:00
|
|
|
|
|
|
|
package executor
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"io"
|
2017-11-23 10:57:26 +00:00
|
|
|
"runtime"
|
2017-11-21 23:51:39 +00:00
|
|
|
"syscall"
|
2016-09-04 21:03:36 +00:00
|
|
|
|
|
|
|
syslog "github.com/RackSec/srslog"
|
2016-03-17 09:53:31 +00:00
|
|
|
|
|
|
|
"github.com/hashicorp/nomad/client/driver/logging"
|
|
|
|
)
|
|
|
|
|
2016-10-12 18:35:29 +00:00
|
|
|
func (e *UniversalExecutor) LaunchSyslogServer() (*SyslogServerState, error) {
|
|
|
|
// Ensure the context has been set first
|
|
|
|
if e.ctx == nil {
|
|
|
|
return nil, fmt.Errorf("SetContext must be called before launching the Syslog Server")
|
|
|
|
}
|
2016-03-26 00:48:05 +00:00
|
|
|
|
2016-03-17 09:53:31 +00:00
|
|
|
e.syslogChan = make(chan *logging.SyslogMessage, 2048)
|
|
|
|
l, err := e.getListener(e.ctx.PortLowerBound, e.ctx.PortUpperBound)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2017-06-16 14:07:34 +00:00
|
|
|
e.logger.Printf("[DEBUG] syslog-server: launching syslog server on addr: %v", l.Addr().String())
|
2016-03-17 09:53:31 +00:00
|
|
|
if err := e.configureLoggers(); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
e.syslogServer = logging.NewSyslogServer(l, e.syslogChan, e.logger)
|
|
|
|
go e.syslogServer.Start()
|
|
|
|
go e.collectLogs(e.lre, e.lro)
|
|
|
|
syslogAddr := fmt.Sprintf("%s://%s", l.Addr().Network(), l.Addr().String())
|
|
|
|
return &SyslogServerState{Addr: syslogAddr}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (e *UniversalExecutor) collectLogs(we io.Writer, wo io.Writer) {
|
|
|
|
for logParts := range e.syslogChan {
|
|
|
|
// If the severity of the log line is err then we write to stderr
|
|
|
|
// otherwise all messages go to stdout
|
|
|
|
if logParts.Severity == syslog.LOG_ERR {
|
|
|
|
e.lre.Write(logParts.Message)
|
|
|
|
e.lre.Write([]byte{'\n'})
|
|
|
|
} else {
|
|
|
|
e.lro.Write(logParts.Message)
|
|
|
|
e.lro.Write([]byte{'\n'})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2017-11-21 23:51:39 +00:00
|
|
|
|
|
|
|
// configure new process group for child process
|
|
|
|
func (e *UniversalExecutor) setNewProcessGroup() error {
|
2017-11-23 10:57:26 +00:00
|
|
|
// We need to check that as build flags includes windows for this file
|
|
|
|
if runtime.GOOS == "windows" {
|
|
|
|
return nil
|
|
|
|
}
|
2017-11-21 23:51:39 +00:00
|
|
|
if e.cmd.SysProcAttr == nil {
|
|
|
|
e.cmd.SysProcAttr = &syscall.SysProcAttr{}
|
|
|
|
}
|
|
|
|
e.cmd.SysProcAttr.Setpgid = true
|
|
|
|
return nil
|
|
|
|
}
|