mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-17 18:35:25 +00:00
4c7a0f424e
* initial implementation with manual code generation * testing generation * refactor to package methods + auto memoize * more memos * fixing signatures * refactor * adding gen util * adding util * regenerate memoized files --------- Co-authored-by: Tarun Koyalwar <tarun@projectdiscovery.io>
115 lines
2.8 KiB
Go
115 lines
2.8 KiB
Go
package rdp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/praetorian-inc/fingerprintx/pkg/plugins"
|
|
"github.com/praetorian-inc/fingerprintx/pkg/plugins/services/rdp"
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/protocols/common/protocolstate"
|
|
)
|
|
|
|
type (
|
|
// IsRDPResponse is the response from the IsRDP function.
|
|
// this is returned by IsRDP function.
|
|
// @example
|
|
// ```javascript
|
|
// const rdp = require('nuclei/rdp');
|
|
// const isRDP = rdp.IsRDP('acme.com', 3389);
|
|
// log(toJSON(isRDP));
|
|
// ```
|
|
IsRDPResponse struct {
|
|
IsRDP bool
|
|
OS string
|
|
}
|
|
)
|
|
|
|
// IsRDP checks if the given host and port are running rdp server.
|
|
// If connection is successful, it returns true.
|
|
// If connection is unsuccessful, it returns false and error.
|
|
// The Name of the OS is also returned if the connection is successful.
|
|
// @example
|
|
// ```javascript
|
|
// const rdp = require('nuclei/rdp');
|
|
// const isRDP = rdp.IsRDP('acme.com', 3389);
|
|
// log(toJSON(isRDP));
|
|
// ```
|
|
func IsRDP(host string, port int) (IsRDPResponse, error) {
|
|
return memoizedisRDP(host, port)
|
|
}
|
|
|
|
// @memo
|
|
func isRDP(host string, port int) (IsRDPResponse, error) {
|
|
resp := IsRDPResponse{}
|
|
|
|
timeout := 5 * time.Second
|
|
conn, err := protocolstate.Dialer.Dial(context.TODO(), "tcp", fmt.Sprintf("%s:%d", host, port))
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
defer conn.Close()
|
|
|
|
server, isRDP, err := rdp.DetectRDP(conn, timeout)
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
if !isRDP {
|
|
return resp, nil
|
|
}
|
|
resp.IsRDP = true
|
|
resp.OS = server
|
|
return resp, nil
|
|
}
|
|
|
|
type (
|
|
// CheckRDPAuthResponse is the response from the CheckRDPAuth function.
|
|
// this is returned by CheckRDPAuth function.
|
|
// @example
|
|
// ```javascript
|
|
// const rdp = require('nuclei/rdp');
|
|
// const checkRDPAuth = rdp.CheckRDPAuth('acme.com', 3389);
|
|
// log(toJSON(checkRDPAuth));
|
|
// ```
|
|
CheckRDPAuthResponse struct {
|
|
PluginInfo *plugins.ServiceRDP
|
|
Auth bool
|
|
}
|
|
)
|
|
|
|
// CheckRDPAuth checks if the given host and port are running rdp server
|
|
// with authentication and returns their metadata.
|
|
// If connection is successful, it returns true.
|
|
// @example
|
|
// ```javascript
|
|
// const rdp = require('nuclei/rdp');
|
|
// const checkRDPAuth = rdp.CheckRDPAuth('acme.com', 3389);
|
|
// log(toJSON(checkRDPAuth));
|
|
// ```
|
|
func CheckRDPAuth(host string, port int) (CheckRDPAuthResponse, error) {
|
|
return memoizedcheckRDPAuth(host, port)
|
|
}
|
|
|
|
// @memo
|
|
func checkRDPAuth(host string, port int) (CheckRDPAuthResponse, error) {
|
|
resp := CheckRDPAuthResponse{}
|
|
|
|
timeout := 5 * time.Second
|
|
conn, err := protocolstate.Dialer.Dial(context.TODO(), "tcp", fmt.Sprintf("%s:%d", host, port))
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
defer conn.Close()
|
|
|
|
pluginInfo, auth, err := rdp.DetectRDPAuth(conn, timeout)
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
if !auth {
|
|
return resp, nil
|
|
}
|
|
resp.Auth = true
|
|
resp.PluginInfo = pluginInfo
|
|
return resp, nil
|
|
}
|