Binbim
b8788c239e
MCP Docker / build-and-push (push) Has been cancelled
- Remove telemetry: disabled all phone-home (installation notify, false positives, behavior tracking, upgrade tips) - Version branding: removed 'ce-' prefix, product name changed to '长亭雷池 WAF' - Detection engine enhancement: strict preset enables all 17 modules with aggressive configs - Multi-user RBAC: admin/operator/viewer roles with password + TOTP 2FA - Rate limiting: nginx limit_req per website (RPS, burst, action) - JS Challenge: browser fingerprint verification page with cookie-based bypass - Security fixes: removed InsecureSkipVerify, hardcoded credentials, NO_AUTH backdoor
42 lines
966 B
Go
42 lines
966 B
Go
package tools
|
|
|
|
import (
|
|
"github.com/chaitin/SafeLine/mcp_server/pkg/logger"
|
|
"github.com/chaitin/SafeLine/mcp_server/pkg/mcp"
|
|
)
|
|
|
|
// By deferring the concretization of generic types to the Register method,
|
|
// we avoid type inference issues.
|
|
|
|
// Each Tool is wrapped in a toolWrapper that knows its concrete type,
|
|
// allowing correct passing of generic parameters during registration.
|
|
type ToolWrapper interface {
|
|
Register(s *mcp.MCPServer) error
|
|
}
|
|
|
|
var (
|
|
tools = []ToolWrapper{}
|
|
)
|
|
|
|
func AppendTool[T any, R any](tool ...mcp.Tool[T, R]) {
|
|
for _, t := range tool {
|
|
tools = append(tools, &toolWrapper[T, R]{tool: t})
|
|
}
|
|
}
|
|
|
|
func Tools() []ToolWrapper {
|
|
return tools
|
|
}
|
|
|
|
type toolWrapper[T any, R any] struct {
|
|
tool mcp.Tool[T, R]
|
|
}
|
|
|
|
func (w *toolWrapper[T, R]) Register(s *mcp.MCPServer) error {
|
|
logger.Info("Registering tool",
|
|
logger.String("name", w.tool.Name()),
|
|
logger.String("description", w.tool.Description()),
|
|
)
|
|
return mcp.RegisterTool(s, w.tool)
|
|
}
|