mirror of
https://github.com/shadow1ng/fscan.git
synced 2025-07-13 12:52:44 +08:00
95 lines
3.2 KiB
Go
95 lines
3.2 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"time"
|
||
|
||
"github.com/shadow1ng/fscan/Common"
|
||
"github.com/shadow1ng/fscan/Core"
|
||
pb "github.com/shadow1ng/fscan/RPC/lib"
|
||
)
|
||
|
||
// FscanService 实现了 pb.FscanServiceServer 接口，用于提供扫描相关的服务。
|
||
type FscanService struct {
|
||
pb.UnimplementedFscanServiceServer
|
||
}
|
||
|
||
// StartScan 用于启动扫描任务。
|
||
// 参数：
|
||
// - ctx：请求上下文，用于控制超时、取消等操作。
|
||
// - req：StartScanRequest，包括扫描目标、端口、模式等参数。
|
||
// 返回值：
|
||
// - StartScanResponse：包含任务 ID 和提示信息。
|
||
// - error：执行中出现的错误信息。
|
||
func (s *FscanService) StartScan(ctx context.Context, req *pb.StartScanRequest) (*pb.StartScanResponse, error) {
|
||
Common.LogDebug("接收到扫描请求，目标: " + req.Arg + ", " + req.Secret)
|
||
// TODO: 在此处实现实际的扫描逻辑，例如调用扫描器、创建任务、存储任务状态等。
|
||
// 可以异步执行扫描逻辑，并生成一个唯一的 taskID 进行标识。
|
||
var info Common.HostInfo
|
||
if err := Common.FlagFromRemote(&info, req.Arg); err != nil {
|
||
return nil, err
|
||
}
|
||
if err := Common.Parse(&info); err != nil {
|
||
return nil, err
|
||
}
|
||
Common.LogDebug("解析参数成功，目标: " + info.Host + ", " + info.Ports)
|
||
go Core.Scan(info) // 建议异步执行
|
||
|
||
return &pb.StartScanResponse{
|
||
TaskId: "task_123456", // TODO: 返回真实生成的 taskID
|
||
Message: "扫描任务已启动",
|
||
}, nil
|
||
}
|
||
|
||
// GetScanResults 用于获取指定任务 ID 的扫描结果。
|
||
// 参数：
|
||
// - ctx：请求上下文。
|
||
// - req：TaskResultsRequest，包含任务 ID。
|
||
// 返回值：
|
||
// - TaskResultsResponse：包含结果列表、任务状态等信息。
|
||
// - error：执行中出现的错误信息。
|
||
func (s *FscanService) GetScanResults(ctx context.Context, req *pb.TaskResultsRequest) (*pb.TaskResultsResponse, error) {
|
||
// TODO: 实现根据任务 ID 查询任务结果，可以从缓存、数据库或临时文件中获取。
|
||
// 此处为模拟数据
|
||
|
||
result := &pb.ScanResult{
|
||
Time: time.Now().Format(time.RFC3339),
|
||
Type: "port",
|
||
Target: "192.168.1.1:80",
|
||
Status: "open",
|
||
DetailsJson: `{"banner":"nginx"}`,
|
||
}
|
||
|
||
return &pb.TaskResultsResponse{
|
||
TaskId: req.TaskId,
|
||
Results: []*pb.ScanResult{result},
|
||
Finished: true, // TODO: 判断任务是否真正完成
|
||
}, nil
|
||
}
|
||
|
||
// StreamScanResults 用于通过流式返回任务扫描结果，适合长时间扫描过程。
|
||
// 参数：
|
||
// - req：TaskResultsRequest，包含任务 ID。
|
||
// - stream：用于向客户端持续推送结果。
|
||
// 返回值：
|
||
// - error：执行中出现的错误信息。
|
||
func (s *FscanService) StreamScanResults(req *pb.TaskResultsRequest, stream pb.FscanService_StreamScanResultsServer) error {
|
||
// TODO: 根据任务 ID 逐步查询任务结果，并通过 stream.Send 发送给客户端。
|
||
// 可以监听任务进度，逐步推送最新结果。
|
||
|
||
for i := 0; i < 5; i++ {
|
||
result := &pb.ScanResult{
|
||
Time: time.Now().Format(time.RFC3339),
|
||
Type: "vuln",
|
||
Target: "192.168.1.1",
|
||
Status: "found",
|
||
DetailsJson: `{"vuln":"CVE-2021-12345"}`,
|
||
}
|
||
if err := stream.Send(result); err != nil {
|
||
return err
|
||
}
|
||
time.Sleep(1 * time.Second) // 模拟异步推送过程
|
||
}
|
||
return nil
|
||
}
|