165 lines
3.7 KiB
Python
Executable File
165 lines
3.7 KiB
Python
Executable File
import os
|
|
import sys
|
|
import stat
|
|
import errno
|
|
import functools as ft
|
|
|
|
# use "import libmount" for in a standard way installed python binding
|
|
import pylibmount as mnt
|
|
|
|
def usage(tss):
|
|
print("\nUsage:\n\t{:s} <test> [testoptions]\nTests:\n".format(sys.argv[0]))
|
|
for i in tss:
|
|
print("\t{15:-s}".format(i[0]))
|
|
if i[2] != "":
|
|
print(" {:s}\n".format(i[2]))
|
|
|
|
print("\n")
|
|
return 1
|
|
|
|
def mnt_run_test(tss, argv):
|
|
rc = -1
|
|
if ((len(argv) < 2) or (argv[1] == "--help") or (argv[1] == "-h")):
|
|
return usage(tss)
|
|
|
|
#mnt_init_debug(0)
|
|
|
|
i=()
|
|
for i in tss:
|
|
if i[0] == argv[1]:
|
|
rc = i[1](i, argv[1:])
|
|
if rc:
|
|
print("FAILED [rc={:d}]".format(rc))
|
|
break
|
|
|
|
if ((rc < 0) and (i == ())):
|
|
return usage(tss)
|
|
return not not rc #because !!rc is too mainstream for python
|
|
|
|
def parser_errcb(tb, fname, line):
|
|
print("{:s}:{:d}: parse error".format(fname, line))
|
|
return 1
|
|
|
|
def create_table(f, comments):
|
|
if not f:
|
|
return None
|
|
|
|
tb = mnt.Table()
|
|
tb.enable_comments(comments)
|
|
tb.errcb = parser_errcb
|
|
|
|
try:
|
|
tb.parse_file(f)
|
|
except Exception:
|
|
print("{:s}: parsing failed".format(f))
|
|
return None
|
|
return tb
|
|
|
|
def test_copy_fs(ts, argv):
|
|
rc = -1
|
|
tb = create_table(argv[1], False)
|
|
fs = tb.find_target("/", mnt.MNT_ITER_FORWARD)
|
|
if not fs:
|
|
return rc
|
|
|
|
print("ORIGINAL:")
|
|
fs.print_debug()
|
|
|
|
fs = fs.copy_fs(None)
|
|
if not fs:
|
|
return rc
|
|
print("COPY:")
|
|
fs.print_debug()
|
|
return 0
|
|
|
|
def test_parse(ts, argv):
|
|
parse_comments = False
|
|
|
|
if len(argv) == 3 and argv[2] == "--comments":
|
|
parse_comments = True
|
|
tb = create_table(argv[1], parse_comments)
|
|
|
|
if tb.intro_comment:
|
|
print("Initial comment:\n\"{:s}\"".format(tb.intro_comment))
|
|
#while ((fs = tb.next_fs()) != None):
|
|
for fs in iter(ft.partial(tb.next_fs), None):
|
|
fs.print_debug()
|
|
if tb.trailing_comment:
|
|
print("Trailing comment:\n\"{:s}\"".format(tb.trailing_comment))
|
|
return 0
|
|
|
|
def test_find(ts, argv, dr):
|
|
if len(argv) != 4:
|
|
print("try --help")
|
|
return -errno.EINVAL
|
|
|
|
f, find, what = argv[1:]
|
|
|
|
tb = create_table(f, False)
|
|
if find.lower() == "source":
|
|
fs = tb.find_source(what, dr)
|
|
elif find.lower() == "target":
|
|
fs = tb.find_target(what, dr)
|
|
|
|
if not fs:
|
|
print("{:s}: not found {:s} '{:s}'".format(f, find, what))
|
|
else:
|
|
fs.print_debug()
|
|
return 0
|
|
|
|
def test_find_fw(ts, argv):
|
|
return test_find(ts, argv, mnt.MNT_ITER_FORWARD)
|
|
|
|
def test_find_bw(ts, argv):
|
|
return test_find(ts, argv, mnt.MNT_ITER_BACKWARD)
|
|
|
|
def test_find_pair(ts, argv):
|
|
rc = -1
|
|
tb = create_table(argv[1], False)
|
|
fs = tb.find_pair(argv[2], argv[3], mnt.MNT_ITER_FORWARD)
|
|
if not fs:
|
|
return rc
|
|
fs.print_debug()
|
|
return 0
|
|
|
|
def test_is_mounted(ts, argv):
|
|
rc = -1
|
|
tb = mnt.Tab(path="/proc/self/mountinfo")
|
|
if not tb:
|
|
print("failed to parse mountinto")
|
|
return rc
|
|
|
|
fstab = create_table(argv[1], False)
|
|
if not fstab:
|
|
return rc
|
|
fs = ()
|
|
for fs in ft.iter(tb.next_fs(), -1):
|
|
if tb.is_fs_mounted(fs):
|
|
print("{:s} already mounted on {:s}".format(fs.source, fs.target))
|
|
else:
|
|
print("{:s} not mounted on {:s}".format(fs.source, fs.target))
|
|
return 0
|
|
|
|
def test_find_mountpoint(ts, argv):
|
|
rc = -1
|
|
tb = mnt.Table("/proc/self/mountinfo")
|
|
if not tb:
|
|
return rc
|
|
fs = tb.find_mountpoint(argv[1], mnt.MNT_ITER_BACKWARD)
|
|
if not fs:
|
|
return rc
|
|
fs.print_debug()
|
|
return 0
|
|
|
|
|
|
tss = (
|
|
( "--parse", test_parse, "<file> [--comments] parse and print(tab" ),
|
|
( "--find-forward", test_find_fw, "<file> <source|target> <string>" ),
|
|
( "--find-backward", test_find_bw, "<file> <source|target> <string>" ),
|
|
( "--find-pair", test_find_pair, "<file> <source> <target>" ),
|
|
( "--find-mountpoint", test_find_mountpoint, "<path>" ),
|
|
( "--copy-fs", test_copy_fs, "<file> copy root FS from the file" ),
|
|
( "--is-mounted", test_is_mounted, "<fstab> check what from <file> are already mounted" ),
|
|
)
|
|
sys.exit(mnt_run_test(tss, sys.argv))
|