#!/usr/bin/python2.7

#

#

# CDDL HEADER START

#

# The contents of this file are subject to the terms of the

# Common Development and Distribution License (the "License").

# You may not use this file except in compliance with the License.

#

# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE

# or http://www.opensolaris.org/os/licensing.

# See the License for the specific language governing permissions

# and limitations under the License.

#

# When distributing Covered Code, include this CDDL HEADER in each

# file and include the License file at usr/src/OPENSOLARIS.LICENSE.

# If applicable, add the following below this CDDL HEADER, with the

# fields enclosed by brackets "[]" replaced with your own identifying

# information: Portions Copyright [yyyy] [name of copyright owner]

#

# CDDL HEADER END

#

#

#

# Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.

#

from rad.bindings.com.oracle.solaris.rad.dlmgr import *

import rad.client as radcli

import rad.connect as radcon

def dlvalue_to_propval(adict):

    pval = None

    for key, value in adict.iteritems():

        for k, v in vars(value).iteritems():

            if k == 'type':

                continue

            if v is not None:

                pval = v

                break

    return pval

def printError(ex):

    dlerr = ex.get_payload()

    if dlerr != None:

        print "Error: %s" % dlerr.errmsg

    else:

        print "Exception: %s" % ex

def deleteVNIC(argv, numargs, dm):

    if numargs !=  3:

        print "usage: %s delete-vnic vnicname" % (argv[0])

        return False

    try:

        dm.deleteVNIC(argv[2])

    except radcli.ObjectError as ex:

        printError(ex)

        return False

    return True

def resetOFPorts(argv, numargs, rc):

    if numargs != 2:

        print "usage: %s reset-ofports" % (argv[0])

        return False

    # Enumerate all the datalinks and reset any ofports that are set

    rval = True

    datalink_name_list = rc.list_objects(Datalink())

    for datalink_name in datalink_name_list:

        datalink = rc.get_object(datalink_name)

        try:

            props = datalink.getProperties(["ofport", "temporary"], ["current"])

        except radcli.ObjectError as ex:

            dlerr = ex.get_payload()

            # ignore if not a supported property

	    if dlerr.err == DLSTATUS.INVALID_ARGUMENT._value:

                continue

            printError(ex)

            rval = False

            continue

        prop = props[0]

        pval =  dlvalue_to_propval(prop.map)

        if pval == None or pval == 0:

            continue

        prop = props[1]

        pval =  dlvalue_to_propval(prop.map)

        try:

            datalink.setProperties({

                'temporary': DLValue(type=DLValueType.BOOLEAN, bval=pval),

                'ofport': DLValue(type=DLValueType.ULONG, ulval=None)

            })

        except radcli.ObjectError as ex:

            dlerr = ex.get_payload()

            printError(ex)

            rval = False

    return rval

def removeFlows(argv, numargs, rc):

    if numargs != 2:

        print "usage: %s remove-flows" % (argv[0])

        return False

    # Enumerate all the datalinks and look for OF links

    rval = True

    datalink_list = rc.list_objects(Datalink())

    for datalink_name in datalink_list:

        datalink = rc.get_object(datalink_name)

        linkname = datalink_name._kvpairs["name"]

	try:

            props = datalink.getProperties(["openvswitch"], ["current"])

        except radcli.ObjectError as ex:

            dlerr = ex.get_payload()

            # ignore if not a supported property

	    if dlerr.err == DLSTATUS.INVALID_ARGUMENT._value:

                continue

            printError(ex)

            rval = False

            continue

        prop = props[0]

        pval =  dlvalue_to_propval(prop.map)

        if pval == str("off"):

            continue

        # Enumerate all the flows for the datalink and remove them

        flow_list = rc.list_objects(Flow(), ADRGlobPattern({'linkname': linkname}))

        for flow_name in flow_list:

            try:

                datalink.removeFlow(flow_name._kvpairs["name"])

            except radcli.ObjectError as ex:

                dlerr = ex.get_payload()

                # ignore if the flow is no longer found

                if dlerr.err == DLSTATUS.NOT_FOUND._value:

                    continue

                printError(ex)

                rval = False

    return rval

def deleteEtherstub(argv, numargs, rc, dm):

    if numargs != 3:

        print "usage: %s delete-etherstub etherstub-name" % (argv[0])

        return False

    name = argv[2]

    try:

        etherstub = rc.get_object(Etherstub(), ADRGlobPattern({'name': name}))

        etherstub.setProperties({

            'temporary': DLValue(type=DLValueType.BOOLEAN, bval="True"),

            'openvswitch': DLValue(type=DLValueType.BOOLEAN, bval=None)

            })

        dm.deleteEtherstub(name)

    except radcli.NotFoundError as ex:

        pass

    except radcli.ObjectError as ex:

        dlerr = ex.get_payload()

        printError(ex)

        return False

    return True

def main():

    numargs = len(sys.argv)

    if numargs < 2:

        print "usage: %s command [args]" % (sys.argv[0])

        print "       %s delete-vnic vnic-name" % (sys.argv[0])

        print "       %s reset-ofports" % (sys.argv[0])

        print "       %s remove-flows" % (sys.argv[0])

        print "       %s delete-etherstub etherstub-name" % (sys.argv[0])

        sys.exit(1)

    rc = radcon.connect_unix()

    dm = rc.get_object(DatalinkManager())

    if sys.argv[1] == "delete-vnic":

        rval = deleteVNIC(sys.argv, numargs, dm)

    elif sys.argv[1] == "reset-ofports":

        rval = resetOFPorts(sys.argv, numargs, rc)

    elif sys.argv[1] == "remove-flows":

        rval = removeFlows(sys.argv, numargs, rc)

    elif sys.argv[1] == "delete-etherstub":

        rval = deleteEtherstub(sys.argv, numargs, rc, dm)

    else:

        print "\nInvalid ovs shutdown function"

        rval = False;

    sys.exit(rval == False)

if __name__ == "__main__":

    main()