mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2025-10-21 21:07:30 +08:00
88 lines
2.7 KiB
Go
88 lines
2.7 KiB
Go
package clam
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/1Panel-dev/1Panel/agent/app/model"
|
|
"github.com/1Panel-dev/1Panel/agent/app/repo"
|
|
"github.com/1Panel-dev/1Panel/agent/app/task"
|
|
"github.com/1Panel-dev/1Panel/agent/constant"
|
|
"github.com/1Panel-dev/1Panel/agent/global"
|
|
"github.com/1Panel-dev/1Panel/agent/i18n"
|
|
"github.com/1Panel-dev/1Panel/agent/utils/cmd"
|
|
"github.com/1Panel-dev/1Panel/agent/utils/systemctl"
|
|
"github.com/robfig/cron/v3"
|
|
)
|
|
|
|
func AddScanTask(taskItem *task.Task, clam model.Clam, timeNow string) {
|
|
taskItem.AddSubTask(i18n.GetWithName("Clamscan", clam.Path), func(t *task.Task) error {
|
|
strategy := ""
|
|
switch clam.InfectedStrategy {
|
|
case "remove":
|
|
strategy = "--remove"
|
|
case "move", "copy":
|
|
dir := path.Join(clam.InfectedDir, "1panel-infected", clam.Name, timeNow)
|
|
taskItem.Log("infected dir: " + dir)
|
|
if _, err := os.Stat(dir); err != nil {
|
|
_ = os.MkdirAll(dir, os.ModePerm)
|
|
}
|
|
strategy = fmt.Sprintf("--%s=%s", clam.InfectedStrategy, dir)
|
|
}
|
|
taskItem.Logf("clamdscan --fdpass %s %s", strategy, clam.Path)
|
|
mgr := cmd.NewCommandMgr(cmd.WithIgnoreExist1(), cmd.WithTimeout(10*time.Hour), cmd.WithTask(*taskItem))
|
|
stdout, err := mgr.RunWithStdoutBashCf("clamdscan --fdpass %s %s", strategy, clam.Path)
|
|
if err != nil {
|
|
return fmt.Errorf("clamdscan failed, stdout: %v, err: %v", stdout, err)
|
|
}
|
|
return nil
|
|
}, nil)
|
|
}
|
|
|
|
func CheckClamIsActive(withCheck bool, clamRepo repo.IClamRepo) bool {
|
|
if withCheck {
|
|
isActive := false
|
|
exist1, _ := systemctl.IsExist(constant.ClamServiceNameCentOs)
|
|
if exist1 {
|
|
isActive, _ = systemctl.IsActive(constant.ClamServiceNameCentOs)
|
|
}
|
|
exist2, _ := systemctl.IsExist(constant.ClamServiceNameUbuntu)
|
|
if exist2 {
|
|
isActive, _ = systemctl.IsActive(constant.ClamServiceNameUbuntu)
|
|
}
|
|
if isActive {
|
|
return true
|
|
}
|
|
}
|
|
clams, _ := clamRepo.List(repo.WithByStatus(constant.StatusEnable))
|
|
for i := 0; i < len(clams); i++ {
|
|
global.Cron.Remove(cron.EntryID(clams[i].EntryID))
|
|
_ = clamRepo.Update(clams[i].ID, map[string]interface{}{"status": constant.StatusDisable, "entry_id": 0})
|
|
}
|
|
return false
|
|
}
|
|
|
|
func AnalysisFromLog(pathItem string, record *model.ClamRecord) {
|
|
file, err := os.ReadFile(pathItem)
|
|
if err != nil {
|
|
return
|
|
}
|
|
lines := strings.Split(string(file), "\n")
|
|
for _, line := range lines {
|
|
if len(line) < 20 {
|
|
continue
|
|
}
|
|
line = line[20:]
|
|
switch {
|
|
case strings.HasPrefix(line, "Infected files: "):
|
|
record.InfectedFiles = strings.TrimPrefix(line, "Infected files: ")
|
|
case strings.HasPrefix(line, "Total errors: "):
|
|
record.TotalError = strings.TrimPrefix(line, "Total errors: ")
|
|
case strings.HasPrefix(line, "Time: "):
|
|
record.ScanTime = strings.TrimPrefix(line, "Time: ")
|
|
}
|
|
}
|
|
}
|