Exemple #1
0
    def __init__(self, verbose):
        super(RainMoveClient, self).__init__()

        #Load Config
        self._conf = RainMoveClientConf()
        self.verbose = verbose
        
        self._port = self._conf.getPort()
        self._serveraddr = self._conf.getServeraddr()
        
        
        self._ca_certs = self._conf.getCaCerts()
        self._certfile = self._conf.getCertFile()
        self._keyfile = self._conf.getKeyFile()
        
        self._connMoveServer = None
        
        self.passwdtype = "ldappassmd5"
        #Setup log
        self._log = fgLog.fgLog(self._conf.getLogFile(), self._conf.getLogLevel(), "Rain Move Client", False)
Exemple #2
0
class RainMoveClient(object):
    def __init__(self, verbose):
        super(RainMoveClient, self).__init__()

        #Load Config
        self._conf = RainMoveClientConf()
        self.verbose = verbose
        
        self._port = self._conf.getPort()
        self._serveraddr = self._conf.getServeraddr()
        
        
        self._ca_certs = self._conf.getCaCerts()
        self._certfile = self._conf.getCertFile()
        self._keyfile = self._conf.getKeyFile()
        
        self._connMoveServer = None
        
        self.passwdtype = "ldappassmd5"
        #Setup log
        self._log = fgLog.fgLog(self._conf.getLogFile(), self._conf.getLogLevel(), "Rain Move Client", False)
        

    def connection(self):
        connected = False
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        try:
            self._connMoveServer = ssl.wrap_socket(s,
                                        ca_certs=self._ca_certs,
                                        certfile=self._certfile,
                                        keyfile=self._keyfile,
                                        cert_reqs=ssl.CERT_REQUIRED,
                                        ssl_version=ssl.PROTOCOL_TLSv1)
            self._log.debug("Connecting server: " + self._serveraddr + ":" + str(self._port))
            self._connMoveServer.connect((self._serveraddr, self._port))   
            connected = True         
        except ssl.SSLError:
            self._log.error("CANNOT establish SSL connection. EXIT")
        except socket.error:
            self._log.error("Error with the socket connection")
        except:
            if self.verbose:
                print "Error CANNOT establish connection with the server"
            self._log.error("ERROR: exception not controlled" + str(sys.exc_info()))
        
        return connected
        #_connMoveServer.write(options) #to be done in each method
    def disconnect(self):
        try:
            self._connMoveServer.shutdown(socket.SHUT_RDWR)
            self._connMoveServer.close()
        except:
            self._log.debug("In disconnect:" + str(sys.exc_info()))

    def check_auth(self, userId, checkauthstat):
        endloop = False
        passed = False
        while not endloop:
            ret = self._connMoveServer.read(1024)
            if (ret == "OK"):
                if self.verbose:
                    print "Authentication OK. Your request is being processed"
                self._log.debug("Authentication OK")
                endloop = True
                passed = True
            elif (ret == "TryAuthAgain"):
                msg = "ERROR: Permission denied, please try again. User is " + userId                    
                self._log.error(msg)
                if self.verbose:
                    print msg                            
                m = hashlib.md5()
                m.update(getpass())
                passwd = m.hexdigest()
                self._connMoveServer.write(passwd)
            elif (ret == "NoActive"):                
                checkauthstat.append("ERROR: The status of the user " + userId + " is not active")
                checkauthstat.append("NoActive")
                self._log.error("The status of the user " + userId + " is not active")
                endloop = True
                passed = False
            elif (ret == "NoUser"):
                checkauthstat.append("ERROR: User " + userId + " does not exist")
                checkauthstat.append("NoUser")
                self._log.error("User " + userId + " does not exist")
                endloop = True
                passed = False
            else:                
                self._log.error(str(ret))
                #if self.verbose:
                #    print ret
                checkauthstat.append(str(ret))
                endloop = True
                passed = False
        return passed

    def cluster(self, user, passwd, subparser_name, operation, arguments, force):
        status=None
        start_all = time.time()
        checkauthstat = []
        msg = str(user) + "|" + str(passwd) + "|" + self.passwdtype + "|" + str(subparser_name) + "|" + str(operation) + "|" + str(arguments) + "|" + str(force)
        self._log.debug("Cluster: " + str(msg))
        self._connMoveServer.write(msg)
        
        if self.check_auth(user, checkauthstat):        
            
            status = self._connMoveServer.read(2048)
            
        else:
            self._log.error(str(checkauthstat[0]))
            if self.verbose:
                status = str(checkauthstat[0])
        
        end_all = time.time()
        self._log.info('TIME walltime rain move client (cluster):' + str(end_all - start_all))
        
        return status
        
    def node(self, user, passwd, subparser_name, operation, arguments, force):
        status=None
        start_all = time.time()
        checkauthstat = []

        msg = str(user) + "|" + str(passwd) + "|" + self.passwdtype + "|" + str(subparser_name) + "|" + str(operation) + "|" + str(arguments) + "|" + str(force)
        self._log.debug("Node: " + str(msg))
        self._connMoveServer.write(msg)
        
        if self.check_auth(user, checkauthstat):
            
            status = self._connMoveServer.read(2048)
            
        else:
            self._log.error(str(checkauthstat[0]))
            if self.verbose:
                status = str(checkauthstat[0])        
        
        end_all = time.time()
        self._log.info('TIME walltime rain move client (node):' + str(end_all - start_all))
        
        return status
        
    def service(self, user, passwd, subparser_name, operation, arguments, force):
        status = None
        start_all = time.time()
        checkauthstat = []
        
        msg = str(user) + "|" + str(passwd) + "|" + self.passwdtype + "|" + str(subparser_name) + "|" + str(operation) + "|" + str(arguments) + "|" + str(force)
        self._log.debug("Service: " + str(msg))
        self._connMoveServer.write(msg)
        
        if self.check_auth(user, checkauthstat):
            
            status = self._connMoveServer.read(2048)
            
        else:
            self._log.error(str(checkauthstat[0]))
            if self.verbose:
                status = str(checkauthstat[0])
        
        end_all = time.time()
        self._log.info('TIME walltime rain move client (service):' + str(end_all - start_all))
        
        return status