mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-17 17:56:56 +00:00
19247ae74b
* setup claude * migrate to using errkit * fix unused imports + lint errors * update settings.json * fix url encoding issue * fix lint error * fix the path fuzzing component * fix lint error
82 lines
2.8 KiB
Go
82 lines
2.8 KiB
Go
package smb
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/js/libs/structs"
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/protocols/common/protocolstate"
|
|
"github.com/projectdiscovery/utils/reader"
|
|
)
|
|
|
|
const (
|
|
pkt = "\x00\x00\x00\xc0\xfeSMB@\x00\x00\x00\x00\x00\x00\x00\x00\x00\x1f\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00$\x00\x08\x00\x01\x00\x00\x00\x7f\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00x\x00\x00\x00\x02\x00\x00\x00\x02\x02\x10\x02\"\x02$\x02\x00\x03\x02\x03\x10\x03\x11\x03\x00\x00\x00\x00\x01\x00&\x00\x00\x00\x00\x00\x01\x00 \x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03\x00\n\x00\x00\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00"
|
|
)
|
|
|
|
// DetectSMBGhost tries to detect SMBGhost vulnerability
|
|
// by using SMBv3 compression feature.
|
|
// If the host is vulnerable, it returns true.
|
|
// @example
|
|
// ```javascript
|
|
// const smb = require('nuclei/smb');
|
|
// const isSMBGhost = smb.DetectSMBGhost('acme.com', 445);
|
|
// ```
|
|
func (c *SMBClient) DetectSMBGhost(ctx context.Context, host string, port int) (bool, error) {
|
|
executionId := ctx.Value("executionId").(string)
|
|
return memoizeddetectSMBGhost(executionId, host, port)
|
|
}
|
|
|
|
// @memo
|
|
func detectSMBGhost(executionId string, host string, port int) (bool, error) {
|
|
if !protocolstate.IsHostAllowed(executionId, host) {
|
|
// host is not valid according to network policy
|
|
return false, protocolstate.ErrHostDenied.Msgf(host)
|
|
}
|
|
addr := net.JoinHostPort(host, strconv.Itoa(port))
|
|
dialer := protocolstate.GetDialersWithId(executionId)
|
|
if dialer == nil {
|
|
return false, fmt.Errorf("dialers not initialized for %s", executionId)
|
|
}
|
|
conn, err := dialer.Fastdialer.Dial(context.TODO(), "tcp", addr)
|
|
if err != nil {
|
|
return false, err
|
|
|
|
}
|
|
defer func() {
|
|
_ = conn.Close()
|
|
}()
|
|
|
|
_, err = conn.Write([]byte(pkt))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
buff, _ := reader.ConnReadNWithTimeout(conn, 4, time.Duration(5)*time.Second)
|
|
args, err := structs.Unpack(">I", buff)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if len(args) != 1 {
|
|
return false, errors.New("invalid response")
|
|
}
|
|
|
|
length := args[0].(int)
|
|
_ = conn.SetReadDeadline(time.Now().Add(2 * time.Second))
|
|
data, err := reader.ConnReadNWithTimeout(conn, int64(length), time.Duration(5)*time.Second)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if len(data) < 72 {
|
|
return false, errors.New("invalid response expected at least 72 bytes")
|
|
}
|
|
|
|
if !bytes.Equal(data[68:70], []byte("\x11\x03")) || !bytes.Equal(data[70:72], []byte("\x02\x00")) {
|
|
return false, nil
|
|
}
|
|
return true, nil
|
|
}
|