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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
#!/usr/bin/python
# Authors: Cedric Bosdonnat <cbosdonnat@suse.com>
#
# Copyright (C) 2017 SUSE, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#
import hashlib
import json
import shutil
import subprocess
import tempfile
import getpass
import os
# default_image_dir - Path where Docker images (tarballs) will be stored
if os.geteuid() == 0:
default_image_dir = "/var/lib/virt-bootstrap/docker_images"
else:
default_image_dir = \
os.environ['HOME'] + "/.local/share/virt-bootstrap/docker_images"
def checksum(path, sum_type, sum_expected):
algorithm = getattr(hashlib, sum_type)
try:
fd = open(path, 'rb')
content = fd.read()
fd.close()
actual = algorithm(content).hexdigest()
return actual == sum_expected
except Exception:
return False
class FileSource:
def __init__(self, url, *args):
self.path = url.path
def unpack(self, dest):
# We assume tar is intelligent enough to find out
# the compression type to use and to strip leading '/',
# not sure if this is safe enough
subprocess.check_call(["tar", "xf", self.path, "-C", dest])
class DockerSource:
def __init__(self, url, username, password, insecure, no_cache):
self.registry = url.netloc
self.image = url.path
self.username = username
self.password = password
self.insecure = insecure
self.no_cache = no_cache
if self.image and not self.image.startswith('/'):
self.image = '/' + self.image
self.url = "docker://" + self.registry + self.image
def unpack(self, dest):
if self.no_cache:
tmp_dest = tempfile.mkdtemp('virt-bootstrap')
images_dir = tmp_dest
else:
if not os.path.exists(default_image_dir):
os.makedirs(default_image_dir)
images_dir = default_image_dir
try:
# Run skopeo copy into a tmp folder
# Note: we don't want to expose --src-cert-dir to users as
# they should place the certificates in the system
# folders for broader enablement
cmd = ["skopeo", "copy",
self.url,
"dir:%s" % images_dir]
if self.insecure:
cmd.append('--src-tls-verify=false')
if self.username:
if not self.password:
self.password = getpass.getpass()
cmd.append('--src-creds=%s:%s' % (self.username,
self.password))
subprocess.check_call(cmd)
# Get the layers list from the manifest
mf = open("%s/manifest.json" % images_dir, "r")
manifest = json.load(mf)
# FIXME We suppose the layers are ordered, is this true?
for layer in manifest['layers']:
sum_type, sum_value = layer['digest'].split(':')
layer_file = "%s/%s.tar" % (images_dir, sum_value)
print('layer_file: (%s) %s' % (sum_type, layer_file))
# Verify the checksum
if not checksum(layer_file, sum_type, sum_value):
raise Exception("Digest not matching: " + layer['digest'])
# untar layer into dest
subprocess.check_call(["tar", "xf", layer_file, "-C", dest])
except Exception:
raise
finally:
# Clean up
if self.no_cache:
shutil.rmtree(tmp_dest)
|
