improved smb_login error codes

10 years ago · 766d92735e
parent effdbbd291
commit 766d92735e
1 changed files with 14 additions and 11 deletions
--- a/patator.py
+++ b/patator.py
@ -2396,7 +2396,13 @@ class SMB_login(TCP_Cache):
      self.reset()

    except impacket_smb.SessionError as e:
-      code = '%04x%04x' % (e.error_class, e.error_code)
+
+      if e.error_class == 0:
+        code = '%04x' % e.error_code
+        mesg = self.error_map.get(e.error_code & 0x0000fffff, '')
+
+      else:
+        code = '%x%04x' % (e.error_class, e.error_code)

        error_class = e.error_classes.get(e.error_class, None) # -> ("ERRNT", nt_msgs)
        if error_class:
@ -2408,9 +2414,6 @@ class SMB_login(TCP_Cache):
          else:
            mesg = '%s' % class_str

-      else:
-        mesg = self.error_map.get(e.error_code & 0x0000fffff, '')
-
    if persistent == '0':
      self.reset()