#!/usr/bin/python
# Copyright 2015 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# pylint: disable=line-too-long

"""Routines for handling Ethernet OUI information.
"""

# The IEEE publishes an enormous list of registered Ethernet OUIs,
# but we don't benefit from having that entire list here. Instead
# we track the OUIs which let us distinguish devices which are
# otherwise very similar, such as distinguishing LG G2 from Samsung
# Galaxy S4.
database = {
    '00:bb:3a': ['amazon'],
    '0c:47:c9': ['amazon'],
    '10:ae:60': ['amazon'],
    '28:ef:01': ['amazon'],
    '44:65:0d': ['amazon'],
    '74:75:48': ['amazon'],
    '74:c2:46': ['amazon'],
    '84:d6:d0': ['amazon'],
    'a0:02:dc': ['amazon'],
    'f0:27:2d': ['amazon'],
    'f0:4f:7c': ['amazon'],
    'f0:a2:25': ['amazon'],

    '08:60:6e': ['asus'],
    '08:62:66': ['asus'],
    '10:bf:48': ['asus'],
    '1c:87:2c': ['asus'],
    '2c:56:dc': ['asus'],
    '30:85:a9': ['asus'],
    '50:46:5d': ['asus'],
    '5c:ff:35': ['asus'],
    '60:a4:4c': ['asus'],
    '74:d0:2b': ['asus'],
    'ac:22:0b': ['asus'],
    'bc:ee:7b': ['asus'],
    'd8:50:e6': ['asus'],
    'f8:32:e4': ['asus'],

    '58:67:1a': ['barnes&noble'],

    '2c:b0:5d': ['dish'],

    '30:8c:fb': ['dropcam'],

    '00:1a:11': ['google'],
    '3c:5a:b4': ['google'],
    '54:60:09': ['google'],
    '94:95:a0': ['google'],
    '94:eb:2c': ['google'],
    'a4:77:33': ['google'],
    'f4:03:04': ['google'],
    'f4:f5:d8': ['google'],
    'f4:f5:e8': ['google'],
    'f8:8f:ca': ['google'],

    # These are registered to AzureWave, but used for Chromecast v1.
    '6c:ad:f8': ['azurewave', 'google'],
    '80:d2:1d': ['azurewave', 'google'],
    'b0:ee:45': ['azurewave', 'google'],
    'd0:e7:82': ['azurewave', 'google'],

    '44:61:32': ['ecobee'],

    '00:23:76': ['htc'],
    '00:ee:bd': ['htc'],
    '18:87:96': ['htc'],
    '1c:b0:94': ['htc'],
    '2c:8a:72': ['htc'],
    '38:e7:d8': ['htc'],
    '50:2e:5c': ['htc'],
    '64:a7:69': ['htc'],
    '7c:61:93': ['htc'],
    '80:01:84': ['htc'],
    '80:7a:bf': ['htc'],
    '84:7a:88': ['htc'],
    '90:e7:c4': ['htc'],
    'a0:f4:50': ['htc'],
    'ac:37:43': ['htc'],
    'b4:ce:f6': ['htc'],
    'd8:b3:77': ['htc'],
    'e8:99:c4': ['htc'],

    '00:1e:b2': ['lg'],
    '00:34:da': ['lg'],
    '0c:48:85': ['lg'],
    '10:68:3f': ['lg'],
    '2c:54:cf': ['lg'],
    '34:4d:f7': ['lg'],
    '34:fc:ef': ['lg'],
    '3c:bd:d8': ['lg'],
    '40:b0:fa': ['lg'],
    '48:59:29': ['lg'],
    '58:3f:54': ['lg'],
    '5c:70:a3': ['lg'],
    '64:89:9a': ['lg'],
    '64:bc:0c': ['lg'],
    '78:f8:82': ['lg'],
    '88:07:4b': ['lg'],
    '88:c9:d0': ['lg'],
    '8c:3a:e3': ['lg'],
    'a0:39:f7': ['lg'],
    'a0:91:69': ['lg'],
    'ac:0d:1b': ['lg'],
    'bc:f5:ac': ['lg'],
    'c4:43:8f': ['lg'],
    'c4:9a:02': ['lg'],
    'cc:fa:00': ['lg'],
    'e8:5b:5b': ['lg'],
    'f8:95:c7': ['lg'],
    'f8:a9:d0': ['lg'],

    'd0:73:d5': ['lifx'],

    '24:fd:52': ['liteon', 'sling'],

    '00:0d:3a': ['microsoft'],
    '00:12:5a': ['microsoft'],
    '00:17:fa': ['microsoft'],
    '00:1d:d8': ['microsoft'],
    '00:22:48': ['microsoft'],
    '00:25:ae': ['microsoft'],
    '00:50:f2': ['microsoft'],
    '28:18:78': ['microsoft'],
    '30:59:b7': ['microsoft'],
    '4c:0b:be': ['microsoft'],
    '50:1a:c5': ['microsoft'],
    '58:82:a8': ['microsoft'],
    '60:45:bd': ['microsoft'],
    '7c:1e:52': ['microsoft'],
    '7c:ed:8d': ['microsoft'],
    '98:5f:d3': ['microsoft'],
    'b4:ae:2b': ['microsoft'],
    'c0:33:5e': ['microsoft'],

    '14:1a:a3': ['motorola'],
    '14:30:c6': ['motorola'],
    '1c:56:fe': ['motorola'],
    '24:da:9b': ['motorola'],
    '3c:43:8e': ['motorola'],
    '40:78:6a': ['motorola'],
    '44:80:eb': ['motorola'],
    '5c:51:88': ['motorola'],
    '60:be:b5': ['motorola'],
    '80:6c:1b': ['motorola'],
    '84:10:0d': ['motorola'],
    '90:68:c3': ['motorola'],
    '98:4b:4a': ['motorola'],
    '9c:d9:17': ['motorola'],
    'cc:c3:ea': ['motorola'],
    'e4:90:7e': ['motorola'],
    'e8:91:20': ['motorola'],
    'ec:88:92': ['motorola'],
    'f8:7b:7a': ['motorola'],
    'f8:cf:c5': ['motorola'],
    'f8:e0:79': ['motorola'],
    'f8:f1:b6': ['motorola'],

    '00:26:e8': ['murata', 'samsung'],
    '00:ae:fa': ['murata', 'samsung'],
    '10:a5:d0': ['murata', 'samsung'],
    '14:7d:c5': ['murata', 'samsung'],
    '1c:99:4c': ['murata', 'samsung'],
    '20:02:af': ['murata', 'samsung'],
    '40:f3:08': ['murata', 'samsung'],
    '44:a7:cf': ['murata', 'samsung'],
    '5c:da:d4': ['murata', 'samsung'],
    '5c:f8:a1': ['murata', 'samsung'],
    '60:f1:89': ['murata', 'samsung'],
    '78:4b:87': ['murata', 'samsung'],
    '90:b6:86': ['murata', 'samsung'],
    '98:f1:70': ['murata', 'samsung'],
    'f0:27:65': ['murata', 'samsung'],
    'fc:c2:de': ['murata', 'samsung'],
    'fc:db:b3': ['murata', 'samsung'],

    '18:b4:30': ['nest', 'dropcam'],

    'c0:ee:fb': ['oneplus'],

    '00:15:99': ['samsung'],
    '00:26:37': ['samsung'],
    '08:d4:2b': ['samsung'],
    '08:ec:a9': ['samsung'],
    '10:30:47': ['samsung'],
    '14:32:d1': ['samsung'],
    '14:49:e0': ['samsung'],
    '18:21:95': ['samsung'],
    '18:22:7e': ['samsung'],
    '20:55:31': ['samsung'],
    '20:6e:9c': ['samsung'],
    '24:4b:81': ['samsung'],
    '28:27:bf': ['samsung'],
    '28:ba:b5': ['samsung'],
    '2c:0e:3d': ['samsung'],
    '2c:ae:2b': ['samsung'],
    '30:19:66': ['samsung'],
    '34:23:ba': ['samsung'],
    '38:2d:e8': ['samsung'],
    '38:aa:3c': ['samsung'],
    '38:d4:0b': ['samsung'],
    '3c:8b:fe': ['samsung'],
    '3c:a1:0d': ['samsung'],
    '3c:bb:fd': ['samsung'],
    '40:0e:85': ['samsung'],
    '48:5a:3f': ['samsung', 'wisol'],
    '4c:3c:16': ['samsung'],
    '4c:bc:a5': ['samsung'],
    '50:cc:f8': ['samsung'],
    '54:40:ad': ['samsung'],
    '54:88:0e': ['samsung'],
    '5c:0a:5b': ['samsung'],
    '5c:f6:dc': ['samsung'],
    '60:af:6d': ['samsung'],
    '6c:2f:2c': ['samsung'],
    '6c:83:36': ['samsung'],
    '78:40:e4': ['samsung'],
    '78:d6:f0': ['samsung'],
    '78:bd:bc': ['samsung'],
    '7c:0b:c6': ['samsung'],
    '80:65:6d': ['samsung'],
    '84:11:9e': ['samsung'],
    '84:25:db': ['samsung'],
    '84:2e:27': ['samsung'],
    '84:38:38': ['samsung'],
    '84:55:a5': ['samsung'],
    '88:32:9b': ['samsung'],
    '8c:77:12': ['samsung'],
    '90:18:7c': ['samsung'],
    '90:f1:aa': ['samsung'],
    '94:35:0a': ['samsung'],
    '94:b1:0a': ['samsung'],
    'a0:0b:ba': ['samsung'],
    'a4:08:ea': ['samsung'],
    'a8:06:00': ['samsung'],
    'ac:36:13': ['samsung'],
    'ac:5f:3e': ['samsung'],
    'b0:47:bf': ['samsung'],
    'b0:c5:59': ['samsung'],
    'b0:df:3a': ['samsung'],
    'b0:ec:71': ['samsung'],
    'b4:07:f9': ['samsung'],
    'b4:79:a7': ['samsung'],
    'b8:57:d8': ['samsung'],
    'b8:5a:73': ['samsung'],
    'b8:5e:7b': ['samsung'],
    'bc:14:85': ['samsung'],
    'bc:20:a4': ['samsung'],
    'bc:72:b1': ['samsung'],
    'bc:8c:cd': ['samsung'],
    'bc:e6:3f': ['samsung'],
    'c0:bd:d1': ['samsung'],
    'c4:42:02': ['samsung'],
    'c4:57:6e': ['samsung'],
    'c4:73:1e': ['samsung'],
    'cc:07:ab': ['samsung'],
    'cc:3a:61': ['samsung'],
    'd0:22:be': ['samsung'],
    'd0:59:e4': ['samsung'],
    'd8:90:e8': ['samsung'],
    'e0:99:71': ['samsung'],
    'e0:db:10': ['samsung'],
    'e4:12:1d': ['samsung'],
    'e4:58:b8': ['samsung'],
    'e4:92:fb': ['samsung'],
    'e8:3a:12': ['samsung'],
    'e8:50:8b': ['samsung'],
    'ec:1f:72': ['samsung'],
    'ec:9b:f3': ['samsung'],
    'f0:25:b7': ['samsung'],
    'f4:09:d8': ['samsung'],
    'f8:04:2e': ['samsung'],
    'fc:19:10': ['samsung'],
    'fc:f1:36': ['samsung'],

    '00:d9:d1': ['sony'],
    '28:0d:fc': ['sony'],
    '30:17:c8': ['sony'],
    '40:b8:37': ['sony'],
    '58:48:22': ['sony'],
    'b4:52:7e': ['sony'],
    'f8:d0:ac': ['sony'],

    '10:08:c1': ['toshiba'],

    '00:6b:9e': ['vizio'],
    'a4:8d:3b': ['vizio'],

    'b4:79:a7': ['wink'],

    '00:24:e4': ['withings'],

    '64:cc:2e': ['xiaomi'],
}


def LookupOUI(mac):
  """Return a list of manufacturer(s) from a MAC address."""
  mac = mac.lower().split(':')
  oui = ':'.join(mac[0:3])
  return database.get(oui, [])