source: rtems-tools/rtemstoolkit/linux.py @ 7e5cdea

Last change on this file since 7e5cdea was 7e5cdea, checked in by Chris Johns <chrisj@…>, on Nov 23, 2018 at 4:02:52 AM

rtemstoolkit: Add unit testing for the python modules

  • Add support to run the unit tests for the rtemstoolkit python modules from waf. Enter './waf test' for the tests to be run on python2 and python3.
  • Update the importing of rtemstoolkit modules to the standard method which works on python2 and python3.
  • Update the README.
  • Property mode set to 100644
File size: 5.9 KB
Line 
1#
2# RTEMS Tools Project (http://www.rtems.org/)
3# Copyright 2010-2017 Chris Johns (chrisj@rtems.org)
4# All rights reserved.
5#
6# This file is part of the RTEMS Tools package in 'rtems-tools'.
7#
8# Redistribution and use in source and binary forms, with or without
9# modification, are permitted provided that the following conditions are met:
10#
11# 1. Redistributions of source code must retain the above copyright notice,
12# this list of conditions and the following disclaimer.
13#
14# 2. Redistributions in binary form must reproduce the above copyright notice,
15# this list of conditions and the following disclaimer in the documentation
16# and/or other materials provided with the distribution.
17#
18# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
22# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28# POSSIBILITY OF SUCH DAMAGE.
29#
30
31#
32# This code is based on what ever doco about spec files I could find and
33# RTEMS project's spec files.
34#
35
36import os
37import platform
38
39from rtemstoolkit import execute
40from rtemstoolkit import path
41
42def cpus():
43    processors = '/bin/grep processor /proc/cpuinfo'
44    e = execute.capture_execution()
45    exit_code, proc, output = e.shell(processors)
46    ncpus = 0
47    if exit_code == 0:
48        try:
49            for l in output.split('\n'):
50                count = l.split(':')[1].strip()
51                if int(count) > ncpus:
52                    ncpus = int(count)
53        except:
54            pass
55    return ncpus + 1
56
57def overrides():
58    uname = os.uname()
59    smp_mflags = ''
60    ncpus = '%d' % cpus()
61    if uname[4].startswith('arm'):
62        cpu = 'arm'
63    else:
64        cpu = uname[4]
65
66    defines = {
67        '_ncpus':         ('none',    'none',     ncpus),
68        '_os':            ('none',    'none',     'linux'),
69        '_host':          ('triplet', 'required', cpu + '-linux-gnu'),
70        '_host_vendor':   ('none',    'none',     'gnu'),
71        '_host_os':       ('none',    'none',     'linux'),
72        '_host_cpu':      ('none',    'none',     cpu),
73        '_host_alias':    ('none',    'none',     '%{nil}'),
74        '_host_arch':     ('none',    'none',     cpu),
75        '_usr':           ('dir',     'required', '/usr'),
76        '_var':           ('dir',     'required', '/var'),
77        '__bzip2':        ('exe',     'required', '/usr/bin/bzip2'),
78        '__gzip':         ('exe',     'required', '/bin/gzip'),
79        '__tar':          ('exe',     'required', '/bin/tar')
80        }
81
82    # Works for LSB distros
83    try:
84        distro = platform.dist()[0]
85        distro_ver = float(platform.dist()[1])
86    except ValueError:
87        # Non LSB distro found, use failover"
88        pass
89
90    # Non LSB - fail over to issue
91    if distro == '':
92        try:
93            issue = open('/etc/issue').read()
94            distro = issue.split(' ')[0]
95            distro_ver = float(issue.split(' ')[2])
96        except:
97            pass
98
99    # Manage distro aliases
100    if distro in ['centos']:
101        distro = 'redhat'
102    elif distro in ['fedora']:
103        if distro_ver < 17:
104            distro = 'redhat'
105    elif distro in ['centos', 'fedora']:
106        distro = 'redhat'
107    elif distro in ['Ubuntu', 'ubuntu']:
108        distro = 'debian'
109    elif distro in ['Arch']:
110        distro = 'arch'
111    elif distro in ['SuSE']:
112        distro = 'suse'
113
114    variations = {
115        'debian' : { '__bzip2':        ('exe',     'required', '/bin/bzip2'),
116                     '__chgrp':        ('exe',     'required', '/bin/chgrp'),
117                     '__chown':        ('exe',     'required', '/bin/chown'),
118                     '__grep':         ('exe',     'required', '/bin/grep'),
119                     '__sed':          ('exe',     'required', '/bin/sed') },
120        'redhat' : { '__bzip2':        ('exe',     'required', '/bin/bzip2'),
121                     '__chgrp':        ('exe',     'required', '/bin/chgrp'),
122                     '__chown':        ('exe',     'required', '/bin/chown'),
123                     '__install_info': ('exe',     'required', '/sbin/install-info'),
124                     '__grep':         ('exe',     'required', '/bin/grep'),
125                     '__sed':          ('exe',     'required', '/bin/sed'),
126                     '__touch':        ('exe',     'required', '/bin/touch') },
127        'fedora' : { '__chown':        ('exe',     'required', '/usr/bin/chown'),
128                     '__install_info': ('exe',     'required', '/usr/sbin/install-info') },
129        'arch'   : { '__gzip':         ('exe',     'required', '/usr/bin/gzip'),
130                     '__chown':        ('exe',     'required', '/usr/bin/chown') },
131        'suse'   : { '__chgrp':        ('exe',     'required', '/usr/bin/chgrp'),
132                     '__chown':        ('exe',     'required', '/usr/sbin/chown') },
133        }
134
135    if distro in variations:
136        for v in variations[distro]:
137            if path.exists(variations[distro][v][2]):
138                defines[v] = variations[distro][v]
139
140    defines['_build']        = defines['_host']
141    defines['_build_vendor'] = defines['_host_vendor']
142    defines['_build_os']     = defines['_host_os']
143    defines['_build_cpu']    = defines['_host_cpu']
144    defines['_build_alias']  = defines['_host_alias']
145    defines['_build_arch']   = defines['_host_arch']
146
147    return defines
148
149if __name__ == '__main__':
150    import pprint
151    pprint.pprint(cpus())
152    pprint.pprint(overrides())
Note: See TracBrowser for help on using the repository browser.