diff options
| -rw-r--r-- | syz-hub/state/state.go | 31 |
1 files changed, 14 insertions, 17 deletions
diff --git a/syz-hub/state/state.go b/syz-hub/state/state.go index 7ccdf1827..a8a1b36ea 100644 --- a/syz-hub/state/state.go +++ b/syz-hub/state/state.go @@ -288,10 +288,21 @@ func (st *State) pendingInputs(mgr *Manager) ([][]byte, int, error) { } maxSeq := st.corpusSeq more := 0 - // Send at most that many records (rounded up to next seq number). - const maxRecords = 100 + const ( + // Send at most that many records (rounded up to next seq number). + maxRecords = 100 + // If we have way too many records to send (more than capRecords), + // cap total number to capRecords and give up sending all. + // Otherwise new managers will never chew all this on a busy hub. + capRecords = 100000 + ) if len(records) > maxRecords { - sort.Sort(recordSeqSorter(records)) + sort.Slice(records, func(i, j int) bool { + return records[i].Seq < records[j].Seq + }) + if len(records) > capRecords { + records = records[len(records)-capRecords:] + } pos := maxRecords maxSeq = records[pos].Seq for pos+1 < len(records) && records[pos+1].Seq == maxSeq { @@ -380,17 +391,3 @@ func loadSeqFile(filename string) uint64 { seq, _ := strconv.ParseUint(string(str), 10, 64) return seq } - -type recordSeqSorter []db.Record - -func (a recordSeqSorter) Len() int { - return len(a) -} - -func (a recordSeqSorter) Less(i, j int) bool { - return a[i].Seq < a[j].Seq -} - -func (a recordSeqSorter) Swap(i, j int) { - a[i], a[j] = a[j], a[i] -} |