system.py

Go to the documentation of this file.

#!/usr/bin/python
# -*- coding: utf-8 -*-
# @cond LICENSE
#
# Copyright [2010-2013] Institute of Nuclear Physics PAN, Krakow, Poland
#
# Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.
#
# @endcond LICENSE
##
# @package src.common.tools.system
# @author Maciej Nabozny <mn@mnabozny.pl>
# 

import os
import pwd, grp

def check_my_uid(uid):
    if os.getuid() != uid:
        raise Exception("Uid is not %d" % uid)

def check_my_gid(gid):
    if os.getgid() != gid:
        raise Exception("Gid is not %d" % gid)

def check_my_login(login):
    try:
        pwlogin = pwd.getpwnam(login)
    except:
        raise Exception("Login %s does not exist" % login)

    check_my_uid(pwlogin.pw_uid)

def check_my_group(group):
    try:
        grpgroup = grp.getgrnam(group)
    except:
        raise Exception("Group %s does not exists" % group)

    check_my_gid(grpgroup.g_gid)

def set_cc1():
    uid_cc1 = pwd.getpwnam('cc1').pw_uid
    gid_cc1 = grp.getgrnam('cc1').gr_gid
    gid_kvm = grp.getgrnam('kvm').gr_gid
    gid_libvirt = grp.getgrnam('libvirt').gr_gid

    if check_my_uid(0):
        os.environ['HOME'] = '/var/lib/cc1/'

        os.setgroups([gid_cc1, gid_kvm, gid_libvirt])
        os.setregid(gid_cc1, gid_cc1)
        os.setreuid(uid_cc1, uid_cc1)
    elif os.getuid() != uid_cc1:
        raise Exception("Not you are not root")