summaryrefslogtreecommitdiff
path: root/drivers/nvme/nvme-uclass.c
blob: 0895bc9c249678d0e4130288836d00538fe2127c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
/*
 * Copyright (C) 2017 NXP Semiconductors
 * Copyright (C) 2017 Bin Meng <bmeng.cn@gmail.com>
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <errno.h>
#include <dm.h>
#include <dm/device.h>
#include "nvme.h"

static int nvme_info_init(struct uclass *uc)
{
	struct nvme_info *info = (struct nvme_info *)uc->priv;

	info->ns_num = 0;
	info->ndev_num = 0;
	INIT_LIST_HEAD(&info->dev_list);
	nvme_info = info;

	return 0;
}

static int nvme_uclass_post_probe(struct udevice *udev)
{
	char name[20];
	char *str;
	struct udevice *ns_udev;
	int i, ret;
	struct nvme_dev *ndev = dev_get_priv(udev);

	/* Create a blk device for each namespace */
	for (i = 0; i < ndev->nn; i++) {
		sprintf(name, "nvme-blk#%d", nvme_info->ns_num);
		str = strdup(name);
		if (!str)
			return -ENOMEM;

		/* The real blksz and size will be set by nvme_blk_probe() */
		ret = blk_create_device(udev, "nvme-blk", str, IF_TYPE_NVME,
					nvme_info->ns_num++, 512, 0, &ns_udev);
		if (ret) {
			free(str);
			nvme_info->ns_num--;

			return ret;
		}
		device_set_name_alloced(ns_udev);
	}

	return 0;
}

UCLASS_DRIVER(nvme) = {
	.name	= "nvme",
	.id	= UCLASS_NVME,
	.init	= nvme_info_init,
	.post_probe = nvme_uclass_post_probe,
	.priv_auto_alloc_size = sizeof(struct nvme_info),
};