| 32 | |
| 33 | |
| 34 | class ProtocolError(Exception): |
| 35 | |
| 36 | def __init__(self, message, data_buf = None, smb_message = None): |
| 37 | self.message = message |
| 38 | self.data_buf = data_buf |
| 39 | self.smb_message = smb_message |
| 40 | |
| 41 | def __str__(self): |
| 42 | b = StringIO() |
| 43 | b.write(self.message + os.linesep) |
| 44 | if self.smb_message: |
| 45 | b.write('=' * 20 + ' SMB Message ' + '=' * 20 + os.linesep) |
| 46 | b.write(str(self.smb_message)) |
| 47 | |
| 48 | if self.data_buf: |
| 49 | b.write('=' * 20 + ' SMB Data Packet (hex) ' + '=' * 20 + os.linesep) |
| 50 | b.write(str(binascii.hexlify(self.data_buf))) |
| 51 | b.write(os.linesep) |
| 52 | |
| 53 | return b.getvalue() |
| 54 | |
| 55 | class SMB2ProtocolHeaderError(ProtocolError): |
| 56 |
no outgoing calls
no test coverage detected