1 # Copyright (C) 2013 Intra2net AG
3 # This program is free software; you can redistribute it and/or modify
4 # it under the terms of the GNU Lesser General Public License as published
5 # by the Free Software Foundation; either version 3 of the License, or
6 # (at your option) any later version.
8 # This program is distributed in the hope that it will be useful,
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # GNU Lesser General Public License for more details.
13 # You should have received a copy of the GNU General Public License
14 # along with this program. If not, see
15 # <http://www.gnu.org/licenses/lgpl-3.0.html>
18 import os, unittest, hashlib, string
21 from deltatar import crypto
27 def new_volume_handler(tarobj, base_name, volume_number, encryption=None):
29 Handles the new volumes
31 volume_path = "%s.%d" % (base_name, volume_number)
32 tarobj.open_volume(volume_path, encryption=encryption)
34 def make_new_encryption_volume_handler(encryption):
36 Handles the new volumes using the right crypto context.
38 return lambda tarobj, base_name, volume_number: \
39 new_volume_handler (tarobj, base_name, volume_number,
40 encryption=encryption)
42 def closing_new_volume_handler(tarobj, base_name, volume_number):
44 Handles the new volumes
46 volume_path = "%s.%d" % (base_name, volume_number)
47 tarobj.fileobj.close()
48 tarobj.open_volume(volume_path)
50 class BaseTest(unittest.TestCase):
52 Test concatenated compression in tarfiles
57 Remove temporal files created by unit tests
59 os.system("rm -rf big big2 small small2 sample.* pdtcrypt-object-*.bin")
61 def create_file_low_entropy(self, path, length):
63 Creates a file with some gibberish inside, returning the md5sum of that
64 file. File path and length are specified as function arguments.
66 data = string.ascii_lowercase + string.digits + "\n"
68 # determine how often need to repeat data and how much part of data is
69 # left in the end to fill file up to length
70 n_blocks, remainder = divmod(length, len(data))
71 with open(path, 'w') as write_handle:
72 for _ in range(n_blocks):
73 write_handle.write(data)
74 write_handle.write(data[:remainder])
75 return self.md5sum(path)
78 def create_file_high_entropy(self, path, length):
80 Create a file with quality random content to counteract compression.
82 fd = os.open (path, os.O_WRONLY | os.O_CREAT | os.O_TRUNC)
86 todo = min (length - p, BLOCK_SIZE)
87 os.write (fd, os.urandom (todo))
92 return self.md5sum (path)
95 def create_file(self, path, length, random=False):
97 Creates a file with some gibberish inside, returning the md5sum of that
98 file. File path and length are specified as function arguments.
101 return self.create_file_high_entropy (path, length)
103 return self.create_file_low_entropy (path, length)
106 def create_symlink(self, linkname, path):
107 os.symlink(linkname, path)
108 return self.md5sum(path, linkname=linkname)
110 def md5sum(self, filename, linkname=None):
112 Returns the md5sum of a file specified by its filename/path or, if
113 ``linkname`` is specified, the hash of both paths (for symlinks).
117 with open(filename,'rb') as f:
118 for chunk in iter(lambda: f.read(128*md5.block_size), b''):
120 else: # symlink; hash paths
121 md5.update(filename.encode("UTF-8"))
122 md5.update(linkname.encode("UTF-8"))
123 return md5.hexdigest()