1 #include <net/if.h> 2 #include <errno.h> 3 #include <string.h> 4 5 #include <netlink/genl/genl.h> 6 #include <netlink/genl/family.h> 7 #include <netlink/genl/ctrl.h> 8 #include <netlink/msg.h> 9 #include <netlink/attr.h> 10 11 #include "nl80211.h" 12 #include "iw.h" 13 14 /* These enums need to be kept in sync with the kernel */ 15 enum hwsim_testmode_attr { 16 __HWSIM_TM_ATTR_INVALID = 0, 17 HWSIM_TM_ATTR_CMD = 1, 18 HWSIM_TM_ATTR_PS = 2, 19 20 /* keep last */ 21 __HWSIM_TM_ATTR_AFTER_LAST, 22 HWSIM_TM_ATTR_MAX = __HWSIM_TM_ATTR_AFTER_LAST - 1 23 }; 24 25 enum hwsim_testmode_cmd { 26 HWSIM_TM_CMD_SET_PS = 0, 27 HWSIM_TM_CMD_GET_PS = 1, 28 HWSIM_TM_CMD_STOP_QUEUES = 2, 29 HWSIM_TM_CMD_WAKE_QUEUES = 3, 30 }; 31 32 33 SECTION(hwsim); 34 35 static int print_hwsim_ps_handler(struct nl_msg *msg, void *arg) 36 { 37 struct nlattr *attrs[NL80211_ATTR_MAX + 1]; 38 struct nlattr *tb[HWSIM_TM_ATTR_MAX + 1]; 39 struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg)); 40 41 nla_parse(attrs, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), 42 genlmsg_attrlen(gnlh, 0), NULL); 43 44 if (!attrs[NL80211_ATTR_TESTDATA]) 45 return NL_SKIP; 46 47 nla_parse(tb, HWSIM_TM_ATTR_MAX, nla_data(attrs[NL80211_ATTR_TESTDATA]), 48 nla_len(attrs[NL80211_ATTR_TESTDATA]), NULL); 49 50 printf("HWSIM PS: %d\n", nla_get_u32(tb[HWSIM_TM_ATTR_PS])); 51 52 return NL_SKIP; 53 } 54 55 static int handle_hwsim_getps(struct nl80211_state *state, struct nl_cb *cb, 56 struct nl_msg *msg, int argc, char **argv, 57 enum id_input id) 58 { 59 struct nlattr *tmdata; 60 61 tmdata = nla_nest_start(msg, NL80211_ATTR_TESTDATA); 62 if (!tmdata) 63 goto nla_put_failure; 64 65 NLA_PUT_U32(msg, HWSIM_TM_ATTR_CMD, HWSIM_TM_CMD_GET_PS); 66 67 nla_nest_end(msg, tmdata); 68 69 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, 70 print_hwsim_ps_handler, NULL); 71 return 0; 72 nla_put_failure: 73 return -ENOBUFS; 74 } 75 COMMAND(hwsim, getps, "", NL80211_CMD_TESTMODE, 0, CIB_PHY, handle_hwsim_getps, ""); 76 77 static int handle_hwsim_setps(struct nl80211_state *state, struct nl_cb *cb, 78 struct nl_msg *msg, int argc, char **argv, 79 enum id_input id) 80 { 81 struct nlattr *tmdata; 82 __u32 ps; 83 char *end; 84 85 if (argc != 1) 86 return 1; 87 88 ps = strtoul(argv[0], &end, 0); 89 if (*end) 90 return 1; 91 92 tmdata = nla_nest_start(msg, NL80211_ATTR_TESTDATA); 93 if (!tmdata) 94 goto nla_put_failure; 95 96 NLA_PUT_U32(msg, HWSIM_TM_ATTR_CMD, HWSIM_TM_CMD_SET_PS); 97 NLA_PUT_U32(msg, HWSIM_TM_ATTR_PS, ps); 98 99 nla_nest_end(msg, tmdata); 100 101 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, 102 print_hwsim_ps_handler, NULL); 103 return 0; 104 nla_put_failure: 105 return -ENOBUFS; 106 } 107 COMMAND(hwsim, setps, "<value>", NL80211_CMD_TESTMODE, 0, CIB_PHY, handle_hwsim_setps, ""); 108 109 static int handle_hwsim_stop_queues(struct nl80211_state *state, struct nl_cb *cb, 110 struct nl_msg *msg, int argc, char **argv, 111 enum id_input id) 112 { 113 struct nlattr *tmdata; 114 115 if (argc != 0) 116 return 1; 117 118 tmdata = nla_nest_start(msg, NL80211_ATTR_TESTDATA); 119 if (!tmdata) 120 goto nla_put_failure; 121 122 NLA_PUT_U32(msg, HWSIM_TM_ATTR_CMD, HWSIM_TM_CMD_STOP_QUEUES); 123 124 nla_nest_end(msg, tmdata); 125 return 0; 126 nla_put_failure: 127 return -ENOBUFS; 128 } 129 COMMAND(hwsim, stopqueues, "", NL80211_CMD_TESTMODE, 0, CIB_PHY, handle_hwsim_stop_queues, ""); 130 131 static int handle_hwsim_wake_queues(struct nl80211_state *state, struct nl_cb *cb, 132 struct nl_msg *msg, int argc, char **argv, 133 enum id_input id) 134 { 135 struct nlattr *tmdata; 136 137 if (argc != 0) 138 return 1; 139 140 tmdata = nla_nest_start(msg, NL80211_ATTR_TESTDATA); 141 if (!tmdata) 142 goto nla_put_failure; 143 144 NLA_PUT_U32(msg, HWSIM_TM_ATTR_CMD, HWSIM_TM_CMD_WAKE_QUEUES); 145 146 nla_nest_end(msg, tmdata); 147 return 0; 148 nla_put_failure: 149 return -ENOBUFS; 150 } 151 COMMAND(hwsim, wakequeues, "", NL80211_CMD_TESTMODE, 0, CIB_PHY, handle_hwsim_wake_queues, ""); 152
