blob: 178514470e8d273e4c8213136eb6992ade196eca [file] [log] [blame]
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001#!/usr/bin/env python
2# -*- coding: iso-8859-1 -*-
3#-------------------------------------------------------------------
4# tarfile.py
5#-------------------------------------------------------------------
6# Copyright (C) 2002 Lars Gustäbel <lars@gustaebel.de>
7# All rights reserved.
8#
9# Permission is hereby granted, free of charge, to any person
10# obtaining a copy of this software and associated documentation
11# files (the "Software"), to deal in the Software without
12# restriction, including without limitation the rights to use,
13# copy, modify, merge, publish, distribute, sublicense, and/or sell
14# copies of the Software, and to permit persons to whom the
15# Software is furnished to do so, subject to the following
16# conditions:
17#
18# The above copyright notice and this permission notice shall be
19# included in all copies or substantial portions of the Software.
20#
21# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
22# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
23# OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
24# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
25# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
26# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
27# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
28# OTHER DEALINGS IN THE SOFTWARE.
29#
30"""Read from and write to tar format archives.
31"""
32
33__version__ = "$Revision$"
34# $Source$
35
Georg Brandl38c6a222006-05-10 16:26:03 +000036version = "0.8.0"
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000037__author__ = "Lars Gustäbel (lars@gustaebel.de)"
38__date__ = "$Date$"
39__cvsid__ = "$Id$"
40__credits__ = "Gustavo Niemeyer, Niels Gustäbel, Richard Townsend."
41
42#---------
43# Imports
44#---------
45import sys
46import os
47import shutil
48import stat
49import errno
50import time
51import struct
Georg Brandl3354f282006-10-29 09:16:12 +000052import copy
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000053
Jack Jansencfc49022003-03-07 13:37:32 +000054if sys.platform == 'mac':
55 # This module needs work for MacOS9, especially in the area of pathname
56 # handling. In many places it is assumed a simple substitution of / by the
57 # local os.path.sep is good enough to convert pathnames, but this does not
58 # work with the mac rooted:path:name versus :nonrooted:path:name syntax
59 raise ImportError, "tarfile does not work for platform==mac"
60
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +000061try:
62 import grp, pwd
63except ImportError:
64 grp = pwd = None
65
66# from tarfile import *
67__all__ = ["TarFile", "TarInfo", "is_tarfile", "TarError"]
68
69#---------------------------------------------------------
70# tar constants
71#---------------------------------------------------------
72NUL = "\0" # the null character
73BLOCKSIZE = 512 # length of processing blocks
74RECORDSIZE = BLOCKSIZE * 20 # length of records
75MAGIC = "ustar" # magic tar string
76VERSION = "00" # version number
77
78LENGTH_NAME = 100 # maximum length of a filename
79LENGTH_LINK = 100 # maximum length of a linkname
80LENGTH_PREFIX = 155 # maximum length of the prefix field
81MAXSIZE_MEMBER = 077777777777L # maximum size of a file (11 octal digits)
82
83REGTYPE = "0" # regular file
84AREGTYPE = "\0" # regular file
85LNKTYPE = "1" # link (inside tarfile)
86SYMTYPE = "2" # symbolic link
87CHRTYPE = "3" # character special device
88BLKTYPE = "4" # block special device
89DIRTYPE = "5" # directory
90FIFOTYPE = "6" # fifo special device
91CONTTYPE = "7" # contiguous file
92
93GNUTYPE_LONGNAME = "L" # GNU tar extension for longnames
94GNUTYPE_LONGLINK = "K" # GNU tar extension for longlink
95GNUTYPE_SPARSE = "S" # GNU tar extension for sparse file
96
97#---------------------------------------------------------
98# tarfile constants
99#---------------------------------------------------------
100SUPPORTED_TYPES = (REGTYPE, AREGTYPE, LNKTYPE, # file types that tarfile
101 SYMTYPE, DIRTYPE, FIFOTYPE, # can cope with.
102 CONTTYPE, CHRTYPE, BLKTYPE,
103 GNUTYPE_LONGNAME, GNUTYPE_LONGLINK,
104 GNUTYPE_SPARSE)
105
106REGULAR_TYPES = (REGTYPE, AREGTYPE, # file types that somehow
107 CONTTYPE, GNUTYPE_SPARSE) # represent regular files
108
109#---------------------------------------------------------
110# Bits used in the mode field, values in octal.
111#---------------------------------------------------------
112S_IFLNK = 0120000 # symbolic link
113S_IFREG = 0100000 # regular file
114S_IFBLK = 0060000 # block device
115S_IFDIR = 0040000 # directory
116S_IFCHR = 0020000 # character device
117S_IFIFO = 0010000 # fifo
118
119TSUID = 04000 # set UID on execution
120TSGID = 02000 # set GID on execution
121TSVTX = 01000 # reserved
122
123TUREAD = 0400 # read by owner
124TUWRITE = 0200 # write by owner
125TUEXEC = 0100 # execute/search by owner
126TGREAD = 0040 # read by group
127TGWRITE = 0020 # write by group
128TGEXEC = 0010 # execute/search by group
129TOREAD = 0004 # read by other
130TOWRITE = 0002 # write by other
131TOEXEC = 0001 # execute/search by other
132
133#---------------------------------------------------------
134# Some useful functions
135#---------------------------------------------------------
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000136
Georg Brandl38c6a222006-05-10 16:26:03 +0000137def stn(s, length):
138 """Convert a python string to a null-terminated string buffer.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000139 """
Georg Brandla32e0a02006-10-24 16:54:16 +0000140 return s[:length] + (length - len(s)) * NUL
Georg Brandl38c6a222006-05-10 16:26:03 +0000141
142def nti(s):
143 """Convert a number field to a python number.
144 """
145 # There are two possible encodings for a number field, see
146 # itn() below.
147 if s[0] != chr(0200):
Georg Brandlded1c4d2006-12-20 11:55:16 +0000148 try:
149 n = int(s.rstrip(NUL + " ") or "0", 8)
150 except ValueError:
151 raise HeaderError("invalid header")
Georg Brandl38c6a222006-05-10 16:26:03 +0000152 else:
153 n = 0L
154 for i in xrange(len(s) - 1):
155 n <<= 8
156 n += ord(s[i + 1])
157 return n
158
159def itn(n, digits=8, posix=False):
160 """Convert a python number to a number field.
161 """
162 # POSIX 1003.1-1988 requires numbers to be encoded as a string of
163 # octal digits followed by a null-byte, this allows values up to
164 # (8**(digits-1))-1. GNU tar allows storing numbers greater than
165 # that if necessary. A leading 0200 byte indicates this particular
166 # encoding, the following digits-1 bytes are a big-endian
167 # representation. This allows values up to (256**(digits-1))-1.
168 if 0 <= n < 8 ** (digits - 1):
169 s = "%0*o" % (digits - 1, n) + NUL
170 else:
171 if posix:
Georg Brandle4751e32006-05-18 06:11:19 +0000172 raise ValueError("overflow in number field")
Georg Brandl38c6a222006-05-10 16:26:03 +0000173
174 if n < 0:
175 # XXX We mimic GNU tar's behaviour with negative numbers,
176 # this could raise OverflowError.
177 n = struct.unpack("L", struct.pack("l", n))[0]
178
179 s = ""
180 for i in xrange(digits - 1):
181 s = chr(n & 0377) + s
182 n >>= 8
183 s = chr(0200) + s
184 return s
185
186def calc_chksums(buf):
187 """Calculate the checksum for a member's header by summing up all
188 characters except for the chksum field which is treated as if
189 it was filled with spaces. According to the GNU tar sources,
190 some tars (Sun and NeXT) calculate chksum with signed char,
191 which will be different if there are chars in the buffer with
192 the high bit set. So we calculate two checksums, unsigned and
193 signed.
194 """
195 unsigned_chksum = 256 + sum(struct.unpack("148B", buf[:148]) + struct.unpack("356B", buf[156:512]))
196 signed_chksum = 256 + sum(struct.unpack("148b", buf[:148]) + struct.unpack("356b", buf[156:512]))
197 return unsigned_chksum, signed_chksum
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000198
199def copyfileobj(src, dst, length=None):
200 """Copy length bytes from fileobj src to fileobj dst.
201 If length is None, copy the entire content.
202 """
203 if length == 0:
204 return
205 if length is None:
206 shutil.copyfileobj(src, dst)
207 return
208
209 BUFSIZE = 16 * 1024
210 blocks, remainder = divmod(length, BUFSIZE)
211 for b in xrange(blocks):
212 buf = src.read(BUFSIZE)
213 if len(buf) < BUFSIZE:
Georg Brandle4751e32006-05-18 06:11:19 +0000214 raise IOError("end of file reached")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000215 dst.write(buf)
216
217 if remainder != 0:
218 buf = src.read(remainder)
219 if len(buf) < remainder:
Georg Brandle4751e32006-05-18 06:11:19 +0000220 raise IOError("end of file reached")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000221 dst.write(buf)
222 return
223
224filemode_table = (
Andrew M. Kuchling8bc462f2004-10-20 11:48:42 +0000225 ((S_IFLNK, "l"),
226 (S_IFREG, "-"),
227 (S_IFBLK, "b"),
228 (S_IFDIR, "d"),
229 (S_IFCHR, "c"),
230 (S_IFIFO, "p")),
231
232 ((TUREAD, "r"),),
233 ((TUWRITE, "w"),),
234 ((TUEXEC|TSUID, "s"),
235 (TSUID, "S"),
236 (TUEXEC, "x")),
237
238 ((TGREAD, "r"),),
239 ((TGWRITE, "w"),),
240 ((TGEXEC|TSGID, "s"),
241 (TSGID, "S"),
242 (TGEXEC, "x")),
243
244 ((TOREAD, "r"),),
245 ((TOWRITE, "w"),),
246 ((TOEXEC|TSVTX, "t"),
247 (TSVTX, "T"),
248 (TOEXEC, "x"))
249)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000250
251def filemode(mode):
252 """Convert a file's mode to a string of the form
253 -rwxrwxrwx.
254 Used by TarFile.list()
255 """
Andrew M. Kuchling8bc462f2004-10-20 11:48:42 +0000256 perm = []
257 for table in filemode_table:
258 for bit, char in table:
259 if mode & bit == bit:
260 perm.append(char)
261 break
262 else:
263 perm.append("-")
264 return "".join(perm)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000265
266if os.sep != "/":
267 normpath = lambda path: os.path.normpath(path).replace(os.sep, "/")
268else:
269 normpath = os.path.normpath
270
271class TarError(Exception):
272 """Base exception."""
273 pass
274class ExtractError(TarError):
275 """General exception for extract errors."""
276 pass
277class ReadError(TarError):
278 """Exception for unreadble tar archives."""
279 pass
280class CompressionError(TarError):
281 """Exception for unavailable compression methods."""
282 pass
283class StreamError(TarError):
284 """Exception for unsupported operations on stream-like TarFiles."""
285 pass
Georg Brandlebbeed72006-12-19 22:06:46 +0000286class HeaderError(TarError):
287 """Exception for invalid headers."""
288 pass
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000289
290#---------------------------
291# internal stream interface
292#---------------------------
293class _LowLevelFile:
294 """Low-level file object. Supports reading and writing.
295 It is used instead of a regular file object for streaming
296 access.
297 """
298
299 def __init__(self, name, mode):
300 mode = {
301 "r": os.O_RDONLY,
302 "w": os.O_WRONLY | os.O_CREAT | os.O_TRUNC,
303 }[mode]
304 if hasattr(os, "O_BINARY"):
305 mode |= os.O_BINARY
306 self.fd = os.open(name, mode)
307
308 def close(self):
309 os.close(self.fd)
310
311 def read(self, size):
312 return os.read(self.fd, size)
313
314 def write(self, s):
315 os.write(self.fd, s)
316
317class _Stream:
318 """Class that serves as an adapter between TarFile and
319 a stream-like object. The stream-like object only
320 needs to have a read() or write() method and is accessed
321 blockwise. Use of gzip or bzip2 compression is possible.
322 A stream-like object could be for example: sys.stdin,
323 sys.stdout, a socket, a tape device etc.
324
325 _Stream is intended to be used only internally.
326 """
327
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000328 def __init__(self, name, mode, comptype, fileobj, bufsize):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000329 """Construct a _Stream object.
330 """
331 self._extfileobj = True
332 if fileobj is None:
333 fileobj = _LowLevelFile(name, mode)
334 self._extfileobj = False
335
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000336 if comptype == '*':
337 # Enable transparent compression detection for the
338 # stream interface
339 fileobj = _StreamProxy(fileobj)
340 comptype = fileobj.getcomptype()
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000341
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000342 self.name = name or ""
343 self.mode = mode
344 self.comptype = comptype
345 self.fileobj = fileobj
346 self.bufsize = bufsize
347 self.buf = ""
348 self.pos = 0L
349 self.closed = False
350
351 if comptype == "gz":
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000352 try:
353 import zlib
354 except ImportError:
Georg Brandle4751e32006-05-18 06:11:19 +0000355 raise CompressionError("zlib module is not available")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000356 self.zlib = zlib
357 self.crc = zlib.crc32("")
358 if mode == "r":
359 self._init_read_gz()
360 else:
361 self._init_write_gz()
362
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000363 if comptype == "bz2":
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000364 try:
365 import bz2
366 except ImportError:
Georg Brandle4751e32006-05-18 06:11:19 +0000367 raise CompressionError("bz2 module is not available")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000368 if mode == "r":
369 self.dbuf = ""
370 self.cmp = bz2.BZ2Decompressor()
371 else:
372 self.cmp = bz2.BZ2Compressor()
373
374 def __del__(self):
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000375 if hasattr(self, "closed") and not self.closed:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000376 self.close()
377
378 def _init_write_gz(self):
379 """Initialize for writing with gzip compression.
380 """
381 self.cmp = self.zlib.compressobj(9, self.zlib.DEFLATED,
382 -self.zlib.MAX_WBITS,
383 self.zlib.DEF_MEM_LEVEL,
384 0)
385 timestamp = struct.pack("<L", long(time.time()))
386 self.__write("\037\213\010\010%s\002\377" % timestamp)
387 if self.name.endswith(".gz"):
388 self.name = self.name[:-3]
389 self.__write(self.name + NUL)
390
391 def write(self, s):
392 """Write string s to the stream.
393 """
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000394 if self.comptype == "gz":
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000395 self.crc = self.zlib.crc32(s, self.crc)
396 self.pos += len(s)
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000397 if self.comptype != "tar":
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000398 s = self.cmp.compress(s)
399 self.__write(s)
400
401 def __write(self, s):
402 """Write string s to the stream if a whole new block
403 is ready to be written.
404 """
405 self.buf += s
406 while len(self.buf) > self.bufsize:
407 self.fileobj.write(self.buf[:self.bufsize])
408 self.buf = self.buf[self.bufsize:]
409
410 def close(self):
411 """Close the _Stream object. No operation should be
412 done on it afterwards.
413 """
414 if self.closed:
415 return
416
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000417 if self.mode == "w" and self.comptype != "tar":
Martin v. Löwisc234a522004-08-22 21:28:33 +0000418 self.buf += self.cmp.flush()
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000419
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000420 if self.mode == "w" and self.buf:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000421 self.fileobj.write(self.buf)
422 self.buf = ""
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000423 if self.comptype == "gz":
Tim Petersa05f6e22006-08-02 05:20:08 +0000424 # The native zlib crc is an unsigned 32-bit integer, but
425 # the Python wrapper implicitly casts that to a signed C
426 # long. So, on a 32-bit box self.crc may "look negative",
427 # while the same crc on a 64-bit box may "look positive".
428 # To avoid irksome warnings from the `struct` module, force
429 # it to look positive on all boxes.
430 self.fileobj.write(struct.pack("<L", self.crc & 0xffffffffL))
Andrew M. Kuchling10a44492003-10-24 17:38:34 +0000431 self.fileobj.write(struct.pack("<L", self.pos & 0xffffFFFFL))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000432
433 if not self._extfileobj:
434 self.fileobj.close()
435
436 self.closed = True
437
438 def _init_read_gz(self):
439 """Initialize for reading a gzip compressed fileobj.
440 """
441 self.cmp = self.zlib.decompressobj(-self.zlib.MAX_WBITS)
442 self.dbuf = ""
443
444 # taken from gzip.GzipFile with some alterations
445 if self.__read(2) != "\037\213":
Georg Brandle4751e32006-05-18 06:11:19 +0000446 raise ReadError("not a gzip file")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000447 if self.__read(1) != "\010":
Georg Brandle4751e32006-05-18 06:11:19 +0000448 raise CompressionError("unsupported compression method")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000449
450 flag = ord(self.__read(1))
451 self.__read(6)
452
453 if flag & 4:
454 xlen = ord(self.__read(1)) + 256 * ord(self.__read(1))
455 self.read(xlen)
456 if flag & 8:
457 while True:
458 s = self.__read(1)
459 if not s or s == NUL:
460 break
461 if flag & 16:
462 while True:
463 s = self.__read(1)
464 if not s or s == NUL:
465 break
466 if flag & 2:
467 self.__read(2)
468
469 def tell(self):
470 """Return the stream's file pointer position.
471 """
472 return self.pos
473
474 def seek(self, pos=0):
475 """Set the stream's file pointer to pos. Negative seeking
476 is forbidden.
477 """
478 if pos - self.pos >= 0:
479 blocks, remainder = divmod(pos - self.pos, self.bufsize)
480 for i in xrange(blocks):
481 self.read(self.bufsize)
482 self.read(remainder)
483 else:
Georg Brandle4751e32006-05-18 06:11:19 +0000484 raise StreamError("seeking backwards is not allowed")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000485 return self.pos
486
487 def read(self, size=None):
488 """Return the next size number of bytes from the stream.
489 If size is not defined, return all bytes of the stream
490 up to EOF.
491 """
492 if size is None:
493 t = []
494 while True:
495 buf = self._read(self.bufsize)
496 if not buf:
497 break
498 t.append(buf)
499 buf = "".join(t)
500 else:
501 buf = self._read(size)
502 self.pos += len(buf)
503 return buf
504
505 def _read(self, size):
506 """Return size bytes from the stream.
507 """
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000508 if self.comptype == "tar":
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000509 return self.__read(size)
510
511 c = len(self.dbuf)
512 t = [self.dbuf]
513 while c < size:
514 buf = self.__read(self.bufsize)
515 if not buf:
516 break
517 buf = self.cmp.decompress(buf)
518 t.append(buf)
519 c += len(buf)
520 t = "".join(t)
521 self.dbuf = t[size:]
522 return t[:size]
523
524 def __read(self, size):
525 """Return size bytes from stream. If internal buffer is empty,
526 read another block from the stream.
527 """
528 c = len(self.buf)
529 t = [self.buf]
530 while c < size:
531 buf = self.fileobj.read(self.bufsize)
532 if not buf:
533 break
534 t.append(buf)
535 c += len(buf)
536 t = "".join(t)
537 self.buf = t[size:]
538 return t[:size]
539# class _Stream
540
Martin v. Löwis78be7df2005-03-05 12:47:42 +0000541class _StreamProxy(object):
542 """Small proxy class that enables transparent compression
543 detection for the Stream interface (mode 'r|*').
544 """
545
546 def __init__(self, fileobj):
547 self.fileobj = fileobj
548 self.buf = self.fileobj.read(BLOCKSIZE)
549
550 def read(self, size):
551 self.read = self.fileobj.read
552 return self.buf
553
554 def getcomptype(self):
555 if self.buf.startswith("\037\213\010"):
556 return "gz"
557 if self.buf.startswith("BZh91"):
558 return "bz2"
559 return "tar"
560
561 def close(self):
562 self.fileobj.close()
563# class StreamProxy
564
Georg Brandl49c8f4c2006-05-15 19:30:35 +0000565class _BZ2Proxy(object):
566 """Small proxy class that enables external file object
567 support for "r:bz2" and "w:bz2" modes. This is actually
568 a workaround for a limitation in bz2 module's BZ2File
569 class which (unlike gzip.GzipFile) has no support for
570 a file object argument.
571 """
572
573 blocksize = 16 * 1024
574
575 def __init__(self, fileobj, mode):
576 self.fileobj = fileobj
577 self.mode = mode
578 self.init()
579
580 def init(self):
581 import bz2
582 self.pos = 0
583 if self.mode == "r":
584 self.bz2obj = bz2.BZ2Decompressor()
585 self.fileobj.seek(0)
586 self.buf = ""
587 else:
588 self.bz2obj = bz2.BZ2Compressor()
589
590 def read(self, size):
591 b = [self.buf]
592 x = len(self.buf)
593 while x < size:
594 try:
595 raw = self.fileobj.read(self.blocksize)
596 data = self.bz2obj.decompress(raw)
597 b.append(data)
598 except EOFError:
599 break
600 x += len(data)
601 self.buf = "".join(b)
602
603 buf = self.buf[:size]
604 self.buf = self.buf[size:]
605 self.pos += len(buf)
606 return buf
607
608 def seek(self, pos):
609 if pos < self.pos:
610 self.init()
611 self.read(pos - self.pos)
612
613 def tell(self):
614 return self.pos
615
616 def write(self, data):
617 self.pos += len(data)
618 raw = self.bz2obj.compress(data)
619 self.fileobj.write(raw)
620
621 def close(self):
622 if self.mode == "w":
623 raw = self.bz2obj.flush()
624 self.fileobj.write(raw)
Georg Brandle8953182006-05-27 14:02:03 +0000625 self.fileobj.close()
Georg Brandl49c8f4c2006-05-15 19:30:35 +0000626# class _BZ2Proxy
627
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000628#------------------------
629# Extraction file object
630#------------------------
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000631class _FileInFile(object):
632 """A thin wrapper around an existing file object that
633 provides a part of its data as an individual file
634 object.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000635 """
636
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000637 def __init__(self, fileobj, offset, size, sparse=None):
638 self.fileobj = fileobj
639 self.offset = offset
640 self.size = size
641 self.sparse = sparse
642 self.position = 0
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000643
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000644 def tell(self):
645 """Return the current file position.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000646 """
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000647 return self.position
648
649 def seek(self, position):
650 """Seek to a position in the file.
651 """
652 self.position = position
653
654 def read(self, size=None):
655 """Read data from the file.
656 """
657 if size is None:
658 size = self.size - self.position
659 else:
660 size = min(size, self.size - self.position)
661
662 if self.sparse is None:
663 return self.readnormal(size)
664 else:
665 return self.readsparse(size)
666
667 def readnormal(self, size):
668 """Read operation for regular files.
669 """
670 self.fileobj.seek(self.offset + self.position)
671 self.position += size
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000672 return self.fileobj.read(size)
673
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000674 def readsparse(self, size):
675 """Read operation for sparse files.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000676 """
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000677 data = []
678 while size > 0:
679 buf = self.readsparsesection(size)
680 if not buf:
681 break
682 size -= len(buf)
683 data.append(buf)
684 return "".join(data)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000685
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000686 def readsparsesection(self, size):
687 """Read a single section of a sparse file.
688 """
689 section = self.sparse.find(self.position)
690
691 if section is None:
692 return ""
693
694 size = min(size, section.offset + section.size - self.position)
695
696 if isinstance(section, _data):
697 realpos = section.realpos + self.position - section.offset
698 self.fileobj.seek(self.offset + realpos)
699 self.position += size
700 return self.fileobj.read(size)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000701 else:
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000702 self.position += size
703 return NUL * size
704#class _FileInFile
705
706
707class ExFileObject(object):
708 """File-like object for reading an archive member.
709 Is returned by TarFile.extractfile().
710 """
711 blocksize = 1024
712
713 def __init__(self, tarfile, tarinfo):
714 self.fileobj = _FileInFile(tarfile.fileobj,
715 tarinfo.offset_data,
716 tarinfo.size,
717 getattr(tarinfo, "sparse", None))
718 self.name = tarinfo.name
719 self.mode = "r"
720 self.closed = False
721 self.size = tarinfo.size
722
723 self.position = 0
724 self.buffer = ""
725
726 def read(self, size=None):
727 """Read at most size bytes from the file. If size is not
728 present or None, read all data until EOF is reached.
729 """
730 if self.closed:
731 raise ValueError("I/O operation on closed file")
732
733 buf = ""
734 if self.buffer:
735 if size is None:
736 buf = self.buffer
737 self.buffer = ""
738 else:
739 buf = self.buffer[:size]
740 self.buffer = self.buffer[size:]
741
742 if size is None:
743 buf += self.fileobj.read()
744 else:
745 buf += self.fileobj.read(size - len(buf))
746
747 self.position += len(buf)
748 return buf
749
750 def readline(self, size=-1):
751 """Read one entire line from the file. If size is present
752 and non-negative, return a string with at most that
753 size, which may be an incomplete line.
754 """
755 if self.closed:
756 raise ValueError("I/O operation on closed file")
757
758 if "\n" in self.buffer:
759 pos = self.buffer.find("\n") + 1
760 else:
761 buffers = [self.buffer]
762 while True:
763 buf = self.fileobj.read(self.blocksize)
764 buffers.append(buf)
765 if not buf or "\n" in buf:
766 self.buffer = "".join(buffers)
767 pos = self.buffer.find("\n") + 1
768 if pos == 0:
769 # no newline found.
770 pos = len(self.buffer)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000771 break
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000772
773 if size != -1:
774 pos = min(size, pos)
775
776 buf = self.buffer[:pos]
777 self.buffer = self.buffer[pos:]
778 self.position += len(buf)
779 return buf
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000780
781 def readlines(self):
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000782 """Return a list with all remaining lines.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000783 """
784 result = []
785 while True:
786 line = self.readline()
787 if not line: break
788 result.append(line)
789 return result
790
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000791 def tell(self):
792 """Return the current file position.
793 """
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000794 if self.closed:
795 raise ValueError("I/O operation on closed file")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000796
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000797 return self.position
798
799 def seek(self, pos, whence=os.SEEK_SET):
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000800 """Seek to a position in the file.
801 """
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000802 if self.closed:
803 raise ValueError("I/O operation on closed file")
804
805 if whence == os.SEEK_SET:
806 self.position = min(max(pos, 0), self.size)
807 elif whence == os.SEEK_CUR:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000808 if pos < 0:
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000809 self.position = max(self.position + pos, 0)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000810 else:
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000811 self.position = min(self.position + pos, self.size)
812 elif whence == os.SEEK_END:
813 self.position = max(min(self.size + pos, self.size), 0)
814 else:
815 raise ValueError("Invalid argument")
816
817 self.buffer = ""
818 self.fileobj.seek(self.position)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000819
820 def close(self):
821 """Close the file object.
822 """
823 self.closed = True
Martin v. Löwisdf241532005-03-03 08:17:42 +0000824
825 def __iter__(self):
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000826 """Get an iterator over the file's lines.
Martin v. Löwisdf241532005-03-03 08:17:42 +0000827 """
Lars Gustäbel6baa5022006-12-23 16:40:13 +0000828 while True:
829 line = self.readline()
830 if not line:
831 break
832 yield line
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000833#class ExFileObject
834
835#------------------
836# Exported Classes
837#------------------
838class TarInfo(object):
839 """Informational class which holds the details about an
840 archive member given by a tar header block.
841 TarInfo objects are returned by TarFile.getmember(),
842 TarFile.getmembers() and TarFile.gettarinfo() and are
843 usually created internally.
844 """
845
846 def __init__(self, name=""):
847 """Construct a TarInfo object. name is the optional name
848 of the member.
849 """
Georg Brandl38c6a222006-05-10 16:26:03 +0000850 self.name = name # member name (dirnames must end with '/')
851 self.mode = 0666 # file permissions
852 self.uid = 0 # user id
853 self.gid = 0 # group id
854 self.size = 0 # file size
855 self.mtime = 0 # modification time
856 self.chksum = 0 # header checksum
857 self.type = REGTYPE # member type
858 self.linkname = "" # link name
859 self.uname = "user" # user name
860 self.gname = "group" # group name
861 self.devmajor = 0 # device major number
862 self.devminor = 0 # device minor number
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000863
Georg Brandl38c6a222006-05-10 16:26:03 +0000864 self.offset = 0 # the tar header starts here
865 self.offset_data = 0 # the file's data starts here
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000866
867 def __repr__(self):
868 return "<%s %r at %#x>" % (self.__class__.__name__,self.name,id(self))
869
Guido van Rossum75b64e62005-01-16 00:16:11 +0000870 @classmethod
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000871 def frombuf(cls, buf):
872 """Construct a TarInfo object from a 512 byte string buffer.
873 """
Georg Brandl38c6a222006-05-10 16:26:03 +0000874 if len(buf) != BLOCKSIZE:
Georg Brandlebbeed72006-12-19 22:06:46 +0000875 raise HeaderError("truncated header")
Georg Brandl38c6a222006-05-10 16:26:03 +0000876 if buf.count(NUL) == BLOCKSIZE:
Georg Brandlebbeed72006-12-19 22:06:46 +0000877 raise HeaderError("empty header")
878
Georg Brandlded1c4d2006-12-20 11:55:16 +0000879 chksum = nti(buf[148:156])
Georg Brandlebbeed72006-12-19 22:06:46 +0000880 if chksum not in calc_chksums(buf):
881 raise HeaderError("bad checksum")
Georg Brandl38c6a222006-05-10 16:26:03 +0000882
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000883 tarinfo = cls()
Georg Brandl38c6a222006-05-10 16:26:03 +0000884 tarinfo.buf = buf
Georg Brandle8953182006-05-27 14:02:03 +0000885 tarinfo.name = buf[0:100].rstrip(NUL)
Georg Brandl38c6a222006-05-10 16:26:03 +0000886 tarinfo.mode = nti(buf[100:108])
887 tarinfo.uid = nti(buf[108:116])
888 tarinfo.gid = nti(buf[116:124])
889 tarinfo.size = nti(buf[124:136])
890 tarinfo.mtime = nti(buf[136:148])
Georg Brandlebbeed72006-12-19 22:06:46 +0000891 tarinfo.chksum = chksum
Georg Brandl38c6a222006-05-10 16:26:03 +0000892 tarinfo.type = buf[156:157]
Georg Brandle8953182006-05-27 14:02:03 +0000893 tarinfo.linkname = buf[157:257].rstrip(NUL)
894 tarinfo.uname = buf[265:297].rstrip(NUL)
895 tarinfo.gname = buf[297:329].rstrip(NUL)
Georg Brandl38c6a222006-05-10 16:26:03 +0000896 tarinfo.devmajor = nti(buf[329:337])
897 tarinfo.devminor = nti(buf[337:345])
Georg Brandl3354f282006-10-29 09:16:12 +0000898 prefix = buf[345:500].rstrip(NUL)
899
900 if prefix and not tarinfo.issparse():
901 tarinfo.name = prefix + "/" + tarinfo.name
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000902
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000903 return tarinfo
904
Georg Brandl38c6a222006-05-10 16:26:03 +0000905 def tobuf(self, posix=False):
Georg Brandl3354f282006-10-29 09:16:12 +0000906 """Return a tar header as a string of 512 byte blocks.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000907 """
Georg Brandl3354f282006-10-29 09:16:12 +0000908 buf = ""
909 type = self.type
910 prefix = ""
911
912 if self.name.endswith("/"):
913 type = DIRTYPE
914
Georg Brandl87fa5592006-12-06 22:21:18 +0000915 if type in (GNUTYPE_LONGNAME, GNUTYPE_LONGLINK):
916 # Prevent "././@LongLink" from being normalized.
917 name = self.name
918 else:
919 name = normpath(self.name)
Georg Brandl3354f282006-10-29 09:16:12 +0000920
921 if type == DIRTYPE:
922 # directories should end with '/'
923 name += "/"
924
925 linkname = self.linkname
926 if linkname:
927 # if linkname is empty we end up with a '.'
928 linkname = normpath(linkname)
929
930 if posix:
931 if self.size > MAXSIZE_MEMBER:
932 raise ValueError("file is too large (>= 8 GB)")
933
934 if len(self.linkname) > LENGTH_LINK:
935 raise ValueError("linkname is too long (>%d)" % (LENGTH_LINK))
936
937 if len(name) > LENGTH_NAME:
938 prefix = name[:LENGTH_PREFIX + 1]
939 while prefix and prefix[-1] != "/":
940 prefix = prefix[:-1]
941
942 name = name[len(prefix):]
943 prefix = prefix[:-1]
944
945 if not prefix or len(name) > LENGTH_NAME:
946 raise ValueError("name is too long")
947
948 else:
949 if len(self.linkname) > LENGTH_LINK:
950 buf += self._create_gnulong(self.linkname, GNUTYPE_LONGLINK)
951
952 if len(name) > LENGTH_NAME:
953 buf += self._create_gnulong(name, GNUTYPE_LONGNAME)
954
Georg Brandl38c6a222006-05-10 16:26:03 +0000955 parts = [
Georg Brandl3354f282006-10-29 09:16:12 +0000956 stn(name, 100),
Georg Brandl38c6a222006-05-10 16:26:03 +0000957 itn(self.mode & 07777, 8, posix),
958 itn(self.uid, 8, posix),
959 itn(self.gid, 8, posix),
960 itn(self.size, 12, posix),
961 itn(self.mtime, 12, posix),
962 " ", # checksum field
Georg Brandl3354f282006-10-29 09:16:12 +0000963 type,
Georg Brandl38c6a222006-05-10 16:26:03 +0000964 stn(self.linkname, 100),
965 stn(MAGIC, 6),
966 stn(VERSION, 2),
967 stn(self.uname, 32),
968 stn(self.gname, 32),
969 itn(self.devmajor, 8, posix),
970 itn(self.devminor, 8, posix),
Georg Brandl3354f282006-10-29 09:16:12 +0000971 stn(prefix, 155)
Georg Brandl38c6a222006-05-10 16:26:03 +0000972 ]
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000973
Georg Brandl3354f282006-10-29 09:16:12 +0000974 buf += struct.pack("%ds" % BLOCKSIZE, "".join(parts))
Georg Brandl87fa5592006-12-06 22:21:18 +0000975 chksum = calc_chksums(buf[-BLOCKSIZE:])[0]
Georg Brandl3354f282006-10-29 09:16:12 +0000976 buf = buf[:-364] + "%06o\0" % chksum + buf[-357:]
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +0000977 self.buf = buf
978 return buf
979
Georg Brandl3354f282006-10-29 09:16:12 +0000980 def _create_gnulong(self, name, type):
981 """Create a GNU longname/longlink header from name.
982 It consists of an extended tar header, with the length
983 of the longname as size, followed by data blocks,
984 which contain the longname as a null terminated string.
985 """
986 name += NUL
987
988 tarinfo = self.__class__()
989 tarinfo.name = "././@LongLink"
990 tarinfo.type = type
991 tarinfo.mode = 0
992 tarinfo.size = len(name)
993
994 # create extended header
995 buf = tarinfo.tobuf()
996 # create name blocks
997 buf += name
998 blocks, remainder = divmod(len(name), BLOCKSIZE)
999 if remainder > 0:
1000 buf += (BLOCKSIZE - remainder) * NUL
1001 return buf
1002
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001003 def isreg(self):
1004 return self.type in REGULAR_TYPES
1005 def isfile(self):
1006 return self.isreg()
1007 def isdir(self):
1008 return self.type == DIRTYPE
1009 def issym(self):
1010 return self.type == SYMTYPE
1011 def islnk(self):
1012 return self.type == LNKTYPE
1013 def ischr(self):
1014 return self.type == CHRTYPE
1015 def isblk(self):
1016 return self.type == BLKTYPE
1017 def isfifo(self):
1018 return self.type == FIFOTYPE
1019 def issparse(self):
1020 return self.type == GNUTYPE_SPARSE
1021 def isdev(self):
1022 return self.type in (CHRTYPE, BLKTYPE, FIFOTYPE)
1023# class TarInfo
1024
1025class TarFile(object):
1026 """The TarFile Class provides an interface to tar archives.
1027 """
1028
1029 debug = 0 # May be set from 0 (no msgs) to 3 (all msgs)
1030
1031 dereference = False # If true, add content of linked file to the
1032 # tar file, else the link.
1033
1034 ignore_zeros = False # If true, skips empty or invalid blocks and
1035 # continues processing.
1036
1037 errorlevel = 0 # If 0, fatal errors only appear in debug
1038 # messages (if debug >= 0). If > 0, errors
1039 # are passed to the caller as exceptions.
1040
Martin v. Löwis75b9da42004-08-18 13:57:44 +00001041 posix = False # If True, generates POSIX.1-1990-compliant
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001042 # archives (no GNU extensions!)
1043
1044 fileobject = ExFileObject
1045
1046 def __init__(self, name=None, mode="r", fileobj=None):
1047 """Open an (uncompressed) tar archive `name'. `mode' is either 'r' to
1048 read from an existing archive, 'a' to append data to an existing
1049 file or 'w' to create a new file overwriting an existing one. `mode'
1050 defaults to 'r'.
1051 If `fileobj' is given, it is used for reading or writing data. If it
1052 can be determined, `mode' is overridden by `fileobj's mode.
1053 `fileobj' is not closed, when TarFile is closed.
1054 """
Lars Gustäbela4b23812006-12-23 17:57:23 +00001055 self.name = os.path.abspath(name)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001056
1057 if len(mode) > 1 or mode not in "raw":
Georg Brandle4751e32006-05-18 06:11:19 +00001058 raise ValueError("mode must be 'r', 'a' or 'w'")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001059 self._mode = mode
1060 self.mode = {"r": "rb", "a": "r+b", "w": "wb"}[mode]
1061
1062 if not fileobj:
1063 fileobj = file(self.name, self.mode)
1064 self._extfileobj = False
1065 else:
1066 if self.name is None and hasattr(fileobj, "name"):
Lars Gustäbela4b23812006-12-23 17:57:23 +00001067 self.name = os.path.abspath(fileobj.name)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001068 if hasattr(fileobj, "mode"):
1069 self.mode = fileobj.mode
1070 self._extfileobj = True
1071 self.fileobj = fileobj
1072
1073 # Init datastructures
Georg Brandl38c6a222006-05-10 16:26:03 +00001074 self.closed = False
1075 self.members = [] # list of members as TarInfo objects
1076 self._loaded = False # flag if all members have been read
1077 self.offset = 0L # current position in the archive file
1078 self.inodes = {} # dictionary caching the inodes of
1079 # archive members already added
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001080
1081 if self._mode == "r":
1082 self.firstmember = None
1083 self.firstmember = self.next()
1084
1085 if self._mode == "a":
1086 # Move to the end of the archive,
1087 # before the first empty block.
1088 self.firstmember = None
1089 while True:
1090 try:
1091 tarinfo = self.next()
1092 except ReadError:
1093 self.fileobj.seek(0)
1094 break
1095 if tarinfo is None:
1096 self.fileobj.seek(- BLOCKSIZE, 1)
1097 break
1098
1099 if self._mode in "aw":
1100 self._loaded = True
1101
1102 #--------------------------------------------------------------------------
1103 # Below are the classmethods which act as alternate constructors to the
1104 # TarFile class. The open() method is the only one that is needed for
1105 # public use; it is the "super"-constructor and is able to select an
1106 # adequate "sub"-constructor for a particular compression using the mapping
1107 # from OPEN_METH.
1108 #
1109 # This concept allows one to subclass TarFile without losing the comfort of
1110 # the super-constructor. A sub-constructor is registered and made available
1111 # by adding it to the mapping in OPEN_METH.
1112
Guido van Rossum75b64e62005-01-16 00:16:11 +00001113 @classmethod
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001114 def open(cls, name=None, mode="r", fileobj=None, bufsize=20*512):
1115 """Open a tar archive for reading, writing or appending. Return
1116 an appropriate TarFile class.
1117
1118 mode:
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001119 'r' or 'r:*' open for reading with transparent compression
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001120 'r:' open for reading exclusively uncompressed
1121 'r:gz' open for reading with gzip compression
1122 'r:bz2' open for reading with bzip2 compression
1123 'a' or 'a:' open for appending
1124 'w' or 'w:' open for writing without compression
1125 'w:gz' open for writing with gzip compression
1126 'w:bz2' open for writing with bzip2 compression
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001127
1128 'r|*' open a stream of tar blocks with transparent compression
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001129 'r|' open an uncompressed stream of tar blocks for reading
1130 'r|gz' open a gzip compressed stream of tar blocks
1131 'r|bz2' open a bzip2 compressed stream of tar blocks
1132 'w|' open an uncompressed stream for writing
1133 'w|gz' open a gzip compressed stream for writing
1134 'w|bz2' open a bzip2 compressed stream for writing
1135 """
1136
1137 if not name and not fileobj:
Georg Brandle4751e32006-05-18 06:11:19 +00001138 raise ValueError("nothing to open")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001139
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001140 if mode in ("r", "r:*"):
1141 # Find out which *open() is appropriate for opening the file.
1142 for comptype in cls.OPEN_METH:
1143 func = getattr(cls, cls.OPEN_METH[comptype])
Lars Gustäbela7ba6fc2006-12-27 10:30:46 +00001144 if fileobj is not None:
1145 saved_pos = fileobj.tell()
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001146 try:
1147 return func(name, "r", fileobj)
1148 except (ReadError, CompressionError):
Lars Gustäbela7ba6fc2006-12-27 10:30:46 +00001149 if fileobj is not None:
1150 fileobj.seek(saved_pos)
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001151 continue
Georg Brandle4751e32006-05-18 06:11:19 +00001152 raise ReadError("file could not be opened successfully")
Martin v. Löwis78be7df2005-03-05 12:47:42 +00001153
1154 elif ":" in mode:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001155 filemode, comptype = mode.split(":", 1)
1156 filemode = filemode or "r"
1157 comptype = comptype or "tar"
1158
1159 # Select the *open() function according to
1160 # given compression.
1161 if comptype in cls.OPEN_METH:
1162 func = getattr(cls, cls.OPEN_METH[comptype])
1163 else:
Georg Brandle4751e32006-05-18 06:11:19 +00001164 raise CompressionError("unknown compression type %r" % comptype)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001165 return func(name, filemode, fileobj)
1166
1167 elif "|" in mode:
1168 filemode, comptype = mode.split("|", 1)
1169 filemode = filemode or "r"
1170 comptype = comptype or "tar"
1171
1172 if filemode not in "rw":
Georg Brandle4751e32006-05-18 06:11:19 +00001173 raise ValueError("mode must be 'r' or 'w'")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001174
1175 t = cls(name, filemode,
1176 _Stream(name, filemode, comptype, fileobj, bufsize))
1177 t._extfileobj = False
1178 return t
1179
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001180 elif mode in "aw":
1181 return cls.taropen(name, mode, fileobj)
1182
Georg Brandle4751e32006-05-18 06:11:19 +00001183 raise ValueError("undiscernible mode")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001184
Guido van Rossum75b64e62005-01-16 00:16:11 +00001185 @classmethod
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001186 def taropen(cls, name, mode="r", fileobj=None):
1187 """Open uncompressed tar archive name for reading or writing.
1188 """
1189 if len(mode) > 1 or mode not in "raw":
Georg Brandle4751e32006-05-18 06:11:19 +00001190 raise ValueError("mode must be 'r', 'a' or 'w'")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001191 return cls(name, mode, fileobj)
1192
Guido van Rossum75b64e62005-01-16 00:16:11 +00001193 @classmethod
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001194 def gzopen(cls, name, mode="r", fileobj=None, compresslevel=9):
1195 """Open gzip compressed tar archive name for reading or writing.
1196 Appending is not allowed.
1197 """
1198 if len(mode) > 1 or mode not in "rw":
Georg Brandle4751e32006-05-18 06:11:19 +00001199 raise ValueError("mode must be 'r' or 'w'")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001200
1201 try:
1202 import gzip
Neal Norwitz4ec68242003-04-11 03:05:56 +00001203 gzip.GzipFile
1204 except (ImportError, AttributeError):
Georg Brandle4751e32006-05-18 06:11:19 +00001205 raise CompressionError("gzip module is not available")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001206
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001207 if fileobj is None:
1208 fileobj = file(name, mode + "b")
1209
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001210 try:
Lars Gustäbela4b23812006-12-23 17:57:23 +00001211 t = cls.taropen(name, mode,
1212 gzip.GzipFile(name, mode, compresslevel, fileobj))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001213 except IOError:
Georg Brandle4751e32006-05-18 06:11:19 +00001214 raise ReadError("not a gzip file")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001215 t._extfileobj = False
1216 return t
1217
Guido van Rossum75b64e62005-01-16 00:16:11 +00001218 @classmethod
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001219 def bz2open(cls, name, mode="r", fileobj=None, compresslevel=9):
1220 """Open bzip2 compressed tar archive name for reading or writing.
1221 Appending is not allowed.
1222 """
1223 if len(mode) > 1 or mode not in "rw":
Georg Brandle4751e32006-05-18 06:11:19 +00001224 raise ValueError("mode must be 'r' or 'w'.")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001225
1226 try:
1227 import bz2
1228 except ImportError:
Georg Brandle4751e32006-05-18 06:11:19 +00001229 raise CompressionError("bz2 module is not available")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001230
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001231 if fileobj is not None:
Georg Brandl49c8f4c2006-05-15 19:30:35 +00001232 fileobj = _BZ2Proxy(fileobj, mode)
1233 else:
1234 fileobj = bz2.BZ2File(name, mode, compresslevel=compresslevel)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001235
1236 try:
Lars Gustäbela4b23812006-12-23 17:57:23 +00001237 t = cls.taropen(name, mode, fileobj)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001238 except IOError:
Georg Brandle4751e32006-05-18 06:11:19 +00001239 raise ReadError("not a bzip2 file")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001240 t._extfileobj = False
1241 return t
1242
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001243 # All *open() methods are registered here.
1244 OPEN_METH = {
1245 "tar": "taropen", # uncompressed tar
1246 "gz": "gzopen", # gzip compressed tar
1247 "bz2": "bz2open" # bzip2 compressed tar
1248 }
1249
1250 #--------------------------------------------------------------------------
1251 # The public methods which TarFile provides:
1252
1253 def close(self):
1254 """Close the TarFile. In write-mode, two finishing zero blocks are
1255 appended to the archive.
1256 """
1257 if self.closed:
1258 return
1259
1260 if self._mode in "aw":
1261 self.fileobj.write(NUL * (BLOCKSIZE * 2))
1262 self.offset += (BLOCKSIZE * 2)
1263 # fill up the end with zero-blocks
1264 # (like option -b20 for tar does)
1265 blocks, remainder = divmod(self.offset, RECORDSIZE)
1266 if remainder > 0:
1267 self.fileobj.write(NUL * (RECORDSIZE - remainder))
1268
1269 if not self._extfileobj:
1270 self.fileobj.close()
1271 self.closed = True
1272
1273 def getmember(self, name):
1274 """Return a TarInfo object for member `name'. If `name' can not be
1275 found in the archive, KeyError is raised. If a member occurs more
1276 than once in the archive, its last occurence is assumed to be the
1277 most up-to-date version.
1278 """
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001279 tarinfo = self._getmember(name)
1280 if tarinfo is None:
Georg Brandle4751e32006-05-18 06:11:19 +00001281 raise KeyError("filename %r not found" % name)
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001282 return tarinfo
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001283
1284 def getmembers(self):
1285 """Return the members of the archive as a list of TarInfo objects. The
1286 list has the same order as the members in the archive.
1287 """
1288 self._check()
1289 if not self._loaded: # if we want to obtain a list of
1290 self._load() # all members, we first have to
1291 # scan the whole archive.
1292 return self.members
1293
1294 def getnames(self):
1295 """Return the members of the archive as a list of their names. It has
1296 the same order as the list returned by getmembers().
1297 """
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001298 return [tarinfo.name for tarinfo in self.getmembers()]
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001299
1300 def gettarinfo(self, name=None, arcname=None, fileobj=None):
1301 """Create a TarInfo object for either the file `name' or the file
1302 object `fileobj' (using os.fstat on its file descriptor). You can
1303 modify some of the TarInfo's attributes before you add it using
1304 addfile(). If given, `arcname' specifies an alternative name for the
1305 file in the archive.
1306 """
1307 self._check("aw")
1308
1309 # When fileobj is given, replace name by
1310 # fileobj's real name.
1311 if fileobj is not None:
1312 name = fileobj.name
1313
1314 # Building the name of the member in the archive.
1315 # Backward slashes are converted to forward slashes,
1316 # Absolute paths are turned to relative paths.
1317 if arcname is None:
1318 arcname = name
1319 arcname = normpath(arcname)
1320 drv, arcname = os.path.splitdrive(arcname)
1321 while arcname[0:1] == "/":
1322 arcname = arcname[1:]
1323
1324 # Now, fill the TarInfo object with
1325 # information specific for the file.
1326 tarinfo = TarInfo()
1327
1328 # Use os.stat or os.lstat, depending on platform
1329 # and if symlinks shall be resolved.
1330 if fileobj is None:
1331 if hasattr(os, "lstat") and not self.dereference:
1332 statres = os.lstat(name)
1333 else:
1334 statres = os.stat(name)
1335 else:
1336 statres = os.fstat(fileobj.fileno())
1337 linkname = ""
1338
1339 stmd = statres.st_mode
1340 if stat.S_ISREG(stmd):
1341 inode = (statres.st_ino, statres.st_dev)
Neal Norwitzb0e32e22005-10-20 04:50:13 +00001342 if not self.dereference and \
1343 statres.st_nlink > 1 and inode in self.inodes:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001344 # Is it a hardlink to an already
1345 # archived file?
1346 type = LNKTYPE
1347 linkname = self.inodes[inode]
1348 else:
1349 # The inode is added only if its valid.
1350 # For win32 it is always 0.
1351 type = REGTYPE
1352 if inode[0]:
1353 self.inodes[inode] = arcname
1354 elif stat.S_ISDIR(stmd):
1355 type = DIRTYPE
1356 if arcname[-1:] != "/":
1357 arcname += "/"
1358 elif stat.S_ISFIFO(stmd):
1359 type = FIFOTYPE
1360 elif stat.S_ISLNK(stmd):
1361 type = SYMTYPE
1362 linkname = os.readlink(name)
1363 elif stat.S_ISCHR(stmd):
1364 type = CHRTYPE
1365 elif stat.S_ISBLK(stmd):
1366 type = BLKTYPE
1367 else:
1368 return None
1369
1370 # Fill the TarInfo object with all
1371 # information we can get.
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001372 tarinfo.name = arcname
1373 tarinfo.mode = stmd
1374 tarinfo.uid = statres.st_uid
1375 tarinfo.gid = statres.st_gid
1376 if stat.S_ISREG(stmd):
Martin v. Löwis61d77e02004-08-20 06:35:46 +00001377 tarinfo.size = statres.st_size
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001378 else:
1379 tarinfo.size = 0L
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001380 tarinfo.mtime = statres.st_mtime
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001381 tarinfo.type = type
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001382 tarinfo.linkname = linkname
1383 if pwd:
1384 try:
1385 tarinfo.uname = pwd.getpwuid(tarinfo.uid)[0]
1386 except KeyError:
1387 pass
1388 if grp:
1389 try:
1390 tarinfo.gname = grp.getgrgid(tarinfo.gid)[0]
1391 except KeyError:
1392 pass
1393
1394 if type in (CHRTYPE, BLKTYPE):
1395 if hasattr(os, "major") and hasattr(os, "minor"):
1396 tarinfo.devmajor = os.major(statres.st_rdev)
1397 tarinfo.devminor = os.minor(statres.st_rdev)
1398 return tarinfo
1399
1400 def list(self, verbose=True):
1401 """Print a table of contents to sys.stdout. If `verbose' is False, only
1402 the names of the members are printed. If it is True, an `ls -l'-like
1403 output is produced.
1404 """
1405 self._check()
1406
1407 for tarinfo in self:
1408 if verbose:
1409 print filemode(tarinfo.mode),
1410 print "%s/%s" % (tarinfo.uname or tarinfo.uid,
1411 tarinfo.gname or tarinfo.gid),
1412 if tarinfo.ischr() or tarinfo.isblk():
1413 print "%10s" % ("%d,%d" \
1414 % (tarinfo.devmajor, tarinfo.devminor)),
1415 else:
1416 print "%10d" % tarinfo.size,
1417 print "%d-%02d-%02d %02d:%02d:%02d" \
1418 % time.localtime(tarinfo.mtime)[:6],
1419
1420 print tarinfo.name,
1421
1422 if verbose:
1423 if tarinfo.issym():
1424 print "->", tarinfo.linkname,
1425 if tarinfo.islnk():
1426 print "link to", tarinfo.linkname,
1427 print
1428
1429 def add(self, name, arcname=None, recursive=True):
1430 """Add the file `name' to the archive. `name' may be any type of file
1431 (directory, fifo, symbolic link, etc.). If given, `arcname'
1432 specifies an alternative name for the file in the archive.
1433 Directories are added recursively by default. This can be avoided by
1434 setting `recursive' to False.
1435 """
1436 self._check("aw")
1437
1438 if arcname is None:
1439 arcname = name
1440
1441 # Skip if somebody tries to archive the archive...
Lars Gustäbela4b23812006-12-23 17:57:23 +00001442 if self.name is not None and os.path.abspath(name) == self.name:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001443 self._dbg(2, "tarfile: Skipped %r" % name)
1444 return
1445
1446 # Special case: The user wants to add the current
1447 # working directory.
1448 if name == ".":
1449 if recursive:
1450 if arcname == ".":
1451 arcname = ""
1452 for f in os.listdir("."):
1453 self.add(f, os.path.join(arcname, f))
1454 return
1455
1456 self._dbg(1, name)
1457
1458 # Create a TarInfo object from the file.
1459 tarinfo = self.gettarinfo(name, arcname)
1460
1461 if tarinfo is None:
1462 self._dbg(1, "tarfile: Unsupported type %r" % name)
1463 return
1464
1465 # Append the tar header and data to the archive.
1466 if tarinfo.isreg():
1467 f = file(name, "rb")
1468 self.addfile(tarinfo, f)
1469 f.close()
1470
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001471 elif tarinfo.isdir():
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001472 self.addfile(tarinfo)
1473 if recursive:
1474 for f in os.listdir(name):
1475 self.add(os.path.join(name, f), os.path.join(arcname, f))
1476
Martin v. Löwis5dbdc592005-08-27 10:07:56 +00001477 else:
1478 self.addfile(tarinfo)
1479
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001480 def addfile(self, tarinfo, fileobj=None):
1481 """Add the TarInfo object `tarinfo' to the archive. If `fileobj' is
1482 given, tarinfo.size bytes are read from it and added to the archive.
1483 You can create TarInfo objects using gettarinfo().
1484 On Windows platforms, `fileobj' should always be opened with mode
1485 'rb' to avoid irritation about the file size.
1486 """
1487 self._check("aw")
1488
Georg Brandl3354f282006-10-29 09:16:12 +00001489 tarinfo = copy.copy(tarinfo)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001490
Georg Brandl3354f282006-10-29 09:16:12 +00001491 buf = tarinfo.tobuf(self.posix)
1492 self.fileobj.write(buf)
1493 self.offset += len(buf)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001494
1495 # If there's data to follow, append it.
1496 if fileobj is not None:
1497 copyfileobj(fileobj, self.fileobj, tarinfo.size)
1498 blocks, remainder = divmod(tarinfo.size, BLOCKSIZE)
1499 if remainder > 0:
1500 self.fileobj.write(NUL * (BLOCKSIZE - remainder))
1501 blocks += 1
1502 self.offset += blocks * BLOCKSIZE
1503
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001504 self.members.append(tarinfo)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001505
Martin v. Löwis00a73e72005-03-04 19:40:34 +00001506 def extractall(self, path=".", members=None):
1507 """Extract all members from the archive to the current working
1508 directory and set owner, modification time and permissions on
1509 directories afterwards. `path' specifies a different directory
1510 to extract to. `members' is optional and must be a subset of the
1511 list returned by getmembers().
1512 """
1513 directories = []
1514
1515 if members is None:
1516 members = self
1517
1518 for tarinfo in members:
1519 if tarinfo.isdir():
1520 # Extract directory with a safe mode, so that
1521 # all files below can be extracted as well.
1522 try:
1523 os.makedirs(os.path.join(path, tarinfo.name), 0777)
1524 except EnvironmentError:
1525 pass
1526 directories.append(tarinfo)
1527 else:
1528 self.extract(tarinfo, path)
1529
1530 # Reverse sort directories.
1531 directories.sort(lambda a, b: cmp(a.name, b.name))
1532 directories.reverse()
1533
1534 # Set correct owner, mtime and filemode on directories.
1535 for tarinfo in directories:
1536 path = os.path.join(path, tarinfo.name)
1537 try:
1538 self.chown(tarinfo, path)
1539 self.utime(tarinfo, path)
1540 self.chmod(tarinfo, path)
1541 except ExtractError, e:
1542 if self.errorlevel > 1:
1543 raise
1544 else:
1545 self._dbg(1, "tarfile: %s" % e)
1546
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001547 def extract(self, member, path=""):
1548 """Extract a member from the archive to the current working directory,
1549 using its full name. Its file information is extracted as accurately
1550 as possible. `member' may be a filename or a TarInfo object. You can
1551 specify a different directory using `path'.
1552 """
1553 self._check("r")
1554
1555 if isinstance(member, TarInfo):
1556 tarinfo = member
1557 else:
1558 tarinfo = self.getmember(member)
1559
Neal Norwitza4f651a2004-07-20 22:07:44 +00001560 # Prepare the link target for makelink().
1561 if tarinfo.islnk():
1562 tarinfo._link_target = os.path.join(path, tarinfo.linkname)
1563
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001564 try:
1565 self._extract_member(tarinfo, os.path.join(path, tarinfo.name))
1566 except EnvironmentError, e:
1567 if self.errorlevel > 0:
1568 raise
1569 else:
1570 if e.filename is None:
1571 self._dbg(1, "tarfile: %s" % e.strerror)
1572 else:
1573 self._dbg(1, "tarfile: %s %r" % (e.strerror, e.filename))
1574 except ExtractError, e:
1575 if self.errorlevel > 1:
1576 raise
1577 else:
1578 self._dbg(1, "tarfile: %s" % e)
1579
1580 def extractfile(self, member):
1581 """Extract a member from the archive as a file object. `member' may be
1582 a filename or a TarInfo object. If `member' is a regular file, a
1583 file-like object is returned. If `member' is a link, a file-like
1584 object is constructed from the link's target. If `member' is none of
1585 the above, None is returned.
1586 The file-like object is read-only and provides the following
1587 methods: read(), readline(), readlines(), seek() and tell()
1588 """
1589 self._check("r")
1590
1591 if isinstance(member, TarInfo):
1592 tarinfo = member
1593 else:
1594 tarinfo = self.getmember(member)
1595
1596 if tarinfo.isreg():
1597 return self.fileobject(self, tarinfo)
1598
1599 elif tarinfo.type not in SUPPORTED_TYPES:
1600 # If a member's type is unknown, it is treated as a
1601 # regular file.
1602 return self.fileobject(self, tarinfo)
1603
1604 elif tarinfo.islnk() or tarinfo.issym():
1605 if isinstance(self.fileobj, _Stream):
1606 # A small but ugly workaround for the case that someone tries
1607 # to extract a (sym)link as a file-object from a non-seekable
1608 # stream of tar blocks.
Georg Brandle4751e32006-05-18 06:11:19 +00001609 raise StreamError("cannot extract (sym)link as file object")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001610 else:
Georg Brandl7eb4b7d2005-07-22 21:49:32 +00001611 # A (sym)link's file object is its target's file object.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001612 return self.extractfile(self._getmember(tarinfo.linkname,
1613 tarinfo))
1614 else:
1615 # If there's no data associated with the member (directory, chrdev,
1616 # blkdev, etc.), return None instead of a file object.
1617 return None
1618
1619 def _extract_member(self, tarinfo, targetpath):
1620 """Extract the TarInfo object tarinfo to a physical
1621 file called targetpath.
1622 """
1623 # Fetch the TarInfo object for the given name
1624 # and build the destination pathname, replacing
1625 # forward slashes to platform specific separators.
1626 if targetpath[-1:] == "/":
1627 targetpath = targetpath[:-1]
1628 targetpath = os.path.normpath(targetpath)
1629
1630 # Create all upper directories.
1631 upperdirs = os.path.dirname(targetpath)
1632 if upperdirs and not os.path.exists(upperdirs):
1633 ti = TarInfo()
1634 ti.name = upperdirs
1635 ti.type = DIRTYPE
1636 ti.mode = 0777
1637 ti.mtime = tarinfo.mtime
1638 ti.uid = tarinfo.uid
1639 ti.gid = tarinfo.gid
1640 ti.uname = tarinfo.uname
1641 ti.gname = tarinfo.gname
1642 try:
1643 self._extract_member(ti, ti.name)
1644 except:
1645 pass
1646
1647 if tarinfo.islnk() or tarinfo.issym():
1648 self._dbg(1, "%s -> %s" % (tarinfo.name, tarinfo.linkname))
1649 else:
1650 self._dbg(1, tarinfo.name)
1651
1652 if tarinfo.isreg():
1653 self.makefile(tarinfo, targetpath)
1654 elif tarinfo.isdir():
1655 self.makedir(tarinfo, targetpath)
1656 elif tarinfo.isfifo():
1657 self.makefifo(tarinfo, targetpath)
1658 elif tarinfo.ischr() or tarinfo.isblk():
1659 self.makedev(tarinfo, targetpath)
1660 elif tarinfo.islnk() or tarinfo.issym():
1661 self.makelink(tarinfo, targetpath)
1662 elif tarinfo.type not in SUPPORTED_TYPES:
1663 self.makeunknown(tarinfo, targetpath)
1664 else:
1665 self.makefile(tarinfo, targetpath)
1666
1667 self.chown(tarinfo, targetpath)
1668 if not tarinfo.issym():
1669 self.chmod(tarinfo, targetpath)
1670 self.utime(tarinfo, targetpath)
1671
1672 #--------------------------------------------------------------------------
1673 # Below are the different file methods. They are called via
1674 # _extract_member() when extract() is called. They can be replaced in a
1675 # subclass to implement other functionality.
1676
1677 def makedir(self, tarinfo, targetpath):
1678 """Make a directory called targetpath.
1679 """
1680 try:
1681 os.mkdir(targetpath)
1682 except EnvironmentError, e:
1683 if e.errno != errno.EEXIST:
1684 raise
1685
1686 def makefile(self, tarinfo, targetpath):
1687 """Make a file called targetpath.
1688 """
1689 source = self.extractfile(tarinfo)
1690 target = file(targetpath, "wb")
1691 copyfileobj(source, target)
1692 source.close()
1693 target.close()
1694
1695 def makeunknown(self, tarinfo, targetpath):
1696 """Make a file from a TarInfo object with an unknown type
1697 at targetpath.
1698 """
1699 self.makefile(tarinfo, targetpath)
1700 self._dbg(1, "tarfile: Unknown file type %r, " \
1701 "extracted as regular file." % tarinfo.type)
1702
1703 def makefifo(self, tarinfo, targetpath):
1704 """Make a fifo called targetpath.
1705 """
1706 if hasattr(os, "mkfifo"):
1707 os.mkfifo(targetpath)
1708 else:
Georg Brandle4751e32006-05-18 06:11:19 +00001709 raise ExtractError("fifo not supported by system")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001710
1711 def makedev(self, tarinfo, targetpath):
1712 """Make a character or block device called targetpath.
1713 """
1714 if not hasattr(os, "mknod") or not hasattr(os, "makedev"):
Georg Brandle4751e32006-05-18 06:11:19 +00001715 raise ExtractError("special devices not supported by system")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001716
1717 mode = tarinfo.mode
1718 if tarinfo.isblk():
1719 mode |= stat.S_IFBLK
1720 else:
1721 mode |= stat.S_IFCHR
1722
1723 os.mknod(targetpath, mode,
1724 os.makedev(tarinfo.devmajor, tarinfo.devminor))
1725
1726 def makelink(self, tarinfo, targetpath):
1727 """Make a (symbolic) link called targetpath. If it cannot be created
1728 (platform limitation), we try to make a copy of the referenced file
1729 instead of a link.
1730 """
1731 linkpath = tarinfo.linkname
1732 try:
1733 if tarinfo.issym():
1734 os.symlink(linkpath, targetpath)
1735 else:
Neal Norwitza4f651a2004-07-20 22:07:44 +00001736 # See extract().
1737 os.link(tarinfo._link_target, targetpath)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001738 except AttributeError:
1739 if tarinfo.issym():
1740 linkpath = os.path.join(os.path.dirname(tarinfo.name),
1741 linkpath)
1742 linkpath = normpath(linkpath)
1743
1744 try:
1745 self._extract_member(self.getmember(linkpath), targetpath)
1746 except (EnvironmentError, KeyError), e:
1747 linkpath = os.path.normpath(linkpath)
1748 try:
1749 shutil.copy2(linkpath, targetpath)
1750 except EnvironmentError, e:
Georg Brandle4751e32006-05-18 06:11:19 +00001751 raise IOError("link could not be created")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001752
1753 def chown(self, tarinfo, targetpath):
1754 """Set owner of targetpath according to tarinfo.
1755 """
1756 if pwd and hasattr(os, "geteuid") and os.geteuid() == 0:
1757 # We have to be root to do so.
1758 try:
1759 g = grp.getgrnam(tarinfo.gname)[2]
1760 except KeyError:
1761 try:
1762 g = grp.getgrgid(tarinfo.gid)[2]
1763 except KeyError:
1764 g = os.getgid()
1765 try:
1766 u = pwd.getpwnam(tarinfo.uname)[2]
1767 except KeyError:
1768 try:
1769 u = pwd.getpwuid(tarinfo.uid)[2]
1770 except KeyError:
1771 u = os.getuid()
1772 try:
1773 if tarinfo.issym() and hasattr(os, "lchown"):
1774 os.lchown(targetpath, u, g)
1775 else:
Andrew MacIntyre7970d202003-02-19 12:51:34 +00001776 if sys.platform != "os2emx":
1777 os.chown(targetpath, u, g)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001778 except EnvironmentError, e:
Georg Brandle4751e32006-05-18 06:11:19 +00001779 raise ExtractError("could not change owner")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001780
1781 def chmod(self, tarinfo, targetpath):
1782 """Set file permissions of targetpath according to tarinfo.
1783 """
Jack Jansen834eff62003-03-07 12:47:06 +00001784 if hasattr(os, 'chmod'):
1785 try:
1786 os.chmod(targetpath, tarinfo.mode)
1787 except EnvironmentError, e:
Georg Brandle4751e32006-05-18 06:11:19 +00001788 raise ExtractError("could not change mode")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001789
1790 def utime(self, tarinfo, targetpath):
1791 """Set modification time of targetpath according to tarinfo.
1792 """
Jack Jansen834eff62003-03-07 12:47:06 +00001793 if not hasattr(os, 'utime'):
Tim Petersf9347782003-03-07 15:36:41 +00001794 return
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001795 if sys.platform == "win32" and tarinfo.isdir():
1796 # According to msdn.microsoft.com, it is an error (EACCES)
1797 # to use utime() on directories.
1798 return
1799 try:
1800 os.utime(targetpath, (tarinfo.mtime, tarinfo.mtime))
1801 except EnvironmentError, e:
Georg Brandle4751e32006-05-18 06:11:19 +00001802 raise ExtractError("could not change modification time")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001803
1804 #--------------------------------------------------------------------------
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001805 def next(self):
1806 """Return the next member of the archive as a TarInfo object, when
1807 TarFile is opened for reading. Return None if there is no more
1808 available.
1809 """
1810 self._check("ra")
1811 if self.firstmember is not None:
1812 m = self.firstmember
1813 self.firstmember = None
1814 return m
1815
1816 # Read the next block.
Andrew M. Kuchling864bba12004-07-10 22:02:11 +00001817 self.fileobj.seek(self.offset)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001818 while True:
1819 buf = self.fileobj.read(BLOCKSIZE)
1820 if not buf:
1821 return None
Georg Brandl38c6a222006-05-10 16:26:03 +00001822
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001823 try:
1824 tarinfo = TarInfo.frombuf(buf)
Tim Peters8a299d22006-05-19 19:16:34 +00001825
Georg Brandl38c6a222006-05-10 16:26:03 +00001826 # Set the TarInfo object's offset to the current position of the
1827 # TarFile and set self.offset to the position where the data blocks
1828 # should begin.
1829 tarinfo.offset = self.offset
1830 self.offset += BLOCKSIZE
1831
1832 tarinfo = self.proc_member(tarinfo)
1833
Georg Brandlebbeed72006-12-19 22:06:46 +00001834 except HeaderError, e:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001835 if self.ignore_zeros:
Georg Brandlebbeed72006-12-19 22:06:46 +00001836 self._dbg(2, "0x%X: %s" % (self.offset, e))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001837 self.offset += BLOCKSIZE
1838 continue
1839 else:
Andrew M. Kuchling864bba12004-07-10 22:02:11 +00001840 if self.offset == 0:
Georg Brandlebbeed72006-12-19 22:06:46 +00001841 raise ReadError(str(e))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001842 return None
1843 break
1844
Georg Brandl38c6a222006-05-10 16:26:03 +00001845 # Some old tar programs represent a directory as a regular
1846 # file with a trailing slash.
1847 if tarinfo.isreg() and tarinfo.name.endswith("/"):
1848 tarinfo.type = DIRTYPE
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001849
Georg Brandl38c6a222006-05-10 16:26:03 +00001850 # Directory names should have a '/' at the end.
1851 if tarinfo.isdir():
1852 tarinfo.name += "/"
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001853
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001854 self.members.append(tarinfo)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001855 return tarinfo
1856
1857 #--------------------------------------------------------------------------
Georg Brandl38c6a222006-05-10 16:26:03 +00001858 # The following are methods that are called depending on the type of a
1859 # member. The entry point is proc_member() which is called with a TarInfo
1860 # object created from the header block from the current offset. The
1861 # proc_member() method can be overridden in a subclass to add custom
1862 # proc_*() methods. A proc_*() method MUST implement the following
1863 # operations:
1864 # 1. Set tarinfo.offset_data to the position where the data blocks begin,
1865 # if there is data that follows.
1866 # 2. Set self.offset to the position where the next member's header will
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001867 # begin.
Georg Brandl38c6a222006-05-10 16:26:03 +00001868 # 3. Return tarinfo or another valid TarInfo object.
1869 def proc_member(self, tarinfo):
1870 """Choose the right processing method for tarinfo depending
1871 on its type and call it.
1872 """
1873 if tarinfo.type in (GNUTYPE_LONGNAME, GNUTYPE_LONGLINK):
1874 return self.proc_gnulong(tarinfo)
1875 elif tarinfo.type == GNUTYPE_SPARSE:
1876 return self.proc_sparse(tarinfo)
1877 else:
1878 return self.proc_builtin(tarinfo)
1879
1880 def proc_builtin(self, tarinfo):
1881 """Process a builtin type member or an unknown member
1882 which will be treated as a regular file.
1883 """
1884 tarinfo.offset_data = self.offset
1885 if tarinfo.isreg() or tarinfo.type not in SUPPORTED_TYPES:
1886 # Skip the following data blocks.
1887 self.offset += self._block(tarinfo.size)
1888 return tarinfo
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001889
1890 def proc_gnulong(self, tarinfo):
Georg Brandl38c6a222006-05-10 16:26:03 +00001891 """Process the blocks that hold a GNU longname
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001892 or longlink member.
1893 """
1894 buf = ""
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001895 count = tarinfo.size
1896 while count > 0:
1897 block = self.fileobj.read(BLOCKSIZE)
1898 buf += block
1899 self.offset += BLOCKSIZE
1900 count -= BLOCKSIZE
1901
Georg Brandl38c6a222006-05-10 16:26:03 +00001902 # Fetch the next header and process it.
1903 b = self.fileobj.read(BLOCKSIZE)
1904 t = TarInfo.frombuf(b)
1905 t.offset = self.offset
1906 self.offset += BLOCKSIZE
1907 next = self.proc_member(t)
Andrew M. Kuchling864bba12004-07-10 22:02:11 +00001908
Georg Brandl38c6a222006-05-10 16:26:03 +00001909 # Patch the TarInfo object from the next header with
1910 # the longname information.
Andrew M. Kuchling864bba12004-07-10 22:02:11 +00001911 next.offset = tarinfo.offset
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001912 if tarinfo.type == GNUTYPE_LONGNAME:
Georg Brandle8953182006-05-27 14:02:03 +00001913 next.name = buf.rstrip(NUL)
Andrew M. Kuchling864bba12004-07-10 22:02:11 +00001914 elif tarinfo.type == GNUTYPE_LONGLINK:
Georg Brandle8953182006-05-27 14:02:03 +00001915 next.linkname = buf.rstrip(NUL)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001916
Andrew M. Kuchling864bba12004-07-10 22:02:11 +00001917 return next
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001918
1919 def proc_sparse(self, tarinfo):
Georg Brandl38c6a222006-05-10 16:26:03 +00001920 """Process a GNU sparse header plus extra headers.
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001921 """
Georg Brandl38c6a222006-05-10 16:26:03 +00001922 buf = tarinfo.buf
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001923 sp = _ringbuffer()
1924 pos = 386
1925 lastpos = 0L
1926 realpos = 0L
1927 # There are 4 possible sparse structs in the
1928 # first header.
1929 for i in xrange(4):
1930 try:
Georg Brandl38c6a222006-05-10 16:26:03 +00001931 offset = nti(buf[pos:pos + 12])
1932 numbytes = nti(buf[pos + 12:pos + 24])
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001933 except ValueError:
1934 break
1935 if offset > lastpos:
1936 sp.append(_hole(lastpos, offset - lastpos))
1937 sp.append(_data(offset, numbytes, realpos))
1938 realpos += numbytes
1939 lastpos = offset + numbytes
1940 pos += 24
1941
1942 isextended = ord(buf[482])
Georg Brandl38c6a222006-05-10 16:26:03 +00001943 origsize = nti(buf[483:495])
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001944
1945 # If the isextended flag is given,
1946 # there are extra headers to process.
1947 while isextended == 1:
1948 buf = self.fileobj.read(BLOCKSIZE)
1949 self.offset += BLOCKSIZE
1950 pos = 0
1951 for i in xrange(21):
1952 try:
Georg Brandl38c6a222006-05-10 16:26:03 +00001953 offset = nti(buf[pos:pos + 12])
1954 numbytes = nti(buf[pos + 12:pos + 24])
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001955 except ValueError:
1956 break
1957 if offset > lastpos:
1958 sp.append(_hole(lastpos, offset - lastpos))
1959 sp.append(_data(offset, numbytes, realpos))
1960 realpos += numbytes
1961 lastpos = offset + numbytes
1962 pos += 24
1963 isextended = ord(buf[504])
1964
1965 if lastpos < origsize:
1966 sp.append(_hole(lastpos, origsize - lastpos))
1967
1968 tarinfo.sparse = sp
1969
1970 tarinfo.offset_data = self.offset
1971 self.offset += self._block(tarinfo.size)
1972 tarinfo.size = origsize
Andrew M. Kuchling864bba12004-07-10 22:02:11 +00001973
Georg Brandl38c6a222006-05-10 16:26:03 +00001974 return tarinfo
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001975
1976 #--------------------------------------------------------------------------
1977 # Little helper methods:
1978
1979 def _block(self, count):
1980 """Round up a byte count by BLOCKSIZE and return it,
1981 e.g. _block(834) => 1024.
1982 """
1983 blocks, remainder = divmod(count, BLOCKSIZE)
1984 if remainder:
1985 blocks += 1
1986 return blocks * BLOCKSIZE
1987
1988 def _getmember(self, name, tarinfo=None):
1989 """Find an archive member by name from bottom to top.
1990 If tarinfo is given, it is used as the starting point.
1991 """
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001992 # Ensure that all members have been loaded.
1993 members = self.getmembers()
1994
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001995 if tarinfo is None:
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001996 end = len(members)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001997 else:
Martin v. Löwisf3c56112004-09-18 09:08:52 +00001998 end = members.index(tarinfo)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00001999
2000 for i in xrange(end - 1, -1, -1):
Martin v. Löwisf3c56112004-09-18 09:08:52 +00002001 if name == members[i].name:
2002 return members[i]
Andrew M. Kuchling864bba12004-07-10 22:02:11 +00002003
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002004 def _load(self):
2005 """Read through the entire archive file and look for readable
2006 members.
2007 """
2008 while True:
2009 tarinfo = self.next()
2010 if tarinfo is None:
2011 break
2012 self._loaded = True
2013
2014 def _check(self, mode=None):
2015 """Check if TarFile is still open, and if the operation's mode
2016 corresponds to TarFile's mode.
2017 """
2018 if self.closed:
Georg Brandle4751e32006-05-18 06:11:19 +00002019 raise IOError("%s is closed" % self.__class__.__name__)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002020 if mode is not None and self._mode not in mode:
Georg Brandle4751e32006-05-18 06:11:19 +00002021 raise IOError("bad operation for mode %r" % self._mode)
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002022
2023 def __iter__(self):
2024 """Provide an iterator object.
2025 """
2026 if self._loaded:
2027 return iter(self.members)
2028 else:
2029 return TarIter(self)
2030
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002031 def _dbg(self, level, msg):
2032 """Write debugging output to sys.stderr.
2033 """
2034 if level <= self.debug:
2035 print >> sys.stderr, msg
2036# class TarFile
2037
2038class TarIter:
2039 """Iterator Class.
2040
2041 for tarinfo in TarFile(...):
2042 suite...
2043 """
2044
2045 def __init__(self, tarfile):
2046 """Construct a TarIter object.
2047 """
2048 self.tarfile = tarfile
Martin v. Löwis637431b2005-03-03 23:12:42 +00002049 self.index = 0
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002050 def __iter__(self):
2051 """Return iterator object.
2052 """
2053 return self
2054 def next(self):
2055 """Return the next item using TarFile's next() method.
2056 When all members have been read, set TarFile as _loaded.
2057 """
Martin v. Löwis637431b2005-03-03 23:12:42 +00002058 # Fix for SF #1100429: Under rare circumstances it can
2059 # happen that getmembers() is called during iteration,
2060 # which will cause TarIter to stop prematurely.
2061 if not self.tarfile._loaded:
2062 tarinfo = self.tarfile.next()
2063 if not tarinfo:
2064 self.tarfile._loaded = True
2065 raise StopIteration
2066 else:
2067 try:
2068 tarinfo = self.tarfile.members[self.index]
2069 except IndexError:
2070 raise StopIteration
2071 self.index += 1
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002072 return tarinfo
2073
2074# Helper classes for sparse file support
2075class _section:
2076 """Base class for _data and _hole.
2077 """
2078 def __init__(self, offset, size):
2079 self.offset = offset
2080 self.size = size
2081 def __contains__(self, offset):
2082 return self.offset <= offset < self.offset + self.size
2083
2084class _data(_section):
2085 """Represent a data section in a sparse file.
2086 """
2087 def __init__(self, offset, size, realpos):
2088 _section.__init__(self, offset, size)
2089 self.realpos = realpos
2090
2091class _hole(_section):
2092 """Represent a hole section in a sparse file.
2093 """
2094 pass
2095
2096class _ringbuffer(list):
2097 """Ringbuffer class which increases performance
2098 over a regular list.
2099 """
2100 def __init__(self):
2101 self.idx = 0
2102 def find(self, offset):
2103 idx = self.idx
2104 while True:
2105 item = self[idx]
2106 if offset in item:
2107 break
2108 idx += 1
2109 if idx == len(self):
2110 idx = 0
2111 if idx == self.idx:
2112 # End of File
2113 return None
2114 self.idx = idx
2115 return item
2116
2117#---------------------------------------------
2118# zipfile compatible TarFile class
2119#---------------------------------------------
2120TAR_PLAIN = 0 # zipfile.ZIP_STORED
2121TAR_GZIPPED = 8 # zipfile.ZIP_DEFLATED
2122class TarFileCompat:
2123 """TarFile class compatible with standard module zipfile's
2124 ZipFile class.
2125 """
2126 def __init__(self, file, mode="r", compression=TAR_PLAIN):
2127 if compression == TAR_PLAIN:
2128 self.tarfile = TarFile.taropen(file, mode)
2129 elif compression == TAR_GZIPPED:
2130 self.tarfile = TarFile.gzopen(file, mode)
2131 else:
Georg Brandle4751e32006-05-18 06:11:19 +00002132 raise ValueError("unknown compression constant")
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002133 if mode[0:1] == "r":
2134 members = self.tarfile.getmembers()
Raymond Hettingera1d09e22005-09-11 16:34:05 +00002135 for m in members:
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002136 m.filename = m.name
2137 m.file_size = m.size
2138 m.date_time = time.gmtime(m.mtime)[:6]
2139 def namelist(self):
2140 return map(lambda m: m.name, self.infolist())
2141 def infolist(self):
2142 return filter(lambda m: m.type in REGULAR_TYPES,
2143 self.tarfile.getmembers())
2144 def printdir(self):
2145 self.tarfile.list()
2146 def testzip(self):
2147 return
2148 def getinfo(self, name):
2149 return self.tarfile.getmember(name)
2150 def read(self, name):
2151 return self.tarfile.extractfile(self.tarfile.getmember(name)).read()
2152 def write(self, filename, arcname=None, compress_type=None):
2153 self.tarfile.add(filename, arcname)
2154 def writestr(self, zinfo, bytes):
Raymond Hettingera6172712004-12-31 19:15:26 +00002155 try:
2156 from cStringIO import StringIO
2157 except ImportError:
2158 from StringIO import StringIO
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002159 import calendar
2160 zinfo.name = zinfo.filename
2161 zinfo.size = zinfo.file_size
2162 zinfo.mtime = calendar.timegm(zinfo.date_time)
Raymond Hettingera6172712004-12-31 19:15:26 +00002163 self.tarfile.addfile(zinfo, StringIO(bytes))
Neal Norwitzb9ef4ae2003-01-05 23:19:43 +00002164 def close(self):
2165 self.tarfile.close()
2166#class TarFileCompat
2167
2168#--------------------
2169# exported functions
2170#--------------------
2171def is_tarfile(name):
2172 """Return True if name points to a tar archive that we
2173 are able to handle, else return False.
2174 """
2175 try:
2176 t = open(name)
2177 t.close()
2178 return True
2179 except TarError:
2180 return False
2181
2182open = TarFile.open