mirror of
https://github.com/projectdiscovery/nuclei
synced 2026-06-08 16:50:47 +00:00
Dwi Siswanto
65d08b33ef
Add shared auth, options, result, redaction, and adapter support for GoExec-backed JavaScript helpers, including: * `nuclei/wmi` * `nuclei/tsch` * `nuclei/scmr` * `nuclei/dcom` through generated bindings, and register them with the JS compiler. Also make JS generators to skip internal helper packages and test files. Signed-off-by: Dwi Siswanto <git@dw1.io>
84 lines
2.6 KiB
Go
84 lines
2.6 KiB
Go
package dcom
|
|
|
|
import (
|
|
"github.com/Mzack9999/goja"
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/js/libs/goexec"
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/js/utils"
|
|
)
|
|
|
|
// Auth exposes DCOM authentication constructors.
|
|
var Auth = goexec.AuthConstructors()
|
|
|
|
// Client is a GoExec-backed DCOM execution client.
|
|
type Client struct {
|
|
target string
|
|
auth *goexec.Auth
|
|
options goexec.ExecutionOptions
|
|
nj *utils.NucleiJS
|
|
}
|
|
|
|
// NewClient constructs a DCOM client.
|
|
//
|
|
// Constructor: constructor(target: string, auth: any)
|
|
func NewClient(call goja.ConstructorCall, runtime *goja.Runtime) *goja.Object {
|
|
nj := utils.NewNucleiJS(runtime)
|
|
nj.ObjectSig = "Client(target, auth)"
|
|
client := &Client{options: goexec.DefaultExecutionOptions(), nj: nj}
|
|
client.target, _ = nj.GetArg(call.Arguments, 0).(string)
|
|
if auth, ok := nj.GetArg(call.Arguments, 1).(*goexec.Auth); ok {
|
|
client.auth = auth
|
|
} else {
|
|
nj.Throw("auth must be created with dcom.Auth")
|
|
}
|
|
obj := runtime.NewObject()
|
|
client.bindAliases(obj, runtime)
|
|
return obj
|
|
}
|
|
|
|
// SetOptions updates default execution options for future client calls.
|
|
func (c *Client) SetOptions(opts interface{}) {
|
|
c.options = goexec.MergeOptions(c.options, opts)
|
|
}
|
|
|
|
// MMC invokes the MMC20.Application DCOM execution method.
|
|
func (c *Client) MMC(executable string, args string, opts interface{}) map[string]interface{} {
|
|
req := goexec.Request{
|
|
Module: "dcom",
|
|
Method: "mmc",
|
|
Target: c.target,
|
|
Auth: c.auth,
|
|
Executable: executable,
|
|
Args: args,
|
|
Options: goexec.MergeOptions(c.options, opts),
|
|
}
|
|
return goexec.Run(goexec.WithExecutionID(c.nj.Context(), c.nj.ExecutionId()), req).Public()
|
|
}
|
|
|
|
func (c *Client) bindAliases(obj *goja.Object, runtime *goja.Runtime) {
|
|
_ = obj.Set("setOptions", func(call goja.FunctionCall) goja.Value {
|
|
c.SetOptions(exportArg(call, 0))
|
|
return goja.Undefined()
|
|
})
|
|
_ = obj.Set("SetOptions", func(call goja.FunctionCall) goja.Value {
|
|
c.SetOptions(exportArg(call, 0))
|
|
return goja.Undefined()
|
|
})
|
|
_ = obj.Set("mmc", func(call goja.FunctionCall) goja.Value {
|
|
executable, _ := exportArg(call, 0).(string)
|
|
args, _ := exportArg(call, 1).(string)
|
|
return runtime.ToValue(c.MMC(executable, args, exportArg(call, 2)))
|
|
})
|
|
_ = obj.Set("MMC", func(call goja.FunctionCall) goja.Value {
|
|
executable, _ := exportArg(call, 0).(string)
|
|
args, _ := exportArg(call, 1).(string)
|
|
return runtime.ToValue(c.MMC(executable, args, exportArg(call, 2)))
|
|
})
|
|
}
|
|
|
|
func exportArg(call goja.FunctionCall, index int) interface{} {
|
|
if index >= len(call.Arguments) || goja.IsUndefined(call.Arguments[index]) || goja.IsNull(call.Arguments[index]) {
|
|
return nil
|
|
}
|
|
return call.Arguments[index].Export()
|
|
}
|