1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
#!/usr/bin/env python
import os, tarfile, hashlib, re, shutil
from snapshot import *
def snap_filename_hash_part(snap):
match = re.match(r".*([a-fA-F\d]{40}).tar.bz2$", snap)
if not match:
raise Exception("unable to find hash in filename: " + snap)
return match.group(1)
def unpack_snapshot(snap):
dl_path = os.path.join(download_dir_base, snap)
print("opening snapshot " + dl_path)
tar = tarfile.open(dl_path)
kernel = get_kernel()
for name in snapshot_files[kernel]:
p = os.path.join("rust-stage0", name)
fp = os.path.join("stage0", name)
print("extracting " + fp)
tar.extract(p, download_unpack_base)
tp = os.path.join(download_unpack_base, p)
shutil.move(tp, fp)
tar.close()
shutil.rmtree(download_unpack_base)
def determine_last_snapshot_for_platform():
lines = open(snapshotfile).readlines();
platform = get_platform()
found = False
hsh = None
date = None
rev = None
for ln in range(len(lines) - 1, -1, -1):
parsed = parse_line(ln, lines[ln])
if (not parsed): continue
if parsed["type"] == "file":
if parsed["platform"] == platform:
hsh = parsed["hash"]
elif parsed["type"] == "snapshot":
date = parsed["date"]
rev = parsed["rev"]
found = True
break
elif parsed["type"] == "transition" and not foundSnapshot:
raise Exception("working on a transition, not updating stage0")
if not found:
raise Exception("no snapshot entries in file")
if not hsh:
raise Exception("no snapshot file found for platform %s, rev %s" %
(platform, rev))
return full_snapshot_name(date, rev, get_kernel(), get_cpu(), hsh)
# Main
snap = determine_last_snapshot_for_platform()
dl = os.path.join(download_dir_base, snap)
url = download_url_base + "/" + snap
print("determined most recent snapshot: " + snap)
if (not os.path.exists(dl)):
get_url_to_file(url, dl)
if (snap_filename_hash_part(snap) == hash_file(dl)):
print("got download with ok hash")
else:
raise Exception("bad hash on download")
unpack_snapshot(snap)
|