mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 15:45:26 +00:00
a677fca192
* js protocol timeout using -timeout flag * fix zgrab smb hang * fix lint error * custom timeout field in js protocol * minor update: bound checking * add 6 * -timeout in code protocol by default
63 lines
2.3 KiB
Go
63 lines
2.3 KiB
Go
package smb
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"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.
|
|
func (c *SMBClient) DetectSMBGhost(host string, port int) (bool, error) {
|
|
if !protocolstate.IsHostAllowed(host) {
|
|
// host is not valid according to network policy
|
|
return false, protocolstate.ErrHostDenied.Msgf(host)
|
|
}
|
|
addr := net.JoinHostPort(host, strconv.Itoa(port))
|
|
conn, err := protocolstate.Dialer.Dial(context.TODO(), "tcp", addr)
|
|
if err != nil {
|
|
return false, err
|
|
|
|
}
|
|
defer 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
|
|
}
|