diff options
| author | Aleksandr Nogikh <nogikh@google.com> | 2024-05-03 13:12:00 +0200 |
|---|---|---|
| committer | Dmitry Vyukov <dvyukov@google.com> | 2024-05-16 15:38:27 +0000 |
| commit | 03820adaef911ce08278d95f034f134c3c0c852e (patch) | |
| tree | 57f87ce0f3dedda459fb1771d3b79ff96e0853bb /pkg/fuzzer/prio_queue.go | |
| parent | ef5d53ed7e3c7d30481a88301f680e37a5cc4775 (diff) | |
pkg/fuzzer: use queue layers
Instead of relying on a fuzzer-internal priority queue, utilize
stackable layers of request-generating steps.
Move the functionality to a separate pkg/fuzzer/queue package.
The pkg/fuzzer/queue package can be reused to add extra processing
layers on top of the fuzzing and to combine machine checking and fuzzing
execution pipelines.
Diffstat (limited to 'pkg/fuzzer/prio_queue.go')
| -rw-r--r-- | pkg/fuzzer/prio_queue.go | 95 |
1 files changed, 0 insertions, 95 deletions
diff --git a/pkg/fuzzer/prio_queue.go b/pkg/fuzzer/prio_queue.go deleted file mode 100644 index c67b6216c..000000000 --- a/pkg/fuzzer/prio_queue.go +++ /dev/null @@ -1,95 +0,0 @@ -// 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 fuzzer - -import ( - "container/heap" - "sync" -) - -type priority []int64 - -func (p priority) greaterThan(other priority) bool { - for i := range p { - if i >= len(other) || p[i] > other[i] { - return true - } - if p[i] < other[i] { - return false - } - } - for i := len(p); i < len(other); i++ { - if other[i] < 0 { - return true - } - if other[i] > 0 { - return false - } - } - return false -} - -type priorityQueue[T any] struct { - impl priorityQueueImpl[T] - mu sync.RWMutex -} - -func makePriorityQueue[T any]() *priorityQueue[T] { - return &priorityQueue[T]{} -} - -func (pq *priorityQueue[T]) Len() int { - pq.mu.RLock() - defer pq.mu.RUnlock() - return pq.impl.Len() -} - -func (pq *priorityQueue[T]) push(item *priorityQueueItem[T]) { - pq.mu.Lock() - defer pq.mu.Unlock() - heap.Push(&pq.impl, item) -} - -func (pq *priorityQueue[T]) tryPop() *priorityQueueItem[T] { - pq.mu.Lock() - defer pq.mu.Unlock() - if len(pq.impl) == 0 { - return nil - } - return heap.Pop(&pq.impl).(*priorityQueueItem[T]) -} - -// The implementation below is based on the example provided -// by https://pkg.go.dev/container/heap. - -type priorityQueueItem[T any] struct { - value T - prio priority -} - -type priorityQueueImpl[T any] []*priorityQueueItem[T] - -func (pq priorityQueueImpl[T]) Len() int { return len(pq) } - -func (pq priorityQueueImpl[T]) Less(i, j int) bool { - // We want Pop to give us the highest, not lowest, - // priority so we use greater than here. - return pq[i].prio.greaterThan(pq[j].prio) -} - -func (pq priorityQueueImpl[T]) Swap(i, j int) { - pq[i], pq[j] = pq[j], pq[i] -} - -func (pq *priorityQueueImpl[T]) Push(x any) { - *pq = append(*pq, x.(*priorityQueueItem[T])) -} - -func (pq *priorityQueueImpl[T]) Pop() any { - n := len(*pq) - item := (*pq)[n-1] - (*pq)[n-1] = nil - *pq = (*pq)[:n-1] - return item -} |
