comment out timing info

[invirt/packages/invirt-web.git] / code / getafsgroups.py

diff --git a/code/getafsgroups.py b/code/getafsgroups.py
index 899de81..13f8cf7 100644 (file)
--- a/code/getafsgroups.py
+++ b/code/getafsgroups.py
@@ -24,41 +24,41 @@ import subprocess
 #             return True
 #     return False
 
 #             return True
 #     return False
 

-class MyException(Exception):
+class AfsProcessError(Exception):

     pass
 
 def getAfsGroupMembers(group, cell):
     pass
 
 def getAfsGroupMembers(group, cell):

-    p = subprocess.Popen(["pts", "membership", group, '-c', cell], 
+    p = subprocess.Popen(["pts", "membership", "-noauth", group, '-c', cell], 

                          stdout=subprocess.PIPE, stderr=subprocess.PIPE)
                          stdout=subprocess.PIPE, stderr=subprocess.PIPE)

-    if p.wait():
-        return []
+    err = p.stderr.read()
+    if err: #Error code doesn't reveal missing groups, but stderr does
+        raise AfsProcessError(err)

     return [line.strip() for line in p.stdout.readlines()[1:]]
 
     return [line.strip() for line in p.stdout.readlines()[1:]]
 

-def checkAfsGroup(user, group, cell):
-    """
-    checkAfsGroup(user, group) returns True if and only if user is in AFS group group in cell cell
-    """
-    return user in getAfsGroupMembers(group, cell)
+def getLockerPath(locker):
+    if '/' in locker or locker in ['.', '..']:
+        raise AfsProcessError("Locker '%s' is invalid." % locker)
+    return '/mit/' + locker

 
 def getCell(locker):
 
 def getCell(locker):

-    p = subprocess.Popen(["fs", "whichcell", "/mit/" + locker], 
+    p = subprocess.Popen(["fs", "whichcell", getLockerPath(locker)], 

                          stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     if p.wait():
                          stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     if p.wait():

-        raise MyException(p.stderr.read())
+        raise AfsProcessError(p.stderr.read())

     return p.stdout.read().split()[-1][1:-1]
 
 def getLockerAcl(locker):
     return p.stdout.read().split()[-1][1:-1]
 
 def getLockerAcl(locker):

-    p = subprocess.Popen(["fs", "listacl", "/mit/" + locker], 
+    p = subprocess.Popen(["fs", "listacl", getLockerPath(locker)], 

                          stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     if p.wait():
                          stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     if p.wait():

-        raise MyException(p.stderr.read())
+        raise AfsProcessError(p.stderr.read())

     lines = p.stdout.readlines()
     values = []
     for line in lines[1:]:
         fields = line.split()
         if fields[0] == 'Negative':
             break
     lines = p.stdout.readlines()
     values = []
     for line in lines[1:]:
         fields = line.split()
         if fields[0] == 'Negative':
             break

-        if 'rlidwka' in fields[1]:
+        if 'a' in fields[1]:

             values.append(fields[0])
     return values
 
             values.append(fields[0])
     return values
 

@@ -72,22 +72,22 @@ def notLockerOwner(user, locker):
     try:
         cell = getCell(locker)
         values = getLockerAcl(locker)
     try:
         cell = getCell(locker)
         values = getLockerAcl(locker)

-    except MyException, e:
+    except AfsProcessError, e:

         return str(e)
 
     for entry in values:
         return str(e)
 
     for entry in values:

-        if entry == user or (entry[0:6] == "system" and 
-                                checkAfsGroup(user, entry, cell)):
+        if entry == user or (entry[0:6] == "system" and
+                                user in getAfsGroupMembers(entry, cell)):

             return False
             return False

-    return "You don't have admin bits on /mit/" + locker
+    return "You don't have admin bits on " + getLockerPath(locker)

 
 
 if __name__ == "__main__":
 #    print list(getldapgroups("tabbott"))
 
 
 if __name__ == "__main__":
 #    print list(getldapgroups("tabbott"))

-    print checkAfsGroup("tabbott", "system:debathena", 'athena.mit.edu')
-    print checkAfsGroup("tabbott", "system:debathena", 'sipb.mit.edu')
-    print checkAfsGroup("tabbott", "system:debathena-root", 'athena.mit.edu')
-    print checkAfsGroup("tabbott", "system:hmmt-request", 'athena.mit.edu')
+    print "tabbott" in getAfsGroupMembers("system:debathena", 'athena.mit.edu')
+    print "tabbott" in getAfsGroupMembers("system:debathena", 'sipb.mit.edu')
+    print "tabbott" in getAfsGroupMembers("system:debathena-root", 'athena.mit.edu')
+    print "tabbott" in getAfsGroupMembers("system:hmmt-request", 'athena.mit.edu')

     print notLockerOwner("tabbott", "tabbott")
     print notLockerOwner("tabbott", "debathena")
     print notLockerOwner("tabbott", "sipb")
     print notLockerOwner("tabbott", "tabbott")
     print notLockerOwner("tabbott", "debathena")
     print notLockerOwner("tabbott", "sipb")