1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
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})
|
