import unittest
import re
import os
import string
from oeqa.oetest import oeRuntimeTest, skipModule
from oeqa.utils.decorators import *

def get_files_dir():
    """Get directory of supporting files"""
    pkgarch = oeRuntimeTest.tc.d.getVar('MACHINE', True)
    deploydir = oeRuntimeTest.tc.d.getVar('DEPLOY_DIR', True)
    return os.path.join(deploydir, "files", "target", pkgarch)

MAX_LABEL_LEN = 255
LABEL = "a" * MAX_LABEL_LEN

def setUpModule():
    if not oeRuntimeTest.hasPackage('security-manager'):
        skipModule(
            "security-manager module skipped: "
            "target doesn't have security-manager installed")

class SecurityManagerBasicTest(oeRuntimeTest):
    ''' base smack test '''
    def setUp(self):
        # TODO: avoid hardcoding path (also in SecurityManager itself)
        self.security_manager_db = '/usr/dbspace/.security-manager.db'
        cmd = "sqlite3 %s 'SELECT name from privilege ORDER BY privilege_id'" % self.security_manager_db
        status, output = self.target.run(cmd)
        self.assertFalse(status, msg="%s failed: %s" % (cmd, output))
        self.privileges = output.split()
        if not self.privileges:
            # Only privileges that map to a Unix group need to be known to
            # SecurityManager. Therefore it is possible that the query above
            # returns nothing. In that case, make up something for the tests.
            self.privileges.append('FoobarPrivilege')
        self.appid = 'test-app-id'
        self.pkgid = 'test-pkg-id'
        self.user = 'security-manager-user'
        idcmd = 'id -u %s' % self.user
        status, output = self.target.run(idcmd)
        if status:
            # -D is from busybox. It disables setting a password.
            createcmd = 'adduser -D %s' % self.user
            status, output = self.target.run(createcmd)
            self.assertFalse(status, msg="%s failed: %s" % (createcmd, output))
            status, output = self.target.run(idcmd)
        self.assertTrue(output.isdigit(), msg="Unexpected output from %s: %s" % (idcmd, output))
        self.uid = output

class SecurityManagerApp(SecurityManagerBasicTest):
    '''Tests covering app installation. Ordering is important, therefore tests are numbered.'''

    @skipUnlessPassed('test_ssh')
    def test_security_manager_01_setup(self):
        '''Check that basic SecurityManager setup is in place.'''
        # If we get this far, then at least the sqlite db must have been in place.
        # This does not mean much, but we need to start somewhere.
        pass

    @skipUnlessPassed('test_security_manager_01_setup')
    def test_security_manager_02_install(self):
        '''Test if installing an app sets up privilege rules for it, also in Cynara.'''
        self.target.copy_to(os.path.join(get_files_dir(), "app-runas"), "/tmp/")
        cmd = '/tmp/app-runas -a %s -p %s -u %s -r %s -i' % \
              (self.appid, self.pkgid, self.uid, self.privileges[0])
        status, output = self.target.run(cmd)
        self.assertFalse(status, msg="%s failed: %s" % (cmd, output))
        cmd = '''sqlite3 %s 'SELECT uid,app_name,pkg_name from app_pkg_view WHERE app_name = "%s"' ''' % \
              (self.security_manager_db, self.appid)
        status, output = self.target.run(cmd)
        self.assertFalse(status, msg="%s failed: %s" % (cmd, output))
        self.assertEqual(output, '|'.join((self.uid, self.appid, self.pkgid)))
        cmd = 'grep -r %s /var/cynara/db/' % self.appid
        status, output = self.target.run(cmd)
        self.assertFalse(status, msg="%s failed: %s" % (cmd, output))
        # User::App:: prefix still hard-coded here because it is not customizable at the moment.
        self.assertEqual(output, '/var/cynara/db/_MANIFESTS:User::App::%s;%s;%s;0xFFFF;' % \
                         (self.appid, self.uid, self.privileges[0]))

    @skipUnlessPassed('test_security_manager_02_install')
    def test_security_manager_03_run(self):
        '''Test running as app. Depends on preparations in test_security_manager_install().'''
        cmd = '''/tmp/app-runas -a %s -u %s -e -- sh -c 'id -u && cat /proc/self/attr/current' ''' % \
              (self.appid, self.uid)
        status, output = self.target.run(cmd)
        self.assertFalse(status, msg="%s failed: %s" % (cmd, output))
        self.assertEqual(output, '%s\nUser::App::%s' % (self.uid, self.appid))

    @skipUnlessPassed('test_security_manager_02_install')
    def test_security_manager_03_uninstall(self):
        '''Test removal of an app.'''
        cmd = '/tmp/app-runas -a %s -p %s -u %s -d' % \
              (self.appid, self.pkgid, self.uid)
        status, output = self.target.run(cmd)
        self.assertFalse(status, msg="%s failed: %s" % (cmd, output))
        cmd = '''sqlite3 %s 'SELECT uid,app_name,pkg_name from app_pkg_view WHERE app_name = "%s"' ''' % \
              (self.security_manager_db, self.appid)
        status, output = self.target.run(cmd)
        self.assertFalse(status, msg="%s failed: %s" % (cmd, output))
        # Entry does not really get removed. Bug filed here:
        # https://github.com/Samsung/security-manager/issues/2
        # self.assertEqual(output, '')
        cmd = 'grep -r %s /var/cynara/db/' % self.appid
        status, output = self.target.run(cmd)
        self.assertFalse(status, msg="%s failed: %s" % (cmd, output))
        # This also does not get removed. Perhaps same root cause.
        # self.assertEqual(output, '')