From da5139d8e2c7b06bf0c1ee92d15507ff0d447afe Mon Sep 17 00:00:00 2001 From: Zac Medico Date: Mon, 14 Jul 2008 23:31:51 +0000 Subject: Split out a generic SubProcess._unregister() method that's called from _wait() to make sure everything is cleaned up before wait() returns. This might solve an intermittent KeyError that is sometimes raised from the main poll() loop when a handler is no longer registered for a poll() event that has occurred. svn path=/main/trunk/; revision=11049 --- pym/_emerge/__init__.py | 98 +++++++++++++++++++++++++++---------------------- 1 file changed, 54 insertions(+), 44 deletions(-) diff --git a/pym/_emerge/__init__.py b/pym/_emerge/__init__.py index a60d33950..b80f60c69 100644 --- a/pym/_emerge/__init__.py +++ b/pym/_emerge/__init__.py @@ -1698,8 +1698,11 @@ class PipeReader(AsynchronousTask): def _wait(self): if self.returncode is not None: return self.returncode + if self.registered: self.scheduler.schedule(self._reg_ids) + self._unregister() + self.returncode = os.EX_OK return self.returncode @@ -1726,16 +1729,27 @@ class PipeReader(AsynchronousTask): if buf: self._read_data.append(buf.tostring()) else: - self.registered = False + self._unregister() + self.wait() + + return self.registered + + def _unregister(self): + """ + Unregister from the scheduler and close open files. + """ + + self.registered = False + + if self._reg_ids is not None: for reg_id in self._reg_ids: self.scheduler.unregister(reg_id) + self._reg_ids = None - for f in files.values(): + if self.input_files is not None: + for f in self.input_files.itervalues(): f.close() - - self.wait() - - return self.registered + self.input_files = None class CompositeTask(AsynchronousTask): @@ -1888,7 +1902,7 @@ class TaskSequence(CompositeTask): class SubProcess(AsynchronousTask): - __slots__ = ("scheduler",) + ("pid", "registered", "_reg_id") + __slots__ = ("scheduler",) + ("files", "pid", "registered", "_reg_id") # A file descriptor is required for the scheduler to monitor changes from # inside a poll() loop. When logging is not enabled, create a pipe just to @@ -1933,10 +1947,16 @@ class SubProcess(AsynchronousTask): self.returncode is None def _wait(self): - if self.registered: - self.scheduler.schedule(self._reg_id) + if self.returncode is not None: return self.returncode + + if self.registered: + self.scheduler.schedule(self._reg_id) + self._unregister() + if self.returncode is not None: + return self.returncode + try: wait_retval = os.waitpid(self.pid, 0) except OSError, e: @@ -1946,8 +1966,25 @@ class SubProcess(AsynchronousTask): self._set_returncode((self.pid, 1)) else: self._set_returncode(wait_retval) + return self.returncode + def _unregister(self): + """ + Unregister from the scheduler and close open files. + """ + + self.registered = False + + if self._reg_id is not None: + self.scheduler.unregister(self._reg_id) + self._reg_id = None + + if self.files is not None: + for f in self.files.itervalues(): + f.close() + self.files = None + def _set_returncode(self, wait_retval): retval = wait_retval[1] @@ -1972,7 +2009,7 @@ class SpawnProcess(SubProcess): "uid", "gid", "groups", "umask", "logfile", "path_lookup", "pre_exec") - __slots__ = ("args", "files") + \ + __slots__ = ("args",) + \ _spawn_kwarg_names _file_names = ("process", "out") @@ -2071,12 +2108,7 @@ class SpawnProcess(SubProcess): buf.tofile(files.out) files.out.flush() else: - self.registered = False - self.scheduler.unregister(self._reg_id) - - for f in files.values(): - f.close() - + self._unregister() self.wait() return self.registered @@ -2095,12 +2127,7 @@ class SpawnProcess(SubProcess): if buf: pass else: - self.registered = False - self.scheduler.unregister(self._reg_id) - - for f in files.values(): - f.close() - + self._unregister() self.wait() return self.registered @@ -2497,7 +2524,7 @@ class EbuildMetadataPhase(SubProcess): __slots__ = ("cpv", "ebuild_path", "fd_pipes", "metadata_callback", "ebuild_mtime", "portdb", "repo_path", "settings") + \ - ("files", "_raw_metadata") + ("_raw_metadata",) _file_names = ("ebuild",) _files_dict = slot_dict_class(_file_names, prefix="") @@ -2565,12 +2592,7 @@ class EbuildMetadataPhase(SubProcess): files = self.files self._raw_metadata.append(files.ebuild.read()) if not self._raw_metadata[-1]: - self.registered = False - self.scheduler.unregister(self._reg_id) - - for f in files.values(): - f.close() - + self._unregister() self.wait() if self.returncode == os.EX_OK: @@ -2584,8 +2606,7 @@ class EbuildMetadataPhase(SubProcess): class EbuildPhase(SubProcess): __slots__ = ("fd_pipes", "phase", "pkg", - "settings", "tree", - "files") + "settings", "tree") _file_names = ("log", "stdout", "ebuild") _files_dict = slot_dict_class(_file_names, prefix="") @@ -2692,12 +2713,7 @@ class EbuildPhase(SubProcess): buf.tofile(files.log) files.log.flush() else: - self.registered = False - self.scheduler.unregister(self._reg_id) - - for f in files.values(): - f.close() - + self._unregister() self.wait() return self.registered @@ -2716,12 +2732,7 @@ class EbuildPhase(SubProcess): if buf: pass else: - self.registered = False - self.scheduler.unregister(self._reg_id) - - for f in files.values(): - f.close() - + self._unregister() self.wait() return self.registered @@ -2743,7 +2754,6 @@ class EbuildPhase(SubProcess): self.returncode = portage._post_src_install_checks(settings) elif self.phase == "preinst": - if self.returncode == os.EX_OK: portage._doebuild_exit_status_unlink( settings.get("EBUILD_EXIT_STATUS_FILE")) -- cgit v1.2.3-1-g7c22