7 from datetime import datetime,timedelta
12 from pcucontrol import reboot
14 from pcucontrol.util import command
15 from monitor import config
17 from monitor.database.info.model import *
19 from monitor.sources import comon
20 from monitor.wrapper import plc, plccache
23 from monitor.common import nmap_port_status, email_exception
25 COMON_COTOPURL= "http://summer.cs.princeton.edu/status/tabulator.cgi?" + \
26 "table=table_nodeview&" + \
27 "dumpcols='name,resptime,sshstatus,uptime,lastcotop,cpuspeed,memsize,disksize'&" + \
30 api = plc.getAuthAPI()
31 plc_lock = threading.Lock()
40 #print "GetPCU from PLC %s" % pcuname
41 l_pcu = plc.GetPCUs({'pcu_id' : pcuname})
47 #print "GetPCU from file %s" % pcuname
48 l_pcus = plccache.l_pcus
50 if i['pcu_id'] == pcuname:
59 def get_nodes(node_ids):
63 l_node = plc.getNodes(node_ids, ['hostname', 'last_contact', 'node_id', 'ports'])
66 plc_nodes = plccache.l_plcnodes
68 if n['node_id'] in node_ids:
80 def get_plc_pcu_values(pcuname):
82 Try to contact PLC to get the PCU info.
83 If that fails, try a backup copy from the last run.
84 If that fails, return None
88 l_pcu = get_pcu(pcuname)
91 site_id = l_pcu['site_id']
92 node_ids = l_pcu['node_ids']
93 l_node = get_nodes(node_ids)
95 if l_node is not None:
97 values[node['hostname']] = node['ports'][0]
99 values['nodenames'] = [node['hostname'] for node in l_node]
101 # NOTE: this is for a dry run later. It doesn't matter which node.
102 values['node_id'] = l_node[0]['node_id']
110 class ScanInterface(object):
113 primarykey = 'hostname'
115 def __init__(self, round=1):
119 def __getattr__(self, name):
120 if 'collect' in name or 'record' in name:
121 method = getattr(self, name, None)
123 raise Exception("No such method %s" % name)
126 raise Exception("No such method %s" % name)
128 def collect(self, nodename, data):
131 def record(self, request, (nodename, values) ):
138 fbnodesync = self.syncclass.findby_or_create(
139 #if_new_set={'round' : self.round},
140 **{ self.primarykey : nodename})
141 # NOTE: This code will either add a new record for the new self.round,
142 # OR it will find the previous value, and update it with new information.
143 # The data that is 'lost' is not that important, b/c older
144 # history still exists.
145 fbrec = self.recordclass.findby_or_create(
146 **{ self.primarykey:nodename})
148 fbrec.set( **values )
152 fbnodesync.round = self.round
155 print "%d %s %s" % (self.count, nodename, values)
160 email_exception(str(nodename))
161 print traceback.print_exc()
164 class ScanNodeInternal(ScanInterface):
165 recordclass = FindbadNodeRecord
166 #syncclass = FindbadNodeRecordSync
168 primarykey = 'hostname'
170 def collectNMAP(self, nodename, cohash):
171 #### RUN NMAP ###############################
174 print "nmap -oG - -P0 -p22,80,806 %s | grep Host:" % nodename
175 (oval,eval) = nmap.run_noexcept("nmap -oG - -P0 -p22,80,806 %s | grep Host:" % nodename)
176 # NOTE: an empty / error value for oval, will still work.
177 (values['port_status'], continue_probe) = nmap_port_status(oval)
179 values['date_checked'] = datetime.now()
181 return (nodename, values)
183 def collectInternal(self, nodename, cohash):
184 ### RUN PING ######################
186 (oval,errval) = ping.run_noexcept("ping -c 1 -q %s | grep rtt" % nodename)
193 values['ping_status'] = False
195 values['ping_status'] = True
198 for port in [22, 806]:
199 ssh = command.SSH('root', nodename, port)
201 (oval, errval) = ssh.run_noexcept2(""" <<\EOF
203 echo ' "kernel_version":"'`uname -a`'",'
204 echo ' "bmlog":"'`ls /tmp/bm.log`'",'
205 echo ' "bootcd_version":"'`cat /mnt/cdrom/bootme/ID`'",'
206 echo ' "nm_status":"'`ps ax | grep nm.py | grep -v grep`'",'
207 echo ' "fs_status":"'`touch /var/log/monitor 2>&1 ; if [ -d /vservers/ ] ; then touch /vservers/monitor.log 2>&1 ; fi ; grep proc /proc/mounts | grep ro,`'",'
208 echo ' "dns_status":"'`host boot.planet-lab.org 2>&1`'",'
209 echo ' "princeton_comon_dir":"'`ls -d /vservers/princeton_comon`'",'
211 ID=`grep princeton_comon /etc/passwd | awk -F : '{if ( $3 > 500 ) { print $3}}'`
212 echo ' "princeton_comon_running":"'`ls -d /proc/virtual/$ID`'",'
213 echo ' "princeton_comon_procs":"'`vps ax | grep $ID | grep -v grep | wc -l`'",'
214 echo ' "rpm_version":"'`rpm -q NodeManager`'",'
218 values['ssh_error'] = errval
220 #print "OVAL: %s" % oval
221 values.update(eval(oval))
222 values['ssh_portused'] = port
225 values.update({'kernel_version': "", 'bmlog' : "", 'bootcd_version' : '',
230 'princeton_comon_dir' : "",
231 'princeton_comon_running' : "",
232 'princeton_comon_procs' : "", 'ssh_portused' : None})
234 print traceback.print_exc()
237 print "RPMVERSION: %s %s" % (nodename, values['rpm_version'])
238 ### RUN SSH ######################
239 b_getbootcd_id = True
241 oval = values['kernel_version']
242 if "2.6.17" in oval or "2.6.2" in oval:
243 values['ssh_status'] = True
244 values['observed_category'] = 'PROD'
245 if "bm.log" in values['bmlog']:
246 values['observed_status'] = 'DEBUG'
248 values['observed_status'] = 'BOOT'
249 elif "2.6.12" in oval or "2.6.10" in oval:
250 values['ssh_status'] = True
251 values['observed_category'] = 'OLDPROD'
252 if "bm.log" in values['bmlog']:
253 values['observed_status'] = 'DEBUG'
255 values['observed_status'] = 'BOOT'
257 # NOTE: on 2.6.8 kernels, with 4.2 bootstrapfs, the chroot
258 # command fails. I have no idea why.
259 elif "2.4" in oval or "2.6.8" in oval:
260 b_getbootcd_id = False
261 values['ssh_status'] = True
262 values['observed_category'] = 'OLDBOOTCD'
263 values['observed_status'] = 'DEBUG'
265 values['ssh_status'] = True
266 values['observed_category'] = 'UNKNOWN'
267 if "bm.log" in values['bmlog']:
268 values['observed_status'] = 'DEBUG'
270 values['observed_status'] = 'BOOT'
273 b_getbootcd_id = False
274 values['ssh_status'] = False
275 values['observed_category'] = 'ERROR'
276 values['observed_status'] = 'DOWN'
278 values['ssh_error'] = val
279 values['kernel_version'] = ""
282 # try to get BootCD for all nodes that are not 2.4 nor inaccessible
283 oval = values['bootcd_version']
285 values['bootcd_version'] = oval
286 if "v2" in oval and \
287 ( nodename is not "planetlab1.cs.unc.edu" and \
288 nodename is not "planetlab2.cs.unc.edu" ):
289 values['observed_category'] = 'OLDBOOTCD'
291 values['bootcd_version'] = ""
293 values['bootcd_version'] = ""
295 oval = values['nm_status']
297 values['nm_status'] = "Y"
299 values['nm_status'] = "N"
301 continue_slice_check = True
302 oval = values['princeton_comon_dir']
303 if "princeton_comon_dir" in oval:
304 values['princeton_comon_dir'] = True
306 values['princeton_comon_dir'] = False
307 continue_slice_check = False
309 if continue_slice_check:
310 oval = values['princeton_comon_running']
311 if len(oval) > len('/proc/virtual/'):
312 values['princeton_comon_running'] = True
314 values['princeton_comon_running'] = False
315 continue_slice_check = False
317 values['princeton_comon_running'] = False
319 if continue_slice_check:
320 oval = values['princeton_comon_procs']
321 values['princeton_comon_procs'] = int(oval)
323 values['princeton_comon_procs'] = None
326 if nodename in cohash:
327 values['comon_stats'] = cohash[nodename]
329 values['comon_stats'] = {'resptime': '-1',
336 # include output value
337 ### GET PLC NODE ######################
341 d_node = plccache.GetNodeByName(nodename)
342 #d_node = plc.getNodes({'hostname': nodename}, ['pcu_ids', 'site_id',
343 # 'date_created', 'last_updated',
344 # 'last_contact', 'boot_state', 'nodegroup_ids'])[0]
346 traceback.print_exc()
348 values['plc_node_stats'] = d_node
350 ##### NMAP ###################
351 (n, v) = self.collectNMAP(nodename, None)
354 ### GET PLC PCU ######################
358 pcu = d_node['pcu_ids']
362 site_id = d_node['site_id']
364 values['plc_pcuid'] = d_pcu
366 ### GET PLC SITE ######################
369 values['loginbase'] = ""
371 d_site = plccache.GetSitesById([ site_id ])[0]
372 #d_site = plc.getSites({'site_id': site_id},
373 # ['max_slices', 'slice_ids', 'node_ids', 'login_base'])[0]
374 values['loginbase'] = d_site['login_base']
376 traceback.print_exc()
379 values['plc_site_stats'] = d_site
380 values['date_checked'] = datetime.now()
382 print traceback.print_exc()
384 return (nodename, values)
386 def internalprobe(hostname):
387 #fbsync = FindbadNodeRecordSync.findby_or_create(hostname="global",
388 # if_new_set={'round' : 1})
389 scannode = ScanNodeInternal() # fbsync.round)
391 (nodename, values) = scannode.collectInternal(hostname, {})
392 scannode.record(None, (nodename, values))
396 print traceback.print_exc()
399 def externalprobe(hostname):
400 #fbsync = FindbadNodeRecordSync.findby_or_create(hostname="global",
401 # if_new_set={'round' : 1})
402 scannode = ScanNodeInternal() # fbsync.round)
404 (nodename, values) = scannode.collectNMAP(hostname, {})
405 scannode.record(None, (nodename, values))
409 print traceback.print_exc()
412 class ScanPCU(ScanInterface):
413 recordclass = FindbadPCURecord
415 primarykey = 'plc_pcuid'
417 def collectInternal(self, pcuname, cohash):
419 continue_probe = True
421 values = {'reboot_trial_status' : 'novalue'}
422 ### GET PCU ######################
426 v = get_plc_pcu_values(pcuname)
427 if v['hostname'] is not None: v['hostname'] = v['hostname'].strip()
428 if v['ip'] is not None: v['ip'] = v['ip'].strip()
431 values['plc_pcu_stats'] = v
433 continue_probe = False
436 traceback.print_exc()
437 continue_probe = False
439 if b_except or not continue_probe: return (None, None, None)
441 #### RUN NMAP ###############################
444 print "nmap -oG - -P0 -p22,23,80,443,5869,9100,16992 %s | grep Host:" % reboot.pcu_name(values['plc_pcu_stats'])
445 (oval,eval) = nmap.run_noexcept("nmap -oG - -P0 -p22,23,80,443,5869,9100,16992 %s | grep Host:" % reboot.pcu_name(values['plc_pcu_stats']))
446 # NOTE: an empty / error value for oval, will still work.
447 (values['port_status'], continue_probe) = nmap_port_status(oval)
449 values['port_status'] = None
451 #### COMPLETE ENTRY #######################
453 values['entry_complete'] = []
454 #if values['protocol'] is None or values['protocol'] is "":
455 # values['entry_complete'] += ["protocol"]
456 if values['plc_pcu_stats']['model'] is None or values['plc_pcu_stats']['model'] is "":
457 values['entry_complete'] += ["model"]
458 # Cannot continue due to this condition
459 continue_probe = False
461 if values['plc_pcu_stats']['password'] is None or values['plc_pcu_stats']['password'] is "":
462 values['entry_complete'] += ["password"]
463 # Cannot continue due to this condition
464 continue_probe = False
466 if len(values['entry_complete']) > 0:
467 continue_probe = False
469 if values['plc_pcu_stats']['hostname'] is None or values['plc_pcu_stats']['hostname'] is "":
470 values['entry_complete'] += ["hostname"]
471 if values['plc_pcu_stats']['ip'] is None or values['plc_pcu_stats']['ip'] is "":
472 values['entry_complete'] += ["ip"]
474 # If there are no nodes associated with this PCU, then we cannot continue.
475 if len(values['plc_pcu_stats']['node_ids']) == 0:
476 continue_probe = False
477 values['entry_complete'] += ['nodeids']
480 #### DNS and IP MATCH #######################
481 if values['plc_pcu_stats']['hostname'] is not None and values['plc_pcu_stats']['hostname'] is not "" and \
482 values['plc_pcu_stats']['ip'] is not None and values['plc_pcu_stats']['ip'] is not "":
484 ipaddr = socket.gethostbyname(values['plc_pcu_stats']['hostname'])
485 if ipaddr == values['plc_pcu_stats']['ip']:
486 values['dns_status'] = "DNS-OK"
488 values['dns_status'] = "DNS-MISMATCH"
489 continue_probe = False
491 except Exception, err:
492 values['dns_status'] = "DNS-NOENTRY"
493 values['plc_pcu_stats']['hostname'] = values['plc_pcu_stats']['ip']
495 if values['plc_pcu_stats']['ip'] is not None and values['plc_pcu_stats']['ip'] is not "":
496 values['dns_status'] = "NOHOSTNAME"
497 values['plc_pcu_stats']['hostname'] = values['plc_pcu_stats']['ip']
499 values['dns_status'] = "NO-DNS-OR-IP"
500 values['plc_pcu_stats']['hostname'] = "No_entry_in_DB"
501 continue_probe = False
504 ###### DRY RUN ############################
505 if continue_probe and 'node_ids' in values['plc_pcu_stats'] and \
506 len(values['plc_pcu_stats']['node_ids']) > 0:
507 rb_ret = reboot.reboot_test_new(values['plc_pcu_stats']['nodenames'][0],
510 rb_ret = "Not_Run" # No nodes to test"
512 values['reboot_trial_status'] = rb_ret
515 print "____________________________________"
518 print "____________________________________"
519 errors['traceback'] = traceback.format_exc()
520 print errors['traceback']
521 values['reboot_trial_status'] = str(errors['traceback'])
524 values['entry_complete']=" ".join(values['entry_complete'])
526 values['date_checked'] = datetime.now()
527 return (pcuname, values)