sgur
2/9/2017 - 3:41 PM

Remove Windows Firewall Rules which is generated by "go run"

Remove Windows Firewall Rules which is generated by "go run"

package main

import (
	"fmt"
	"log"
	"os"
	"path/filepath"
	"strings"

	"github.com/go-ole/go-ole"
	"github.com/go-ole/go-ole/oleutil"
)

func main() {
	pattern := filepath.Join(os.TempDir(), "go-build")

	ole.CoInitialize(0)

	unk, err := oleutil.CreateObject("HNetCfg.FwPolicy2")
	if err != nil {
		log.Fatal(err)
	}
	dsp, err := unk.QueryInterface(ole.IID_IDispatch)
	if err != nil {
		log.Fatal(err)
	}
	rules := oleutil.MustGetProperty(dsp, "Rules").ToIDispatch()

	err = oleutil.ForEach(rules, func(v *ole.VARIANT) error {
		rule := v.ToIDispatch()
		appname := oleutil.MustGetProperty(rule, "Applicationname").ToString()
		if strings.HasPrefix(strings.ToLower(appname), strings.ToLower(pattern)) {
			_, rerr := oleutil.CallMethod(rules, "Remove", appname)
			if rerr != nil {
				log.Fatal(rerr)
			}
			fmt.Println(appname)
		}
		return nil
	})
	if err != nil {
		log.Fatal(err)
	}
}