yak

package
v0.0.0-...-9a26220 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Apr 6, 2025 License: AGPL-3.0 Imports: 89 Imported by: 0

Documentation

Index

Constants

View Source 
const (
	// func mirrorHTTPFlow(isHttps, url, request, response, body)
	//     mirror hijacked by filtered http flows
	HOOK_MirrorFilteredHTTPFlow = "mirrorFilteredHTTPFlow"

	// func mirrorHTTPFlow(isHttps, url, request, response, body)
	//     mirror hijacked all
	HOOK_MirrorHTTPFlow = "mirrorHTTPFlow"

	// func mirrorNewWebsite(isHttps, url, request, response, body)
	HOOK_MirrorNewWebsite = "mirrorNewWebsite" // schema + addr

	// func mirrorNewWebsitePath(isHttps, url, request, response, body)
	HOOK_MirrorNewWebsitePath = "mirrorNewWebsitePath" // schema + addr + path (remove params)

	// func mirrorNewWebsitePathParams(isHttps, url, request, response, body)
	HOOK_MirrorNewWebsitePathParams = "mirrorNewWebsitePathParams" // schema + addr + path + param_names

	// func hijackHTTPRequest(isHttps, url, request, forward/*func(modified []byte)*/, drop /*func()*/)
	HOOK_HijackHTTPRequest = "hijackHTTPRequest"

	// func hijackHTTPRequest(isHttps, url, response, forward/*func(modified []byte)*/, drop /*func()*/)
	HOOK_HijackHTTPResponse = "hijackHTTPResponse"
	// func hijackHTTPRequest(isHttps, url, request, response, forward/*func(modified []byte)*/, drop /*func()*/)
	HOOK_HijackHTTPResponseEx = "hijackHTTPResponseEx"

	// func hijackSaveHTTPFlow(record *httpFlow, forward func(*httpFlow), drop func()) return (*httpFlow)
	HOOK_hijackSaveHTTPFlow = "hijackSaveHTTPFlow"

	// func handle(r *fp.MatchResult)
	HOOK_PortScanHandle = "handle"

	// func execNuclei(target)
	HOOK_NucleiScanHandle = "execNuclei"

	HOOK_NaslScanHandle           = "execNasl"
	HOOK_LoadNaslScriptByNameFunc = "loadNaslScriptByNameFunc"

	// beforeRequest, afterRequest
	HOOK_BeforeRequest = "beforeRequest"
	HOOK_AfterRequest  = "afterRequest"

	HOOK_CLAER = "clear"
)

Variables

View Source 
var (
	MITMAndPortScanHooks = []string{
		HOOK_MirrorFilteredHTTPFlow,
		HOOK_MirrorHTTPFlow,
		HOOK_MirrorNewWebsite,
		HOOK_MirrorNewWebsitePath,
		HOOK_MirrorNewWebsitePathParams,
		HOOK_CLAER,

		HOOK_HijackHTTPRequest,
		HOOK_HijackHTTPResponse,
		HOOK_HijackHTTPResponseEx,
		HOOK_hijackSaveHTTPFlow,

		HOOK_PortScanHandle,

		HOOK_BeforeRequest,
		HOOK_AfterRequest,
	}
	HotPatchScriptName = "@HotPatchCode"
)
View Source 
var (
	CRYPTO_KEY_SIZE = 16
)
View Source 
var EvalExports = map[string]interface{}{
	"Eval":            QuickEvalWithoutContext,
	"LoadVarFromFile": LoadingVariableFrom,
	"Import":          ImportVarFromFile,
	"IsYakFunc":       yaklib.IsYakFunction,
	"params":          setYakEvalParams,
	"recursive":       setYakBatchImportRecursiveParams,
}
View Source 
var GlobalEvalExports = map[string]interface{}{

	"import": ImportVarFromFile,
}
View Source 
var HooksExports = map[string]interface{}{
	"NewManager":                   NewYakToCallerManager,
	"NewMixPluginCaller":           NewMixPluginCaller,
	"NewMixPluginCallerWithFilter": NewMixPluginCallerWithFilter,
	"RemoveYakitPluginByName":      removeScriptByNameCtx,
	"LoadYakitPluginContext":       loadScriptCtx,
	"LoadYakitPlugin":              loadScript,
	"LoadYakitPluginByName":        loadScriptByName,
	"CallYakitPluginFunc":          CallYakitPluginFunc,
}
View Source 
var SandboxExports = map[string]any{
	"Create":  NewSandbox,
	"library": WithSandbox_ExternalLib,
}
View Source 
var YAK_TEMPLATE_NUCLEI_EXECUTOR string

Functions

func BindYakitPluginContextToEngine 

func BindYakitPluginContextToEngine(nIns *antlr4yak.Engine, pluginContext *YakitPluginContext)

func CallYakitPluginFunc 

func CallYakitPluginFunc(scriptName string, hookName string) (interface{}, error)

func ClearHelper 

func ClearHelper(helper *yakdoc.DocumentHelper)

func DefaultAfterRequestCallArgumentHook 

func DefaultAfterRequestCallArgumentHook(name string, numIn int, args []any) []any

func DefaultBeforeRequestCallArgumentHook 

func DefaultBeforeRequestCallArgumentHook(name string, numIn int, args []any) []any

func DocumentHelperWithVerboseInfo 

func DocumentHelperWithVerboseInfo(funcMap map[string]any) *yakdoc.DocumentHelper

func EngineToDocumentHelperWithVerboseInfo 

func EngineToDocumentHelperWithVerboseInfo(engine *antlr4yak.Engine) *yakdoc.DocumentHelper

func EngineToLibDocuments 

func EngineToLibDocuments(engine *antlr4yak.Engine) []yakdocument.LibDoc

! 老接口

func Execute 

func Execute(code string, params ...map[string]any) (*antlr4yak.Engine, error)

func FeedbackFactory 

func FeedbackFactory(db *gorm.DB, caller func(result *ypb.ExecResult) error, saveToDb bool, yakScriptName string) func(i interface{}, items ...interface{})

func Fuzz_WithHotPatch 

func Fuzz_WithHotPatch(ctx context.Context, code string) mutate.FuzzConfigOpt

func GetDeprecatedFunctionDecls 

func GetDeprecatedFunctionDecls(helper *yakdoc.DocumentHelper) []*yakdoc.DeprecateFunction

func GetHookCliApp 

func GetHookCliApp(tempArg []string) *cli.CliApp

func GetInterfaceDocumentFromAST 

func GetInterfaceDocumentFromAST(pkg *ast.Package, interfaceName string) map[string]string

func GetMethodFuncDeclFromAST 

func GetMethodFuncDeclFromAST(pkg *ast.Package, libName, structName, methodName, yakFuncName string, fset *token.FileSet, funcRefType reflect.Type) *yakdoc.FuncDecl

func ImportVarFromFile 

func ImportVarFromFile(file string, exportsName string) (interface{}, error)

func ImportVarFromScript 

func ImportVarFromScript(engine *antlr4yak.Engine, script string, exportsName string) (interface{}, error)

func ImportVarFromYakFile 

func ImportVarFromYakFile(path string, exportsName string) (interface{}, error)

func InitYaklangLib 

func InitYaklangLib()

func IsSameTypeName 

func IsSameTypeName(typName1, typName2 string) bool

func LoadingVariableFrom 

func LoadingVariableFrom(path string, exportsName string, opts ...yakEvalConfigOpt) ([]*yakVariable, error)

func MutateHookCaller 

func MutateHookCaller(ctx context.Context, raw string, caller YakitCallerIf, params ...*ypb.ExecParamItem) (func(https bool, originReq []byte, req []byte) []byte, func(https bool, originReq []byte, req []byte, originRsp []byte, rsp []byte) []byte, func([]byte, []byte, map[string]string) map[string]string)

func MutateWithParamsGetter 

func MutateWithParamsGetter(raw string) func() *mutate.RegexpMutateCondition

func MutateWithYaklang 

func MutateWithYaklang(raw string) *mutate.RegexpMutateCondition

func NewFetchFuncFromSrcCodeConfig 

func NewFetchFuncFromSrcCodeConfig() *fetchFuncFromSrcCodeConfig

func NewYakitFetchFuncFromSrcCodeConfig 

func NewYakitFetchFuncFromSrcCodeConfig() *fetchFuncFromSrcCodeConfig

func QuickEvalWithoutContext 

func QuickEvalWithoutContext(i interface{}) error

func SetNaslExports 

func SetNaslExports(lib map[string]interface{})

func StaticAnalyze 

func StaticAnalyze(code string, opts ...StaticAnalyzeOption) []*result.StaticAnalyzeResult

func WithFetchCallArgumentHook 

func WithFetchCallArgumentHook(name string, hook callArgumentHookFunc) fetchFuncFromSrcCodeOptions

func WithFetchCode 

func WithFetchCode(code string) fetchFuncFromSrcCodeOptions

func WithFetchEngineHook 

func WithFetchEngineHook(hook func(e *antlr4yak.Engine) error) fetchFuncFromSrcCodeOptions

func WithFetchFunctionNames 

func WithFetchFunctionNames(names ...string) fetchFuncFromSrcCodeOptions

func WithFetchScript 

func WithFetchScript(script *schema.YakScript) fetchFuncFromSrcCodeOptions

Types

type CallConfig 

type CallConfig struct {
	// contains filtered or unexported fields
}

func NewCallConfig 

func NewCallConfig() *CallConfig

type CallOpt 

type CallOpt func(*CallConfig)

func WithCallConfigCallback 

func WithCallConfigCallback(callback func()) CallOpt

func WithCallConfigForceSync 

func WithCallConfigForceSync(forceSync bool) CallOpt

func WithCallConfigItemFuncs 

func WithCallConfigItemFuncs(itemFuncs ...func() interface{}) CallOpt

func WithCallConfigItems 

func WithCallConfigItems(items ...interface{}) CallOpt

func WithCallConfigPluginId 

func WithCallConfigPluginId(pluginId string) CallOpt

func WithCallConfigRuntimeCtx 

func WithCallConfigRuntimeCtx(ctx context.Context) CallOpt

type Caller 

type Caller struct {
	Core    *YakFunctionCaller
	Hash    string
	Id      string
	Verbose string
	Engine  *antlr4yak.Engine
}

type CallerHookDescription 

type CallerHookDescription struct {
	// 这两个是
	YakScriptId   string
	YakScriptName string
	VerboseName   string
}

type CallerHooks 

type CallerHooks struct {
	HookName string

	Hooks []*CallerHookDescription
}

type EmbedFieldTypeAndMethod 

type EmbedFieldTypeAndMethod struct {
	FieldType reflect.Type
	Method    reflect.Method
}

type InstanceMethodHandler 

type InstanceMethodHandler struct {
	// contains filtered or unexported fields
}

type LibFuncWithFrameType 

type LibFuncWithFrameType func(*yakvm.Frame) interface{}

type MixPluginCaller 

type MixPluginCaller struct {
	// contains filtered or unexported fields
}

func NewMixPluginCaller 

func NewMixPluginCaller() (*MixPluginCaller, error)

func NewMixPluginCallerWithFilter 

func NewMixPluginCallerWithFilter(webFilter filter.Filterable) (*MixPluginCaller, error)

func (*MixPluginCaller) CallAfterRequestWithCtx 

func (m *MixPluginCaller) CallAfterRequestWithCtx(
	runtimeCtx context.Context,
	isHttps bool, u string,
	originReq, req []byte,
	originRsp, rsp []byte,
) []byte

func (*MixPluginCaller) CallBeforeRequestWithCtx 

func (m *MixPluginCaller) CallBeforeRequestWithCtx(
	runtimeCtx context.Context,
	isHttps bool, u string,
	originReq, req []byte,
) []byte

func (*MixPluginCaller) CallHijackRequest 

func (m *MixPluginCaller) CallHijackRequest(
	isHttps bool, u string, getRequest func() interface{},
	reject func() interface{},
	drop func() interface{},
)

func (*MixPluginCaller) CallHijackRequestWithCtx 

func (m *MixPluginCaller) CallHijackRequestWithCtx(
	runtimeCtx context.Context,
	isHttps bool, u string, getRequest func() interface{},
	reject func() interface{},
	drop func() interface{},
)

func (*MixPluginCaller) CallHijackResponse 

func (m *MixPluginCaller) CallHijackResponse(
	isHttps bool, u string, getResponse,
	reject, drop func() interface{},
)

func (*MixPluginCaller) CallHijackResponseEx 

func (m *MixPluginCaller) CallHijackResponseEx(
	isHttps bool, u string, getRequest, getResponse,
	reject, drop func() interface{},
)

func (*MixPluginCaller) CallHijackResponseExWithCtx 

func (m *MixPluginCaller) CallHijackResponseExWithCtx(
	runtimeCtx context.Context,
	isHttps bool, u string, getRequest, getResponse,
	reject, drop func() interface{},
)

func (*MixPluginCaller) CallHijackResponseWithCtx 

func (m *MixPluginCaller) CallHijackResponseWithCtx(
	runtimeCtx context.Context,
	isHttps bool, u string, getResponse,
	reject, drop func() interface{},
)

func (*MixPluginCaller) FeedbackOrdinary 

func (c *MixPluginCaller) FeedbackOrdinary(i interface{})

func (*MixPluginCaller) GetFingerprintMatcher 

func (m *MixPluginCaller) GetFingerprintMatcher() *fp.Matcher

func (*MixPluginCaller) GetNativeCaller 

func (m *MixPluginCaller) GetNativeCaller() *YakToCallerManager

func (*MixPluginCaller) HandleServiceScanResult 

func (m *MixPluginCaller) HandleServiceScanResult(r *fp.MatchResult)

func (*MixPluginCaller) HijackSaveHTTPFlow 

func (m *MixPluginCaller) HijackSaveHTTPFlow(flow *schema.HTTPFlow, reject func(httpFlow *schema.HTTPFlow), drop func())

func (*MixPluginCaller) HijackSaveHTTPFlowEx 

func (m *MixPluginCaller) HijackSaveHTTPFlowEx(runtimeCtx context.Context, flow *schema.HTTPFlow, callback func(), reject func(httpFlow *schema.HTTPFlow), drop func())

func (*MixPluginCaller) IsPassed 

func (c *MixPluginCaller) IsPassed(target string) bool

func (*MixPluginCaller) LastErr 

func (m *MixPluginCaller) LastErr() error

func (*MixPluginCaller) LoadHotPatch 

func (c *MixPluginCaller) LoadHotPatch(ctx context.Context, params []*ypb.ExecParamItem, code string) error

func (*MixPluginCaller) LoadPlugin 

func (m *MixPluginCaller) LoadPlugin(scriptName string, params ...*ypb.ExecParamItem) error

func (*MixPluginCaller) LoadPluginByName 

func (m *MixPluginCaller) LoadPluginByName(ctx context.Context, name string, params []*ypb.ExecParamItem, codes ...string) error

LoadPluginByName 基于脚本名加载插件，如果没有指定代码，则从数据库中加载，如果指定了代码，则默认视为mitm插件执行

func (*MixPluginCaller) LoadPluginEx 

func (m *MixPluginCaller) LoadPluginEx(ctx context.Context, script *schema.YakScript, params ...*ypb.ExecParamItem) error

func (*MixPluginCaller) MirrorHTTPFlow 

func (m *MixPluginCaller) MirrorHTTPFlow(
	isHttps bool, u string, req, rsp, body []byte,
	filters ...bool,
)

func (*MixPluginCaller) MirrorHTTPFlowEx 

func (m *MixPluginCaller) MirrorHTTPFlowEx(
	scanPort bool,
	isHttps bool, u string, req, rsp, body []byte,
	filters ...bool,
)

func (*MixPluginCaller) MirrorHTTPFlowExSync 

func (m *MixPluginCaller) MirrorHTTPFlowExSync(
	scanPort bool,
	isHttps bool, u string, req, rsp, body []byte,
	filters ...bool,
)

func (*MixPluginCaller) MirrorHTTPFlowWithCtx 

func (m *MixPluginCaller) MirrorHTTPFlowWithCtx(
	runtimeCtx context.Context,
	isHttps bool, u string, req, rsp, body []byte,
	filters ...bool,
)

func (*MixPluginCaller) ResetFilter 

func (c *MixPluginCaller) ResetFilter()

func (*MixPluginCaller) SetCache 

func (m *MixPluginCaller) SetCache(b bool)

func (*MixPluginCaller) SetCallPluginTimeout 

func (c *MixPluginCaller) SetCallPluginTimeout(i float64)

func (*MixPluginCaller) SetConcurrent 

func (c *MixPluginCaller) SetConcurrent(i int) error

func (*MixPluginCaller) SetCtx 

func (m *MixPluginCaller) SetCtx(ctx context.Context)

func (*MixPluginCaller) SetDividedContext 

func (m *MixPluginCaller) SetDividedContext(b bool)

func (*MixPluginCaller) SetFeedback 

func (m *MixPluginCaller) SetFeedback(i func(i *ypb.ExecResult) error)

func (*MixPluginCaller) SetLoadPluginTimeout 

func (c *MixPluginCaller) SetLoadPluginTimeout(i float64)

func (*MixPluginCaller) SetProxy 

func (m *MixPluginCaller) SetProxy(s string)

func (*MixPluginCaller) SetRuntimeId 

func (m *MixPluginCaller) SetRuntimeId(s string)

func (*MixPluginCaller) Wait 

func (c *MixPluginCaller) Wait()

type PalmScriptEngineHelper 

type PalmScriptEngineHelper struct {
	Libs             map[string]*PalmScriptLib
	BuildInFunctions map[string]*PalmScriptLibFunc
	UserFunctions    map[string]*PalmScriptLibFunc
	Instances        map[string]*PalmScriptLibInstance
}

func EngineToHelper 

func EngineToHelper(engine *antlr4yak.Engine) *PalmScriptEngineHelper

func (*PalmScriptEngineHelper) GetAllLibs 

func (p *PalmScriptEngineHelper) GetAllLibs() []string

func (*PalmScriptEngineHelper) HelpInfo 

func (p *PalmScriptEngineHelper) HelpInfo() string

func (*PalmScriptEngineHelper) LibHelpInfo 

func (p *PalmScriptEngineHelper) LibHelpInfo(name string) string

func (*PalmScriptEngineHelper) ShowHelpInfo 

func (p *PalmScriptEngineHelper) ShowHelpInfo()

func (*PalmScriptEngineHelper) ShowLibHelpInfo 

func (p *PalmScriptEngineHelper) ShowLibHelpInfo(name string)

type PalmScriptLib 

type PalmScriptLib struct {
	Name             string
	Values           map[string]interface{}
	ElementDocs      []string
	FuncElements     []*PalmScriptLibFunc
	InstanceElements []*PalmScriptLibInstance
}

func (*PalmScriptLib) String 

func (p *PalmScriptLib) String() string

type PalmScriptLibFunc 

type PalmScriptLibFunc struct {
	LibName    string
	MethodName string
	Params     []string
	Returns    []string
}

func (*PalmScriptLibFunc) String 

func (p *PalmScriptLibFunc) String() string

type PalmScriptLibInstance 

type PalmScriptLibInstance struct {
	LibName      string
	InstanceName string
	Type         string
	Value        interface{}
}

func (*PalmScriptLibInstance) String 

func (p *PalmScriptLibInstance) String() string

type Sandbox 

type Sandbox struct {
	// contains filtered or unexported fields
}

func NewSandbox 

func NewSandbox(opts ...SandboxOption) *Sandbox

func (*Sandbox) ExecuteAsBoolean 

func (s *Sandbox) ExecuteAsBoolean(code string, vars ...any) (ret bool, err error)

func (*Sandbox) ExecuteAsExpression 

func (s *Sandbox) ExecuteAsExpression(code string, vars ...any) (ret any, err error)

func (Sandbox) ExecuteAsExpressionRaw 

func (s Sandbox) ExecuteAsExpressionRaw(code string, vars map[string]any) (ret any, err error)

type SandboxConfig 

type SandboxConfig struct {
	// contains filtered or unexported fields
}

type SandboxOption 

type SandboxOption func(*SandboxConfig)

func WithSandbox_ExternalLib 

func WithSandbox_ExternalLib(lib map[string]any) SandboxOption

func WithYaklang_Libs 

func WithYaklang_Libs(b bool) SandboxOption

type ScriptEngine 

type ScriptEngine struct {

	// 设定几个 hook
	RegisterLogHook          yaklib.RegisterOutputFuncType
	UnregisterLogHook        yaklib.UnregisterOutputFuncType
	RegisterLogConsoleHook   yaklib.RegisterOutputFuncType
	UnregisterLogConsoleHook yaklib.UnregisterOutputFuncType
	RegisterOutputHook       yaklib.RegisterOutputFuncType
	UnregisterOutputHook     yaklib.UnregisterOutputFuncType
	RegisterFailedHook       yaklib.RegisterOutputFuncType
	UnregisterFailedHook     yaklib.UnregisterOutputFuncType
	RegisterFinishHook       yaklib.RegisterOutputFuncType
	UnregisterFinishHook     yaklib.UnregisterOutputFuncType
	RegisterAlertHook        yaklib.RegisterOutputFuncType
	UnregisterAlertHook      yaklib.UnregisterOutputFuncType
	// contains filtered or unexported fields
}

func NewScriptEngine 

func NewScriptEngine(maxConcurrent int) *ScriptEngine

func NewYakitVirtualClientScriptEngine 

func NewYakitVirtualClientScriptEngine(client *yaklib.YakitClient) *ScriptEngine

func (*ScriptEngine) Compile 

func (e *ScriptEngine) Compile(code string) ([]byte, error)

func (*ScriptEngine) Execute 

func (e *ScriptEngine) Execute(code string) error

func (*ScriptEngine) ExecuteEx 

func (e *ScriptEngine) ExecuteEx(code string, params map[string]interface{}) (*antlr4yak.Engine, error)

func (*ScriptEngine) ExecuteExWithContext 

func (e *ScriptEngine) ExecuteExWithContext(ctx context.Context, code string, params map[string]interface{}) (_ *antlr4yak.Engine, fErr error)

func (*ScriptEngine) ExecuteMain 

func (e *ScriptEngine) ExecuteMain(code string, AbsFile string) error

func (*ScriptEngine) ExecuteMainWithContext 

func (e *ScriptEngine) ExecuteMainWithContext(ctx context.Context, code string, AbsFile string) error

func (*ScriptEngine) ExecuteWithContext 

func (e *ScriptEngine) ExecuteWithContext(ctx context.Context, code string) error

func (*ScriptEngine) ExecuteWithTaskID 

func (e *ScriptEngine) ExecuteWithTaskID(taskId, code string) error

func (*ScriptEngine) ExecuteWithTaskIDAndContext 

func (e *ScriptEngine) ExecuteWithTaskIDAndContext(ctx context.Context, taskId, code string) error

func (*ScriptEngine) ExecuteWithTaskIDAndParams 

func (e *ScriptEngine) ExecuteWithTaskIDAndParams(ctx context.Context, taskId, code string, params map[string]interface{}) error

func (*ScriptEngine) ExecuteWithTemplate 

func (e *ScriptEngine) ExecuteWithTemplate(codeTmp string, i map[string][]string) error

func (*ScriptEngine) ExecuteWithoutCache 

func (e *ScriptEngine) ExecuteWithoutCache(code string, params map[string]interface{}) (*antlr4yak.Engine, error)

func (*ScriptEngine) GetTaskByTaskID 

func (s *ScriptEngine) GetTaskByTaskID(id string) (*Task, error)

func (*ScriptEngine) HookOsExit 

func (e *ScriptEngine) HookOsExit()

func (*ScriptEngine) RegisterEngineHooks 

func (e *ScriptEngine) RegisterEngineHooks(f func(engine *antlr4yak.Engine) error)

func (*ScriptEngine) SaveTask 

func (s *ScriptEngine) SaveTask(task *Task) error

func (*ScriptEngine) SetCryptoKey 

func (s *ScriptEngine) SetCryptoKey(key []byte) error

func (*ScriptEngine) SetDebug 

func (s *ScriptEngine) SetDebug(debug bool)

func (*ScriptEngine) SetDebugCallback 

func (s *ScriptEngine) SetDebugCallback(callback func(*yakvm.Debugger))

func (*ScriptEngine) SetDebugInit 

func (s *ScriptEngine) SetDebugInit(callback func(*yakvm.Debugger))

func (*ScriptEngine) SetYakitClient 

func (e *ScriptEngine) SetYakitClient(client *yaklib.YakitClient)

func (*ScriptEngine) Status 

func (s *ScriptEngine) Status() map[string]*Task

type StaticAnalyzeConfig 

type StaticAnalyzeConfig struct {
	// contains filtered or unexported fields
}

func NewStaticAnalyzeConfig 

func NewStaticAnalyzeConfig() *StaticAnalyzeConfig

type StaticAnalyzeOption 

type StaticAnalyzeOption func(*StaticAnalyzeConfig)

func WithStaticAnalyzeKindAnalyze 

func WithStaticAnalyzeKindAnalyze() StaticAnalyzeOption

func WithStaticAnalyzeKindScore 

func WithStaticAnalyzeKindScore() StaticAnalyzeOption

func WithStaticAnalyzePluginType 

func WithStaticAnalyzePluginType(typ string) StaticAnalyzeOption

type Task 

type Task struct {
	TaskID string
	Code   string

	Output   []string
	Log      []string
	Alert    []string
	Finished []string
	Failed   []string
	// contains filtered or unexported fields
}

func (*Task) IsFinished 

func (t *Task) IsFinished() bool

func (*Task) IsRunning 

func (t *Task) IsRunning() bool

type TextHandlingScript 

type TextHandlingScript struct {
	RuleID         string `json:"rule_id" yaml:"rule_id"`
	MatchingScript string `json:"matching_script" yaml:"matching_script"`
	ParsingScript  string `json:"parsing_script" yaml:"parsing_script"`
}

type TextParser 

type TextParser struct {
	Scripts map[string]*TextHandlingScript
	// contains filtered or unexported fields
}

func NewTextParser 

func NewTextParser() *TextParser

NewTextParser 创建一个新的文本解析器

func (*TextParser) ParseLine 

func (t *TextParser) ParseLine(r io.Reader, handler func(line string, r map[string]string, data []map[string]string)) error

type YakFunctionCaller 

type YakFunctionCaller struct {
	Handler func(callback func(*yakvm.Frame), args ...any) any
}

type YakToCallerManager 

type YakToCallerManager struct {
	ContextCancelFuncs *sync.Map
	Err                error
	// contains filtered or unexported fields
}

func NewYakToCallerManager 

func NewYakToCallerManager() *YakToCallerManager

func (*YakToCallerManager) Add 

func (y *YakToCallerManager) Add(ctx context.Context, script *schema.YakScript, paramMap map[string]any, code string, hook func(*antlr4yak.Engine) error, funcName ...string) (retError error)

func (*YakToCallerManager) AddForYakit 

func (y *YakToCallerManager) AddForYakit(
	ctx context.Context, script *schema.YakScript,
	paramMap map[string]any,
	code string, callerIf interface {
		Send(result *ypb.ExecResult) error
	},
	hooks ...string,
) error

func (*YakToCallerManager) AddGoNative 

func (y *YakToCallerManager) AddGoNative(id string, name string, cb func(...interface{}))

deprecated

func (*YakToCallerManager) Call 

func (y *YakToCallerManager) Call(name string, opts ...CallOpt) (results []any)

only forceSync can get results

func (*YakToCallerManager) CallByName 

func (y *YakToCallerManager) CallByName(name string, items ...interface{})

func (*YakToCallerManager) CallByNameExSync 

func (y *YakToCallerManager) CallByNameExSync(name string, items ...func() interface{})

func (*YakToCallerManager) CallByNameSync 

func (y *YakToCallerManager) CallByNameSync(name string, items ...interface{})

func (*YakToCallerManager) CallByNameWithCallback 

func (y *YakToCallerManager) CallByNameWithCallback(name string, callback func(), items ...interface{})

func (*YakToCallerManager) CallPluginKeyByName 

func (y *YakToCallerManager) CallPluginKeyByName(pluginId string, name string, items ...interface{})

func (*YakToCallerManager) CallPluginKeyByNameEx 

func (y *YakToCallerManager) CallPluginKeyByNameEx(pluginId string, name string, callback func(), itemsFuncs ...func() interface{})

func (*YakToCallerManager) CallPluginKeyByNameExWithAsync 

func (y *YakToCallerManager) CallPluginKeyByNameExWithAsync(runtimeCtx context.Context, forceSync bool, pluginId string, name string, callback func(), itemsFuncs ...func() interface{})

func (*YakToCallerManager) CallPluginKeyByNameSyncWithCallback 

func (y *YakToCallerManager) CallPluginKeyByNameSyncWithCallback(pluginId string, name string, callback func(), items ...interface{})

func (*YakToCallerManager) CallPluginKeyByNameWithCallback 

func (y *YakToCallerManager) CallPluginKeyByNameWithCallback(pluginId string, name string, callback func(), items ...interface{})

func (*YakToCallerManager) GetCurrentHooks 

func (y *YakToCallerManager) GetCurrentHooks() []*CallerHooks

func (*YakToCallerManager) GetCurrentHooksGRPCModel 

func (y *YakToCallerManager) GetCurrentHooksGRPCModel() []*ypb.YakScriptHooks

func (*YakToCallerManager) GetWaitingEventCount 

func (c *YakToCallerManager) GetWaitingEventCount() int

func (*YakToCallerManager) LoadPlugin 

func (y *YakToCallerManager) LoadPlugin(t string, hooks ...string) error

func (*YakToCallerManager) LoadPluginContext 

func (y *YakToCallerManager) LoadPluginContext(ctx context.Context, t string, hooks ...string) error

func (*YakToCallerManager) Remove 

func (y *YakToCallerManager) Remove(params *ypb.RemoveHookParams)

func (*YakToCallerManager) Set 

func (y *YakToCallerManager) Set(ctx context.Context, code string, paramMap map[string]any, hook func(engine *antlr4yak.Engine) error, funcName ...string) (retError error)

func (*YakToCallerManager) SetCallPluginTimeout 

func (c *YakToCallerManager) SetCallPluginTimeout(i float64)

func (*YakToCallerManager) SetConcurrent 

func (m *YakToCallerManager) SetConcurrent(i int) error

func (*YakToCallerManager) SetDividedContext 

func (y *YakToCallerManager) SetDividedContext(b bool)

func (*YakToCallerManager) SetForYakit 

func (y *YakToCallerManager) SetForYakit(
	ctx context.Context,
	code string,
	paramMap map[string]any, callerIf interface {
		Send(result *ypb.ExecResult) error
	},
	hooks ...string,
) error

func (*YakToCallerManager) SetLoadPluginTimeout 

func (c *YakToCallerManager) SetLoadPluginTimeout(i float64)

func (*YakToCallerManager) ShouldCallByName 

func (y *YakToCallerManager) ShouldCallByName(name string, callbacks ...func()) (ret bool)

func (*YakToCallerManager) SyncCallPluginKeyByNameEx 

func (y *YakToCallerManager) SyncCallPluginKeyByNameEx(pluginId string, name string, callback func(), itemsFuncs ...func() interface{})

func (*YakToCallerManager) Wait 

func (y *YakToCallerManager) Wait()

func (*YakToCallerManager) WithVulFilter 

func (y *YakToCallerManager) WithVulFilter(filter filter.Filterable) *YakToCallerManager

type YakitCallerIf 

type YakitCallerIf func(result *ypb.ExecResult) error

func (YakitCallerIf) Send 

func (y YakitCallerIf) Send(i *ypb.ExecResult) error

type YakitPluginContext 

type YakitPluginContext struct {
	yakit.YakitPluginInfo
	Proxy  string
	Ctx    context.Context
	CliApp *cli.CliApp
	Cancel context.CancelFunc

	YakitClient *yaklib.YakitClient
	// contains filtered or unexported fields
}

func CreateYakitPluginContext 

func CreateYakitPluginContext(runtimeId string) *YakitPluginContext

func (*YakitPluginContext) WithCliApp 

func (y *YakitPluginContext) WithCliApp(cliApp *cli.CliApp) *YakitPluginContext

func (*YakitPluginContext) WithContext 

func (y *YakitPluginContext) WithContext(ctx context.Context) *YakitPluginContext

func (*YakitPluginContext) WithContextCancel 

func (y *YakitPluginContext) WithContextCancel(cancel context.CancelFunc) *YakitPluginContext

func (*YakitPluginContext) WithPluginName 

func (y *YakitPluginContext) WithPluginName(id string) *YakitPluginContext

func (*YakitPluginContext) WithPluginUUID 

func (y *YakitPluginContext) WithPluginUUID(uuid string) *YakitPluginContext

func (*YakitPluginContext) WithProxy 

func (y *YakitPluginContext) WithProxy(proxy string) *YakitPluginContext

func (*YakitPluginContext) WithVulFilter 

func (y *YakitPluginContext) WithVulFilter(filter filter.Filterable) *YakitPluginContext

func (*YakitPluginContext) WithYakitClient 

func (y *YakitPluginContext) WithYakitClient(yakitClient *yaklib.YakitClient) *YakitPluginContext

Source Files

View all Source files

Directories

Path Synopsis
antlr4JS
parser
infrastructure
luaast
parser
antlr4go
parser
buildin_script
executor
executor/nasl_type
lib
parser
script_core
visitors
cmd command
dap
parser
yakast
yakvm
yakvm/vmstack
cmd
yakcmds
Package yakcmds @Author bcy2007 2024/3/12 16:06
 Package yakcmds @Author bcy2007 2024/3/12 16:06
utils
java
freemarker
freemarker/parser
java2ssa
jsp
jsp/parser
parser
spel
spel/parser
template2java
php
parser
php2ssa
ssa
ssadb
ssautil
ssareducer
test/ssatest
information
plugin_type
result
rules
score_rules
ssa_option
doc
generate_doc command
generate_web_doc command
lib/builtin
spec
spec/types
codec
codec/codegrpc
codec/codegrpc/doc command
domain
palmscanlib
tools
tools/dicts
yakhttp

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.