コード例 #1
0
ファイル: rqnetwork.py プロジェクト: xinobi/OpenCue
 def kill(self, message=""):
     """Kills the frame"""
     log.info("Request recieved: kill")
     if self.frameAttendantThread is None:
         log.warning(
             "Kill requested before frameAttendantThread is created "
             "for: %s" % self.frameId)
     elif self.frameAttendantThread.isAlive() and self.pid is None:
         log.warning("Kill requested before pid is available for: %s" %
                     self.frameId)
     elif self.frameAttendantThread.isAlive():
         try:
             if not self.killMessage and message:
                 self.killMessage = message
             rqutil.permissionsHigh()
             try:
                 if platform.system() == "Windows":
                     subprocess.Popen('taskkill /F /T /PID %i' % self.pid,
                                      shell=True)
                 else:
                     os.killpg(self.pid, rqconstants.KILL_SIGNAL)
             finally:
                 rqutil.permissionsLow()
         except OSError, e:
             log.warning("kill() tried to kill a non-existant pid for: %s "
                         "Error: %s" % (self.frameId, e))
         except Exception, e:
             log.warning("kill() encountered an unknown error: %s" % e)
コード例 #2
0
    def runDarwin(self):
        """The steps required to handle a frame under mac"""
        frameInfo = self.frameInfo

        self.__createEnvVariables()
        self.__writeHeader()

        rqutil.permissionsHigh()
        try:
            tempCommand = [
                "/usr/bin/su", frameInfo.runFrame.user_name, "-c",
                '"' + self._createCommandFile(frameInfo.runFrame.command) + '"'
            ]

            frameInfo.forkedCommand = subprocess.Popen(
                tempCommand,
                env=self.frameEnv,
                cwd=self.rqCore.machine.getTempPath(),
                stdin=subprocess.PIPE,
                stdout=self.rqlog,
                stderr=self.rqlog,
                preexec_fn=os.setsid)
        finally:
            rqutil.permissionsLow()

        frameInfo.pid = frameInfo.forkedCommand.pid

        if not self.rqCore.updateRssThread.isAlive():
            self.rqCore.updateRssThread = threading.Timer(
                rqconstants.RSS_UPDATE_INTERVAL, self.rqCore.updateRss)
            self.rqCore.updateRssThread.start()

        frameInfo.forkedCommand.wait()

        # Find exitStatus and exitSignal
        returncode = frameInfo.forkedCommand.returncode
        if os.WIFEXITED(returncode):
            frameInfo.exitStatus = os.WEXITSTATUS(returncode)
        else:
            frameInfo.exitStatus = 1
        if os.WIFSIGNALED(returncode):
            frameInfo.exitSignal = os.WTERMSIG(returncode)

        self.__writeFooter()
        self.__cleanup()
コード例 #3
0
ファイル: rqnimby.py プロジェクト: yazici/OpenCue
    def _openEvents(self):
        """Opens the /dev/input/event* files so nimby can monitor them"""
        self._closeEvents()

        rqutil.permissionsHigh()
        try:
            for device in os.listdir("/dev/input/"):
                if device.startswith("event") or device.startswith("mice"):
                    log.debug("Found device: %s" % device)
                    try:
                        self.fileObjList.append(
                            open("/dev/input/%s" % device, "rb"))
                    except IOError, e:
                        # Bad device found
                        log.debug("IOError: Failed to open %s, %s" %
                                  ("/dev/input/%s" % device, e))
        finally:
            rqutil.permissionsLow()
コード例 #4
0
ファイル: rqcore.py プロジェクト: timkeller/OpenCue
    def __cleanup(self):
        """Cleans up temporary files"""
        rqutil.permissionsHigh()
        try:
            for location in self._tempLocations:
                if os.path.isfile(location):
                    try:
                        os.remove(location)
                    except Exception, e:
                        log.warning("Unable to delete file: %s due to %s at %s" % (
                            location, e, traceback.extract_tb(sys.exc_info()[2])))
        finally:
            rqutil.permissionsLow()

        # Close log file
        try:
            self.rqlog.close()
        except Exception, e:
            log.warning("Unable to close file: %s due to %s at %s" % (
                self.runFrame.log_file, e, traceback.extract_tb(sys.exc_info()[2])))
コード例 #5
0
ファイル: rqcore.py プロジェクト: timkeller/OpenCue
    def runLinux(self):
        """The steps required to handle a frame under linux"""
        frameInfo = self.frameInfo
        runFrame = self.runFrame

        self.__createEnvVariables()
        self.__writeHeader()
        if rqconstants.RQD_CREATE_USER_IF_NOT_EXISTS:
            rqutil.permissionsHigh()
            rqutil.checkAndCreateUser(runFrame.user_name)
            rqutil.permissionsLow()

        tempStatFile = "%srqd-stat-%s-%s" % (self.rqCore.machine.getTempPath(),
                                             frameInfo.frameId,
                                             time.time())
        self._tempLocations.append(tempStatFile)
        tempCommand = []
        if self.rqCore.machine.isDesktop():
            tempCommand += ["/bin/nice"]
        tempCommand += ["/usr/bin/time", "-p", "-o", tempStatFile]

        if 'CPU_LIST' in runFrame.attributes:
            tempCommand += ['taskset', '-c', runFrame.attributes['CPU_LIST']]

        rqutil.permissionsHigh()
        try:
            tempCommand += ["/bin/su", runFrame.user_name, rqconstants.SU_ARGUEMENT,
                            '"' + self._createCommandFile(runFrame.command) + '"']

            # Actual cwd is set by /shots/SHOW/home/perl/etc/qwrap.cuerun
            frameInfo.forkedCommand = subprocess.Popen(tempCommand,
                                                       env=self.frameEnv,
                                                       cwd=self.rqCore.machine.getTempPath(),
                                                       stdin=subprocess.PIPE,
                                                       stdout=self.rqlog,
                                                       stderr=self.rqlog,
                                                       close_fds=True,
                                                       preexec_fn=os.setsid)
        finally:
            rqutil.permissionsLow()

        frameInfo.pid = frameInfo.forkedCommand.pid

        if not self.rqCore.updateRssThread.isAlive():
            self.rqCore.updateRssThread = threading.Timer(rqconstants.RSS_UPDATE_INTERVAL,
                                                           self.rqCore.updateRss)
            self.rqCore.updateRssThread.start()

        returncode = frameInfo.forkedCommand.wait()

        # Find exitStatus and exitSignal
        if returncode < 0:
            # Exited with a signal
            frameInfo.exitStatus = 1
            frameInfo.exitSignal = -returncode
        else:
            frameInfo.exitStatus = returncode
            frameInfo.exitSignal = 0

        try:
            statFile  = open(tempStatFile,"r")
            frameInfo.realtime = statFile.readline().split()[1]
            frameInfo.utime = statFile.readline().split()[1]
            frameInfo.stime = statFile.readline().split()[1]
            statFile.close()
        except Exception:
            pass # This happens when frames are killed

        self.__writeFooter()
        self.__cleanup()