iwd/test/simple-agent

#!/usr/bin/python

from gi.repository import GLib

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 = 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 = GLib.MainLoop()
    mainloop.run()