summaryrefslogtreecommitdiff
blob: 1d04580cbf01b1046e742bcf008641cdb24e841e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
import xml.dom.minidom
import os, portage, types
import subprocess as sp
from os.path import join as pjoin, dirname as pdirname, realpath as prealpath

from .. import benchconfig as cfg
from .. import PortageUtils as pu


def readFile(fs):
    result = {}

    # If fs is a filename, open it
    if type(fs) != types.FileType:
        fs = file(pjoin(cfg.curdir, fs))

    # Read line by line
    for l in fs.readlines():
        try:
            k,v = l.split('=', 1)
            result[k.strip()] = v.strip()
        except:
            pass

    return result


def readScript(fname):
    fname = pjoin(cfg.curdir, fname)

    # Execute script with void environment
    proc = sp.Popen('. ' + fname + ' &> /dev/null; env', shell=True,
      stdout=sp.PIPE, env={})
    result = readFile(proc.stdout)

    # Remove useless variables
    for k in ('SHLVL', 'PWD', '_'):
        if result.has_key(k):
            del result[k]
    return result

def getEnvFromNode(node, envName):
    envs = node.getElementsByTagName(envName)

    # Check number of envs
    if len(envs) > 1:
        errstr = "Error: no more than one " + envName + " element is allowed!"
        raise Exception(errstr)
    elif len(envs) < 1:
        return {}

    e = envs[0]

    # Check attribute "append"
    if (e.attributes.has_key('append')):
        append = e.attributes['append'].value == '1'
    else:
        append = False

    if append:
        env = os.environ
    else:
        env = {}

    # Check attribute script
    # the script is run with a void environment
    if (e.attributes.has_key('script')):
        for k,v in readScript(e.getAttribute('script')).items():
            env[k] = v

    # Check attribute file
    # the file must contain lines with key=value pairs (each line one pair)
    if (e.attributes.has_key('file')):
        for k,v in readFile(e.getAttribute('file')).items():
            env[k] = v

    # Get Variables
    for v in e.getElementsByTagName('var'):
        envname = v.getAttribute('name')
        envvalue = v.firstChild.data
        env[envname] = envvalue

    return env


def parseConf(fname):
    testNodes = xml.dom.minidom.parse(fname).getElementsByTagName('test')

    tests = {}

    for t in testNodes:
        tid = t.getAttribute('id')

        # Get description
        descr = None
        if len(t.getElementsByTagName('descr')) != 0:
            descr = t.getElementsByTagName('descr')[0].firstChild.data

        # Get package
        pkg = portage.catpkgsplit(
          t.getElementsByTagName('pkg')[0].firstChild.data)
        normPkg = pu.normalize_cpv(pkg)

        # Skip implementations
        # TODO: add regexp
        skip = [i.firstChild.data for i in t.getElementsByTagName('skip')]

        # Requirements
        requires = {}
        for i in t.getElementsByTagName('required'):
            requires[i.getAttribute('name').strip()] = i.firstChild.data.strip()

        # Environments
        dependenv = getEnvFromNode(t, 'dependenv')
        emergeenv = getEnvFromNode(t, 'emergeenv')
        compileenv = getEnvFromNode(t, 'compileenv')
        runenv = getEnvFromNode(t, 'runenv')

        # Adjust PATH
        if runenv.has_key('PATH'):
            runenv['PATH'] += ':' + os.environ['PATH']
        else:
            runenv['PATH'] = os.environ['PATH']

        # Build test dictionary
        tests[tid] = dict(
          descr = descr,
          package = pkg,
          normalizedPackage = normPkg,
          skip = skip,
          requires = requires,

          dependenv = dependenv,
          emergeenv = emergeenv,
          compileenv = compileenv,
          runenv = runenv,

          pkgdir = pjoin(cfg.pkgsdir, tid),
          archive = pjoin(cfg.pkgsdir, tid, normPkg+'.tbz2'),
          root = pjoin(cfg.rootsdir, tid),
          testdir = pjoin(cfg.testsdir, tid),
          logdir = pjoin(cfg.logdir, tid)
        )

    return tests