def __init__(self): self.__log = Logger.getLogger() self.__servers = ServerList() self.__servers.parseList() self.__commandParser = CommandParser() self.__transacationId = str(self.__generateId()) self.__transactionManager = TransactionManager(self.__transacationId)
def __init__(self, serverName, socket, transactionId):
super(TransactionThread, self).__init__()
self.__name = transactionId
self.__log = Logger.getLogger()
self.__log.debug("Creating transaction thread [%s]", self.__name)
self.__serverName = serverName
self.__socket = socket
self.__transactionId = transactionId
self.__messageParser = MessageParser()
self.__file = FileWorker(self.__serverName)
self.__walRegister = WALRegister(self.__serverName)
self.__resource = ResourceQueue()
self.__resourceOwned = False
# create functions mapping
self.__handlers = {
"begin": self.__begin,
"end": self.__end,
"read": self.__read,
"write": self.__write,
"commit": self.__commit,
"globalCommit": self.__globalCommit,
"abort": self.__abort,
"invalid": self.__reject
}
def start(self, serverIP, serverPort):
# start logging
log = Logger.getLogger()
log.info("Starting TrackMe-Server...")
# start listening for incomming messages
listener = ConnectionListener()
listener.listen(serverIP, serverPort)
def __init__(self, serverName): self.__log = Logger.getLogger() # create thread manager self.__serverName = serverName # create a TCP/IP socket self.__socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
def main(argv = None):
if(argv is None):
argv = sys.argv
Logger.initLogger()
log = Logger.getLogger()
client = ClientCore()
try:
client.start()
except IOError:
log.error("Client terminated")
except KeyboardInterrupt:
log.info("Client terminated")
def __init__(self):
self.SUCCESS = 1
self.FAILED = 0
self.__log = Logger.getLogger()
self.__responseCode = self.FAILED
self.__writeCounter = 1
self.clear()
# create functions mapping
self.__saveHandlers = {
"txt": self.__saveToTXT
}
def __init__(self):
self.__dbName = RESOURCE_PATH + SERVER_DB_FILE
self.__log = Logger.getLogger()
# check if database file exists
if(os.path.exists(self.__dbName) == False):
self.__log.debug("DB file doesn't exists, creating...")
dbConnection = sqlite3.connect(self.__dbName) # @UndefinedVariable
# check if ResourceQueue table exists within database
c = dbConnection.cursor()
c.execute("CREATE TABLE IF NOT EXISTS ResourceQueue (id VARCHAR(20), timestamp VARCHAR(20))")
dbConnection.close()
def __init__(self, serverName):
self.__dbName = RESOURCE_PATH + SERVER_DB_FILE
self.__log = Logger.getLogger()
self.__file = FileWorker(serverName)
self.__resource = ResourceQueue()
# check if database file exists
if(os.path.exists(self.__dbName) == False):
self.__log.debug("DB file doesn't exists, creating...")
dbConnection = sqlite3.connect(self.__dbName) # @UndefinedVariable
c = dbConnection.cursor()
# check if WAL table exists within database
c.execute("CREATE TABLE IF NOT EXISTS WALRegister (id VARCHAR(20), command VARCHAR(5), before VARCHAR(200), after VARCHAR(200), timestamp VARCHAR(20))")
dbConnection.close()
def __init__(self):
self.__dbName = RESOURCE_PATH + SERVER_DB_FILE
self.__log = Logger.getLogger()
# check if database file exists
if(os.path.exists(self.__dbName) == False):
self.__log.debug("DB file doesn't exists, creating...")
dbConnection = sqlite3.connect(self.__dbName) # @UndefinedVariable
c = dbConnection.cursor()
# check if tables exist within database
c.execute("CREATE TABLE IF NOT EXISTS UserLocation (userId VARCHAR(30), longtitude VARCHAR(30), latitude VARCHAR(30))")
c.execute("CREATE TABLE IF NOT EXISTS TrackingPermission (userId VARCHAR(30), viewerId VARCHAR(30))")
c.execute("CREATE TABLE IF NOT EXISTS User (userId VARCHAR(30), password VARCHAR(30))")
dbConnection.close()
def __init__(self):
self.__log = Logger.getLogger()
self.__servers = ServerList()
self.__servers.parseList()
self.__sockets = {}
for name in self.__servers.getNames():
try:
serverAddress = self.__servers.getAddress(name)
# create a TCP/IP socket and connect to remote server
serverSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
serverSocket.connect(serverAddress)
# save socket
self.__sockets[name] = serverSocket
except socket.error, e:
self.__log.error("Socket error: " + str(e))
raise IOError
def __init__(self, socket, clientId):
super(ClientThread, self).__init__()
self.__name = clientId
self.__log = Logger.getLogger()
self.__socket = socket
self.__messageParser = MessageParser()
self.__database = ServerDatabase()
self.__log.debug("Creating client thread [%s]", self.__name)
# create functions mapping
self.__handlers = {
"setLocation": self.__setLocation,
"getLocation": self.__getLocation,
"allowTracking": self.__allowTracking,
"registerUser": self.__registerUser,
"loginUser": self.__loginUser,
"checkPermission": self.__checkPermission,
"invalid": self.__invalidMessage
}
def main(argv = None):
if(argv is None):
argv = sys.argv
Logger.initLogger()
log = Logger.getLogger()
if(len(argv) != 3):
log.error("IP, port or server name not specified!")
log.info("usage: TrackMe-Server <IP> <Port>")
return 1
server = ServerCore()
serverIP = argv[1]
serverPort = int(argv[2])
try:
server.start(serverIP, serverPort)
except IOError:
log.error("TrackMe-Server terminated")
except KeyboardInterrupt:
log.info("TrackMe-Server terminated")
def main(argv = None):
if(argv is None):
argv = sys.argv
Logger.initLogger()
log = Logger.getLogger()
#if(len(argv) != 3):
# log.error("malware directory not specified!")
# log.info("usage: VirusTotalAPI <directory>")
# return 1
#filesDirectory = argv[1]
filesDirectory = "D:\\Virtual Machines\\shared\\Win7\\outputSafe\\EncryptionEngine8x_delay20\\"
try:
virusTotalClient = VirusTotalClient()
virusTotalClient.run(filesDirectory)
except IOError:
log.error("VirusTotal API terminated")
except KeyboardInterrupt:
log.info("VirusTotal API terminated")
def __init__(self):
self.__log = Logger.getLogger()
self.__resourceDict = {}
def __init__(self): self.__log = Logger.getLogger() # create a TCP/IP socket self.__socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
