archive/fscan
1
0
Fork 0
mirror of https://github.com/shadow1ng/fscan.git synced 2026-09-25 04:31:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Harden scan robustness and tests

Browse Source
This commit is contained in:
ZacharyZcR
2026-06-14 22:23:48 +08:00
parent 5ad914a1bb
commit c49c23c7f0
100 changed files with 4483 additions and 412 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/services/smb_protocol.go
+3 -12
View File
@@ -696,13 +696,9 @@ func classifySMBError(err error) ErrorType {
// readSMBMessage 从连接读取NetBIOS消息
func readSMBMessage(conn net.Conn) ([]byte, error) {
headerBuf := make([]byte, 4)
n, err := conn.Read(headerBuf)
if err != nil {
if _, err := io.ReadFull(conn, headerBuf); err != nil {
return nil, err
}
if n != 4 {
return nil, fmt.Errorf(i18n.GetText("netbios_header_too_short")+": %d", n)
}
messageLength := int(headerBuf[0])<<24 | int(headerBuf[1])<<16 | int(headerBuf[2])<<8 | int(headerBuf[3])
@@ -715,13 +711,8 @@ func readSMBMessage(conn net.Conn) ([]byte, error) {
}
messageBuf := make([]byte, messageLength)
totalRead := 0
for totalRead < messageLength {
n, err := conn.Read(messageBuf[totalRead:])
if err != nil {
return nil, err
}
totalRead += n
if _, err := io.ReadFull(conn, messageBuf); err != nil {
return nil, err
}
result := make([]byte, 0, 4+messageLength)