wpa_supplicant/examples/p2p/p2p_stop_find.py

   1 #!/usr/bin/python
   2 # Tests p2p_stop_find
   3 ######### MAY NEED TO RUN AS SUDO #############
   4
   5 import dbus
   6 import sys, os
   7 import time
   8 import gobject
   9 import threading
  10 import getopt
  11 from dbus.mainloop.glib import DBusGMainLoop
  12
  13 def usage():
  14         print "Usage:"
  15         print "  %s -i <interface_name> \ " \
  16                 % sys.argv[0]
  17         print "                 [-w <wpas_dbus_interface>]"
  18         print "Options:"
  19         print "  -i = interface name"
  20         print "  -w = wpas dbus interface = fi.w1.wpa_supplicant1"
  21         print "Example:"
  22         print "  %s -i wlan0" % sys.argv[0]
  23
  24 # Required Signals
  25 def deviceLost(devicepath):
  26         print "Device lost: %s" % (devicepath)
  27
  28 def p2pStateChange(status):
  29         print status
  30         os._exit(0)
  31
  32 class P2P_Stop_Find (threading.Thread):
  33         # Needed Variables
  34         global bus
  35         global wpas_object
  36         global interface_object
  37         global p2p_interface
  38         global interface_name
  39         global wpas
  40         global wpas_dbus_interface
  41         global path
  42         global timeout
  43
  44         # Dbus Paths
  45         global wpas_dbus_opath
  46         global wpas_dbus_interfaces_opath
  47         global wpas_dbus_interfaces_interface
  48         global wpas_dbus_interfaces_p2pdevice
  49
  50         # Constructor
  51         def __init__(self,interface_name,wpas_dbus_interface,timeout):
  52                 # Initializes variables and threads
  53                 self.interface_name = interface_name
  54                 self.wpas_dbus_interface = wpas_dbus_interface
  55                 self.timeout = timeout
  56
  57                 # Initializes thread and daemon allows for ctrl-c kill
  58                 threading.Thread.__init__(self)
  59                 self.daemon = True
  60
  61                 # Generating interface/object paths
  62                 self.wpas_dbus_opath = "/" + \
  63                                 self.wpas_dbus_interface.replace(".","/")
  64                 self.wpas_wpas_dbus_interfaces_opath = self.wpas_dbus_opath + \
  65                                 "/Interfaces"
  66                 self.wpas_dbus_interfaces_interface = \
  67                                 self.wpas_dbus_interface + ".Interface"
  68                 self.wpas_dbus_interfaces_p2pdevice = \
  69                                 self.wpas_dbus_interfaces_interface \
  70                                 + ".P2PDevice"
  71
  72                 # Getting interfaces and objects
  73                 DBusGMainLoop(set_as_default=True)
  74                 self.bus = dbus.SystemBus()
  75                 self.wpas_object = self.bus.get_object(
  76                                 self.wpas_dbus_interface,
  77                                 self.wpas_dbus_opath)
  78                 self.wpas = dbus.Interface(self.wpas_object,
  79                                 self.wpas_dbus_interface)
  80
  81                 # Try to see if supplicant knows about interface
  82                 # If not, throw an exception
  83                 try:
  84                         self.path = self.wpas.GetInterface(
  85                                         self.interface_name)
  86                 except dbus.DBusException as exc:
  87                         error = 'Error:\n  Interface ' + self.interface_name \
  88                                 + ' was not found'
  89                         print error
  90                         usage()
  91                         os._exit(0)
  92
  93                 self.interface_object = self.bus.get_object(
  94                                 self.wpas_dbus_interface, self.path)
  95                 self.p2p_interface = dbus.Interface(self.interface_object,
  96                                 self.wpas_dbus_interfaces_p2pdevice)
  97
  98                 # Signals
  99                 self.bus.add_signal_receiver(deviceLost,
 100                         dbus_interface=self.wpas_dbus_interfaces_p2pdevice,
 101                         signal_name="DeviceLost")
 102                 self.bus.add_signal_receiver(p2pStateChange,
 103                         dbus_interface=self.wpas_dbus_interfaces_p2pdevice,
 104                         signal_name="P2PStateChanged")
 105
 106         # Runs p2p_stop_find
 107         def run(self):
 108                 # Allows other threads to keep working while MainLoop runs
 109                 # Required for timeout implementation
 110                 gobject.MainLoop().get_context().iteration(True)
 111                 gobject.threads_init()
 112                 self.p2p_interface.StopFind()
 113                 gobject.MainLoop().run()
 114
 115
 116 if __name__ == "__main__":
 117         # Needed because P2PStateChanged signal is not caught
 118         timeout = 5
 119         # Defaults for optional inputs
 120         wpas_dbus_interface = 'fi.w1.wpa_supplicant1'
 121
 122         # interface_name is required
 123         interface_name = None
 124
 125         # Using getopts to handle options
 126         try:
 127                 options, args = getopt.getopt(sys.argv[1:],"ht:i:w:")
 128
 129         except getopt.GetoptError:
 130                 usage()
 131                 quit()
 132
 133         # If theres a switch, override default option
 134         for key, value in options:
 135                 # Help
 136                 if (key == "-h"):
 137                         usage()
 138                         quit()
 139                 # Interface Name
 140                 elif (key == "-i"):
 141                         interface_name = value
 142                 # Dbus interface
 143                 elif (key == "-w"):
 144                         wpas_dbus_interface = value
 145                 else:
 146                         assert False, "unhandled option"
 147
 148         # Interface name is required and was not given
 149         if (interface_name == None):
 150                 print "Error:\n  interface_name is required"
 151                 usage()
 152                 quit()
 153
 154         # Constructor
 155         try:
 156                 p2p_stop_find_test = P2P_Stop_Find(interface_name,
 157                                                 wpas_dbus_interface,timeout)
 158
 159         except:
 160                 print "Error:\n  Invalid wpas_dbus_interface"
 161                 usage()
 162                 quit()
 163
 164         # Start P2P_Find
 165         p2p_stop_find_test.start()
 166
 167         try:
 168                 time.sleep(int(p2p_stop_find_test.timeout))
 169
 170         except:
 171                 pass
 172
 173         print "p2p find stopped"
 174         quit()