blob: f53e3ea0dbe57ea9ae19c856f7a5ae11b4332e26 [file] [log] [blame]
#! /usr/bin/env python
import pprint
import os
import sys
import subprocess
import json
import argparse
import io
import time
from flask import Flask, json, Response, render_template, make_response, request
## Global Var ##
ControllerIP="127.0.0.1"
ControllerPort=8080
DEBUG=0
pp = pprint.PrettyPrinter(indent=4)
app = Flask(__name__)
## Worker Functions ##
def log_error(txt):
print '%s' % (txt)
def debug(txt):
if DEBUG:
print '%s' % (txt)
# @app.route("/wm/fprog/setrate/<dpid>/<rate>/json")
# Sample output:
# "true"
def set_rate(dpid,rate):
try:
command = "curl -s \"http://%s:%s/wm/fprog/setrate/%s/%s/json\"" % (ControllerIP, ControllerPort, dpid, rate)
debug("set_rate %s" % command)
result = os.popen(command).read()
debug("result %s" % result)
if result == "false":
print "Failed to set rate"
return;
except:
log_error("Controller IF has issue")
exit(1)
print "Sending rate to %s is successfully set to %s" % (dpid, rate)
def suspend(dpid):
try:
command = "curl -s \"http://%s:%s/wm/fprog/suspend/%s/json\"" % (ControllerIP, ControllerPort, dpid)
debug("suspend %s" % command)
result = os.popen(command).read()
debug("result %s" % result)
if result == "false":
print "Failed to suspend"
return;
except:
log_error("Controller IF has issue")
exit(1)
print "DPID %s is successfully suspended" % dpid
def resume(dpid):
try:
command = "curl -s \"http://%s:%s/wm/fprog/resume/%s/json\"" % (ControllerIP, ControllerPort, dpid)
debug("resume %s" % command)
result = os.popen(command).read()
debug("result %s" % result)
if result == "false":
print "Failed to resume"
return;
except:
log_error("Controller IF has issue")
exit(1)
print "DPID %s is successfully resumed" % dpid
def barrier(dpid):
try:
command = "curl -s \"http://%s:%s/wm/fprog/barrier/%s/json\"" % (ControllerIP, ControllerPort, dpid)
debug("barrier %s" % command)
result = os.popen(command).read()
debug("result %s" % result)
if result == "false":
print "Failed to send barrier"
return;
except:
log_error("Controller IF has issue")
exit(1)
print "Barrier reply from %s : %s" % (dpid, result)
if __name__ == "__main__":
usage_msg1 = "Usage:\n"
usage_msg2 = "%s rate <dpid> <rate> : Set sending rate[bytes/ms] to the switch\n" % (sys.argv[0])
usage_msg3 = " suspend <dpid> : Suspend sending message to the switch\n"
usage_msg4 = " resume <dpid> : Resume sending message to the switch\n"
usage_msg5 = " barrier <dpid> : Send barrier message to the switch\n"
usage_msg = usage_msg1 + usage_msg2 + usage_msg3 + usage_msg4 + usage_msg5;
app.debug = True;
# Usage info
if len(sys.argv) > 1 and (sys.argv[1] == "-h" or sys.argv[1] == "--help"):
print(usage_msg)
exit(0)
# Check arguments
if len(sys.argv) < 2:
log_error(usage_msg)
exit(1)
# Do the work
if sys.argv[1] == "rate":
if len(sys.argv) < 4:
log_error(usage_msg)
exit(1)
set_rate(sys.argv[2], sys.argv[3])
elif sys.argv[1] == "suspend":
if len(sys.argv) < 3:
log_error(usage_msg)
exit(1)
suspend(sys.argv[2])
elif sys.argv[1] == "resume":
if len(sys.argv) < 3:
log_error(usage_msg)
exit(1)
resume(sys.argv[2])
elif sys.argv[1] == "barrier":
if len(sys.argv) < 3:
log_error(usage_msg)
exit(1)
barrier(sys.argv[2])
else:
log_error(usage_msg)
exit(1)