From c564fb09097648a7f28d701c9ccea05d7dc49b01 Mon Sep 17 00:00:00 2001 From: Chris Larson Date: Fri, 18 Mar 2011 09:42:33 -0700 Subject: runqueue: simplify fakeroot environment handling (Bitbake rev: 88f0d1db88bdba95e488fba34f40dc0b72a85c33) Signed-off-by: Chris Larson Signed-off-by: Richard Purdie --- bitbake/lib/bb/runqueue.py | 32 +++++++++++++++++--------------- 1 file changed, 17 insertions(+), 15 deletions(-) diff --git a/bitbake/lib/bb/runqueue.py b/bitbake/lib/bb/runqueue.py index dcb2e0a86b..9000c5cafc 100644 --- a/bitbake/lib/bb/runqueue.py +++ b/bitbake/lib/bb/runqueue.py @@ -1060,23 +1060,24 @@ class RunQueueExecute: return def fork_off_task(self, fn, task, taskname, quieterrors=False): + # We need to setup the environment BEFORE the fork, since + # a fork() or exec*() activates PSEUDO... - envbackup = os.environ.copy() - env = {} + envbackup = {} taskdep = self.rqdata.dataCache.task_deps[fn] if 'fakeroot' in taskdep and taskname in taskdep['fakeroot']: envvars = (self.rqdata.dataCache.fakerootenv[fn] or "").split() - for var in envvars: - comps = var.split("=") - env[comps[0]] = comps[1] + for key, value in (var.split('=') for var in envvars): + envbackup[key] = os.environ.get(key) + os.environ[key] = value fakedirs = (self.rqdata.dataCache.fakerootdirs[fn] or "").split() for p in fakedirs: - bb.mkdirhier(p) - logger.debug(2, "Running %s:%s under fakeroot, state dir is %s" % (fn, taskname, fakedirs)) - for e in env: - os.putenv(e, env[e]) + bb.utils.mkdirhier(p) + + logger.debug(2, 'Running %s:%s under fakeroot, fakedirs: %s' % + (fn, taskname, ', '.join(fakedirs))) sys.stdout.flush() sys.stderr.flush() @@ -1087,6 +1088,7 @@ class RunQueueExecute: pid = os.fork() except OSError as e: bb.msg.fatal(bb.msg.domain.RunQueue, "fork failed: %d (%s)" % (e.errno, e.strerror)) + if pid == 0: pipein.close() @@ -1143,12 +1145,12 @@ class RunQueueExecute: os._exit(ret) except: os._exit(1) - - for e in env: - os.unsetenv(e) - for e in envbackup: - if e in env: - os.putenv(e, envbackup[e]) + else: + for key, value in envbackup.iteritems(): + if value is None: + del os.environ[key] + else: + os.environ[key] = value return pid, pipein, pipeout -- cgit v1.2.3-54-g00ecf