* @defined:           Bitmap of endpoints defined in config data
  * @available:         Bitmap of endpoints supported by hardware
  * @filtered:          Bitmap of endpoints that support filtering
- * @set_up:            Bit mask indicating endpoints set up
+ * @set_up:            Bitmap of endpoints that are set up for use
  * @enabled:           Bit mask indicating endpoints enabled
  * @modem_tx_count:    Number of defined modem TX endoints
  * @endpoint:          Array of endpoint information
        unsigned long *defined;         /* Defined in configuration data */
        unsigned long *available;       /* Supported by hardware */
        u64 filtered;                   /* Support filtering (AP and modem) */
-       u32 set_up;
+       unsigned long *set_up;
        u32 enabled;
 
        u32 modem_tx_count;
 
 
        ipa_endpoint_program(endpoint);
 
-       endpoint->ipa->set_up |= BIT(endpoint->endpoint_id);
+       __set_bit(endpoint->endpoint_id, endpoint->ipa->set_up);
 }
 
 static void ipa_endpoint_teardown_one(struct ipa_endpoint *endpoint)
 {
-       endpoint->ipa->set_up &= ~BIT(endpoint->endpoint_id);
+       __clear_bit(endpoint->endpoint_id, endpoint->ipa->set_up);
 
        if (!endpoint->toward_ipa)
                cancel_delayed_work_sync(&endpoint->replenish_work);
 {
        u32 endpoint_id;
 
-       ipa->set_up = 0;
        for_each_set_bit(endpoint_id, ipa->defined, ipa->endpoint_count)
                ipa_endpoint_setup_one(&ipa->endpoint[endpoint_id]);
 }
 
 void ipa_endpoint_teardown(struct ipa *ipa)
 {
-       u32 set_up = ipa->set_up;
-
-       while (set_up) {
-               u32 endpoint_id = __fls(set_up);
-
-               set_up ^= BIT(endpoint_id);
+       u32 endpoint_id;
 
+       for_each_set_bit(endpoint_id, ipa->set_up, ipa->endpoint_count)
                ipa_endpoint_teardown_one(&ipa->endpoint[endpoint_id]);
-       }
-       ipa->set_up = 0;
 }
 
 void ipa_endpoint_deconfig(struct ipa *ipa)
        for_each_set_bit(endpoint_id, ipa->defined, ipa->endpoint_count)
                ipa_endpoint_exit_one(&ipa->endpoint[endpoint_id]);
 
+       bitmap_free(ipa->set_up);
+       ipa->set_up = NULL;
        bitmap_free(ipa->defined);
        ipa->defined = NULL;
 
        if (!ipa->endpoint_count)
                return -EINVAL;
 
-       /* Initialize the defined endpoint bitmap */
+       /* Initialize endpoint state bitmaps */
        ipa->defined = bitmap_zalloc(ipa->endpoint_count, GFP_KERNEL);
        if (!ipa->defined)
                return -ENOMEM;
 
+       ipa->set_up = bitmap_zalloc(ipa->endpoint_count, GFP_KERNEL);
+       if (!ipa->set_up)
+               goto err_free_defined;
+
        filtered = 0;
        for (name = 0; name < count; name++, data++) {
                if (ipa_gsi_endpoint_data_empty(data))
                        ipa->modem_tx_count++;
        }
 
-       if (!ipa_filtered_valid(ipa, filtered))
-               goto err_endpoint_exit;
+       /* Make sure the set of filtered endpoints is valid */
+       if (!ipa_filtered_valid(ipa, filtered)) {
+               ipa_endpoint_exit(ipa);
+
+               return -EINVAL;
+       }
 
        ipa->filtered = filtered;
 
        return 0;
 
-err_endpoint_exit:
-       ipa_endpoint_exit(ipa);
+err_free_defined:
+       bitmap_free(ipa->defined);
+       ipa->defined = NULL;
 
-       return -EINVAL;
+       return -ENOMEM;
 }