ar71xx: add support for TP-Link TL-WDR6500 v2

[openwrt.git] / target / linux / ar71xx / files / arch / mips / ath79 / routerboot.c

diff --git a/target/linux/ar71xx/files/arch/mips/ath79/routerboot.c b/target/linux/ar71xx/files/arch/mips/ath79/routerboot.c
index dbdec02..76776e1 100644 (file)
--- a/target/linux/ar71xx/files/arch/mips/ath79/routerboot.c
+++ b/target/linux/ar71xx/files/arch/mips/ath79/routerboot.c
@@ -11,15 +11,18 @@
 #define pr_fmt(fmt) "rb: " fmt
 
 #include <linux/kernel.h>
+#include <linux/kobject.h>
 #include <linux/slab.h>
 #include <linux/errno.h>
 #include <linux/routerboot.h>
 #include <linux/rle.h>
+#include <linux/lzo.h>
 
 #include "routerboot.h"
 
 #define RB_BLOCK_SIZE          0x1000
 #define RB_ART_SIZE            0x10000
+#define RB_MAGIC_ERD           0x00455244      /* extended radio data */
 
 static struct rb_info rb_info;
 
@@ -63,6 +66,7 @@ routerboot_find_tag(u8 *buf, unsigned int buflen, u16 tag_id,
                    u8 **tag_data, u16 *tag_len)
 {
        uint32_t magic;
+       bool align = false;
        int ret;
 
        if (buflen < 4)
@@ -70,6 +74,9 @@ routerboot_find_tag(u8 *buf, unsigned int buflen, u16 tag_id,
 
        magic = get_u32(buf);
        switch (magic) {
+       case RB_MAGIC_ERD:
+               align = true;
+               /* fall trough */
        case RB_MAGIC_HARD:
                /* skip magic value */
                buf += 4;
@@ -121,6 +128,9 @@ routerboot_find_tag(u8 *buf, unsigned int buflen, u16 tag_id,
                        break;
                }
 
+               if (align)
+                       len = (len + 3) / 4;
+
                buf += len;
                buflen -= len;
        }
@@ -168,13 +178,16 @@ rb_get_hw_options(void)
        return get_u32(tag);
 }
 
-__init void *
-rb_get_wlan_data(void)
+static void * __init
+__rb_get_wlan_data(u16 id)
 {
        u16 tag_len;
        u8 *tag;
        void *buf;
        int err;
+       u32 magic;
+       size_t src_done;
+       size_t dst_done;
 
        err = rb_find_hard_cfg_tag(RB_ID_WLAN_DATA, &tag, &tag_len);
        if (err) {
@@ -188,11 +201,38 @@ rb_get_wlan_data(void)
                goto err;
        }
 
-       err = rle_decode((char *) tag, tag_len, buf, RB_ART_SIZE,
-                        NULL, NULL);
-       if (err) {
-               pr_err("unable to decode calibration data\n");
-               goto err_free;
+       magic = get_u32(tag);
+       if (magic == RB_MAGIC_ERD) {
+               u8 *erd_data;
+               u16 erd_len;
+
+               if (id == 0)
+                       goto err_free;
+
+               err = routerboot_find_tag(tag, tag_len, id,
+                                         &erd_data, &erd_len);
+               if (err) {
+                       pr_err("no ERD data found for id %u\n", id);
+                       goto err_free;
+               }
+
+               dst_done = RB_ART_SIZE;
+               err = lzo1x_decompress_safe(erd_data, erd_len, buf, &dst_done);
+               if (err) {
+                       pr_err("unable to decompress calibration data %d\n",
+                              err);
+                       goto err_free;
+               }
+       } else {
+               if (id != 0)
+                       goto err_free;
+
+               err = rle_decode((char *) tag, tag_len, buf, RB_ART_SIZE,
+                                &src_done, &dst_done);
+               if (err) {
+                       pr_err("unable to decode calibration data\n");
+                       goto err_free;
+               }
        }
 
        return buf;
@@ -203,6 +243,18 @@ err:
        return NULL;
 }
 
+__init void *
+rb_get_wlan_data(void)
+{
+       return __rb_get_wlan_data(0);
+}
+
+__init void *
+rb_get_ext_wlan_data(u16 id)
+{
+       return __rb_get_wlan_data(id);
+}
+
 __init const struct rb_info *
 rb_init_info(void *data, unsigned int size)
 {
@@ -247,3 +299,60 @@ rb_init_info(void *data, unsigned int size)
 
        return &rb_info;
 }
+
+static char *rb_ext_wlan_data;
+
+static ssize_t
+rb_ext_wlan_data_read(struct file *filp, struct kobject *kobj,
+                     struct bin_attribute *attr, char *buf,
+                     loff_t off, size_t count)
+{
+         if (off + count > attr->size)
+                 return -EFBIG;
+
+         memcpy(buf, &rb_ext_wlan_data[off], count);
+
+         return count;
+}
+
+static const struct bin_attribute rb_ext_wlan_data_attr = {
+       .attr = {
+               .name = "ext_wlan_data",
+               .mode = S_IRUSR | S_IWUSR,
+       },
+       .read = rb_ext_wlan_data_read,
+       .size = RB_ART_SIZE,
+};
+
+static int __init rb_sysfs_init(void)
+{
+       struct kobject *rb_kobj;
+       int ret;
+
+       rb_ext_wlan_data = rb_get_ext_wlan_data(1);
+       if (rb_ext_wlan_data == NULL)
+               return -ENOENT;
+
+       rb_kobj = kobject_create_and_add("routerboot", firmware_kobj);
+       if (rb_kobj == NULL) {
+               ret = -ENOMEM;
+               pr_err("unable to create sysfs entry\n");
+               goto err_free_wlan_data;
+       }
+
+       ret = sysfs_create_bin_file(rb_kobj, &rb_ext_wlan_data_attr);
+       if (ret) {
+               pr_err("unable to create sysfs file, %d\n", ret);
+               goto err_put_kobj;
+       }
+
+       return 0;
+
+err_put_kobj:
+       kobject_put(rb_kobj);
+err_free_wlan_data:
+       kfree(rb_ext_wlan_data);
+       return ret;
+}
+
+late_initcall(rb_sysfs_init);