230 lines
8 KiB
Go
230 lines
8 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/cwhub"
|
|
"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/require"
|
|
)
|
|
|
|
func backupHub(dirPath string) error {
|
|
var err error
|
|
var itemDirectory string
|
|
var upstreamParsers []string
|
|
|
|
for _, itemType := range cwhub.ItemTypes {
|
|
clog := log.WithFields(log.Fields{
|
|
"type": itemType,
|
|
})
|
|
itemMap := cwhub.GetItemMap(itemType)
|
|
if itemMap == nil {
|
|
clog.Infof("No %s to backup.", itemType)
|
|
continue
|
|
}
|
|
itemDirectory = fmt.Sprintf("%s/%s/", dirPath, itemType)
|
|
if err := os.MkdirAll(itemDirectory, os.ModePerm); err != nil {
|
|
return fmt.Errorf("error while creating %s : %s", itemDirectory, err)
|
|
}
|
|
upstreamParsers = []string{}
|
|
for k, v := range itemMap {
|
|
clog = clog.WithFields(log.Fields{
|
|
"file": v.Name,
|
|
})
|
|
if !v.Installed { //only backup installed ones
|
|
clog.Debugf("[%s] : not installed", k)
|
|
continue
|
|
}
|
|
|
|
//for the local/tainted ones, we back up the full file
|
|
if v.Tainted || v.Local || !v.UpToDate {
|
|
//we need to back up stages for parsers
|
|
if itemType == cwhub.PARSERS || itemType == cwhub.PARSERS_OVFLW {
|
|
fstagedir := fmt.Sprintf("%s%s", itemDirectory, v.Stage)
|
|
if err := os.MkdirAll(fstagedir, os.ModePerm); err != nil {
|
|
return fmt.Errorf("error while creating stage dir %s : %s", fstagedir, err)
|
|
}
|
|
}
|
|
clog.Debugf("[%s] : backuping file (tainted:%t local:%t up-to-date:%t)", k, v.Tainted, v.Local, v.UpToDate)
|
|
tfile := fmt.Sprintf("%s%s/%s", itemDirectory, v.Stage, v.FileName)
|
|
if err = CopyFile(v.LocalPath, tfile); err != nil {
|
|
return fmt.Errorf("failed copy %s %s to %s : %s", itemType, v.LocalPath, tfile, err)
|
|
}
|
|
clog.Infof("local/tainted saved %s to %s", v.LocalPath, tfile)
|
|
continue
|
|
}
|
|
clog.Debugf("[%s] : from hub, just backup name (up-to-date:%t)", k, v.UpToDate)
|
|
clog.Infof("saving, version:%s, up-to-date:%t", v.Version, v.UpToDate)
|
|
upstreamParsers = append(upstreamParsers, v.Name)
|
|
}
|
|
//write the upstream items
|
|
upstreamParsersFname := fmt.Sprintf("%s/upstream-%s.json", itemDirectory, itemType)
|
|
upstreamParsersContent, err := json.MarshalIndent(upstreamParsers, "", " ")
|
|
if err != nil {
|
|
return fmt.Errorf("failed marshaling upstream parsers : %s", err)
|
|
}
|
|
err = os.WriteFile(upstreamParsersFname, upstreamParsersContent, 0644)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to write to %s %s : %s", itemType, upstreamParsersFname, err)
|
|
}
|
|
clog.Infof("Wrote %d entries for %s to %s", len(upstreamParsers), itemType, upstreamParsersFname)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
/*
|
|
Backup crowdsec configurations to directory <dirPath>:
|
|
|
|
- Main config (config.yaml)
|
|
- Profiles config (profiles.yaml)
|
|
- Simulation config (simulation.yaml)
|
|
- Backup of API credentials (local API and online API)
|
|
- List of scenarios, parsers, postoverflows and collections that are up-to-date
|
|
- Tainted/local/out-of-date scenarios, parsers, postoverflows and collections
|
|
- Acquisition files (acquis.yaml, acquis.d/*.yaml)
|
|
*/
|
|
func backupConfigToDirectory(dirPath string) error {
|
|
var err error
|
|
|
|
if dirPath == "" {
|
|
return fmt.Errorf("directory path can't be empty")
|
|
}
|
|
|
|
log.Infof("Starting configuration backup")
|
|
|
|
/*if parent directory doesn't exist, bail out. create final dir with Mkdir*/
|
|
parentDir := filepath.Dir(dirPath)
|
|
if _, err := os.Stat(parentDir); err != nil {
|
|
return fmt.Errorf("while checking parent directory %s existence: %w", parentDir, err)
|
|
}
|
|
|
|
if err = os.Mkdir(dirPath, 0o700); err != nil {
|
|
return fmt.Errorf("while creating %s: %w", dirPath, err)
|
|
}
|
|
|
|
if csConfig.ConfigPaths.SimulationFilePath != "" {
|
|
backupSimulation := filepath.Join(dirPath, "simulation.yaml")
|
|
if err = CopyFile(csConfig.ConfigPaths.SimulationFilePath, backupSimulation); err != nil {
|
|
return fmt.Errorf("failed copy %s to %s: %w", csConfig.ConfigPaths.SimulationFilePath, backupSimulation, err)
|
|
}
|
|
|
|
log.Infof("Saved simulation to %s", backupSimulation)
|
|
}
|
|
|
|
/*
|
|
- backup AcquisitionFilePath
|
|
- backup the other files of acquisition directory
|
|
*/
|
|
if csConfig.Crowdsec != nil && csConfig.Crowdsec.AcquisitionFilePath != "" {
|
|
backupAcquisition := filepath.Join(dirPath, "acquis.yaml")
|
|
if err = CopyFile(csConfig.Crowdsec.AcquisitionFilePath, backupAcquisition); err != nil {
|
|
return fmt.Errorf("failed copy %s to %s: %s", csConfig.Crowdsec.AcquisitionFilePath, backupAcquisition, err)
|
|
}
|
|
}
|
|
|
|
acquisBackupDir := filepath.Join(dirPath, "acquis")
|
|
if err = os.Mkdir(acquisBackupDir, 0o700); err != nil {
|
|
return fmt.Errorf("error while creating %s: %s", acquisBackupDir, err)
|
|
}
|
|
|
|
if csConfig.Crowdsec != nil && len(csConfig.Crowdsec.AcquisitionFiles) > 0 {
|
|
for _, acquisFile := range csConfig.Crowdsec.AcquisitionFiles {
|
|
/*if it was the default one, it was already backup'ed*/
|
|
if csConfig.Crowdsec.AcquisitionFilePath == acquisFile {
|
|
continue
|
|
}
|
|
|
|
targetFname, err := filepath.Abs(filepath.Join(acquisBackupDir, filepath.Base(acquisFile)))
|
|
if err != nil {
|
|
return fmt.Errorf("while saving %s to %s: %w", acquisFile, acquisBackupDir, err)
|
|
}
|
|
|
|
if err = CopyFile(acquisFile, targetFname); err != nil {
|
|
return fmt.Errorf("failed copy %s to %s: %w", acquisFile, targetFname, err)
|
|
}
|
|
|
|
log.Infof("Saved acquis %s to %s", acquisFile, targetFname)
|
|
}
|
|
}
|
|
|
|
if ConfigFilePath != "" {
|
|
backupMain := fmt.Sprintf("%s/config.yaml", dirPath)
|
|
if err = CopyFile(ConfigFilePath, backupMain); err != nil {
|
|
return fmt.Errorf("failed copy %s to %s: %s", ConfigFilePath, backupMain, err)
|
|
}
|
|
|
|
log.Infof("Saved default yaml to %s", backupMain)
|
|
}
|
|
|
|
if csConfig.API != nil && csConfig.API.Server != nil && csConfig.API.Server.OnlineClient != nil && csConfig.API.Server.OnlineClient.CredentialsFilePath != "" {
|
|
backupCAPICreds := fmt.Sprintf("%s/online_api_credentials.yaml", dirPath)
|
|
if err = CopyFile(csConfig.API.Server.OnlineClient.CredentialsFilePath, backupCAPICreds); err != nil {
|
|
return fmt.Errorf("failed copy %s to %s: %s", csConfig.API.Server.OnlineClient.CredentialsFilePath, backupCAPICreds, err)
|
|
}
|
|
|
|
log.Infof("Saved online API credentials to %s", backupCAPICreds)
|
|
}
|
|
|
|
if csConfig.API != nil && csConfig.API.Client != nil && csConfig.API.Client.CredentialsFilePath != "" {
|
|
backupLAPICreds := fmt.Sprintf("%s/local_api_credentials.yaml", dirPath)
|
|
if err = CopyFile(csConfig.API.Client.CredentialsFilePath, backupLAPICreds); err != nil {
|
|
return fmt.Errorf("failed copy %s to %s: %s", csConfig.API.Client.CredentialsFilePath, backupLAPICreds, err)
|
|
}
|
|
|
|
log.Infof("Saved local API credentials to %s", backupLAPICreds)
|
|
}
|
|
|
|
if csConfig.API != nil && csConfig.API.Server != nil && csConfig.API.Server.ProfilesPath != "" {
|
|
backupProfiles := fmt.Sprintf("%s/profiles.yaml", dirPath)
|
|
if err = CopyFile(csConfig.API.Server.ProfilesPath, backupProfiles); err != nil {
|
|
return fmt.Errorf("failed copy %s to %s: %s", csConfig.API.Server.ProfilesPath, backupProfiles, err)
|
|
}
|
|
|
|
log.Infof("Saved profiles to %s", backupProfiles)
|
|
}
|
|
|
|
if err = backupHub(dirPath); err != nil {
|
|
return fmt.Errorf("failed to backup hub config: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func runConfigBackup(cmd *cobra.Command, args []string) error {
|
|
if err := require.Hub(csConfig); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := backupConfigToDirectory(args[0]); err != nil {
|
|
return fmt.Errorf("failed to backup config: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func NewConfigBackupCmd() *cobra.Command {
|
|
cmdConfigBackup := &cobra.Command{
|
|
Use: `backup "directory"`,
|
|
Short: "Backup current config",
|
|
Long: `Backup the current crowdsec configuration including :
|
|
|
|
- Main config (config.yaml)
|
|
- Simulation config (simulation.yaml)
|
|
- Profiles config (profiles.yaml)
|
|
- List of scenarios, parsers, postoverflows and collections that are up-to-date
|
|
- Tainted/local/out-of-date scenarios, parsers, postoverflows and collections
|
|
- Backup of API credentials (local API and online API)`,
|
|
Example: `cscli config backup ./my-backup`,
|
|
Args: cobra.ExactArgs(1),
|
|
DisableAutoGenTag: true,
|
|
RunE: runConfigBackup,
|
|
}
|
|
|
|
return cmdConfigBackup
|
|
}
|