test/py/tests/test_fs/test_squashfs/sqfs_common.py

   1 # SPDX-License-Identifier: GPL-2.0
   2 # Copyright (C) 2020 Bootlin
   3 # Author: Joao Marcos Costa <joaomarcos.costa@bootlin.com>
   4
   5 import os
   6 import random
   7 import string
   8
   9 def sqfs_get_random_letters(size):
  10     letters = []
  11     for i in range(0, size):
  12             letters.append(random.choice(string.ascii_letters))
  13
  14     return ''.join(letters)
  15
  16 def sqfs_generate_file(path, size):
  17     content = sqfs_get_random_letters(size)
  18     file = open(path, "w")
  19     file.write(content)
  20     file.close()
  21
  22 # generate image with three files and a symbolic link
  23 def sqfs_generate_image(cons):
  24     src = os.path.join(cons.config.build_dir, "sqfs_src/")
  25     dest = os.path.join(cons.config.build_dir, "sqfs")
  26     os.mkdir(src)
  27     sqfs_generate_file(src + "frag_only", 100)
  28     sqfs_generate_file(src + "blks_frag", 5100)
  29     sqfs_generate_file(src + "blks_only", 4096)
  30     os.symlink("frag_only", src + "sym")
  31     os.system("mksquashfs " + src + " " + dest + " -b 4096 -always-use-fragments")
  32
  33 # removes all files created by sqfs_generate_image()
  34 def sqfs_clean(cons):
  35     src = os.path.join(cons.config.build_dir, "sqfs_src/")
  36     dest = os.path.join(cons.config.build_dir, "sqfs")
  37     os.remove(src + "frag_only")
  38     os.remove(src + "blks_frag")
  39     os.remove(src + "blks_only")
  40     os.remove(src + "sym")
  41     os.rmdir(src)
  42     os.remove(dest)