source: genesis/tools/check-batch-cmd@ 10729

#!/usr/bin/env python
# vim:ts=2:et:sw=2:ai
#
# Check configs with remote addresses
#
# Rick van der Zwet <info@rickvanderzwet.nl>
#
import argparse
import gformat
import getpass
import os
import paramiko
import socket
import sys
import time

SSHPASS = None
import pysnmp
from pysnmp.entity.rfc3413.oneliner import cmdgen

def snmp_test():
  errorIndication, errorStatus, errorIndex, varBinds = cmdgen.CommandGenerator().getCmd(
      # SNMP v1
  #    cmdgen.CommunityData('test-agent', 'public', 0),
      # SNMP v2
      cmdgen.CommunityData('test-agent', 'public'),
      # SNMP v3
  #    cmdgen.UsmUserData('test-user', 'authkey1', 'privkey1'),
      cmdgen.UdpTransportTarget(('localhost', 161)),
      # Plain OID
      (1,3,6,1,2,1,1,1,0),
      # ((mib-name, mib-symbol), instance-id)
      (('SNMPv2-MIB', 'sysObjectID'), 0)
      )
  
  if errorIndication:
      print errorIndication
  else:
      if errorStatus:
          print '%s at %s\n' % (
              errorStatus.prettyPrint(),
              errorIndex and varBinds[int(errorIndex)-1] or '?'
              )
      else:
          for name, val in varBinds:
              print '%s = %s' % (name.prettyPrint(), val.prettyPrint())



class CmdError(Exception):
  pass

class ConnectError(Exception):
  pass



def host_ssh_cmd(hostname, cmd):
  try:
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(hostname, username='root', password=SSHPASS,timeout=3)
    stdin, stdout, stderr = ssh.exec_command(cmd)
    stdout = stdout.readlines()
    stderr = stderr.readlines()
    ssh.close()
    if stderr:
      raise CmdError((stderr, stdout))
    return stdout
  except socket.error, e:
    raise ConnectError(e)

def parse_ini(lines):
  return dict(map(lambda x: x.strip().split('='),lines))

def ubnt_probe(hostname):
  items = parse_ini(host_ssh_cmd(hostname, 'cat /etc/board.info'))
  print items


def get_bridge_type(host):
  """ Both NS and NS Mx uses a slighly different OID"""
  var_list = netsnmp.VarList(
   *map(lambda x: netsnmp.Varbind(x), 
    ['.1.2.840.10036.3.1.2.1.3.6', '.1.2.840.10036.3.1.2.1.3.7']))
  
  sess = netsnmp.Session(Version=1, DestHost=host, Community='public', Timeout=2 * 100000, Retries=1)
  retval = sess.get(var_list)
  if sess.ErrorInd < 0:
    raise CmdError('SNMP Failed -- [%(ErrorInd)s] %(ErrorStr)s (%(DestHost)s)' % vars(sess))
  return filter(None, retval)[0]



def node_check(host):
  print "# Processing host", host
  datadump = gformat.get_yaml(host)
  output = host_ssh_cmd(datadump['autogen_fqdn'], 'cat /var/run/dmesg.boot')

  # Get board Type
  for line in [x.strip() for x in output]:
    if line.startswith('CPU:'):
      print line
    elif line.startswith('Geode LX:'):
      datadump['board'] = 'ALIX2'
      print line
    elif line.startswith('real memory'):
      print line
    elif line.startswith('Elan-mmcr'):
      datadump['board'] = 'net45xx'
  #for iface_key in datadump['autogen_iface_keys']:
  #  ifacedump = datadump[iface_key]
  #  if ifacedump.has_key('ns_ip') and ifacedump['ns_ip']:
  #    addr = ifacedump['ns_ip'].split('/')[0]
  #    print "## Bridge IP: %(ns_ip)s at %(interface)s" % ifacedump
  #    try:
  #      socket.create_connection((addr,80),2)
  #      bridge_type = get_bridge_type(addr)
  #      datadump[iface_key]['bridge_type'] = bridge_type
  #    except (socket.timeout, socket.error) as e:
  #      print "### %s (%s)" % (e, addr)
  #    except paramiko.AuthenticationException:
  #      print "### Conection failed (invalid username/password)"
  #    except CmdError, e: 
  #      print "### Command error: %s" % e
  gformat.store_yaml(datadump)


def make_output(stdout, stderr):
  def p(prefix, lines):
    return ''.join(["#%s: %s" % (prefix, line) for line in lines])
  output = p('STDOUT', stdout)
  output += p('STDERR', stderr)
  return output

def ubnt_snmp(hostname):
  lines = """\
snmp.community=public
snmp.contact=beheer@lijst.wirelessleiden.nl
snmp.location=WL
snmp.status=enabled\
"""
  cmd = 'mca-config get /tmp/get.cfg && grep -v snmp /tmp/get.cfg > /tmp/new.cfg && echo "%s" >> /tmp/new.cfg \
    && mca-config activate /tmp/new.cfg 1>/dev/null 2>/dev/null && echo "ALL DONE"' % lines
  ssh = paramiko.SSHClient()
  ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  ssh.connect(hostname, username='root', password=SSHPASS,timeout=3)
  stdin, stdout, stderr = ssh.exec_command(cmd)
  stdout = stdout.readlines()
  stderr = stderr.readlines()
  print make_output(stdout, stderr)
  ssh.close()

def ubnt_keys(hostname):
  keys = open(os.path.join(gformat.NODE_DIR,'global_keys'),'r').read()
  ssh = paramiko.SSHClient()
  ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  ssh.connect(hostname, username='root', password=SSHPASS,timeout=3)
  cmd = 'test -d .ssh || mkdir .ssh;\
    cat > .ssh/authorized_keys && \
    chmod 0700 .ssh && \
    chmod 0755 . && cfgmtd -p /etc -w'
  stdin, stdout, stderr = ssh.exec_command(cmd)
  stdin.write(keys)
  stdin.flush()
  stdin.channel.shutdown_write()
  stdout = stdout.readlines()
  stderr = stderr.readlines()
  print make_output(stdout, stderr)
  ssh.close()

if __name__ == '__main__':
  # create the top-level parser
  parser = argparse.ArgumentParser(prog='Various WL management tools')
  parser.add_argument('--ask-pass', dest="ask_pass", action='store_true', help='Ask password if SSHPASS is not found')
  parser.add_argument('--filter', dest="use_filter", action='store_true', help='Thread the host definition as an filter')
  subparsers = parser.add_subparsers(help='sub-command help')
  
  parser_snmp = subparsers.add_parser('bridge', help='UBNT Bridge Management')
  parser_snmp.add_argument('action', type=str, choices=['keys', 'snmp', 'probe'])
  parser_snmp.add_argument('host',type=str)
  parser_snmp.set_defaults(func='bridge')
  
  parser_node = subparsers.add_parser('node', help='Proxy/Node/Hybrid Management')
  parser_node.add_argument('action', type=str, choices=['check',])
  parser_node.add_argument('host', type=str)
  parser_node.set_defaults(func='node')

  args = parser.parse_args()

  try:
    SSHPASS = os.environ['SSHPASS']
  except KeyError:
    print "#WARN: SSHPASS environ variable not found"
    if args.ask_pass:
      SSHPASS = getpass.getpass("WL root password: ")


  if args.use_filter:
    hosts = []
    for host in gformat.get_hostlist():
      if args.host in host:
        hosts.append(host)
  else:
    hosts = [args.host]


  for host in hosts:
    try:
      if args.func == 'bridge':
        if args.action == 'keys':
          ubnt_keys(host)
        elif args.action == 'snmp':
          ubnt_snmp(host)
        elif args.action == 'probe':
          ubnt_probe(host)
      elif args.func == 'node':
        if args.action == 'check':
          node_check(host)
    except ConnectError:
      print "#ERR: Connection failed to host %s" % host