firmware-utils: tplink-safeloader: add struct device_info

This struct stores all device specific info which allows making do_eap
function more generic.

Signed-off-by: Rafał Miłecki <rafal@milecki.pl>

diff --git a/tools/firmware-utils/src/tplink-safeloader.c b/tools/firmware-utils/src/tplink-safeloader.c
index fdf93d75d5052bbaa3eb9016b0e0f6a2ad4b7921..a1b8c31c70b62256382c72c1e88e6c2e10c25589 100644 (file)
--- a/tools/firmware-utils/src/tplink-safeloader.c
+++ b/tools/firmware-utils/src/tplink-safeloader.c
@@ -67,6 +67,12 @@ struct flash_partition_entry {
        uint32_t size;
 };
 
+struct device_info {
+       const char *vendor;
+       const char *support_list;
+       const struct flash_partition_entry *partitions;
+       void *(*generate_sysupgrade_image)(const struct flash_partition_entry *flash_parts, const struct image_partition_entry *image_parts, size_t *len);
+};
 
 /** The content of the soft-version structure */
 struct __attribute__((__packed__)) soft_version {
@@ -575,6 +581,13 @@ static void *generate_sysupgrade_image_eap120(const struct flash_partition_entry
        return image;
 }
 
+struct device_info eap120_info = {
+       .vendor = eap120_vendor,
+       .support_list = eap120_support_list,
+       .partitions = eap120_partitions,
+       .generate_sysupgrade_image = &generate_sysupgrade_image_eap120,
+};
+
 /** Generates an image for CPE210/220/510/520 and writes it to a file */
 static void do_cpe(const char *output,
                const char *kernel_image,
@@ -655,21 +668,21 @@ static void do_eap(const char *output,
                uint32_t rev,
                bool add_jffs2_eof,
                bool sysupgrade,
-               const char *support_list) {
+               struct device_info *info) {
        struct image_partition_entry parts[6] = {};
 
-       parts[0] = make_partition_table(eap120_partitions);
+       parts[0] = make_partition_table(info->partitions);
        parts[1] = make_soft_version(rev);
-       parts[2] = make_support_list(support_list, false);
+       parts[2] = make_support_list(info->support_list, false);
        parts[3] = read_file("os-image", kernel_image, false);
        parts[4] = read_file("file-system", rootfs_image, add_jffs2_eof);
 
        size_t len;
        void *image;
        if (sysupgrade)
-               image = generate_sysupgrade_image_eap120(eap120_partitions, parts, &len);
+               image = info->generate_sysupgrade_image(info->partitions, parts, &len);
        else
-               image = generate_factory_image(eap120_vendor, parts, &len);
+               image = generate_factory_image(info->vendor, parts, &len);
 
        FILE *file = fopen(output, "wb");
        if (!file)
@@ -773,7 +786,7 @@ int main(int argc, char *argv[]) {
        else if (strcmp(board, "C2600") == 0)
                do_c2600(output, kernel_image, rootfs_image, rev, add_jffs2_eof, sysupgrade);
        else if (strcmp(board, "EAP120") == 0)
-               do_eap(output, kernel_image, rootfs_image, rev, add_jffs2_eof, sysupgrade, eap120_support_list);
+               do_eap(output, kernel_image, rootfs_image, rev, add_jffs2_eof, sysupgrade, &eap120_info);
        else
                error(1, 0, "unsupported board %s", board);