1 // SPDX-License-Identifier: GPL-2.0-only
2
3 #include "netlink.h"
4 #include "common.h"
5 #include "bitset.h"
6
7 struct wol_req_info {
8 struct ethnl_req_info base;
9 };
10
11 struct wol_reply_data {
12 struct ethnl_reply_data base;
13 struct ethtool_wolinfo wol;
14 bool show_sopass;
15 };
16
17 #define WOL_REPDATA(__reply_base) \
18 container_of(__reply_base, struct wol_reply_data, base)
19
20 const struct nla_policy ethnl_wol_get_policy[] = {
21 [ETHTOOL_A_WOL_HEADER] =
22 NLA_POLICY_NESTED(ethnl_header_policy),
23 };
24
wol_prepare_data(const struct ethnl_req_info * req_base,struct ethnl_reply_data * reply_base,struct genl_info * info)25 static int wol_prepare_data(const struct ethnl_req_info *req_base,
26 struct ethnl_reply_data *reply_base,
27 struct genl_info *info)
28 {
29 struct wol_reply_data *data = WOL_REPDATA(reply_base);
30 struct net_device *dev = reply_base->dev;
31 int ret;
32
33 if (!dev->ethtool_ops->get_wol)
34 return -EOPNOTSUPP;
35
36 ret = ethnl_ops_begin(dev);
37 if (ret < 0)
38 return ret;
39 dev->ethtool_ops->get_wol(dev, &data->wol);
40 ethnl_ops_complete(dev);
41 /* do not include password in notifications */
42 data->show_sopass = info && (data->wol.supported & WAKE_MAGICSECURE);
43
44 return 0;
45 }
46
wol_reply_size(const struct ethnl_req_info * req_base,const struct ethnl_reply_data * reply_base)47 static int wol_reply_size(const struct ethnl_req_info *req_base,
48 const struct ethnl_reply_data *reply_base)
49 {
50 bool compact = req_base->flags & ETHTOOL_FLAG_COMPACT_BITSETS;
51 const struct wol_reply_data *data = WOL_REPDATA(reply_base);
52 int len;
53
54 len = ethnl_bitset32_size(&data->wol.wolopts, &data->wol.supported,
55 WOL_MODE_COUNT, wol_mode_names, compact);
56 if (len < 0)
57 return len;
58 if (data->show_sopass)
59 len += nla_total_size(sizeof(data->wol.sopass));
60
61 return len;
62 }
63
wol_fill_reply(struct sk_buff * skb,const struct ethnl_req_info * req_base,const struct ethnl_reply_data * reply_base)64 static int wol_fill_reply(struct sk_buff *skb,
65 const struct ethnl_req_info *req_base,
66 const struct ethnl_reply_data *reply_base)
67 {
68 bool compact = req_base->flags & ETHTOOL_FLAG_COMPACT_BITSETS;
69 const struct wol_reply_data *data = WOL_REPDATA(reply_base);
70 int ret;
71
72 ret = ethnl_put_bitset32(skb, ETHTOOL_A_WOL_MODES, &data->wol.wolopts,
73 &data->wol.supported, WOL_MODE_COUNT,
74 wol_mode_names, compact);
75 if (ret < 0)
76 return ret;
77 if (data->show_sopass &&
78 nla_put(skb, ETHTOOL_A_WOL_SOPASS, sizeof(data->wol.sopass),
79 data->wol.sopass))
80 return -EMSGSIZE;
81
82 return 0;
83 }
84
85 const struct ethnl_request_ops ethnl_wol_request_ops = {
86 .request_cmd = ETHTOOL_MSG_WOL_GET,
87 .reply_cmd = ETHTOOL_MSG_WOL_GET_REPLY,
88 .hdr_attr = ETHTOOL_A_WOL_HEADER,
89 .req_info_size = sizeof(struct wol_req_info),
90 .reply_data_size = sizeof(struct wol_reply_data),
91
92 .prepare_data = wol_prepare_data,
93 .reply_size = wol_reply_size,
94 .fill_reply = wol_fill_reply,
95 };
96
97 /* WOL_SET */
98
99 const struct nla_policy ethnl_wol_set_policy[] = {
100 [ETHTOOL_A_WOL_HEADER] =
101 NLA_POLICY_NESTED(ethnl_header_policy),
102 [ETHTOOL_A_WOL_MODES] = { .type = NLA_NESTED },
103 [ETHTOOL_A_WOL_SOPASS] = { .type = NLA_BINARY,
104 .len = SOPASS_MAX },
105 };
106
ethnl_set_wol(struct sk_buff * skb,struct genl_info * info)107 int ethnl_set_wol(struct sk_buff *skb, struct genl_info *info)
108 {
109 struct ethtool_wolinfo wol = { .cmd = ETHTOOL_GWOL };
110 struct ethnl_req_info req_info = {};
111 struct nlattr **tb = info->attrs;
112 struct net_device *dev;
113 bool mod = false;
114 int ret;
115
116 ret = ethnl_parse_header_dev_get(&req_info, tb[ETHTOOL_A_WOL_HEADER],
117 genl_info_net(info), info->extack,
118 true);
119 if (ret < 0)
120 return ret;
121 dev = req_info.dev;
122 ret = -EOPNOTSUPP;
123 if (!dev->ethtool_ops->get_wol || !dev->ethtool_ops->set_wol)
124 goto out_dev;
125
126 rtnl_lock();
127 ret = ethnl_ops_begin(dev);
128 if (ret < 0)
129 goto out_rtnl;
130
131 dev->ethtool_ops->get_wol(dev, &wol);
132 ret = ethnl_update_bitset32(&wol.wolopts, WOL_MODE_COUNT,
133 tb[ETHTOOL_A_WOL_MODES], wol_mode_names,
134 info->extack, &mod);
135 if (ret < 0)
136 goto out_ops;
137 if (wol.wolopts & ~wol.supported) {
138 NL_SET_ERR_MSG_ATTR(info->extack, tb[ETHTOOL_A_WOL_MODES],
139 "cannot enable unsupported WoL mode");
140 ret = -EINVAL;
141 goto out_ops;
142 }
143 if (tb[ETHTOOL_A_WOL_SOPASS]) {
144 if (!(wol.supported & WAKE_MAGICSECURE)) {
145 NL_SET_ERR_MSG_ATTR(info->extack,
146 tb[ETHTOOL_A_WOL_SOPASS],
147 "magicsecure not supported, cannot set password");
148 ret = -EINVAL;
149 goto out_ops;
150 }
151 ethnl_update_binary(wol.sopass, sizeof(wol.sopass),
152 tb[ETHTOOL_A_WOL_SOPASS], &mod);
153 }
154
155 if (!mod)
156 goto out_ops;
157 ret = dev->ethtool_ops->set_wol(dev, &wol);
158 if (ret)
159 goto out_ops;
160 dev->wol_enabled = !!wol.wolopts;
161 ethtool_notify(dev, ETHTOOL_MSG_WOL_NTF, NULL);
162
163 out_ops:
164 ethnl_ops_complete(dev);
165 out_rtnl:
166 rtnl_unlock();
167 out_dev:
168 dev_put(dev);
169 return ret;
170 }
171