mirror of
https://github.com/Motorhead1991/qemu.git
synced 2025-07-27 12:23:53 -06:00
cfcb4484fc
More archive related code will be added shortly, so having a separate file makes more sense. The utils.py imports the functions from archive.py, so that existing callers don't need to be modified. This avoids redundant code churn until later in the series when all calls will be adapted for other reasons. Reviewed-by: Thomas Huth <thuth@redhat.com> Signed-off-by: Daniel P. Berrangé <berrange@redhat.com> Message-ID: <20241217155953.3950506-17-berrange@redhat.com> Signed-off-by: Thomas Huth <thuth@redhat.com>
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
# Utilities for python-based QEMU tests
|
|
#
|
|
# Copyright 2024 Red Hat, Inc.
|
|
#
|
|
# Authors:
|
|
# Thomas Huth <thuth@redhat.com>
|
|
#
|
|
# This work is licensed under the terms of the GNU GPL, version 2 or
|
|
# later. See the COPYING file in the top-level directory.
|
|
|
|
import gzip
|
|
import lzma
|
|
import os
|
|
import shutil
|
|
|
|
from .archive import tar_extract as archive_extract
|
|
from .archive import cpio_extract
|
|
|
|
"""
|
|
Round up to next power of 2
|
|
"""
|
|
def pow2ceil(x):
|
|
return 1 if x == 0 else 2**(x - 1).bit_length()
|
|
|
|
def file_truncate(path, size):
|
|
if size != os.path.getsize(path):
|
|
with open(path, 'ab+') as fd:
|
|
fd.truncate(size)
|
|
|
|
"""
|
|
Expand file size to next power of 2
|
|
"""
|
|
def image_pow2ceil_expand(path):
|
|
size = os.path.getsize(path)
|
|
size_aligned = pow2ceil(size)
|
|
if size != size_aligned:
|
|
with open(path, 'ab+') as fd:
|
|
fd.truncate(size_aligned)
|
|
|
|
def gzip_uncompress(gz_path, output_path):
|
|
if os.path.exists(output_path):
|
|
return
|
|
with gzip.open(gz_path, 'rb') as gz_in:
|
|
try:
|
|
with open(output_path, 'wb') as raw_out:
|
|
shutil.copyfileobj(gz_in, raw_out)
|
|
except:
|
|
os.remove(output_path)
|
|
raise
|
|
|
|
def lzma_uncompress(xz_path, output_path):
|
|
if os.path.exists(output_path):
|
|
return
|
|
with lzma.open(xz_path, 'rb') as lzma_in:
|
|
try:
|
|
with open(output_path, 'wb') as raw_out:
|
|
shutil.copyfileobj(lzma_in, raw_out)
|
|
except:
|
|
os.remove(output_path)
|
|
raise
|