Try a fix

Author: jrife

net/ipv4/tcp_ipv4.c

@@ -3112,6 +3112,9 @@ static struct sock *bpf_iter_tcp_resume_listening(struct seq_file *seq)
 	int resume_bucket = st->bucket;
 	struct sock *sk;
 
+	if (end_cookie && find_cookie == end_cookie)
+		++st->bucket;
+
 	sk = listening_get_first(seq);
 	iter->cur_sk = 0;
 	iter->end_sk = 0;
@@ -3139,6 +3142,9 @@ static struct sock *bpf_iter_tcp_resume_established(struct seq_file *seq)
 	int resume_bucket = st->bucket;
 	struct sock *sk;
 
+	if (end_cookie && find_cookie == end_cookie)
+		++st->bucket;
+
 	sk = established_get_first(seq);
 	iter->cur_sk = 0;
 	iter->end_sk = 0;
@@ -3162,12 +3168,6 @@ static struct sock *bpf_iter_tcp_resume(struct seq_file *seq)
 	struct tcp_iter_state *st = &iter->state;
 	struct sock *sk = NULL;
 
-	/* Don't waste time finding the first socket in this bucket if we're
-	 * already done with it.
-	 */
-	if (iter->end_sk && iter->cur_sk == iter->end_sk)
-		++st->bucket;
-
 	switch (st->state) {
 	case TCP_SEQ_STATE_LISTENING:
 		sk = bpf_iter_tcp_resume_listening(seq);