name should not have hrefs + cosmetic
[myplc.git] / db-config
index 4f994a6..d522bfd 100755 (executable)
--- a/db-config
+++ b/db-config
@@ -7,8 +7,8 @@
 # Mark Huang <mlhuang@cs.princeton.edu>
 # Copyright (C) 2006 The Trustees of Princeton University
 #
-# $Id$
-#
+# $Id: db-config 7454 2007-12-11 18:55:00Z faiyaza $
+# $HeadURL$
 
 from plc_config import PLCConfiguration
 import sys
@@ -142,18 +142,6 @@ def main():
          'error_cmd': '',
          'ignore_cmd_errors': False,
          'always_update': False},
-        {'enabled': True,
-         'source': 'PlanetLabConf/keys.php?role=admin',
-         'dest': '/home/pl_admin/.ssh/authorized_keys',
-         'file_permissions': '644',
-         'file_owner': 'pl_admin',
-         'file_group': 'pl_admin',
-         'preinstall_cmd': 'grep -q pl_admin /etc/passwd',
-         'postinstall_cmd': '/bin/chmod 700 /home/pl_admin/.ssh',
-         'error_cmd': '',
-         'ignore_cmd_errors': False,
-         'always_update': False},
-
         # Log rotation configuration
         {'enabled': True,
          'source': 'PlanetLabConf/logrotate.conf',
@@ -250,52 +238,7 @@ def main():
          'ignore_cmd_errors': False,
          'always_update': False},
 
-        # Node Manager configuration
-        {'enabled': True,
-         'source': 'PlanetLabConf/pl_nm.conf',
-         'dest': '/etc/planetlab/pl_nm.conf',
-         'file_permissions': '644',
-         'file_owner': 'root',
-         'file_group': 'root',
-         'preinstall_cmd': '',
-         'postinstall_cmd': '/etc/init.d/pl_nm restart',
-         'error_cmd': '',
-         'ignore_cmd_errors': False,
-         'always_update': False},
-        {'enabled': True,
-         'source': 'PlanetLabConf/RootResources/plc_slice_pool.php',
-         'dest': '/home/pl_nm/RootResources/plc_slice_pool',
-         'file_permissions': '644',
-         'file_owner': 'pl_nm',
-         'file_group': 'pl_nm',
-         'preinstall_cmd': '',
-         'postinstall_cmd': '',
-         'error_cmd': '',
-         'ignore_cmd_errors': False,
-         'always_update': False},
-        {'enabled': True,
-         'source': 'PlanetLabConf/RootResources/pl_conf.py',
-         'dest': '/home/pl_nm/RootResources/pl_conf',
-         'file_permissions': '644',
-         'file_owner': 'pl_nm',
-         'file_group': 'pl_nm',
-         'preinstall_cmd': '',
-         'postinstall_cmd': '/etc/init.d/pl_nm restart',
-         'error_cmd': '',
-         'ignore_cmd_errors': False,
-         'always_update': False},
-        {'enabled': True,
-         'source': 'PlanetLabConf/RootResources/pl_netflow.py',
-         'dest': '/home/pl_nm/RootResources/pl_netflow',
-         'file_permissions': '644',
-         'file_owner': 'pl_nm',
-         'file_group': 'pl_nm',
-         'preinstall_cmd': '',
-         'postinstall_cmd': '',
-         'error_cmd': '',
-         'ignore_cmd_errors': False,
-         'always_update': False},
-
+        # XXX Required for old Node Manager
         # Proper configuration
         {'enabled': True,
          'source': 'PlanetLabConf/propd.conf',
@@ -309,6 +252,7 @@ def main():
          'ignore_cmd_errors': True,
          'always_update': False},
 
+        # XXX Required for old Node Manager
         # Bandwidth cap
         {'enabled': True,
          'source': 'PlanetLabConf/bwlimit.php',
@@ -317,7 +261,7 @@ def main():
          'file_owner': 'root',
          'file_group': 'root',
          'preinstall_cmd': '',
-         'postinstall_cmd': '/etc/init.d/pl_nm restart',
+         'postinstall_cmd': '',
          'error_cmd': '',
          'ignore_cmd_errors': True,
          'always_update': False},
@@ -357,7 +301,7 @@ def main():
          'postinstall_cmd': '/sbin/iptables-restore --noflush < /etc/planetlab/blacklist',
          'error_cmd': '',
          'ignore_cmd_errors': True,
-         'always_update': True},
+         'always_update': False},
 
         # /etc/issue
         {'enabled': True,
@@ -383,7 +327,7 @@ def main():
          'postinstall_cmd': '/sbin/sysctl -e -p /etc/sysctl.conf',
          'error_cmd': '',
          'ignore_cmd_errors': False,
-         'always_update': True},
+         'always_update': False},
 
         # Sendmail configuration
         {'enabled': True,
@@ -434,6 +378,7 @@ def main():
          'always_update': False},
 
         # Ping of death configuration
+        # the 'restart' postcommand doesn't work, b/c the pod script doesn't support it.
         {'enabled': True,
          'source': 'PlanetLabConf/ipod.conf.php',
          'dest': '/etc/ipod.conf',
@@ -441,7 +386,7 @@ def main():
          'file_owner': 'root',
          'file_group': 'root',
          'preinstall_cmd': '',
-         'postinstall_cmd': '',
+         'postinstall_cmd': '/etc/init.d/pod start',
          'error_cmd': '',
          'ignore_cmd_errors': False,
          'always_update': False},
@@ -505,54 +450,65 @@ def main():
          'min_role_id': 10},
 
         # CPU share
-        {'name': "cpu_min",
-         'description': "Minimum CPU share (ms/s)",
+        {'name': "cpu_pct",
+         'description': "Reserved CPU percent",
          'min_role_id': 10},
         {'name': "cpu_share",
          'description': "Number of CPU shares",
          'min_role_id': 10},
 
         # Bandwidth limits
-        {'name': "net_min",
-         'description': "Minimum bandwidth (bps)",
+        {'name': "net_min_rate",
+         'description': "Minimum bandwidth (kbps)",
          'min_role_id': 10},
-        {'name': "net_max",
-         'description': "Maximum bandwidth (bps)",
+        {'name': "net_max_rate",
+         'description': "Maximum bandwidth (kbps)",
          'min_role_id': 10},
-        {'name': "net_avg",
-         'description': "Average bandwidth (bps)",
+        {'name': "net_i2_min_rate",
+         'description': "Minimum bandwidth over I2 routes (kbps)",
          'min_role_id': 10},
-        {'name': "net_share",
-         'description': "Number of bandwidth shares",
+        {'name': "net_i2_max_rate",
+         'description': "Maximum bandwidth over I2 routes (kbps)",
          'min_role_id': 10},
-        {'name': "net2_min",
-         'description': "Minimum bandwidth over routes exempt from node bandwidth limits (bps)",
+        {'name': "net_max_kbyte",
+         'description': "Maximum daily network Tx KByte limit.",
          'min_role_id': 10},
-        {'name': "net2_max",
-         'description': "Maximum bandwidth over routes exempt from node bandwidth limits (bps)",
+        {'name': "net_thresh_kbyte",
+         'description': "KByte limit before warning and throttling.",
          'min_role_id': 10},
-        {'name': "net2_avg",
-         'description': "Average bandwidth over routes exempt from node bandwidth limits (bps)",
+        {'name': "net_i2_max_kbyte",
+         'description': "Maximum daily network Tx KByte limit to I2 hosts.",
          'min_role_id': 10},
-        {'name': "net2_share",
-         'description': "Number of bandwidth shares over routes exempt from node bandwidth limits",
+        {'name': "net_i2_thresh_kbyte",
+         'description': "KByte limit to I2 hosts before warning and throttling.",
          'min_role_id': 10},
-
+        {'name': "net_share",
+         'description': "Number of bandwidth shares",
+         'min_role_id': 10},
+        {'name': "net_i2_share",
+         'description': "Number of bandwidth shares over I2 routes",
+         'min_role_id': 10},
         # Disk quota
         {'name': "disk_max",
          'description': "Disk quota (1k disk blocks)",
          'min_role_id': 10},
 
-        # Special attributes applicable to Slice Creation Service (pl_conf) slice
-        {'name': "plc_slice_type",
-         'description': "Type of slice rspec to be created",
-         'min_role_id': 20},
-        {'name': "plc_agent_version",
-         'description': "Version of PLC agent (slice creation service) software to be deployed",
+        # Proper operations
+        {'name': "proper_op",
+         'description': "Proper operation (e.g. bind_socket)",
+         'min_role_id': 10},
+
+        # VServer capabilities 
+        {'name': "capabilities",
+         'description': "VServer bcapabilities (separate by commas)",
          'min_role_id': 10},
-        {'name': "plc_ticket_pubkey",
-         'description': "Public key used to verify PLC-signed tickets",
+
+               # Vsys
+        {'name': "vsys",
+         'description': "Bind vsys script fd's to a slice's vsys directory.",
          'min_role_id': 10}
         ]
 
     # Get list of existing attribute types
@@ -566,51 +522,108 @@ def main():
         else:
             UpdateSliceAttributeType(default_attribute_type['name'], default_attribute_type)
 
-    # Get contents of SSL public certificate used for signing slice tickets
-    try:
-        plc_ticket_pubkey = ""
-        for line in file(plc_ma_sa['ca_ssl_key_pub']):
-            # Skip comments
-            if line[0:5] != "-----":
-                # XXX The embedded newlines matter, do not strip()!
-                plc_ticket_pubkey += line
-    except:
-        plc_ticket_pubkey = '%KEY%'
+    # Default Initscripts
+    default_initscripts = []
+
+    # Find initscripts and add them to the db
+    for (root, dirs, files) in os.walk("/etc/plc_sliceinitscripts"):
+        for f in files:
+            # Read the file
+            file = open(root + "/" + f, "ro")
+            default_initscripts.append({"name": plc['slice_prefix'] + "_" + f,
+                                        "enabled": True,
+                                        "script": file.read()})
+            file.close()
+
+    # Get list of existing initscripts
+    oldinitscripts = GetInitScripts()
+    oldinitscripts = [script['name'] for script in oldinitscripts]
+
+    for initscript in default_initscripts:
+        if initscript['name'] not in oldinitscripts:  AddInitScript(initscript)
+
+    # Setup default slice attribute types
+    default_setting_types = [
+
+        {'category' : "general",
+         'name' : "ifname",
+         'description': "Set interface name, instead of eth0 or the like",
+         'min_role_id' : 40},
+        {'category' : "general",
+         'name' : "driver",
+         'description': "Use this to specify an alternate driver",
+         'min_role_id' : 40 },
+        {'category' : "general",
+         'name' : "alias",
+         'description': "Allows to reuse an interface as eth0:alias",
+         'min_role_id' : 40},
+
+        {'category' : "hidden",
+         'name' : "backdoor",
+         'description': "For testing new settings",
+         'min_role_id' : 10},
+        ] + [
+        { "category" : "WiFi",
+          "name" : x,
+          "description" : "802.11 %s -- see %s"%(y,z),
+          "min_role_id" : 40 } for (x,y,z) in [
+            ("mode","Mode","iwconfig"),
+            ("essid","ESSID","iwconfig"),
+            ("nw","Network Id","iwconfig"),
+            ("freq","Frequency","iwconfig"),
+            ("channel","Channel","iwconfig"),
+            ("sens","sensitivity threshold","iwconfig"),
+            ("rate","Rate","iwconfig"),
+            ("key","key","iwconfig key"),
+            ("key1","key1","iwconfig key [1]"),
+            ("key2","key2","iwconfig key [2]"),
+            ("key3","key3","iwconfig key [3]"),
+            ("key4","key4","iwconfig key [4]"),
+            ("securitymode","Security mode","iwconfig enc"),
+            ("iwconfig","Additional parameters to iwconfig","ifup-wireless"),
+            ("iwpriv","Additional parameters to iwpriv","ifup-wireless"),
+            ]
+        ]
+
+
+    # Get list of existing attribute types
+    setting_types = GetNodeNetworkSettingTypes()
+    setting_types = [setting_type['name'] for setting_type in setting_types]
+
+    # Create/update default slice setting types
+    for default_setting_type in default_setting_types:
+        if default_setting_type['name'] not in setting_types:
+            AddNodeNetworkSettingType(default_setting_type)
+        else:
+            UpdateNodeNetworkSettingType(default_setting_type['name'], default_setting_type)
 
     # Create/update system slices
     default_slices = [
-        # Required for old Node Manager
-        {'name': "pl_conf",
-         'description': "PlanetLab Slice Creation Service (SCS)",
-         'url': url,
-         'instantiation': "plc-instantiated",
-         # Renew forever
-         'expires': sys.maxint,
-         'attributes': {'plc_slice_type': "VServerSlice",
-                        'plc_agent_version': "1.0",
-                        'plc_ticket_pubkey': plc_ticket_pubkey}},
-
-        # Required for old Node Manager
-        {'name': "pl_conf_vserverslice",
-         'description': "Default attributes for vserver slices",
-         'url': url,
-         'instantiation': "plc-instantiated",
-         # Renew forever
-         'expires': sys.maxint,
-         'attributes': {'cpu_share': "32",
-                        'plc_slice_type': "VServerSlice",
-                        'disk_max': "5000000"}},
-
          # PlanetFlow
         {'name': plc['slice_prefix'] + "_netflow",
          'description': "PlanetFlow Traffic Auditing Service",
          'url': url,
          'instantiation': "plc-instantiated",
-         # Renew forever
-         'expires': sys.maxint,
-         'attributes': {'system': "1", 'vref': "planetflow"}},
+         # Renew forever (minus one day, work around date conversion weirdness)
+         'expires': sys.maxint - (60 * 60 * 24),
+         'attributes': [('system', "1"),
+                        ('vref', "planetflow"),
+                        ('proper_op', "open file=/etc/passwd, flags=r"),
+                        ('proper_op', "create_socket"),
+                        ('proper_op', "bind_socket")]},
+          # Sirius
+        {'name': plc['slice_prefix'] + "_sirius",
+         'description': 'The Sirius Calendar Service.\n\nSirius provides system-wide reservations of 25% CPU and 2Mb/s outgoing\nbandwidth.  Sign up for hour-long slots using the Web GUI at the\nPlanetLab website.\n\nThis slice should not generate traffic external to PlanetLab.\n',
+         'url': url,
+         'instantiation': "plc-instantiated",
+         # Renew forever (minus one day, work around date conversion weirdness)
+         'expires': sys.maxint - (60 * 60 * 24),
+         'attributes': [('system', "1"),
+                        ('net_min_rate', "2000"),
+                        ('cpu_pct', "25"),
+                        ('initscript', plc['slice_prefix'] + "_sirius")]}
         ]
-
+    
     for default_slice in default_slices:
         slices = GetSlices([default_slice['name']])
         if slices:
@@ -621,16 +634,390 @@ def main():
             slice = GetSlices([default_slice['name']])[0]
 
         # Create/update all attributes
-        slice_attributes = {}
+        slice_attributes = []
         if slice['slice_attribute_ids']:
+            # Delete unknown attributes
             for slice_attribute in GetSliceAttributes(slice['slice_attribute_ids']):
-                slice_attributes[slice_attribute['name']] = slice_attribute
+                if (slice_attribute['name'], slice_attribute['value']) \
+                   not in default_slice['attributes']:
+                    DeleteSliceAttribute(slice_attribute['slice_attribute_id'])
+                else:
+                    slice_attributes.append((slice_attribute['name'], slice_attribute['value']))
 
-        for name, value in default_slice['attributes'].iteritems():
-            if name not in slice_attributes:
+        for (name, value) in default_slice['attributes']:
+            if (name, value) not in slice_attributes:
                 AddSliceAttribute(slice['name'], name, value)
-            else:
-                UpdateSliceAttribute(slice_attributes[name]['slice_attribute_id'], value)
+
+    installfailed = """
+Once the node meets these requirements, please reinitiate the install
+by visiting:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
+
+Update the BootState to 'Reinstall', then reboot the node.
+
+If you have already performed this step and are still receiving this
+message, please reply so that we may investigate the problem.
+"""
+
+    # Load default message templates
+    message_templates = [
+        {'message_id': 'Verify account',
+         'subject': "Verify account registration",
+         'template': """
+Please verify that you registered for a %(PLC_NAME)s account with the
+username %(email)s by visiting:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/register.php?id=%(person_id)d&key=%(verification_key)s
+
+If you did not register for a %(PLC_NAME)s account, please ignore this
+message, or contact %(PLC_NAME)s Support <%(PLC_MAIL_SUPPORT_ADDRESS)s>.
+"""
+         },
+
+        {'message_id': 'New PI account',
+         'subject': "New PI account registration from %(first_name)s %(last_name)s <%(email)s> at %(site_name)s",
+         'template': """
+%(first_name)s %(last_name)s <%(email)s> has signed up for a new
+%(PLC_NAME)s account at %(site_name)s and has requested a PI role. PIs
+are responsible for enabling user accounts, creating slices, and
+ensuring that all users abide by the %(PLC_NAME)s Acceptable Use
+Policy.
+
+Only %(PLC_NAME)s administrators may enable new PI accounts. If you
+are a PI at %(site_name)s, please respond and indicate whether this
+registration is acceptable.
+
+To view the request, visit:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/index.php?id=%(person_id)d
+"""
+         },
+
+        {'message_id': 'New account',
+         'subject': "New account registration from %(first_name)s %(last_name)s <%(email)s> at %(site_name)s",
+         'template': """
+%(first_name)s %(last_name)s <%(email)s> has signed up for a new
+%(PLC_NAME)s account at %(site_name)s and has requested the following
+roles: %(roles)s.
+
+To deny the request or enable the account, visit:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/index.php?id=%(person_id)d
+"""
+         },
+
+        {'message_id': 'Password reset requested',
+         'subject': "Password reset requested",
+         'template': """
+Someone has requested that the password of your %(PLC_NAME)s account
+%(email)s be reset. If this person was you, you may continue with the
+reset by visiting:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/reset_password.php?id=%(person_id)d&key=%(verification_key)s
+
+If you did not request that your password be reset, please contact
+%(PLC_NAME)s Support <%(PLC_MAIL_SUPPORT_ADDRESS)s>. Do not quote or
+otherwise include any of this text in any correspondence.
+"""
+         },
+
+        {'message_id': 'Password reset',
+         'subject': "Password reset",
+         'template': """
+The password of your %(PLC_NAME)s account %(email)s has been
+temporarily reset to:
+
+%(password)s
+
+Please change it at as soon as possible by visiting:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/index.php?id=%(person_id)d
+
+If you did not request that your password be reset, please contact
+%(PLC_NAME)s Support <%(PLC_MAIL_SUPPORT_ADDRESS)s>. Do not quote or
+otherwise include any of this text in any correspondence.
+"""
+         },
+
+        # Boot Manager messages
+        {'message_id': "installfinished",
+         'subject': "%(hostname)s completed installation",
+         'template': """
+%(hostname)s just completed installation.
+
+The node should be usable in a couple of minutes if installation was
+successful.
+"""
+         },
+
+        {'message_id': "insufficientdisk",
+         'subject': "%(hostname)s does not have sufficient disk space",
+         'template': """
+%(hostname)s failed to boot because it does not have sufficent disk
+space, or because its disk controller was not recognized.
+
+Please replace the current disk or disk controller or install
+additional disks to meet the current hardware requirements.
+""" + installfailed
+         },
+
+        {'message_id': "insufficientmemory",
+         'subject': "%(hostname)s does not have sufficient memory",
+         'template': """
+%(hostname)s failed to boot because it does not have sufficent
+memory.
+
+Please install additional memory to meet the current hardware
+requirements.
+""" + installfailed
+         },
+
+        {'message_id': "authfail",
+         'subject': "%(hostname)s failed to authenticate",
+         'template':
+"""
+%(hostname)s failed to authenticate for the following reason:
+
+%(fault)s
+
+The most common reason for authentication failure is that the
+authentication key stored in the node configuration file, does not
+match the key on record. 
+
+There are two possible steps to resolve the problem.
+
+1. If you have used an All-in-one BootCD that includes the plnode.txt file,
+    then please check your machine for any old boot media, either in the
+    floppy drive, or on a USB stick.  It is likely that an old configuration
+    is being used instead of the new configuration stored on the BootCD.
+Or, 
+2. If you are using Generic BootCD image, then regenerate the node 
+    configuration file by visiting:
+
+    https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
+
+    Under 'Download', follow the 'Download plnode.txt file for %(hostname)s'
+    option, and save the downloaded file as plnode.txt on either a floppy 
+    disk or a USB flash drive.  Be sure the 'Boot State' is set to 'Boot', 
+    and, then reboot the node.
+
+If you have already performed this step and are still receiving this
+message, please reply so that we can help investigate the problem.
+"""
+         },
+
+        {'message_id': "notinstalled",
+         'subject': "%(hostname)s is not installed",
+         'template':
+"""
+%(hostname)s failed to boot because it has either never been
+installed, or the installation is corrupt.
+
+Please check if the hard drive has failed, and replace it if so. After
+doing so, visit:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
+
+Change the 'Boot State' to 'Reinstall', and then reboot the node.
+
+If you have already performed this step and are still receiving this
+message, please reply so that we may investigate the problem.
+"""
+         },
+
+        {'message_id': "hostnamenotresolve",
+         'subject': "%(hostname)s does not resolve",
+         'template':
+"""
+%(hostname)s failed to boot because its hostname does not resolve, or
+does resolve but does not match its configured IP address.
+
+Please check the network settings for the node, especially its
+hostname, IP address, and DNS servers, by visiting:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
+
+Correct any errors, and change the 'Boot State' to 'Reinstall', and then
+reboot the node.
+
+If you have already performed this step and are still receiving this
+message, please reply so that we may investigate the problem.
+"""
+         },
+
+        # XXX N.B. I don't think these are necessary, since there's no
+        # way that the Boot Manager would even be able to contact the
+        # API to send these messages.
+
+        {'message_id': "noconfig",
+         'subject': "%(hostname)s does not have a configuration file",
+         'template': """
+%(hostname)s failed to boot because it could not find a PlanetLab
+configuration file. To create this file, visit:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
+
+Click the Configuration File link, and save the downloaded file as
+plnode.txt on either a floppy disk or a USB flash drive.  Change the 
+'Boot State' to 'Reinstall', and then reboot the node.
+
+If you have already performed this step and are still receiving this
+message, please reply so that we may investigate the problem.
+"""
+         },
+
+        {'message_id': "nodetectednetwork",
+         'subject': "%(hostname)s has unsupported network hardware",
+         'template':
+"""
+
+%(hostname)s failed to boot because it has network hardware that is
+unsupported by the current production kernel. If it has booted
+successfully in the past, please try re-installing it by visiting:
+
+https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
+
+Change the 'Boot State' to 'Reinstall', and then reboot the node.
+
+If you have already performed this step and are still receiving this
+message, please reply so that we may investigate the problem.
+"""
+         },
+        ]
+
+    for template in message_templates:
+        messages = GetMessages([template['message_id']])
+        if not messages:
+            AddMessage(template)
+
+    
+    ### Setup Initial PCU information
+    pcu_types = [{'model': 'AP79xx',
+          'name': 'APC AP79xx',
+          'pcu_protocol_types': [{ 'port': 80,
+                                  'protocol': 'APC79xxHttp',
+                                  'supported': False},
+                                 { 'port': 23,
+                                  'protocol': 'APC79xx',
+                                  'supported': True},
+                                 { 'port': 22,
+                                  'protocol': 'APC79xx',
+                                  'supported': True}],
+          },
+         {'model': 'Masterswitch',
+          'name': 'APC Masterswitch',
+          'pcu_protocol_types': [{ 'port': 80,
+                                  'protocol': 'APCMasterHttp',
+                                  'supported': False},
+                                 { 'port': 23,
+                                  'protocol': 'APCMaster',
+                                  'supported': True},
+                                 { 'port': 22,
+                                  'protocol': 'APCMaster',
+                                  'supported': True}],
+          },
+         {'model': 'DS4-RPC',
+          'name': 'BayTech DS4-RPC',
+          'pcu_protocol_types': [{ 'port': 80,
+                                  'protocol': 'BayTechHttp',
+                                  'supported': False},
+                                 { 'port': 23,
+                                  'protocol': 'BayTech',
+                                  'supported': True},
+                                 { 'port': 22,
+                                  'protocol': 'BayTech',
+                                  'supported': True}],
+          },
+         {'model': 'IP-41x_IP-81x',
+          'name': 'Dataprobe IP-41x & IP-81x',
+          'pcu_protocol_types': [ { 'port': 23,
+                                  'protocol': 'IPALTelnet',
+                                  'supported': True},
+                                  { 'port': 80,
+                                  'protocol': 'IPALHttp',
+                                  'supported': False}],
+          },
+         {'model': 'DRAC3',
+          'name': 'Dell RAC Version 3',
+          'pcu_protocol_types': [],
+          },
+         {'model': 'DRAC4',
+          'name': 'Dell RAC Version 4',
+          'pcu_protocol_types': [{ 'port': 443,
+                                  'protocol': 'DRACRacAdm',
+                                  'supported': True},
+                                 { 'port': 80,
+                                  'protocol': 'DRACRacAdm',
+                                  'supported': False},
+                                 { 'port': 22,
+                                  'protocol': 'DRAC',
+                                  'supported': True}],
+          },
+         {'model': 'ePowerSwitch',
+          'name': 'ePowerSwitch 1/4/8x',
+          'pcu_protocol_types': [{ 'port': 80,
+                                  'protocol': 'ePowerSwitch',
+                                  'supported': True}],
+          },
+         {'model': 'ilo2',
+          'name': 'HP iLO2 (Integrated Lights-Out)',
+          'pcu_protocol_types': [{ 'port': 443,
+                                  'protocol': 'HPiLOHttps',
+                                  'supported': True},
+                                 { 'port': 22,
+                                  'protocol': 'HPiLO',
+                                  'supported': True}],
+          },
+         {'model': 'ilo1',
+          'name': 'HP iLO version 1',
+          'pcu_protocol_types': [],
+          },
+         {'model': 'PM211-MIP',
+          'name': 'Infratec PM221-MIP',
+          'pcu_protocol_types': [],
+          },
+         {'model': 'AMT2.5',
+          'name': 'Intel AMT v2.5 (Active Management Technology)',
+          'pcu_protocol_types': [],
+          },
+         {'model': 'AMT3.0',
+          'name': 'Intel AMT v3.0 (Active Management Technology)',
+          'pcu_protocol_types': [],
+          },
+         {'model': 'WTI_IPS-4',
+          'name': 'Western Telematic (WTI IPS-4)',
+          'pcu_protocol_types': [],
+          },
+         {'model': 'unknown',
+          'name': 'Unknown Vendor or Model',
+          'pcu_protocol_types': [{ 'port': 443,
+                                  'protocol': 'UnknownPCU',
+                                  'supported': False},
+                                 { 'port': 80,
+                                  'protocol': 'UnknownPCU',
+                                  'supported': False},
+                                 { 'port': 23,
+                                  'protocol': 'UnknownPCU',
+                                  'supported': False},
+                                 { 'port': 22,
+                                  'protocol': 'UnknownPCU',
+                                  'supported': False}],
+          }]
+
+    # Get all model names
+    pcu_models = [type['model'] for type in GetPCUTypes()]
+    for type in pcu_types:
+        protocol_types = type['pcu_protocol_types']
+        # Take this value out of the struct.
+        del type['pcu_protocol_types']
+        if type['model'] not in pcu_models:
+            # Add the name/model info into DB
+            id = AddPCUType(type)
+            # for each protocol, also add this.
+            for ptype in protocol_types:
+                AddPCUProtocolType(id, ptype)
+
 
 if __name__ == '__main__':
     main()