3
0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-11-17 17:39:28 +01:00
iwd/test/simple-agent
Jukka Rissanen 7b0c04db48 test: Python script that implements a basic agent
Not meant for anything serious but shows how to use the agent
interface.
2015-02-27 10:45:40 -06:00

101 lines
2.2 KiB
Python
Executable File

#!/usr/bin/python
import gobject
import dbus
import dbus.service
import dbus.mainloop.glib
import sys
from random import randrange
class Canceled(dbus.DBusException):
_dbus_error_name = "net.connman.iwd.Error.Canceled"
class Agent(dbus.service.Object):
passphrase = None
@dbus.service.method("net.connman.iwd.Agent",
in_signature='', out_signature='')
def Release(self):
print("Release")
mainloop.quit()
def input_passphrase(self):
response = {}
if not self.passphrase:
print "Service credentials requested, type cancel to cancel"
args = raw_input('Answer: ')
for arg in args.split():
if arg.startswith("cancel"):
response["Error"] = arg
if arg.startswith("Passphrase="):
passphrase = arg.replace("Passphrase=", "", 1)
response["Passphrase"] = passphrase
else:
if self.passphrase:
response["Passphrase"] = self.passphrase
return response
@dbus.service.method("net.connman.iwd.Agent",
in_signature='o',
out_signature='s')
def RequestPassphrase(self, path):
print "RequestPassphrase (%s)" % (path)
response = {}
response.update(self.input_passphrase())
if response.has_key("Error"):
if response["Error"] == "cancel":
raise Canceled("canceled")
return
print "returning (%s)" % (response["Passphrase"])
return response["Passphrase"]
@dbus.service.method("net.connman.iwd.Agent",
in_signature='', out_signature='')
def Cancel(self):
print "Cancel"
def print_usage():
print "Usage:"
print "For WPA input:"
print "%s Passphrase=<passphrase>" % (sys.argv[0])
print "Help: %s help" % (sys.argv[0])
sys.exit(1)
if __name__ == '__main__':
if len(sys.argv) == 2 and sys.argv[1] == "help":
print_usage()
dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
bus = dbus.SystemBus()
manager = dbus.Interface(bus.get_object('net.connman.iwd', "/"),
'net.connman.iwd.Manager')
path = "/test/agent/" + str(randrange(100))
object = Agent(bus, path)
if len(sys.argv) >= 2:
for arg in sys.argv[1:]:
if arg.startswith("Passphrase="):
object.passphrase = arg.replace("Passphrase=", "", 1)
else:
print_usage()
try:
manager.RegisterAgent(path)
except:
print "Cannot register iwd agent."
mainloop = gobject.MainLoop()
mainloop.run()