mirror of
https://github.com/safing/portmaster
synced 2025-09-02 02:29:12 +00:00
91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
//go:build windows
|
|
// +build windows
|
|
|
|
package windowskext
|
|
|
|
import "golang.org/x/sys/windows"
|
|
|
|
const (
|
|
METHOD_BUFFERED = 0
|
|
METHOD_IN_DIRECT = 1
|
|
METHOD_OUT_DIRECT = 2
|
|
METHOD_NEITHER = 3
|
|
|
|
SIOCTL_TYPE = 40000
|
|
)
|
|
|
|
var (
|
|
IOCTL_HELLO = ctlCode(SIOCTL_TYPE, 0x800, METHOD_BUFFERED, windows.FILE_READ_DATA|windows.FILE_WRITE_DATA)
|
|
IOCTL_RECV_VERDICT_REQ_POLL = ctlCode(SIOCTL_TYPE, 0x801, METHOD_BUFFERED, windows.FILE_READ_DATA|windows.FILE_WRITE_DATA)
|
|
IOCTL_RECV_VERDICT_REQ = ctlCode(SIOCTL_TYPE, 0x802, METHOD_BUFFERED, windows.FILE_READ_DATA|windows.FILE_WRITE_DATA)
|
|
IOCTL_SET_VERDICT = ctlCode(SIOCTL_TYPE, 0x803, METHOD_BUFFERED, windows.FILE_READ_DATA|windows.FILE_WRITE_DATA)
|
|
IOCTL_GET_PAYLOAD = ctlCode(SIOCTL_TYPE, 0x804, METHOD_BUFFERED, windows.FILE_READ_DATA|windows.FILE_WRITE_DATA)
|
|
IOCTL_CLEAR_CACHE = ctlCode(SIOCTL_TYPE, 0x805, METHOD_BUFFERED, windows.FILE_READ_DATA|windows.FILE_WRITE_DATA)
|
|
)
|
|
|
|
func ctlCode(device_type, function, method, access uint32) uint32 {
|
|
return (device_type << 16) | (access << 14) | (function << 2) | method
|
|
}
|
|
|
|
func deviceIoControlRead(handle windows.Handle, code uint32, data []byte) (uint32, error) {
|
|
var bytesReturned uint32
|
|
|
|
var dataPtr *byte = nil
|
|
var dataSize uint32 = 0
|
|
if data != nil {
|
|
dataPtr = &data[0]
|
|
dataSize = uint32(len(data))
|
|
}
|
|
|
|
err := windows.DeviceIoControl(handle,
|
|
code,
|
|
nil, 0,
|
|
dataPtr, dataSize,
|
|
&bytesReturned, nil)
|
|
|
|
return bytesReturned, err
|
|
}
|
|
|
|
func deviceIoControlWrite(handle windows.Handle, code uint32, data []byte) (uint32, error) {
|
|
var bytesReturned uint32
|
|
|
|
var dataPtr *byte = nil
|
|
var dataSize uint32 = 0
|
|
if data != nil {
|
|
dataPtr = &data[0]
|
|
dataSize = uint32(len(data))
|
|
}
|
|
|
|
err := windows.DeviceIoControl(handle,
|
|
code,
|
|
dataPtr, dataSize,
|
|
nil, 0,
|
|
&bytesReturned, nil)
|
|
|
|
return bytesReturned, err
|
|
}
|
|
|
|
func deviceIoControlReadWrite(handle windows.Handle, code uint32, inData []byte, outData []byte) (uint32, error) {
|
|
var bytesReturned uint32
|
|
|
|
var inDataPtr *byte = nil
|
|
var inDataSize uint32 = 0
|
|
if inData != nil {
|
|
inDataPtr = &inData[0]
|
|
inDataSize = uint32(len(inData))
|
|
}
|
|
|
|
var outDataPtr *byte = nil
|
|
var outDataSize uint32 = 0
|
|
if outData != nil {
|
|
outDataPtr = &outData[0]
|
|
outDataSize = uint32(len(outData))
|
|
}
|
|
err := windows.DeviceIoControl(handle,
|
|
code,
|
|
inDataPtr, inDataSize,
|
|
outDataPtr, outDataSize,
|
|
&bytesReturned, nil)
|
|
|
|
return bytesReturned, err
|
|
}
|