Skip to content

Commit

Permalink
Auditbeat: stop monitoring excluded paths (elastic#21282)
Browse files Browse the repository at this point in the history
If the files match the excluded pattern, recursiveWatcher does not emit events for them, as well as errors if any.
  • Loading branch information
newly12 authored Nov 9, 2020
1 parent e0765e4 commit af8eedd
Show file tree
Hide file tree
Showing 5 changed files with 131 additions and 17 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.next.asciidoc
Original file line number Diff line number Diff line change
Expand Up @@ -208,6 +208,7 @@ https://github.com/elastic/beats/compare/v7.0.0-alpha2...master[Check the HEAD d
- auditd: Fix typo in `event.action` of `used-suspicious-link`. {pull}19300[19300]
- system/socket: Fix kprobe grouping to allow running more than one instance. {pull}20325[20325]
- system/socket: Fixed a crash due to concurrent map read and write. {issue}21192[21192] {pull}21690[21690]
- file_integrity: stop monitoring excluded paths {issue}21278[21278] {pull}21282[21282]

*Filebeat*

Expand Down
2 changes: 1 addition & 1 deletion auditbeat/module/file_integrity/eventreader_fsnotify.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ func NewEventReader(c Config) (EventProducer, error) {
}

func (r *reader) Start(done <-chan struct{}) (<-chan Event, error) {
watcher, err := monitor.New(r.config.Recursive)
watcher, err := monitor.New(r.config.Recursive, r.config.IsExcludedPath)
if err != nil {
return nil, err
}
Expand Down
10 changes: 5 additions & 5 deletions auditbeat/module/file_integrity/monitor/monitor.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,15 +37,15 @@ type Watcher interface {

// New creates a new Watcher backed by fsnotify with optional recursive
// logic.
func New(recursive bool) (Watcher, error) {
fsnotify, err := fsnotify.NewWatcher()
func New(recursive bool, IsExcludedPath func(path string) bool) (Watcher, error) {
watcher, err := fsnotify.NewWatcher()
if err != nil {
return nil, err
}
// Use our simulated recursive watches unless the fsnotify implementation
// supports OS-provided recursive watches
if recursive && fsnotify.SetRecursive() != nil {
return newRecursiveWatcher(fsnotify), nil
if recursive && watcher.SetRecursive() != nil {
return newRecursiveWatcher(watcher, IsExcludedPath), nil
}
return (*nonRecursiveWatcher)(fsnotify), nil
return (*nonRecursiveWatcher)(watcher), nil
}
110 changes: 106 additions & 4 deletions auditbeat/module/file_integrity/monitor/monitor_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,10 @@ import (
"github.com/stretchr/testify/assert"
)

func alwaysInclude(path string) bool {
return false
}

func TestNonRecursive(t *testing.T) {
dir, err := ioutil.TempDir("", "monitor")
assertNoError(t, err)
Expand All @@ -44,7 +48,7 @@ func TestNonRecursive(t *testing.T) {
}
defer os.RemoveAll(dir)

watcher, err := New(false)
watcher, err := New(false, alwaysInclude)
assertNoError(t, err)

assertNoError(t, watcher.Add(dir))
Expand Down Expand Up @@ -92,7 +96,7 @@ func TestRecursive(t *testing.T) {
}
defer os.RemoveAll(dir)

watcher, err := New(true)
watcher, err := New(true, alwaysInclude)
assertNoError(t, err)

assertNoError(t, watcher.Add(dir))
Expand Down Expand Up @@ -147,7 +151,7 @@ func TestRecursiveNoFollowSymlink(t *testing.T) {

// Start the watcher

watcher, err := New(true)
watcher, err := New(true, alwaysInclude)
assertNoError(t, err)

assertNoError(t, watcher.Add(dir))
Expand Down Expand Up @@ -208,7 +212,7 @@ func TestRecursiveSubdirPermissions(t *testing.T) {

// Setup watches on watched dir

watcher, err := New(true)
watcher, err := New(true, alwaysInclude)
assertNoError(t, err)

assertNoError(t, watcher.Start())
Expand Down Expand Up @@ -263,6 +267,104 @@ func TestRecursiveSubdirPermissions(t *testing.T) {
}
}

func TestRecursiveExcludedPaths(t *testing.T) {
if runtime.GOOS == "windows" {
t.Skip("Skipping permissions test on Windows")
}

// Create dir to be watched

dir, err := ioutil.TempDir("", "monitor")
assertNoError(t, err)
if runtime.GOOS == "darwin" {
if dirAlt, err := filepath.EvalSymlinks(dir); err == nil {
dir = dirAlt
}
}
defer os.RemoveAll(dir)

// Create not watched dir

outDir, err := ioutil.TempDir("", "non-watched")
assertNoError(t, err)
if runtime.GOOS == "darwin" {
if dirAlt, err := filepath.EvalSymlinks(outDir); err == nil {
outDir = dirAlt
}
}
defer os.RemoveAll(outDir)

// Populate not watched subdir

for _, name := range []string{"a", "b", "c"} {
path := filepath.Join(outDir, name)
assertNoError(t, os.Mkdir(path, 0755))
assertNoError(t, ioutil.WriteFile(filepath.Join(path, name), []byte("Hello"), 0644))
}

// excludes file/dir named "b"
selectiveExclude := func(path string) bool {
r := filepath.Base(path) == "b"
t.Logf("path: %v, excluded: %v\n", path, r)
return r
}

// Setup watches on watched dir

watcher, err := New(true, selectiveExclude)
assertNoError(t, err)

assertNoError(t, watcher.Start())
assertNoError(t, watcher.Add(dir))

defer func() {
assertNoError(t, watcher.Close())
}()

// No event is received

ev, err := readTimeout(t, watcher)
assert.Equal(t, errReadTimeout, err)
if err != errReadTimeout {
t.Fatalf("Expected timeout, got event %+v", ev)
}

// Move the outside directory into the watched

dest := filepath.Join(dir, "subdir")
assertNoError(t, os.Rename(outDir, dest))

// Receive all events

var evs []fsnotify.Event
for {
// No event is received
ev, err := readTimeout(t, watcher)
if err == errReadTimeout {
break
}
assertNoError(t, err)
evs = append(evs, ev)
}

// Verify that events for all accessible files are received
// "b" and "b/b" are missing as they are excluded

expected := map[string]fsnotify.Op{
dest: fsnotify.Create,
filepath.Join(dest, "a"): fsnotify.Create,
filepath.Join(dest, "a/a"): fsnotify.Create,
filepath.Join(dest, "c"): fsnotify.Create,
filepath.Join(dest, "c/c"): fsnotify.Create,
}
assert.Len(t, evs, len(expected))
for _, ev := range evs {
op, found := expected[ev.Name]
assert.True(t, found, ev.Name)
assert.Equal(t, op, ev.Op)
}
}

func testDirOps(t *testing.T, dir string, watcher Watcher) {
fpath := filepath.Join(dir, "file.txt")
fpath2 := filepath.Join(dir, "file2.txt")
Expand Down
25 changes: 18 additions & 7 deletions auditbeat/module/file_integrity/monitor/recursive.go
Original file line number Diff line number Diff line change
Expand Up @@ -36,16 +36,19 @@ type recursiveWatcher struct {
addC chan string
addErrC chan error
log *logp.Logger

isExcludedPath func(path string) bool
}

func newRecursiveWatcher(inner *fsnotify.Watcher) *recursiveWatcher {
func newRecursiveWatcher(inner *fsnotify.Watcher, IsExcludedPath func(path string) bool) *recursiveWatcher {
return &recursiveWatcher{
inner: inner,
tree: FileTree{},
eventC: make(chan fsnotify.Event, 1),
addC: make(chan string),
addErrC: make(chan error),
log: logp.NewLogger(moduleName),
inner: inner,
tree: FileTree{},
eventC: make(chan fsnotify.Event, 1),
addC: make(chan string),
addErrC: make(chan error),
log: logp.NewLogger(moduleName),
isExcludedPath: IsExcludedPath,
}
}

Expand Down Expand Up @@ -82,8 +85,16 @@ func (watcher *recursiveWatcher) ErrorChannel() <-chan error {
}

func (watcher *recursiveWatcher) addRecursive(path string) error {
if watcher.isExcludedPath(path) {
return nil
}

var errs multierror.Errors
err := filepath.Walk(path, func(path string, info os.FileInfo, fnErr error) error {
if watcher.isExcludedPath(path) {
return nil
}

if fnErr != nil {
errs = append(errs, errors.Wrapf(fnErr, "error walking path '%s'", path))
// If FileInfo is not nil, the directory entry can be processed
Expand Down

0 comments on commit af8eedd

Please sign in to comment.