External/pmg
1
0
Fork 0
mirror of https://github.com/safedep/pmg.git synced 2026-08-03 07:24:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add audit system with sink based dispatcher (#211)

Browse Source 
* chore: Dependency update

* feat: Add audit system with eventlog as a sink

* fix: Linter fixes

* fix: Code review fixes
This commit is contained in:
Abhisek Datta
2026-04-10 19:07:15 +05:30
committed by GitHub
parent e72ff6aeaf
commit 0355a5d4fd
19 changed files with 872 additions and 284 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/eventlog/eventlog_test.go
+21 -9
View File
@@ -85,12 +85,10 @@ func TestLogEvent(t *testing.T) {
assert.Equal(t, "evil-package", loggedEvent.PackageName)
}
func TestLogMalwareBlocked(t *testing.T) {
// Create a temporary directory for testing
func TestLogEventMalwareBlocked(t *testing.T) {
tmpDir := t.TempDir()
logDir := filepath.Join(tmpDir, ".pmg", "logs")
// Initialize logger
err := reinitializeForTest(logDir)
assert.NoError(t, err, "Failed to initialize logger")
defer func() {
@@ -98,10 +96,16 @@ func TestLogMalwareBlocked(t *testing.T) {
assert.NoError(t, err)
}()
// Log malware blocked event
LogMalwareBlocked("malicious-pkg", "2.0.0", "pypi", "Contains known malware", nil)
err = LogEvent(Event{
EventType: EventTypeMalwareBlocked,
Message: "Blocked installation of malicious package: malicious-pkg@2.0.0",
PackageName: "malicious-pkg",
Version: "2.0.0",
Ecosystem: "pypi",
Details: map[string]interface{}{"reason": "Contains known malware"},
})
assert.NoError(t, err)
// Read and verify
logFilePath := filepath.Join(logDir, time.Now().Format("20060102")+"-pmg.log")
data, err := os.ReadFile(logFilePath)
assert.NoError(t, err, "Failed to read log file")
@@ -115,7 +119,7 @@ func TestLogMalwareBlocked(t *testing.T) {
assert.Equal(t, "pypi", event.Ecosystem)
}
func TestLogProxyHostObserved(t *testing.T) {
func TestLogEventProxyHostObserved(t *testing.T) {
tmpDir := t.TempDir()
logDir := filepath.Join(tmpDir, ".pmg", "logs")
@@ -126,9 +130,17 @@ func TestLogProxyHostObserved(t *testing.T) {
assert.NoError(t, err)
}()
LogProxyHostObserved("example.com", "CONNECT", "connect_tunnel_no_interceptor", map[string]interface{}{
"request_id": "abc123",
err = LogEvent(Event{
EventType: EventTypeProxyHostObserved,
Message: "Proxy observed outbound host: example.com",
Details: map[string]interface{}{
"hostname": "example.com",
"method": "CONNECT",
"reason": "connect_tunnel_no_interceptor",
"request_id": "abc123",
},
})
assert.NoError(t, err)
logFilePath := filepath.Join(logDir, time.Now().Format("20060102")+"-pmg.log")
data, err := os.ReadFile(logFilePath)