"aborting.\n");
                        return -ENODEV;
                }
-       } else if (mdev->hca_type == TAVOR)
+       } else if (!(mdev->mthca_flags & MTHCA_FLAG_PCIE))
                mthca_info(mdev, "No PCI-X capability, not setting RBC.\n");
 
        cap = pci_find_capability(mdev->pdev, PCI_CAP_ID_EXP);
                                  "register, aborting.\n");
                        return -ENODEV;
                }
-       } else if (mdev->hca_type == ARBEL_NATIVE ||
-                  mdev->hca_type == ARBEL_COMPAT)
+       } else if (mdev->mthca_flags & MTHCA_FLAG_PCIE)
                mthca_info(mdev, "No PCI Express capability, "
                           "not setting Max Read Request Size.\n");
 
        if (!mdev->qp_table.rdb_table) {
                mthca_err(mdev, "Failed to map RDB context memory, aborting\n");
                err = -ENOMEM;
-               goto err_unmap_eqp;
+               goto err_unmap_rdb;
        }
 
        mdev->cq_table.table = mthca_alloc_icm_table(mdev, init_hca->cqc_base,
 
 err_free_icm:
        mthca_free_icm_table(mdev, mdev->cq_table.table);
+       mthca_free_icm_table(mdev, mdev->qp_table.rdb_table);
        mthca_free_icm_table(mdev, mdev->qp_table.eqp_table);
        mthca_free_icm_table(mdev, mdev->qp_table.qp_table);
        mthca_free_icm_table(mdev, mdev->mr_table.mpt_table);
 
        if (mthca_is_memfree(mdev)) {
                mthca_free_icm_table(mdev, mdev->cq_table.table);
+               mthca_free_icm_table(mdev, mdev->qp_table.rdb_table);
                mthca_free_icm_table(mdev, mdev->qp_table.eqp_table);
                mthca_free_icm_table(mdev, mdev->qp_table.qp_table);
                mthca_free_icm_table(mdev, mdev->mr_table.mpt_table);
                mthca_SYS_DIS(mdev, &status);
 }
 
+/* Types of supported HCA */
+enum {
+       TAVOR,                  /* MT23108                        */
+       ARBEL_COMPAT,           /* MT25208 in Tavor compat mode   */
+       ARBEL_NATIVE,           /* MT25208 with extended features */
+       SINAI                   /* MT25204 */
+};
+
+#define MTHCA_FW_VER(major, minor, subminor) \
+       (((u64) (major) << 32) | ((u64) (minor) << 16) | (u64) (subminor))
+
+static struct {
+       u64 latest_fw;
+       int is_memfree;
+       int is_pcie;
+} mthca_hca_table[] = {
+       [TAVOR]        = { .latest_fw = MTHCA_FW_VER(3, 3, 2), .is_memfree = 0, .is_pcie = 0 },
+       [ARBEL_COMPAT] = { .latest_fw = MTHCA_FW_VER(4, 6, 2), .is_memfree = 0, .is_pcie = 1 },
+       [ARBEL_NATIVE] = { .latest_fw = MTHCA_FW_VER(5, 0, 1), .is_memfree = 1, .is_pcie = 1 },
+       [SINAI]        = { .latest_fw = MTHCA_FW_VER(1, 0, 1), .is_memfree = 1, .is_pcie = 1 }
+};
+
 static int __devinit mthca_init_one(struct pci_dev *pdev,
                                    const struct pci_device_id *id)
 {
        static int mthca_version_printed = 0;
-       static int mthca_memfree_warned = 0;
        int ddr_hidden = 0;
        int err;
        struct mthca_dev *mdev;
        printk(KERN_INFO PFX "Initializing %s (%s)\n",
               pci_pretty_name(pdev), pci_name(pdev));
 
+       if (id->driver_data >= ARRAY_SIZE(mthca_hca_table)) {
+               printk(KERN_ERR PFX "%s (%s) has invalid driver data %lx\n",
+                      pci_pretty_name(pdev), pci_name(pdev), id->driver_data);
+               return -ENODEV;
+       }
+
        err = pci_enable_device(pdev);
        if (err) {
                dev_err(&pdev->dev, "Cannot enable PCI device, "
                goto err_free_res;
        }
 
-       mdev->pdev     = pdev;
-       mdev->hca_type = id->driver_data;
-
-       if (mthca_is_memfree(mdev) && !mthca_memfree_warned++)
-               mthca_warn(mdev, "Warning: native MT25208 mode support is incomplete.  "
-                          "Your HCA may not work properly.\n");
+       mdev->pdev = pdev;
 
        if (ddr_hidden)
                mdev->mthca_flags |= MTHCA_FLAG_DDR_HIDDEN;
+       if (mthca_hca_table[id->driver_data].is_memfree)
+               mdev->mthca_flags |= MTHCA_FLAG_MEMFREE;
+       if (mthca_hca_table[id->driver_data].is_pcie)
+               mdev->mthca_flags |= MTHCA_FLAG_PCIE;
 
        /*
         * Now reset the HCA before we touch the PCI capabilities or
        if (err)
                goto err_iounmap;
 
+       if (mdev->fw_ver < mthca_hca_table[id->driver_data].latest_fw) {
+               mthca_warn(mdev, "HCA FW version %x.%x.%x is old (%x.%x.%x is current).\n",
+                          (int) (mdev->fw_ver >> 32), (int) (mdev->fw_ver >> 16) & 0xffff,
+                          (int) (mdev->fw_ver & 0xffff),
+                          (int) (mthca_hca_table[id->driver_data].latest_fw >> 32),
+                          (int) (mthca_hca_table[id->driver_data].latest_fw >> 16) & 0xffff,
+                          (int) (mthca_hca_table[id->driver_data].latest_fw & 0xffff));
+               mthca_warn(mdev, "If you have problems, try updating your HCA FW.\n");
+       }
+
        err = mthca_setup_hca(mdev);
        if (err)
                goto err_close;
          .driver_data = ARBEL_NATIVE },
        { PCI_DEVICE(PCI_VENDOR_ID_TOPSPIN, PCI_DEVICE_ID_MELLANOX_ARBEL),
          .driver_data = ARBEL_NATIVE },
+       { PCI_DEVICE(PCI_VENDOR_ID_MELLANOX, PCI_DEVICE_ID_MELLANOX_SINAI),
+         .driver_data = SINAI },
+       { PCI_DEVICE(PCI_VENDOR_ID_TOPSPIN, PCI_DEVICE_ID_MELLANOX_SINAI),
+         .driver_data = SINAI },
+       { PCI_DEVICE(PCI_VENDOR_ID_MELLANOX, PCI_DEVICE_ID_MELLANOX_SINAI_OLD),
+         .driver_data = SINAI },
+       { PCI_DEVICE(PCI_VENDOR_ID_TOPSPIN, PCI_DEVICE_ID_MELLANOX_SINAI_OLD),
+         .driver_data = SINAI },
        { 0, }
 };
 
 
 static ssize_t show_hca(struct class_device *cdev, char *buf)
 {
        struct mthca_dev *dev = container_of(cdev, struct mthca_dev, ib_dev.class_dev);
-       switch (dev->hca_type) {
-       case TAVOR:        return sprintf(buf, "MT23108\n");
-       case ARBEL_COMPAT: return sprintf(buf, "MT25208 (MT23108 compat mode)\n");
-       case ARBEL_NATIVE: return sprintf(buf, "MT25208\n");
-       default:           return sprintf(buf, "unknown\n");
+       switch (dev->pdev->device) {
+       case PCI_DEVICE_ID_MELLANOX_TAVOR:
+               return sprintf(buf, "MT23108\n");
+       case PCI_DEVICE_ID_MELLANOX_ARBEL_COMPAT:
+               return sprintf(buf, "MT25208 (MT23108 compat mode)\n");
+       case PCI_DEVICE_ID_MELLANOX_ARBEL:
+               return sprintf(buf, "MT25208\n");
+       case PCI_DEVICE_ID_MELLANOX_SINAI:
+       case PCI_DEVICE_ID_MELLANOX_SINAI_OLD:
+               return sprintf(buf, "MT25204\n");
+       default:
+               return sprintf(buf, "unknown\n");
        }
 }