def write_file(volume,
file,
split_before,
split_after,
name,
is_unicode,
dictsize,
host_os,
attr,
accum_crc,
dostime,
xtime=None,
size=None,
pack_size=None):
size_64 = size_64_encode(pack_size, size)
header_size = file_hdr_size(name, xtime, size_64)
volume.seek(+header_size, io.SEEK_CUR)
left = pack_size
crc = 0 if split_after else accum_crc
while left > 0:
chunk = file.read(min(FILE_COPY_CRC_BUF, left))
left -= len(chunk)
volume.write(chunk)
crc = crc32(chunk, crc)
if split_after:
accum_crc = crc32(chunk, accum_crc)
parts = list()
flags = (RAR_LONG_BLOCK ^ split_before * RAR_FILE_SPLIT_BEFORE
^ split_after * RAR_FILE_SPLIT_AFTER ^ dictsize
^ is_unicode * RAR_FILE_UNICODE)
parts.append(
S_FILE_HDR.pack(pack_size & bitmask(32), size & bitmask(32), host_os,
crc, dostime, 20, ord("0"), len(name), attr))
if size_64 is not None:
flags ^= RAR_FILE_LARGE
parts.append(size_64)
parts.append(name)
if xtime is not None:
flags ^= RAR_FILE_EXTTIME
parts.append(xtime)
volume.seek(-pack_size - header_size, io.SEEK_CUR)
write_block(volume, RAR_BLOCK_FILE, flags, parts)
volume.seek(+pack_size, io.SEEK_CUR)
if split_after: return accum_crc
def write_file(volume, file, split_before, split_after, name, is_unicode,
dict, host_os, attr, accum_crc, dostime, xtime=None, size=None,
pack_size=None):
size_64 = size_64_encode(pack_size, size)
header_size = file_hdr_size(name, xtime, size_64)
volume.seek(+header_size, io.SEEK_CUR)
left = pack_size
crc = 0 if split_after else accum_crc
while left > 0:
chunk = file.read(min(FILE_COPY_CRC_BUF, left))
left -= len(chunk)
volume.write(chunk)
crc = crc32(chunk, crc)
if split_after:
accum_crc = crc32(chunk, accum_crc)
parts = list()
flags = (RAR_LONG_BLOCK ^ split_before * RAR_FILE_SPLIT_BEFORE ^
split_after * RAR_FILE_SPLIT_AFTER ^ dict ^
is_unicode * RAR_FILE_UNICODE)
parts.append(S_FILE_HDR.pack(
pack_size & bitmask(32), size & bitmask(32),
host_os, crc, dostime, 20, ord("0"), len(name), attr
))
if size_64 is not None:
flags ^= RAR_FILE_LARGE
parts.append(size_64)
parts.append(name)
if xtime is not None:
flags ^= RAR_FILE_EXTTIME
parts.append(xtime)
volume.seek(-pack_size - header_size, io.SEEK_CUR)
write_block(volume, RAR_BLOCK_FILE, flags, parts)
volume.seek(+pack_size, io.SEEK_CUR)
if split_after: return accum_crc
def write_rr(version, host_os, volume, rr_count):
prot_size = volume.tell()
(rr_crcs, rr_sects) = rr_calc(volume, rr_count, prot_size)
crc = crc32(rr_crcs, RR_CRC_INIT)
for s in rr_sects:
crc = crc32(s.buffer(), crc)
prot_sect_count = len(rr_crcs) // S_SHORT.size
size = prot_sect_count * RR_CRC_SIZE + rr_count * RR_SECT_SIZE
if version < 3:
write_block(volume,
type=RAR_BLOCK_OLD_RECOVERY,
flags=RAR_LONG_BLOCK ^ RAR_SKIP_IF_UNKNOWN,
data=(
S_LONG.pack(size),
(20,),
S_SHORT.pack(rr_count),
S_LONG.pack(prot_sect_count),
RR_PROTECT_2,
))
else:
write_block(volume,
type=RAR_BLOCK_SUB,
flags=RAR_LONG_BLOCK ^ RAR_SKIP_IF_UNKNOWN,
data=(
S_FILE_HDR.pack(size, size, host_os, crc, 0, 29, ord("0"),
len(RR_SUB_NAME), 0),
RR_SUB_NAME,
RR_PROTECT_3,
S_LONG.pack(rr_count),
struct.pack("<Q", prot_sect_count),
))
volume.write(rr_crcs)
for s in rr_sects:
volume.write(s.buffer())
def write_rr(version, host_os, volume, rr_count):
prot_size = volume.tell()
(rr_crcs, rr_sects) = rr_calc(volume, rr_count, prot_size)
crc = crc32(rr_crcs, RR_CRC_INIT)
for s in rr_sects:
crc = crc32(s.buffer(), crc)
prot_sect_count = len(rr_crcs) // S_SHORT.size
size = prot_sect_count * RR_CRC_SIZE + rr_count * RR_SECT_SIZE
if version < 3:
write_block(volume,
type=RAR_BLOCK_OLD_RECOVERY,
flags=RAR_LONG_BLOCK ^ RAR_SKIP_IF_UNKNOWN,
data=(
S_LONG.pack(size),
(20, ),
S_SHORT.pack(rr_count),
S_LONG.pack(prot_sect_count),
RR_PROTECT_2,
))
else:
write_block(volume,
type=RAR_BLOCK_SUB,
flags=RAR_LONG_BLOCK ^ RAR_SKIP_IF_UNKNOWN,
data=(
S_FILE_HDR.pack(size, size, host_os, crc, 0, 29,
ord("0"), len(RR_SUB_NAME), 0),
RR_SUB_NAME,
RR_PROTECT_3,
S_LONG.pack(rr_count),
struct.pack("<Q", prot_sect_count),
))
volume.write(rr_crcs)
for s in rr_sects:
volume.write(s.buffer())