mirror of
https://github.com/theupdateframework/python-tuf
synced 2026-05-24 10:08:28 +00:00
028d1bc9f7
Currently, we are importing the "utils" module in tests/utils with "import utils". This could become a problem when there is another module with the same general name "utils" and could lead to import mistakes. Signed-off-by: Martin Vrachev <mvrachev@vmware.com>
245 lines
10 KiB
Python
Executable file
245 lines
10 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
# Copyright 2012 - 2017, New York University and the TUF contributors
|
|
# SPDX-License-Identifier: MIT OR Apache-2.0
|
|
|
|
"""
|
|
<Program Name>
|
|
test_mix_and_match_attack.py
|
|
|
|
<Author>
|
|
Konstantin Andrianov.
|
|
|
|
<Started>
|
|
March 27, 2012.
|
|
|
|
April 6, 2014.
|
|
Refactored to use the 'unittest' module (test conditions in code, rather
|
|
than verifying text output), use pre-generated repository files, and
|
|
discontinue use of the old repository tools. Modify the previous scenario
|
|
simulated for the mix-and-match attack. -vladimir.v.diaz
|
|
|
|
<Copyright>
|
|
See LICENSE-MIT OR LICENSE for licensing information.
|
|
|
|
<Purpose>
|
|
Simulate a mix-and-match attack. In a mix-and-match attack, an attacker is
|
|
able to trick clients into using a combination of metadata that never existed
|
|
together on the repository at the same time.
|
|
|
|
Note: There is no difference between 'updates' and 'target' files.
|
|
"""
|
|
|
|
# Help with Python 3 compatibility, where the print statement is a function, an
|
|
# implicit relative import is invalid, and the '/' operator performs true
|
|
# division. Example: print 'hello world' raises a 'SyntaxError' exception.
|
|
from __future__ import print_function
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import tempfile
|
|
import shutil
|
|
import logging
|
|
import unittest
|
|
import sys
|
|
|
|
import tuf.exceptions
|
|
import tuf.log
|
|
import tuf.client.updater as updater
|
|
import tuf.repository_tool as repo_tool
|
|
import tuf.unittest_toolbox as unittest_toolbox
|
|
import tuf.roledb
|
|
import tuf.keydb
|
|
|
|
from tests import utils
|
|
|
|
import six
|
|
|
|
# The repository tool is imported and logs console messages by default.
|
|
# Disable console log messages generated by this unit test.
|
|
repo_tool.disable_console_log_messages()
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
class TestMixAndMatchAttack(unittest_toolbox.Modified_TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
# Create a temporary directory to store the repository, metadata, and
|
|
# target files. 'temporary_directory' must be deleted in TearDownModule()
|
|
# so that temporary files are always removed, even when exceptions occur.
|
|
cls.temporary_directory = tempfile.mkdtemp(dir=os.getcwd())
|
|
|
|
# Launch a SimpleHTTPServer (serves files in the current directory).
|
|
# Test cases will request metadata and target files that have been
|
|
# pre-generated in 'tuf/tests/repository_data', which will be served by the
|
|
# SimpleHTTPServer launched here. The test cases of this unit test assume
|
|
# the pre-generated metadata files have a specific structure, such
|
|
# as a delegated role 'targets/role1', three target files, five key files,
|
|
# etc.
|
|
cls.server_process_handler = utils.TestServerProcess(log=logger)
|
|
|
|
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
# Cleans the resources and flush the logged lines (if any).
|
|
cls.server_process_handler.clean()
|
|
|
|
# Remove the temporary repository directory, which should contain all the
|
|
# metadata, targets, and key files generated of all the test cases.
|
|
shutil.rmtree(cls.temporary_directory)
|
|
|
|
|
|
|
|
|
|
def setUp(self):
|
|
# We are inheriting from custom class.
|
|
unittest_toolbox.Modified_TestCase.setUp(self)
|
|
|
|
self.repository_name = 'test_repository1'
|
|
|
|
# Copy the original repository files provided in the test folder so that
|
|
# any modifications made to repository files are restricted to the copies.
|
|
# The 'repository_data' directory is expected to exist in 'tuf/tests/'.
|
|
original_repository_files = os.path.join(os.getcwd(), 'repository_data')
|
|
temporary_repository_root = \
|
|
self.make_temp_directory(directory=self.temporary_directory)
|
|
|
|
# The original repository, keystore, and client directories will be copied
|
|
# for each test case.
|
|
original_repository = os.path.join(original_repository_files, 'repository')
|
|
original_client = os.path.join(original_repository_files, 'client')
|
|
original_keystore = os.path.join(original_repository_files, 'keystore')
|
|
|
|
# Save references to the often-needed client repository directories.
|
|
# Test cases need these references to access metadata and target files.
|
|
self.repository_directory = \
|
|
os.path.join(temporary_repository_root, 'repository')
|
|
self.client_directory = os.path.join(temporary_repository_root, 'client')
|
|
self.keystore_directory = os.path.join(temporary_repository_root, 'keystore')
|
|
|
|
# Copy the original 'repository', 'client', and 'keystore' directories
|
|
# to the temporary repository the test cases can use.
|
|
shutil.copytree(original_repository, self.repository_directory)
|
|
shutil.copytree(original_client, self.client_directory)
|
|
shutil.copytree(original_keystore, self.keystore_directory)
|
|
|
|
# Set the url prefix required by the 'tuf/client/updater.py' updater.
|
|
# 'path/to/tmp/repository' -> 'localhost:8001/tmp/repository'.
|
|
repository_basepath = self.repository_directory[len(os.getcwd()):]
|
|
url_prefix = 'http://localhost:' \
|
|
+ str(self.server_process_handler.port) + repository_basepath
|
|
|
|
# Setting 'tuf.settings.repository_directory' with the temporary client
|
|
# directory copied from the original repository files.
|
|
tuf.settings.repositories_directory = self.client_directory
|
|
self.repository_mirrors = {'mirror1': {'url_prefix': url_prefix,
|
|
'metadata_path': 'metadata',
|
|
'targets_path': 'targets'}}
|
|
|
|
# Create the repository instance. The test cases will use this client
|
|
# updater to refresh metadata, fetch target files, etc.
|
|
self.repository_updater = updater.Updater(self.repository_name,
|
|
self.repository_mirrors)
|
|
|
|
|
|
def tearDown(self):
|
|
# Modified_TestCase.tearDown() automatically deletes temporary files and
|
|
# directories that may have been created during each test case.
|
|
unittest_toolbox.Modified_TestCase.tearDown(self)
|
|
tuf.roledb.clear_roledb(clear_all=True)
|
|
tuf.keydb.clear_keydb(clear_all=True)
|
|
|
|
# Logs stdout and stderr from the sever subprocess.
|
|
self.server_process_handler.flush_log()
|
|
|
|
|
|
def test_with_tuf(self):
|
|
# Scenario:
|
|
# An attacker tries to trick the client into installing files indicated by
|
|
# a previous release of its corresponding metatadata. The outdated metadata
|
|
# is properly named and was previously valid, but is no longer current
|
|
# according to the latest 'snapshot.json' role. Generate a new snapshot of
|
|
# the repository after modifying a target file of 'role1.json'.
|
|
# Backup 'role1.json' (the delegated role to be updated, and then inserted
|
|
# again for the mix-and-match attack.)
|
|
role1_path = os.path.join(self.repository_directory, 'metadata', 'role1.json')
|
|
backup_role1 = os.path.join(self.repository_directory, 'role1.json.backup')
|
|
shutil.copy(role1_path, backup_role1)
|
|
|
|
# Backup 'file3.txt', specified by 'role1.json'.
|
|
file3_path = os.path.join(self.repository_directory, 'targets', 'file3.txt')
|
|
shutil.copy(file3_path, file3_path + '.backup')
|
|
|
|
# Re-generate the required metadata on the remote repository. The affected
|
|
# metadata must be properly updated and signed with 'repository_tool.py',
|
|
# otherwise the client will reject them as invalid metadata. The resulting
|
|
# metadata should be valid metadata.
|
|
repository = repo_tool.load_repository(self.repository_directory)
|
|
|
|
# Load the signing keys so that newly generated metadata is properly signed.
|
|
timestamp_keyfile = os.path.join(self.keystore_directory, 'timestamp_key')
|
|
role1_keyfile = os.path.join(self.keystore_directory, 'delegation_key')
|
|
snapshot_keyfile = os.path.join(self.keystore_directory, 'snapshot_key')
|
|
timestamp_private = \
|
|
repo_tool.import_ed25519_privatekey_from_file(timestamp_keyfile, 'password')
|
|
role1_private = \
|
|
repo_tool.import_ed25519_privatekey_from_file(role1_keyfile, 'password')
|
|
snapshot_private = \
|
|
repo_tool.import_ed25519_privatekey_from_file(snapshot_keyfile, 'password')
|
|
|
|
repository.targets('role1').load_signing_key(role1_private)
|
|
repository.snapshot.load_signing_key(snapshot_private)
|
|
repository.timestamp.load_signing_key(timestamp_private)
|
|
|
|
# Modify a 'role1.json' target file, and add it to its metadata so that a
|
|
# new version is generated.
|
|
with open(file3_path, 'wt') as file_object:
|
|
file_object.write('This is role2\'s target file.')
|
|
repository.targets('role1').add_target(os.path.basename(file3_path))
|
|
|
|
repository.writeall()
|
|
|
|
# Move the staged metadata to the "live" metadata.
|
|
shutil.rmtree(os.path.join(self.repository_directory, 'metadata'))
|
|
shutil.copytree(os.path.join(self.repository_directory, 'metadata.staged'),
|
|
os.path.join(self.repository_directory, 'metadata'))
|
|
|
|
# Insert the previously valid 'role1.json'. The TUF client should reject it.
|
|
shutil.move(backup_role1, role1_path)
|
|
|
|
# Verify that the TUF client detects unexpected metadata (previously valid,
|
|
# but not up-to-date with the latest snapshot of the repository) and
|
|
# refuses to continue the update process. Refresh top-level metadata so
|
|
# that the client is aware of the latest snapshot of the repository.
|
|
self.repository_updater.refresh()
|
|
|
|
try:
|
|
with utils.ignore_deprecation_warnings('tuf.client.updater'):
|
|
self.repository_updater.targets_of_role('role1')
|
|
|
|
# Verify that the specific
|
|
# 'tuf.exceptions.BadVersionNumberError' exception is raised by
|
|
# each mirror.
|
|
except tuf.exceptions.NoWorkingMirrorError as exception:
|
|
for mirror_url, mirror_error in six.iteritems(exception.mirror_errors):
|
|
url_prefix = self.repository_mirrors['mirror1']['url_prefix']
|
|
url_file = os.path.join(url_prefix, 'metadata', 'role1.json')
|
|
|
|
# Verify that 'role1.json' is the culprit.
|
|
self.assertEqual(url_file.replace('\\', '/'), mirror_url)
|
|
self.assertTrue(isinstance(
|
|
mirror_error, tuf.exceptions.BadVersionNumberError))
|
|
|
|
else:
|
|
self.fail('TUF did not prevent a mix-and-match attack.')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
utils.configure_test_logging(sys.argv)
|
|
unittest.main()
|