Changeset 6439741 in mainline for uspace/srv/bd/vbd/disk.c

Timestamp:

2015-07-30T16:20:22Z (9 years ago)

Author:

Jiri Svoboda <jiri@…>

Branches:

lfn, master, serial, ticket/834-toolchain-update, topic/msim-upgrade, topic/simplify-dev-export

Children:

22747e8

Parents:

100b1d1

Message:

VBD must re-register logical partitions whose index has changed under new service names.

File:

• uspace/srv/bd/vbd/disk.c (modified) (8 diffs)

uspace/srv/bd/vbd/disk.c

@@ -62 +62 @@
 static int vbds_bsa_translate(vbds_part_t *, aoff64_t, size_t, aoff64_t *);
 
+static int vbds_part_svc_register(vbds_part_t *);
+static int vbds_part_svc_unregister(vbds_part_t *);
+static int vbds_part_indices_update(vbds_disk_t *);
+
 static vbd_part_id_t vbds_part_id = 1;
 
@@ -138 +142 @@
         service_id_t psid = 0;
         label_part_info_t lpinfo;
-        char *name;
         int rc;
 
@@ -150 +153 @@
                 log_msg(LOG_DEFAULT, LVL_ERROR, "Out of memory.");
                 return ENOMEM;
-        }
-
-        if (lpinfo.pkind != lpk_extended) {
-                /* XXX Proper service name */
-                rc = asprintf(&name, "%sp%u", disk->svc_name, lpart->index);
-                if (rc < 0) {
-                        log_msg(LOG_DEFAULT, LVL_ERROR, "Out of memory.");
-                        return ENOMEM;
-                }
-
-                rc = loc_service_register(name, &psid);
-                if (rc != EOK) {
-                        log_msg(LOG_DEFAULT, LVL_ERROR, "Failed registering "
-                            "service %s.", name);
-                        free(name);
-                        free(part);
-                        return EIO;
-                }
-
-                free(name);
         }
 
@@ -183 +166 @@
         part->bds.sarg = part;
 
+        if (lpinfo.pkind != lpk_extended) {
+                rc = vbds_part_svc_register(part);
+                if (rc != EOK)
+                        return EIO;
+        }
+
         list_append(&part->ldisk, &disk->parts);
         list_append(&part->lparts, &vbds_parts);
@@ -207 +196 @@
 
         if (part->svc_id != 0) {
-                rc = loc_service_unregister(part->svc_id);
+                rc = vbds_part_svc_unregister(part);
                 if (rc != EOK)
                         return EIO;
@@ -533 +522 @@
         }
 
+        rc = vbds_part_indices_update(disk);
+        if (rc != EOK) {
+                log_msg(LOG_DEFAULT, LVL_ERROR, "Error updating partition indices");
+                goto error;
+        }
+
         rc = vbds_part_add(disk, lpart, &part);
         if (rc != EOK) {
@@ -581 +576 @@
         }
 
+        rc = vbds_part_indices_update(disk);
+        if (rc != EOK) {
+                log_msg(LOG_DEFAULT, LVL_ERROR, "Error updating partition indices");
+                return EIO;
+        }
+
         return EOK;
 }
@@ -707 +708 @@
 }
 
+/** Register service for partition */
+static int vbds_part_svc_register(vbds_part_t *part)
+{
+        char *name;
+        service_id_t psid;
+        int idx;
+        int rc;
+
+        idx = part->lpart->index;
+
+        rc = asprintf(&name, "%sp%u", part->disk->svc_name, idx);
+        if (rc < 0) {
+                log_msg(LOG_DEFAULT, LVL_ERROR, "Out of memory.");
+                return ENOMEM;
+        }
+
+        rc = loc_service_register(name, &psid);
+        if (rc != EOK) {
+                log_msg(LOG_DEFAULT, LVL_ERROR, "Failed registering "
+                    "service %s.", name);
+                free(name);
+                free(part);
+                return EIO;
+        }
+
+        free(name);
+        part->svc_id = psid;
+        part->reg_idx = idx;
+        return EOK;
+}
+
+/** Unregister service for partition */
+static int vbds_part_svc_unregister(vbds_part_t *part)
+{
+        int rc;
+
+        rc = loc_service_unregister(part->svc_id);
+        if (rc != EOK)
+                return EIO;
+
+        part->svc_id = 0;
+        part->reg_idx = 0;
+        return EOK;
+}
+
+/** Update service names for any partition whose index has changed. */
+static int vbds_part_indices_update(vbds_disk_t *disk)
+{
+        label_part_info_t lpinfo;
+        int rc;
+
+        log_msg(LOG_DEFAULT, LVL_NOTE, "vbds_part_indices_update()");
+
+        /* First unregister services for partitions whose index has changed */
+        list_foreach(disk->parts, ldisk, vbds_part_t, part) {
+                if (part->svc_id != 0 && part->lpart->index != part->reg_idx) {
+                        rc = vbds_part_svc_unregister(part);
+                        if (rc != EOK) {
+                                log_msg(LOG_DEFAULT, LVL_NOTE, "Error "
+                                    "un-registering partition.");
+                                return EIO;
+                        }
+                }
+        }
+
+        /* Now re-register those services under the new indices */
+        list_foreach(disk->parts, ldisk, vbds_part_t, part) {
+                label_part_get_info(part->lpart, &lpinfo);
+                if (part->svc_id == 0 && lpinfo.pkind != lpk_extended) {
+                        rc = vbds_part_svc_register(part);
+                        if (rc != EOK) {
+                                log_msg(LOG_DEFAULT, LVL_NOTE, "Error "
+                                    "re-registering partition.");
+                                return EIO;
+                        }
+                }
+        }
+
+        return EOK;
+}
+
 /** @}
  */