main.go

// Hardentools
// Copyright (C) 2017  Security Without Borders
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package main

import (
	"github.com/lxn/walk"
	. "github.com/lxn/walk/declarative"
	"golang.org/x/sys/windows/registry"
	"os"
)

var window *walk.MainWindow
var events *walk.TextEdit
var progress *walk.ProgressBar

const harden_key_path = "SOFTWARE\\Security Without Borders\\"

func check_status() bool {
	key, err := registry.OpenKey(registry.CURRENT_USER, harden_key_path, registry.READ)
	if err != nil {
		return false
	}

	value, _, err := key.GetIntegerValue("Harden")
	if err != nil {
		return false
	}

	if value == 1 {
		return true
	} else {
		return false
	}
}

func mark_status(hardened bool) {
	key, _, err := registry.CreateKey(registry.CURRENT_USER, harden_key_path, registry.WRITE)
	if err != nil {
		panic(err)
	}

	if hardened {
		key.SetDWordValue("Harden", 1)
	} else {
		key.SetDWordValue("Harden", 0)
	}
}

func harden_all() {
	trigger_all(true)
	mark_status(true)

	walk.MsgBox(window, "Done!", "I have hardened all risky features!\nFor all changes to take effect please restart Windows.", walk.MsgBoxIconInformation)
	os.Exit(0)
}

func restore_all() {
	trigger_all(false)
	mark_status(false)

	walk.MsgBox(window, "Done!", "I have restored all risky features!\nFor all changes to take effect please restart Windows.", walk.MsgBoxIconExclamation)
	os.Exit(0)
}

func trigger_all(harden bool) {
	trigger_wsh(harden)
	trigger_ole(harden)
	trigger_macro(harden)
	trigger_activex(harden)
	trigger_pdf_js(harden)
	trigger_pdf_objects(harden)
	trigger_autorun(harden)
	trigger_powershell(harden)
	trigger_uac(harden)
	trigger_fileassoc(harden)
	progress.SetValue(100)
}

func main() {
	var label_text, button_text, events_text string
	var button_func func()

	if check_status() == false {
		button_text = "Harden!"
		button_func = harden_all
		label_text = "Ready to harden some features of your system?"
	} else {
		button_text = "Restore..."
		button_func = restore_all
		label_text = "We have already hardened some risky features, do you want to restore them?"
	}

	MainWindow{
		AssignTo: &window,
		Title:    "Harden - Security Without Borders",
		MinSize:  Size{400, 300},
		Layout:   VBox{},
		Children: []Widget{
			Label{Text: label_text},
			PushButton{
				Text:      button_text,
				OnClicked: button_func,
			},
			ProgressBar{
				AssignTo: &progress,
			},
			TextEdit{
				AssignTo: &events,
				Text:     events_text,
				ReadOnly: true,
			},
		},
	}.Create()

	window.Run()
}