Merge pull request #675 from MrAnde7son/patch-12

Add support for all info types query\set info request
fortra · Sep 23, 2019 · 02089dc · 02089dc
2 parents afa2825 + 149ba19
commit 02089dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/impacket/smb3.py b/impacket/smb3.py
@@ -1310,7 +1310,7 @@ def queryInfo(self, treeId, fileId, inputBlob = '', infoType = SMB2_0_INFO_FILE,
 
         queryInfo = SMB2QueryInfo()
         queryInfo['FileID']                = fileId
-        queryInfo['InfoType']              = SMB2_0_INFO_FILE 
+        queryInfo['InfoType']              = infoType 
         queryInfo['FileInfoClass']         = fileInfoClass 
         queryInfo['OutputBufferLength']    = 65535
         queryInfo['AdditionalInformation'] = additionalInformation
@@ -1341,7 +1341,7 @@ def setInfo(self, treeId, fileId, inputBlob = '', infoType = SMB2_0_INFO_FILE, f
         packet['TreeID']  = treeId
 
         setInfo = SMB2SetInfo()
-        setInfo['InfoType']              = SMB2_0_INFO_FILE 
+        setInfo['InfoType']              = infoType 
         setInfo['FileInfoClass']         = fileInfoClass 
         setInfo['BufferLength']          = len(inputBlob)
         setInfo['AdditionalInformation'] = additionalInformation