backtrace/bk/listen_windows.go

//go:build windows
// +build windows

package backtrace

import (
	"net"

	"github.com/oneclickvirt/backtrace/model"
	. "github.com/oneclickvirt/defaultset"
	"golang.org/x/sys/windows"
)

func (t *Tracer) listen(network string, laddr *net.IPAddr) (*net.IPConn, error) {
	if model.EnableLoger {
		InitLogger()
		defer Logger.Sync()
	}
	conn, err := net.ListenIP(network, laddr)
	if err != nil {
		if model.EnableLoger {
			Logger.Info(err.Error())
		}
		return nil, err
	}
	raw, err := conn.SyscallConn()
	if err != nil {
		if model.EnableLoger {
			Logger.Info(err.Error())
		}
		conn.Close()
		return nil, err
	}
	_ = raw.Control(func(fd uintptr) {
		err = windows.SetsockoptInt(windows.Handle(fd), windows.IPPROTO_IP, windows.IP_HDRINCL, 1)
	})
	if err != nil {
		if model.EnableLoger {
			Logger.Info(err.Error())
		}
		conn.Close()
		return nil, err
	}
	return conn, nil
}