-
Notifications
You must be signed in to change notification settings - Fork 4.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
33 changed files
with
1,353 additions
and
206 deletions.
There are no files selected for viewing
File renamed without changes.
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -6,4 +6,3 @@ module/*/_meta/config.yml | |
/auditbeat | ||
/auditbeat.test | ||
/docs/html_docs | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,40 @@ | ||
// Licensed to Elasticsearch B.V. under one or more contributor | ||
// license agreements. See the NOTICE file distributed with | ||
// this work for additional information regarding copyright | ||
// ownership. Elasticsearch B.V. licenses this file to you under | ||
// the Apache License, Version 2.0 (the "License"); you may | ||
// not use this file except in compliance with the License. | ||
// You may obtain a copy of the License at | ||
// | ||
// http://www.apache.org/licenses/LICENSE-2.0 | ||
// | ||
// Unless required by applicable law or agreed to in writing, | ||
// software distributed under the License is distributed on an | ||
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | ||
// KIND, either express or implied. See the License for the | ||
// specific language governing permissions and limitations | ||
// under the License. | ||
|
||
//go:build linux | ||
|
||
package ebpf | ||
|
||
import ( | ||
"runtime" | ||
|
||
"github.com/elastic/beats/v7/libbeat/common/seccomp" | ||
) | ||
|
||
func init() { | ||
switch runtime.GOARCH { | ||
case "amd64", "arm64": | ||
syscalls := []string{ | ||
"bpf", | ||
"eventfd2", // needed by ringbuf | ||
"perf_event_open", // needed by tracepoints | ||
} | ||
if err := seccomp.ModifyDefaultPolicy(seccomp.AddSyscall, syscalls...); err != nil { | ||
panic(err) | ||
} | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,27 @@ | ||
// Licensed to Elasticsearch B.V. under one or more contributor | ||
// license agreements. See the NOTICE file distributed with | ||
// this work for additional information regarding copyright | ||
// ownership. Elasticsearch B.V. licenses this file to you under | ||
// the Apache License, Version 2.0 (the "License"); you may | ||
// not use this file except in compliance with the License. | ||
// You may obtain a copy of the License at | ||
// | ||
// http://www.apache.org/licenses/LICENSE-2.0 | ||
// | ||
// Unless required by applicable law or agreed to in writing, | ||
// software distributed under the License is distributed on an | ||
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | ||
// KIND, either express or implied. See the License for the | ||
// specific language governing permissions and limitations | ||
// under the License. | ||
|
||
package ebpf | ||
|
||
import "github.com/elastic/ebpfevents" | ||
|
||
type EventMask uint64 | ||
|
||
type Watcher interface { | ||
Subscribe(string, EventMask) (<-chan ebpfevents.Event, <-chan error) | ||
Unsubscribe(string) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,170 @@ | ||
// Licensed to Elasticsearch B.V. under one or more contributor | ||
// license agreements. See the NOTICE file distributed with | ||
// this work for additional information regarding copyright | ||
// ownership. Elasticsearch B.V. licenses this file to you under | ||
// the Apache License, Version 2.0 (the "License"); you may | ||
// not use this file except in compliance with the License. | ||
// You may obtain a copy of the License at | ||
// | ||
// http://www.apache.org/licenses/LICENSE-2.0 | ||
// | ||
// Unless required by applicable law or agreed to in writing, | ||
// software distributed under the License is distributed on an | ||
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | ||
// KIND, either express or implied. See the License for the | ||
// specific language governing permissions and limitations | ||
// under the License. | ||
|
||
//go:build linux | ||
|
||
package ebpf | ||
|
||
import ( | ||
"context" | ||
"fmt" | ||
"sync" | ||
|
||
"github.com/elastic/ebpfevents" | ||
) | ||
|
||
var ( | ||
gWatcherOnce sync.Once | ||
gWatcherErr error | ||
gWatcher watcher | ||
) | ||
|
||
type client struct { | ||
name string | ||
mask EventMask | ||
events chan ebpfevents.Event | ||
errors chan error | ||
} | ||
|
||
type watcher struct { | ||
sync.Mutex | ||
ctx context.Context | ||
cancel context.CancelFunc | ||
loader *ebpfevents.Loader | ||
clients map[string]client | ||
status status | ||
} | ||
|
||
type status int | ||
|
||
const ( | ||
stopped status = iota | ||
started | ||
) | ||
|
||
func GetWatcher() (Watcher, error) { | ||
gWatcher.Lock() | ||
defer gWatcher.Unlock() | ||
|
||
// Try to load the probe once on startup so consumers can error out. | ||
gWatcherOnce.Do(func() { | ||
if gWatcher.status == stopped { | ||
l, err := ebpfevents.NewLoader() | ||
if err != nil { | ||
gWatcherErr = fmt.Errorf("init ebpf loader: %w", err) | ||
return | ||
} | ||
_ = l.Close() | ||
} | ||
}) | ||
|
||
return &gWatcher, gWatcherErr | ||
} | ||
|
||
func (w *watcher) Subscribe(name string, events EventMask) (<-chan ebpfevents.Event, <-chan error) { | ||
w.Lock() | ||
defer w.Unlock() | ||
|
||
if w.status == stopped { | ||
startLocked() | ||
} | ||
|
||
w.clients[name] = client{ | ||
name: name, | ||
mask: events, | ||
events: make(chan ebpfevents.Event), | ||
errors: make(chan error), | ||
} | ||
|
||
return w.clients[name].events, w.clients[name].errors | ||
} | ||
|
||
func (w *watcher) Unsubscribe(name string) { | ||
w.Lock() | ||
defer w.Unlock() | ||
|
||
delete(w.clients, name) | ||
|
||
if w.nclients() == 0 { | ||
stopLocked() | ||
} | ||
} | ||
|
||
func startLocked() { | ||
loader, err := ebpfevents.NewLoader() | ||
if err != nil { | ||
gWatcherErr = fmt.Errorf("start ebpf loader: %w", err) | ||
return | ||
} | ||
|
||
gWatcher.loader = loader | ||
gWatcher.clients = make(map[string]client) | ||
|
||
events := make(chan ebpfevents.Event) | ||
errors := make(chan error) | ||
gWatcher.ctx, gWatcher.cancel = context.WithCancel(context.Background()) | ||
|
||
go gWatcher.loader.EventLoop(gWatcher.ctx, events, errors) | ||
go func() { | ||
for { | ||
select { | ||
case err := <-errors: | ||
for _, client := range gWatcher.clients { | ||
client.errors <- err | ||
} | ||
continue | ||
case ev := <-events: | ||
for _, client := range gWatcher.clients { | ||
if client.mask&EventMask(ev.Type) != 0 { | ||
client.events <- ev | ||
} | ||
} | ||
continue | ||
case <-gWatcher.ctx.Done(): | ||
return | ||
} | ||
} | ||
}() | ||
|
||
gWatcher.status = started | ||
} | ||
|
||
func stopLocked() { | ||
_ = gWatcher.close() | ||
gWatcher.status = stopped | ||
} | ||
|
||
func (w *watcher) nclients() int { | ||
return len(w.clients) | ||
} | ||
|
||
func (w *watcher) close() error { | ||
if w.cancel != nil { | ||
w.cancel() | ||
} | ||
|
||
if w.loader != nil { | ||
_ = w.loader.Close() | ||
} | ||
|
||
for _, cl := range w.clients { | ||
close(cl.events) | ||
close(cl.errors) | ||
} | ||
|
||
return nil | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,28 @@ | ||
// Licensed to Elasticsearch B.V. under one or more contributor | ||
// license agreements. See the NOTICE file distributed with | ||
// this work for additional information regarding copyright | ||
// ownership. Elasticsearch B.V. licenses this file to you under | ||
// the Apache License, Version 2.0 (the "License"); you may | ||
// not use this file except in compliance with the License. | ||
// You may obtain a copy of the License at | ||
// | ||
// http://www.apache.org/licenses/LICENSE-2.0 | ||
// | ||
// Unless required by applicable law or agreed to in writing, | ||
// software distributed under the License is distributed on an | ||
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | ||
// KIND, either express or implied. See the License for the | ||
// specific language governing permissions and limitations | ||
// under the License. | ||
|
||
//go:build !linux | ||
|
||
package ebpf | ||
|
||
import "errors" | ||
|
||
var ErrNotSupported = errors.New("not supported") | ||
|
||
func NewWatcher() (Watcher, error) { | ||
return nil, ErrNotSupported | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,61 @@ | ||
// Licensed to Elasticsearch B.V. under one or more contributor | ||
// license agreements. See the NOTICE file distributed with | ||
// this work for additional information regarding copyright | ||
// ownership. Elasticsearch B.V. licenses this file to you under | ||
// the Apache License, Version 2.0 (the "License"); you may | ||
// not use this file except in compliance with the License. | ||
// You may obtain a copy of the License at | ||
// | ||
// http://www.apache.org/licenses/LICENSE-2.0 | ||
// | ||
// Unless required by applicable law or agreed to in writing, | ||
// software distributed under the License is distributed on an | ||
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | ||
// KIND, either express or implied. See the License for the | ||
// specific language governing permissions and limitations | ||
// under the License. | ||
|
||
//go:build linux | ||
|
||
package ebpf | ||
|
||
import ( | ||
"math" | ||
"testing" | ||
|
||
"github.com/stretchr/testify/assert" | ||
) | ||
|
||
const allEvents = EventMask(math.MaxUint64) | ||
|
||
func TestWatcherStartStop(t *testing.T) { | ||
w, err := GetWatcher() | ||
if err != nil { | ||
t.Skipf("skipping ebpf watcher test: %v", err) | ||
} | ||
assert.Equal(t, gWatcher.status, stopped) | ||
assert.Equal(t, 0, gWatcher.nclients()) | ||
|
||
_, _ = w.Subscribe("test-1", allEvents) | ||
assert.Equal(t, gWatcher.status, started) | ||
assert.Equal(t, 1, gWatcher.nclients()) | ||
|
||
_, _ = w.Subscribe("test-2", allEvents) | ||
assert.Equal(t, 2, gWatcher.nclients()) | ||
|
||
w.Unsubscribe("test-2") | ||
assert.Equal(t, 1, gWatcher.nclients()) | ||
|
||
w.Unsubscribe("dummy") | ||
assert.Equal(t, 1, gWatcher.nclients()) | ||
|
||
assert.Equal(t, gWatcher.status, started) | ||
w.Unsubscribe("test-1") | ||
assert.Equal(t, 0, gWatcher.nclients()) | ||
assert.Equal(t, gWatcher.status, stopped) | ||
|
||
_, _ = w.Subscribe("new", allEvents) | ||
assert.Equal(t, 1, gWatcher.nclients()) | ||
assert.Equal(t, gWatcher.status, started) | ||
w.Unsubscribe("new") | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.