Ejemplo n.º 1
0
Archivo: trace.py Proyecto: oksbsb/lcm
def tcpdump_vm(args):
    try:
        vm = VM.get(VM.id == args['id'])
    except Exception:
        print >> sys.stderr, 'Error: get vm from db failed'
        return

    try:
        host = HostDevice.get(HostDevice.ip == vm.launch_server)
    except Exception:
        print >> sys.stderr, \
            'Error: get launch_server %s from db failed' % vm.launch_server
        return

    try:
        vif = VInterface.get(VInterface.devicetype == VIF_DEVICE_TYPE_VM,
                             VInterface.deviceid == args['id'],
                             VInterface.ifindex == args['if_index'])
    except Exception:
        print >> sys.stderr, 'Error: get vif if_index=%s from db failed' % \
            args['if_index']
        return

    cmd = 'sshpass -p %s ssh %s %s@%s "ovs-vsctl --bare -- --columns=name ' \
        'find interface \'external_ids:attached-mac=\\\"%s\\\"\'"' % \
        (host.user_passwd, SSH_OPTION_STRING, host.user_name, host.ip, vif.mac)
    rc, output = commands.getstatusoutput(cmd)
    if rc:
        print >> sys.stderr, \
            'Error: cmd [%s] failed, rc=%d out=%s' % \
            (cmd.replace(host.user_passwd, '*'), rc, output)
        return
    if_name = output

    tcpdump_cmds = [
        'tcpdump', '-i', if_name, '-l', '-nn', '-e', '-v', '-c', args['count'],
        args['filter']
    ]
    print ' '.join(tcpdump_cmds)
    for line in call_system_streaming([
            '/usr/bin/timeout',
            args['timeout'],
            '/usr/bin/sshpass',
            '-p',
            host.user_passwd,
            'ssh',
    ] + SSH_OPTION_LIST + ['%s@%s' % (host.user_name, host.ip)] + tcpdump_cmds,
                                      ignore_err=False):
        print line,  # do not print duplicate newline
Ejemplo n.º 2
0
def tcpdump_vm(args):
    try:
        vm = VM.get(VM.id == args['id'])
    except Exception:
        print >> sys.stderr, 'Error: get vm from db failed'
        return

    try:
        host = HostDevice.get(HostDevice.ip == vm.launch_server)
    except Exception:
        print >> sys.stderr, \
            'Error: get launch_server %s from db failed' % vm.launch_server
        return

    try:
        vif = VInterface.get(
            VInterface.devicetype == VIF_DEVICE_TYPE_VM,
            VInterface.deviceid == args['id'],
            VInterface.ifindex == args['if_index'])
    except Exception:
        print >> sys.stderr, 'Error: get vif if_index=%s from db failed' % \
            args['if_index']
        return

    cmd = 'sshpass -p %s ssh %s %s@%s "ovs-vsctl --bare -- --columns=name ' \
        'find interface \'external_ids:attached-mac=\\\"%s\\\"\'"' % \
        (host.user_passwd, SSH_OPTION_STRING, host.user_name, host.ip, vif.mac)
    rc, output = commands.getstatusoutput(cmd)
    if rc:
        print >> sys.stderr, \
            'Error: cmd [%s] failed, rc=%d out=%s' % \
            (cmd.replace(host.user_passwd, '*'), rc, output)
        return
    if_name = output

    tcpdump_cmds = [
        'tcpdump', '-i', if_name,
        '-l', '-nn', '-e', '-v',
        '-c', args['count'], args['filter']]
    print ' '.join(tcpdump_cmds)
    for line in call_system_streaming(
            ['/usr/bin/timeout', args['timeout'],
             '/usr/bin/sshpass', '-p', host.user_passwd, 'ssh',
             ] + SSH_OPTION_LIST +
            ['%s@%s' % (host.user_name, host.ip)] + tcpdump_cmds,
            ignore_err=False):
        print line,  # do not print duplicate newline
Ejemplo n.º 3
0
Archivo: trace.py Proyecto: oksbsb/lcm
def tcpdump_vgateway(args):
    try:
        vgw = VGateway.get(VGateway.id == args['id'])
    except Exception:
        print >> sys.stderr, 'Error: get vgw from db failed'
        return

    try:
        host = HostDevice.get(HostDevice.ip == vgw.gw_launch_server)
    except Exception:
        print >> sys.stderr, \
            'Error: get gw_launch_server %s from db failed' % \
            vgw.gw_launch_server
        return

    try:
        vif = VInterface.get(VInterface.devicetype == VIF_DEVICE_TYPE_VGATEWAY,
                             VInterface.deviceid == args['id'],
                             VInterface.ifindex == args['if_index'])
    except Exception:
        print >> sys.stderr, 'Error: get vif if_index=%s from db failed' % \
            args['if_index']
        return

    if vif.iftype == VINTERFACE_TYPE_WAN:
        if_name = '%d-w-%d' % (vgw.id, vif.ifindex)
    else:
        if_name = '%d-l-%d' % (vgw.id, vif.ifindex)

    tcpdump_cmds = [
        'tcpdump', '-i', if_name, '-l', '-nn', '-e', '-v', '-c', args['count'],
        args['filter']
    ]
    print ' '.join(tcpdump_cmds)
    for line in call_system_streaming([
            '/usr/bin/timeout',
            args['timeout'],
            '/usr/bin/sshpass',
            '-p',
            host.user_passwd,
            'ssh',
    ] + SSH_OPTION_LIST + ['%s@%s' % (host.user_name, host.ip)] + tcpdump_cmds,
                                      ignore_err=False):
        print line,  # do not print duplicate newline
Ejemplo n.º 4
0
def tcpdump_vgateway(args):
    try:
        vgw = VGateway.get(VGateway.id == args['id'])
    except Exception:
        print >> sys.stderr, 'Error: get vgw from db failed'
        return

    try:
        host = HostDevice.get(HostDevice.ip == vgw.gw_launch_server)
    except Exception:
        print >> sys.stderr, \
            'Error: get gw_launch_server %s from db failed' % \
            vgw.gw_launch_server
        return

    try:
        vif = VInterface.get(
            VInterface.devicetype == VIF_DEVICE_TYPE_VGATEWAY,
            VInterface.deviceid == args['id'],
            VInterface.ifindex == args['if_index'])
    except Exception:
        print >> sys.stderr, 'Error: get vif if_index=%s from db failed' % \
            args['if_index']
        return

    if vif.iftype == VINTERFACE_TYPE_WAN:
        if_name = '%d-w-%d' % (vgw.id, vif.ifindex)
    else:
        if_name = '%d-l-%d' % (vgw.id, vif.ifindex)

    tcpdump_cmds = [
        'tcpdump', '-i', if_name,
        '-l', '-nn', '-e', '-v',
        '-c', args['count'], args['filter']]
    print ' '.join(tcpdump_cmds)
    for line in call_system_streaming(
            ['/usr/bin/timeout', args['timeout'],
             '/usr/bin/sshpass', '-p', host.user_passwd, 'ssh',
             ] + SSH_OPTION_LIST +
            ['%s@%s' % (host.user_name, host.ip)] + tcpdump_cmds,
            ignore_err=False):
        print line,  # do not print duplicate newline
Ejemplo n.º 5
0
Archivo: trace.py Proyecto: oksbsb/lcm
def find_vif_by_id(vifid):
    try:
        vif = VInterface.get(VInterface.id == vifid)
    except Exception:
        return (None, None, None, 'Can not find vif id=%d in db' % vifid)

    try:
        if vif.devicetype == VIF_DEVICE_TYPE_VM:
            ins = VM.get(VM.id == vif.deviceid)
            host = HostDevice.get(HostDevice.ip == ins.launch_server)
        elif vif.deviceid == VIF_DEVICE_TYPE_VGATEWAY:
            ins = VGateway.get(VGateway.id == vif.deviceid)
            host = HostDevice.get(HostDevice.ip == ins.gw_launch_server)
        else:
            return (None, None, vif,
                    'Can not find device/host of %r' % vif.mac)
    except Exception:
        return (None, None, vif, 'Can not find device/host of %r' % vif.mac)

    return (host, ins, vif, '')
Ejemplo n.º 6
0
def find_vif_by_id(vifid):
    try:
        vif = VInterface.get(VInterface.id == vifid)
    except Exception:
        return (None, None, None, 'Can not find vif id=%d in db' % vifid)

    try:
        if vif.devicetype == VIF_DEVICE_TYPE_VM:
            ins = VM.get(VM.id == vif.deviceid)
            host = HostDevice.get(HostDevice.ip == ins.launch_server)
        elif vif.deviceid == VIF_DEVICE_TYPE_VGATEWAY:
            ins = VGateway.get(VGateway.id == vif.deviceid)
            host = HostDevice.get(HostDevice.ip == ins.gw_launch_server)
        else:
            return (None, None, vif,
                    'Can not find device/host of %r' % vif.mac)
    except Exception:
        return (None, None, vif, 'Can not find device/host of %r' % vif.mac)

    return (host, ins, vif, '')
Ejemplo n.º 7
0
def vif_conifg(args):
    assert isinstance(args, dict)
    assert args['devicetype'] in VINTERFACE_DEVICETYPE.keys()

    devicetype = VINTERFACE_DEVICETYPE[args['devicetype']]
    try:
        vif = VInterface.get((VInterface.devicetype == devicetype) &
                             (VInterface.deviceid == args['deviceid']) &
                             (VInterface.ifindex == args['ifindex']))
    except Exception:
        print >>sys.stderr, 'Error: vif not found'
        return -1

    try:
        if args['devicetype'] == 'VM':
            vdevice = VM.get(VM.id == args['deviceid'])
        else:
            vdevice = VGateway.get(VGateway.id == args['deviceid'])
    except Exception:
        print >>sys.stderr, 'Error: VM/VGATEWAY %s not found' % \
            args['deviceid']
        return -1
    launch_server = vdevice.launch_server if args['devicetype'] == 'VM' else \
        vdevice.gw_launch_server
    try:
        host = HostDevice.get(HostDevice.ip == launch_server)
    except Exception:
        print >>sys.stderr, 'Error: Host device %s not found' % launch_server
        return -1
    cmd = 'sshpass -p %s ssh -o ConnectTimeout=9 %s@%s ' % (host.user_passwd,
                                                            host.user_name,
                                                            launch_server)

    if 'vlantag' in args:
        if vif.state != VINTERFACE_STATE_ATTACH:
            print >>sys.stderr, 'Error: vif is detached'
            return -1
        if vif.iftype != VINTERFACE_TYPE_WAN:
            print >>sys.stderr, 'Error: vif is not WAN'
            return -1

        if args['devicetype'] == 'VM':
            cmd += 'sh /usr/local/livecloud/pyagexec/script/vport.sh UPDATE '\
                'vlantag %s %s' % (vif.mac, args['vlantag'])
        else:
            cmd += 'sh /usr/local/livegate/script/router.sh update '\
                'vlantag %s %s' % (vif.mac, args['vlantag'])
        rc, output = commands.getstatusoutput(cmd)
        if rc:
            print >>sys.stderr, 'Error: "%s" failed' % cmd
            print >>sys.stderr, 'Error: %s' % output
            vif_syslog(
                viftype=vif.iftype, devicetype=args['devicetype'],
                deviceid=vif.deviceid, ifindex=vif.ifindex, vifid=vif.id,
                loginfo='vlantag config failed',
                level=VINTERFACE_CONFIG_VLANTAG_LEVEL)
            return -1
        ips = IP.select().where(IP.vifid == vif.id)
        for ip in ips:
            ip.vlantag = args['vlantag']
            ip.save()
        old_vlantag = vif.vlantag
        vif.vlantag = args['vlantag']
        vif.save()
        vif_syslog(
            viftype=vif.iftype, devicetype=args['devicetype'],
            deviceid=vif.deviceid, ifindex=vif.ifindex, vifid=vif.id,
            loginfo='vlantag config successful on launch_server %s,'
                    ' from %d to %s' % (launch_server, old_vlantag,
                                        args['vlantag']),
            level=VINTERFACE_CONFIG_VLANTAG_LEVEL)
        print "SUCCESS"
        return

    if 'broadc_bandw' in args or 'broadc_ceil_bandw' in args:
        if vif.state != VINTERFACE_STATE_ATTACH:
            print >>sys.stderr, 'Error: vif is detached'
            return -1
        if args['devicetype'] != 'VM' and vif.iftype != VINTERFACE_TYPE_WAN:
            print >>sys.stderr, 'Error: vif is not VGATEWAY WAN or VM DATA'
            return -1
        if 'broadc_bandw' not in args:
            print >>sys.stderr, 'Error: broadc_bandw is also required '\
                'for broadc_ceil_bandw'
            return -1
        if 'broadc_ceil_bandw' not in args:
            print >>sys.stderr, 'Error: broadc_ceil_bandw is also '\
                'required for broadc_bandw'
            return -1
        if int(args['broadc_bandw']) > int(args['broadc_ceil_bandw']):
            print >>sys.stderr, 'Error: broadc_bandw cannot be larger '\
                'than broadc_ceil_bandw'
            return -1

        if int(args['broadc_ceil_bandw']) > 0:
            if args['devicetype'] == 'VM':
                cmd += 'sh /usr/local/livecloud/pyagexec/script/vport.sh ADD-BROADCAST-QOS '\
                    '%s %s %s' % (vif.mac,
                                  args['broadc_bandw'],
                                  args['broadc_ceil_bandw'])
            else:
                cmd += 'sh /usr/local/livegate/script/router.sh add broadcast-qos '\
                    '%s %s %s %s' % (vif.deviceid, vif.ifindex,
                                     args['broadc_bandw'],
                                     args['broadc_ceil_bandw'])
        else:
            if args['devicetype'] == 'VM':
                cmd += 'sh /usr/local/livecloud/pyagexec/script/vport.sh DEL-BROADCAST-QOS '\
                    '%s' % vif.mac
            else:
                cmd += 'sh /usr/local/livegate/script/router.sh delete broadcast-qos '\
                    '%s %s' % (vif.deviceid, vif.ifindex)
        rc, output = commands.getstatusoutput(cmd)
        if rc:
            print >>sys.stderr, 'Error: "%s" failed' % cmd
            print >>sys.stderr, 'Error: %s' % output
            vif_syslog(
                viftype=vif.iftype, devicetype=args['devicetype'],
                deviceid=vif.deviceid, ifindex=vif.ifindex, vifid=vif.id,
                loginfo='broadcast bandwidth config failed',
                level=VINTERFACE_CONFIG_BROADC_BANDW_LEVEL)
            return -1
        vif.broadc_bandw = args['broadc_bandw']
        vif.broadc_ceil_bandw = args['broadc_ceil_bandw']
        vif.save()
        vif_syslog(
            viftype=vif.iftype, devicetype=args['devicetype'],
            deviceid=vif.deviceid, ifindex=vif.ifindex, vifid=vif.id,
            loginfo='broadcast bandwidth config successful on '
                    'launch_server %s' % launch_server,
            level=VINTERFACE_CONFIG_BROADC_BANDW_LEVEL)
        print "SUCCESS"
        return

    print "ERROR: No config parameter is given"