162306a36Sopenharmony_ci/*
262306a36Sopenharmony_ci * Copyright (c) 2016, NVIDIA CORPORATION. All rights reserved.
362306a36Sopenharmony_ci *
462306a36Sopenharmony_ci * Permission is hereby granted, free of charge, to any person obtaining a
562306a36Sopenharmony_ci * copy of this software and associated documentation files (the "Software"),
662306a36Sopenharmony_ci * to deal in the Software without restriction, including without limitation
762306a36Sopenharmony_ci * the rights to use, copy, modify, merge, publish, distribute, sublicense,
862306a36Sopenharmony_ci * and/or sell copies of the Software, and to permit persons to whom the
962306a36Sopenharmony_ci * Software is furnished to do so, subject to the following conditions:
1062306a36Sopenharmony_ci *
1162306a36Sopenharmony_ci * The above copyright notice and this permission notice shall be included in
1262306a36Sopenharmony_ci * all copies or substantial portions of the Software.
1362306a36Sopenharmony_ci *
1462306a36Sopenharmony_ci * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
1562306a36Sopenharmony_ci * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
1662306a36Sopenharmony_ci * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
1762306a36Sopenharmony_ci * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
1862306a36Sopenharmony_ci * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
1962306a36Sopenharmony_ci * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
2062306a36Sopenharmony_ci * DEALINGS IN THE SOFTWARE.
2162306a36Sopenharmony_ci */
2262306a36Sopenharmony_ci#include "priv.h"
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_ci#include <subdev/mc.h>
2562306a36Sopenharmony_ci#include <subdev/timer.h>
2662306a36Sopenharmony_ci#include <subdev/top.h>
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_cistatic const struct nvkm_falcon_func_dma *
2962306a36Sopenharmony_cinvkm_falcon_dma(struct nvkm_falcon *falcon, enum nvkm_falcon_mem *mem_type, u32 *mem_base)
3062306a36Sopenharmony_ci{
3162306a36Sopenharmony_ci	switch (*mem_type) {
3262306a36Sopenharmony_ci	case IMEM: return falcon->func->imem_dma;
3362306a36Sopenharmony_ci	case DMEM: return falcon->func->dmem_dma;
3462306a36Sopenharmony_ci	default:
3562306a36Sopenharmony_ci		return NULL;
3662306a36Sopenharmony_ci	}
3762306a36Sopenharmony_ci}
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ciint
4062306a36Sopenharmony_cinvkm_falcon_dma_wr(struct nvkm_falcon *falcon, const u8 *img, u64 dma_addr, u32 dma_base,
4162306a36Sopenharmony_ci		   enum nvkm_falcon_mem mem_type, u32 mem_base, int len, bool sec)
4262306a36Sopenharmony_ci{
4362306a36Sopenharmony_ci	const struct nvkm_falcon_func_dma *dma = nvkm_falcon_dma(falcon, &mem_type, &mem_base);
4462306a36Sopenharmony_ci	const char *type = nvkm_falcon_mem(mem_type);
4562306a36Sopenharmony_ci	const int dmalen = 256;
4662306a36Sopenharmony_ci	u32 dma_start = 0;
4762306a36Sopenharmony_ci	u32 dst, src, cmd;
4862306a36Sopenharmony_ci	int ret, i;
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	if (WARN_ON(!dma->xfer))
5162306a36Sopenharmony_ci		return -EINVAL;
5262306a36Sopenharmony_ci
5362306a36Sopenharmony_ci	if (mem_type == DMEM) {
5462306a36Sopenharmony_ci		dma_start = dma_base;
5562306a36Sopenharmony_ci		dma_addr += dma_base;
5662306a36Sopenharmony_ci	}
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci	FLCN_DBG(falcon, "%s %08x <- %08x bytes at %08x (%010llx %08x)",
5962306a36Sopenharmony_ci		 type, mem_base, len, dma_base, dma_addr - dma_base, dma_start);
6062306a36Sopenharmony_ci	if (WARN_ON(!len || (len & (dmalen - 1))))
6162306a36Sopenharmony_ci		return -EINVAL;
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	ret = dma->init(falcon, dma_addr, dmalen, mem_type, sec, &cmd);
6462306a36Sopenharmony_ci	if (ret)
6562306a36Sopenharmony_ci		return ret;
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci	dst = mem_base;
6862306a36Sopenharmony_ci	src = dma_base;
6962306a36Sopenharmony_ci	if (len) {
7062306a36Sopenharmony_ci		while (len >= dmalen) {
7162306a36Sopenharmony_ci			dma->xfer(falcon, dst, src - dma_start, cmd);
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci			if (img && nvkm_printk_ok(falcon->owner, falcon->user, NV_DBG_TRACE)) {
7462306a36Sopenharmony_ci				for (i = 0; i < dmalen; i += 4, mem_base += 4) {
7562306a36Sopenharmony_ci					const int w = 8, x = (i / 4) % w;
7662306a36Sopenharmony_ci
7762306a36Sopenharmony_ci					if (x == 0)
7862306a36Sopenharmony_ci						printk(KERN_INFO "%s %08x <-", type, mem_base);
7962306a36Sopenharmony_ci					printk(KERN_CONT " %08x", *(u32 *)(img + src + i));
8062306a36Sopenharmony_ci					if (x == (w - 1) || ((i + 4) == dmalen))
8162306a36Sopenharmony_ci						printk(KERN_CONT " <- %08x+%08x", dma_base,
8262306a36Sopenharmony_ci						       src + i - dma_base - (x * 4));
8362306a36Sopenharmony_ci					if (i == (7 * 4))
8462306a36Sopenharmony_ci						printk(KERN_CONT " *");
8562306a36Sopenharmony_ci				}
8662306a36Sopenharmony_ci			}
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci			if (nvkm_msec(falcon->owner->device, 2000,
8962306a36Sopenharmony_ci				if (dma->done(falcon))
9062306a36Sopenharmony_ci					break;
9162306a36Sopenharmony_ci			) < 0)
9262306a36Sopenharmony_ci				return -ETIMEDOUT;
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_ci			src += dmalen;
9562306a36Sopenharmony_ci			dst += dmalen;
9662306a36Sopenharmony_ci			len -= dmalen;
9762306a36Sopenharmony_ci		}
9862306a36Sopenharmony_ci		WARN_ON(len);
9962306a36Sopenharmony_ci	}
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_ci	return 0;
10262306a36Sopenharmony_ci}
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_cistatic const struct nvkm_falcon_func_pio *
10562306a36Sopenharmony_cinvkm_falcon_pio(struct nvkm_falcon *falcon, enum nvkm_falcon_mem *mem_type, u32 *mem_base)
10662306a36Sopenharmony_ci{
10762306a36Sopenharmony_ci	switch (*mem_type) {
10862306a36Sopenharmony_ci	case IMEM:
10962306a36Sopenharmony_ci		return falcon->func->imem_pio;
11062306a36Sopenharmony_ci	case DMEM:
11162306a36Sopenharmony_ci		if (!falcon->func->emem_addr || *mem_base < falcon->func->emem_addr)
11262306a36Sopenharmony_ci			return falcon->func->dmem_pio;
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_ci		*mem_base -= falcon->func->emem_addr;
11562306a36Sopenharmony_ci		fallthrough;
11662306a36Sopenharmony_ci	case EMEM:
11762306a36Sopenharmony_ci		return falcon->func->emem_pio;
11862306a36Sopenharmony_ci	default:
11962306a36Sopenharmony_ci		return NULL;
12062306a36Sopenharmony_ci	}
12162306a36Sopenharmony_ci}
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_ciint
12462306a36Sopenharmony_cinvkm_falcon_pio_rd(struct nvkm_falcon *falcon, u8 port, enum nvkm_falcon_mem mem_type, u32 mem_base,
12562306a36Sopenharmony_ci		   const u8 *img, u32 img_base, int len)
12662306a36Sopenharmony_ci{
12762306a36Sopenharmony_ci	const struct nvkm_falcon_func_pio *pio = nvkm_falcon_pio(falcon, &mem_type, &mem_base);
12862306a36Sopenharmony_ci	const char *type = nvkm_falcon_mem(mem_type);
12962306a36Sopenharmony_ci	int xfer_len;
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_ci	if (WARN_ON(!pio || !pio->rd))
13262306a36Sopenharmony_ci		return -EINVAL;
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_ci	FLCN_DBG(falcon, "%s %08x -> %08x bytes at %08x", type, mem_base, len, img_base);
13562306a36Sopenharmony_ci	if (WARN_ON(!len || (len & (pio->min - 1))))
13662306a36Sopenharmony_ci		return -EINVAL;
13762306a36Sopenharmony_ci
13862306a36Sopenharmony_ci	pio->rd_init(falcon, port, mem_base);
13962306a36Sopenharmony_ci	do {
14062306a36Sopenharmony_ci		xfer_len = min(len, pio->max);
14162306a36Sopenharmony_ci		pio->rd(falcon, port, img, xfer_len);
14262306a36Sopenharmony_ci
14362306a36Sopenharmony_ci		if (nvkm_printk_ok(falcon->owner, falcon->user, NV_DBG_TRACE)) {
14462306a36Sopenharmony_ci			for (img_base = 0; img_base < xfer_len; img_base += 4, mem_base += 4) {
14562306a36Sopenharmony_ci				if (((img_base / 4) % 8) == 0)
14662306a36Sopenharmony_ci					printk(KERN_INFO "%s %08x ->", type, mem_base);
14762306a36Sopenharmony_ci				printk(KERN_CONT " %08x", *(u32 *)(img + img_base));
14862306a36Sopenharmony_ci			}
14962306a36Sopenharmony_ci		}
15062306a36Sopenharmony_ci
15162306a36Sopenharmony_ci		img += xfer_len;
15262306a36Sopenharmony_ci		len -= xfer_len;
15362306a36Sopenharmony_ci	} while (len);
15462306a36Sopenharmony_ci
15562306a36Sopenharmony_ci	return 0;
15662306a36Sopenharmony_ci}
15762306a36Sopenharmony_ci
15862306a36Sopenharmony_ciint
15962306a36Sopenharmony_cinvkm_falcon_pio_wr(struct nvkm_falcon *falcon, const u8 *img, u32 img_base, u8 port,
16062306a36Sopenharmony_ci		   enum nvkm_falcon_mem mem_type, u32 mem_base, int len, u16 tag, bool sec)
16162306a36Sopenharmony_ci{
16262306a36Sopenharmony_ci	const struct nvkm_falcon_func_pio *pio = nvkm_falcon_pio(falcon, &mem_type, &mem_base);
16362306a36Sopenharmony_ci	const char *type = nvkm_falcon_mem(mem_type);
16462306a36Sopenharmony_ci	int xfer_len;
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_ci	if (WARN_ON(!pio || !pio->wr))
16762306a36Sopenharmony_ci		return -EINVAL;
16862306a36Sopenharmony_ci
16962306a36Sopenharmony_ci	FLCN_DBG(falcon, "%s %08x <- %08x bytes at %08x", type, mem_base, len, img_base);
17062306a36Sopenharmony_ci	if (WARN_ON(!len || (len & (pio->min - 1))))
17162306a36Sopenharmony_ci		return -EINVAL;
17262306a36Sopenharmony_ci
17362306a36Sopenharmony_ci	pio->wr_init(falcon, port, sec, mem_base);
17462306a36Sopenharmony_ci	do {
17562306a36Sopenharmony_ci		xfer_len = min(len, pio->max);
17662306a36Sopenharmony_ci		pio->wr(falcon, port, img, xfer_len, tag++);
17762306a36Sopenharmony_ci
17862306a36Sopenharmony_ci		if (nvkm_printk_ok(falcon->owner, falcon->user, NV_DBG_TRACE)) {
17962306a36Sopenharmony_ci			for (img_base = 0; img_base < xfer_len; img_base += 4, mem_base += 4) {
18062306a36Sopenharmony_ci				if (((img_base / 4) % 8) == 0)
18162306a36Sopenharmony_ci					printk(KERN_INFO "%s %08x <-", type, mem_base);
18262306a36Sopenharmony_ci				printk(KERN_CONT " %08x", *(u32 *)(img + img_base));
18362306a36Sopenharmony_ci				if ((img_base / 4) == 7 && mem_type == IMEM)
18462306a36Sopenharmony_ci					printk(KERN_CONT " %04x", tag - 1);
18562306a36Sopenharmony_ci			}
18662306a36Sopenharmony_ci		}
18762306a36Sopenharmony_ci
18862306a36Sopenharmony_ci		img += xfer_len;
18962306a36Sopenharmony_ci		len -= xfer_len;
19062306a36Sopenharmony_ci	} while (len);
19162306a36Sopenharmony_ci
19262306a36Sopenharmony_ci	return 0;
19362306a36Sopenharmony_ci}
19462306a36Sopenharmony_ci
19562306a36Sopenharmony_civoid
19662306a36Sopenharmony_cinvkm_falcon_load_imem(struct nvkm_falcon *falcon, void *data, u32 start,
19762306a36Sopenharmony_ci		      u32 size, u16 tag, u8 port, bool secure)
19862306a36Sopenharmony_ci{
19962306a36Sopenharmony_ci	if (secure && !falcon->secret) {
20062306a36Sopenharmony_ci		nvkm_warn(falcon->user,
20162306a36Sopenharmony_ci			  "writing with secure tag on a non-secure falcon!\n");
20262306a36Sopenharmony_ci		return;
20362306a36Sopenharmony_ci	}
20462306a36Sopenharmony_ci
20562306a36Sopenharmony_ci	falcon->func->load_imem(falcon, data, start, size, tag, port,
20662306a36Sopenharmony_ci				secure);
20762306a36Sopenharmony_ci}
20862306a36Sopenharmony_ci
20962306a36Sopenharmony_civoid
21062306a36Sopenharmony_cinvkm_falcon_load_dmem(struct nvkm_falcon *falcon, void *data, u32 start,
21162306a36Sopenharmony_ci		      u32 size, u8 port)
21262306a36Sopenharmony_ci{
21362306a36Sopenharmony_ci	mutex_lock(&falcon->dmem_mutex);
21462306a36Sopenharmony_ci
21562306a36Sopenharmony_ci	falcon->func->load_dmem(falcon, data, start, size, port);
21662306a36Sopenharmony_ci
21762306a36Sopenharmony_ci	mutex_unlock(&falcon->dmem_mutex);
21862306a36Sopenharmony_ci}
21962306a36Sopenharmony_ci
22062306a36Sopenharmony_civoid
22162306a36Sopenharmony_cinvkm_falcon_start(struct nvkm_falcon *falcon)
22262306a36Sopenharmony_ci{
22362306a36Sopenharmony_ci	falcon->func->start(falcon);
22462306a36Sopenharmony_ci}
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ciint
22762306a36Sopenharmony_cinvkm_falcon_reset(struct nvkm_falcon *falcon)
22862306a36Sopenharmony_ci{
22962306a36Sopenharmony_ci	int ret;
23062306a36Sopenharmony_ci
23162306a36Sopenharmony_ci	ret = falcon->func->disable(falcon);
23262306a36Sopenharmony_ci	if (WARN_ON(ret))
23362306a36Sopenharmony_ci		return ret;
23462306a36Sopenharmony_ci
23562306a36Sopenharmony_ci	return nvkm_falcon_enable(falcon);
23662306a36Sopenharmony_ci}
23762306a36Sopenharmony_ci
23862306a36Sopenharmony_cistatic int
23962306a36Sopenharmony_cinvkm_falcon_oneinit(struct nvkm_falcon *falcon)
24062306a36Sopenharmony_ci{
24162306a36Sopenharmony_ci	const struct nvkm_falcon_func *func = falcon->func;
24262306a36Sopenharmony_ci	const struct nvkm_subdev *subdev = falcon->owner;
24362306a36Sopenharmony_ci	u32 reg;
24462306a36Sopenharmony_ci
24562306a36Sopenharmony_ci	if (!falcon->addr) {
24662306a36Sopenharmony_ci		falcon->addr = nvkm_top_addr(subdev->device, subdev->type, subdev->inst);
24762306a36Sopenharmony_ci		if (WARN_ON(!falcon->addr))
24862306a36Sopenharmony_ci			return -ENODEV;
24962306a36Sopenharmony_ci	}
25062306a36Sopenharmony_ci
25162306a36Sopenharmony_ci	reg = nvkm_falcon_rd32(falcon, 0x12c);
25262306a36Sopenharmony_ci	falcon->version = reg & 0xf;
25362306a36Sopenharmony_ci	falcon->secret = (reg >> 4) & 0x3;
25462306a36Sopenharmony_ci	falcon->code.ports = (reg >> 8) & 0xf;
25562306a36Sopenharmony_ci	falcon->data.ports = (reg >> 12) & 0xf;
25662306a36Sopenharmony_ci
25762306a36Sopenharmony_ci	reg = nvkm_falcon_rd32(falcon, 0x108);
25862306a36Sopenharmony_ci	falcon->code.limit = (reg & 0x1ff) << 8;
25962306a36Sopenharmony_ci	falcon->data.limit = (reg & 0x3fe00) >> 1;
26062306a36Sopenharmony_ci
26162306a36Sopenharmony_ci	if (func->debug) {
26262306a36Sopenharmony_ci		u32 val = nvkm_falcon_rd32(falcon, func->debug);
26362306a36Sopenharmony_ci		falcon->debug = (val >> 20) & 0x1;
26462306a36Sopenharmony_ci	}
26562306a36Sopenharmony_ci
26662306a36Sopenharmony_ci	return 0;
26762306a36Sopenharmony_ci}
26862306a36Sopenharmony_ci
26962306a36Sopenharmony_civoid
27062306a36Sopenharmony_cinvkm_falcon_put(struct nvkm_falcon *falcon, struct nvkm_subdev *user)
27162306a36Sopenharmony_ci{
27262306a36Sopenharmony_ci	if (unlikely(!falcon))
27362306a36Sopenharmony_ci		return;
27462306a36Sopenharmony_ci
27562306a36Sopenharmony_ci	mutex_lock(&falcon->mutex);
27662306a36Sopenharmony_ci	if (falcon->user == user) {
27762306a36Sopenharmony_ci		nvkm_debug(falcon->user, "released %s falcon\n", falcon->name);
27862306a36Sopenharmony_ci		falcon->user = NULL;
27962306a36Sopenharmony_ci	}
28062306a36Sopenharmony_ci	mutex_unlock(&falcon->mutex);
28162306a36Sopenharmony_ci}
28262306a36Sopenharmony_ci
28362306a36Sopenharmony_ciint
28462306a36Sopenharmony_cinvkm_falcon_get(struct nvkm_falcon *falcon, struct nvkm_subdev *user)
28562306a36Sopenharmony_ci{
28662306a36Sopenharmony_ci	int ret = 0;
28762306a36Sopenharmony_ci
28862306a36Sopenharmony_ci	mutex_lock(&falcon->mutex);
28962306a36Sopenharmony_ci	if (falcon->user) {
29062306a36Sopenharmony_ci		nvkm_error(user, "%s falcon already acquired by %s!\n",
29162306a36Sopenharmony_ci			   falcon->name, falcon->user->name);
29262306a36Sopenharmony_ci		mutex_unlock(&falcon->mutex);
29362306a36Sopenharmony_ci		return -EBUSY;
29462306a36Sopenharmony_ci	}
29562306a36Sopenharmony_ci
29662306a36Sopenharmony_ci	nvkm_debug(user, "acquired %s falcon\n", falcon->name);
29762306a36Sopenharmony_ci	if (!falcon->oneinit)
29862306a36Sopenharmony_ci		ret = nvkm_falcon_oneinit(falcon);
29962306a36Sopenharmony_ci	falcon->user = user;
30062306a36Sopenharmony_ci	mutex_unlock(&falcon->mutex);
30162306a36Sopenharmony_ci	return ret;
30262306a36Sopenharmony_ci}
30362306a36Sopenharmony_ci
30462306a36Sopenharmony_civoid
30562306a36Sopenharmony_cinvkm_falcon_dtor(struct nvkm_falcon *falcon)
30662306a36Sopenharmony_ci{
30762306a36Sopenharmony_ci}
30862306a36Sopenharmony_ci
30962306a36Sopenharmony_ciint
31062306a36Sopenharmony_cinvkm_falcon_ctor(const struct nvkm_falcon_func *func,
31162306a36Sopenharmony_ci		 struct nvkm_subdev *subdev, const char *name, u32 addr,
31262306a36Sopenharmony_ci		 struct nvkm_falcon *falcon)
31362306a36Sopenharmony_ci{
31462306a36Sopenharmony_ci	falcon->func = func;
31562306a36Sopenharmony_ci	falcon->owner = subdev;
31662306a36Sopenharmony_ci	falcon->name = name;
31762306a36Sopenharmony_ci	falcon->addr = addr;
31862306a36Sopenharmony_ci	falcon->addr2 = func->addr2;
31962306a36Sopenharmony_ci	mutex_init(&falcon->mutex);
32062306a36Sopenharmony_ci	mutex_init(&falcon->dmem_mutex);
32162306a36Sopenharmony_ci	return 0;
32262306a36Sopenharmony_ci}
323