From 240b6636f9619a265ab06fa4d0f61c89b3c324c8 Mon Sep 17 00:00:00 2001
From: Richard Purdie <rpurdie@linux.intel.com>
Date: Thu, 7 Oct 2010 15:42:56 +0100
Subject: sstate: Fix SSTAGE_MIRRORS handling and improve example in
 local.conf.sample

Signed-off-by: Richard Purdie <rpurdie@linux.intel.com>
---
 meta/classes/sstate.bbclass | 52 +++++++++++++++++++++++++++++++++++++--------
 meta/conf/local.conf.sample |  4 +++-
 2 files changed, 46 insertions(+), 10 deletions(-)

diff --git a/meta/classes/sstate.bbclass b/meta/classes/sstate.bbclass
index 7f897ae54d..4a63d37d20 100644
--- a/meta/classes/sstate.bbclass
+++ b/meta/classes/sstate.bbclass
@@ -254,14 +254,18 @@ def pstaging_fetch(sstatepkg, d):
     import bb.fetch
 
     # only try and fetch if the user has configured a mirror
-    if bb.data.getVar('SSTATE_MIRROR', d) != "":
+
+    mirrors = bb.data.getVar('SSTATE_MIRRORS', d, True)
+    if mirrors:
         # Copy the data object and override DL_DIR and SRC_URI
-        pd = d.createCopy()
-        dldir = bb.data.expand("${SSTATE_DIR}", pd)
-        mirror = bb.data.expand("${SSTATE_MIRROR}/", pd)
-        srcuri = mirror + os.path.basename(sstatepkg)
-        bb.data.setVar('DL_DIR', dldir, pd)
-        bb.data.setVar('SRC_URI', srcuri, pd)
+        localdata = bb.data.createCopy(d)
+        bb.data.update_data(localdata)
+
+        dldir = bb.data.expand("${SSTATE_DIR}", localdata)
+        srcuri = "file://" + os.path.basename(sstatepkg)
+        bb.data.setVar('DL_DIR', dldir, localdata)
+        bb.data.setVar('PREMIRRORS', mirrors, localdata)
+        bb.data.setVar('SRC_URI', srcuri, localdata)
 
         # Try a fetch from the sstate mirror, if it fails just return and
         # we will build the package
@@ -269,7 +273,7 @@ def pstaging_fetch(sstatepkg, d):
             bb.fetch.init([srcuri], pd)
             bb.fetch.go(pd, [srcuri])
         except:
-            return
+            pass
 
 def sstate_setscene(d):
     shared_state = sstate_state_fromvars(d)
@@ -319,6 +323,7 @@ sstate_unpack_package () {
 BB_HASHCHECK_FUNCTION = "sstate_checkhashes"
 
 def sstate_checkhashes(sq_fn, sq_task, sq_hash, sq_hashfn, d):
+
     ret = []
     # This needs to go away, FIXME
     mapping = {
@@ -332,10 +337,39 @@ def sstate_checkhashes(sq_fn, sq_task, sq_hash, sq_hashfn, d):
 
     for task in range(len(sq_fn)):
         sstatefile = bb.data.expand("${SSTATE_DIR}/" + sq_hashfn[task] + "_" + mapping[sq_task[task]] + ".tgz", d)
-        sstatefile= sstatefile.replace("${BB_TASKHASH}", sq_hash[task])
+        sstatefile = sstatefile.replace("${BB_TASKHASH}", sq_hash[task])
         #print("Checking for %s" % sstatefile)
         if os.path.exists(sstatefile):
             ret.append(task)
+            continue
+
+    mirrors = bb.data.getVar("SSTATE_MIRRORS", d, True)
+    if mirrors:
+        # Copy the data object and override DL_DIR and SRC_URI
+        localdata = bb.data.createCopy(d)
+        bb.data.update_data(localdata)
+
+        dldir = bb.data.expand("${SSTATE_DIR}", localdata)
+        bb.data.setVar('DL_DIR', dldir, localdata)
+        bb.data.setVar('PREMIRRORS', mirrors, localdata)
+
+        for task in range(len(sq_fn)):
+            if task in ret:
+                continue
+
+            sstatefile = bb.data.expand("${SSTATE_DIR}/" + sq_hashfn[task] + "_" + mapping[sq_task[task]] + ".tgz", d)
+            sstatefile = sstatefile.replace("${BB_TASKHASH}", sq_hash[task])
+
+            srcuri = "file://" + os.path.basename(sstatefile)
+            bb.data.setVar('SRC_URI', srcuri, localdata)
+            bb.note(str(srcuri))
+
+            try:
+                bb.fetch.init(srcuri.split(), d)
+                bb.fetch.checkstatus(localdata)
+                ret.append(task)
+            except:
+                pass     
 
     return ret
 
diff --git a/meta/conf/local.conf.sample b/meta/conf/local.conf.sample
index 5fab4494bc..fc6f13932f 100644
--- a/meta/conf/local.conf.sample
+++ b/meta/conf/local.conf.sample
@@ -142,7 +142,9 @@ ENABLE_BINARY_LOCALE_GENERATION = "1"
 
 # Poky can try and fetch packaged-staging packages from a http, https or ftp
 # mirror. Set this variable to the root of a pstage directory on a server.
-#SSTATE_MIRROR ?= "http://someserver.tld/share/pstage"
+#SSTATE_MIRRORS ?= "\
+#file://.* http://someserver.tld/share/sstate/ \n \
+#file://.* file:///some/local/dir/sstate/"
 
 # Set IMAGETEST to qemu if you want to build testcases and start 
 # testing in qemu after do_rootfs.
-- 
cgit v1.2.3-54-g00ecf