e7da45d0f7400f37b9974c10492dc03a7423e3a5
[oweals/openwrt.git] / target / linux / generic / files / drivers / net / phy / swconfig.c
1 /*
2  * swconfig.c: Switch configuration API
3  *
4  * Copyright (C) 2008 Felix Fietkau <nbd@nbd.name>
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  */
16
17 #include <linux/types.h>
18 #include <linux/module.h>
19 #include <linux/init.h>
20 #include <linux/list.h>
21 #include <linux/if.h>
22 #include <linux/if_ether.h>
23 #include <linux/capability.h>
24 #include <linux/skbuff.h>
25 #include <linux/switch.h>
26 #include <linux/of.h>
27 #include <linux/version.h>
28 #include <uapi/linux/mii.h>
29
30 #define SWCONFIG_DEVNAME        "switch%d"
31
32 #include "swconfig_leds.c"
33
34 MODULE_AUTHOR("Felix Fietkau <nbd@nbd.name>");
35 MODULE_LICENSE("GPL");
36
37 static int swdev_id;
38 static struct list_head swdevs;
39 static DEFINE_MUTEX(swdevs_lock);
40 struct swconfig_callback;
41
42 struct swconfig_callback {
43         struct sk_buff *msg;
44         struct genlmsghdr *hdr;
45         struct genl_info *info;
46         int cmd;
47
48         /* callback for filling in the message data */
49         int (*fill)(struct swconfig_callback *cb, void *arg);
50
51         /* callback for closing the message before sending it */
52         int (*close)(struct swconfig_callback *cb, void *arg);
53
54         struct nlattr *nest[4];
55         int args[4];
56 };
57
58 /* defaults */
59
60 static int
61 swconfig_get_vlan_ports(struct switch_dev *dev, const struct switch_attr *attr,
62                         struct switch_val *val)
63 {
64         int ret;
65         if (val->port_vlan >= dev->vlans)
66                 return -EINVAL;
67
68         if (!dev->ops->get_vlan_ports)
69                 return -EOPNOTSUPP;
70
71         ret = dev->ops->get_vlan_ports(dev, val);
72         return ret;
73 }
74
75 static int
76 swconfig_set_vlan_ports(struct switch_dev *dev, const struct switch_attr *attr,
77                         struct switch_val *val)
78 {
79         struct switch_port *ports = val->value.ports;
80         const struct switch_dev_ops *ops = dev->ops;
81         int i;
82
83         if (val->port_vlan >= dev->vlans)
84                 return -EINVAL;
85
86         /* validate ports */
87         if (val->len > dev->ports)
88                 return -EINVAL;
89
90         if (!ops->set_vlan_ports)
91                 return -EOPNOTSUPP;
92
93         for (i = 0; i < val->len; i++) {
94                 if (ports[i].id >= dev->ports)
95                         return -EINVAL;
96
97                 if (ops->set_port_pvid &&
98                     !(ports[i].flags & (1 << SWITCH_PORT_FLAG_TAGGED)))
99                         ops->set_port_pvid(dev, ports[i].id, val->port_vlan);
100         }
101
102         return ops->set_vlan_ports(dev, val);
103 }
104
105 static int
106 swconfig_set_pvid(struct switch_dev *dev, const struct switch_attr *attr,
107                         struct switch_val *val)
108 {
109         if (val->port_vlan >= dev->ports)
110                 return -EINVAL;
111
112         if (!dev->ops->set_port_pvid)
113                 return -EOPNOTSUPP;
114
115         return dev->ops->set_port_pvid(dev, val->port_vlan, val->value.i);
116 }
117
118 static int
119 swconfig_get_pvid(struct switch_dev *dev, const struct switch_attr *attr,
120                         struct switch_val *val)
121 {
122         if (val->port_vlan >= dev->ports)
123                 return -EINVAL;
124
125         if (!dev->ops->get_port_pvid)
126                 return -EOPNOTSUPP;
127
128         return dev->ops->get_port_pvid(dev, val->port_vlan, &val->value.i);
129 }
130
131 static int
132 swconfig_set_link(struct switch_dev *dev, const struct switch_attr *attr,
133                         struct switch_val *val)
134 {
135         if (!dev->ops->set_port_link)
136                 return -EOPNOTSUPP;
137
138         return dev->ops->set_port_link(dev, val->port_vlan, val->value.link);
139 }
140
141 static int
142 swconfig_get_link(struct switch_dev *dev, const struct switch_attr *attr,
143                         struct switch_val *val)
144 {
145         struct switch_port_link *link = val->value.link;
146
147         if (val->port_vlan >= dev->ports)
148                 return -EINVAL;
149
150         if (!dev->ops->get_port_link)
151                 return -EOPNOTSUPP;
152
153         memset(link, 0, sizeof(*link));
154         return dev->ops->get_port_link(dev, val->port_vlan, link);
155 }
156
157 static int
158 swconfig_apply_config(struct switch_dev *dev, const struct switch_attr *attr,
159                         struct switch_val *val)
160 {
161         /* don't complain if not supported by the switch driver */
162         if (!dev->ops->apply_config)
163                 return 0;
164
165         return dev->ops->apply_config(dev);
166 }
167
168 static int
169 swconfig_reset_switch(struct switch_dev *dev, const struct switch_attr *attr,
170                         struct switch_val *val)
171 {
172         /* don't complain if not supported by the switch driver */
173         if (!dev->ops->reset_switch)
174                 return 0;
175
176         return dev->ops->reset_switch(dev);
177 }
178
179 enum global_defaults {
180         GLOBAL_APPLY,
181         GLOBAL_RESET,
182 };
183
184 enum vlan_defaults {
185         VLAN_PORTS,
186 };
187
188 enum port_defaults {
189         PORT_PVID,
190         PORT_LINK,
191 };
192
193 static struct switch_attr default_global[] = {
194         [GLOBAL_APPLY] = {
195                 .type = SWITCH_TYPE_NOVAL,
196                 .name = "apply",
197                 .description = "Activate changes in the hardware",
198                 .set = swconfig_apply_config,
199         },
200         [GLOBAL_RESET] = {
201                 .type = SWITCH_TYPE_NOVAL,
202                 .name = "reset",
203                 .description = "Reset the switch",
204                 .set = swconfig_reset_switch,
205         }
206 };
207
208 static struct switch_attr default_port[] = {
209         [PORT_PVID] = {
210                 .type = SWITCH_TYPE_INT,
211                 .name = "pvid",
212                 .description = "Primary VLAN ID",
213                 .set = swconfig_set_pvid,
214                 .get = swconfig_get_pvid,
215         },
216         [PORT_LINK] = {
217                 .type = SWITCH_TYPE_LINK,
218                 .name = "link",
219                 .description = "Get port link information",
220                 .set = swconfig_set_link,
221                 .get = swconfig_get_link,
222         }
223 };
224
225 static struct switch_attr default_vlan[] = {
226         [VLAN_PORTS] = {
227                 .type = SWITCH_TYPE_PORTS,
228                 .name = "ports",
229                 .description = "VLAN port mapping",
230                 .set = swconfig_set_vlan_ports,
231                 .get = swconfig_get_vlan_ports,
232         },
233 };
234
235 static const struct switch_attr *
236 swconfig_find_attr_by_name(const struct switch_attrlist *alist,
237                                 const char *name)
238 {
239         int i;
240
241         for (i = 0; i < alist->n_attr; i++)
242                 if (strcmp(name, alist->attr[i].name) == 0)
243                         return &alist->attr[i];
244
245         return NULL;
246 }
247
248 static void swconfig_defaults_init(struct switch_dev *dev)
249 {
250         const struct switch_dev_ops *ops = dev->ops;
251
252         dev->def_global = 0;
253         dev->def_vlan = 0;
254         dev->def_port = 0;
255
256         if (ops->get_vlan_ports || ops->set_vlan_ports)
257                 set_bit(VLAN_PORTS, &dev->def_vlan);
258
259         if (ops->get_port_pvid || ops->set_port_pvid)
260                 set_bit(PORT_PVID, &dev->def_port);
261
262         if (ops->get_port_link &&
263             !swconfig_find_attr_by_name(&ops->attr_port, "link"))
264                 set_bit(PORT_LINK, &dev->def_port);
265
266         /* always present, can be no-op */
267         set_bit(GLOBAL_APPLY, &dev->def_global);
268         set_bit(GLOBAL_RESET, &dev->def_global);
269 }
270
271
272 static struct genl_family switch_fam;
273
274 static const struct nla_policy switch_policy[SWITCH_ATTR_MAX+1] = {
275         [SWITCH_ATTR_ID] = { .type = NLA_U32 },
276         [SWITCH_ATTR_OP_ID] = { .type = NLA_U32 },
277         [SWITCH_ATTR_OP_PORT] = { .type = NLA_U32 },
278         [SWITCH_ATTR_OP_VLAN] = { .type = NLA_U32 },
279         [SWITCH_ATTR_OP_VALUE_INT] = { .type = NLA_U32 },
280         [SWITCH_ATTR_OP_VALUE_STR] = { .type = NLA_NUL_STRING },
281         [SWITCH_ATTR_OP_VALUE_PORTS] = { .type = NLA_NESTED },
282         [SWITCH_ATTR_TYPE] = { .type = NLA_U32 },
283 };
284
285 static const struct nla_policy port_policy[SWITCH_PORT_ATTR_MAX+1] = {
286         [SWITCH_PORT_ID] = { .type = NLA_U32 },
287         [SWITCH_PORT_FLAG_TAGGED] = { .type = NLA_FLAG },
288 };
289
290 static struct nla_policy link_policy[SWITCH_LINK_ATTR_MAX] = {
291         [SWITCH_LINK_FLAG_DUPLEX] = { .type = NLA_FLAG },
292         [SWITCH_LINK_FLAG_ANEG] = { .type = NLA_FLAG },
293         [SWITCH_LINK_SPEED] = { .type = NLA_U32 },
294 };
295
296 static inline void
297 swconfig_lock(void)
298 {
299         mutex_lock(&swdevs_lock);
300 }
301
302 static inline void
303 swconfig_unlock(void)
304 {
305         mutex_unlock(&swdevs_lock);
306 }
307
308 static struct switch_dev *
309 swconfig_get_dev(struct genl_info *info)
310 {
311         struct switch_dev *dev = NULL;
312         struct switch_dev *p;
313         int id;
314
315         if (!info->attrs[SWITCH_ATTR_ID])
316                 goto done;
317
318         id = nla_get_u32(info->attrs[SWITCH_ATTR_ID]);
319         swconfig_lock();
320         list_for_each_entry(p, &swdevs, dev_list) {
321                 if (id != p->id)
322                         continue;
323
324                 dev = p;
325                 break;
326         }
327         if (dev)
328                 mutex_lock(&dev->sw_mutex);
329         else
330                 pr_debug("device %d not found\n", id);
331         swconfig_unlock();
332 done:
333         return dev;
334 }
335
336 static inline void
337 swconfig_put_dev(struct switch_dev *dev)
338 {
339         mutex_unlock(&dev->sw_mutex);
340 }
341
342 static int
343 swconfig_dump_attr(struct swconfig_callback *cb, void *arg)
344 {
345         struct switch_attr *op = arg;
346         struct genl_info *info = cb->info;
347         struct sk_buff *msg = cb->msg;
348         int id = cb->args[0];
349         void *hdr;
350
351         hdr = genlmsg_put(msg, info->snd_portid, info->snd_seq, &switch_fam,
352                         NLM_F_MULTI, SWITCH_CMD_NEW_ATTR);
353         if (IS_ERR(hdr))
354                 return -1;
355
356         if (nla_put_u32(msg, SWITCH_ATTR_OP_ID, id))
357                 goto nla_put_failure;
358         if (nla_put_u32(msg, SWITCH_ATTR_OP_TYPE, op->type))
359                 goto nla_put_failure;
360         if (nla_put_string(msg, SWITCH_ATTR_OP_NAME, op->name))
361                 goto nla_put_failure;
362         if (op->description)
363                 if (nla_put_string(msg, SWITCH_ATTR_OP_DESCRIPTION,
364                         op->description))
365                         goto nla_put_failure;
366
367         genlmsg_end(msg, hdr);
368         return msg->len;
369 nla_put_failure:
370         genlmsg_cancel(msg, hdr);
371         return -EMSGSIZE;
372 }
373
374 /* spread multipart messages across multiple message buffers */
375 static int
376 swconfig_send_multipart(struct swconfig_callback *cb, void *arg)
377 {
378         struct genl_info *info = cb->info;
379         int restart = 0;
380         int err;
381
382         do {
383                 if (!cb->msg) {
384                         cb->msg = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
385                         if (cb->msg == NULL)
386                                 goto error;
387                 }
388
389                 if (!(cb->fill(cb, arg) < 0))
390                         break;
391
392                 /* fill failed, check if this was already the second attempt */
393                 if (restart)
394                         goto error;
395
396                 /* try again in a new message, send the current one */
397                 restart = 1;
398                 if (cb->close) {
399                         if (cb->close(cb, arg) < 0)
400                                 goto error;
401                 }
402                 err = genlmsg_reply(cb->msg, info);
403                 cb->msg = NULL;
404                 if (err < 0)
405                         goto error;
406
407         } while (restart);
408
409         return 0;
410
411 error:
412         if (cb->msg)
413                 nlmsg_free(cb->msg);
414         return -1;
415 }
416
417 static int
418 swconfig_list_attrs(struct sk_buff *skb, struct genl_info *info)
419 {
420         struct genlmsghdr *hdr = nlmsg_data(info->nlhdr);
421         const struct switch_attrlist *alist;
422         struct switch_dev *dev;
423         struct swconfig_callback cb;
424         int err = -EINVAL;
425         int i;
426
427         /* defaults */
428         struct switch_attr *def_list;
429         unsigned long *def_active;
430         int n_def;
431
432         dev = swconfig_get_dev(info);
433         if (!dev)
434                 return -EINVAL;
435
436         switch (hdr->cmd) {
437         case SWITCH_CMD_LIST_GLOBAL:
438                 alist = &dev->ops->attr_global;
439                 def_list = default_global;
440                 def_active = &dev->def_global;
441                 n_def = ARRAY_SIZE(default_global);
442                 break;
443         case SWITCH_CMD_LIST_VLAN:
444                 alist = &dev->ops->attr_vlan;
445                 def_list = default_vlan;
446                 def_active = &dev->def_vlan;
447                 n_def = ARRAY_SIZE(default_vlan);
448                 break;
449         case SWITCH_CMD_LIST_PORT:
450                 alist = &dev->ops->attr_port;
451                 def_list = default_port;
452                 def_active = &dev->def_port;
453                 n_def = ARRAY_SIZE(default_port);
454                 break;
455         default:
456                 WARN_ON(1);
457                 goto out;
458         }
459
460         memset(&cb, 0, sizeof(cb));
461         cb.info = info;
462         cb.fill = swconfig_dump_attr;
463         for (i = 0; i < alist->n_attr; i++) {
464                 if (alist->attr[i].disabled)
465                         continue;
466                 cb.args[0] = i;
467                 err = swconfig_send_multipart(&cb, (void *) &alist->attr[i]);
468                 if (err < 0)
469                         goto error;
470         }
471
472         /* defaults */
473         for (i = 0; i < n_def; i++) {
474                 if (!test_bit(i, def_active))
475                         continue;
476                 cb.args[0] = SWITCH_ATTR_DEFAULTS_OFFSET + i;
477                 err = swconfig_send_multipart(&cb, (void *) &def_list[i]);
478                 if (err < 0)
479                         goto error;
480         }
481         swconfig_put_dev(dev);
482
483         if (!cb.msg)
484                 return 0;
485
486         return genlmsg_reply(cb.msg, info);
487
488 error:
489         if (cb.msg)
490                 nlmsg_free(cb.msg);
491 out:
492         swconfig_put_dev(dev);
493         return err;
494 }
495
496 static const struct switch_attr *
497 swconfig_lookup_attr(struct switch_dev *dev, struct genl_info *info,
498                 struct switch_val *val)
499 {
500         struct genlmsghdr *hdr = nlmsg_data(info->nlhdr);
501         const struct switch_attrlist *alist;
502         const struct switch_attr *attr = NULL;
503         unsigned int attr_id;
504
505         /* defaults */
506         struct switch_attr *def_list;
507         unsigned long *def_active;
508         int n_def;
509
510         if (!info->attrs[SWITCH_ATTR_OP_ID])
511                 goto done;
512
513         switch (hdr->cmd) {
514         case SWITCH_CMD_SET_GLOBAL:
515         case SWITCH_CMD_GET_GLOBAL:
516                 alist = &dev->ops->attr_global;
517                 def_list = default_global;
518                 def_active = &dev->def_global;
519                 n_def = ARRAY_SIZE(default_global);
520                 break;
521         case SWITCH_CMD_SET_VLAN:
522         case SWITCH_CMD_GET_VLAN:
523                 alist = &dev->ops->attr_vlan;
524                 def_list = default_vlan;
525                 def_active = &dev->def_vlan;
526                 n_def = ARRAY_SIZE(default_vlan);
527                 if (!info->attrs[SWITCH_ATTR_OP_VLAN])
528                         goto done;
529                 val->port_vlan = nla_get_u32(info->attrs[SWITCH_ATTR_OP_VLAN]);
530                 if (val->port_vlan >= dev->vlans)
531                         goto done;
532                 break;
533         case SWITCH_CMD_SET_PORT:
534         case SWITCH_CMD_GET_PORT:
535                 alist = &dev->ops->attr_port;
536                 def_list = default_port;
537                 def_active = &dev->def_port;
538                 n_def = ARRAY_SIZE(default_port);
539                 if (!info->attrs[SWITCH_ATTR_OP_PORT])
540                         goto done;
541                 val->port_vlan = nla_get_u32(info->attrs[SWITCH_ATTR_OP_PORT]);
542                 if (val->port_vlan >= dev->ports)
543                         goto done;
544                 break;
545         default:
546                 WARN_ON(1);
547                 goto done;
548         }
549
550         if (!alist)
551                 goto done;
552
553         attr_id = nla_get_u32(info->attrs[SWITCH_ATTR_OP_ID]);
554         if (attr_id >= SWITCH_ATTR_DEFAULTS_OFFSET) {
555                 attr_id -= SWITCH_ATTR_DEFAULTS_OFFSET;
556                 if (attr_id >= n_def)
557                         goto done;
558                 if (!test_bit(attr_id, def_active))
559                         goto done;
560                 attr = &def_list[attr_id];
561         } else {
562                 if (attr_id >= alist->n_attr)
563                         goto done;
564                 attr = &alist->attr[attr_id];
565         }
566
567         if (attr->disabled)
568                 attr = NULL;
569
570 done:
571         if (!attr)
572                 pr_debug("attribute lookup failed\n");
573         val->attr = attr;
574         return attr;
575 }
576
577 static int
578 swconfig_parse_ports(struct sk_buff *msg, struct nlattr *head,
579                 struct switch_val *val, int max)
580 {
581         struct nlattr *nla;
582         int rem;
583
584         val->len = 0;
585         nla_for_each_nested(nla, head, rem) {
586                 struct nlattr *tb[SWITCH_PORT_ATTR_MAX+1];
587                 struct switch_port *port;
588
589                 if (val->len >= max)
590                         return -EINVAL;
591
592                 port = &val->value.ports[val->len];
593
594 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
595                 if (nla_parse_nested_deprecated(tb, SWITCH_PORT_ATTR_MAX, nla,
596                                 port_policy, NULL))
597 #elif LINUX_VERSION_CODE >= KERNEL_VERSION(4,12,0)
598                 if (nla_parse_nested(tb, SWITCH_PORT_ATTR_MAX, nla,
599                                 port_policy, NULL))
600 #else
601                 if (nla_parse_nested(tb, SWITCH_PORT_ATTR_MAX, nla,
602                                 port_policy))
603 #endif
604                         return -EINVAL;
605
606                 if (!tb[SWITCH_PORT_ID])
607                         return -EINVAL;
608
609                 port->id = nla_get_u32(tb[SWITCH_PORT_ID]);
610                 if (tb[SWITCH_PORT_FLAG_TAGGED])
611                         port->flags |= (1 << SWITCH_PORT_FLAG_TAGGED);
612                 val->len++;
613         }
614
615         return 0;
616 }
617
618 static int
619 swconfig_parse_link(struct sk_buff *msg, struct nlattr *nla,
620                     struct switch_port_link *link)
621 {
622         struct nlattr *tb[SWITCH_LINK_ATTR_MAX + 1];
623
624 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
625         if (nla_parse_nested_deprecated(tb, SWITCH_LINK_ATTR_MAX, nla, link_policy, NULL))
626 #elif LINUX_VERSION_CODE >= KERNEL_VERSION(4,12,0)
627         if (nla_parse_nested(tb, SWITCH_LINK_ATTR_MAX, nla, link_policy, NULL))
628 #else
629         if (nla_parse_nested(tb, SWITCH_LINK_ATTR_MAX, nla, link_policy))
630 #endif
631                 return -EINVAL;
632
633         link->duplex = !!tb[SWITCH_LINK_FLAG_DUPLEX];
634         link->aneg = !!tb[SWITCH_LINK_FLAG_ANEG];
635         link->speed = nla_get_u32(tb[SWITCH_LINK_SPEED]);
636
637         return 0;
638 }
639
640 static int
641 swconfig_set_attr(struct sk_buff *skb, struct genl_info *info)
642 {
643         const struct switch_attr *attr;
644         struct switch_dev *dev;
645         struct switch_val val;
646         int err = -EINVAL;
647
648         if (!capable(CAP_NET_ADMIN))
649                 return -EPERM;
650
651         dev = swconfig_get_dev(info);
652         if (!dev)
653                 return -EINVAL;
654
655         memset(&val, 0, sizeof(val));
656         attr = swconfig_lookup_attr(dev, info, &val);
657         if (!attr || !attr->set)
658                 goto error;
659
660         val.attr = attr;
661         switch (attr->type) {
662         case SWITCH_TYPE_NOVAL:
663                 break;
664         case SWITCH_TYPE_INT:
665                 if (!info->attrs[SWITCH_ATTR_OP_VALUE_INT])
666                         goto error;
667                 val.value.i =
668                         nla_get_u32(info->attrs[SWITCH_ATTR_OP_VALUE_INT]);
669                 break;
670         case SWITCH_TYPE_STRING:
671                 if (!info->attrs[SWITCH_ATTR_OP_VALUE_STR])
672                         goto error;
673                 val.value.s =
674                         nla_data(info->attrs[SWITCH_ATTR_OP_VALUE_STR]);
675                 break;
676         case SWITCH_TYPE_PORTS:
677                 val.value.ports = dev->portbuf;
678                 memset(dev->portbuf, 0,
679                         sizeof(struct switch_port) * dev->ports);
680
681                 /* TODO: implement multipart? */
682                 if (info->attrs[SWITCH_ATTR_OP_VALUE_PORTS]) {
683                         err = swconfig_parse_ports(skb,
684                                 info->attrs[SWITCH_ATTR_OP_VALUE_PORTS],
685                                 &val, dev->ports);
686                         if (err < 0)
687                                 goto error;
688                 } else {
689                         val.len = 0;
690                         err = 0;
691                 }
692                 break;
693         case SWITCH_TYPE_LINK:
694                 val.value.link = &dev->linkbuf;
695                 memset(&dev->linkbuf, 0, sizeof(struct switch_port_link));
696
697                 if (info->attrs[SWITCH_ATTR_OP_VALUE_LINK]) {
698                         err = swconfig_parse_link(skb,
699                                                   info->attrs[SWITCH_ATTR_OP_VALUE_LINK],
700                                                   val.value.link);
701                         if (err < 0)
702                                 goto error;
703                 } else {
704                         val.len = 0;
705                         err = 0;
706                 }
707                 break;
708         default:
709                 goto error;
710         }
711
712         err = attr->set(dev, attr, &val);
713 error:
714         swconfig_put_dev(dev);
715         return err;
716 }
717
718 static int
719 swconfig_close_portlist(struct swconfig_callback *cb, void *arg)
720 {
721         if (cb->nest[0])
722                 nla_nest_end(cb->msg, cb->nest[0]);
723         return 0;
724 }
725
726 static int
727 swconfig_send_port(struct swconfig_callback *cb, void *arg)
728 {
729         const struct switch_port *port = arg;
730         struct nlattr *p = NULL;
731
732         if (!cb->nest[0]) {
733                 cb->nest[0] = nla_nest_start(cb->msg, cb->cmd);
734                 if (!cb->nest[0])
735                         return -1;
736         }
737
738         p = nla_nest_start(cb->msg, SWITCH_ATTR_PORT);
739         if (!p)
740                 goto error;
741
742         if (nla_put_u32(cb->msg, SWITCH_PORT_ID, port->id))
743                 goto nla_put_failure;
744         if (port->flags & (1 << SWITCH_PORT_FLAG_TAGGED)) {
745                 if (nla_put_flag(cb->msg, SWITCH_PORT_FLAG_TAGGED))
746                         goto nla_put_failure;
747         }
748
749         nla_nest_end(cb->msg, p);
750         return 0;
751
752 nla_put_failure:
753                 nla_nest_cancel(cb->msg, p);
754 error:
755         nla_nest_cancel(cb->msg, cb->nest[0]);
756         return -1;
757 }
758
759 static int
760 swconfig_send_ports(struct sk_buff **msg, struct genl_info *info, int attr,
761                 const struct switch_val *val)
762 {
763         struct swconfig_callback cb;
764         int err = 0;
765         int i;
766
767         if (!val->value.ports)
768                 return -EINVAL;
769
770         memset(&cb, 0, sizeof(cb));
771         cb.cmd = attr;
772         cb.msg = *msg;
773         cb.info = info;
774         cb.fill = swconfig_send_port;
775         cb.close = swconfig_close_portlist;
776
777         cb.nest[0] = nla_nest_start(cb.msg, cb.cmd);
778         for (i = 0; i < val->len; i++) {
779                 err = swconfig_send_multipart(&cb, &val->value.ports[i]);
780                 if (err)
781                         goto done;
782         }
783         err = val->len;
784         swconfig_close_portlist(&cb, NULL);
785         *msg = cb.msg;
786
787 done:
788         return err;
789 }
790
791 static int
792 swconfig_send_link(struct sk_buff *msg, struct genl_info *info, int attr,
793                    const struct switch_port_link *link)
794 {
795         struct nlattr *p = NULL;
796         int err = 0;
797
798         p = nla_nest_start(msg, attr);
799         if (link->link) {
800                 if (nla_put_flag(msg, SWITCH_LINK_FLAG_LINK))
801                         goto nla_put_failure;
802         }
803         if (link->duplex) {
804                 if (nla_put_flag(msg, SWITCH_LINK_FLAG_DUPLEX))
805                         goto nla_put_failure;
806         }
807         if (link->aneg) {
808                 if (nla_put_flag(msg, SWITCH_LINK_FLAG_ANEG))
809                         goto nla_put_failure;
810         }
811         if (link->tx_flow) {
812                 if (nla_put_flag(msg, SWITCH_LINK_FLAG_TX_FLOW))
813                         goto nla_put_failure;
814         }
815         if (link->rx_flow) {
816                 if (nla_put_flag(msg, SWITCH_LINK_FLAG_RX_FLOW))
817                         goto nla_put_failure;
818         }
819         if (nla_put_u32(msg, SWITCH_LINK_SPEED, link->speed))
820                 goto nla_put_failure;
821         if (link->eee & ADVERTISED_100baseT_Full) {
822                 if (nla_put_flag(msg, SWITCH_LINK_FLAG_EEE_100BASET))
823                         goto nla_put_failure;
824         }
825         if (link->eee & ADVERTISED_1000baseT_Full) {
826                 if (nla_put_flag(msg, SWITCH_LINK_FLAG_EEE_1000BASET))
827                         goto nla_put_failure;
828         }
829         nla_nest_end(msg, p);
830
831         return err;
832
833 nla_put_failure:
834         nla_nest_cancel(msg, p);
835         return -1;
836 }
837
838 static int
839 swconfig_get_attr(struct sk_buff *skb, struct genl_info *info)
840 {
841         struct genlmsghdr *hdr = nlmsg_data(info->nlhdr);
842         const struct switch_attr *attr;
843         struct switch_dev *dev;
844         struct sk_buff *msg = NULL;
845         struct switch_val val;
846         int err = -EINVAL;
847         int cmd = hdr->cmd;
848
849         dev = swconfig_get_dev(info);
850         if (!dev)
851                 return -EINVAL;
852
853         memset(&val, 0, sizeof(val));
854         attr = swconfig_lookup_attr(dev, info, &val);
855         if (!attr || !attr->get)
856                 goto error;
857
858         if (attr->type == SWITCH_TYPE_PORTS) {
859                 val.value.ports = dev->portbuf;
860                 memset(dev->portbuf, 0,
861                         sizeof(struct switch_port) * dev->ports);
862         } else if (attr->type == SWITCH_TYPE_LINK) {
863                 val.value.link = &dev->linkbuf;
864                 memset(&dev->linkbuf, 0, sizeof(struct switch_port_link));
865         }
866
867         err = attr->get(dev, attr, &val);
868         if (err)
869                 goto error;
870
871         msg = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
872         if (!msg)
873                 goto error;
874
875         hdr = genlmsg_put(msg, info->snd_portid, info->snd_seq, &switch_fam,
876                         0, cmd);
877         if (IS_ERR(hdr))
878                 goto nla_put_failure;
879
880         switch (attr->type) {
881         case SWITCH_TYPE_INT:
882                 if (nla_put_u32(msg, SWITCH_ATTR_OP_VALUE_INT, val.value.i))
883                         goto nla_put_failure;
884                 break;
885         case SWITCH_TYPE_STRING:
886                 if (nla_put_string(msg, SWITCH_ATTR_OP_VALUE_STR, val.value.s))
887                         goto nla_put_failure;
888                 break;
889         case SWITCH_TYPE_PORTS:
890                 err = swconfig_send_ports(&msg, info,
891                                 SWITCH_ATTR_OP_VALUE_PORTS, &val);
892                 if (err < 0)
893                         goto nla_put_failure;
894                 break;
895         case SWITCH_TYPE_LINK:
896                 err = swconfig_send_link(msg, info,
897                                          SWITCH_ATTR_OP_VALUE_LINK, val.value.link);
898                 if (err < 0)
899                         goto nla_put_failure;
900                 break;
901         default:
902                 pr_debug("invalid type in attribute\n");
903                 err = -EINVAL;
904                 goto nla_put_failure;
905         }
906         genlmsg_end(msg, hdr);
907         err = msg->len;
908         if (err < 0)
909                 goto nla_put_failure;
910
911         swconfig_put_dev(dev);
912         return genlmsg_reply(msg, info);
913
914 nla_put_failure:
915         if (msg)
916                 nlmsg_free(msg);
917 error:
918         swconfig_put_dev(dev);
919         if (!err)
920                 err = -ENOMEM;
921         return err;
922 }
923
924 static int
925 swconfig_send_switch(struct sk_buff *msg, u32 pid, u32 seq, int flags,
926                 const struct switch_dev *dev)
927 {
928         struct nlattr *p = NULL, *m = NULL;
929         void *hdr;
930         int i;
931
932         hdr = genlmsg_put(msg, pid, seq, &switch_fam, flags,
933                         SWITCH_CMD_NEW_ATTR);
934         if (IS_ERR(hdr))
935                 return -1;
936
937         if (nla_put_u32(msg, SWITCH_ATTR_ID, dev->id))
938                 goto nla_put_failure;
939         if (nla_put_string(msg, SWITCH_ATTR_DEV_NAME, dev->devname))
940                 goto nla_put_failure;
941         if (nla_put_string(msg, SWITCH_ATTR_ALIAS, dev->alias))
942                 goto nla_put_failure;
943         if (nla_put_string(msg, SWITCH_ATTR_NAME, dev->name))
944                 goto nla_put_failure;
945         if (nla_put_u32(msg, SWITCH_ATTR_VLANS, dev->vlans))
946                 goto nla_put_failure;
947         if (nla_put_u32(msg, SWITCH_ATTR_PORTS, dev->ports))
948                 goto nla_put_failure;
949         if (nla_put_u32(msg, SWITCH_ATTR_CPU_PORT, dev->cpu_port))
950                 goto nla_put_failure;
951
952         m = nla_nest_start(msg, SWITCH_ATTR_PORTMAP);
953         if (!m)
954                 goto nla_put_failure;
955         for (i = 0; i < dev->ports; i++) {
956                 p = nla_nest_start(msg, SWITCH_ATTR_PORTS);
957                 if (!p)
958                         continue;
959                 if (dev->portmap[i].s) {
960                         if (nla_put_string(msg, SWITCH_PORTMAP_SEGMENT,
961                                                 dev->portmap[i].s))
962                                 goto nla_put_failure;
963                         if (nla_put_u32(msg, SWITCH_PORTMAP_VIRT,
964                                                 dev->portmap[i].virt))
965                                 goto nla_put_failure;
966                 }
967                 nla_nest_end(msg, p);
968         }
969         nla_nest_end(msg, m);
970         genlmsg_end(msg, hdr);
971         return msg->len;
972 nla_put_failure:
973         genlmsg_cancel(msg, hdr);
974         return -EMSGSIZE;
975 }
976
977 static int swconfig_dump_switches(struct sk_buff *skb,
978                 struct netlink_callback *cb)
979 {
980         struct switch_dev *dev;
981         int start = cb->args[0];
982         int idx = 0;
983
984         swconfig_lock();
985         list_for_each_entry(dev, &swdevs, dev_list) {
986                 if (++idx <= start)
987                         continue;
988                 if (swconfig_send_switch(skb, NETLINK_CB(cb->skb).portid,
989                                 cb->nlh->nlmsg_seq, NLM_F_MULTI,
990                                 dev) < 0)
991                         break;
992         }
993         swconfig_unlock();
994         cb->args[0] = idx;
995
996         return skb->len;
997 }
998
999 static int
1000 swconfig_done(struct netlink_callback *cb)
1001 {
1002         return 0;
1003 }
1004
1005 static struct genl_ops swconfig_ops[] = {
1006         {
1007                 .cmd = SWITCH_CMD_LIST_GLOBAL,
1008 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1009                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1010 #endif
1011                 .doit = swconfig_list_attrs,
1012 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1013                 .policy = switch_policy,
1014 #endif
1015         },
1016         {
1017                 .cmd = SWITCH_CMD_LIST_VLAN,
1018 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1019                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1020 #endif
1021                 .doit = swconfig_list_attrs,
1022 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1023                 .policy = switch_policy,
1024 #endif
1025         },
1026         {
1027                 .cmd = SWITCH_CMD_LIST_PORT,
1028 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1029                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1030 #endif
1031                 .doit = swconfig_list_attrs,
1032 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1033                 .policy = switch_policy,
1034 #endif
1035         },
1036         {
1037                 .cmd = SWITCH_CMD_GET_GLOBAL,
1038 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1039                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1040 #endif
1041                 .doit = swconfig_get_attr,
1042 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1043                 .policy = switch_policy,
1044 #endif
1045         },
1046         {
1047                 .cmd = SWITCH_CMD_GET_VLAN,
1048 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1049                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1050 #endif
1051                 .doit = swconfig_get_attr,
1052 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1053                 .policy = switch_policy,
1054 #endif
1055         },
1056         {
1057                 .cmd = SWITCH_CMD_GET_PORT,
1058 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1059                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1060 #endif
1061                 .doit = swconfig_get_attr,
1062 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1063                 .policy = switch_policy,
1064 #endif
1065         },
1066         {
1067                 .cmd = SWITCH_CMD_SET_GLOBAL,
1068 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1069                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1070 #endif
1071                 .flags = GENL_ADMIN_PERM,
1072                 .doit = swconfig_set_attr,
1073 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1074                 .policy = switch_policy,
1075 #endif
1076         },
1077         {
1078                 .cmd = SWITCH_CMD_SET_VLAN,
1079 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1080                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1081 #endif
1082                 .flags = GENL_ADMIN_PERM,
1083                 .doit = swconfig_set_attr,
1084 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1085                 .policy = switch_policy,
1086 #endif
1087         },
1088         {
1089                 .cmd = SWITCH_CMD_SET_PORT,
1090 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1091                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1092 #endif
1093                 .flags = GENL_ADMIN_PERM,
1094                 .doit = swconfig_set_attr,
1095 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1096                 .policy = switch_policy,
1097 #endif
1098         },
1099         {
1100                 .cmd = SWITCH_CMD_GET_SWITCH,
1101 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5,2,0)
1102                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1103 #endif
1104                 .dumpit = swconfig_dump_switches,
1105 #if LINUX_VERSION_CODE < KERNEL_VERSION(5, 2, 0)
1106                 .policy = switch_policy,
1107 #endif
1108                 .done = swconfig_done,
1109         }
1110 };
1111
1112 static struct genl_family switch_fam = {
1113 #if LINUX_VERSION_CODE < KERNEL_VERSION(4,10,0)
1114         .id = GENL_ID_GENERATE,
1115 #endif
1116         .name = "switch",
1117         .hdrsize = 0,
1118         .version = 1,
1119         .maxattr = SWITCH_ATTR_MAX,
1120 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5, 2, 0)
1121         .policy = switch_policy,
1122 #endif
1123         .module = THIS_MODULE,
1124         .ops = swconfig_ops,
1125         .n_ops = ARRAY_SIZE(swconfig_ops),
1126 };
1127
1128 #ifdef CONFIG_OF
1129 void
1130 of_switch_load_portmap(struct switch_dev *dev)
1131 {
1132         struct device_node *port;
1133
1134         if (!dev->of_node)
1135                 return;
1136
1137         for_each_child_of_node(dev->of_node, port) {
1138                 const __be32 *prop;
1139                 const char *segment;
1140                 int size, phys;
1141
1142                 if (!of_device_is_compatible(port, "swconfig,port"))
1143                         continue;
1144
1145                 if (of_property_read_string(port, "swconfig,segment", &segment))
1146                         continue;
1147
1148                 prop = of_get_property(port, "swconfig,portmap", &size);
1149                 if (!prop)
1150                         continue;
1151
1152                 if (size != (2 * sizeof(*prop))) {
1153                         pr_err("%s: failed to parse port mapping\n",
1154                                         port->name);
1155                         continue;
1156                 }
1157
1158                 phys = be32_to_cpup(prop++);
1159                 if ((phys < 0) | (phys >= dev->ports)) {
1160                         pr_err("%s: physical port index out of range\n",
1161                                         port->name);
1162                         continue;
1163                 }
1164
1165                 dev->portmap[phys].s = kstrdup(segment, GFP_KERNEL);
1166                 dev->portmap[phys].virt = be32_to_cpup(prop);
1167                 pr_debug("Found port: %s, physical: %d, virtual: %d\n",
1168                         segment, phys, dev->portmap[phys].virt);
1169         }
1170 }
1171 #endif
1172
1173 int
1174 register_switch(struct switch_dev *dev, struct net_device *netdev)
1175 {
1176         struct switch_dev *sdev;
1177         const int max_switches = 8 * sizeof(unsigned long);
1178         unsigned long in_use = 0;
1179         int err;
1180         int i;
1181
1182         INIT_LIST_HEAD(&dev->dev_list);
1183         if (netdev) {
1184                 dev->netdev = netdev;
1185                 if (!dev->alias)
1186                         dev->alias = netdev->name;
1187         }
1188         BUG_ON(!dev->alias);
1189
1190         /* Make sure swdev_id doesn't overflow */
1191         if (swdev_id == INT_MAX) {
1192                 return -ENOMEM;
1193         }
1194
1195         if (dev->ports > 0) {
1196                 dev->portbuf = kzalloc(sizeof(struct switch_port) *
1197                                 dev->ports, GFP_KERNEL);
1198                 if (!dev->portbuf)
1199                         return -ENOMEM;
1200                 dev->portmap = kzalloc(sizeof(struct switch_portmap) *
1201                                 dev->ports, GFP_KERNEL);
1202                 if (!dev->portmap) {
1203                         kfree(dev->portbuf);
1204                         return -ENOMEM;
1205                 }
1206         }
1207         swconfig_defaults_init(dev);
1208         mutex_init(&dev->sw_mutex);
1209         swconfig_lock();
1210         dev->id = ++swdev_id;
1211
1212         list_for_each_entry(sdev, &swdevs, dev_list) {
1213                 if (!sscanf(sdev->devname, SWCONFIG_DEVNAME, &i))
1214                         continue;
1215                 if (i < 0 || i > max_switches)
1216                         continue;
1217
1218                 set_bit(i, &in_use);
1219         }
1220         i = find_first_zero_bit(&in_use, max_switches);
1221
1222         if (i == max_switches) {
1223                 swconfig_unlock();
1224                 return -ENFILE;
1225         }
1226
1227 #ifdef CONFIG_OF
1228         if (dev->ports)
1229                 of_switch_load_portmap(dev);
1230 #endif
1231
1232         /* fill device name */
1233         snprintf(dev->devname, IFNAMSIZ, SWCONFIG_DEVNAME, i);
1234
1235         list_add_tail(&dev->dev_list, &swdevs);
1236         swconfig_unlock();
1237
1238         err = swconfig_create_led_trigger(dev);
1239         if (err)
1240                 return err;
1241
1242         return 0;
1243 }
1244 EXPORT_SYMBOL_GPL(register_switch);
1245
1246 void
1247 unregister_switch(struct switch_dev *dev)
1248 {
1249         swconfig_destroy_led_trigger(dev);
1250         kfree(dev->portbuf);
1251         mutex_lock(&dev->sw_mutex);
1252         swconfig_lock();
1253         list_del(&dev->dev_list);
1254         swconfig_unlock();
1255         mutex_unlock(&dev->sw_mutex);
1256 }
1257 EXPORT_SYMBOL_GPL(unregister_switch);
1258
1259 int
1260 switch_generic_set_link(struct switch_dev *dev, int port,
1261                         struct switch_port_link *link)
1262 {
1263         if (WARN_ON(!dev->ops->phy_write16))
1264                 return -ENOTSUPP;
1265
1266         /* Generic implementation */
1267         if (link->aneg) {
1268                 dev->ops->phy_write16(dev, port, MII_BMCR, 0x0000);
1269                 dev->ops->phy_write16(dev, port, MII_BMCR, BMCR_ANENABLE | BMCR_ANRESTART);
1270         } else {
1271                 u16 bmcr = 0;
1272
1273                 if (link->duplex)
1274                         bmcr |= BMCR_FULLDPLX;
1275
1276                 switch (link->speed) {
1277                 case SWITCH_PORT_SPEED_10:
1278                         break;
1279                 case SWITCH_PORT_SPEED_100:
1280                         bmcr |= BMCR_SPEED100;
1281                         break;
1282                 case SWITCH_PORT_SPEED_1000:
1283                         bmcr |= BMCR_SPEED1000;
1284                         break;
1285                 default:
1286                         return -ENOTSUPP;
1287                 }
1288
1289                 dev->ops->phy_write16(dev, port, MII_BMCR, bmcr);
1290         }
1291
1292         return 0;
1293 }
1294 EXPORT_SYMBOL_GPL(switch_generic_set_link);
1295
1296 static int __init
1297 swconfig_init(void)
1298 {
1299         INIT_LIST_HEAD(&swdevs);
1300
1301 #if LINUX_VERSION_CODE < KERNEL_VERSION(4,10,0)
1302         return genl_register_family_with_ops(&switch_fam, swconfig_ops);
1303 #else
1304         return genl_register_family(&switch_fam);
1305 #endif
1306 }
1307
1308 static void __exit
1309 swconfig_exit(void)
1310 {
1311         genl_unregister_family(&switch_fam);
1312 }
1313
1314 module_init(swconfig_init);
1315 module_exit(swconfig_exit);