From 348c67d932820fa8ef44075bc6d4e014bc82d8ef Mon Sep 17 00:00:00 2001 From: jarnpher_rice Date: Wed, 1 Jun 2022 01:36:19 +0800 Subject: [PATCH] Add WithStdout option for osquery's Runner (#5637) * Add WithStdout option for osquery's Runner fetch the osquery output to do something in other flow * Create osquery-runner-withstdout * Update osquery.go --- orbit/changes/osquery-runner-withstdout | 1 + orbit/pkg/osquery/osquery.go | 8 ++++++++ 2 files changed, 9 insertions(+) create mode 100644 orbit/changes/osquery-runner-withstdout diff --git a/orbit/changes/osquery-runner-withstdout b/orbit/changes/osquery-runner-withstdout new file mode 100644 index 0000000000..18b102bb23 --- /dev/null +++ b/orbit/changes/osquery-runner-withstdout @@ -0,0 +1 @@ +* Add WithStdout option for osquery's Runner diff --git a/orbit/pkg/osquery/osquery.go b/orbit/pkg/osquery/osquery.go index 92e47f595d..2c51df894f 100644 --- a/orbit/pkg/osquery/osquery.go +++ b/orbit/pkg/osquery/osquery.go @@ -115,6 +115,14 @@ func WithStderr(w io.Writer) Option { } } +// WithStdout sets the runner's cmd's stdout to the given writer. +func WithStdout(w io.Writer) Option { + return func(r *Runner) error { + r.cmd.Stdout = w + return nil + } +} + func WithLogPath(path string) Option { return func(r *Runner) error { if err := secure.MkdirAll(path, constant.DefaultDirMode); err != nil {