2 files changed, 60 insertions, 1 deletions

diff --git a/scripts/mod/file2alias.c b/scripts/mod/file2alias.c
index e26e2fb462d4..f936d1fa969d 100644
--- a/scripts/mod/file2alias.c
+++ b/scripts/mod/file2alias.c
@@ -735,6 +735,27 @@ static int do_virtio_entry(const char *filename, struct virtio_device_id *id,
         return 1;
 }
 
+/*
+ * Looks like: vmbus:guid
+ * Each byte of the guid will be represented by two hex characters
+ * in the name.
+ */
+
+static int do_vmbus_entry(const char *filename, struct hv_vmbus_device_id *id,
+                          char *alias)
+{
+        int i;
+        char guid_name[((sizeof(id->guid) + 1)) * 2];
+
+        for (i = 0; i < (sizeof(id->guid) * 2); i += 2)
+                sprintf(&guid_name[i], "%02x", id->guid[i/2]);
+
+        strcpy(alias, "vmbus:");
+        strcat(alias, guid_name);
+
+        return 1;
+}
+
 /* Looks like: i2c:S */
 static int do_i2c_entry(const char *filename, struct i2c_device_id *id,
                         char *alias)
@@ -994,6 +1015,10 @@ void handle_moddevtable(struct module *mod, struct elf_info *info,
                 do_table(symval, sym->st_size,
                          sizeof(struct virtio_device_id), "virtio",
                          do_virtio_entry, mod);
+        else if (sym_is(symname, "__mod_vmbus_device_table"))
+                do_table(symval, sym->st_size,
+                         sizeof(struct hv_vmbus_device_id), "vmbus",
+                         do_vmbus_entry, mod);
         else if (sym_is(symname, "__mod_i2c_device_table"))
                 do_table(symval, sym->st_size,
                          sizeof(struct i2c_device_id), "i2c",
diff --git a/scripts/mod/modpost.c b/scripts/mod/modpost.c
index 413c53693e62..2bd594e6d1b4 100644
--- a/scripts/mod/modpost.c
+++ b/scripts/mod/modpost.c
@@ -254,6 +254,28 @@ static enum export export_no(const char *s)
         return export_unknown;
 }
 
+static const char *sec_name(struct elf_info *elf, int secindex);
+
+#define strstarts(str, prefix) (strncmp(str, prefix, strlen(prefix)) == 0)
+
+static enum export export_from_secname(struct elf_info *elf, unsigned int sec)
+{
+        const char *secname = sec_name(elf, sec);
+
+        if (strstarts(secname, "___ksymtab+"))
+                return export_plain;
+        else if (strstarts(secname, "___ksymtab_unused+"))
+                return export_unused;
+        else if (strstarts(secname, "___ksymtab_gpl+"))
+                return export_gpl;
+        else if (strstarts(secname, "___ksymtab_unused_gpl+"))
+                return export_unused_gpl;
+        else if (strstarts(secname, "___ksymtab_gpl_future+"))
+                return export_gpl_future;
+        else
+                return export_unknown;
+}
+
 static enum export export_from_sec(struct elf_info *elf, unsigned int sec)
 {
         if (sec == elf->export_sec)
@@ -563,7 +585,12 @@ static void handle_modversions(struct module *mod, struct elf_info *info,
                                Elf_Sym *sym, const char *symname)
 {
         unsigned int crc;
-        enum export export = export_from_sec(info, get_secindex(info, sym));
+        enum export export;
+
+        if (!is_vmlinux(mod->name) && strncmp(symname, "__ksymtab", 9) == 0)
+                export = export_from_secname(info, get_secindex(info, sym));
+        else
+                export = export_from_sec(info, get_secindex(info, sym));
 
         switch (sym->st_shndx) {
         case SHN_COMMON:
@@ -1822,6 +1849,12 @@ static void add_header(struct buffer *b, struct module *mod)
         buf_printf(b, "};\n");
 }
 
+static void add_intree_flag(struct buffer *b, int is_intree)
+{
+        if (is_intree)
+                buf_printf(b, "\nMODULE_INFO(intree, \"Y\");\n");
+}
+
 static void add_staging_flag(struct buffer *b, const char *name)
 {
         static const char *staging_dir = "drivers/staging";
@@ -2142,6 +2175,7 @@ int main(int argc, char **argv)
                 buf.pos = 0;
 
                 add_header(&buf, mod);
+                add_intree_flag(&buf, !external_module);
                 add_staging_flag(&buf, mod->name);
                 err |= add_versions(&buf, mod);
                 add_depends(&buf, mod, modules);