fix: bugs

Author: jayantxie

connstate/conn.go

@@ -58,7 +58,7 @@ func ListenConnState(conn net.Conn) (ConnStater, error) {
 	err = rawConn.Control(func(fileDescriptor uintptr) {
 		fd = pollcache.alloc()
 		fd.fd = int(fileDescriptor)
-		fd.conn = unsafe.Pointer(&connStater{fd: unsafe.Pointer(fd)})
+		atomic.StorePointer(&fd.conn, unsafe.Pointer(&connStater{fd: unsafe.Pointer(fd)}))
 		opAddErr = poll.control(fd, opAdd)
 	})
 	if fd != nil {
@@ -75,7 +75,7 @@ func ListenConnState(conn net.Conn) (ConnStater, error) {
 	if opAddErr != nil {
 		return nil, opAddErr
 	}
-	return (*connStater)(fd.conn), nil
+	return (*connStater)(atomic.LoadPointer(&fd.conn)), nil
 }
 
 type connStater struct {

connstate/poll_bsd.go

@@ -27,6 +27,7 @@ type kqueue struct {
 	fd int
 }
 
+//go:nocheckptr
 func (p *kqueue) wait() error {
 	events := make([]syscall.Kevent_t, 1024)
 	for {

connstate/poll_linux.go

@@ -28,6 +28,7 @@ type epoller struct {
 	epfd int
 }
 
+//go:nocheckptr
 func (p *epoller) wait() error {
 	events := make([]syscall.EpollEvent, 1024)
 	for {