test/simple-agent

   1 #!/usr/bin/python
   2
   3 import gobject
   4
   5 import sys
   6 import dbus
   7 import dbus.service
   8 import dbus.mainloop.glib
   9
  10 class Rejected(dbus.DBusException):
  11         _dbus_error_name = "org.bluez.Error.Rejected"
  12
  13 class Agent(dbus.service.Object):
  14         exit_on_release = True
  15
  16         def set_exit_on_release(self, exit_on_release):
  17                 self.exit_on_release = exit_on_release
  18
  19         @dbus.service.method("org.bluez.Agent",
  20                                         in_signature="", out_signature="")
  21         def Release(self):
  22                 print "Release"
  23                 if self.exit_on_release:
  24                         mainloop.quit()
  25
  26         @dbus.service.method("org.bluez.Agent",
  27                                         in_signature="os", out_signature="")
  28         def Authorize(self, device, uuid):
  29                 print "Authorize (%s, %s)" % (device, uuid)
  30                 authorize = raw_input("Authorize connection (yes/no): ")
  31                 if (authorize == "yes"):
  32                         return
  33                 raise Rejected("Connection rejected by user")
  34
  35         @dbus.service.method("org.bluez.Agent",
  36                                         in_signature="o", out_signature="s")
  37         def RequestPinCode(self, device):
  38                 print "RequestPinCode (%s)" % (device)
  39                 return raw_input("Enter PIN Code: ")
  40
  41         @dbus.service.method("org.bluez.Agent",
  42                                         in_signature="o", out_signature="u")
  43         def RequestPasskey(self, device):
  44                 print "RequestPasskey (%s)" % (device)
  45                 passkey = raw_input("Enter passkey: ")
  46                 return dbus.UInt32(passkey)
  47
  48         @dbus.service.method("org.bluez.Agent",
  49                                         in_signature="ou", out_signature="")
  50         def DisplayPasskey(self, device, passkey):
  51                 print "DisplayPasskey (%s, %d)" % (device, passkey)
  52
  53         @dbus.service.method("org.bluez.Agent",
  54                                         in_signature="ou", out_signature="")
  55         def RequestConfirmation(self, device, passkey):
  56                 print "RequestConfirmation (%s, %d)" % (device, passkey)
  57                 confirm = raw_input("Confirm passkey (yes/no): ")
  58                 if (confirm == "yes"):
  59                         return
  60                 raise Rejected("Passkey doesn't match")
  61
  62         @dbus.service.method("org.bluez.Agent",
  63                                         in_signature="s", out_signature="")
  64         def ConfirmModeChange(self, mode):
  65                 print "ConfirmModeChange (%s)" % (mode)
  66                 authorize = raw_input("Authorize mode change (yes/no): ")
  67                 if (authorize == "yes"):
  68                         return
  69                 raise Rejected("Mode change by user")
  70
  71         @dbus.service.method("org.bluez.Agent",
  72                                         in_signature="", out_signature="")
  73         def Cancel(self):
  74                 print "Cancel"
  75
  76 def create_device_reply(device):
  77         print "New device (%s)" % (device)
  78         mainloop.quit()
  79
  80 def create_device_error(error):
  81         print "Creating device failed: %s" % (error)
  82         mainloop.quit()
  83
  84 if __name__ == '__main__':
  85         dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
  86
  87         bus = dbus.SystemBus()
  88         manager = dbus.Interface(bus.get_object("org.bluez", "/"),
  89                                                         "org.bluez.Manager")
  90
  91         if len(sys.argv) > 1:
  92                 path = manager.FindAdapter(sys.argv[1])
  93         else:
  94                 path = manager.DefaultAdapter()
  95
  96         adapter = dbus.Interface(bus.get_object("org.bluez", path),
  97                                                         "org.bluez.Adapter")
  98
  99         path = "/test/agent"
 100         agent = Agent(bus, path)
 101
 102         mainloop = gobject.MainLoop()
 103
 104         if len(sys.argv) > 2:
 105                 if len(sys.argv) > 3:
 106                         device = adapter.FindDevice(sys.argv[2])
 107                         adapter.RemoveDevice(device)
 108
 109                 agent.set_exit_on_release(False)
 110                 adapter.CreatePairedDevice(sys.argv[2], path, "DisplayYesNo",
 111                                         reply_handler=create_device_reply,
 112                                         error_handler=create_device_error)
 113         else:
 114                 adapter.RegisterAgent(path, "DisplayYesNo")
 115                 print "Agent registered"
 116
 117         mainloop.run()
 118
 119         #adapter.UnregisterAgent(path)
 120         #print "Agent unregistered"