onos_websocket.py 文件源码

def run(params):
  signal.signal(signal.SIGINT, signal_handler) #Assign the signal handler

  if(sdnpwn.checkArg(["-t", "--target"], params)):

    port = str(sdnpwn.getArg(["-p", "--port"], params, 8181))
    cookie = sdnpwn.getArg(["-c", "--cookie"], params, None)

    wsURL = "ws://" + sdnpwn.getArg(["-t", "--target"], params) + ":" + port + "/onos/ui/websock/core"
    #websocket.enableTrace(True)
    ws = websocket.WebSocketApp(wsURL)
    ws.on_open = onOpen
    ws.on_message = onMessage
    ws.on_error = onError
    ws.on_close = onClose
    ws.sdnpwn_params = params
    ws.sdnpwn_expected_events = 1 #Execting initial bootstrap event

    if(cookie is not None):
      ws.cookie = cookie

    sdnpwn.printNormal("Attempting connection to " + wsURL)

    ws.run_forever()

  else:
    print(usage())