go-divert

golang client for windivert

Documnet

Example:

package main

import (
    "fmt"
    "log"

    "github.com/lysShub/divert-go"
    "gvisor.dev/gvisor/pkg/tcpip/header" // go get gvisor.dev/gvisor@go
)

func main() {
    divert.MustLoad(divert.DLL)
    defer divert.Release()

    d, err := divert.Open("tcp.Syn and !loopback", divert.Network, 0, divert.Sniff|divert.ReadOnly)
    if err != nil {
        log.Fatal(err)
    }

    var b = make([]byte, 1536)
    var addr divert.Address
    for {
        n, err := d.Recv(b[:cap(b)], &addr)
        if err != nil {
            if errors.Is(err, windows.ERROR_INSUFFICIENT_BUFFER) {
                continue
            }
            log.Fatal(err)
        }

        if !addr.IPv6() {
            if n >= header.IPv4MinimumSize+header.TCPMinimumSize {
                iphdr := header.IPv4(b[:n])
                tcphdr := header.TCP(iphdr[iphdr.HeaderLength():])

                fmt.Printf("%s:%d --> %s:%d \n",
                    iphdr.SourceAddress().String(),
                    tcphdr.SourcePort(),
                    iphdr.DestinationAddress().String(),
                    tcphdr.DestinationPort(),
                )
            }
        } else {
            if n >= header.IPv6MinimumSize+header.TCPMinimumSize {
                iphdr := header.IPv6(b[:n])
                tcphdr := header.TCP(iphdr[header.IPv6MinimumSize:])

                fmt.Printf("%s:%d --> %s:%d \n",
                    iphdr.SourceAddress().String(),
                    tcphdr.SourcePort(),
                    iphdr.DestinationAddress().String(),
                    tcphdr.DestinationPort(),
                )
            }
        }
    }
}