1 files changed, 22 insertions, 3 deletions

diff --git a/arch/sparc64/kernel/irq.c b/arch/sparc64/kernel/irq.c
index 51b8875a13a8..23956096b3bf 100644
--- a/arch/sparc64/kernel/irq.c
+++ b/arch/sparc64/kernel/irq.c
@@ -217,8 +217,27 @@ struct irq_handler_data {
         void            (*pre_handler)(unsigned int, void *, void *);
         void            *pre_handler_arg1;
         void            *pre_handler_arg2;
+
+        u32             msi;
 };
 
+void sparc64_set_msi(unsigned int virt_irq, u32 msi)
+{
+        struct irq_handler_data *data = get_irq_chip_data(virt_irq);
+
+        if (data)
+                data->msi = msi;
+}
+
+u32 sparc64_get_msi(unsigned int virt_irq)
+{
+        struct irq_handler_data *data = get_irq_chip_data(virt_irq);
+
+        if (data)
+                return data->msi;
+        return 0xffffffff;
+}
+
 static inline struct ino_bucket *virt_irq_to_bucket(unsigned int virt_irq)
 {
         unsigned int real_irq = virt_to_real_irq(virt_irq);
@@ -741,7 +760,7 @@ unsigned int sun4v_build_msi(u32 devhandle, unsigned int *virt_irq_p,
                         break;
         }
         if (devino >= msi_end)
-                return 0;
+                return -ENOSPC;
 
         sysino = sun4v_devino_to_sysino(devhandle, devino);
         bucket = &ivector_table[sysino];
@@ -755,8 +774,8 @@ unsigned int sun4v_build_msi(u32 devhandle, unsigned int *virt_irq_p,
 
         data = kzalloc(sizeof(struct irq_handler_data), GFP_ATOMIC);
         if (unlikely(!data)) {
-                prom_printf("IRQ: kzalloc(irq_handler_data) failed.\n");
-                prom_halt();
+                virt_irq_free(*virt_irq_p);
+                return -ENOMEM;
         }
         set_irq_chip_data(bucket->virt_irq, data);