import commands
import random
import re
alphabet = 'azertyuiopqsdfghjklmwxcvbnAZERTYUIOPQSDFGHJKLMWXCVBN123456789-_'
a_length = len(alphabet)
""" ACL support attribute """
ACL4_SUPPORT_ALLOW_ACL = 0x00000001
ACL4_SUPPORT_DENY_ACL = 0x00000002
ACL4_SUPPORT_AUDIT_ACL = 0x00000004
ACL4_SUPPORT_ALARM_ACL = 0x00000008
class RandomGen(object):
""" List of ACE possible who fields """
ace_who=["OWNER@","GROUP@","EVERYONE@","ANONYMOUS@","AUTHENTICATED@"]
""" List of GID than can be used to do the tests """
gList=[]
gListSize = len(gList)
uList = []
uListSize = len(uList)
fList=[]
fListSize = len(fList)
""" Create a user in available groups to do the tests """
def createUser(self,username):
group = self.gList[random.randint(0,len(self.gList)-1)][0]
opts = "-g" + group + " -p" + "1pilot" + " -m " + username
u = commands.getoutput('/usr/sbin/useradd '+ opts)
if u != "":
print "create user " + username + "failed" + u
def createFile(self,path,n):
for i in range(n):
fName = 'file' + str(i)
u = commands.getoutput('touch ' + path + '/'+ fName)
self.fList.append(fName)
def createGroup(self, grpname, gid):
u = commands.getoutput('/usr/sbin/groupadd -g' + gid + " " + grpname)
if u != "":
print u
def createNGroup(self, n):
for i in range(n):
gName = 'grp' + str(i)
gid = str(500+i)
self.createGroup(gName, gid)
""" Random creation of n user """
def createNUser(self,n):
for i in range(n):
userName= "user" + str(i)
self.createUser(userName)
""" clean all users created to do the tests """
def cleanUsers(self):
for name in self.uList:
u = commands.getoutput('/usr/sbin/userdel -r '+ name)
self.uList = []
""" clean all users created to do the tests """
def cleanGroups(self):
for name in self.gList:
u = commands.getoutput('/usr/sbin/groupdel '+ name[0])
self.gList = []
""" Retrieve the list of user from /etc/passwd file """
def getUserList(self):
f = open('/etc/passwd','r')
lines = f.readlines()
for line in lines:
splitedline = line.split(':')
userName = splitedline[0]
gid = splitedline[3]
# TO FIX: verify that the group is OK (in the right range)
NameOK = re.match("user",userName)
# We keep only usernames starting with "user"
if NameOK != None:
self.uList.append(userName)
f.close()
def getFileList(self,path):
u = commands.getoutput('ls ' + path)
tmp = u.split('\n')
for i in range (len(tmp)-1):
NameOK = re.match("file",tmp[i])
if NameOK != None:
self.fList.append(tmp[i])
def getNUserList(self,nb):
f = open('/etc/passwd','r')
lines = f.readlines()
n = 0
for line in lines:
splitedline = line.split(':');
userName = splitedline[0]
gid = splitedline[3]
# TO FIX: verify that the group is OK (in the right range)
NameOK = re.match("user",userName)
# We keep only usernames starting with "user"
if NameOK != None:
self.uList.append(userName)
n = n+1
if n==nb:
break;
f.close()
""" Get group list """
def getGroupList(self):
f = open('/etc/group','r')
lines = f.readlines()
for line in lines:
splitedline = line.split(':');
groupName = splitedline[0]
gid = splitedline[2]
NameOK = re.match("grp",groupName)
if NameOK != None:
self.gList.append([groupName,gid])
f.close()
""" Get a list of n group """
def getNGroupList(self,nb):
f = open('/etc/group','r')
lines = f.readlines()
n = 0
for line in lines:
splitedline = line.split(':');
groupName = splitedline[0]
gid = splitedline[2]
NameOK = re.match("grp",groupName)
if NameOK != None:
self.gList.append([groupName,gid])
n = n+1
if n==nb:
break;
f.close()
def printUserList(self):
print self.uList
def printGroupList(self):
print self.gList
""" Create a random name of random length """
def createOneNameRandomLength(self,maxlength):
outputString =""
l=random.randint(0,maxlength)
for i in range(l):
a = random.randint(0,a_length-1)
outputString =outputString + alphabet[a]
return outputString
""" Create a random name of fixed length """
def createOneName(self,lenght):
outputString =""
for i in range(length):
a = random.randint(0,a_length-1)
outputString = outputString + alphabet[a]
return outputString
""" Create Random User List with fixed length user names """
def createRandomUserList(self,listlength,usernamelength):
userlist = []
for i in range(listlength):
user = createOneName(lenght)
userlist.append(user)
return userlist
""" Create Random ACE for a file and a given usr """
def createRandomACE(self,user):
type = ace_type[random.randint(0,len(ace_type))]
flag = ace_flags[random.randint(0,len(ace_flags))]
mask = ace_mask[random.randint(0,len(ace_mask))]
who = ace_who[random.randint(0,len(ace_who))]
return nfsace4(type, flag, mask, who)
""" Create Random ACL for a file with a fixed number a entries """
def createRandomACL(self,acl_size):
acl = []
userList = uList
userListSize = uListSize
for i in range(acl_size):
n = random.randint(0,userListSize-1)
usr = userList.pop(n)
newace = createRandomACE(usr)
acl.append(newace)
return acl
""" Return a mode string like 'xwr' or 'x' """
def createRandomMode(self):
out_str = ""
while (out_str == ""):
if random.randint(0,1) == 1:
out_str += 'x'
if random.randint(0,1) == 1:
out_str += 'w'
if random.randint(0,1) == 1:
out_str += 'r'
return out_str
""" Create a random ACL operation (delete / remove / modify on user / group ) """
def randomOp(self,path):
a = random.randint(1,4)
mode = self.createRandomMode()
file = self.fList[random.randint(0,len(self.fList)-1)]
if a == 1: # creation/modification
user = self.uList[random.randint(0,len(self.uList)-1)]
u = commands.getoutput('setfacl -m u:' + user + ':' + mode + " " + path + "/" + file)
if a == 2: # with group
group = self.gList[random.randint(0,len(self.gList)-1)][0]
u = commands.getoutput('setfacl -m g:' + group + ':' + mode + " " + path + "/" + file)
if a == 3: # deletation
user = self.uList[random.randint(0,len(self.uList)-1)]
u = commands.getoutput('setfacl -x u:' + user + " " + path + "/" + file)
if a == 4: # with group
group = self.gList[random.randint(0,len(self.gList)-1)][0]
u = commands.getoutput('setfacl -x g:' + group + " " + path + "/" + file)
# request on a unexisting group
'''if a == 5:
group = self.createOneNameRandomLength(16)
print 'setfacl -x g:' + group + " " + path + "/" + file
u = commands.getoutput('setfacl -x g:' + group + " " + path + "/" + file)
if a == 6:
user = self.createOneNameRandomLength(16)
u = commands.getoutput('setfacl -x u:' + user + " " + path + "/" + file)
if a == 7: # creation/modification
user = self.createOneNameRandomLength(16)
u = commands.getoutput('setfacl -m u:' + user + ':' + mode + " " + path + "/" + file)
if a == 8: # with group
group = self.createOneNameRandomLength(16)
u = commands.getoutput('setfacl -m g:' + group + ':' + mode + " " + path + "/" + file)
if a == 9: #Copying the ACL of one file to another
file2 = self.fList[random.randint(0,len(self.fList)-1)]
u = commands.getoutput('getfacl ' + path + "/" + file + "| setfacl --set-file=- " + path + "/" + file2)
if u!="":
print u'''