Signed-off-by: Glenn Washburn <developm...@efficientek.com>
---
grub-core/fs/proc.c | 91 +++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 91 insertions(+)
diff --git a/grub-core/fs/proc.c b/grub-core/fs/proc.c
index 5f516502d..9baabb7d4 100644
--- a/grub-core/fs/proc.c
+++ b/grub-core/fs/proc.c
@@ -190,10 +190,101 @@ static struct grub_fs grub_procfs_fs =
.next = 0
};
+/* Context for grub_device_iterate. */
+struct grub_device_iterate_ctx
+{
+ char *str;
+ grub_size_t len, size;
+};
+
+#define ALLOC_SIZE 512
+static int
+iterate_disk (const char *disk_name, void *data)
+{
+ struct grub_device_iterate_ctx *ctx = data;
+ grub_device_t dev;
+ int size = 0, ret = 0;
+ char *buf;
+
+ dev = grub_device_open (disk_name);
+ if (! dev)
+ {
+ grub_errno = GRUB_ERR_NONE;
+ return 0;
+ }
+
+ if (!dev->disk || dev->disk->partition)
+ goto err;
+
+ do {
+ grub_disk_t disk = dev->disk;
+
+ size = grub_snprintf(ctx->str + ctx->len, ctx->size - ctx->len,
+ "\n%s %s %"PRIuGRUB_UINT64_T" %u",
+ disk->name, (disk->dev) ? disk->dev->name : "none",
+ disk->total_sectors, 1 << disk->log_sector_size);
+
+ if ((size + ctx->len) <= ctx->size)
+ {
+ ctx->len += size;
+ break;
+ }
+
+ /* Not enough space left in string, reallocate more and try again. */
+ buf = grub_realloc(ctx->str, ctx->size + ALLOC_SIZE);
+ if (!buf)
+ goto err;
+
+ ctx->str = buf;
+ ctx->size += ALLOC_SIZE;
+ } while (1);
+
+err:
+ grub_device_close (dev);
+ return ret;
+}
+
+static char *
+devices_get (grub_size_t *sz)
+{
+ struct grub_device_iterate_ctx ctx = { NULL, 0, 0 };
+ static const char header[] = N_("<name> <type> <nsectors> <sector size>");
+
+ *sz = 0;
+ ctx.str = grub_malloc(ALLOC_SIZE);
+ ctx.size = ALLOC_SIZE;
+ if (ctx.str == NULL)
+ {
+ grub_print_error();
+ return NULL;
+ }
+
+ ctx.len = grub_stpcpy (ctx.str, header) - ctx.str;
+
+ if (grub_disk_dev_iterate (iterate_disk, &ctx))
+ grub_print_error();
+
+ *sz = ctx.len;
+ return ctx.str;
+}
+
+struct grub_procfs_entry procfs[] =
+{
+ {
+ .name = "devices",
+ .get_contents = devices_get
+ }
+};
+
GRUB_MOD_INIT (procfs)
{
+ struct grub_procfs_entry *pe;
+
grub_disk_dev_register (&grub_procfs_dev);
grub_fs_register (&grub_procfs_fs);
+
+ for (pe = procfs + (sizeof (procfs)/sizeof (*procfs)) - 1; (pe + 1) !=
procfs; pe--)
+ grub_procfs_register (pe->name, pe);
}
GRUB_MOD_FINI (procfs)
--
2.27.0
_______________________________________________
Grub-devel mailing list
Grub-devel@gnu.org
https://lists.gnu.org/mailman/listinfo/grub-devel
