From 8e75011239e95eb718e7f5baf800b33423aa39ba Mon Sep 17 00:00:00 2001
From: KOLANICH <kolan_n@mail.ru>
Date: Sun, 18 Dec 2016 03:23:21 +0300
Subject: Added symlink fixer. It fixes the profiles in order to give access to
 symlinked binaries (for example if sh -> dash and dash is not in private-bin,
 you can't use sh)

---
 etc/fix_private-bin_for_symlinked_sh.py | 68 +++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 etc/fix_private-bin_for_symlinked_sh.py

(limited to 'etc/fix_private-bin_for_symlinked_sh.py')

diff --git a/etc/fix_private-bin_for_symlinked_sh.py b/etc/fix_private-bin_for_symlinked_sh.py
new file mode 100644
index 000000000..705e46e46
--- /dev/null
+++ b/etc/fix_private-bin_for_symlinked_sh.py
@@ -0,0 +1,68 @@
+#!/usr/bin/python3
+
+import sys, os, glob, re
+
+privRx=re.compile("^(?:#\s*)?private-bin")
+
+def fixSymlinkedBins(files, replMap):
+	rxs=dict()
+	for (old,new) in replMap.items():
+		rxs[old]=re.compile("\\b"+old+"\\b")
+		rxs[new]=re.compile("\\b"+new+"\\b")
+	print(rxs)
+	
+	for filename in files:
+		lines=None
+		with open(filename,"r") as file:
+			lines=file.readlines()
+		
+		shouldUpdate=False
+		for (i,line) in enumerate(lines):
+			if privRx.search(line):
+				for (old,new) in replMap.items():
+					if rxs[old].search(line) and not rxs[new].search(line):
+						lines[i]=rxs[old].sub(old+","+new, line)
+						shouldUpdate=True
+						print(lines[i])
+		
+		if shouldUpdate:
+			with open(filename,"w") as file:
+				file.writelines(lines)
+				pass
+
+def createListOfBinaries(files):
+	s=set()
+	for filename in files:
+		lines=None
+		with open(filename,"r") as file:
+			for line in file:
+				if privRx.search(line):
+					bins=line.split(",")
+					bins[0]=bins[0].split(" ")[-1]
+					bins = [n.strip() for n in bins]
+					s=s|set(bins)
+	return s
+
+def createSymlinkTable(binDirs, binariesSet):
+	m=dict()
+	for sh in binariesSet:
+		for bD in binDirs:
+			p=bD+os.path.sep+sh
+			if os.path.exists(p):
+				if os.path.islink(p):
+					m[sh]=os.readlink(p)
+				else:
+					pass
+				break
+	return m
+
+
+sh="sh"
+binDirs=["/bin","/usr/bin","/usr/sbin","/usr/local/bin","/usr/local/sbin"]
+profilesPath="."
+files=glob.glob(profilesPath+os.path.sep+"*.profile")
+
+bins=createListOfBinaries(files)
+stbl=createSymlinkTable(binDirs,bins)
+print(stbl)
+fixSymlinkedBins(files,{a[0]:a[1] for a in stbl.items() if a[0].find("/") < 0 and a[1].find("/")<0})
-- 
cgit v1.2.3-54-g00ecf