[PATCH 2/2] iw: allow to configure dynamic power save timeout

[Stanislaw Gruszka <sgruszka@redhat.com>]

Signed-off-by: Stanislaw Gruszka <sgruszka@redhat.com>
---
 ps.c | 38 +++++++++++++++++++++++++++++---------
 1 file changed, 29 insertions(+), 9 deletions(-)

diff --git a/ps.c b/ps.c
index 4c8e2d1..83ee9d5 100644
--- a/ps.c
+++ b/ps.c
@@ -15,19 +15,29 @@ static int set_power_save(struct nl80211_state *state,
 			  enum id_input id)
 {
 	enum nl80211_ps_state ps_state;
+	int timeout;
+	char *endp;
 
-	if (argc != 1) {
-		printf("Invalid parameters!\n");
-		return 2;
-	}
+	if (argc != 1 && argc != 3)
+		goto invalid_parameters;
 
 	if (strcmp(argv[0], "on") == 0)
 		ps_state = NL80211_PS_ENABLED;
 	else if (strcmp(argv[0], "off") == 0)
 		ps_state = NL80211_PS_DISABLED;
-	else {
-		printf("Invalid parameter: %s\n", argv[0]);
-		return 2;
+	else
+		goto invalid_parameters;
+
+	if (argc == 3) {
+		if (ps_state != NL80211_PS_ENABLED)
+			goto invalid_parameters;
+		if (strcmp(argv[1], "timeout") != 0)
+			goto invalid_parameters;
+		timeout = (int) strtol(argv[2], &endp, 10);
+		if (endp == argv[2] || (timeout <= 0 && timeout != -1))
+			goto invalid_parameters;
+
+		NLA_PUT_U32(msg, NL80211_ATTR_PS_TIMEOUT, timeout);
 	}
 
 	NLA_PUT_U32(msg, NL80211_ATTR_PS_STATE, ps_state);
@@ -36,9 +46,13 @@ static int set_power_save(struct nl80211_state *state,
 
  nla_put_failure:
 	return -ENOBUFS;
+
+ invalid_parameters:
+	printf("Invalid parameters!\n");
+	return 2;
 }
 
-COMMAND(set, power_save, "<on|off>",
+COMMAND(set, power_save, "<on [timeout <value>] | off>",
 	NL80211_CMD_SET_POWER_SAVE, 0, CIB_NETDEV, set_power_save,
 	"Set power save state to on or off.");
 
@@ -47,6 +61,7 @@ static int print_power_save_handler(struct nl_msg *msg, void *arg)
 	struct nlattr *attrs[NL80211_ATTR_MAX + 1];
 	struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 	const char *s;
+	int timeout = -1;
 
 	nla_parse(attrs, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 		  genlmsg_attrlen(gnlh, 0), NULL);
@@ -57,6 +72,8 @@ static int print_power_save_handler(struct nl_msg *msg, void *arg)
 	switch (nla_get_u32(attrs[NL80211_ATTR_PS_STATE])) {
 	case NL80211_PS_ENABLED:
 		s = "on";
+		if (attrs[NL80211_ATTR_PS_TIMEOUT])
+			timeout = nla_get_u32(attrs[NL80211_ATTR_PS_TIMEOUT]);
 		break;
 	case NL80211_PS_DISABLED:
 	default:
@@ -64,7 +81,10 @@ static int print_power_save_handler(struct nl_msg *msg, void *arg)
 		break;
 	}
 
-	printf("Power save: %s\n", s);
+	if (timeout > 0)
+		printf("Power save: on, timeout %d ms\n", timeout);
+	else
+		printf("Power save: %s\n", s);
 
 	return NL_SKIP;
 }
-- 
1.8.3.1