Skip to content

Commit a57aefb

Browse files
committed
Add specs for QUERY information level
1 parent 96c0f18 commit a57aefb

File tree

2 files changed

+424
-3
lines changed
  • lib/msf/core/exploit/smb/server/share/information_level
  • spec/lib/msf/core/exploit/smb/server/share/information_level

2 files changed

+424
-3
lines changed

lib/msf/core/exploit/smb/server/share/information_level/query.rb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@ module Query
1515
def smb_cmd_trans_query_file_info_basic(c, fid)
1616
smb = @state[c]
1717

18-
if fid.eql?smb[:file_id].to_i
18+
if fid == smb[:file_id].to_i
1919
attrib = CONST::SMB_EXT_FILE_ATTR_NORMAL
20-
elsif fid.nil? || fid.empty? || fid == "\x00" # empty fid
20+
elsif fid.nil? || fid == 0 || fid == smb[:dir_id].to_i # empty fid
2121
attrib = CONST::SMB_EXT_FILE_ATTR_DIRECTORY
2222
else
2323
return smb_error(CONST::SMB_COM_TRANSACTION2, c, CONST::SMB_STATUS_OBJECT_NAME_NOT_FOUND, true)
@@ -100,7 +100,7 @@ def smb_cmd_trans_query_path_info_standard(c, path)
100100
def smb_cmd_trans_query_path_info_network(c, path)
101101

102102
if path && path.include?(file_name)
103-
attrib = 0
103+
attrib = CONST::SMB_EXT_FILE_ATTR_NORMAL
104104
elsif path && path == path_name
105105
attrib = CONST::SMB_EXT_FILE_ATTR_DIRECTORY
106106
elsif path.nil? || path.empty? || path == "\x00" # empty path

0 commit comments

Comments
 (0)