From d0709b885c5da2c81e21d2021fd60c6437b4cd95 Mon Sep 17 00:00:00 2001
From: Ian Gulliver <flamingcow@gmail.com>
Date: Sat, 10 Jan 2015 16:39:24 +0000
Subject: [PATCH] Add Query()

---
 nl80211.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/nl80211.py b/nl80211.py
index c1ffbea..cbe27f3 100755
--- a/nl80211.py
+++ b/nl80211.py
@@ -313,6 +313,10 @@ class GenericNetlink(object):
       parser = self._msgtypes_by_id[msgtype]['parser']
       yield (genlhdr['cmd'], parser.Unpack(iterator))
 
+  def Query(self, msgtype, flags, cmd, version, **attrs):
+    self.Send(msgtype, flags, cmd, version, **attrs)
+    return self.Recv()
+
 
 def RegisterNL80211(gnl):
   rate_info = Attributes({
@@ -383,5 +387,4 @@ def GetIfIndex(if_name):
 
 gnl = GenericNetlink()
 RegisterNL80211(gnl)
-gnl.Send('nl80211', ['dump'], 'get_station', 0, ifindex=GetIfIndex('wlan0'))
-print list(gnl.Recv())
+print list(gnl.Query('nl80211', ['dump'], 'get_station', 0, ifindex=GetIfIndex('wlan0')))