import types, os, sys, isys, select, string, stat, signal def getArch (): arch = os.uname ()[4] if (len (arch) == 4 and arch[0] == 'i' and arch[2:4] == "86"): arch = "i386" if arch == "sparc64": arch = "sparc" return arch def getfd(filespec, readOnly = 0): if type(filespec) == types.IntType: return filespec if filespec == None: filespec = "/dev/null" flags = os.O_RDWR | os.O_CREAT if (readOnly): flags = os.O_RDONLY return os.open(filespec, flags) def execWithRedirect(command, argv, stdin = 0, stdout = 1, stderr = 2, searchPath = 0, root = '/', newPgrp = 0, ignoreTermSigs = 0): stdin = getfd(stdin) if stdout == stderr: stdout = getfd(stdout) stderr = stdout else: stdout = getfd(stdout) stderr = getfd(stderr) if not os.access (root + command, os.X_OK): if not os.access (command, os.X_OK): raise RuntimeError, command + " can not be run" else: root = "" childpid = os.fork() if (not childpid): if (root and root != '/'): isys.chroot (root) os.chdir("/") if ignoreTermSigs: signal.signal(signal.SIGTSTP, signal.SIG_IGN) signal.signal(signal.SIGINT, signal.SIG_IGN) if type(stdin) == type("a"): stdin == os.open(stdin, os.O_RDONLY) if type(stdout) == type("a"): stdout == os.open(stdout, os.O_RDWR) if type(stderr) == type("a"): stderr = os.open(stderr, os.O_RDWR) if stdin != 0: os.dup2(stdin, 0) os.close(stdin) if stdout != 1: os.dup2(stdout, 1) if stdout != stderr: os.close(stdout) if stderr != 2: os.dup2(stderr, 2) os.close(stderr) if (searchPath): os.execvp(command, argv) else: os.execv(command, argv) sys.exit(1) if newPgrp: os.setpgid(childpid, childpid) oldPgrp = os.tcgetpgrp(0) os.tcsetpgrp(0, childpid) (pid, status) = os.waitpid(childpid, 0) if newPgrp: os.tcsetpgrp(0, oldPgrp) return status def execWithCapture(command, argv, searchPath = 0, root = '/', stdin = 0): if not os.access (root + command, os.X_OK): if not os.access (command, os.X_OK): raise RuntimeError, command + " can not be run" else: root = "" (read, write) = os.pipe() childpid = os.fork() if (not childpid): if (root and root != '/'): isys.chroot (root) os.dup2(write, 1) if stdin: os.dup2(stdin, 0) os.close(stdin) if (searchPath): os.execvp(command, argv) else: os.execv(command, argv) sys.exit(1) os.close(write) rc = "" s = "1" while (s): select.select([read], [], []) s = os.read(read, 1000) rc = rc + s os.close(read) os.waitpid(childpid, 0) return rc def copyFile(source, to): f = os.open(source, os.O_RDONLY) t = os.open(to, os.O_RDWR | os.O_TRUNC | os.O_CREAT) count = os.read(f, 16384) while (count): os.write(t, count) count = os.read(f, 16384) os.close(f) os.close(t) def memInstalled(): f = open("/proc/meminfo", "r") mem = f.readlines()[1] del f # patch from hjl fields = string.split(mem) try: mem = int(fields[1]) / 1024 except: mem = 2097151 return mem # this is a mkdir that won't fail if a directory already exists and will # happily make all of the directories leading up to it. def mkdirChain(dir): if (os.path.isdir(dir)): return elements = string.splitfields(dir, "/") if (len(elements[0])): which = 1 path = elements[0] else: which = 2 path = "/" + elements[1] if (not os.path.isdir(path)): os.mkdir(path, 0755) while (which < len(elements)): path = path + "/" + elements[which] which = which + 1 if (not os.path.isdir(path)): os.mkdir(path, 0755) # # get default runlevel - only for use in reconfig mode # def getDefaultRunlevel (): inittab = open ('/etc/inittab', 'r') lines = inittab.readlines () inittab.close () for line in lines: if len (line) > 3 and line[:3] == "id:": fields = string.split (line, ':') return fields[1] return None def makerelname(relpath, filename): if relpath != '': return relpath+'/'+filename else: return filename def findtz(basepath, relpath): tzdata = [] for n in os.listdir(basepath+'/'+relpath): timezone = makerelname(relpath, n) if relpath != '': timezone = relpath+'/'+n else: timezone = n filestat = os.lstat(basepath+'/'+timezone) [filemode] = filestat[:1] if (not (stat.S_ISLNK(filemode) or stat.S_ISREG(filemode) or stat.S_ISDIR(filemode))): continue elif n[:1] >= 'A' and n[:1] <= 'Z': if stat.S_ISDIR(filemode): tmptzdata = findtz(basepath, timezone) else: tmptzdata = [timezone] for m in tmptzdata: if tzdata == []: tzdata = [m] else: tzdata.append(m) tzdata.sort() return tzdata def rmrf (path): # this is only the very simple case. files = os.listdir (path) for file in files: if os.path.isdir(path + '/' + file): rmrf (path + '/' + file) else: os.unlink (path + '/' + file) os.rmdir (path) def validUser (user): if len (user) > 8: return 0 if not user[0] in string.letters: return 0 for letter in user: if (letter == ':' or letter == ',' or letter == '\n' or ord (letter) < 33): return 0 return 1 def setClock (root): # eeeeew, inline shell. ;) args = ("bash", "-c", """ if [ -f /etc/sysconfig/clock ]; then . /etc/sysconfig/clock # convert old style clock config to new values if [ "${CLOCKMODE}" = "GMT" ]; then UTC=true elif [ "${CLOCKMODE}" = "ARC" ]; then ARC=true fi fi CLOCKFLAGS="--hctosys" case "$UTC" in yes|true) CLOCKFLAGS="$CLOCKFLAGS -u"; ;; esac case "$ARC" in yes|true) CLOCKFLAGS="$CLOCKFLAGS -A"; ;; esac case "$SRM" in yes|true) CLOCKFLAGS="$CLOCKFLAGS -S"; ;; esac /sbin/hwclock $CLOCKFLAGS """) execWithRedirect('/bin/sh', args, stdin = None, stdout = None, stderr = None, root = root) def defaultZone(): if os.access("lang-table", os.R_OK): f = open("lang-table", "r") elif os.access("/etc/lang-table", os.R_OK): f = open("/etc/lang-table", "r") else: f = open("/usr/lib/anaconda/lang-table", "r") lines = f.readlines () f.close() for line in lines: string.strip(line) l = string.split(line) if l[4] == os.environ["LANG"]: return l[6] return "America/New_York" def defaultKeyboard(): if os.access("lang-table", os.R_OK): f = open("lang-table", "r") elif os.access("/etc/lang-table", os.R_OK): f = open("/etc/lang-table", "r") else: f = open("/usr/lib/anaconda/lang-table", "r") lines = f.readlines () f.close() for line in lines: string.strip(line) l = string.split(line) if l[4] == os.environ["LANG"]: return l[5] return "us"