mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 09:48:05 +03:00
cf36295fa4
Summary: This diff updates all license headers to use the new text and style. Also, a few internal files were missing the header, but now they have it. `fbcode/common/rust/netstring/` had the internal header, but now it has GPLV2PLUS - since that goes to Mononoke's Github too. Differential Revision: D17881539 fbshipit-source-id: b70d2ee41d2019fc7c2fe458627f0f7c01978186
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License found in the LICENSE file in the root
|
|
# directory of this source tree.
|
|
|
|
'''Utilities for dealing with tar files.'''
|
|
|
|
import os
|
|
import tarfile
|
|
|
|
from typing import List
|
|
|
|
|
|
def from_dirs(dirs: List[str], tar_filename: str):
|
|
'''Create a .tar.gz file from a given list of directories.'''
|
|
with tarfile.open(tar_filename, 'w:gz') as tar:
|
|
for dir in dirs:
|
|
tar.add(dir, arcname=os.path.basename(dir))
|
|
|
|
|
|
def extractall_safe(path: str, dest: str) -> str:
|
|
'''Extract a tar safely. Raise an exception if there are any bad paths.
|
|
|
|
Returns the subdirectory the files were extracted to.'''
|
|
tar = tarfile.open(path)
|
|
firstdir = _check_members(path, tar)
|
|
tar.extractall(dest)
|
|
return os.path.join(dest, firstdir)
|
|
|
|
|
|
def _check_members(path: str, tar: tarfile.TarFile) -> str:
|
|
firstdir = None
|
|
for finfo in tar:
|
|
if _badpath(finfo.name):
|
|
raise RuntimeError('{} has bad path: {}'.format(path, finfo.name))
|
|
if firstdir is None:
|
|
firstdir = _firstdir(finfo.name)
|
|
elif firstdir != _firstdir(finfo.name):
|
|
raise RuntimeError(
|
|
'{}: expected path {} to begin with {}'.
|
|
format(path, finfo.name, firstdir)
|
|
)
|
|
|
|
if firstdir is None:
|
|
raise RuntimeError('{}: empty tar file'.format(path))
|
|
|
|
return firstdir
|
|
|
|
|
|
def _badpath(path: str) -> bool:
|
|
return path.startswith('..') or path.startswith('/')
|
|
|
|
|
|
def _firstdir(rel_path: str) -> str:
|
|
return rel_path.split('/', 1)[0]
|