fix slice prefix.
[myplc.git] / db-config
1 #!/usr/bin/env /usr/bin/plcsh
2 #
3 # Bootstraps the PLC database with a default administrator account and
4 # a default site, defines default slice attribute types, and
5 # creates/updates default system slices.
6 #
7 # Mark Huang <mlhuang@cs.princeton.edu>
8 # Copyright (C) 2006 The Trustees of Princeton University
9 #
10 # $Id: db-config 7454 2007-12-11 18:55:00Z faiyaza $
11 # $HeadURL$
12
13 from plc_config import PLCConfiguration
14 import sys
15
16 def main():
17     cfg = PLCConfiguration()
18     cfg.load()
19     variables = cfg.variables()
20
21     # Load variables into dictionaries
22     for category_id, (category, variablelist) in variables.iteritems():
23         globals()[category_id] = dict(zip(variablelist.keys(),
24                                        [variable['value'] for variable in variablelist.values()]))
25
26     # Create/update the default administrator account (should be
27     # person_id 2).
28     admin = { 'person_id': 2,
29               'first_name': "Default",
30               'last_name': "Administrator",
31               'email': plc['root_user'],
32               'password': plc['root_password'] }
33     persons = GetPersons([admin['person_id']])
34     if not persons:
35         person_id = AddPerson(admin)
36         if person_id != admin['person_id']:
37             # Huh? Someone deleted the account manually from the database.
38             DeletePerson(person_id)
39             raise Exception, "Someone deleted the \"%s %s\" account from the database!" % \
40                   (admin['first_name'], admin['last_name'])
41         UpdatePerson(person_id, { 'enabled': True })
42     else:
43         person_id = persons[0]['person_id']
44         UpdatePerson(person_id, admin)
45
46     # Create/update the default site (should be site_id 1)
47     if plc_www['port'] == '80':
48         url = "http://" + plc_www['host'] + "/"
49     elif plc_www['port'] == '443':
50         url = "https://" + plc_www['host'] + "/"
51     else:
52         url = "http://" + plc_www['host'] + ":" + plc_www['port'] + "/"
53     site = { 'site_id': 1,
54              'name': plc['name'] + " Central",
55              'abbreviated_name': plc['name'],
56              'login_base': plc['slice_prefix'],
57              'is_public': False,
58              'url': url,
59              'max_slices': 100 }
60
61     sites = GetSites([site['site_id']])
62     if not sites:
63         site_id = AddSite(site['name'], site['abbreviated_name'], site['login_base'], site)
64         if site_id != site['site_id']:
65             DeleteSite(site_id)
66             raise Exception, "Someone deleted the \"%s\" site from the database!" % \
67                   site['name']
68         sites = [site]
69
70     # Must call UpdateSite() even after AddSite() to update max_slices
71     site_id = sites[0]['site_id']
72     UpdateSite(site_id, site)
73
74     # The default administrator account must be associated with a site
75     # in order to login.
76     AddPersonToSite(admin['person_id'], site['site_id'])
77     SetPersonPrimarySite(admin['person_id'], site['site_id'])
78
79     # Grant admin and PI roles to the default administrator account
80     AddRoleToPerson(10, admin['person_id'])
81     AddRoleToPerson(20, admin['person_id'])
82
83     # Setup default PlanetLabConf entries
84     default_conf_files = [
85         # NTP configuration
86         {'enabled': True,
87          'source': 'PlanetLabConf/ntp.conf.php',
88          'dest': '/etc/ntp.conf',
89          'file_permissions': '644',
90          'file_owner': 'root',
91          'file_group': 'root',
92          'preinstall_cmd': '',
93          'postinstall_cmd': '/etc/rc.d/init.d/ntpd restart',
94          'error_cmd': '',
95          'ignore_cmd_errors': False,
96          'always_update': False},
97         {'enabled': True,
98          'source': 'PlanetLabConf/ntp/step-tickers.php',
99          'dest': '/etc/ntp/step-tickers',
100          'file_permissions': '644',
101          'file_owner': 'root',
102          'file_group': 'root',
103          'preinstall_cmd': '',
104          'postinstall_cmd': '/etc/rc.d/init.d/ntpd restart',
105          'error_cmd': '',
106          'ignore_cmd_errors': False,
107          'always_update': False},
108
109         # SSH server configuration
110         {'enabled': True,
111          'source': 'PlanetLabConf/sshd_config',
112          'dest': '/etc/ssh/sshd_config',
113          'file_permissions': '600',
114          'file_owner': 'root',
115          'file_group': 'root',
116          'preinstall_cmd': '',
117          'postinstall_cmd': '/etc/init.d/sshd restart',
118          'error_cmd': '',
119          'ignore_cmd_errors': False,
120          'always_update': False},
121
122         # Administrative SSH keys
123         {'enabled': True,
124          'source': 'PlanetLabConf/keys.php?root',
125          'dest': '/root/.ssh/authorized_keys',
126          'file_permissions': '644',
127          'file_owner': 'root',
128          'file_group': 'root',
129          'preinstall_cmd': '',
130          'postinstall_cmd': '/bin/chmod 700 /root/.ssh',
131          'error_cmd': '',
132          'ignore_cmd_errors': False,
133          'always_update': False},
134         {'enabled': True,
135          'source': 'PlanetLabConf/keys.php?site_admin',
136          'dest': '/home/site_admin/.ssh/authorized_keys',
137          'file_permissions': '644',
138          'file_owner': 'site_admin',
139          'file_group': 'site_admin',
140          'preinstall_cmd': 'grep -q site_admin /etc/passwd',
141          'postinstall_cmd': '/bin/chmod 700 /home/site_admin/.ssh',
142          'error_cmd': '',
143          'ignore_cmd_errors': False,
144          'always_update': False},
145         # Log rotation configuration
146         {'enabled': True,
147          'source': 'PlanetLabConf/logrotate.conf',
148          'dest': '/etc/logrotate.conf',
149          'file_permissions': '644',
150          'file_owner': 'root',
151          'file_group': 'root',
152          'preinstall_cmd': '',
153          'postinstall_cmd': '',
154          'error_cmd': '',
155          'ignore_cmd_errors': False,
156          'always_update': False},
157
158         # updatedb/locate nightly cron job
159         {'enabled': True,
160          'source': 'PlanetLabConf/slocate.cron',
161          'dest': '/etc/cron.daily/slocate.cron',
162          'file_permissions': '755',
163          'file_owner': 'root',
164          'file_group': 'root',
165          'preinstall_cmd': '',
166          'postinstall_cmd': '',
167          'error_cmd': '',
168          'ignore_cmd_errors': False,
169          'always_update': False},
170
171         # YUM configuration
172         {'enabled': True,
173          'source': 'PlanetLabConf/yum.conf.php?gpgcheck=1',
174          'dest': '/etc/yum.conf',
175          'file_permissions': '644',
176          'file_owner': 'root',
177          'file_group': 'root',
178          'preinstall_cmd': '',
179          'postinstall_cmd': '',
180          'error_cmd': '',
181          'ignore_cmd_errors': False,
182          'always_update': False},
183         {'enabled': True,
184          'source': 'PlanetLabConf/delete-rpm-list-production',
185          'dest': '/etc/planetlab/delete-rpm-list',
186          'file_permissions': '644',
187          'file_owner': 'root',
188          'file_group': 'root',
189          'preinstall_cmd': '',
190          'postinstall_cmd': '',
191          'error_cmd': '',
192          'ignore_cmd_errors': False,
193          'always_update': False},
194
195         # PLC configuration
196         {'enabled': True,
197          'source': 'PlanetLabConf/get_plc_config.php',
198          'dest': '/etc/planetlab/plc_config',
199          'file_permissions': '644',
200          'file_owner': 'root',
201          'file_group': 'root',
202          'preinstall_cmd': '',
203          'postinstall_cmd': '',
204          'error_cmd': '',
205          'ignore_cmd_errors': False,
206          'always_update': False},
207         {'enabled': True,
208          'source': 'PlanetLabConf/get_plc_config.php?python',
209          'dest': '/etc/planetlab/plc_config.py',
210          'file_permissions': '644',
211          'file_owner': 'root',
212          'file_group': 'root',
213          'preinstall_cmd': '',
214          'postinstall_cmd': '',
215          'error_cmd': '',
216          'ignore_cmd_errors': False,
217          'always_update': False},
218         {'enabled': True,
219          'source': 'PlanetLabConf/get_plc_config.php?perl',
220          'dest': '/etc/planetlab/plc_config.pl',
221          'file_permissions': '644',
222          'file_owner': 'root',
223          'file_group': 'root',
224          'preinstall_cmd': '',
225          'postinstall_cmd': '',
226          'error_cmd': '',
227          'ignore_cmd_errors': False,
228          'always_update': False},
229         {'enabled': True,
230          'source': 'PlanetLabConf/get_plc_config.php?php',
231          'dest': '/etc/planetlab/php/plc_config.php',
232          'file_permissions': '644',
233          'file_owner': 'root',
234          'file_group': 'root',
235          'preinstall_cmd': '',
236          'postinstall_cmd': '',
237          'error_cmd': '',
238          'ignore_cmd_errors': False,
239          'always_update': False},
240
241         # XXX Required for old Node Manager
242         # Proper configuration
243         {'enabled': True,
244          'source': 'PlanetLabConf/propd.conf',
245          'dest': '/etc/proper/propd.conf',
246          'file_permissions': '644',
247          'file_owner': 'root',
248          'file_group': 'root',
249          'preinstall_cmd': '',
250          'postinstall_cmd': '/etc/init.d/proper restart',
251          'error_cmd': '',
252          'ignore_cmd_errors': True,
253          'always_update': False},
254
255         # XXX Required for old Node Manager
256         # Bandwidth cap
257         {'enabled': True,
258          'source': 'PlanetLabConf/bwlimit.php',
259          'dest': '/etc/planetlab/bwcap',
260          'file_permissions': '644',
261          'file_owner': 'root',
262          'file_group': 'root',
263          'preinstall_cmd': '',
264          'postinstall_cmd': '',
265          'error_cmd': '',
266          'ignore_cmd_errors': True,
267          'always_update': False},
268
269         # Proxy ARP setup
270         {'enabled': True,
271          'source': 'PlanetLabConf/proxies.php',
272          'dest': '/etc/planetlab/proxies',
273          'file_permissions': '644',
274          'file_owner': 'root',
275          'file_group': 'root',
276          'preinstall_cmd': '',
277          'postinstall_cmd': '',
278          'error_cmd': '',
279          'ignore_cmd_errors': False,
280          'always_update': False},
281
282         # Firewall configuration
283         {'enabled': True,
284          'source': 'PlanetLabConf/iptables',
285          'dest': '/etc/sysconfig/iptables',
286          'file_permissions': '600',
287          'file_owner': 'root',
288          'file_group': 'root',
289          'preinstall_cmd': '',
290          'postinstall_cmd': '',
291          'error_cmd': '',
292          'ignore_cmd_errors': False,
293          'always_update': False},
294         {'enabled': True,
295          'source': 'PlanetLabConf/blacklist.php',
296          'dest': '/etc/planetlab/blacklist',
297          'file_permissions': '600',
298          'file_owner': 'root',
299          'file_group': 'root',
300          'preinstall_cmd': '',
301          'postinstall_cmd': '/sbin/iptables-restore --noflush < /etc/planetlab/blacklist',
302          'error_cmd': '',
303          'ignore_cmd_errors': True,
304          'always_update': False},
305
306         # /etc/issue
307         {'enabled': True,
308          'source': 'PlanetLabConf/issue.php',
309          'dest': '/etc/issue',
310          'file_permissions': '644',
311          'file_owner': 'root',
312          'file_group': 'root',
313          'preinstall_cmd': '',
314          'postinstall_cmd': '',
315          'error_cmd': '',
316          'ignore_cmd_errors': False,
317          'always_update': False},
318
319         # Kernel parameters
320         {'enabled': True,
321          'source': 'PlanetLabConf/sysctl.php',
322          'dest': '/etc/sysctl.conf',
323          'file_permissions': '644',
324          'file_owner': 'root',
325          'file_group': 'root',
326          'preinstall_cmd': '',
327          'postinstall_cmd': '/sbin/sysctl -e -p /etc/sysctl.conf',
328          'error_cmd': '',
329          'ignore_cmd_errors': False,
330          'always_update': False},
331
332         # Sendmail configuration
333         {'enabled': True,
334          'source': 'PlanetLabConf/sendmail.mc',
335          'dest': '/etc/mail/sendmail.mc',
336          'file_permissions': '644',
337          'file_owner': 'root',
338          'file_group': 'root',
339          'preinstall_cmd': '',
340          'postinstall_cmd': '',
341          'error_cmd': '',
342          'ignore_cmd_errors': False,
343          'always_update': False},
344         {'enabled': True,
345          'source': 'PlanetLabConf/sendmail.cf',
346          'dest': '/etc/mail/sendmail.cf',
347          'file_permissions': '644',
348          'file_owner': 'root',
349          'file_group': 'root',
350          'preinstall_cmd': '',
351          'postinstall_cmd': 'service sendmail restart',
352          'error_cmd': '',
353          'ignore_cmd_errors': False,
354          'always_update': False},
355
356         # GPG signing keys
357         {'enabled': True,
358          'source': 'PlanetLabConf/RPM-GPG-KEY-fedora',
359          'dest': '/etc/pki/rpm-gpg/RPM-GPG-KEY-fedora',
360          'file_permissions': '644',
361          'file_owner': 'root',
362          'file_group': 'root',
363          'preinstall_cmd': '',
364          'postinstall_cmd': 'rpm --import /etc/pki/rpm-gpg/RPM-GPG-KEY-fedora',
365          'error_cmd': '',
366          'ignore_cmd_errors': False,
367          'always_update': False},
368         {'enabled': True,
369          'source': 'PlanetLabConf/get_gpg_key.php',
370          'dest': '/etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab',
371          'file_permissions': '644',
372          'file_owner': 'root',
373          'file_group': 'root',
374          'preinstall_cmd': '',
375          'postinstall_cmd': 'rpm --import /etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab',
376          'error_cmd': '',
377          'ignore_cmd_errors': False,
378          'always_update': False},
379
380         # Ping of death configuration
381         # the 'restart' postcommand doesn't work, b/c the pod script doesn't support it.
382         {'enabled': True,
383          'source': 'PlanetLabConf/ipod.conf.php',
384          'dest': '/etc/ipod.conf',
385          'file_permissions': '644',
386          'file_owner': 'root',
387          'file_group': 'root',
388          'preinstall_cmd': '',
389          'postinstall_cmd': '/etc/init.d/pod start',
390          'error_cmd': '',
391          'ignore_cmd_errors': False,
392          'always_update': False},
393
394         # sudo configuration
395         {'enabled': True,
396          'source': 'PlanetLabConf/sudoers',
397          'dest': '/etc/sudoers',
398          'file_permissions': '440',
399          'file_owner': 'root',
400          'file_group': 'root',
401          'preinstall_cmd': '',
402          'postinstall_cmd': '/usr/sbin/visudo -c',
403          'error_cmd': '',
404          'ignore_cmd_errors': False,
405          'always_update': False}
406         ]
407
408     # Get list of existing (enabled, global) files
409     conf_files = GetConfFiles()
410     conf_files = filter(lambda conf_file: conf_file['enabled'] and \
411                                           not conf_file['node_ids'] and \
412                                           not conf_file['nodegroup_ids'],
413                         conf_files)
414     dests = [conf_file['dest'] for conf_file in conf_files]
415     conf_files = dict(zip(dests, conf_files))
416
417     # Create/update default PlanetLabConf entries
418     for default_conf_file in default_conf_files:
419         if default_conf_file['dest'] not in dests:
420             AddConfFile(default_conf_file)
421         else:
422             conf_file = conf_files[default_conf_file['dest']]
423             UpdateConfFile(conf_file['conf_file_id'], default_conf_file)
424
425     # Setup default slice attribute types
426     default_attribute_types = [
427         # Slice type (only vserver is supported)
428         {'name': "type",
429          'description': "Type of slice (e.g. vserver)",
430          'min_role_id': 20},
431
432         # System slice
433         {'name': "system",
434          'description': "Is a default system slice (1) or not (0 or unset)",
435          'min_role_id': 10},
436
437         # Slice enabled (1) or suspended (0)
438         {'name': "enabled",
439          'description': "Slice enabled (1 or unset) or suspended (0)",
440          'min_role_id': 10},
441
442         # Slice reference image
443         {'name': "vref",
444          'description': "Reference image",
445          'min_role_id': 30},
446
447         # Slice initialization script
448         {'name': "initscript",
449          'description': "Slice initialization script",
450          'min_role_id': 10},
451
452         # CPU share
453         {'name': "cpu_pct",
454          'description': "Reserved CPU percent",
455          'min_role_id': 10},
456         {'name': "cpu_share",
457          'description': "Number of CPU shares",
458          'min_role_id': 10},
459
460         # Bandwidth limits
461         {'name': "net_min_rate",
462          'description': "Minimum bandwidth (kbps)",
463          'min_role_id': 10},
464         {'name': "net_max_rate",
465          'description': "Maximum bandwidth (kbps)",
466          'min_role_id': 10},
467         {'name': "net_i2_min_rate",
468          'description': "Minimum bandwidth over I2 routes (kbps)",
469          'min_role_id': 10},
470         {'name': "net_i2_max_rate",
471          'description': "Maximum bandwidth over I2 routes (kbps)",
472          'min_role_id': 10},
473         {'name': "net_max_kbyte",
474          'description': "Maximum daily network Tx KByte limit.",
475          'min_role_id': 10},
476         {'name': "net_thresh_kbyte",
477          'description': "KByte limit before warning and throttling.",
478          'min_role_id': 10},
479         {'name': "net_i2_max_kbyte",
480          'description': "Maximum daily network Tx KByte limit to I2 hosts.",
481          'min_role_id': 10},
482         {'name': "net_i2_thresh_kbyte",
483          'description': "KByte limit to I2 hosts before warning and throttling.",
484          'min_role_id': 10},
485         {'name': "net_share",
486          'description': "Number of bandwidth shares",
487          'min_role_id': 10},
488         {'name': "net_i2_share",
489          'description': "Number of bandwidth shares over I2 routes",
490          'min_role_id': 10},
491  
492         # Disk quota
493         {'name': "disk_max",
494          'description': "Disk quota (1k disk blocks)",
495          'min_role_id': 10},
496
497         # Proper operations
498         {'name': "proper_op",
499          'description': "Proper operation (e.g. bind_socket)",
500          'min_role_id': 10},
501
502         # VServer capabilities 
503         {'name': "capabilities",
504          'description': "VServer bcapabilities (separate by commas)",
505          'min_role_id': 10},
506
507                 # Vsys
508         {'name': "vsys",
509          'description': "Bind vsys script fd's to a slice's vsys directory.",
510          'min_role_id': 10},
511
512         # CoDemux
513         {'name': "codemux",
514          'description': "Demux HTTP between slices using localhost ports. Value in the form 'host, localhost port'.",
515          'min_role_id': 10},
516
517         ]
518
519     # Get list of existing attribute types
520     attribute_types = GetSliceAttributeTypes()
521     attribute_types = [attribute_type['name'] for attribute_type in attribute_types]
522
523     # Create/update default slice attribute types
524     for default_attribute_type in default_attribute_types:
525         if default_attribute_type['name'] not in attribute_types:
526             AddSliceAttributeType(default_attribute_type)
527         else:
528             UpdateSliceAttributeType(default_attribute_type['name'], default_attribute_type)
529
530     # Default Initscripts
531     default_initscripts = []
532
533     # Find initscripts and add them to the db
534     for (root, dirs, files) in os.walk("/etc/plc_sliceinitscripts"):
535         for f in files:
536             # Read the file
537             file = open(root + "/" + f, "ro")
538             default_initscripts.append({"name": plc['slice_prefix'] + "_" + f,
539                                         "enabled": True,
540                                         "script": file.read().replace("XXXSITEXXX", url).replace("XXXPREFIXXXX", plc['slice_prefix'])})
541             file.close()
542
543     # Get list of existing initscripts
544     oldinitscripts = GetInitScripts()
545     oldinitscripts = [script['name'] for script in oldinitscripts]
546
547     for initscript in default_initscripts:
548         if initscript['name'] not in oldinitscripts:  AddInitScript(initscript)
549
550     # Setup default slice attribute types
551     default_setting_types = [
552
553         {'category' : "general",
554          'name' : "ifname",
555          'description': "Set interface name, instead of eth0 or the like",
556          'min_role_id' : 40},
557         {'category' : "general",
558          'name' : "driver",
559          'description': "Use this to specify an alternate driver",
560          'min_role_id' : 40 },
561         {'category' : "general",
562          'name' : "alias",
563          'description': "Allows to reuse an interface as eth0:alias",
564          'min_role_id' : 40},
565
566         {'category' : "hidden",
567          'name' : "backdoor",
568          'description': "For testing new settings",
569          'min_role_id' : 10},
570         ] + [
571         { "category" : "WiFi",
572           "name" : x,
573           "description" : "802.11 %s -- see %s"%(y,z),
574           "min_role_id" : 40 } for (x,y,z) in [
575             ("mode","Mode","iwconfig"),
576             ("essid","ESSID","iwconfig"),
577             ("nw","Network Id","iwconfig"),
578             ("freq","Frequency","iwconfig"),
579             ("channel","Channel","iwconfig"),
580             ("sens","sensitivity threshold","iwconfig"),
581             ("rate","Rate","iwconfig"),
582             ("key","key","iwconfig key"),
583             ("key1","key1","iwconfig key [1]"),
584             ("key2","key2","iwconfig key [2]"),
585             ("key3","key3","iwconfig key [3]"),
586             ("key4","key4","iwconfig key [4]"),
587             ("securitymode","Security mode","iwconfig enc"),
588             ("iwconfig","Additional parameters to iwconfig","ifup-wireless"),
589             ("iwpriv","Additional parameters to iwpriv","ifup-wireless"),
590             ]
591         ]
592
593
594     # Get list of existing attribute types
595     setting_types = GetNodeNetworkSettingTypes()
596     setting_types = [setting_type['name'] for setting_type in setting_types]
597
598     # Create/update default slice setting types
599     for default_setting_type in default_setting_types:
600         if default_setting_type['name'] not in setting_types:
601             AddNodeNetworkSettingType(default_setting_type)
602         else:
603             UpdateNodeNetworkSettingType(default_setting_type['name'], default_setting_type)
604
605     # Create/update system slices
606     default_slices = [
607          # PlanetFlow
608         {'name': plc['slice_prefix'] + "_netflow",
609          'description': "PlanetFlow Traffic Auditing Service",
610          'url': url,
611          'instantiation': "plc-instantiated",
612          # Renew forever (minus one day, work around date conversion weirdness)
613          'expires': sys.maxint - (60 * 60 * 24),
614          'attributes': [('system', "1"),
615                         ('vref', "planetflow"),
616                         ('proper_op', "open file=/etc/passwd, flags=r"),
617                         ('proper_op', "create_socket"),
618                         ('proper_op', "bind_socket")]},
619           # Sirius
620         {'name': plc['slice_prefix'] + "_sirius",
621          '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',
622          'url': url,
623          'instantiation': "plc-instantiated",
624          # Renew forever (minus one day, work around date conversion weirdness)
625          'expires': sys.maxint - (60 * 60 * 24),
626          'attributes': [('system', "1"),
627                         ('net_min_rate', "2000"),
628                         ('cpu_pct', "25"),
629                         ('initscript', plc['slice_prefix'] + "_sirius")]}
630         ]
631     
632     for default_slice in default_slices:
633         slices = GetSlices([default_slice['name']])
634         if slices:
635             slice = slices[0]
636             UpdateSlice(slice['slice_id'], default_slice)
637         else:
638             AddSlice(default_slice)
639             slice = GetSlices([default_slice['name']])[0]
640
641         # Create/update all attributes
642         slice_attributes = []
643         if slice['slice_attribute_ids']:
644             # Delete unknown attributes
645             for slice_attribute in GetSliceAttributes(slice['slice_attribute_ids']):
646                 if (slice_attribute['name'], slice_attribute['value']) \
647                    not in default_slice['attributes']:
648                     DeleteSliceAttribute(slice_attribute['slice_attribute_id'])
649                 else:
650                     slice_attributes.append((slice_attribute['name'], slice_attribute['value']))
651
652         for (name, value) in default_slice['attributes']:
653             if (name, value) not in slice_attributes:
654                 AddSliceAttribute(slice['name'], name, value)
655
656     installfailed = """
657 Once the node meets these requirements, please reinitiate the install
658 by visiting:
659
660 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
661
662 Update the BootState to 'Reinstall', then reboot the node.
663
664 If you have already performed this step and are still receiving this
665 message, please reply so that we may investigate the problem.
666 """
667
668     # Load default message templates
669     message_templates = [
670         {'message_id': 'Verify account',
671          'subject': "Verify account registration",
672          'template': """
673 Please verify that you registered for a %(PLC_NAME)s account with the
674 username %(email)s by visiting:
675
676 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/register.php?id=%(person_id)d&key=%(verification_key)s
677
678 If you did not register for a %(PLC_NAME)s account, please ignore this
679 message, or contact %(PLC_NAME)s Support <%(PLC_MAIL_SUPPORT_ADDRESS)s>.
680 """
681          },
682
683         {'message_id': 'New PI account',
684          'subject': "New PI account registration from %(first_name)s %(last_name)s <%(email)s> at %(site_name)s",
685          'template': """
686 %(first_name)s %(last_name)s <%(email)s> has signed up for a new
687 %(PLC_NAME)s account at %(site_name)s and has requested a PI role. PIs
688 are responsible for enabling user accounts, creating slices, and
689 ensuring that all users abide by the %(PLC_NAME)s Acceptable Use
690 Policy.
691
692 Only %(PLC_NAME)s administrators may enable new PI accounts. If you
693 are a PI at %(site_name)s, please respond and indicate whether this
694 registration is acceptable.
695
696 To view the request, visit:
697
698 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/index.php?id=%(person_id)d
699 """
700          },
701
702         {'message_id': 'New account',
703          'subject': "New account registration from %(first_name)s %(last_name)s <%(email)s> at %(site_name)s",
704          'template': """
705 %(first_name)s %(last_name)s <%(email)s> has signed up for a new
706 %(PLC_NAME)s account at %(site_name)s and has requested the following
707 roles: %(roles)s.
708
709 To deny the request or enable the account, visit:
710
711 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/index.php?id=%(person_id)d
712 """
713          },
714
715         {'message_id': 'Password reset requested',
716          'subject': "Password reset requested",
717          'template': """
718 Someone has requested that the password of your %(PLC_NAME)s account
719 %(email)s be reset. If this person was you, you may continue with the
720 reset by visiting:
721
722 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/reset_password.php?id=%(person_id)d&key=%(verification_key)s
723
724 If you did not request that your password be reset, please contact
725 %(PLC_NAME)s Support <%(PLC_MAIL_SUPPORT_ADDRESS)s>. Do not quote or
726 otherwise include any of this text in any correspondence.
727 """
728          },
729
730         {'message_id': 'Password reset',
731          'subject': "Password reset",
732          'template': """
733 The password of your %(PLC_NAME)s account %(email)s has been
734 temporarily reset to:
735
736 %(password)s
737
738 Please change it at as soon as possible by visiting:
739
740 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/persons/index.php?id=%(person_id)d
741
742 If you did not request that your password be reset, please contact
743 %(PLC_NAME)s Support <%(PLC_MAIL_SUPPORT_ADDRESS)s>. Do not quote or
744 otherwise include any of this text in any correspondence.
745 """
746          },
747
748         # Boot Manager messages
749         {'message_id': "installfinished",
750          'subject': "%(hostname)s completed installation",
751          'template': """
752 %(hostname)s just completed installation.
753
754 The node should be usable in a couple of minutes if installation was
755 successful.
756 """
757          },
758
759         {'message_id': "insufficientdisk",
760          'subject': "%(hostname)s does not have sufficient disk space",
761          'template': """
762 %(hostname)s failed to boot because it does not have sufficent disk
763 space, or because its disk controller was not recognized.
764
765 Please replace the current disk or disk controller or install
766 additional disks to meet the current hardware requirements.
767 """ + installfailed
768          },
769
770         {'message_id': "insufficientmemory",
771          'subject': "%(hostname)s does not have sufficient memory",
772          'template': """
773 %(hostname)s failed to boot because it does not have sufficent
774 memory.
775
776 Please install additional memory to meet the current hardware
777 requirements.
778 """ + installfailed
779          },
780
781         {'message_id': "authfail",
782          'subject': "%(hostname)s failed to authenticate",
783          'template':
784 """
785 %(hostname)s failed to authenticate for the following reason:
786
787 %(fault)s
788
789 The most common reason for authentication failure is that the
790 authentication key stored in the node configuration file, does not
791 match the key on record. 
792
793 There are two possible steps to resolve the problem.
794
795 1. If you have used an All-in-one BootCD that includes the plnode.txt file,
796     then please check your machine for any old boot media, either in the
797     floppy drive, or on a USB stick.  It is likely that an old configuration
798     is being used instead of the new configuration stored on the BootCD.
799 Or, 
800 2. If you are using Generic BootCD image, then regenerate the node 
801     configuration file by visiting:
802
803     https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
804
805     Under 'Download', follow the 'Download plnode.txt file for %(hostname)s'
806     option, and save the downloaded file as plnode.txt on either a floppy 
807     disk or a USB flash drive.  Be sure the 'Boot State' is set to 'Boot', 
808     and, then reboot the node.
809
810 If you have already performed this step and are still receiving this
811 message, please reply so that we can help investigate the problem.
812 """
813          },
814
815         {'message_id': "notinstalled",
816          'subject': "%(hostname)s is not installed",
817          'template':
818 """
819 %(hostname)s failed to boot because it has either never been
820 installed, or the installation is corrupt.
821
822 Please check if the hard drive has failed, and replace it if so. After
823 doing so, visit:
824
825 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
826
827 Change the 'Boot State' to 'Reinstall', and then reboot the node.
828
829 If you have already performed this step and are still receiving this
830 message, please reply so that we may investigate the problem.
831 """
832          },
833
834         {'message_id': "hostnamenotresolve",
835          'subject': "%(hostname)s does not resolve",
836          'template':
837 """
838 %(hostname)s failed to boot because its hostname does not resolve, or
839 does resolve but does not match its configured IP address.
840
841 Please check the network settings for the node, especially its
842 hostname, IP address, and DNS servers, by visiting:
843
844 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
845
846 Correct any errors, and change the 'Boot State' to 'Reinstall', and then
847 reboot the node.
848
849 If you have already performed this step and are still receiving this
850 message, please reply so that we may investigate the problem.
851 """
852          },
853
854         # XXX N.B. I don't think these are necessary, since there's no
855         # way that the Boot Manager would even be able to contact the
856         # API to send these messages.
857
858         {'message_id': "noconfig",
859          'subject': "%(hostname)s does not have a configuration file",
860          'template': """
861 %(hostname)s failed to boot because it could not find a PlanetLab
862 configuration file. To create this file, visit:
863
864 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
865
866 Click the Configuration File link, and save the downloaded file as
867 plnode.txt on either a floppy disk or a USB flash drive.  Change the 
868 'Boot State' to 'Reinstall', and then reboot the node.
869
870 If you have already performed this step and are still receiving this
871 message, please reply so that we may investigate the problem.
872 """
873          },
874
875         {'message_id': "nodetectednetwork",
876          'subject': "%(hostname)s has unsupported network hardware",
877          'template':
878 """
879
880 %(hostname)s failed to boot because it has network hardware that is
881 unsupported by the current production kernel. If it has booted
882 successfully in the past, please try re-installing it by visiting:
883
884 https://%(PLC_WWW_HOST)s:%(PLC_WWW_SSL_PORT)d/db/nodes/?id=%(node_id)d
885
886 Change the 'Boot State' to 'Reinstall', and then reboot the node.
887
888 If you have already performed this step and are still receiving this
889 message, please reply so that we may investigate the problem.
890 """
891          },
892         ]
893
894     for template in message_templates:
895         messages = GetMessages([template['message_id']])
896         if not messages:
897             AddMessage(template)
898
899     
900     ### Setup Initial PCU information
901     pcu_types = [{'model': 'AP79xx',
902           'name': 'APC AP79xx',
903           'pcu_protocol_types': [{ 'port': 80,
904                                   'protocol': 'APC79xxHttp',
905                                   'supported': False},
906                                  { 'port': 23,
907                                   'protocol': 'APC79xx',
908                                   'supported': True},
909                                  { 'port': 22,
910                                   'protocol': 'APC79xx',
911                                   'supported': True}],
912           },
913          {'model': 'Masterswitch',
914           'name': 'APC Masterswitch',
915           'pcu_protocol_types': [{ 'port': 80,
916                                   'protocol': 'APCMasterHttp',
917                                   'supported': False},
918                                  { 'port': 23,
919                                   'protocol': 'APCMaster',
920                                   'supported': True},
921                                  { 'port': 22,
922                                   'protocol': 'APCMaster',
923                                   'supported': True}],
924           },
925          {'model': 'DS4-RPC',
926           'name': 'BayTech DS4-RPC',
927           'pcu_protocol_types': [{ 'port': 80,
928                                   'protocol': 'BayTechHttp',
929                                   'supported': False},
930                                  { 'port': 23,
931                                   'protocol': 'BayTech',
932                                   'supported': True},
933                                  { 'port': 22,
934                                   'protocol': 'BayTech',
935                                   'supported': True}],
936           },
937          {'model': 'IP-41x_IP-81x',
938           'name': 'Dataprobe IP-41x & IP-81x',
939           'pcu_protocol_types': [ { 'port': 23,
940                                   'protocol': 'IPALTelnet',
941                                   'supported': True},
942                                   { 'port': 80,
943                                   'protocol': 'IPALHttp',
944                                   'supported': False}],
945           },
946          {'model': 'DRAC3',
947           'name': 'Dell RAC Version 3',
948           'pcu_protocol_types': [],
949           },
950          {'model': 'DRAC4',
951           'name': 'Dell RAC Version 4',
952           'pcu_protocol_types': [{ 'port': 443,
953                                   'protocol': 'DRACRacAdm',
954                                   'supported': True},
955                                  { 'port': 80,
956                                   'protocol': 'DRACRacAdm',
957                                   'supported': False},
958                                  { 'port': 22,
959                                   'protocol': 'DRAC',
960                                   'supported': True}],
961           },
962          {'model': 'ePowerSwitch',
963           'name': 'ePowerSwitch 1/4/8x',
964           'pcu_protocol_types': [{ 'port': 80,
965                                   'protocol': 'ePowerSwitch',
966                                   'supported': True}],
967           },
968          {'model': 'ilo2',
969           'name': 'HP iLO2 (Integrated Lights-Out)',
970           'pcu_protocol_types': [{ 'port': 443,
971                                   'protocol': 'HPiLOHttps',
972                                   'supported': True},
973                                  { 'port': 22,
974                                   'protocol': 'HPiLO',
975                                   'supported': True}],
976           },
977          {'model': 'ilo1',
978           'name': 'HP iLO version 1',
979           'pcu_protocol_types': [],
980           },
981          {'model': 'PM211-MIP',
982           'name': 'Infratec PM221-MIP',
983           'pcu_protocol_types': [],
984           },
985          {'model': 'AMT2.5',
986           'name': 'Intel AMT v2.5 (Active Management Technology)',
987           'pcu_protocol_types': [],
988           },
989          {'model': 'AMT3.0',
990           'name': 'Intel AMT v3.0 (Active Management Technology)',
991           'pcu_protocol_types': [],
992           },
993          {'model': 'WTI_IPS-4',
994           'name': 'Western Telematic (WTI IPS-4)',
995           'pcu_protocol_types': [],
996           },
997          {'model': 'unknown',
998           'name': 'Unknown Vendor or Model',
999           'pcu_protocol_types': [{ 'port': 443,
1000                                   'protocol': 'UnknownPCU',
1001                                   'supported': False},
1002                                  { 'port': 80,
1003                                   'protocol': 'UnknownPCU',
1004                                   'supported': False},
1005                                  { 'port': 23,
1006                                   'protocol': 'UnknownPCU',
1007                                   'supported': False},
1008                                  { 'port': 22,
1009                                   'protocol': 'UnknownPCU',
1010                                   'supported': False}],
1011           }]
1012
1013     # Get all model names
1014     pcu_models = [type['model'] for type in GetPCUTypes()]
1015     for type in pcu_types:
1016         protocol_types = type['pcu_protocol_types']
1017         # Take this value out of the struct.
1018         del type['pcu_protocol_types']
1019         if type['model'] not in pcu_models:
1020             # Add the name/model info into DB
1021             id = AddPCUType(type)
1022             # for each protocol, also add this.
1023             for ptype in protocol_types:
1024                 AddPCUProtocolType(id, ptype)
1025
1026
1027 if __name__ == '__main__':
1028     main()
1029
1030 # Local variables:
1031 # tab-width: 4
1032 # mode: python
1033 # End: