From: Ansgar Burchardt Date: Thu, 19 Jun 2014 14:34:09 +0000 (+0200) Subject: Implement architecture matching rules. X-Git-Url: https://git.decadent.org.uk/gitweb/?p=dak.git;a=commitdiff_plain;h=9e28920e7356504ca30dfc4e2717d1e61b324500 Implement architecture matching rules. --- diff --git a/daklib/architecture.py b/daklib/architecture.py new file mode 100644 index 00000000..110d522f --- /dev/null +++ b/daklib/architecture.py @@ -0,0 +1,96 @@ +"""architecture matching + +@copyright: 2014, Ansgar Burchardt +@license: GPL-2+ +""" + +# This program 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 2 of the License, or +# (at your option) any later version. +# +# This program 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 this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + +def _load_table(path): + table = [] + with open(path, 'r') as fh: + for line in fh: + if not line or line.startswith('#'): + continue + table.append(line.split()) + return table + +_cached_cputable = None +def _cputable(): + global _cached_cputable + if _cached_cputable is None: + _cached_cputable = _load_table('/usr/share/dpkg/cputable') + return _cached_cputable + +_cached_arch2triplet = None +_cached_triplet2arch = None +def _triplettable(): + global _cached_arch2triplet, _cached_triplet2arch + if _cached_arch2triplet is None or _cached_triplet2arch is None: + table = _load_table('/usr/share/dpkg/triplettable') + arch2triplet = {} + triplet2arch = {} + for row in table: + if '' in row[0] or '' in row[1]: + for cpu in _cputable(): + replaced_row = [ column.replace('', cpu[0]) for column in row ] + arch2triplet[replaced_row[1]] = replaced_row[0] + triplet2arch[replaced_row[0]] = replaced_row[1] + else: + arch2triplet[row[1]] = row[0] + triplet2arch[row[0]] = row[1] + _cached_arch2triplet = arch2triplet + _cached_triplet2arch = triplet2arch + return _cached_triplet2arch, _cached_arch2triplet + +class InvalidArchitecture(Exception): + pass + +def Debian_arch_to_Debian_triplet(arch): + parts = arch.split('-') + + # Handle architecture wildcards + if 'any' in parts: + if len(parts) == 3: + return parts + elif len(parts) == 2: + return 'any', parts[0], parts[1] + else: + return 'any', 'any', 'any' + + triplet = _triplettable()[1].get(arch, None) + if triplet is None: + return None + return triplet.split('-', 2) + +def match_architecture(arch, wildcard): + # 'all' has no valid triplet + if arch == 'all' or wildcard == 'all': + return arch == wildcard + if wildcard is 'any' or arch == wildcard: + return True + + triplet_arch = Debian_arch_to_Debian_triplet(arch) + triplet_wildcard = Debian_arch_to_Debian_triplet(wildcard) + + if len(triplet_arch) != 3: + raise InvalidArchitecture('{0} is not a valid architecture name'.format(arch)) + if len(triplet_wildcard) != 3: + raise InvalidArchitecture('{0} is not a valid architecture name or wildcard'.format(wildcard)) + + for i in range(0,3): + if triplet_arch[i] != triplet_wildcard[i] and triplet_wildcard[i] != 'any': + return False + return True diff --git a/tests/test_architecture.py b/tests/test_architecture.py new file mode 100755 index 00000000..8fd2d3a5 --- /dev/null +++ b/tests/test_architecture.py @@ -0,0 +1,49 @@ +#! /usr/bin/python +# +# Copyright (C) 2014, Ansgar Burchardt +# +# This program 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 2 of the License, or +# (at your option) any later version. +# +# This program 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 this program; if not, write to the Free Software Foundation, Inc., +# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + +from base_test import DakTestCase + +import unittest + +from daklib.architecture import match_architecture + +class MatchArchitecture(DakTestCase): + def testEqual(self): + self.assert_(match_architecture('amd64', 'amd64')) + self.assert_(not match_architecture('amd64', 'i386')) + self.assert_(match_architecture('kfreebsd-amd64', 'kfreebsd-amd64')) + self.assert_(not match_architecture('kfreebsd-amd64', 'amd64')) + def testAny(self): + self.assert_(match_architecture('amd64', 'any')) + self.assert_(match_architecture('amd64', 'any-amd64')) + self.assert_(match_architecture('x32', 'any-amd64')) + self.assert_(match_architecture('kfreebsd-amd64', 'any-amd64')) + self.assert_(not match_architecture('amd64', 'any-i386')) + + self.assert_(match_architecture('kfreebsd-amd64', 'kfreebsd-any')) + self.assert_(not match_architecture('amd64', 'kfreebsd-any')) + def testAll(self): + self.assert_(match_architecture('all', 'all')) + + self.assert_(not match_architecture('amd64', 'all')) + self.assert_(not match_architecture('all', 'amd64')) + + self.assert_(not match_architecture('all', 'any')) + +if __name__ == '__main__': + unittest.main()