blob: 7d43b1ecc0da7ae1292f375650243f9ff74442a0 [file] [log] [blame]
Benjamin Peterson90f5ba52010-03-11 22:53:45 +00001#!/usr/bin/env python3
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002#-------------------------------------------------------------------
3# tarfile.py
4#-------------------------------------------------------------------
Christian Heimes9c1257e2007-11-04 11:37:22 +00005# Copyright (C) 2002 Lars Gustaebel <lars@gustaebel.de>
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00006# All rights reserved.
7#
8# Permission is hereby granted, free of charge, to any person
9# obtaining a copy of this software and associated documentation
10# files (the "Software"), to deal in the Software without
11# restriction, including without limitation the rights to use,
12# copy, modify, merge, publish, distribute, sublicense, and/or sell
13# copies of the Software, and to permit persons to whom the
14# Software is furnished to do so, subject to the following
15# conditions:
16#
17# The above copyright notice and this permission notice shall be
18# included in all copies or substantial portions of the Software.
19#
20# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
21# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
22# OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
23# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
24# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
25# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
26# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
27# OTHER DEALINGS IN THE SOFTWARE.
28#
29"""Read from and write to tar format archives.
30"""
31
Guido van Rossumd8faa362007-04-27 19:54:29 +000032version = "0.9.0"
Guido van Rossum98297ee2007-11-06 21:34:58 +000033__author__ = "Lars Gust\u00e4bel (lars@gustaebel.de)"
Senthil Kumaran7c9719c2011-07-28 22:32:49 +080034__date__ = "$Date: 2011-02-25 17:42:01 +0200 (Fri, 25 Feb 2011) $"
35__cvsid__ = "$Id: tarfile.py 88586 2011-02-25 15:42:01Z marc-andre.lemburg $"
Guido van Rossum98297ee2007-11-06 21:34:58 +000036__credits__ = "Gustavo Niemeyer, Niels Gust\u00e4bel, Richard Townsend."
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000037
38#---------
39# Imports
40#---------
Serhiy Storchakacf4a2f22015-03-11 17:18:03 +020041from builtins import open as bltn_open
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000042import sys
43import os
Eli Bendersky74c503b2012-01-03 06:26:13 +020044import io
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000045import shutil
46import stat
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000047import time
48import struct
Thomas Wouters89f507f2006-12-13 04:49:30 +000049import copy
Guido van Rossumd8faa362007-04-27 19:54:29 +000050import re
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000051
52try:
53 import grp, pwd
Brett Cannoncd171c82013-07-04 17:43:24 -040054except ImportError:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000055 grp = pwd = None
56
Brian Curtin16633fa2010-07-09 13:54:27 +000057# os.symlink on Windows prior to 6.0 raises NotImplementedError
58symlink_exception = (AttributeError, NotImplementedError)
59try:
Andrew Svetlov2606a6f2012-12-19 14:33:35 +020060 # OSError (winerror=1314) will be raised if the caller does not hold the
Brian Curtin16633fa2010-07-09 13:54:27 +000061 # SeCreateSymbolicLinkPrivilege privilege
Andrew Svetlov2606a6f2012-12-19 14:33:35 +020062 symlink_exception += (OSError,)
Brian Curtin16633fa2010-07-09 13:54:27 +000063except NameError:
64 pass
65
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000066# from tarfile import *
Martin Panter104dcda2016-01-16 06:59:13 +000067__all__ = ["TarFile", "TarInfo", "is_tarfile", "TarError", "ReadError",
68 "CompressionError", "StreamError", "ExtractError", "HeaderError",
69 "ENCODING", "USTAR_FORMAT", "GNU_FORMAT", "PAX_FORMAT",
70 "DEFAULT_FORMAT", "open"]
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000071
72#---------------------------------------------------------
73# tar constants
74#---------------------------------------------------------
Lars Gustäbelb506dc32007-08-07 18:36:16 +000075NUL = b"\0" # the null character
Guido van Rossumd8faa362007-04-27 19:54:29 +000076BLOCKSIZE = 512 # length of processing blocks
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000077RECORDSIZE = BLOCKSIZE * 20 # length of records
Lars Gustäbelb506dc32007-08-07 18:36:16 +000078GNU_MAGIC = b"ustar \0" # magic gnu tar string
79POSIX_MAGIC = b"ustar\x0000" # magic posix tar string
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000080
Guido van Rossumd8faa362007-04-27 19:54:29 +000081LENGTH_NAME = 100 # maximum length of a filename
82LENGTH_LINK = 100 # maximum length of a linkname
83LENGTH_PREFIX = 155 # maximum length of the prefix field
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000084
Lars Gustäbelb506dc32007-08-07 18:36:16 +000085REGTYPE = b"0" # regular file
86AREGTYPE = b"\0" # regular file
87LNKTYPE = b"1" # link (inside tarfile)
88SYMTYPE = b"2" # symbolic link
89CHRTYPE = b"3" # character special device
90BLKTYPE = b"4" # block special device
91DIRTYPE = b"5" # directory
92FIFOTYPE = b"6" # fifo special device
93CONTTYPE = b"7" # contiguous file
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000094
Lars Gustäbelb506dc32007-08-07 18:36:16 +000095GNUTYPE_LONGNAME = b"L" # GNU tar longname
96GNUTYPE_LONGLINK = b"K" # GNU tar longlink
97GNUTYPE_SPARSE = b"S" # GNU tar sparse file
Guido van Rossumd8faa362007-04-27 19:54:29 +000098
Lars Gustäbelb506dc32007-08-07 18:36:16 +000099XHDTYPE = b"x" # POSIX.1-2001 extended header
100XGLTYPE = b"g" # POSIX.1-2001 global header
101SOLARIS_XHDTYPE = b"X" # Solaris extended header
Guido van Rossumd8faa362007-04-27 19:54:29 +0000102
103USTAR_FORMAT = 0 # POSIX.1-1988 (ustar) format
104GNU_FORMAT = 1 # GNU tar format
105PAX_FORMAT = 2 # POSIX.1-2001 (pax) format
106DEFAULT_FORMAT = GNU_FORMAT
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000107
108#---------------------------------------------------------
109# tarfile constants
110#---------------------------------------------------------
Guido van Rossumd8faa362007-04-27 19:54:29 +0000111# File types that tarfile supports:
112SUPPORTED_TYPES = (REGTYPE, AREGTYPE, LNKTYPE,
113 SYMTYPE, DIRTYPE, FIFOTYPE,
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000114 CONTTYPE, CHRTYPE, BLKTYPE,
115 GNUTYPE_LONGNAME, GNUTYPE_LONGLINK,
116 GNUTYPE_SPARSE)
117
Guido van Rossumd8faa362007-04-27 19:54:29 +0000118# File types that will be treated as a regular file.
119REGULAR_TYPES = (REGTYPE, AREGTYPE,
120 CONTTYPE, GNUTYPE_SPARSE)
121
122# File types that are part of the GNU tar format.
123GNU_TYPES = (GNUTYPE_LONGNAME, GNUTYPE_LONGLINK,
124 GNUTYPE_SPARSE)
125
126# Fields from a pax header that override a TarInfo attribute.
127PAX_FIELDS = ("path", "linkpath", "size", "mtime",
128 "uid", "gid", "uname", "gname")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000129
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000130# Fields from a pax header that are affected by hdrcharset.
131PAX_NAME_FIELDS = {"path", "linkpath", "uname", "gname"}
132
Guido van Rossume7ba4952007-06-06 23:52:48 +0000133# Fields in a pax header that are numbers, all other fields
134# are treated as strings.
135PAX_NUMBER_FIELDS = {
136 "atime": float,
137 "ctime": float,
138 "mtime": float,
139 "uid": int,
140 "gid": int,
141 "size": int
142}
143
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000144#---------------------------------------------------------
Guido van Rossumd8faa362007-04-27 19:54:29 +0000145# initialization
146#---------------------------------------------------------
Victor Stinner0f35e2c2010-06-11 23:46:47 +0000147if os.name in ("nt", "ce"):
148 ENCODING = "utf-8"
149else:
150 ENCODING = sys.getfilesystemencoding()
Guido van Rossumd8faa362007-04-27 19:54:29 +0000151
152#---------------------------------------------------------
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000153# Some useful functions
154#---------------------------------------------------------
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000155
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000156def stn(s, length, encoding, errors):
157 """Convert a string to a null-terminated bytes object.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000158 """
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000159 s = s.encode(encoding, errors)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000160 return s[:length] + (length - len(s)) * NUL
Thomas Wouters477c8d52006-05-27 19:21:47 +0000161
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000162def nts(s, encoding, errors):
163 """Convert a null-terminated bytes object to a string.
Guido van Rossumd8faa362007-04-27 19:54:29 +0000164 """
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000165 p = s.find(b"\0")
166 if p != -1:
167 s = s[:p]
168 return s.decode(encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000169
Thomas Wouters477c8d52006-05-27 19:21:47 +0000170def nti(s):
171 """Convert a number field to a python number.
172 """
173 # There are two possible encodings for a number field, see
174 # itn() below.
Lars Gustäbelac3d1372011-10-14 12:46:40 +0200175 if s[0] in (0o200, 0o377):
176 n = 0
177 for i in range(len(s) - 1):
178 n <<= 8
179 n += s[i + 1]
180 if s[0] == 0o377:
181 n = -(256 ** (len(s) - 1) - n)
182 else:
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000183 try:
Lars Gustäbelb7a688b2015-07-02 19:38:38 +0200184 s = nts(s, "ascii", "strict")
185 n = int(s.strip() or "0", 8)
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000186 except ValueError:
Lars Gustäbel9520a432009-11-22 18:48:49 +0000187 raise InvalidHeaderError("invalid header")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000188 return n
189
Guido van Rossumd8faa362007-04-27 19:54:29 +0000190def itn(n, digits=8, format=DEFAULT_FORMAT):
Thomas Wouters477c8d52006-05-27 19:21:47 +0000191 """Convert a python number to a number field.
192 """
193 # POSIX 1003.1-1988 requires numbers to be encoded as a string of
194 # octal digits followed by a null-byte, this allows values up to
195 # (8**(digits-1))-1. GNU tar allows storing numbers greater than
Lars Gustäbelac3d1372011-10-14 12:46:40 +0200196 # that if necessary. A leading 0o200 or 0o377 byte indicate this
197 # particular encoding, the following digits-1 bytes are a big-endian
198 # base-256 representation. This allows values up to (256**(digits-1))-1.
199 # A 0o200 byte indicates a positive number, a 0o377 byte a negative
200 # number.
Thomas Wouters477c8d52006-05-27 19:21:47 +0000201 if 0 <= n < 8 ** (digits - 1):
Ethan Furmandf3ed242014-01-05 06:50:30 -0800202 s = bytes("%0*o" % (digits - 1, int(n)), "ascii") + NUL
Lars Gustäbelac3d1372011-10-14 12:46:40 +0200203 elif format == GNU_FORMAT and -256 ** (digits - 1) <= n < 256 ** (digits - 1):
204 if n >= 0:
205 s = bytearray([0o200])
206 else:
207 s = bytearray([0o377])
208 n = 256 ** digits + n
Thomas Wouters477c8d52006-05-27 19:21:47 +0000209
Guido van Rossum805365e2007-05-07 22:24:25 +0000210 for i in range(digits - 1):
Lars Gustäbelac3d1372011-10-14 12:46:40 +0200211 s.insert(1, n & 0o377)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000212 n >>= 8
Lars Gustäbelac3d1372011-10-14 12:46:40 +0200213 else:
214 raise ValueError("overflow in number field")
215
Thomas Wouters477c8d52006-05-27 19:21:47 +0000216 return s
217
218def calc_chksums(buf):
219 """Calculate the checksum for a member's header by summing up all
220 characters except for the chksum field which is treated as if
221 it was filled with spaces. According to the GNU tar sources,
222 some tars (Sun and NeXT) calculate chksum with signed char,
223 which will be different if there are chars in the buffer with
224 the high bit set. So we calculate two checksums, unsigned and
225 signed.
226 """
Ross Lagerwall468ff4c2012-05-17 19:49:27 +0200227 unsigned_chksum = 256 + sum(struct.unpack_from("148B8x356B", buf))
228 signed_chksum = 256 + sum(struct.unpack_from("148b8x356b", buf))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000229 return unsigned_chksum, signed_chksum
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000230
Lars Gustäbel03572682015-07-06 09:27:24 +0200231def copyfileobj(src, dst, length=None, exception=OSError):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000232 """Copy length bytes from fileobj src to fileobj dst.
233 If length is None, copy the entire content.
234 """
235 if length == 0:
236 return
237 if length is None:
238 shutil.copyfileobj(src, dst)
239 return
240
241 BUFSIZE = 16 * 1024
242 blocks, remainder = divmod(length, BUFSIZE)
Guido van Rossum805365e2007-05-07 22:24:25 +0000243 for b in range(blocks):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000244 buf = src.read(BUFSIZE)
245 if len(buf) < BUFSIZE:
Lars Gustäbel03572682015-07-06 09:27:24 +0200246 raise exception("unexpected end of data")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000247 dst.write(buf)
248
249 if remainder != 0:
250 buf = src.read(remainder)
251 if len(buf) < remainder:
Lars Gustäbel03572682015-07-06 09:27:24 +0200252 raise exception("unexpected end of data")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000253 dst.write(buf)
254 return
255
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000256def filemode(mode):
Giampaolo Rodola'ffa1d0b2012-05-15 15:30:25 +0200257 """Deprecated in this location; use stat.filemode."""
258 import warnings
259 warnings.warn("deprecated in favor of stat.filemode",
260 DeprecationWarning, 2)
261 return stat.filemode(mode)
262
Serhiy Storchaka3b4f1592014-02-05 20:53:36 +0200263def _safe_print(s):
264 encoding = getattr(sys.stdout, 'encoding', None)
265 if encoding is not None:
266 s = s.encode(encoding, 'backslashreplace').decode(encoding)
267 print(s, end=' ')
268
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000269
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000270class TarError(Exception):
271 """Base exception."""
272 pass
273class ExtractError(TarError):
274 """General exception for extract errors."""
275 pass
276class ReadError(TarError):
Ezio Melotti30b9d5d2013-08-17 15:50:46 +0300277 """Exception for unreadable tar archives."""
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000278 pass
279class CompressionError(TarError):
280 """Exception for unavailable compression methods."""
281 pass
282class StreamError(TarError):
283 """Exception for unsupported operations on stream-like TarFiles."""
284 pass
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000285class HeaderError(TarError):
Lars Gustäbel9520a432009-11-22 18:48:49 +0000286 """Base exception for header errors."""
287 pass
288class EmptyHeaderError(HeaderError):
289 """Exception for empty headers."""
290 pass
291class TruncatedHeaderError(HeaderError):
292 """Exception for truncated headers."""
293 pass
294class EOFHeaderError(HeaderError):
295 """Exception for end of file headers."""
296 pass
297class InvalidHeaderError(HeaderError):
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000298 """Exception for invalid headers."""
299 pass
Lars Gustäbel9520a432009-11-22 18:48:49 +0000300class SubsequentHeaderError(HeaderError):
301 """Exception for missing and invalid extended headers."""
302 pass
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000303
304#---------------------------
305# internal stream interface
306#---------------------------
307class _LowLevelFile:
308 """Low-level file object. Supports reading and writing.
309 It is used instead of a regular file object for streaming
310 access.
311 """
312
313 def __init__(self, name, mode):
314 mode = {
315 "r": os.O_RDONLY,
316 "w": os.O_WRONLY | os.O_CREAT | os.O_TRUNC,
317 }[mode]
318 if hasattr(os, "O_BINARY"):
319 mode |= os.O_BINARY
Lars Gustäbeld6eb70b2010-04-29 15:37:02 +0000320 self.fd = os.open(name, mode, 0o666)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000321
322 def close(self):
323 os.close(self.fd)
324
325 def read(self, size):
326 return os.read(self.fd, size)
327
328 def write(self, s):
329 os.write(self.fd, s)
330
331class _Stream:
332 """Class that serves as an adapter between TarFile and
333 a stream-like object. The stream-like object only
334 needs to have a read() or write() method and is accessed
335 blockwise. Use of gzip or bzip2 compression is possible.
336 A stream-like object could be for example: sys.stdin,
337 sys.stdout, a socket, a tape device etc.
338
339 _Stream is intended to be used only internally.
340 """
341
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000342 def __init__(self, name, mode, comptype, fileobj, bufsize):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000343 """Construct a _Stream object.
344 """
345 self._extfileobj = True
346 if fileobj is None:
347 fileobj = _LowLevelFile(name, mode)
348 self._extfileobj = False
349
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000350 if comptype == '*':
351 # Enable transparent compression detection for the
352 # stream interface
353 fileobj = _StreamProxy(fileobj)
354 comptype = fileobj.getcomptype()
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000355
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000356 self.name = name or ""
357 self.mode = mode
358 self.comptype = comptype
359 self.fileobj = fileobj
360 self.bufsize = bufsize
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000361 self.buf = b""
Guido van Rossume2a383d2007-01-15 16:59:06 +0000362 self.pos = 0
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000363 self.closed = False
364
Antoine Pitrou605c2932010-09-23 20:15:14 +0000365 try:
366 if comptype == "gz":
367 try:
368 import zlib
Brett Cannoncd171c82013-07-04 17:43:24 -0400369 except ImportError:
Antoine Pitrou605c2932010-09-23 20:15:14 +0000370 raise CompressionError("zlib module is not available")
371 self.zlib = zlib
372 self.crc = zlib.crc32(b"")
373 if mode == "r":
374 self._init_read_gz()
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +0100375 self.exception = zlib.error
Antoine Pitrou605c2932010-09-23 20:15:14 +0000376 else:
377 self._init_write_gz()
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000378
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +0100379 elif comptype == "bz2":
Antoine Pitrou605c2932010-09-23 20:15:14 +0000380 try:
381 import bz2
Brett Cannoncd171c82013-07-04 17:43:24 -0400382 except ImportError:
Antoine Pitrou605c2932010-09-23 20:15:14 +0000383 raise CompressionError("bz2 module is not available")
384 if mode == "r":
385 self.dbuf = b""
386 self.cmp = bz2.BZ2Decompressor()
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200387 self.exception = OSError
Antoine Pitrou605c2932010-09-23 20:15:14 +0000388 else:
389 self.cmp = bz2.BZ2Compressor()
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +0100390
391 elif comptype == "xz":
392 try:
393 import lzma
Brett Cannoncd171c82013-07-04 17:43:24 -0400394 except ImportError:
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +0100395 raise CompressionError("lzma module is not available")
396 if mode == "r":
397 self.dbuf = b""
398 self.cmp = lzma.LZMADecompressor()
399 self.exception = lzma.LZMAError
400 else:
401 self.cmp = lzma.LZMACompressor()
402
403 elif comptype != "tar":
404 raise CompressionError("unknown compression type %r" % comptype)
405
Antoine Pitrou605c2932010-09-23 20:15:14 +0000406 except:
407 if not self._extfileobj:
408 self.fileobj.close()
409 self.closed = True
410 raise
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000411
412 def __del__(self):
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000413 if hasattr(self, "closed") and not self.closed:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000414 self.close()
415
416 def _init_write_gz(self):
417 """Initialize for writing with gzip compression.
418 """
419 self.cmp = self.zlib.compressobj(9, self.zlib.DEFLATED,
420 -self.zlib.MAX_WBITS,
421 self.zlib.DEF_MEM_LEVEL,
422 0)
Guido van Rossume2a383d2007-01-15 16:59:06 +0000423 timestamp = struct.pack("<L", int(time.time()))
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000424 self.__write(b"\037\213\010\010" + timestamp + b"\002\377")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000425 if self.name.endswith(".gz"):
426 self.name = self.name[:-3]
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000427 # RFC1952 says we must use ISO-8859-1 for the FNAME field.
428 self.__write(self.name.encode("iso-8859-1", "replace") + NUL)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000429
430 def write(self, s):
431 """Write string s to the stream.
432 """
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000433 if self.comptype == "gz":
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000434 self.crc = self.zlib.crc32(s, self.crc)
435 self.pos += len(s)
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000436 if self.comptype != "tar":
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000437 s = self.cmp.compress(s)
438 self.__write(s)
439
440 def __write(self, s):
441 """Write string s to the stream if a whole new block
442 is ready to be written.
443 """
444 self.buf += s
445 while len(self.buf) > self.bufsize:
446 self.fileobj.write(self.buf[:self.bufsize])
447 self.buf = self.buf[self.bufsize:]
448
449 def close(self):
450 """Close the _Stream object. No operation should be
451 done on it afterwards.
452 """
453 if self.closed:
454 return
455
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000456 self.closed = True
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300457 try:
458 if self.mode == "w" and self.comptype != "tar":
459 self.buf += self.cmp.flush()
460
461 if self.mode == "w" and self.buf:
462 self.fileobj.write(self.buf)
463 self.buf = b""
464 if self.comptype == "gz":
Martin Panterb82032f2015-12-11 05:19:29 +0000465 self.fileobj.write(struct.pack("<L", self.crc))
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300466 self.fileobj.write(struct.pack("<L", self.pos & 0xffffFFFF))
467 finally:
468 if not self._extfileobj:
469 self.fileobj.close()
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000470
471 def _init_read_gz(self):
472 """Initialize for reading a gzip compressed fileobj.
473 """
474 self.cmp = self.zlib.decompressobj(-self.zlib.MAX_WBITS)
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000475 self.dbuf = b""
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000476
477 # taken from gzip.GzipFile with some alterations
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000478 if self.__read(2) != b"\037\213":
Thomas Wouters477c8d52006-05-27 19:21:47 +0000479 raise ReadError("not a gzip file")
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000480 if self.__read(1) != b"\010":
Thomas Wouters477c8d52006-05-27 19:21:47 +0000481 raise CompressionError("unsupported compression method")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000482
483 flag = ord(self.__read(1))
484 self.__read(6)
485
486 if flag & 4:
487 xlen = ord(self.__read(1)) + 256 * ord(self.__read(1))
488 self.read(xlen)
489 if flag & 8:
490 while True:
491 s = self.__read(1)
492 if not s or s == NUL:
493 break
494 if flag & 16:
495 while True:
496 s = self.__read(1)
497 if not s or s == NUL:
498 break
499 if flag & 2:
500 self.__read(2)
501
502 def tell(self):
503 """Return the stream's file pointer position.
504 """
505 return self.pos
506
507 def seek(self, pos=0):
508 """Set the stream's file pointer to pos. Negative seeking
509 is forbidden.
510 """
511 if pos - self.pos >= 0:
512 blocks, remainder = divmod(pos - self.pos, self.bufsize)
Guido van Rossum805365e2007-05-07 22:24:25 +0000513 for i in range(blocks):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000514 self.read(self.bufsize)
515 self.read(remainder)
516 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000517 raise StreamError("seeking backwards is not allowed")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000518 return self.pos
519
520 def read(self, size=None):
521 """Return the next size number of bytes from the stream.
522 If size is not defined, return all bytes of the stream
523 up to EOF.
524 """
525 if size is None:
526 t = []
527 while True:
528 buf = self._read(self.bufsize)
529 if not buf:
530 break
531 t.append(buf)
532 buf = "".join(t)
533 else:
534 buf = self._read(size)
535 self.pos += len(buf)
536 return buf
537
538 def _read(self, size):
539 """Return size bytes from the stream.
540 """
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000541 if self.comptype == "tar":
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000542 return self.__read(size)
543
544 c = len(self.dbuf)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000545 while c < size:
546 buf = self.__read(self.bufsize)
547 if not buf:
548 break
Guido van Rossumd8faa362007-04-27 19:54:29 +0000549 try:
550 buf = self.cmp.decompress(buf)
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +0100551 except self.exception:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000552 raise ReadError("invalid compressed data")
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000553 self.dbuf += buf
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000554 c += len(buf)
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000555 buf = self.dbuf[:size]
556 self.dbuf = self.dbuf[size:]
557 return buf
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000558
559 def __read(self, size):
560 """Return size bytes from stream. If internal buffer is empty,
561 read another block from the stream.
562 """
563 c = len(self.buf)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000564 while c < size:
565 buf = self.fileobj.read(self.bufsize)
566 if not buf:
567 break
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000568 self.buf += buf
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000569 c += len(buf)
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000570 buf = self.buf[:size]
571 self.buf = self.buf[size:]
572 return buf
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000573# class _Stream
574
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000575class _StreamProxy(object):
576 """Small proxy class that enables transparent compression
577 detection for the Stream interface (mode 'r|*').
578 """
579
580 def __init__(self, fileobj):
581 self.fileobj = fileobj
582 self.buf = self.fileobj.read(BLOCKSIZE)
583
584 def read(self, size):
585 self.read = self.fileobj.read
586 return self.buf
587
588 def getcomptype(self):
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +0100589 if self.buf.startswith(b"\x1f\x8b\x08"):
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000590 return "gz"
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +0100591 elif self.buf[0:3] == b"BZh" and self.buf[4:10] == b"1AY&SY":
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000592 return "bz2"
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +0100593 elif self.buf.startswith((b"\x5d\x00\x00\x80", b"\xfd7zXZ")):
594 return "xz"
595 else:
596 return "tar"
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000597
598 def close(self):
599 self.fileobj.close()
600# class StreamProxy
601
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000602#------------------------
603# Extraction file object
604#------------------------
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000605class _FileInFile(object):
606 """A thin wrapper around an existing file object that
607 provides a part of its data as an individual file
608 object.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000609 """
610
Lars Gustäbel9cbdd752010-10-29 09:08:19 +0000611 def __init__(self, fileobj, offset, size, blockinfo=None):
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000612 self.fileobj = fileobj
613 self.offset = offset
614 self.size = size
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000615 self.position = 0
Lars Gustäbel7a919e92012-05-05 18:15:03 +0200616 self.name = getattr(fileobj, "name", None)
617 self.closed = False
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000618
Lars Gustäbel9cbdd752010-10-29 09:08:19 +0000619 if blockinfo is None:
620 blockinfo = [(0, size)]
621
622 # Construct a map with data and zero blocks.
623 self.map_index = 0
624 self.map = []
625 lastpos = 0
626 realpos = self.offset
627 for offset, size in blockinfo:
628 if offset > lastpos:
629 self.map.append((False, lastpos, offset, None))
630 self.map.append((True, offset, offset + size, realpos))
631 realpos += size
632 lastpos = offset + size
633 if lastpos < self.size:
634 self.map.append((False, lastpos, self.size, None))
635
Lars Gustäbel7a919e92012-05-05 18:15:03 +0200636 def flush(self):
637 pass
638
639 def readable(self):
640 return True
641
642 def writable(self):
643 return False
644
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000645 def seekable(self):
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000646 return self.fileobj.seekable()
647
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000648 def tell(self):
649 """Return the current file position.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000650 """
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000651 return self.position
652
Lars Gustäbel7a919e92012-05-05 18:15:03 +0200653 def seek(self, position, whence=io.SEEK_SET):
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000654 """Seek to a position in the file.
655 """
Lars Gustäbel7a919e92012-05-05 18:15:03 +0200656 if whence == io.SEEK_SET:
657 self.position = min(max(position, 0), self.size)
658 elif whence == io.SEEK_CUR:
659 if position < 0:
660 self.position = max(self.position + position, 0)
661 else:
662 self.position = min(self.position + position, self.size)
663 elif whence == io.SEEK_END:
664 self.position = max(min(self.size + position, self.size), 0)
665 else:
666 raise ValueError("Invalid argument")
667 return self.position
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000668
669 def read(self, size=None):
670 """Read data from the file.
671 """
672 if size is None:
673 size = self.size - self.position
674 else:
675 size = min(size, self.size - self.position)
676
Lars Gustäbel9cbdd752010-10-29 09:08:19 +0000677 buf = b""
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000678 while size > 0:
Lars Gustäbel9cbdd752010-10-29 09:08:19 +0000679 while True:
680 data, start, stop, offset = self.map[self.map_index]
681 if start <= self.position < stop:
682 break
683 else:
684 self.map_index += 1
685 if self.map_index == len(self.map):
686 self.map_index = 0
687 length = min(size, stop - self.position)
688 if data:
Lars Gustäbeldd071042011-02-23 11:42:22 +0000689 self.fileobj.seek(offset + (self.position - start))
Lars Gustäbel03572682015-07-06 09:27:24 +0200690 b = self.fileobj.read(length)
691 if len(b) != length:
692 raise ReadError("unexpected end of data")
693 buf += b
Lars Gustäbel9cbdd752010-10-29 09:08:19 +0000694 else:
695 buf += NUL * length
696 size -= length
697 self.position += length
698 return buf
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000699
Lars Gustäbel7a919e92012-05-05 18:15:03 +0200700 def readinto(self, b):
701 buf = self.read(len(b))
702 b[:len(buf)] = buf
703 return len(buf)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000704
705 def close(self):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000706 self.closed = True
Lars Gustäbel7a919e92012-05-05 18:15:03 +0200707#class _FileInFile
Martin v. Löwisdf241532005-03-03 08:17:42 +0000708
Lars Gustäbelb062a2f2012-05-14 13:18:16 +0200709class ExFileObject(io.BufferedReader):
710
711 def __init__(self, tarfile, tarinfo):
712 fileobj = _FileInFile(tarfile.fileobj, tarinfo.offset_data,
713 tarinfo.size, tarinfo.sparse)
714 super().__init__(fileobj)
715#class ExFileObject
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000716
717#------------------
718# Exported Classes
719#------------------
720class TarInfo(object):
721 """Informational class which holds the details about an
722 archive member given by a tar header block.
723 TarInfo objects are returned by TarFile.getmember(),
724 TarFile.getmembers() and TarFile.gettarinfo() and are
725 usually created internally.
726 """
727
Lars Gustäbelc2ea8c62008-04-14 10:05:48 +0000728 __slots__ = ("name", "mode", "uid", "gid", "size", "mtime",
729 "chksum", "type", "linkname", "uname", "gname",
730 "devmajor", "devminor",
731 "offset", "offset_data", "pax_headers", "sparse",
732 "tarfile", "_sparse_structs", "_link_target")
733
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000734 def __init__(self, name=""):
735 """Construct a TarInfo object. name is the optional name
736 of the member.
737 """
Guido van Rossumd8faa362007-04-27 19:54:29 +0000738 self.name = name # member name
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000739 self.mode = 0o644 # file permissions
Thomas Wouters477c8d52006-05-27 19:21:47 +0000740 self.uid = 0 # user id
741 self.gid = 0 # group id
742 self.size = 0 # file size
743 self.mtime = 0 # modification time
744 self.chksum = 0 # header checksum
745 self.type = REGTYPE # member type
746 self.linkname = "" # link name
Lars Gustäbel331b8002010-10-04 15:18:47 +0000747 self.uname = "" # user name
748 self.gname = "" # group name
Thomas Wouters477c8d52006-05-27 19:21:47 +0000749 self.devmajor = 0 # device major number
750 self.devminor = 0 # device minor number
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000751
Thomas Wouters477c8d52006-05-27 19:21:47 +0000752 self.offset = 0 # the tar header starts here
753 self.offset_data = 0 # the file's data starts here
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000754
Lars Gustäbelc2ea8c62008-04-14 10:05:48 +0000755 self.sparse = None # sparse member information
Guido van Rossumd8faa362007-04-27 19:54:29 +0000756 self.pax_headers = {} # pax header information
757
758 # In pax headers the "name" and "linkname" field are called
759 # "path" and "linkpath".
760 def _getpath(self):
761 return self.name
762 def _setpath(self, name):
763 self.name = name
764 path = property(_getpath, _setpath)
765
766 def _getlinkpath(self):
767 return self.linkname
768 def _setlinkpath(self, linkname):
769 self.linkname = linkname
770 linkpath = property(_getlinkpath, _setlinkpath)
771
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000772 def __repr__(self):
773 return "<%s %r at %#x>" % (self.__class__.__name__,self.name,id(self))
774
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000775 def get_info(self):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000776 """Return the TarInfo's attributes as a dictionary.
777 """
778 info = {
Lars Gustäbelbfdfdda2009-08-28 19:59:59 +0000779 "name": self.name,
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000780 "mode": self.mode & 0o7777,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000781 "uid": self.uid,
782 "gid": self.gid,
783 "size": self.size,
784 "mtime": self.mtime,
785 "chksum": self.chksum,
786 "type": self.type,
Lars Gustäbelbfdfdda2009-08-28 19:59:59 +0000787 "linkname": self.linkname,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000788 "uname": self.uname,
789 "gname": self.gname,
790 "devmajor": self.devmajor,
791 "devminor": self.devminor
792 }
793
794 if info["type"] == DIRTYPE and not info["name"].endswith("/"):
795 info["name"] += "/"
796
797 return info
798
Victor Stinnerde629d42010-05-05 21:43:57 +0000799 def tobuf(self, format=DEFAULT_FORMAT, encoding=ENCODING, errors="surrogateescape"):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000800 """Return a tar header as a string of 512 byte blocks.
801 """
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000802 info = self.get_info()
Guido van Rossume7ba4952007-06-06 23:52:48 +0000803
Guido van Rossumd8faa362007-04-27 19:54:29 +0000804 if format == USTAR_FORMAT:
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000805 return self.create_ustar_header(info, encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000806 elif format == GNU_FORMAT:
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000807 return self.create_gnu_header(info, encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000808 elif format == PAX_FORMAT:
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000809 return self.create_pax_header(info, encoding)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000810 else:
811 raise ValueError("invalid format")
812
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000813 def create_ustar_header(self, info, encoding, errors):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000814 """Return the object as a ustar header block.
815 """
Guido van Rossumd8faa362007-04-27 19:54:29 +0000816 info["magic"] = POSIX_MAGIC
817
818 if len(info["linkname"]) > LENGTH_LINK:
819 raise ValueError("linkname is too long")
820
821 if len(info["name"]) > LENGTH_NAME:
822 info["prefix"], info["name"] = self._posix_split_name(info["name"])
823
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000824 return self._create_header(info, USTAR_FORMAT, encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000825
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000826 def create_gnu_header(self, info, encoding, errors):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000827 """Return the object as a GNU header block sequence.
828 """
Guido van Rossumd8faa362007-04-27 19:54:29 +0000829 info["magic"] = GNU_MAGIC
830
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000831 buf = b""
Guido van Rossumd8faa362007-04-27 19:54:29 +0000832 if len(info["linkname"]) > LENGTH_LINK:
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000833 buf += self._create_gnu_long_header(info["linkname"], GNUTYPE_LONGLINK, encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000834
835 if len(info["name"]) > LENGTH_NAME:
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000836 buf += self._create_gnu_long_header(info["name"], GNUTYPE_LONGNAME, encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000837
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000838 return buf + self._create_header(info, GNU_FORMAT, encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000839
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000840 def create_pax_header(self, info, encoding):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000841 """Return the object as a ustar header block. If it cannot be
842 represented this way, prepend a pax extended header sequence
843 with supplement information.
844 """
Guido van Rossumd8faa362007-04-27 19:54:29 +0000845 info["magic"] = POSIX_MAGIC
846 pax_headers = self.pax_headers.copy()
847
848 # Test string fields for values that exceed the field length or cannot
849 # be represented in ASCII encoding.
850 for name, hname, length in (
851 ("name", "path", LENGTH_NAME), ("linkname", "linkpath", LENGTH_LINK),
852 ("uname", "uname", 32), ("gname", "gname", 32)):
853
Guido van Rossume7ba4952007-06-06 23:52:48 +0000854 if hname in pax_headers:
855 # The pax header has priority.
856 continue
857
Guido van Rossumd8faa362007-04-27 19:54:29 +0000858 # Try to encode the string as ASCII.
859 try:
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000860 info[name].encode("ascii", "strict")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000861 except UnicodeEncodeError:
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000862 pax_headers[hname] = info[name]
Guido van Rossumd8faa362007-04-27 19:54:29 +0000863 continue
864
Guido van Rossume7ba4952007-06-06 23:52:48 +0000865 if len(info[name]) > length:
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000866 pax_headers[hname] = info[name]
Guido van Rossumd8faa362007-04-27 19:54:29 +0000867
868 # Test number fields for values that exceed the field limit or values
869 # that like to be stored as float.
870 for name, digits in (("uid", 8), ("gid", 8), ("size", 12), ("mtime", 12)):
Guido van Rossume7ba4952007-06-06 23:52:48 +0000871 if name in pax_headers:
872 # The pax header has priority. Avoid overflow.
873 info[name] = 0
874 continue
875
Guido van Rossumd8faa362007-04-27 19:54:29 +0000876 val = info[name]
877 if not 0 <= val < 8 ** (digits - 1) or isinstance(val, float):
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000878 pax_headers[name] = str(val)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000879 info[name] = 0
880
Guido van Rossume7ba4952007-06-06 23:52:48 +0000881 # Create a pax extended header if necessary.
Guido van Rossumd8faa362007-04-27 19:54:29 +0000882 if pax_headers:
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000883 buf = self._create_pax_generic_header(pax_headers, XHDTYPE, encoding)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000884 else:
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000885 buf = b""
Guido van Rossumd8faa362007-04-27 19:54:29 +0000886
Lars Gustäbel3741eff2007-08-21 12:17:05 +0000887 return buf + self._create_header(info, USTAR_FORMAT, "ascii", "replace")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000888
889 @classmethod
Lars Gustäbel3741eff2007-08-21 12:17:05 +0000890 def create_pax_global_header(cls, pax_headers):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000891 """Return the object as a pax global header block sequence.
892 """
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000893 return cls._create_pax_generic_header(pax_headers, XGLTYPE, "utf-8")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000894
895 def _posix_split_name(self, name):
896 """Split a name longer than 100 chars into a prefix
897 and a name part.
898 """
899 prefix = name[:LENGTH_PREFIX + 1]
900 while prefix and prefix[-1] != "/":
901 prefix = prefix[:-1]
902
903 name = name[len(prefix):]
904 prefix = prefix[:-1]
905
906 if not prefix or len(name) > LENGTH_NAME:
907 raise ValueError("name is too long")
908 return prefix, name
909
910 @staticmethod
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000911 def _create_header(info, format, encoding, errors):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000912 """Return a header block. info is a dictionary with file
913 information, format must be one of the *_FORMAT constants.
914 """
915 parts = [
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000916 stn(info.get("name", ""), 100, encoding, errors),
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000917 itn(info.get("mode", 0) & 0o7777, 8, format),
Guido van Rossumd8faa362007-04-27 19:54:29 +0000918 itn(info.get("uid", 0), 8, format),
919 itn(info.get("gid", 0), 8, format),
920 itn(info.get("size", 0), 12, format),
921 itn(info.get("mtime", 0), 12, format),
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000922 b" ", # checksum field
Guido van Rossumd8faa362007-04-27 19:54:29 +0000923 info.get("type", REGTYPE),
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000924 stn(info.get("linkname", ""), 100, encoding, errors),
925 info.get("magic", POSIX_MAGIC),
Lars Gustäbel331b8002010-10-04 15:18:47 +0000926 stn(info.get("uname", ""), 32, encoding, errors),
927 stn(info.get("gname", ""), 32, encoding, errors),
Guido van Rossumd8faa362007-04-27 19:54:29 +0000928 itn(info.get("devmajor", 0), 8, format),
929 itn(info.get("devminor", 0), 8, format),
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000930 stn(info.get("prefix", ""), 155, encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000931 ]
932
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000933 buf = struct.pack("%ds" % BLOCKSIZE, b"".join(parts))
Guido van Rossumd8faa362007-04-27 19:54:29 +0000934 chksum = calc_chksums(buf[-BLOCKSIZE:])[0]
Lars Gustäbela280ca752007-08-28 07:34:33 +0000935 buf = buf[:-364] + bytes("%06o\0" % chksum, "ascii") + buf[-357:]
Guido van Rossumd8faa362007-04-27 19:54:29 +0000936 return buf
937
938 @staticmethod
939 def _create_payload(payload):
940 """Return the string payload filled with zero bytes
941 up to the next 512 byte border.
942 """
943 blocks, remainder = divmod(len(payload), BLOCKSIZE)
944 if remainder > 0:
945 payload += (BLOCKSIZE - remainder) * NUL
946 return payload
947
948 @classmethod
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000949 def _create_gnu_long_header(cls, name, type, encoding, errors):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000950 """Return a GNUTYPE_LONGNAME or GNUTYPE_LONGLINK sequence
951 for name.
952 """
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000953 name = name.encode(encoding, errors) + NUL
Guido van Rossumd8faa362007-04-27 19:54:29 +0000954
955 info = {}
956 info["name"] = "././@LongLink"
957 info["type"] = type
958 info["size"] = len(name)
959 info["magic"] = GNU_MAGIC
960
961 # create extended header + name blocks.
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000962 return cls._create_header(info, USTAR_FORMAT, encoding, errors) + \
Guido van Rossumd8faa362007-04-27 19:54:29 +0000963 cls._create_payload(name)
964
965 @classmethod
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000966 def _create_pax_generic_header(cls, pax_headers, type, encoding):
967 """Return a POSIX.1-2008 extended or global header sequence
Guido van Rossumd8faa362007-04-27 19:54:29 +0000968 that contains a list of keyword, value pairs. The values
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000969 must be strings.
Guido van Rossumd8faa362007-04-27 19:54:29 +0000970 """
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000971 # Check if one of the fields contains surrogate characters and thereby
972 # forces hdrcharset=BINARY, see _proc_pax() for more information.
973 binary = False
974 for keyword, value in pax_headers.items():
975 try:
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000976 value.encode("utf-8", "strict")
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000977 except UnicodeEncodeError:
978 binary = True
979 break
980
Lars Gustäbelb506dc32007-08-07 18:36:16 +0000981 records = b""
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000982 if binary:
983 # Put the hdrcharset field at the beginning of the header.
984 records += b"21 hdrcharset=BINARY\n"
985
Guido van Rossumd8faa362007-04-27 19:54:29 +0000986 for keyword, value in pax_headers.items():
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000987 keyword = keyword.encode("utf-8")
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000988 if binary:
989 # Try to restore the original byte representation of `value'.
990 # Needless to say, that the encoding must match the string.
991 value = value.encode(encoding, "surrogateescape")
992 else:
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000993 value = value.encode("utf-8")
Lars Gustäbel1465cc22010-05-17 18:02:50 +0000994
Guido van Rossumd8faa362007-04-27 19:54:29 +0000995 l = len(keyword) + len(value) + 3 # ' ' + '=' + '\n'
996 n = p = 0
997 while True:
998 n = l + len(str(p))
999 if n == p:
1000 break
1001 p = n
Lars Gustäbela280ca752007-08-28 07:34:33 +00001002 records += bytes(str(p), "ascii") + b" " + keyword + b"=" + value + b"\n"
Guido van Rossumd8faa362007-04-27 19:54:29 +00001003
1004 # We use a hardcoded "././@PaxHeader" name like star does
1005 # instead of the one that POSIX recommends.
1006 info = {}
1007 info["name"] = "././@PaxHeader"
1008 info["type"] = type
1009 info["size"] = len(records)
1010 info["magic"] = POSIX_MAGIC
1011
1012 # Create pax header + record blocks.
Lars Gustäbel3741eff2007-08-21 12:17:05 +00001013 return cls._create_header(info, USTAR_FORMAT, "ascii", "replace") + \
Guido van Rossumd8faa362007-04-27 19:54:29 +00001014 cls._create_payload(records)
1015
Guido van Rossum75b64e62005-01-16 00:16:11 +00001016 @classmethod
Lars Gustäbelb506dc32007-08-07 18:36:16 +00001017 def frombuf(cls, buf, encoding, errors):
1018 """Construct a TarInfo object from a 512 byte bytes object.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001019 """
Lars Gustäbel9520a432009-11-22 18:48:49 +00001020 if len(buf) == 0:
1021 raise EmptyHeaderError("empty header")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001022 if len(buf) != BLOCKSIZE:
Lars Gustäbel9520a432009-11-22 18:48:49 +00001023 raise TruncatedHeaderError("truncated header")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001024 if buf.count(NUL) == BLOCKSIZE:
Lars Gustäbel9520a432009-11-22 18:48:49 +00001025 raise EOFHeaderError("end of file header")
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001026
1027 chksum = nti(buf[148:156])
1028 if chksum not in calc_chksums(buf):
Lars Gustäbel9520a432009-11-22 18:48:49 +00001029 raise InvalidHeaderError("bad checksum")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001030
Guido van Rossumd8faa362007-04-27 19:54:29 +00001031 obj = cls()
Lars Gustäbelb506dc32007-08-07 18:36:16 +00001032 obj.name = nts(buf[0:100], encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001033 obj.mode = nti(buf[100:108])
1034 obj.uid = nti(buf[108:116])
1035 obj.gid = nti(buf[116:124])
1036 obj.size = nti(buf[124:136])
1037 obj.mtime = nti(buf[136:148])
1038 obj.chksum = chksum
1039 obj.type = buf[156:157]
Lars Gustäbelb506dc32007-08-07 18:36:16 +00001040 obj.linkname = nts(buf[157:257], encoding, errors)
1041 obj.uname = nts(buf[265:297], encoding, errors)
1042 obj.gname = nts(buf[297:329], encoding, errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001043 obj.devmajor = nti(buf[329:337])
1044 obj.devminor = nti(buf[337:345])
Lars Gustäbelb506dc32007-08-07 18:36:16 +00001045 prefix = nts(buf[345:500], encoding, errors)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001046
Guido van Rossumd8faa362007-04-27 19:54:29 +00001047 # Old V7 tar format represents a directory as a regular
1048 # file with a trailing slash.
1049 if obj.type == AREGTYPE and obj.name.endswith("/"):
1050 obj.type = DIRTYPE
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001051
Lars Gustäbelc2ea8c62008-04-14 10:05:48 +00001052 # The old GNU sparse format occupies some of the unused
1053 # space in the buffer for up to 4 sparse structures.
1054 # Save the them for later processing in _proc_sparse().
1055 if obj.type == GNUTYPE_SPARSE:
1056 pos = 386
1057 structs = []
1058 for i in range(4):
1059 try:
1060 offset = nti(buf[pos:pos + 12])
1061 numbytes = nti(buf[pos + 12:pos + 24])
1062 except ValueError:
1063 break
1064 structs.append((offset, numbytes))
1065 pos += 24
1066 isextended = bool(buf[482])
1067 origsize = nti(buf[483:495])
1068 obj._sparse_structs = (structs, isextended, origsize)
1069
Guido van Rossumd8faa362007-04-27 19:54:29 +00001070 # Remove redundant slashes from directories.
1071 if obj.isdir():
1072 obj.name = obj.name.rstrip("/")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001073
Guido van Rossumd8faa362007-04-27 19:54:29 +00001074 # Reconstruct a ustar longname.
1075 if prefix and obj.type not in GNU_TYPES:
1076 obj.name = prefix + "/" + obj.name
1077 return obj
1078
1079 @classmethod
1080 def fromtarfile(cls, tarfile):
1081 """Return the next TarInfo object from TarFile object
1082 tarfile.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001083 """
Guido van Rossumd8faa362007-04-27 19:54:29 +00001084 buf = tarfile.fileobj.read(BLOCKSIZE)
Lars Gustäbelb506dc32007-08-07 18:36:16 +00001085 obj = cls.frombuf(buf, tarfile.encoding, tarfile.errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001086 obj.offset = tarfile.fileobj.tell() - BLOCKSIZE
1087 return obj._proc_member(tarfile)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001088
Guido van Rossumd8faa362007-04-27 19:54:29 +00001089 #--------------------------------------------------------------------------
1090 # The following are methods that are called depending on the type of a
1091 # member. The entry point is _proc_member() which can be overridden in a
1092 # subclass to add custom _proc_*() methods. A _proc_*() method MUST
1093 # implement the following
1094 # operations:
1095 # 1. Set self.offset_data to the position where the data blocks begin,
1096 # if there is data that follows.
1097 # 2. Set tarfile.offset to the position where the next member's header will
1098 # begin.
1099 # 3. Return self or another valid TarInfo object.
1100 def _proc_member(self, tarfile):
1101 """Choose the right processing method depending on
1102 the type and call it.
Thomas Wouters89f507f2006-12-13 04:49:30 +00001103 """
Guido van Rossumd8faa362007-04-27 19:54:29 +00001104 if self.type in (GNUTYPE_LONGNAME, GNUTYPE_LONGLINK):
1105 return self._proc_gnulong(tarfile)
1106 elif self.type == GNUTYPE_SPARSE:
1107 return self._proc_sparse(tarfile)
1108 elif self.type in (XHDTYPE, XGLTYPE, SOLARIS_XHDTYPE):
1109 return self._proc_pax(tarfile)
1110 else:
1111 return self._proc_builtin(tarfile)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001112
Guido van Rossumd8faa362007-04-27 19:54:29 +00001113 def _proc_builtin(self, tarfile):
1114 """Process a builtin type or an unknown type which
1115 will be treated as a regular file.
1116 """
1117 self.offset_data = tarfile.fileobj.tell()
1118 offset = self.offset_data
1119 if self.isreg() or self.type not in SUPPORTED_TYPES:
1120 # Skip the following data blocks.
1121 offset += self._block(self.size)
1122 tarfile.offset = offset
Thomas Wouters89f507f2006-12-13 04:49:30 +00001123
Guido van Rossume7ba4952007-06-06 23:52:48 +00001124 # Patch the TarInfo object with saved global
Guido van Rossumd8faa362007-04-27 19:54:29 +00001125 # header information.
Guido van Rossume7ba4952007-06-06 23:52:48 +00001126 self._apply_pax_info(tarfile.pax_headers, tarfile.encoding, tarfile.errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001127
1128 return self
1129
1130 def _proc_gnulong(self, tarfile):
1131 """Process the blocks that hold a GNU longname
1132 or longlink member.
1133 """
1134 buf = tarfile.fileobj.read(self._block(self.size))
1135
1136 # Fetch the next header and process it.
Lars Gustäbel9520a432009-11-22 18:48:49 +00001137 try:
1138 next = self.fromtarfile(tarfile)
1139 except HeaderError:
1140 raise SubsequentHeaderError("missing or bad subsequent header")
Guido van Rossumd8faa362007-04-27 19:54:29 +00001141
1142 # Patch the TarInfo object from the next header with
1143 # the longname information.
1144 next.offset = self.offset
1145 if self.type == GNUTYPE_LONGNAME:
Lars Gustäbelb506dc32007-08-07 18:36:16 +00001146 next.name = nts(buf, tarfile.encoding, tarfile.errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001147 elif self.type == GNUTYPE_LONGLINK:
Lars Gustäbelb506dc32007-08-07 18:36:16 +00001148 next.linkname = nts(buf, tarfile.encoding, tarfile.errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001149
1150 return next
1151
1152 def _proc_sparse(self, tarfile):
1153 """Process a GNU sparse header plus extra headers.
1154 """
Lars Gustäbelc2ea8c62008-04-14 10:05:48 +00001155 # We already collected some sparse structures in frombuf().
1156 structs, isextended, origsize = self._sparse_structs
1157 del self._sparse_structs
Guido van Rossumd8faa362007-04-27 19:54:29 +00001158
Lars Gustäbelc2ea8c62008-04-14 10:05:48 +00001159 # Collect sparse structures from extended header blocks.
Lars Gustäbelb506dc32007-08-07 18:36:16 +00001160 while isextended:
Guido van Rossumd8faa362007-04-27 19:54:29 +00001161 buf = tarfile.fileobj.read(BLOCKSIZE)
1162 pos = 0
Guido van Rossum805365e2007-05-07 22:24:25 +00001163 for i in range(21):
Guido van Rossumd8faa362007-04-27 19:54:29 +00001164 try:
1165 offset = nti(buf[pos:pos + 12])
1166 numbytes = nti(buf[pos + 12:pos + 24])
1167 except ValueError:
1168 break
Lars Gustäbel9cbdd752010-10-29 09:08:19 +00001169 if offset and numbytes:
1170 structs.append((offset, numbytes))
Guido van Rossumd8faa362007-04-27 19:54:29 +00001171 pos += 24
Lars Gustäbelb506dc32007-08-07 18:36:16 +00001172 isextended = bool(buf[504])
Lars Gustäbel9cbdd752010-10-29 09:08:19 +00001173 self.sparse = structs
Guido van Rossumd8faa362007-04-27 19:54:29 +00001174
1175 self.offset_data = tarfile.fileobj.tell()
1176 tarfile.offset = self.offset_data + self._block(self.size)
1177 self.size = origsize
Guido van Rossumd8faa362007-04-27 19:54:29 +00001178 return self
1179
1180 def _proc_pax(self, tarfile):
1181 """Process an extended or global header as described in
Lars Gustäbel1465cc22010-05-17 18:02:50 +00001182 POSIX.1-2008.
Guido van Rossumd8faa362007-04-27 19:54:29 +00001183 """
1184 # Read the header information.
1185 buf = tarfile.fileobj.read(self._block(self.size))
1186
1187 # A pax header stores supplemental information for either
1188 # the following file (extended) or all following files
1189 # (global).
1190 if self.type == XGLTYPE:
1191 pax_headers = tarfile.pax_headers
1192 else:
1193 pax_headers = tarfile.pax_headers.copy()
1194
Lars Gustäbel1465cc22010-05-17 18:02:50 +00001195 # Check if the pax header contains a hdrcharset field. This tells us
1196 # the encoding of the path, linkpath, uname and gname fields. Normally,
1197 # these fields are UTF-8 encoded but since POSIX.1-2008 tar
1198 # implementations are allowed to store them as raw binary strings if
1199 # the translation to UTF-8 fails.
1200 match = re.search(br"\d+ hdrcharset=([^\n]+)\n", buf)
1201 if match is not None:
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001202 pax_headers["hdrcharset"] = match.group(1).decode("utf-8")
Lars Gustäbel1465cc22010-05-17 18:02:50 +00001203
1204 # For the time being, we don't care about anything other than "BINARY".
1205 # The only other value that is currently allowed by the standard is
1206 # "ISO-IR 10646 2000 UTF-8" in other words UTF-8.
1207 hdrcharset = pax_headers.get("hdrcharset")
1208 if hdrcharset == "BINARY":
1209 encoding = tarfile.encoding
1210 else:
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001211 encoding = "utf-8"
Lars Gustäbel1465cc22010-05-17 18:02:50 +00001212
Guido van Rossumd8faa362007-04-27 19:54:29 +00001213 # Parse pax header information. A record looks like that:
1214 # "%d %s=%s\n" % (length, keyword, value). length is the size
1215 # of the complete record including the length field itself and
Guido van Rossume7ba4952007-06-06 23:52:48 +00001216 # the newline. keyword and value are both UTF-8 encoded strings.
Antoine Pitroufd036452008-08-19 17:56:33 +00001217 regex = re.compile(br"(\d+) ([^=]+)=")
Guido van Rossumd8faa362007-04-27 19:54:29 +00001218 pos = 0
1219 while True:
1220 match = regex.match(buf, pos)
1221 if not match:
1222 break
1223
1224 length, keyword = match.groups()
1225 length = int(length)
1226 value = buf[match.end(2) + 1:match.start(1) + length - 1]
1227
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001228 # Normally, we could just use "utf-8" as the encoding and "strict"
Lars Gustäbel1465cc22010-05-17 18:02:50 +00001229 # as the error handler, but we better not take the risk. For
1230 # example, GNU tar <= 1.23 is known to store filenames it cannot
1231 # translate to UTF-8 as raw strings (unfortunately without a
1232 # hdrcharset=BINARY header).
1233 # We first try the strict standard encoding, and if that fails we
1234 # fall back on the user's encoding and error handler.
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001235 keyword = self._decode_pax_field(keyword, "utf-8", "utf-8",
Lars Gustäbel1465cc22010-05-17 18:02:50 +00001236 tarfile.errors)
1237 if keyword in PAX_NAME_FIELDS:
1238 value = self._decode_pax_field(value, encoding, tarfile.encoding,
1239 tarfile.errors)
1240 else:
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001241 value = self._decode_pax_field(value, "utf-8", "utf-8",
Lars Gustäbel1465cc22010-05-17 18:02:50 +00001242 tarfile.errors)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001243
1244 pax_headers[keyword] = value
1245 pos += length
1246
Guido van Rossume7ba4952007-06-06 23:52:48 +00001247 # Fetch the next header.
Lars Gustäbel9520a432009-11-22 18:48:49 +00001248 try:
1249 next = self.fromtarfile(tarfile)
1250 except HeaderError:
1251 raise SubsequentHeaderError("missing or bad subsequent header")
Guido van Rossumd8faa362007-04-27 19:54:29 +00001252
Lars Gustäbel9cbdd752010-10-29 09:08:19 +00001253 # Process GNU sparse information.
1254 if "GNU.sparse.map" in pax_headers:
1255 # GNU extended sparse format version 0.1.
1256 self._proc_gnusparse_01(next, pax_headers)
1257
1258 elif "GNU.sparse.size" in pax_headers:
1259 # GNU extended sparse format version 0.0.
1260 self._proc_gnusparse_00(next, pax_headers, buf)
1261
1262 elif pax_headers.get("GNU.sparse.major") == "1" and pax_headers.get("GNU.sparse.minor") == "0":
1263 # GNU extended sparse format version 1.0.
1264 self._proc_gnusparse_10(next, pax_headers, tarfile)
1265
Guido van Rossume7ba4952007-06-06 23:52:48 +00001266 if self.type in (XHDTYPE, SOLARIS_XHDTYPE):
Guido van Rossume7ba4952007-06-06 23:52:48 +00001267 # Patch the TarInfo object with the extended header info.
1268 next._apply_pax_info(pax_headers, tarfile.encoding, tarfile.errors)
1269 next.offset = self.offset
1270
1271 if "size" in pax_headers:
1272 # If the extended header replaces the size field,
1273 # we need to recalculate the offset where the next
1274 # header starts.
1275 offset = next.offset_data
1276 if next.isreg() or next.type not in SUPPORTED_TYPES:
1277 offset += next._block(next.size)
1278 tarfile.offset = offset
1279
1280 return next
1281
Lars Gustäbel9cbdd752010-10-29 09:08:19 +00001282 def _proc_gnusparse_00(self, next, pax_headers, buf):
1283 """Process a GNU tar extended sparse header, version 0.0.
1284 """
1285 offsets = []
1286 for match in re.finditer(br"\d+ GNU.sparse.offset=(\d+)\n", buf):
1287 offsets.append(int(match.group(1)))
1288 numbytes = []
1289 for match in re.finditer(br"\d+ GNU.sparse.numbytes=(\d+)\n", buf):
1290 numbytes.append(int(match.group(1)))
1291 next.sparse = list(zip(offsets, numbytes))
1292
1293 def _proc_gnusparse_01(self, next, pax_headers):
1294 """Process a GNU tar extended sparse header, version 0.1.
1295 """
1296 sparse = [int(x) for x in pax_headers["GNU.sparse.map"].split(",")]
1297 next.sparse = list(zip(sparse[::2], sparse[1::2]))
1298
1299 def _proc_gnusparse_10(self, next, pax_headers, tarfile):
1300 """Process a GNU tar extended sparse header, version 1.0.
1301 """
1302 fields = None
1303 sparse = []
1304 buf = tarfile.fileobj.read(BLOCKSIZE)
1305 fields, buf = buf.split(b"\n", 1)
1306 fields = int(fields)
1307 while len(sparse) < fields * 2:
1308 if b"\n" not in buf:
1309 buf += tarfile.fileobj.read(BLOCKSIZE)
1310 number, buf = buf.split(b"\n", 1)
1311 sparse.append(int(number))
1312 next.offset_data = tarfile.fileobj.tell()
1313 next.sparse = list(zip(sparse[::2], sparse[1::2]))
1314
Guido van Rossume7ba4952007-06-06 23:52:48 +00001315 def _apply_pax_info(self, pax_headers, encoding, errors):
1316 """Replace fields with supplemental information from a previous
1317 pax extended or global header.
1318 """
1319 for keyword, value in pax_headers.items():
Lars Gustäbel9cbdd752010-10-29 09:08:19 +00001320 if keyword == "GNU.sparse.name":
1321 setattr(self, "path", value)
1322 elif keyword == "GNU.sparse.size":
1323 setattr(self, "size", int(value))
1324 elif keyword == "GNU.sparse.realsize":
1325 setattr(self, "size", int(value))
1326 elif keyword in PAX_FIELDS:
1327 if keyword in PAX_NUMBER_FIELDS:
1328 try:
1329 value = PAX_NUMBER_FIELDS[keyword](value)
1330 except ValueError:
1331 value = 0
1332 if keyword == "path":
1333 value = value.rstrip("/")
1334 setattr(self, keyword, value)
Guido van Rossume7ba4952007-06-06 23:52:48 +00001335
1336 self.pax_headers = pax_headers.copy()
Guido van Rossumd8faa362007-04-27 19:54:29 +00001337
Lars Gustäbel1465cc22010-05-17 18:02:50 +00001338 def _decode_pax_field(self, value, encoding, fallback_encoding, fallback_errors):
1339 """Decode a single field from a pax record.
1340 """
1341 try:
1342 return value.decode(encoding, "strict")
1343 except UnicodeDecodeError:
1344 return value.decode(fallback_encoding, fallback_errors)
1345
Guido van Rossumd8faa362007-04-27 19:54:29 +00001346 def _block(self, count):
1347 """Round up a byte count by BLOCKSIZE and return it,
1348 e.g. _block(834) => 1024.
1349 """
1350 blocks, remainder = divmod(count, BLOCKSIZE)
1351 if remainder:
1352 blocks += 1
1353 return blocks * BLOCKSIZE
Thomas Wouters89f507f2006-12-13 04:49:30 +00001354
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001355 def isreg(self):
1356 return self.type in REGULAR_TYPES
1357 def isfile(self):
1358 return self.isreg()
1359 def isdir(self):
1360 return self.type == DIRTYPE
1361 def issym(self):
1362 return self.type == SYMTYPE
1363 def islnk(self):
1364 return self.type == LNKTYPE
1365 def ischr(self):
1366 return self.type == CHRTYPE
1367 def isblk(self):
1368 return self.type == BLKTYPE
1369 def isfifo(self):
1370 return self.type == FIFOTYPE
1371 def issparse(self):
Lars Gustäbel9cbdd752010-10-29 09:08:19 +00001372 return self.sparse is not None
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001373 def isdev(self):
1374 return self.type in (CHRTYPE, BLKTYPE, FIFOTYPE)
1375# class TarInfo
1376
1377class TarFile(object):
1378 """The TarFile Class provides an interface to tar archives.
1379 """
1380
1381 debug = 0 # May be set from 0 (no msgs) to 3 (all msgs)
1382
1383 dereference = False # If true, add content of linked file to the
1384 # tar file, else the link.
1385
1386 ignore_zeros = False # If true, skips empty or invalid blocks and
1387 # continues processing.
1388
Lars Gustäbel365aff32009-12-13 11:42:29 +00001389 errorlevel = 1 # If 0, fatal errors only appear in debug
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001390 # messages (if debug >= 0). If > 0, errors
1391 # are passed to the caller as exceptions.
1392
Guido van Rossumd8faa362007-04-27 19:54:29 +00001393 format = DEFAULT_FORMAT # The format to use when creating an archive.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001394
Guido van Rossume7ba4952007-06-06 23:52:48 +00001395 encoding = ENCODING # Encoding for 8-bit character strings.
1396
1397 errors = None # Error handler for unicode conversion.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001398
Guido van Rossumd8faa362007-04-27 19:54:29 +00001399 tarinfo = TarInfo # The default TarInfo class to use.
1400
Lars Gustäbelb062a2f2012-05-14 13:18:16 +02001401 fileobject = ExFileObject # The file-object for extractfile().
Guido van Rossumd8faa362007-04-27 19:54:29 +00001402
1403 def __init__(self, name=None, mode="r", fileobj=None, format=None,
1404 tarinfo=None, dereference=None, ignore_zeros=None, encoding=None,
Victor Stinnerde629d42010-05-05 21:43:57 +00001405 errors="surrogateescape", pax_headers=None, debug=None, errorlevel=None):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001406 """Open an (uncompressed) tar archive `name'. `mode' is either 'r' to
1407 read from an existing archive, 'a' to append data to an existing
1408 file or 'w' to create a new file overwriting an existing one. `mode'
1409 defaults to 'r'.
1410 If `fileobj' is given, it is used for reading or writing data. If it
1411 can be determined, `mode' is overridden by `fileobj's mode.
1412 `fileobj' is not closed, when TarFile is closed.
1413 """
Berker Peksag0fe63252015-02-13 21:02:12 +02001414 modes = {"r": "rb", "a": "r+b", "w": "wb", "x": "xb"}
Serhiy Storchaka53ad0cd2014-01-18 15:35:37 +02001415 if mode not in modes:
Berker Peksag0fe63252015-02-13 21:02:12 +02001416 raise ValueError("mode must be 'r', 'a', 'w' or 'x'")
Guido van Rossumd8faa362007-04-27 19:54:29 +00001417 self.mode = mode
Serhiy Storchaka53ad0cd2014-01-18 15:35:37 +02001418 self._mode = modes[mode]
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001419
1420 if not fileobj:
Guido van Rossumd8faa362007-04-27 19:54:29 +00001421 if self.mode == "a" and not os.path.exists(name):
Thomas Wouterscf297e42007-02-23 15:07:44 +00001422 # Create nonexistent files in append mode.
Guido van Rossumd8faa362007-04-27 19:54:29 +00001423 self.mode = "w"
1424 self._mode = "wb"
Guido van Rossume7ba4952007-06-06 23:52:48 +00001425 fileobj = bltn_open(name, self._mode)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001426 self._extfileobj = False
1427 else:
Serhiy Storchaka2c6a3ae2014-07-16 23:58:58 +03001428 if (name is None and hasattr(fileobj, "name") and
1429 isinstance(fileobj.name, (str, bytes))):
Guido van Rossumd8faa362007-04-27 19:54:29 +00001430 name = fileobj.name
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001431 if hasattr(fileobj, "mode"):
Guido van Rossumd8faa362007-04-27 19:54:29 +00001432 self._mode = fileobj.mode
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001433 self._extfileobj = True
Thomas Woutersed03b412007-08-28 21:37:11 +00001434 self.name = os.path.abspath(name) if name else None
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001435 self.fileobj = fileobj
1436
Guido van Rossumd8faa362007-04-27 19:54:29 +00001437 # Init attributes.
1438 if format is not None:
1439 self.format = format
1440 if tarinfo is not None:
1441 self.tarinfo = tarinfo
1442 if dereference is not None:
1443 self.dereference = dereference
1444 if ignore_zeros is not None:
1445 self.ignore_zeros = ignore_zeros
1446 if encoding is not None:
1447 self.encoding = encoding
Victor Stinnerde629d42010-05-05 21:43:57 +00001448 self.errors = errors
Guido van Rossume7ba4952007-06-06 23:52:48 +00001449
1450 if pax_headers is not None and self.format == PAX_FORMAT:
1451 self.pax_headers = pax_headers
1452 else:
1453 self.pax_headers = {}
1454
Guido van Rossumd8faa362007-04-27 19:54:29 +00001455 if debug is not None:
1456 self.debug = debug
1457 if errorlevel is not None:
1458 self.errorlevel = errorlevel
1459
1460 # Init datastructures.
Thomas Wouters477c8d52006-05-27 19:21:47 +00001461 self.closed = False
1462 self.members = [] # list of members as TarInfo objects
1463 self._loaded = False # flag if all members have been read
Christian Heimesd8654cf2007-12-02 15:22:16 +00001464 self.offset = self.fileobj.tell()
1465 # current position in the archive file
Thomas Wouters477c8d52006-05-27 19:21:47 +00001466 self.inodes = {} # dictionary caching the inodes of
1467 # archive members already added
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001468
Lars Gustäbel7b465392009-11-18 20:29:25 +00001469 try:
1470 if self.mode == "r":
1471 self.firstmember = None
1472 self.firstmember = self.next()
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001473
Lars Gustäbel7b465392009-11-18 20:29:25 +00001474 if self.mode == "a":
1475 # Move to the end of the archive,
1476 # before the first empty block.
Lars Gustäbel7b465392009-11-18 20:29:25 +00001477 while True:
Lars Gustäbel9520a432009-11-22 18:48:49 +00001478 self.fileobj.seek(self.offset)
1479 try:
1480 tarinfo = self.tarinfo.fromtarfile(self)
1481 self.members.append(tarinfo)
1482 except EOFHeaderError:
1483 self.fileobj.seek(self.offset)
Lars Gustäbel7b465392009-11-18 20:29:25 +00001484 break
Lars Gustäbel9520a432009-11-22 18:48:49 +00001485 except HeaderError as e:
1486 raise ReadError(str(e))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001487
Lars Gustäbel20703c62015-05-27 12:53:44 +02001488 if self.mode in ("a", "w", "x"):
Lars Gustäbel7b465392009-11-18 20:29:25 +00001489 self._loaded = True
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001490
Lars Gustäbel7b465392009-11-18 20:29:25 +00001491 if self.pax_headers:
1492 buf = self.tarinfo.create_pax_global_header(self.pax_headers.copy())
1493 self.fileobj.write(buf)
1494 self.offset += len(buf)
1495 except:
1496 if not self._extfileobj:
1497 self.fileobj.close()
1498 self.closed = True
1499 raise
Guido van Rossumd8faa362007-04-27 19:54:29 +00001500
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001501 #--------------------------------------------------------------------------
1502 # Below are the classmethods which act as alternate constructors to the
1503 # TarFile class. The open() method is the only one that is needed for
1504 # public use; it is the "super"-constructor and is able to select an
1505 # adequate "sub"-constructor for a particular compression using the mapping
1506 # from OPEN_METH.
1507 #
1508 # This concept allows one to subclass TarFile without losing the comfort of
1509 # the super-constructor. A sub-constructor is registered and made available
1510 # by adding it to the mapping in OPEN_METH.
1511
Guido van Rossum75b64e62005-01-16 00:16:11 +00001512 @classmethod
Guido van Rossumd8faa362007-04-27 19:54:29 +00001513 def open(cls, name=None, mode="r", fileobj=None, bufsize=RECORDSIZE, **kwargs):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001514 """Open a tar archive for reading, writing or appending. Return
1515 an appropriate TarFile class.
1516
1517 mode:
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001518 'r' or 'r:*' open for reading with transparent compression
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001519 'r:' open for reading exclusively uncompressed
1520 'r:gz' open for reading with gzip compression
1521 'r:bz2' open for reading with bzip2 compression
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001522 'r:xz' open for reading with lzma compression
Thomas Wouterscf297e42007-02-23 15:07:44 +00001523 'a' or 'a:' open for appending, creating the file if necessary
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001524 'w' or 'w:' open for writing without compression
1525 'w:gz' open for writing with gzip compression
1526 'w:bz2' open for writing with bzip2 compression
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001527 'w:xz' open for writing with lzma compression
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001528
Berker Peksag0fe63252015-02-13 21:02:12 +02001529 'x' or 'x:' create a tarfile exclusively without compression, raise
1530 an exception if the file is already created
1531 'x:gz' create an gzip compressed tarfile, raise an exception
1532 if the file is already created
1533 'x:bz2' create an bzip2 compressed tarfile, raise an exception
1534 if the file is already created
1535 'x:xz' create an lzma compressed tarfile, raise an exception
1536 if the file is already created
1537
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001538 'r|*' open a stream of tar blocks with transparent compression
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001539 'r|' open an uncompressed stream of tar blocks for reading
1540 'r|gz' open a gzip compressed stream of tar blocks
1541 'r|bz2' open a bzip2 compressed stream of tar blocks
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001542 'r|xz' open an lzma compressed stream of tar blocks
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001543 'w|' open an uncompressed stream for writing
1544 'w|gz' open a gzip compressed stream for writing
1545 'w|bz2' open a bzip2 compressed stream for writing
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001546 'w|xz' open an lzma compressed stream for writing
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001547 """
1548
1549 if not name and not fileobj:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001550 raise ValueError("nothing to open")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001551
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001552 if mode in ("r", "r:*"):
1553 # Find out which *open() is appropriate for opening the file.
1554 for comptype in cls.OPEN_METH:
1555 func = getattr(cls, cls.OPEN_METH[comptype])
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001556 if fileobj is not None:
1557 saved_pos = fileobj.tell()
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001558 try:
Guido van Rossumd8faa362007-04-27 19:54:29 +00001559 return func(name, "r", fileobj, **kwargs)
1560 except (ReadError, CompressionError) as e:
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001561 if fileobj is not None:
1562 fileobj.seek(saved_pos)
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001563 continue
Thomas Wouters477c8d52006-05-27 19:21:47 +00001564 raise ReadError("file could not be opened successfully")
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001565
1566 elif ":" in mode:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001567 filemode, comptype = mode.split(":", 1)
1568 filemode = filemode or "r"
1569 comptype = comptype or "tar"
1570
1571 # Select the *open() function according to
1572 # given compression.
1573 if comptype in cls.OPEN_METH:
1574 func = getattr(cls, cls.OPEN_METH[comptype])
1575 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001576 raise CompressionError("unknown compression type %r" % comptype)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001577 return func(name, filemode, fileobj, **kwargs)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001578
1579 elif "|" in mode:
1580 filemode, comptype = mode.split("|", 1)
1581 filemode = filemode or "r"
1582 comptype = comptype or "tar"
1583
Serhiy Storchaka53ad0cd2014-01-18 15:35:37 +02001584 if filemode not in ("r", "w"):
Thomas Wouters477c8d52006-05-27 19:21:47 +00001585 raise ValueError("mode must be 'r' or 'w'")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001586
Antoine Pitrou605c2932010-09-23 20:15:14 +00001587 stream = _Stream(name, filemode, comptype, fileobj, bufsize)
1588 try:
1589 t = cls(name, filemode, stream, **kwargs)
1590 except:
1591 stream.close()
1592 raise
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001593 t._extfileobj = False
1594 return t
1595
Berker Peksag0fe63252015-02-13 21:02:12 +02001596 elif mode in ("a", "w", "x"):
Guido van Rossumd8faa362007-04-27 19:54:29 +00001597 return cls.taropen(name, mode, fileobj, **kwargs)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001598
Thomas Wouters477c8d52006-05-27 19:21:47 +00001599 raise ValueError("undiscernible mode")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001600
Guido van Rossum75b64e62005-01-16 00:16:11 +00001601 @classmethod
Guido van Rossumd8faa362007-04-27 19:54:29 +00001602 def taropen(cls, name, mode="r", fileobj=None, **kwargs):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001603 """Open uncompressed tar archive name for reading or writing.
1604 """
Berker Peksag0fe63252015-02-13 21:02:12 +02001605 if mode not in ("r", "a", "w", "x"):
1606 raise ValueError("mode must be 'r', 'a', 'w' or 'x'")
Guido van Rossumd8faa362007-04-27 19:54:29 +00001607 return cls(name, mode, fileobj, **kwargs)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001608
Guido van Rossum75b64e62005-01-16 00:16:11 +00001609 @classmethod
Guido van Rossumd8faa362007-04-27 19:54:29 +00001610 def gzopen(cls, name, mode="r", fileobj=None, compresslevel=9, **kwargs):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001611 """Open gzip compressed tar archive name for reading or writing.
1612 Appending is not allowed.
1613 """
Berker Peksag0fe63252015-02-13 21:02:12 +02001614 if mode not in ("r", "w", "x"):
1615 raise ValueError("mode must be 'r', 'w' or 'x'")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001616
1617 try:
1618 import gzip
Neal Norwitz4ec68242003-04-11 03:05:56 +00001619 gzip.GzipFile
1620 except (ImportError, AttributeError):
Thomas Wouters477c8d52006-05-27 19:21:47 +00001621 raise CompressionError("gzip module is not available")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001622
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001623 try:
Antoine Pitroue1eca4e2010-10-29 23:49:49 +00001624 fileobj = gzip.GzipFile(name, mode + "b", compresslevel, fileobj)
Serhiy Storchakac2d01422014-01-18 16:14:10 +02001625 except OSError:
1626 if fileobj is not None and mode == 'r':
1627 raise ReadError("not a gzip file")
1628 raise
1629
1630 try:
Antoine Pitroue1eca4e2010-10-29 23:49:49 +00001631 t = cls.taropen(name, mode, fileobj, **kwargs)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001632 except OSError:
Serhiy Storchakac2d01422014-01-18 16:14:10 +02001633 fileobj.close()
1634 if mode == 'r':
1635 raise ReadError("not a gzip file")
1636 raise
Antoine Pitroue1eca4e2010-10-29 23:49:49 +00001637 except:
Serhiy Storchakac2d01422014-01-18 16:14:10 +02001638 fileobj.close()
Antoine Pitroue1eca4e2010-10-29 23:49:49 +00001639 raise
Serhiy Storchaka9fbec7a2014-01-18 15:53:05 +02001640 t._extfileobj = False
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001641 return t
1642
Guido van Rossum75b64e62005-01-16 00:16:11 +00001643 @classmethod
Guido van Rossumd8faa362007-04-27 19:54:29 +00001644 def bz2open(cls, name, mode="r", fileobj=None, compresslevel=9, **kwargs):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001645 """Open bzip2 compressed tar archive name for reading or writing.
1646 Appending is not allowed.
1647 """
Berker Peksag0fe63252015-02-13 21:02:12 +02001648 if mode not in ("r", "w", "x"):
1649 raise ValueError("mode must be 'r', 'w' or 'x'")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001650
1651 try:
1652 import bz2
Brett Cannoncd171c82013-07-04 17:43:24 -04001653 except ImportError:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001654 raise CompressionError("bz2 module is not available")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001655
Nadeem Vawdaaebcdba2012-06-04 23:31:20 +02001656 fileobj = bz2.BZ2File(fileobj or name, mode,
1657 compresslevel=compresslevel)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001658
1659 try:
Guido van Rossumd8faa362007-04-27 19:54:29 +00001660 t = cls.taropen(name, mode, fileobj, **kwargs)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02001661 except (OSError, EOFError):
Antoine Pitrou95f55602010-09-23 18:36:46 +00001662 fileobj.close()
Serhiy Storchakac2d01422014-01-18 16:14:10 +02001663 if mode == 'r':
1664 raise ReadError("not a bzip2 file")
1665 raise
Serhiy Storchakae413cde2014-01-18 16:28:08 +02001666 except:
1667 fileobj.close()
1668 raise
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001669 t._extfileobj = False
1670 return t
1671
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001672 @classmethod
Lars Gustäbelc5e11992012-01-18 14:01:17 +01001673 def xzopen(cls, name, mode="r", fileobj=None, preset=None, **kwargs):
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001674 """Open lzma compressed tar archive name for reading or writing.
1675 Appending is not allowed.
1676 """
Berker Peksag0fe63252015-02-13 21:02:12 +02001677 if mode not in ("r", "w", "x"):
1678 raise ValueError("mode must be 'r', 'w' or 'x'")
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001679
1680 try:
1681 import lzma
Brett Cannoncd171c82013-07-04 17:43:24 -04001682 except ImportError:
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001683 raise CompressionError("lzma module is not available")
1684
Nadeem Vawda33c34da2012-06-04 23:34:07 +02001685 fileobj = lzma.LZMAFile(fileobj or name, mode, preset=preset)
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001686
1687 try:
1688 t = cls.taropen(name, mode, fileobj, **kwargs)
1689 except (lzma.LZMAError, EOFError):
1690 fileobj.close()
Serhiy Storchakac2d01422014-01-18 16:14:10 +02001691 if mode == 'r':
1692 raise ReadError("not an lzma file")
1693 raise
Serhiy Storchakae413cde2014-01-18 16:28:08 +02001694 except:
1695 fileobj.close()
1696 raise
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001697 t._extfileobj = False
1698 return t
1699
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001700 # All *open() methods are registered here.
1701 OPEN_METH = {
1702 "tar": "taropen", # uncompressed tar
1703 "gz": "gzopen", # gzip compressed tar
Lars Gustäbel0a9dd2f2011-12-10 20:38:14 +01001704 "bz2": "bz2open", # bzip2 compressed tar
1705 "xz": "xzopen" # lzma compressed tar
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001706 }
1707
1708 #--------------------------------------------------------------------------
1709 # The public methods which TarFile provides:
1710
1711 def close(self):
1712 """Close the TarFile. In write-mode, two finishing zero blocks are
1713 appended to the archive.
1714 """
1715 if self.closed:
1716 return
1717
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001718 self.closed = True
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001719 try:
Lars Gustäbel20703c62015-05-27 12:53:44 +02001720 if self.mode in ("a", "w", "x"):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +03001721 self.fileobj.write(NUL * (BLOCKSIZE * 2))
1722 self.offset += (BLOCKSIZE * 2)
1723 # fill up the end with zero-blocks
1724 # (like option -b20 for tar does)
1725 blocks, remainder = divmod(self.offset, RECORDSIZE)
1726 if remainder > 0:
1727 self.fileobj.write(NUL * (RECORDSIZE - remainder))
1728 finally:
1729 if not self._extfileobj:
1730 self.fileobj.close()
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001731
1732 def getmember(self, name):
1733 """Return a TarInfo object for member `name'. If `name' can not be
1734 found in the archive, KeyError is raised. If a member occurs more
Mark Dickinson934896d2009-02-21 20:59:32 +00001735 than once in the archive, its last occurrence is assumed to be the
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001736 most up-to-date version.
1737 """
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001738 tarinfo = self._getmember(name)
1739 if tarinfo is None:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001740 raise KeyError("filename %r not found" % name)
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001741 return tarinfo
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001742
1743 def getmembers(self):
1744 """Return the members of the archive as a list of TarInfo objects. The
1745 list has the same order as the members in the archive.
1746 """
1747 self._check()
1748 if not self._loaded: # if we want to obtain a list of
1749 self._load() # all members, we first have to
1750 # scan the whole archive.
1751 return self.members
1752
1753 def getnames(self):
1754 """Return the members of the archive as a list of their names. It has
1755 the same order as the list returned by getmembers().
1756 """
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001757 return [tarinfo.name for tarinfo in self.getmembers()]
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001758
1759 def gettarinfo(self, name=None, arcname=None, fileobj=None):
1760 """Create a TarInfo object for either the file `name' or the file
1761 object `fileobj' (using os.fstat on its file descriptor). You can
1762 modify some of the TarInfo's attributes before you add it using
1763 addfile(). If given, `arcname' specifies an alternative name for the
1764 file in the archive.
1765 """
Berker Peksag0fe63252015-02-13 21:02:12 +02001766 self._check("awx")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001767
1768 # When fileobj is given, replace name by
1769 # fileobj's real name.
1770 if fileobj is not None:
1771 name = fileobj.name
1772
1773 # Building the name of the member in the archive.
1774 # Backward slashes are converted to forward slashes,
1775 # Absolute paths are turned to relative paths.
1776 if arcname is None:
1777 arcname = name
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001778 drv, arcname = os.path.splitdrive(arcname)
Lars Gustäbelbfdfdda2009-08-28 19:59:59 +00001779 arcname = arcname.replace(os.sep, "/")
1780 arcname = arcname.lstrip("/")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001781
1782 # Now, fill the TarInfo object with
1783 # information specific for the file.
Guido van Rossumd8faa362007-04-27 19:54:29 +00001784 tarinfo = self.tarinfo()
1785 tarinfo.tarfile = self
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001786
1787 # Use os.stat or os.lstat, depending on platform
1788 # and if symlinks shall be resolved.
1789 if fileobj is None:
1790 if hasattr(os, "lstat") and not self.dereference:
1791 statres = os.lstat(name)
1792 else:
1793 statres = os.stat(name)
1794 else:
1795 statres = os.fstat(fileobj.fileno())
1796 linkname = ""
1797
1798 stmd = statres.st_mode
1799 if stat.S_ISREG(stmd):
1800 inode = (statres.st_ino, statres.st_dev)
Guido van Rossumd8faa362007-04-27 19:54:29 +00001801 if not self.dereference and statres.st_nlink > 1 and \
1802 inode in self.inodes and arcname != self.inodes[inode]:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001803 # Is it a hardlink to an already
1804 # archived file?
1805 type = LNKTYPE
1806 linkname = self.inodes[inode]
1807 else:
1808 # The inode is added only if its valid.
1809 # For win32 it is always 0.
1810 type = REGTYPE
1811 if inode[0]:
1812 self.inodes[inode] = arcname
1813 elif stat.S_ISDIR(stmd):
1814 type = DIRTYPE
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001815 elif stat.S_ISFIFO(stmd):
1816 type = FIFOTYPE
1817 elif stat.S_ISLNK(stmd):
1818 type = SYMTYPE
1819 linkname = os.readlink(name)
1820 elif stat.S_ISCHR(stmd):
1821 type = CHRTYPE
1822 elif stat.S_ISBLK(stmd):
1823 type = BLKTYPE
1824 else:
1825 return None
1826
1827 # Fill the TarInfo object with all
1828 # information we can get.
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001829 tarinfo.name = arcname
1830 tarinfo.mode = stmd
1831 tarinfo.uid = statres.st_uid
1832 tarinfo.gid = statres.st_gid
Lars Gustäbel2470ff12010-06-03 10:11:52 +00001833 if type == REGTYPE:
Martin v. Löwis61d77e02004-08-20 06:35:46 +00001834 tarinfo.size = statres.st_size
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001835 else:
Guido van Rossume2a383d2007-01-15 16:59:06 +00001836 tarinfo.size = 0
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001837 tarinfo.mtime = statres.st_mtime
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001838 tarinfo.type = type
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001839 tarinfo.linkname = linkname
1840 if pwd:
1841 try:
1842 tarinfo.uname = pwd.getpwuid(tarinfo.uid)[0]
1843 except KeyError:
1844 pass
1845 if grp:
1846 try:
1847 tarinfo.gname = grp.getgrgid(tarinfo.gid)[0]
1848 except KeyError:
1849 pass
1850
1851 if type in (CHRTYPE, BLKTYPE):
1852 if hasattr(os, "major") and hasattr(os, "minor"):
1853 tarinfo.devmajor = os.major(statres.st_rdev)
1854 tarinfo.devminor = os.minor(statres.st_rdev)
1855 return tarinfo
1856
Serhiy Storchakaa7eb7462014-08-21 10:01:16 +03001857 def list(self, verbose=True, *, members=None):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001858 """Print a table of contents to sys.stdout. If `verbose' is False, only
1859 the names of the members are printed. If it is True, an `ls -l'-like
Serhiy Storchakaa7eb7462014-08-21 10:01:16 +03001860 output is produced. `members' is optional and must be a subset of the
1861 list returned by getmembers().
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001862 """
1863 self._check()
1864
Serhiy Storchakaa7eb7462014-08-21 10:01:16 +03001865 if members is None:
1866 members = self
1867 for tarinfo in members:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001868 if verbose:
Serhiy Storchaka3b4f1592014-02-05 20:53:36 +02001869 _safe_print(stat.filemode(tarinfo.mode))
1870 _safe_print("%s/%s" % (tarinfo.uname or tarinfo.uid,
1871 tarinfo.gname or tarinfo.gid))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001872 if tarinfo.ischr() or tarinfo.isblk():
Serhiy Storchaka3b4f1592014-02-05 20:53:36 +02001873 _safe_print("%10s" %
1874 ("%d,%d" % (tarinfo.devmajor, tarinfo.devminor)))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001875 else:
Serhiy Storchaka3b4f1592014-02-05 20:53:36 +02001876 _safe_print("%10d" % tarinfo.size)
1877 _safe_print("%d-%02d-%02d %02d:%02d:%02d" \
1878 % time.localtime(tarinfo.mtime)[:6])
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001879
Serhiy Storchaka3b4f1592014-02-05 20:53:36 +02001880 _safe_print(tarinfo.name + ("/" if tarinfo.isdir() else ""))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001881
1882 if verbose:
1883 if tarinfo.issym():
Serhiy Storchaka3b4f1592014-02-05 20:53:36 +02001884 _safe_print("-> " + tarinfo.linkname)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001885 if tarinfo.islnk():
Serhiy Storchaka3b4f1592014-02-05 20:53:36 +02001886 _safe_print("link to " + tarinfo.linkname)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001887 print()
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001888
Raymond Hettingera63a3122011-01-26 20:34:14 +00001889 def add(self, name, arcname=None, recursive=True, exclude=None, *, filter=None):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001890 """Add the file `name' to the archive. `name' may be any type of file
1891 (directory, fifo, symbolic link, etc.). If given, `arcname'
1892 specifies an alternative name for the file in the archive.
1893 Directories are added recursively by default. This can be avoided by
Guido van Rossum486364b2007-06-30 05:01:58 +00001894 setting `recursive' to False. `exclude' is a function that should
Lars Gustäbel049d2aa2009-09-12 10:44:00 +00001895 return True for each filename to be excluded. `filter' is a function
1896 that expects a TarInfo object argument and returns the changed
1897 TarInfo object, if it returns None the TarInfo object will be
1898 excluded from the archive.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001899 """
Berker Peksag0fe63252015-02-13 21:02:12 +02001900 self._check("awx")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001901
1902 if arcname is None:
1903 arcname = name
1904
Guido van Rossum486364b2007-06-30 05:01:58 +00001905 # Exclude pathnames.
Lars Gustäbel049d2aa2009-09-12 10:44:00 +00001906 if exclude is not None:
1907 import warnings
1908 warnings.warn("use the filter argument instead",
1909 DeprecationWarning, 2)
1910 if exclude(name):
1911 self._dbg(2, "tarfile: Excluded %r" % name)
1912 return
Guido van Rossum486364b2007-06-30 05:01:58 +00001913
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001914 # Skip if somebody tries to archive the archive...
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001915 if self.name is not None and os.path.abspath(name) == self.name:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001916 self._dbg(2, "tarfile: Skipped %r" % name)
1917 return
1918
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001919 self._dbg(1, name)
1920
1921 # Create a TarInfo object from the file.
1922 tarinfo = self.gettarinfo(name, arcname)
1923
1924 if tarinfo is None:
1925 self._dbg(1, "tarfile: Unsupported type %r" % name)
1926 return
1927
Lars Gustäbel049d2aa2009-09-12 10:44:00 +00001928 # Change or exclude the TarInfo object.
1929 if filter is not None:
1930 tarinfo = filter(tarinfo)
1931 if tarinfo is None:
1932 self._dbg(2, "tarfile: Excluded %r" % name)
1933 return
1934
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001935 # Append the tar header and data to the archive.
1936 if tarinfo.isreg():
Andrew Svetlov718df1d2012-11-29 14:20:47 +02001937 with bltn_open(name, "rb") as f:
1938 self.addfile(tarinfo, f)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001939
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001940 elif tarinfo.isdir():
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001941 self.addfile(tarinfo)
1942 if recursive:
1943 for f in os.listdir(name):
Lars Gustäbel049d2aa2009-09-12 10:44:00 +00001944 self.add(os.path.join(name, f), os.path.join(arcname, f),
Raymond Hettingera63a3122011-01-26 20:34:14 +00001945 recursive, exclude, filter=filter)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001946
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001947 else:
1948 self.addfile(tarinfo)
1949
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001950 def addfile(self, tarinfo, fileobj=None):
1951 """Add the TarInfo object `tarinfo' to the archive. If `fileobj' is
1952 given, tarinfo.size bytes are read from it and added to the archive.
1953 You can create TarInfo objects using gettarinfo().
1954 On Windows platforms, `fileobj' should always be opened with mode
1955 'rb' to avoid irritation about the file size.
1956 """
Berker Peksag0fe63252015-02-13 21:02:12 +02001957 self._check("awx")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001958
Thomas Wouters89f507f2006-12-13 04:49:30 +00001959 tarinfo = copy.copy(tarinfo)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001960
Guido van Rossume7ba4952007-06-06 23:52:48 +00001961 buf = tarinfo.tobuf(self.format, self.encoding, self.errors)
Thomas Wouters89f507f2006-12-13 04:49:30 +00001962 self.fileobj.write(buf)
1963 self.offset += len(buf)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001964
1965 # If there's data to follow, append it.
1966 if fileobj is not None:
1967 copyfileobj(fileobj, self.fileobj, tarinfo.size)
1968 blocks, remainder = divmod(tarinfo.size, BLOCKSIZE)
1969 if remainder > 0:
1970 self.fileobj.write(NUL * (BLOCKSIZE - remainder))
1971 blocks += 1
1972 self.offset += blocks * BLOCKSIZE
1973
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001974 self.members.append(tarinfo)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001975
Eric V. Smith7a803892015-04-15 10:27:58 -04001976 def extractall(self, path=".", members=None, *, numeric_owner=False):
Martin v. Löwis00a73e72005-03-04 19:40:34 +00001977 """Extract all members from the archive to the current working
1978 directory and set owner, modification time and permissions on
1979 directories afterwards. `path' specifies a different directory
1980 to extract to. `members' is optional and must be a subset of the
Eric V. Smith7a803892015-04-15 10:27:58 -04001981 list returned by getmembers(). If `numeric_owner` is True, only
1982 the numbers for user/group names are used and not the names.
Martin v. Löwis00a73e72005-03-04 19:40:34 +00001983 """
1984 directories = []
1985
1986 if members is None:
1987 members = self
1988
1989 for tarinfo in members:
1990 if tarinfo.isdir():
Christian Heimes2202f872008-02-06 14:31:34 +00001991 # Extract directories with a safe mode.
Martin v. Löwis00a73e72005-03-04 19:40:34 +00001992 directories.append(tarinfo)
Christian Heimes2202f872008-02-06 14:31:34 +00001993 tarinfo = copy.copy(tarinfo)
1994 tarinfo.mode = 0o700
Martin v. Löwis16f344d2010-11-01 21:39:13 +00001995 # Do not set_attrs directories, as we will do that further down
Eric V. Smith7a803892015-04-15 10:27:58 -04001996 self.extract(tarinfo, path, set_attrs=not tarinfo.isdir(),
1997 numeric_owner=numeric_owner)
Martin v. Löwis00a73e72005-03-04 19:40:34 +00001998
1999 # Reverse sort directories.
Raymond Hettingerd4cb56d2008-01-30 02:55:10 +00002000 directories.sort(key=lambda a: a.name)
Martin v. Löwis00a73e72005-03-04 19:40:34 +00002001 directories.reverse()
2002
2003 # Set correct owner, mtime and filemode on directories.
2004 for tarinfo in directories:
Christian Heimesfaf2f632008-01-06 16:59:19 +00002005 dirpath = os.path.join(path, tarinfo.name)
Martin v. Löwis00a73e72005-03-04 19:40:34 +00002006 try:
Eric V. Smith7a803892015-04-15 10:27:58 -04002007 self.chown(tarinfo, dirpath, numeric_owner=numeric_owner)
Christian Heimesfaf2f632008-01-06 16:59:19 +00002008 self.utime(tarinfo, dirpath)
2009 self.chmod(tarinfo, dirpath)
Guido van Rossumb940e112007-01-10 16:19:56 +00002010 except ExtractError as e:
Martin v. Löwis00a73e72005-03-04 19:40:34 +00002011 if self.errorlevel > 1:
2012 raise
2013 else:
2014 self._dbg(1, "tarfile: %s" % e)
2015
Eric V. Smith7a803892015-04-15 10:27:58 -04002016 def extract(self, member, path="", set_attrs=True, *, numeric_owner=False):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002017 """Extract a member from the archive to the current working directory,
2018 using its full name. Its file information is extracted as accurately
2019 as possible. `member' may be a filename or a TarInfo object. You can
Martin v. Löwis16f344d2010-11-01 21:39:13 +00002020 specify a different directory using `path'. File attributes (owner,
Eric V. Smith7a803892015-04-15 10:27:58 -04002021 mtime, mode) are set unless `set_attrs' is False. If `numeric_owner`
2022 is True, only the numbers for user/group names are used and not
2023 the names.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002024 """
2025 self._check("r")
2026
Guido van Rossum3172c5d2007-10-16 18:12:55 +00002027 if isinstance(member, str):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002028 tarinfo = self.getmember(member)
Guido van Rossumd8faa362007-04-27 19:54:29 +00002029 else:
2030 tarinfo = member
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002031
Neal Norwitza4f651a2004-07-20 22:07:44 +00002032 # Prepare the link target for makelink().
2033 if tarinfo.islnk():
2034 tarinfo._link_target = os.path.join(path, tarinfo.linkname)
2035
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002036 try:
Martin v. Löwis16f344d2010-11-01 21:39:13 +00002037 self._extract_member(tarinfo, os.path.join(path, tarinfo.name),
Eric V. Smith7a803892015-04-15 10:27:58 -04002038 set_attrs=set_attrs,
2039 numeric_owner=numeric_owner)
Andrew Svetlov3438fa42012-12-17 23:35:18 +02002040 except OSError as e:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002041 if self.errorlevel > 0:
2042 raise
2043 else:
2044 if e.filename is None:
2045 self._dbg(1, "tarfile: %s" % e.strerror)
2046 else:
2047 self._dbg(1, "tarfile: %s %r" % (e.strerror, e.filename))
Guido van Rossumb940e112007-01-10 16:19:56 +00002048 except ExtractError as e:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002049 if self.errorlevel > 1:
2050 raise
2051 else:
2052 self._dbg(1, "tarfile: %s" % e)
2053
2054 def extractfile(self, member):
2055 """Extract a member from the archive as a file object. `member' may be
Lars Gustäbel7a919e92012-05-05 18:15:03 +02002056 a filename or a TarInfo object. If `member' is a regular file or a
2057 link, an io.BufferedReader object is returned. Otherwise, None is
2058 returned.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002059 """
2060 self._check("r")
2061
Guido van Rossum3172c5d2007-10-16 18:12:55 +00002062 if isinstance(member, str):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002063 tarinfo = self.getmember(member)
Guido van Rossumd8faa362007-04-27 19:54:29 +00002064 else:
2065 tarinfo = member
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002066
Lars Gustäbel7a919e92012-05-05 18:15:03 +02002067 if tarinfo.isreg() or tarinfo.type not in SUPPORTED_TYPES:
2068 # Members with unknown types are treated as regular files.
Lars Gustäbelb062a2f2012-05-14 13:18:16 +02002069 return self.fileobject(self, tarinfo)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002070
2071 elif tarinfo.islnk() or tarinfo.issym():
2072 if isinstance(self.fileobj, _Stream):
2073 # A small but ugly workaround for the case that someone tries
2074 # to extract a (sym)link as a file-object from a non-seekable
2075 # stream of tar blocks.
Thomas Wouters477c8d52006-05-27 19:21:47 +00002076 raise StreamError("cannot extract (sym)link as file object")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002077 else:
Georg Brandl7eb4b7d2005-07-22 21:49:32 +00002078 # A (sym)link's file object is its target's file object.
Lars Gustäbel1b512722010-06-03 12:45:16 +00002079 return self.extractfile(self._find_link_target(tarinfo))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002080 else:
2081 # If there's no data associated with the member (directory, chrdev,
2082 # blkdev, etc.), return None instead of a file object.
2083 return None
2084
Eric V. Smith7a803892015-04-15 10:27:58 -04002085 def _extract_member(self, tarinfo, targetpath, set_attrs=True,
2086 numeric_owner=False):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002087 """Extract the TarInfo object tarinfo to a physical
2088 file called targetpath.
2089 """
2090 # Fetch the TarInfo object for the given name
2091 # and build the destination pathname, replacing
2092 # forward slashes to platform specific separators.
Lars Gustäbelbfdfdda2009-08-28 19:59:59 +00002093 targetpath = targetpath.rstrip("/")
2094 targetpath = targetpath.replace("/", os.sep)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002095
2096 # Create all upper directories.
2097 upperdirs = os.path.dirname(targetpath)
2098 if upperdirs and not os.path.exists(upperdirs):
Christian Heimes2202f872008-02-06 14:31:34 +00002099 # Create directories that are not part of the archive with
2100 # default permissions.
Thomas Woutersb2137042007-02-01 18:02:27 +00002101 os.makedirs(upperdirs)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002102
2103 if tarinfo.islnk() or tarinfo.issym():
2104 self._dbg(1, "%s -> %s" % (tarinfo.name, tarinfo.linkname))
2105 else:
2106 self._dbg(1, tarinfo.name)
2107
2108 if tarinfo.isreg():
2109 self.makefile(tarinfo, targetpath)
2110 elif tarinfo.isdir():
2111 self.makedir(tarinfo, targetpath)
2112 elif tarinfo.isfifo():
2113 self.makefifo(tarinfo, targetpath)
2114 elif tarinfo.ischr() or tarinfo.isblk():
2115 self.makedev(tarinfo, targetpath)
2116 elif tarinfo.islnk() or tarinfo.issym():
2117 self.makelink(tarinfo, targetpath)
2118 elif tarinfo.type not in SUPPORTED_TYPES:
2119 self.makeunknown(tarinfo, targetpath)
2120 else:
2121 self.makefile(tarinfo, targetpath)
2122
Martin v. Löwis16f344d2010-11-01 21:39:13 +00002123 if set_attrs:
Eric V. Smith7a803892015-04-15 10:27:58 -04002124 self.chown(tarinfo, targetpath, numeric_owner)
Martin v. Löwis16f344d2010-11-01 21:39:13 +00002125 if not tarinfo.issym():
2126 self.chmod(tarinfo, targetpath)
2127 self.utime(tarinfo, targetpath)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002128
2129 #--------------------------------------------------------------------------
2130 # Below are the different file methods. They are called via
2131 # _extract_member() when extract() is called. They can be replaced in a
2132 # subclass to implement other functionality.
2133
2134 def makedir(self, tarinfo, targetpath):
2135 """Make a directory called targetpath.
2136 """
2137 try:
Christian Heimes2202f872008-02-06 14:31:34 +00002138 # Use a safe mode for the directory, the real mode is set
2139 # later in _extract_member().
2140 os.mkdir(targetpath, 0o700)
Florent Xicluna68f71a32011-10-28 16:06:23 +02002141 except FileExistsError:
2142 pass
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002143
2144 def makefile(self, tarinfo, targetpath):
2145 """Make a file called targetpath.
2146 """
Lars Gustäbel9cbdd752010-10-29 09:08:19 +00002147 source = self.fileobj
2148 source.seek(tarinfo.offset_data)
Andrew Svetlov718df1d2012-11-29 14:20:47 +02002149 with bltn_open(targetpath, "wb") as target:
2150 if tarinfo.sparse is not None:
2151 for offset, size in tarinfo.sparse:
2152 target.seek(offset)
Lars Gustäbel03572682015-07-06 09:27:24 +02002153 copyfileobj(source, target, size, ReadError)
Andrew Svetlov718df1d2012-11-29 14:20:47 +02002154 else:
Lars Gustäbel03572682015-07-06 09:27:24 +02002155 copyfileobj(source, target, tarinfo.size, ReadError)
Andrew Svetlov718df1d2012-11-29 14:20:47 +02002156 target.seek(tarinfo.size)
2157 target.truncate()
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002158
2159 def makeunknown(self, tarinfo, targetpath):
2160 """Make a file from a TarInfo object with an unknown type
2161 at targetpath.
2162 """
2163 self.makefile(tarinfo, targetpath)
2164 self._dbg(1, "tarfile: Unknown file type %r, " \
2165 "extracted as regular file." % tarinfo.type)
2166
2167 def makefifo(self, tarinfo, targetpath):
2168 """Make a fifo called targetpath.
2169 """
2170 if hasattr(os, "mkfifo"):
2171 os.mkfifo(targetpath)
2172 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +00002173 raise ExtractError("fifo not supported by system")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002174
2175 def makedev(self, tarinfo, targetpath):
2176 """Make a character or block device called targetpath.
2177 """
2178 if not hasattr(os, "mknod") or not hasattr(os, "makedev"):
Thomas Wouters477c8d52006-05-27 19:21:47 +00002179 raise ExtractError("special devices not supported by system")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002180
2181 mode = tarinfo.mode
2182 if tarinfo.isblk():
2183 mode |= stat.S_IFBLK
2184 else:
2185 mode |= stat.S_IFCHR
2186
2187 os.mknod(targetpath, mode,
2188 os.makedev(tarinfo.devmajor, tarinfo.devminor))
2189
2190 def makelink(self, tarinfo, targetpath):
2191 """Make a (symbolic) link called targetpath. If it cannot be created
2192 (platform limitation), we try to make a copy of the referenced file
2193 instead of a link.
2194 """
Brian Curtind40e6f72010-07-08 21:39:08 +00002195 try:
Lars Gustäbel1b512722010-06-03 12:45:16 +00002196 # For systems that support symbolic and hard links.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002197 if tarinfo.issym():
Lars Gustäbelbfdfdda2009-08-28 19:59:59 +00002198 os.symlink(tarinfo.linkname, targetpath)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002199 else:
Neal Norwitza4f651a2004-07-20 22:07:44 +00002200 # See extract().
Lars Gustäbel1b512722010-06-03 12:45:16 +00002201 if os.path.exists(tarinfo._link_target):
2202 os.link(tarinfo._link_target, targetpath)
2203 else:
Brian Curtin82df53e2010-09-24 21:04:05 +00002204 self._extract_member(self._find_link_target(tarinfo),
2205 targetpath)
Brian Curtin16633fa2010-07-09 13:54:27 +00002206 except symlink_exception:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002207 try:
Brian Curtin16633fa2010-07-09 13:54:27 +00002208 self._extract_member(self._find_link_target(tarinfo),
2209 targetpath)
Lars Gustäbel1b512722010-06-03 12:45:16 +00002210 except KeyError:
2211 raise ExtractError("unable to resolve link inside archive")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002212
Eric V. Smith7a803892015-04-15 10:27:58 -04002213 def chown(self, tarinfo, targetpath, numeric_owner):
2214 """Set owner of targetpath according to tarinfo. If numeric_owner
2215 is True, use .gid/.uid instead of .gname/.uname.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002216 """
2217 if pwd and hasattr(os, "geteuid") and os.geteuid() == 0:
2218 # We have to be root to do so.
Eric V. Smith7a803892015-04-15 10:27:58 -04002219 if numeric_owner:
Lars Gustäbel2e7ddd32011-09-05 16:58:14 +02002220 g = tarinfo.gid
Lars Gustäbel2e7ddd32011-09-05 16:58:14 +02002221 u = tarinfo.uid
Eric V. Smith7a803892015-04-15 10:27:58 -04002222 else:
2223 try:
2224 g = grp.getgrnam(tarinfo.gname)[2]
2225 except KeyError:
2226 g = tarinfo.gid
2227 try:
2228 u = pwd.getpwnam(tarinfo.uname)[2]
2229 except KeyError:
2230 u = tarinfo.uid
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002231 try:
2232 if tarinfo.issym() and hasattr(os, "lchown"):
2233 os.lchown(targetpath, u, g)
2234 else:
Jesus Cea4791a242012-10-05 03:15:39 +02002235 os.chown(targetpath, u, g)
Andrew Svetlov3438fa42012-12-17 23:35:18 +02002236 except OSError as e:
Thomas Wouters477c8d52006-05-27 19:21:47 +00002237 raise ExtractError("could not change owner")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002238
2239 def chmod(self, tarinfo, targetpath):
2240 """Set file permissions of targetpath according to tarinfo.
2241 """
Jack Jansen834eff62003-03-07 12:47:06 +00002242 if hasattr(os, 'chmod'):
2243 try:
2244 os.chmod(targetpath, tarinfo.mode)
Andrew Svetlov3438fa42012-12-17 23:35:18 +02002245 except OSError as e:
Thomas Wouters477c8d52006-05-27 19:21:47 +00002246 raise ExtractError("could not change mode")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002247
2248 def utime(self, tarinfo, targetpath):
2249 """Set modification time of targetpath according to tarinfo.
2250 """
Jack Jansen834eff62003-03-07 12:47:06 +00002251 if not hasattr(os, 'utime'):
Tim Petersf9347782003-03-07 15:36:41 +00002252 return
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002253 try:
2254 os.utime(targetpath, (tarinfo.mtime, tarinfo.mtime))
Andrew Svetlov3438fa42012-12-17 23:35:18 +02002255 except OSError as e:
Thomas Wouters477c8d52006-05-27 19:21:47 +00002256 raise ExtractError("could not change modification time")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002257
2258 #--------------------------------------------------------------------------
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002259 def next(self):
2260 """Return the next member of the archive as a TarInfo object, when
2261 TarFile is opened for reading. Return None if there is no more
2262 available.
2263 """
2264 self._check("ra")
2265 if self.firstmember is not None:
2266 m = self.firstmember
2267 self.firstmember = None
2268 return m
2269
Lars Gustäbel03572682015-07-06 09:27:24 +02002270 # Advance the file pointer.
2271 if self.offset != self.fileobj.tell():
2272 self.fileobj.seek(self.offset - 1)
2273 if not self.fileobj.read(1):
2274 raise ReadError("unexpected end of data")
2275
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002276 # Read the next block.
Lars Gustäbel9520a432009-11-22 18:48:49 +00002277 tarinfo = None
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002278 while True:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002279 try:
Guido van Rossumd8faa362007-04-27 19:54:29 +00002280 tarinfo = self.tarinfo.fromtarfile(self)
Lars Gustäbel9520a432009-11-22 18:48:49 +00002281 except EOFHeaderError as e:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002282 if self.ignore_zeros:
Thomas Wouters902d6eb2007-01-09 23:18:33 +00002283 self._dbg(2, "0x%X: %s" % (self.offset, e))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002284 self.offset += BLOCKSIZE
2285 continue
Lars Gustäbel9520a432009-11-22 18:48:49 +00002286 except InvalidHeaderError as e:
2287 if self.ignore_zeros:
2288 self._dbg(2, "0x%X: %s" % (self.offset, e))
2289 self.offset += BLOCKSIZE
2290 continue
2291 elif self.offset == 0:
2292 raise ReadError(str(e))
2293 except EmptyHeaderError:
2294 if self.offset == 0:
2295 raise ReadError("empty file")
2296 except TruncatedHeaderError as e:
2297 if self.offset == 0:
2298 raise ReadError(str(e))
2299 except SubsequentHeaderError as e:
2300 raise ReadError(str(e))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002301 break
2302
Lars Gustäbel9520a432009-11-22 18:48:49 +00002303 if tarinfo is not None:
2304 self.members.append(tarinfo)
2305 else:
2306 self._loaded = True
2307
Thomas Wouters477c8d52006-05-27 19:21:47 +00002308 return tarinfo
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002309
2310 #--------------------------------------------------------------------------
2311 # Little helper methods:
2312
Lars Gustäbel1b512722010-06-03 12:45:16 +00002313 def _getmember(self, name, tarinfo=None, normalize=False):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002314 """Find an archive member by name from bottom to top.
2315 If tarinfo is given, it is used as the starting point.
2316 """
Martin v. Löwisf3c56112004-09-18 09:08:52 +00002317 # Ensure that all members have been loaded.
2318 members = self.getmembers()
2319
Lars Gustäbel1b512722010-06-03 12:45:16 +00002320 # Limit the member search list up to tarinfo.
2321 if tarinfo is not None:
2322 members = members[:members.index(tarinfo)]
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002323
Lars Gustäbel1b512722010-06-03 12:45:16 +00002324 if normalize:
2325 name = os.path.normpath(name)
2326
2327 for member in reversed(members):
2328 if normalize:
2329 member_name = os.path.normpath(member.name)
2330 else:
2331 member_name = member.name
2332
2333 if name == member_name:
2334 return member
Andrew M. Kuchling864bba12004-07-10 22:02:11 +00002335
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002336 def _load(self):
2337 """Read through the entire archive file and look for readable
2338 members.
2339 """
2340 while True:
2341 tarinfo = self.next()
2342 if tarinfo is None:
2343 break
2344 self._loaded = True
2345
2346 def _check(self, mode=None):
2347 """Check if TarFile is still open, and if the operation's mode
2348 corresponds to TarFile's mode.
2349 """
2350 if self.closed:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02002351 raise OSError("%s is closed" % self.__class__.__name__)
Guido van Rossumd8faa362007-04-27 19:54:29 +00002352 if mode is not None and self.mode not in mode:
Andrew Svetlovf7a17b42012-12-25 16:47:37 +02002353 raise OSError("bad operation for mode %r" % self.mode)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002354
Lars Gustäbel1b512722010-06-03 12:45:16 +00002355 def _find_link_target(self, tarinfo):
2356 """Find the target member of a symlink or hardlink member in the
2357 archive.
2358 """
2359 if tarinfo.issym():
2360 # Always search the entire archive.
Lars Gustäbel1ef9eda2012-04-24 21:04:40 +02002361 linkname = "/".join(filter(None, (os.path.dirname(tarinfo.name), tarinfo.linkname)))
Lars Gustäbel1b512722010-06-03 12:45:16 +00002362 limit = None
2363 else:
2364 # Search the archive before the link, because a hard link is
2365 # just a reference to an already archived file.
2366 linkname = tarinfo.linkname
2367 limit = tarinfo
2368
2369 member = self._getmember(linkname, tarinfo=limit, normalize=True)
2370 if member is None:
2371 raise KeyError("linkname %r not found" % linkname)
2372 return member
2373
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002374 def __iter__(self):
2375 """Provide an iterator object.
2376 """
2377 if self._loaded:
Serhiy Storchakaa2549212015-12-19 09:43:14 +02002378 yield from self.members
2379 return
2380
2381 # Yield items using TarFile's next() method.
2382 # When all members have been read, set TarFile as _loaded.
2383 index = 0
2384 # Fix for SF #1100429: Under rare circumstances it can
2385 # happen that getmembers() is called during iteration,
2386 # which will have already exhausted the next() method.
2387 if self.firstmember is not None:
2388 tarinfo = self.next()
2389 index += 1
2390 yield tarinfo
2391
2392 while True:
2393 if index < len(self.members):
2394 tarinfo = self.members[index]
2395 elif not self._loaded:
2396 tarinfo = self.next()
2397 if not tarinfo:
2398 self._loaded = True
2399 return
2400 else:
2401 return
2402 index += 1
2403 yield tarinfo
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002404
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002405 def _dbg(self, level, msg):
2406 """Write debugging output to sys.stderr.
2407 """
2408 if level <= self.debug:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00002409 print(msg, file=sys.stderr)
Lars Gustäbel01385812010-03-03 12:08:54 +00002410
2411 def __enter__(self):
2412 self._check()
2413 return self
2414
2415 def __exit__(self, type, value, traceback):
2416 if type is None:
2417 self.close()
2418 else:
2419 # An exception occurred. We must not call close() because
2420 # it would try to write end-of-archive blocks and padding.
2421 if not self._extfileobj:
2422 self.fileobj.close()
2423 self.closed = True
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002424
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002425#--------------------
2426# exported functions
2427#--------------------
2428def is_tarfile(name):
2429 """Return True if name points to a tar archive that we
2430 are able to handle, else return False.
2431 """
2432 try:
2433 t = open(name)
2434 t.close()
2435 return True
2436 except TarError:
2437 return False
2438
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002439open = TarFile.open
Serhiy Storchakad27b4552013-11-24 01:53:29 +02002440
2441
2442def main():
2443 import argparse
2444
2445 description = 'A simple command line interface for tarfile module.'
2446 parser = argparse.ArgumentParser(description=description)
2447 parser.add_argument('-v', '--verbose', action='store_true', default=False,
2448 help='Verbose output')
2449 group = parser.add_mutually_exclusive_group()
2450 group.add_argument('-l', '--list', metavar='<tarfile>',
2451 help='Show listing of a tarfile')
2452 group.add_argument('-e', '--extract', nargs='+',
2453 metavar=('<tarfile>', '<output_dir>'),
2454 help='Extract tarfile into target dir')
2455 group.add_argument('-c', '--create', nargs='+',
2456 metavar=('<name>', '<file>'),
2457 help='Create tarfile from sources')
2458 group.add_argument('-t', '--test', metavar='<tarfile>',
2459 help='Test if a tarfile is valid')
2460 args = parser.parse_args()
2461
2462 if args.test:
2463 src = args.test
2464 if is_tarfile(src):
2465 with open(src, 'r') as tar:
2466 tar.getmembers()
2467 print(tar.getmembers(), file=sys.stderr)
2468 if args.verbose:
2469 print('{!r} is a tar archive.'.format(src))
2470 else:
2471 parser.exit(1, '{!r} is not a tar archive.\n'.format(src))
2472
2473 elif args.list:
2474 src = args.list
2475 if is_tarfile(src):
2476 with TarFile.open(src, 'r:*') as tf:
2477 tf.list(verbose=args.verbose)
2478 else:
2479 parser.exit(1, '{!r} is not a tar archive.\n'.format(src))
2480
2481 elif args.extract:
2482 if len(args.extract) == 1:
2483 src = args.extract[0]
2484 curdir = os.curdir
2485 elif len(args.extract) == 2:
2486 src, curdir = args.extract
2487 else:
2488 parser.exit(1, parser.format_help())
2489
2490 if is_tarfile(src):
2491 with TarFile.open(src, 'r:*') as tf:
2492 tf.extractall(path=curdir)
2493 if args.verbose:
2494 if curdir == '.':
2495 msg = '{!r} file is extracted.'.format(src)
2496 else:
2497 msg = ('{!r} file is extracted '
2498 'into {!r} directory.').format(src, curdir)
2499 print(msg)
2500 else:
2501 parser.exit(1, '{!r} is not a tar archive.\n'.format(src))
2502
2503 elif args.create:
2504 tar_name = args.create.pop(0)
2505 _, ext = os.path.splitext(tar_name)
2506 compressions = {
2507 # gz
Serhiy Storchaka832dd5f2015-02-10 08:45:53 +02002508 '.gz': 'gz',
2509 '.tgz': 'gz',
Serhiy Storchakad27b4552013-11-24 01:53:29 +02002510 # xz
Serhiy Storchaka832dd5f2015-02-10 08:45:53 +02002511 '.xz': 'xz',
2512 '.txz': 'xz',
Serhiy Storchakad27b4552013-11-24 01:53:29 +02002513 # bz2
Serhiy Storchaka832dd5f2015-02-10 08:45:53 +02002514 '.bz2': 'bz2',
2515 '.tbz': 'bz2',
2516 '.tbz2': 'bz2',
2517 '.tb2': 'bz2',
Serhiy Storchakad27b4552013-11-24 01:53:29 +02002518 }
2519 tar_mode = 'w:' + compressions[ext] if ext in compressions else 'w'
2520 tar_files = args.create
2521
2522 with TarFile.open(tar_name, tar_mode) as tf:
2523 for file_name in tar_files:
2524 tf.add(file_name)
2525
2526 if args.verbose:
2527 print('{!r} file created.'.format(tar_name))
2528
2529 else:
2530 parser.exit(1, parser.format_help())
2531
2532if __name__ == '__main__':
2533 main()