162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-or-later
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Probing flash chips with QINFO records.
462306a36Sopenharmony_ci * (C) 2008 Korolev Alexey <akorolev@infradead.org>
562306a36Sopenharmony_ci * (C) 2008 Vasiliy Leonenko <vasiliy.leonenko@gmail.com>
662306a36Sopenharmony_ci */
762306a36Sopenharmony_ci#include <linux/module.h>
862306a36Sopenharmony_ci#include <linux/types.h>
962306a36Sopenharmony_ci#include <linux/kernel.h>
1062306a36Sopenharmony_ci#include <linux/init.h>
1162306a36Sopenharmony_ci#include <linux/errno.h>
1262306a36Sopenharmony_ci#include <linux/slab.h>
1362306a36Sopenharmony_ci#include <linux/interrupt.h>
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci#include <linux/mtd/xip.h>
1662306a36Sopenharmony_ci#include <linux/mtd/map.h>
1762306a36Sopenharmony_ci#include <linux/mtd/pfow.h>
1862306a36Sopenharmony_ci#include <linux/mtd/qinfo.h>
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_cistatic int lpddr_chip_setup(struct map_info *map, struct lpddr_private *lpddr);
2162306a36Sopenharmony_cistruct mtd_info *lpddr_probe(struct map_info *map);
2262306a36Sopenharmony_cistatic struct lpddr_private *lpddr_probe_chip(struct map_info *map);
2362306a36Sopenharmony_cistatic int lpddr_pfow_present(struct map_info *map,
2462306a36Sopenharmony_ci			struct lpddr_private *lpddr);
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_cistatic struct qinfo_query_info qinfo_array[] = {
2762306a36Sopenharmony_ci	/* General device info */
2862306a36Sopenharmony_ci	{0, 0, "DevSizeShift", "Device size 2^n bytes"},
2962306a36Sopenharmony_ci	{0, 3, "BufSizeShift", "Program buffer size 2^n bytes"},
3062306a36Sopenharmony_ci	/* Erase block information */
3162306a36Sopenharmony_ci	{1, 1, "TotalBlocksNum", "Total number of blocks"},
3262306a36Sopenharmony_ci	{1, 2, "UniformBlockSizeShift", "Uniform block size 2^n bytes"},
3362306a36Sopenharmony_ci	/* Partition information */
3462306a36Sopenharmony_ci	{2, 1, "HWPartsNum", "Number of hardware partitions"},
3562306a36Sopenharmony_ci	/* Optional features */
3662306a36Sopenharmony_ci	{5, 1, "SuspEraseSupp", "Suspend erase supported"},
3762306a36Sopenharmony_ci	/* Operation typical time */
3862306a36Sopenharmony_ci	{10, 0, "SingleWordProgTime", "Single word program 2^n u-sec"},
3962306a36Sopenharmony_ci	{10, 1, "ProgBufferTime", "Program buffer write 2^n u-sec"},
4062306a36Sopenharmony_ci	{10, 2, "BlockEraseTime", "Block erase 2^n m-sec"},
4162306a36Sopenharmony_ci	{10, 3, "FullChipEraseTime", "Full chip erase 2^n m-sec"},
4262306a36Sopenharmony_ci};
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_cistatic long lpddr_get_qinforec_pos(struct map_info *map, char *id_str)
4562306a36Sopenharmony_ci{
4662306a36Sopenharmony_ci	int qinfo_lines = ARRAY_SIZE(qinfo_array);
4762306a36Sopenharmony_ci	int i;
4862306a36Sopenharmony_ci	int bankwidth = map_bankwidth(map) * 8;
4962306a36Sopenharmony_ci	int major, minor;
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci	for (i = 0; i < qinfo_lines; i++) {
5262306a36Sopenharmony_ci		if (strcmp(id_str, qinfo_array[i].id_str) == 0) {
5362306a36Sopenharmony_ci			major = qinfo_array[i].major & ((1 << bankwidth) - 1);
5462306a36Sopenharmony_ci			minor = qinfo_array[i].minor & ((1 << bankwidth) - 1);
5562306a36Sopenharmony_ci			return minor | (major << bankwidth);
5662306a36Sopenharmony_ci		}
5762306a36Sopenharmony_ci	}
5862306a36Sopenharmony_ci	printk(KERN_ERR"%s qinfo id string is wrong! \n", map->name);
5962306a36Sopenharmony_ci	BUG();
6062306a36Sopenharmony_ci	return -1;
6162306a36Sopenharmony_ci}
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_cistatic uint16_t lpddr_info_query(struct map_info *map, char *id_str)
6462306a36Sopenharmony_ci{
6562306a36Sopenharmony_ci	unsigned int dsr, val;
6662306a36Sopenharmony_ci	int bits_per_chip = map_bankwidth(map) * 8;
6762306a36Sopenharmony_ci	unsigned long adr = lpddr_get_qinforec_pos(map, id_str);
6862306a36Sopenharmony_ci	int attempts = 20;
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_ci	/* Write a request for the PFOW record */
7162306a36Sopenharmony_ci	map_write(map, CMD(LPDDR_INFO_QUERY),
7262306a36Sopenharmony_ci			map->pfow_base + PFOW_COMMAND_CODE);
7362306a36Sopenharmony_ci	map_write(map, CMD(adr & ((1 << bits_per_chip) - 1)),
7462306a36Sopenharmony_ci			map->pfow_base + PFOW_COMMAND_ADDRESS_L);
7562306a36Sopenharmony_ci	map_write(map, CMD(adr >> bits_per_chip),
7662306a36Sopenharmony_ci			map->pfow_base + PFOW_COMMAND_ADDRESS_H);
7762306a36Sopenharmony_ci	map_write(map, CMD(LPDDR_START_EXECUTION),
7862306a36Sopenharmony_ci			map->pfow_base + PFOW_COMMAND_EXECUTE);
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_ci	while ((attempts--) > 0) {
8162306a36Sopenharmony_ci		dsr = CMDVAL(map_read(map, map->pfow_base + PFOW_DSR));
8262306a36Sopenharmony_ci		if (dsr & DSR_READY_STATUS)
8362306a36Sopenharmony_ci			break;
8462306a36Sopenharmony_ci		udelay(10);
8562306a36Sopenharmony_ci	}
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_ci	val = CMDVAL(map_read(map, map->pfow_base + PFOW_COMMAND_DATA));
8862306a36Sopenharmony_ci	return val;
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_cistatic int lpddr_pfow_present(struct map_info *map, struct lpddr_private *lpddr)
9262306a36Sopenharmony_ci{
9362306a36Sopenharmony_ci	map_word pfow_val[4];
9462306a36Sopenharmony_ci
9562306a36Sopenharmony_ci	/* Check identification string */
9662306a36Sopenharmony_ci	pfow_val[0] = map_read(map, map->pfow_base + PFOW_QUERY_STRING_P);
9762306a36Sopenharmony_ci	pfow_val[1] = map_read(map, map->pfow_base + PFOW_QUERY_STRING_F);
9862306a36Sopenharmony_ci	pfow_val[2] = map_read(map, map->pfow_base + PFOW_QUERY_STRING_O);
9962306a36Sopenharmony_ci	pfow_val[3] = map_read(map, map->pfow_base + PFOW_QUERY_STRING_W);
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_ci	if (!map_word_equal(map, CMD('P'), pfow_val[0]))
10262306a36Sopenharmony_ci		goto out;
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_ci	if (!map_word_equal(map, CMD('F'), pfow_val[1]))
10562306a36Sopenharmony_ci		goto out;
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	if (!map_word_equal(map, CMD('O'), pfow_val[2]))
10862306a36Sopenharmony_ci		goto out;
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_ci	if (!map_word_equal(map, CMD('W'), pfow_val[3]))
11162306a36Sopenharmony_ci		goto out;
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_ci	return 1;	/* "PFOW" is found */
11462306a36Sopenharmony_ciout:
11562306a36Sopenharmony_ci	printk(KERN_WARNING"%s: PFOW string at 0x%lx is not found \n",
11662306a36Sopenharmony_ci					map->name, map->pfow_base);
11762306a36Sopenharmony_ci	return 0;
11862306a36Sopenharmony_ci}
11962306a36Sopenharmony_ci
12062306a36Sopenharmony_cistatic int lpddr_chip_setup(struct map_info *map, struct lpddr_private *lpddr)
12162306a36Sopenharmony_ci{
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_ci	lpddr->qinfo = kzalloc(sizeof(struct qinfo_chip), GFP_KERNEL);
12462306a36Sopenharmony_ci	if (!lpddr->qinfo)
12562306a36Sopenharmony_ci		return 0;
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_ci	/* Get the ManuID */
12862306a36Sopenharmony_ci	lpddr->ManufactId = CMDVAL(map_read(map, map->pfow_base + PFOW_MANUFACTURER_ID));
12962306a36Sopenharmony_ci	/* Get the DeviceID */
13062306a36Sopenharmony_ci	lpddr->DevId = CMDVAL(map_read(map, map->pfow_base + PFOW_DEVICE_ID));
13162306a36Sopenharmony_ci	/* read parameters from chip qinfo table */
13262306a36Sopenharmony_ci	lpddr->qinfo->DevSizeShift = lpddr_info_query(map, "DevSizeShift");
13362306a36Sopenharmony_ci	lpddr->qinfo->TotalBlocksNum = lpddr_info_query(map, "TotalBlocksNum");
13462306a36Sopenharmony_ci	lpddr->qinfo->BufSizeShift = lpddr_info_query(map, "BufSizeShift");
13562306a36Sopenharmony_ci	lpddr->qinfo->HWPartsNum = lpddr_info_query(map, "HWPartsNum");
13662306a36Sopenharmony_ci	lpddr->qinfo->UniformBlockSizeShift =
13762306a36Sopenharmony_ci				lpddr_info_query(map, "UniformBlockSizeShift");
13862306a36Sopenharmony_ci	lpddr->qinfo->SuspEraseSupp = lpddr_info_query(map, "SuspEraseSupp");
13962306a36Sopenharmony_ci	lpddr->qinfo->SingleWordProgTime =
14062306a36Sopenharmony_ci				lpddr_info_query(map, "SingleWordProgTime");
14162306a36Sopenharmony_ci	lpddr->qinfo->ProgBufferTime = lpddr_info_query(map, "ProgBufferTime");
14262306a36Sopenharmony_ci	lpddr->qinfo->BlockEraseTime = lpddr_info_query(map, "BlockEraseTime");
14362306a36Sopenharmony_ci	return 1;
14462306a36Sopenharmony_ci}
14562306a36Sopenharmony_cistatic struct lpddr_private *lpddr_probe_chip(struct map_info *map)
14662306a36Sopenharmony_ci{
14762306a36Sopenharmony_ci	struct lpddr_private lpddr;
14862306a36Sopenharmony_ci	struct lpddr_private *retlpddr;
14962306a36Sopenharmony_ci	int numvirtchips;
15062306a36Sopenharmony_ci
15162306a36Sopenharmony_ci
15262306a36Sopenharmony_ci	if ((map->pfow_base + 0x1000) >= map->size) {
15362306a36Sopenharmony_ci		printk(KERN_NOTICE"%s Probe at base (0x%08lx) past the end of"
15462306a36Sopenharmony_ci				"the map(0x%08lx)\n", map->name,
15562306a36Sopenharmony_ci				(unsigned long)map->pfow_base, map->size - 1);
15662306a36Sopenharmony_ci		return NULL;
15762306a36Sopenharmony_ci	}
15862306a36Sopenharmony_ci	memset(&lpddr, 0, sizeof(struct lpddr_private));
15962306a36Sopenharmony_ci	if (!lpddr_pfow_present(map, &lpddr))
16062306a36Sopenharmony_ci		return NULL;
16162306a36Sopenharmony_ci
16262306a36Sopenharmony_ci	if (!lpddr_chip_setup(map, &lpddr))
16362306a36Sopenharmony_ci		return NULL;
16462306a36Sopenharmony_ci
16562306a36Sopenharmony_ci	/* Ok so we found a chip */
16662306a36Sopenharmony_ci	lpddr.chipshift = lpddr.qinfo->DevSizeShift;
16762306a36Sopenharmony_ci	lpddr.numchips = 1;
16862306a36Sopenharmony_ci
16962306a36Sopenharmony_ci	numvirtchips = lpddr.numchips * lpddr.qinfo->HWPartsNum;
17062306a36Sopenharmony_ci	retlpddr = kzalloc(struct_size(retlpddr, chips, numvirtchips),
17162306a36Sopenharmony_ci			   GFP_KERNEL);
17262306a36Sopenharmony_ci	if (!retlpddr)
17362306a36Sopenharmony_ci		return NULL;
17462306a36Sopenharmony_ci
17562306a36Sopenharmony_ci	memcpy(retlpddr, &lpddr, sizeof(struct lpddr_private));
17662306a36Sopenharmony_ci
17762306a36Sopenharmony_ci	retlpddr->numchips = numvirtchips;
17862306a36Sopenharmony_ci	retlpddr->chipshift = retlpddr->qinfo->DevSizeShift -
17962306a36Sopenharmony_ci				__ffs(retlpddr->qinfo->HWPartsNum);
18062306a36Sopenharmony_ci
18162306a36Sopenharmony_ci	return retlpddr;
18262306a36Sopenharmony_ci}
18362306a36Sopenharmony_ci
18462306a36Sopenharmony_cistruct mtd_info *lpddr_probe(struct map_info *map)
18562306a36Sopenharmony_ci{
18662306a36Sopenharmony_ci	struct mtd_info *mtd = NULL;
18762306a36Sopenharmony_ci	struct lpddr_private *lpddr;
18862306a36Sopenharmony_ci
18962306a36Sopenharmony_ci	/* First probe the map to see if we havecan open PFOW here */
19062306a36Sopenharmony_ci	lpddr = lpddr_probe_chip(map);
19162306a36Sopenharmony_ci	if (!lpddr)
19262306a36Sopenharmony_ci		return NULL;
19362306a36Sopenharmony_ci
19462306a36Sopenharmony_ci	map->fldrv_priv = lpddr;
19562306a36Sopenharmony_ci	mtd = lpddr_cmdset(map);
19662306a36Sopenharmony_ci	if (mtd) {
19762306a36Sopenharmony_ci		if (mtd->size > map->size) {
19862306a36Sopenharmony_ci			printk(KERN_WARNING "Reducing visibility of %ldKiB chip"
19962306a36Sopenharmony_ci				"to %ldKiB\n", (unsigned long)mtd->size >> 10,
20062306a36Sopenharmony_ci				(unsigned long)map->size >> 10);
20162306a36Sopenharmony_ci			mtd->size = map->size;
20262306a36Sopenharmony_ci		}
20362306a36Sopenharmony_ci		return mtd;
20462306a36Sopenharmony_ci	}
20562306a36Sopenharmony_ci
20662306a36Sopenharmony_ci	kfree(lpddr->qinfo);
20762306a36Sopenharmony_ci	kfree(lpddr);
20862306a36Sopenharmony_ci	map->fldrv_priv = NULL;
20962306a36Sopenharmony_ci	return NULL;
21062306a36Sopenharmony_ci}
21162306a36Sopenharmony_ci
21262306a36Sopenharmony_cistatic struct mtd_chip_driver lpddr_chipdrv = {
21362306a36Sopenharmony_ci	.probe		= lpddr_probe,
21462306a36Sopenharmony_ci	.name		= "qinfo_probe",
21562306a36Sopenharmony_ci	.module		= THIS_MODULE
21662306a36Sopenharmony_ci};
21762306a36Sopenharmony_ci
21862306a36Sopenharmony_cistatic int __init lpddr_probe_init(void)
21962306a36Sopenharmony_ci{
22062306a36Sopenharmony_ci	register_mtd_chip_driver(&lpddr_chipdrv);
22162306a36Sopenharmony_ci	return 0;
22262306a36Sopenharmony_ci}
22362306a36Sopenharmony_ci
22462306a36Sopenharmony_cistatic void __exit lpddr_probe_exit(void)
22562306a36Sopenharmony_ci{
22662306a36Sopenharmony_ci	unregister_mtd_chip_driver(&lpddr_chipdrv);
22762306a36Sopenharmony_ci}
22862306a36Sopenharmony_ci
22962306a36Sopenharmony_cimodule_init(lpddr_probe_init);
23062306a36Sopenharmony_cimodule_exit(lpddr_probe_exit);
23162306a36Sopenharmony_ci
23262306a36Sopenharmony_ciMODULE_LICENSE("GPL");
23362306a36Sopenharmony_ciMODULE_AUTHOR("Vasiliy Leonenko <vasiliy.leonenko@gmail.com>");
23462306a36Sopenharmony_ciMODULE_DESCRIPTION("Driver to probe qinfo flash chips");
23562306a36Sopenharmony_ci
236