Add option to create a new teamserver database

teamserver/cmd/cmd.go

@@ -32,6 +32,7 @@ func init() {
 	// server flags
 	CobraServer.Flags().SortFlags = false
 	CobraServer.Flags().StringVarP(&flags.Server.Profile, "profile", "", "", "set havoc teamserver profile")
+	CobraServer.Flags().StringVarP(&flags.Server.Database, "database", "", "data/teamserver.db", "set havoc teamserver database")
 	CobraServer.Flags().BoolVarP(&flags.Server.Debug, "debug", "", false, "enable debug mode")
 	CobraServer.Flags().BoolVarP(&flags.Server.DebugDev, "debug-dev", "", false, "enable debug mode for developers (compiles the agent with the debug mode/macro enabled)")
 	CobraServer.Flags().BoolVarP(&flags.Server.SendLogs, "send-logs", "", false, "the agent will send logs over http(s) to the teamserver")

teamserver/cmd/server.go

@@ -33,7 +33,10 @@ var CobraServer = &cobra.Command{
 			}
 			os.Exit(0)
 		}
-
+
+		if flags.Server.Database != "" {
+			DatabasePath = (flags.Server.Database)
+		}
 		Server = server.NewTeamserver(DatabasePath)
 		Server.SetServerFlags(flags)
 

teamserver/cmd/server/types.go

@@ -42,6 +42,7 @@ type serverFlags struct {
 	Host string
 	Port string
 
+	Database string
 	Profile  string
 	Verbose  bool
 	Debug    bool