From 0ef2645af8871b1722608c36b4f5cdcb8d38067e Mon Sep 17 00:00:00 2001 From: Tomas Hrnciar Date: Sun, 19 Apr 2020 20:06:52 +0200 Subject: [PATCH 1/2] Prevent infinite recursion with pip wheel with $TMPDIR in $PWD During a build of extension module within `pip wheel` the source directory is recursively copied in a temporary directory. See https://github.com/pypa/pip/issues/7555 When the temporary directory is inside the source directory (for example by setting `TMPDIR=$PWD/tmp`) this caused an infinite recursion that ended in: [Errno 36] File name too long We prevent that buy never copying the target to the target in _copy_source_tree. Fixes https://github.com/pypa/pip/issues/7872 --- news/7872.bugfix | 1 + src/pip/_internal/download.py | 23 ++++++++++++++++++----- tests/data/src/extension/extension.c | 0 tests/data/src/extension/setup.py | 4 ++++ tests/functional/test_wheel.py | 11 +++++++++++ 5 files changed, 34 insertions(+), 5 deletions(-) create mode 100644 news/7872.bugfix create mode 100644 tests/data/src/extension/extension.c create mode 100644 tests/data/src/extension/setup.py diff --git a/news/7872.bugfix b/news/7872.bugfix new file mode 100644 index 00000000..3550d573 --- /dev/null +++ b/news/7872.bugfix @@ -0,0 +1 @@ +Prevent an infinite recursion with ``pip wheel`` when ``$TMPDIR`` is within the source directory. diff --git a/src/pip/_internal/download.py b/src/pip/_internal/download.py index 6567fc37..d5c01a41 100644 --- a/src/pip/_internal/download.py +++ b/src/pip/_internal/download.py @@ -350,12 +350,25 @@ def _copy2_ignoring_special_files(src, dest): def _copy_source_tree(source, target): # type: (str, str) -> None + target_abspath = os.path.abspath(target) + target_basename = os.path.basename(target_abspath) + target_dirname = os.path.dirname(target_abspath) + def ignore(d, names): - # Pulling in those directories can potentially be very slow, - # exclude the following directories if they appear in the top - # level dir (and only it). - # See discussion at https://github.com/pypa/pip/pull/6770 - return ['.tox', '.nox'] if d == source else [] + # type: (str, List[str]) -> List[str] + skipped = [] # type: List[str] + if d == source: + # Pulling in those directories can potentially be very slow, + # exclude the following directories if they appear in the top + # level dir (and only it). + # See discussion at https://github.com/pypa/pip/pull/6770 + skipped += ['.tox', '.nox'] + if os.path.abspath(d) == target_dirname: + # Prevent an infinite recursion if the target is in source. + # This can happen when TMPDIR is set to ${PWD}/... + # and we copy PWD to TMPDIR. + skipped += [target_basename] + return skipped kwargs = dict(ignore=ignore, symlinks=True) # type: CopytreeKwargs diff --git a/tests/data/src/extension/extension.c b/tests/data/src/extension/extension.c new file mode 100644 index 00000000..e69de29b diff --git a/tests/data/src/extension/setup.py b/tests/data/src/extension/setup.py new file mode 100644 index 00000000..b26302b0 --- /dev/null +++ b/tests/data/src/extension/setup.py @@ -0,0 +1,4 @@ +from setuptools import Extension, setup + +module = Extension('extension', sources=['extension.c']) +setup(name='extension', version='0.0.1', ext_modules = [module]) diff --git a/tests/functional/test_wheel.py b/tests/functional/test_wheel.py index 5ebc9ea4..9923c023 100644 --- a/tests/functional/test_wheel.py +++ b/tests/functional/test_wheel.py @@ -228,6 +228,17 @@ def test_pip_wheel_with_user_set_in_config(script, data, common_wheels): assert "Successfully built withpyproject" in result.stdout, result.stdout +def test_pip_wheel_ext_module_with_tmpdir_inside(script, data, common_wheels): + tmpdir = data.src / 'extension/tmp' + tmpdir.mkdir() + script.environ['TMPDIR'] = str(tmpdir) + result = script.pip( + 'wheel', data.src / 'extension', + '--no-index', '-f', common_wheels + ) + assert "Successfully built extension" in result.stdout, result.stdout + + @pytest.mark.network def test_pep517_wheels_are_not_confused_with_other_files(script, tmpdir, data): """Check correct wheels are copied. (#6196) From 168e98adb0f5d03887c4d88fc1733308761c3b8a Mon Sep 17 00:00:00 2001 From: Tomas Hrnciar Date: Sun, 19 Apr 2020 20:31:47 +0200 Subject: [PATCH 2/2] Avoid a test dependency on a C compiler, skip the test on Windows --- tests/functional/test_wheel.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/functional/test_wheel.py b/tests/functional/test_wheel.py index 9923c023..fa3df239 100644 --- a/tests/functional/test_wheel.py +++ b/tests/functional/test_wheel.py @@ -1,6 +1,7 @@ """'pip wheel' tests""" import os import re +import sys from os.path import exists import pytest @@ -228,10 +229,17 @@ def test_pip_wheel_with_user_set_in_config(script, data, common_wheels): assert "Successfully built withpyproject" in result.stdout, result.stdout +@pytest.mark.skipif(sys.platform.startswith('win'), + reason='The empty extension module does not work on Win') def test_pip_wheel_ext_module_with_tmpdir_inside(script, data, common_wheels): tmpdir = data.src / 'extension/tmp' tmpdir.mkdir() script.environ['TMPDIR'] = str(tmpdir) + + # To avoid a test dependency on a C compiler, we set the env vars to "noop" + # The .c source is empty anyway + script.environ['CC'] = script.environ['LDSHARED'] = str('true') + result = script.pip( 'wheel', data.src / 'extension', '--no-index', '-f', common_wheels