test_config.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

""" This file is part of B{Domogik} project (U{http://www.domogik.org}).

License
=======

B{Domogik} is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

B{Domogik} is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Domogik. If not, see U{http://www.gnu.org/licenses}.

Module purpose
==============

Test domogik configuration

Implements
==========


@author: Maxence Dunnewind <maxence@dunnewind.net>
@copyright: (C) 2007-2012 Domogik project
@license: GPL(v3)
@organization: Domogik
"""

import os
import pwd
import sys
from multiprocessing import Process, Pipe
from socket import gethostbyname, gethostname

BLUE = '\033[94m'
OK = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'

user = ''

def info(msg):
    print("%s [ %s ] %s" % (BLUE,msg,ENDC))
def ok(msg):
    print("%s ==> %s  %s" % (OK,msg,ENDC))
def warning(msg):
    print("%s ==> %s  %s" % (WARNING,msg,ENDC))
def fail(msg):
    print("%s ==> %s  %s" % (FAIL,msg,ENDC))

def am_i_root():
    info("Check this script is started as root")
    assert os.getuid() == 0, "This script must be started as root"
    ok("Correctly started with root privileges.")

def test_imports():
    good = True
    info("Test imports")
    try:
        import domogik
    except ImportError:
        warning("domogik package can not be imported, did you made ./setup.py develop or ./setup.py install ?")
        good = False
    try:
        import OpenSSL
    except ImportError:
        warning("OpenSSL can't be imported, please exec ./setup.py develop or ./setup.py install and check it \
                downloads and install pyOpenSSL correctly.")
        good = False
    try:
        import sqlalchemy
    except ImportError:
        warning("Can't import sqlalchemy, please install it")
        good = False
    try:
        import httplib
    except ImportError:
        warning("Can't import httplib, please install it by hand (>= 2) or exec ./setup.py develop or ./setup.py install")
        good = False
    try:
        import simplejson
    except ImportError:
        warning("Can't import simplejson, please install it by hand (>= 1.1) or exec ./setup.py develop or ./setup.py install")
        good = False
    assert good, "One or more import have failed, please install required packages and restart this script."
    ok("Imports are good")

def test_config_files():
    global user
    info("Test global config file")
    assert os.path.isfile("/etc/conf.d/domogik") or os.path.isfile("/etc/default/domogik"), \
            "No global config file found, please exec install.sh if you did not exec it before."
    assert not (os.path.isfile("/etc/conf.d/domogik") and os.path.isfile("/etc/default/domogik")), \
            "Global config file found at 2 locations. Please put it only at /etc/default/domogik or \
            /etc/conf.d/domogik then restart test_config.py as root"
    if os.path.isfile("/etc/default/domogik"):
        file = "/etc/default/domogik"
    else:
        file = "/etc/conf.d/domogik"
    f = open(file,"r")
    r = f.readlines()
    lines = filter(lambda x: not x.startswith('#') and x != '\n',r)
    f.close()
    #user = ''
    manager_params = ''
    custom_path = ''
    hub_iface = ''
    launch_domogik_xpl_hub = ''
    for line in lines:
        item,value = line.strip().split("=")
        if item.strip() == "DOMOGIK_USER":
            user = value
        elif item.strip() == "MANAGER_PARAMS":
            manager_params = value
        elif item.strip() == "CUSTOM_PATH":
            custom_path = value
        elif item.strip() == "HUB_IFACE":
            hub_iface = value
        elif item.strip() == "LAUNCH_DOMOGIK_XPL_HUB":
            launch_domogik_xpl_hub = value
        elif item.strip() == "DOMOGIK_XPL_HUB":
            domogik_xpl_hub = value
        elif item.strip() == "DIETIME":
            pass
        elif item.strip() == "STARTTIME":
            pass
        else:
            warning("Unknown config value in the main config file : %s" % item)
    ok("Global config file exists and contains right stuff")

    if launch_domogik_xpl_hub not in ["true", "false"]:
        warning("LAUNCH_DOMOGIK_XPL_HUB= must be 'true' or 'false'.")

    #Check manager params
    info("Check manager params")
    params = True
    if "-d" not in manager_params:
        warning("No -d option in MANAGER_PARAMS. You should have it unless you are using domogik on more than one computer.")
        params = False
    if "-r" not in manager_params:
        warning("No -r option in MANAGER_PARAMS. You should have it unless you are using domogik on more than one computer.")
        params = False
    if params:
        ok("Manager params seem good")

    #Check if we can find xPL_Hub
    info("Check xPL_Hub is in path")
    path = os.environ['PATH'].split(':')
    path.append(custom_path)
    l = [p for p in path if os.path.exists(os.path.join(p, 'xPL_Hub'))]
    assert l != [], "xPL_Hub can't be found, please double check CUSTOM_PATH is correctly defined if you are in development mode. In install mode, check your architecture is supported or check src/domogik/xpl/tools/COMPILE.txt, then restart test_config.py"
    ok("xPL_Hub found in the path")

    info("Test user / config file")

    #Check user config file
    try:
        user_entry = pwd.getpwnam(user)
    except KeyError:
        raise KeyError("The user %s does not exists, you MUST create it or change the DOMOGIK_USER parameter in %s. Please report this as a bug if you used install.sh." % (user, file))
    user_home = user_entry.pw_dir
    assert os.path.isfile("/etc/domogik/domogik.cfg"), "The domogik config file /etc/domogik/domogik.cfg does not exist. Please report this as a bug if you used install.sh." % user_home
    ok("Domogik's user exists and has a config file")
    
    test_user_config_file(user_home, user_entry)

def _test_user_can_write(conn, path, user_entry):
    os.setgid(user_entry.pw_gid)
    os.setuid(user_entry.pw_uid)
    conn.send(os.access(path, os.W_OK))
    conn.close()

def _check_port_availability(s_ip, s_port, udp = False):
    """ Parse /proc/net/tcp to check if something listen on the port"""
    ip = gethostbyname(s_ip).split('.')
    port = "%04X" % int(s_port)
    ip = "%02X%02X%02X%02X" % (int(ip[3]),int(ip[2]),int(ip[1]),int(ip[0]))
    if udp == False:
        f = open("/proc/net/tcp")
    else:
        f = open("/proc/net/udp")
    lines = f.readlines()
    f.close()
    lines.pop(0)
    for line in lines:
        data = line.split()
        d_ip = data[1].split(':')[0]
        d_port = data[1].split(':')[1]
        if d_port == port:
            assert d_ip != ip and ip != "00000000", "A service already listen on ip %s and port %s. Stop it and restart test_config.py" % (s_ip, s_port)

def test_user_config_file(user_home, user_entry):
    info("Check user config file contents")
    import ConfigParser
    config = ConfigParser.ConfigParser()
    config.read("/etc/domogik/domogik.cfg")
    
    #check [domogik] section
    dmg = dict(config.items('domogik'))
    database = dict(config.items('database'))
    rest = dict(config.items('rest'))
    ok("Config file correctly loaded")

    info("Parse [domogik] section")
    import domogik

    #Check ix xpl port is not used
    _check_port_availability("0.0.0.0", 3865, udp = True)
    ok("xPL hub IP/port is not bound by anything else")

    parent_conn, child_conn = Pipe()
    p = Process(target=_test_user_can_write, args=(child_conn, dmg['log_dir_path'],user_entry,))
    p.start()
    p.join()
    assert parent_conn.recv(), "The directory %s for log does not exist or does not have right permissions" % dmg['log_dir_path']

    assert dmg['log_level'] in ['debug','info','warning','error','critical'], "The log_level parameter does not have a good value. Must \
            be one of debug,info,warning,error,critical"
    ### obsolete
    #if not os.path.isdir(dmg['src_prefix'] + '/share/domogik'):
    #    try:
    #        f = os.listdir("%s/share/domogik" % dmg['src_prefix'])
    #        f.close()
    #    except OSError:
    #        fail("Can't access %s/share/domogik. Check %s is available for domogik user (if you are in development mode, be sure the directory which contains the sources is available for domogik user)." % (dmg['src_prefix'],dmg['src_prefix']))
    #        exit()
    ok("[domogik] section seems good")

    # check [database] section
    info("Parse [database] section")
    assert database['db_type'] == 'mysql', "Only mysql database type is supported at the moment"

    uid = user_entry.pw_uid
    os.setreuid(0,uid)
    old_home = os.environ['HOME']
    os.environ['HOME'] = user_home
    from domogik.common.database import DbHelper
    d = DbHelper()
    os.setreuid(0,0)
    os.environ['HOME'] = old_home
    assert d.get_engine() != None, "Engine is not set, it seems something went wrong during connection to the database"
        
    ok("[database] section seems good")
    
    # Check [rest] section
    info("Parse [rest] section")
    _check_port_availability(rest['rest_server_ip'], rest['rest_server_port'])
    ok("Rest server IP/port is not bound by anything else")

def test_init():
    info("Check init.d / rc.d")
    assert os.access("/etc/init.d/domogik", os.X_OK) or os.access("/etc/rc.d/domogik", os.X_OK), "/etc/init.d/domogik and /etc/rc.d/domogik do not \
            exist or can't be executed.\
            Please copy src/domogik/examples/init/domogik to /etc/init.d or /etc/rc.d depending on your system, and chmod +x /etc/init.d/domogik"
    ok("/etc/init.d/domogik or /etc/rc.d/domogik found with good permissions")

def test_version():
    info("Check python version")
    v = sys.version_info
    assert not (v[0] == 2 and v[1] < 6), "Python version is %s.%s, it must be >= 2.6, please upgrade" % (v[0], v[1])
    ok("Python version is >= 2.6")

def test_hostname():
    info("Check hostname")
    assert len(gethostname().split('.')[0]) < 16, "Your hostname length is > 16, because it is used into xpl messages, it must be < 16).\
            Please change it in /etc/hostname and /etc/hosts, logout and login, then run ./test_config.py again."
    ok("Hostname length is < 16.")
    if gethostname().count("-") > 0:
        warning("Your hostname is '%s'. It shouldn't contain the character '-'." % gethostname())
    else:
        ok("Hostname characters are OK")

try:
    am_i_root()
    test_imports()
    test_hostname()
    test_config_files()
    test_init()
    test_version()
    print("\n\n")
    ok("================================================== <==")
    ok(" Everything seems ok, you should be able to start  <==")
    ok("      Domogik with /etc/init.d/domogik start       <==")
    ok("            or /etc/rc.d/domogik start             <==")
    ok(" You can now install Domoweb User Interface        <==")
    ok("================================================== <==")
except:
    fail(sys.exc_info()[1])