1 files changed, 59 insertions, 0 deletions
diff --git a/clover/cloverctl/src/cloverctl/cmd/create_idsrules.go b/clover/cloverctl/src/cloverctl/cmd/create_idsrules.go
new file mode 100644
index 0000000..6e59297
--- /dev/null
+++ b/clover/cloverctl/src/cloverctl/cmd/create_idsrules.go
@@ -0,0 +1,59 @@
+// Copyright (c) Authors of Clover
+//
+// All rights reserved. This program and the accompanying materials
+// are made available under the terms of the Apache License, Version 2.0
+// which accompanies this distribution, and is available at
+// http://www.apache.org/licenses/LICENSE-2.0
+
+package cmd
+
+import (
+    "fmt"
+    "os"
+    "io/ioutil"
+    "gopkg.in/resty.v1"
+    "github.com/ghodss/yaml"
+    "github.com/spf13/cobra"
+)
+
+
+var idsrulesCmd = &cobra.Command{
+    Use:   "idsrules",
+    Short: "Create one or many snort IDS rules from yaml file",
+    Long: ``,
+    Run: func(cmd *cobra.Command, args []string) {
+        createIDSRules()
+    },
+}
+
+func init() {
+    createCmd.AddCommand(idsrulesCmd)
+    idsrulesCmd.Flags().StringVarP(&cloverFile, "file", "f", "",
+                                   "Input yaml file to add IDS rules")
+    idsrulesCmd.MarkFlagRequired("file")
+
+}
+
+func createIDSRules() {
+    checkControllerIP()
+    url := controllerIP + "/snort/addrule"
+    in, err := ioutil.ReadFile(cloverFile)
+    if err != nil {
+        fmt.Println("Please specify a valid rule definition yaml file")
+        os.Exit(1)
+    }
+    out_json, err := yaml.YAMLToJSON(in)
+    if err != nil {
+        fmt.Printf("Invalid yaml: %v\n", err)
+        os.Exit(1)
+    }
+    resp, err := resty.R().
+    SetHeader("Content-Type", "application/json").
+    SetBody(out_json).
+    Post(url)
+    if err != nil {
+        fmt.Printf("Cannot connect to controller: %v\n", err)
+        os.Exit(1)
+    }
+    fmt.Printf("\n%v\n", resp)
+}