mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-19 13:45: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>
64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package vnc
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/praetorian-inc/fingerprintx/pkg/plugins"
|
|
"github.com/praetorian-inc/fingerprintx/pkg/plugins/services/vnc"
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/protocols/common/protocolstate"
|
|
)
|
|
|
|
type (
|
|
// IsVNCResponse is the response from the IsVNC function.
|
|
// @example
|
|
// ```javascript
|
|
// const vnc = require('nuclei/vnc');
|
|
// const isVNC = vnc.IsVNC('acme.com', 5900);
|
|
// log(toJSON(isVNC));
|
|
// ```
|
|
IsVNCResponse struct {
|
|
IsVNC bool
|
|
Banner string
|
|
}
|
|
)
|
|
|
|
// IsVNC checks if a host is running a VNC server.
|
|
// It returns a boolean indicating if the host is running a VNC server
|
|
// and the banner of the VNC server.
|
|
// @example
|
|
// ```javascript
|
|
// const vnc = require('nuclei/vnc');
|
|
// const isVNC = vnc.IsVNC('acme.com', 5900);
|
|
// log(toJSON(isVNC));
|
|
// ```
|
|
func IsVNC(host string, port int) (IsVNCResponse, error) {
|
|
return memoizedisVNC(host, port)
|
|
}
|
|
|
|
// @memo
|
|
func isVNC(host string, port int) (IsVNCResponse, error) {
|
|
resp := IsVNCResponse{}
|
|
|
|
timeout := 5 * time.Second
|
|
conn, err := protocolstate.Dialer.Dial(context.TODO(), "tcp", net.JoinHostPort(host, strconv.Itoa(port)))
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
defer conn.Close()
|
|
|
|
vncPlugin := vnc.VNCPlugin{}
|
|
service, err := vncPlugin.Run(conn, timeout, plugins.Target{Host: host})
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
if service == nil {
|
|
return resp, nil
|
|
}
|
|
resp.Banner = service.Version
|
|
resp.IsVNC = true
|
|
return resp, nil
|
|
}
|