0.17

1 files changed, 173 insertions, 0 deletions

diff --git a/share/Perms.py b/share/Perms.py
new file mode 100755
index 0000000..8a8c983
--- /dev/null
+++ b/share/Perms.py
@@ -0,0 +1,173 @@
+#!/usr/bin/python -O
+#---------------------------------------------------------------
+# Project         : Mandrake Linux
+# Module          : msec
+# File            : Perms.py
+# Version         : $Id$
+# Author          : Frederic Lepied
+# Created On      : Fri Dec  7 23:33:49 2001
+# Purpose         : fix permissions and owner/group of files
+#                   and directories.
+#---------------------------------------------------------------
+
+import glob
+import re
+import string
+import os
+import stat
+import pwd
+import grp
+import Config
+from Log import *
+import gettext
+
+try:
+    cat = gettext.Catalog('msec')
+    _ = cat.gettext
+except IOError:
+    _ = str
+
+comment_regex = re.compile('^\s*#|^\s*$')
+
+user = {}
+group = {}
+userid = {}
+groupid = {}
+
+def get_user_id(name):
+    try:
+        return user[name]
+    except KeyError:
+        try:
+            user[name] = pwd.getpwnam(name)[2]
+        except KeyError:
+            error(_('user name %s not found') % name)
+            user[name] = -1
+    return user[name]
+
+def get_user_name(id):
+    try:
+        return userid[id]
+    except KeyError:
+        try:
+            userid[id] = pwd.getpwuid(id)[0]
+        except KeyError:
+            error(_('user name not found for id %d') % id)
+            userid[id] = str(id)
+    return userid[id]
+
+def get_group_id(name):
+    try:
+        return group[name]
+    except KeyError:
+        try:
+            group[name] = grp.getgrnam(name)[2]
+        except KeyError:
+            error(_('group name %s not found') % name)
+            group[name] = -1
+    return group[name]
+
+def get_group_name(id):
+    try:
+        return groupid[id]
+    except KeyError:
+        try:
+            groupid[id] = grp.getgrgid(id)[0]
+        except KeyError:
+            error(_('group name not found for id %d') % id)
+            groupid[id] = str(id)
+    return groupid[id]
+
+assoc = {}
+
+def fix_perms(path):
+    try:
+        file = open(path, 'r')
+    except IOError:
+        return
+    root = Config.get_config('root', '')
+    lineno = 0
+    for line in file.readlines():
+        lineno = lineno + 1
+        if comment_regex.search(line):
+            continue
+        fields = re.split('\s*', line)
+        newmode = int(fields[2], 8)
+        if fields[1] == 'current':
+            user = group = -1
+            user_str = group_str = ''
+        else:
+            (user_str, group_str) = string.split(fields[1], '.')
+            user = get_user_id(user_str)
+            group = get_group_id(group_str)
+        if len(fields) == 4:
+            for f in glob.glob(fields[0]):
+                try:
+                    full = os.stat(f)
+                except OSError:
+                    continue
+                mode = stat.S_IMODE(full[stat.ST_MODE])
+                newperm = newmode
+                if stat.S_ISDIR(full[stat.ST_MODE]):
+                    if newperm & 0400:
+                        newperm = newperm | 0100
+                    if newperm & 0040:
+                        newperm = newperm | 0010
+                    if newperm & 0004:
+                        newperm = newperm | 0001
+                uid = full[stat.ST_UID]
+                gid = full[stat.ST_GID]
+                if f != '/' and f[-1] == '/':
+                    f = f[:-1]
+                if f[-2:] == '/.':
+                    f = f[:-2]
+                assoc[f] = (mode, uid, gid, newperm, user, group, user_str, group_str)
+        else:
+            error(_('invalid syntax in %s line %d') % (path, lineno))
+    file.close()
+
+def act():
+    for f in assoc.keys():
+        (mode, uid, gid, newperm, user, group, user_str, group_str) = assoc[f]
+        #print f, (mode, uid, gid, newperm, user, group)
+        if mode != newperm:
+            log(_('changed mode of %s from %o to %o') % (f, mode, newperm))
+            os.chmod(f, newperm)
+        if user != -1:
+            if user != uid:
+                log(_('changed owner of %s from %s to %s') % (f, get_user_name(uid), user_str))
+                os.chown(f, user, -1)
+        if group != -1:
+            if group != gid:
+                log(_('changed group of %s from %s to %s') % (f, get_group_name(gid), group_str))
+                os.chown(f, -1, group)
+
+def chmod(f, newperm):
+    try:
+        full = os.stat(f)
+    except OSError:
+        return 0
+    mode = stat.S_IMODE(full[stat.ST_MODE])
+    if stat.S_ISDIR(full[stat.ST_MODE]):
+        if newperm & 0400:
+            newperm = newperm | 0100
+        if newperm & 0040:
+            newperm = newperm | 0010
+        if newperm & 0004:
+            newperm = newperm | 0001
+    if mode != newperm:
+        log(_('changed mode of %s from %o to %o') % (f, mode, newperm))
+        os.chmod(f, newperm)
+    return 1
+
+if __name__ == '__main__':
+    import sys
+
+    initlog('msec')
+    
+    for p in sys.argv[1:]:
+        fix_perms(p)
+
+    act()
+    
+# Perms.py ends here