change greylist such that the underlying logic will choose the 8139too driver over...
[pypcilib.git] / pypci.py
1 # Copyright 2008 The Trustees of Princeton University
2 # Author: Daniel Hokka Zakrisson
3 # $Id$
4 # vim:ts=4:expandtab
5
6 # Redistribution and use in source and binary forms, with or without
7 # modification, are permitted provided that the following conditions
8 # are met: 
9
10 # * Redistributions of source code must retain the above copyright
11 # notice, this list of conditions and the following disclaimer.
12 #       
13 # * Redistributions in binary form must reproduce the above
14 # copyright notice, this list of conditions and the following
15 # disclaimer in the documentation and/or other materials provided
16 # with the distribution.
17 #       
18 # * Neither the name of the copyright holder nor the names of its
19 # contributors may be used to endorse or promote products derived
20 # from this software without specific prior written permission.
21 #       
22 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
23 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
24 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
25 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL PRINCETON
26 # UNIVERSITY OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
27 # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
28 # BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
29 # OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
30 # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY
32 # WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
33 # POSSIBILITY OF SUCH DAMAGE. 
34
35 PCI_BASE_CLASS_NETWORK=0x02L
36 PCI_BASE_CLASS_STORAGE=0x01L
37 PCI_ANY=0xffffffffL
38
39 def get_devices():
40     """ This is a replacement to the pypciscan library."""
41     import os
42     pci_cmd = os.popen("""/sbin/lspci -nvm | sed -e 's/\t/ /g' -e 's/ Class / /' -e 's/^/"/' -e 's/$/"/' -e 's/$/,/' -e 's/^"",$/],[/'""", 'r')
43     pci_str = "[" + pci_cmd.read() + "]"
44     pci_list = eval(pci_str)
45
46     pci_devlist = []
47     # convert each entry into a dict. and convert strings to ints.
48     for dev in pci_list:
49         rec = {}
50         for field in dev:
51             s = field.split(":")
52             if len(s) > 2:
53                 # There are two 'device' fields in the output. Append
54                 # 'addr' for the bus address, identified by the extra ':'.
55                 end=":".join(s[1:])
56                 value = end.strip()
57                 key = s[0].lower() + "addr"
58             else:
59                 value = int(s[1].strip(), 16)
60                 key = s[0].lower()
61
62             rec[key] = value
63
64         pci_devlist.append(rec)
65
66     ret = {}
67     # convert this list of devices into the format expected by the
68     # consumer of get_devices()
69     for dev in pci_devlist:
70         if 'deviceaddr' not in dev:
71             continue
72
73         subdev = dev.get('sdevice',PCI_ANY)
74         subvend = dev.get('svendor',PCI_ANY)
75         progif = dev.get('progif',0)
76
77         value = (dev['vendor'], dev['device'], subvend, subdev, dev['class'] << 8 | progif)
78         ret[dev['deviceaddr']] = value
79
80     return  ret
81
82 # for convenience for the clients of pypci
83 import pypcimap