/usr/lib/python2.7/dist-packages/libarchive/write.py is in python-libarchive-c 2.1-3.1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 | # This file is part of a program licensed under the terms of the GNU Lesser
# General Public License version 2 (or at your option any later version)
# as published by the Free Software Foundation: http://www.gnu.org/licenses/
from __future__ import division, print_function, unicode_literals
from contextlib import contextmanager
from ctypes import byref, cast, c_char, c_size_t, c_void_p, POINTER
from . import ffi
from .entry import ArchiveEntry, new_archive_entry
from .ffi import (
OPEN_CALLBACK, WRITE_CALLBACK, CLOSE_CALLBACK, VOID_CB,
ARCHIVE_EOF, page_size,
entry_sourcepath, entry_clear,
read_disk_new, read_disk_open_w, read_next_header2, read_disk_descend,
read_free, write_header, write_data, write_finish_entry,
)
@contextmanager
def new_archive_read_disk(path):
archive_p = read_disk_new()
read_disk_open_w(archive_p, path)
try:
yield archive_p
finally:
read_free(archive_p)
class ArchiveWrite(object):
def __init__(self, archive_p):
self._pointer = archive_p
def add_entries(self, entries):
"""Add the given entries to the archive.
"""
write_p = self._pointer
for entry in entries:
write_header(write_p, entry._entry_p)
for block in entry.get_blocks():
write_data(write_p, block, len(block))
write_finish_entry(write_p)
def add_files(self, *paths):
"""Read the given paths from disk and add them to the archive.
"""
write_p = self._pointer
block_size = ffi.write_get_bytes_per_block(write_p)
if block_size <= 0:
block_size = 10240 # pragma: no cover
with new_archive_entry() as entry_p:
entry = ArchiveEntry(None, entry_p)
for path in paths:
with new_archive_read_disk(path) as read_p:
while 1:
r = read_next_header2(read_p, entry_p)
if r == ARCHIVE_EOF:
break
entry.pathname = entry.pathname.lstrip('/')
read_disk_descend(read_p)
write_header(write_p, entry_p)
try:
with open(entry_sourcepath(entry_p), 'rb') as f:
while 1:
data = f.read(block_size)
if not data:
break
write_data(write_p, data, len(data))
except IOError as e:
if e.errno != 21:
raise # pragma: no cover
write_finish_entry(write_p)
entry_clear(entry_p)
@contextmanager
def new_archive_write(format_name, filter_name=None):
archive_p = ffi.write_new()
getattr(ffi, 'write_set_format_'+format_name)(archive_p)
if filter_name:
getattr(ffi, 'write_add_filter_'+filter_name)(archive_p)
try:
yield archive_p
finally:
ffi.write_close(archive_p)
ffi.write_free(archive_p)
@contextmanager
def custom_writer(write_func, format_name, filter_name=None,
open_func=VOID_CB, close_func=VOID_CB, block_size=page_size):
def write_cb_internal(archive_p, context, buffer_, length):
data = cast(buffer_, POINTER(c_char * length))[0]
return write_func(data)
open_cb = OPEN_CALLBACK(open_func)
write_cb = WRITE_CALLBACK(write_cb_internal)
close_cb = CLOSE_CALLBACK(close_func)
with new_archive_write(format_name, filter_name) as archive_p:
ffi.write_set_bytes_in_last_block(archive_p, 1)
ffi.write_set_bytes_per_block(archive_p, block_size)
ffi.write_open(archive_p, None, open_cb, write_cb, close_cb)
yield ArchiveWrite(archive_p)
@contextmanager
def fd_writer(fd, format_name, filter_name=None):
with new_archive_write(format_name, filter_name) as archive_p:
ffi.write_open_fd(archive_p, fd)
yield ArchiveWrite(archive_p)
@contextmanager
def file_writer(filepath, format_name, filter_name=None):
with new_archive_write(format_name, filter_name) as archive_p:
ffi.write_open_filename_w(archive_p, filepath)
yield ArchiveWrite(archive_p)
@contextmanager
def memory_writer(buf, format_name, filter_name=None):
with new_archive_write(format_name, filter_name) as archive_p:
used = byref(c_size_t())
buf_p = cast(buf, c_void_p)
ffi.write_open_memory(archive_p, buf_p, len(buf), used)
yield ArchiveWrite(archive_p)
|