Innovation/ufpsutil
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Begin actual work on ufpsutil (currently only started argument management)

Browse source
This commit is contained in:
Innovation 2024-12-15 23:06:30 -06:00
parent 5ffd8c6551
commit fdaa77c872
1 changed files with 55 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

62
ufpsutil.py
View file

@ -1,12 +1,60 @@
from libufps import ufps
import libufps
from inspect import getmembers, isfunction
print(getmembers(ufps, isfunction))
import os
import argparse
basedir = "."
parser = argparse.ArgumentParser(
prog='ufpsutil',
description=' Unix File Permission Supplicant Utility - A simple utility that indexes directories and records their Unix permission and ownership data.'
)
parser.add_argument('path', help = "Path to create index of or restore index to")
parser.add_argument("-c", "--create", help = "Create index", action = "store_true")
parser.add_argument("-r", "--restore", help = "Restore permissions from index", action = "store_true")
parser.add_argument("-i", "--index", help = "Index file to read from/write to (default is index.ufps)", default='index.ufps')
parser.add_argument("--i-know-what-im-doing", help = "Force the utility to run without root", action = "store_true")
ufps = ufps.ufps(basedir)
args = parser.parse_args()
print("Regular dir: " + str(ufps.getFilePermissions("tests/testfiles/")) + " | " + str(ufps.getFileOwner("tests/testfiles/")))
print("Regular file: " + str(ufps.getFilePermissions("tests/testfiles/0644")) + " | " + str(ufps.getFileOwner("tests/testfiles/0644")))
print("Linked dir: " + str(ufps.getFilePermissions("tests/testfiles/linktests/link")) + " | " + str(ufps.getFileOwner("tests/testfiles/linktests/link")))
print("Linked file: " + str(ufps.getFilePermissions("tests/testfiles/linktests/linkfile")) + " | " + str(ufps.getFileOwner("tests/testfiles/linktests/linkfile")))
# Just make argument variables a bit more human readable.
indexFile = args.index
m_create = args.create
m_restore = args.restore
forceWithoutRoot = args.i_know_what_im_doing
# Check that both -c and -r arent specified.
if m_create and m_restore:
exit("[Fatal] Both --create and --restore specified. That doesn't make sense!")
# That said, at least -c or -r must be specified.
if not m_create and not m_restore:
exit("[Fatal] I'm not sure what you want me to do. Must specify --create or --restore.")
# Check if user is running as root.
uid = os.geteuid()
if uid != 0:
# We make a special case for when the user is forcing the program to proceed.
# Otherwise, the program doesn't run without root.
if forceWithoutRoot == True:
print("[Warning] Running without root. This may cause issues when encountering files owned by other users!")
else:
exit("[Fatal] Root privileges are usually required for this utility to do it's job. If you know what you're doing, try running with --i-know-what-im-doing")
#print(args)
#basedir = "tests/testfiles/"
#basedirbad = "tests/testfiles_bad/"
#ufps = ufps.ufps(basedir)
#print("Regular dir: " + ufps.getInformationRecursive("."))
#print("Regular file: " + ufps.getInformationRecursive("0644"))
#print("Linked dir: " + ufps.getInformationRecursive("linktests/link"))
#print("Linked file: " + ufps.getInformationRecursive("linktests/linkfile"))
#index = ufps.createIndex("")
#file = open("index.ufps", "w")
#file.write(index)
#file.close()