1 files changed, 42 insertions, 0 deletions
diff --git a/pkg/symbolizer/cache.go b/pkg/symbolizer/cache.go
new file mode 100644
index 000000000..dcade3929
--- /dev/null
+++ b/pkg/symbolizer/cache.go
@@ -0,0 +1,42 @@
+// Copyright 2024 syzkaller project authors. All rights reserved.
+// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
+
+package symbolizer
+
+import (
+	"sync"
+)
+
+// Cache caches symbolization results from Symbolizer in a thread-safe way.
+type Cache struct {
+	mu    sync.RWMutex
+	cache map[cacheKey]cacheVal
+}
+
+type cacheKey struct {
+	bin string
+	pc  uint64
+}
+
+type cacheVal struct {
+	frames []Frame
+	err    error
+}
+
+func (c *Cache) Symbolize(inner func(string, uint64) ([]Frame, error), bin string, pc uint64) ([]Frame, error) {
+	key := cacheKey{bin, pc}
+	c.mu.RLock()
+	val, ok := c.cache[key]
+	c.mu.RUnlock()
+	if ok {
+		return val.frames, val.err
+	}
+	frames, err := inner(bin, pc)
+	c.mu.Lock()
+	if c.cache == nil {
+		c.cache = make(map[cacheKey]cacheVal)
+	}
+	c.cache[key] = cacheVal{frames, err}
+	c.mu.Unlock()
+	return frames, err
+}