Add tests for wepoll to debug

Author: nekohasekai

common/bufio/fd_demux_windows.go

@@ -6,6 +6,7 @@ import (
 	"context"
 	"sync"
 	"sync/atomic"
+	"unsafe"
 
 	"github.com/sagernet/sing/common/wepoll"
 
@@ -36,7 +37,7 @@ type FDDemultiplexer struct {
 	mutex               sync.Mutex
 	entries             map[int]*fdDemuxEntry
 	registrationCounter uint64
-	registrationToFD    map[uint64]int
+	iosbToFD            map[uintptr]int
 	running             bool
 	closed              atomic.Bool
 	wg                  sync.WaitGroup
@@ -56,12 +57,12 @@ func NewFDDemultiplexer(ctx context.Context) (*FDDemultiplexer, error) {
 
 	ctx, cancel := context.WithCancel(ctx)
 	demux := &FDDemultiplexer{
-		ctx:              ctx,
-		cancel:           cancel,
-		iocp:             iocp,
-		afd:              afd,
-		entries:          make(map[int]*fdDemuxEntry),
-		registrationToFD: make(map[uint64]int),
+		ctx:      ctx,
+		cancel:   cancel,
+		iocp:     iocp,
+		afd:      afd,
+		entries:  make(map[int]*fdDemuxEntry),
+		iosbToFD: make(map[uintptr]int),
 	}
 	return demux, nil
 }
@@ -94,14 +95,14 @@ func (p *FDDemultiplexer) Add(stream *reactorStream, fd int) error {
 	entry.pinner.Pin(&entry.state)
 
 	events := uint32(wepoll.AFD_POLL_RECEIVE | wepoll.AFD_POLL_DISCONNECT | wepoll.AFD_POLL_ABORT | wepoll.AFD_POLL_LOCAL_CLOSE)
-	err = p.afd.Poll(baseHandle, events, &entry.state.iosb, &entry.state.pollInfo, uintptr(regID))
+	err = p.afd.Poll(baseHandle, events, &entry.state.iosb, &entry.state.pollInfo)
 	if err != nil {
 		entry.pinner.Unpin()
 		return err
 	}
 
 	p.entries[fd] = entry
-	p.registrationToFD[regID] = fd
+	p.iosbToFD[uintptr(unsafe.Pointer(&entry.state.iosb))] = fd
 
 	if !p.running {
 		p.running = true
@@ -122,7 +123,9 @@ func (p *FDDemultiplexer) Remove(fd int) {
 	}
 
 	entry.cancelled = true
-	p.afd.Cancel(&entry.state.iosb)
+	if p.afd != nil {
+		p.afd.Cancel(&entry.state.iosb)
+	}
 }
 
 func (p *FDDemultiplexer) wakeup() {
@@ -178,27 +181,28 @@ func (p *FDDemultiplexer) run() {
 
 		for i := uint32(0); i < numRemoved; i++ {
 			ev := entries[i]
-			regID := uint64(ev.CompletionKey)
 
-			if regID == 0 {
+			if ev.Overlapped == nil {
 				continue
 			}
 
+			iosbPtr := uintptr(unsafe.Pointer(ev.Overlapped))
+
 			p.mutex.Lock()
-			fd, ok := p.registrationToFD[regID]
+			fd, ok := p.iosbToFD[iosbPtr]
 			if !ok {
 				p.mutex.Unlock()
 				continue
 			}
 
 			entry := p.entries[fd]
-			if entry == nil || entry.registrationID != regID {
+			if entry == nil {
 				p.mutex.Unlock()
 				continue
 			}
 
 			entry.pinner.Unpin()
-			delete(p.registrationToFD, regID)
+			delete(p.iosbToFD, iosbPtr)
 			delete(p.entries, fd)
 
 			if entry.cancelled {