[mirror_ubuntu-bionic-kernel.git] / drivers / misc / sgi-gru / grukdump.c

/*
 * SN Platform GRU Driver
 *
 *            Dump GRU State
 *
 *  Copyright (c) 2008 Silicon Graphics, Inc.  All Rights Reserved.
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
 */

#include <linux/kernel.h>
#include <linux/mm.h>
#include <linux/spinlock.h>
#include <linux/uaccess.h>
#include <linux/delay.h>
#include <linux/bitops.h>
#include <asm/uv/uv_hub.h>

#include <linux/nospec.h>

#include "gru.h"
#include "grutables.h"
#include "gruhandles.h"
#include "grulib.h"

#define CCH_LOCK_ATTEMPTS	10

static int gru_user_copy_handle(void __user **dp, void *s)
{
	if (copy_to_user(*dp, s, GRU_HANDLE_BYTES))
		return -1;
	*dp += GRU_HANDLE_BYTES;
	return 0;
}

static int gru_dump_context_data(void *grubase,
			struct gru_context_configuration_handle *cch,
			void __user *ubuf, int ctxnum, int dsrcnt,
			int flush_cbrs)
{
	void *cb, *cbe, *tfh, *gseg;
	int i, scr;

	gseg = grubase + ctxnum * GRU_GSEG_STRIDE;
	cb = gseg + GRU_CB_BASE;
	cbe = grubase + GRU_CBE_BASE;
	tfh = grubase + GRU_TFH_BASE;

	for_each_cbr_in_allocation_map(i, &cch->cbr_allocation_map, scr) {
		if (flush_cbrs)
			gru_flush_cache(cb);
		if (gru_user_copy_handle(&ubuf, cb))
			goto fail;
		if (gru_user_copy_handle(&ubuf, tfh + i * GRU_HANDLE_STRIDE))
			goto fail;
		if (gru_user_copy_handle(&ubuf, cbe + i * GRU_HANDLE_STRIDE))
			goto fail;
		cb += GRU_HANDLE_STRIDE;
	}
	if (dsrcnt)
		memcpy(ubuf, gseg + GRU_DS_BASE, dsrcnt * GRU_HANDLE_STRIDE);
	return 0;

fail:
	return -EFAULT;
}

static int gru_dump_tfm(struct gru_state *gru,
		void __user *ubuf, void __user *ubufend)
{
	struct gru_tlb_fault_map *tfm;
	int i;

	if (GRU_NUM_TFM * GRU_CACHE_LINE_BYTES > ubufend - ubuf)
		return -EFBIG;

	for (i = 0; i < GRU_NUM_TFM; i++) {
		tfm = get_tfm(gru->gs_gru_base_vaddr, i);
		if (gru_user_copy_handle(&ubuf, tfm))
			goto fail;
	}
	return GRU_NUM_TFM * GRU_CACHE_LINE_BYTES;

fail:
	return -EFAULT;
}

static int gru_dump_tgh(struct gru_state *gru,
		void __user *ubuf, void __user *ubufend)
{
	struct gru_tlb_global_handle *tgh;
	int i;

	if (GRU_NUM_TGH * GRU_CACHE_LINE_BYTES > ubufend - ubuf)
		return -EFBIG;

	for (i = 0; i < GRU_NUM_TGH; i++) {
		tgh = get_tgh(gru->gs_gru_base_vaddr, i);
		if (gru_user_copy_handle(&ubuf, tgh))
			goto fail;
	}
	return GRU_NUM_TGH * GRU_CACHE_LINE_BYTES;

fail:
	return -EFAULT;
}

static int gru_dump_context(struct gru_state *gru, int ctxnum,
		void __user *ubuf, void __user *ubufend, char data_opt,
		char lock_cch, char flush_cbrs)
{
	struct gru_dump_context_header hdr;
	struct gru_dump_context_header __user *uhdr = ubuf;
	struct gru_context_configuration_handle *cch, *ubufcch;
	struct gru_thread_state *gts;
	int try, cch_locked, cbrcnt = 0, dsrcnt = 0, bytes = 0, ret = 0;
	void *grubase;

	memset(&hdr, 0, sizeof(hdr));
	grubase = gru->gs_gru_base_vaddr;
	cch = get_cch(grubase, ctxnum);
	for (try = 0; try < CCH_LOCK_ATTEMPTS; try++) {
		cch_locked =  trylock_cch_handle(cch);
		if (cch_locked)
			break;
		msleep(1);
	}

	ubuf += sizeof(hdr);
	ubufcch = ubuf;
	if (gru_user_copy_handle(&ubuf, cch)) {
		if (cch_locked)
			unlock_cch_handle(cch);
		return -EFAULT;
	}
	if (cch_locked)
		ubufcch->delresp = 0;
	bytes = sizeof(hdr) + GRU_CACHE_LINE_BYTES;

	if (cch_locked || !lock_cch) {
		gts = gru->gs_gts[ctxnum];
		if (gts && gts->ts_vma) {
			hdr.pid = gts->ts_tgid_owner;
			hdr.vaddr = gts->ts_vma->vm_start;
		}
		if (cch->state != CCHSTATE_INACTIVE) {
			cbrcnt = hweight64(cch->cbr_allocation_map) *
						GRU_CBR_AU_SIZE;
			dsrcnt = data_opt ? hweight32(cch->dsr_allocation_map) *
						GRU_DSR_AU_CL : 0;
		}
		bytes += (3 * cbrcnt + dsrcnt) * GRU_CACHE_LINE_BYTES;
		if (bytes > ubufend - ubuf)
			ret = -EFBIG;
		else
			ret = gru_dump_context_data(grubase, cch, ubuf, ctxnum,
							dsrcnt, flush_cbrs);
	}
	if (cch_locked)
		unlock_cch_handle(cch);
	if (ret)
		return ret;

	hdr.magic = GRU_DUMP_MAGIC;
	hdr.gid = gru->gs_gid;
	hdr.ctxnum = ctxnum;
	hdr.cbrcnt = cbrcnt;
	hdr.dsrcnt = dsrcnt;
	hdr.cch_locked = cch_locked;
	if (copy_to_user(uhdr, &hdr, sizeof(hdr)))
		return -EFAULT;

	return bytes;
}

int gru_dump_chiplet_request(unsigned long arg)
{
	struct gru_state *gru;
	struct gru_dump_chiplet_state_req req;
	void __user *ubuf;
	void __user *ubufend;
	int ctxnum, ret, cnt = 0;

	if (copy_from_user(&req, (void __user *)arg, sizeof(req)))
		return -EFAULT;

	/* Currently, only dump by gid is implemented */
	if (req.gid >= gru_max_gids)
		return -EINVAL;
	req.gid = array_index_nospec(req.gid, gru_max_gids);

	gru = GID_TO_GRU(req.gid);
	ubuf = req.buf;
	ubufend = req.buf + req.buflen;

	ret = gru_dump_tfm(gru, ubuf, ubufend);
	if (ret < 0)
		goto fail;
	ubuf += ret;

	ret = gru_dump_tgh(gru, ubuf, ubufend);
	if (ret < 0)
		goto fail;
	ubuf += ret;

	for (ctxnum = 0; ctxnum < GRU_NUM_CCH; ctxnum++) {
		if (req.ctxnum == ctxnum || req.ctxnum < 0) {
			ret = gru_dump_context(gru, ctxnum, ubuf, ubufend,
						req.data_opt, req.lock_cch,
						req.flush_cbrs);
			if (ret < 0)
				goto fail;
			ubuf += ret;
			cnt++;
		}
	}

	if (copy_to_user((void __user *)arg, &req, sizeof(req)))
		return -EFAULT;
	return cnt;

fail:
	return ret;
}
Commit	Line	Data
9cc9b056 JS	1	/*
	2	* SN Platform GRU Driver
	3	*
	4	* Dump GRU State
	5	*
8820f27a	6	* Copyright (c) 2008 Silicon Graphics, Inc. All Rights Reserved.
9cc9b056	7	*
8820f27a JS	8	* This program is free software; you can redistribute it and/or modify
	9	* it under the terms of the GNU General Public License as published by
	10	* the Free Software Foundation; either version 2 of the License, or
	11	* (at your option) any later version.
	12	*
	13	* This program is distributed in the hope that it will be useful,
	14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	16	* GNU General Public License for more details.
	17	*
	18	* You should have received a copy of the GNU General Public License
	19	* along with this program; if not, write to the Free Software
	20	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
9cc9b056 JS	21	*/
	22
	23	#include <linux/kernel.h>
	24	#include <linux/mm.h>
	25	#include <linux/spinlock.h>
	26	#include <linux/uaccess.h>
	27	#include <linux/delay.h>
	28	#include <linux/bitops.h>
	29	#include <asm/uv/uv_hub.h>
073fd29a GS	30
	31	#include <linux/nospec.h>
	32
9cc9b056 JS	33	#include "gru.h"
	34	#include "grutables.h"
	35	#include "gruhandles.h"
	36	#include "grulib.h"
	37
	38	#define CCH_LOCK_ATTEMPTS 10
	39
	40	static int gru_user_copy_handle(void __user *dp, void s)
	41	{
2b702b28	42	if (copy_to_user(*dp, s, GRU_HANDLE_BYTES))
9cc9b056 JS	43	return -1;
	44	*dp += GRU_HANDLE_BYTES;
	45	return 0;
	46	}
	47
	48	static int gru_dump_context_data(void *grubase,
	49	struct gru_context_configuration_handle *cch,
b8229bed JS	50	void __user *ubuf, int ctxnum, int dsrcnt,
b8229bed JS	51	int flush_cbrs)
9cc9b056 JS	52	{
	53	void cb, cbe, tfh, gseg;
	54	int i, scr;
	55
	56	gseg = grubase + ctxnum * GRU_GSEG_STRIDE;
	57	cb = gseg + GRU_CB_BASE;
	58	cbe = grubase + GRU_CBE_BASE;
	59	tfh = grubase + GRU_TFH_BASE;
	60
	61	for_each_cbr_in_allocation_map(i, &cch->cbr_allocation_map, scr) {
b8229bed JS	62	if (flush_cbrs)
b8229bed JS	63	gru_flush_cache(cb);
9cc9b056 JS	64	if (gru_user_copy_handle(&ubuf, cb))
	65	goto fail;
	66	if (gru_user_copy_handle(&ubuf, tfh + i * GRU_HANDLE_STRIDE))
	67	goto fail;
	68	if (gru_user_copy_handle(&ubuf, cbe + i * GRU_HANDLE_STRIDE))
	69	goto fail;
	70	cb += GRU_HANDLE_STRIDE;
	71	}
	72	if (dsrcnt)
	73	memcpy(ubuf, gseg + GRU_DS_BASE, dsrcnt * GRU_HANDLE_STRIDE);
	74	return 0;
	75
	76	fail:
	77	return -EFAULT;
	78	}
	79
	80	static int gru_dump_tfm(struct gru_state *gru,
	81	void __user ubuf, void __user ubufend)
	82	{
	83	struct gru_tlb_fault_map *tfm;
a010d276	84	int i;
9cc9b056	85
a010d276 SM	86	if (GRU_NUM_TFM * GRU_CACHE_LINE_BYTES > ubufend - ubuf)
a010d276 SM	87	return -EFBIG;
9cc9b056 JS	88
	89	for (i = 0; i < GRU_NUM_TFM; i++) {
	90	tfm = get_tfm(gru->gs_gru_base_vaddr, i);
	91	if (gru_user_copy_handle(&ubuf, tfm))
	92	goto fail;
	93	}
	94	return GRU_NUM_TFM * GRU_CACHE_LINE_BYTES;
	95
	96	fail:
	97	return -EFAULT;
	98	}
	99
	100	static int gru_dump_tgh(struct gru_state *gru,
	101	void __user ubuf, void __user ubufend)
	102	{
	103	struct gru_tlb_global_handle *tgh;
a010d276	104	int i;
9cc9b056	105
a010d276 SM	106	if (GRU_NUM_TGH * GRU_CACHE_LINE_BYTES > ubufend - ubuf)
a010d276 SM	107	return -EFBIG;
9cc9b056 JS	108
	109	for (i = 0; i < GRU_NUM_TGH; i++) {
	110	tgh = get_tgh(gru->gs_gru_base_vaddr, i);
	111	if (gru_user_copy_handle(&ubuf, tgh))
	112	goto fail;
	113	}
	114	return GRU_NUM_TGH * GRU_CACHE_LINE_BYTES;
	115
	116	fail:
	117	return -EFAULT;
	118	}
	119
	120	static int gru_dump_context(struct gru_state *gru, int ctxnum,
	121	void __user ubuf, void __user ubufend, char data_opt,
b8229bed	122	char lock_cch, char flush_cbrs)
9cc9b056 JS	123	{
	124	struct gru_dump_context_header hdr;
	125	struct gru_dump_context_header __user *uhdr = ubuf;
2b702b28	126	struct gru_context_configuration_handle cch, ubufcch;
9cc9b056 JS	127	struct gru_thread_state *gts;
	128	int try, cch_locked, cbrcnt = 0, dsrcnt = 0, bytes = 0, ret = 0;
	129	void *grubase;
	130
	131	memset(&hdr, 0, sizeof(hdr));
	132	grubase = gru->gs_gru_base_vaddr;
	133	cch = get_cch(grubase, ctxnum);
	134	for (try = 0; try < CCH_LOCK_ATTEMPTS; try++) {
	135	cch_locked = trylock_cch_handle(cch);
	136	if (cch_locked)
	137	break;
	138	msleep(1);
	139	}
	140
	141	ubuf += sizeof(hdr);
2b702b28	142	ubufcch = ubuf;
49d3d6c3 DC	143	if (gru_user_copy_handle(&ubuf, cch)) {
	144	if (cch_locked)
	145	unlock_cch_handle(cch);
	146	return -EFAULT;
	147	}
2b702b28 JS	148	if (cch_locked)
2b702b28 JS	149	ubufcch->delresp = 0;
9cc9b056 JS	150	bytes = sizeof(hdr) + GRU_CACHE_LINE_BYTES;
	151
	152	if (cch_locked \|\| !lock_cch) {
	153	gts = gru->gs_gts[ctxnum];
836ce679	154	if (gts && gts->ts_vma) {
9cc9b056 JS	155	hdr.pid = gts->ts_tgid_owner;
	156	hdr.vaddr = gts->ts_vma->vm_start;
	157	}
	158	if (cch->state != CCHSTATE_INACTIVE) {
	159	cbrcnt = hweight64(cch->cbr_allocation_map) *
	160	GRU_CBR_AU_SIZE;
	161	dsrcnt = data_opt ? hweight32(cch->dsr_allocation_map) *
	162	GRU_DSR_AU_CL : 0;
	163	}
	164	bytes += (3 * cbrcnt + dsrcnt) * GRU_CACHE_LINE_BYTES;
	165	if (bytes > ubufend - ubuf)
	166	ret = -EFBIG;
	167	else
	168	ret = gru_dump_context_data(grubase, cch, ubuf, ctxnum,
b8229bed	169	dsrcnt, flush_cbrs);
9cc9b056 JS	170	}
	171	if (cch_locked)
	172	unlock_cch_handle(cch);
	173	if (ret)
	174	return ret;
	175
	176	hdr.magic = GRU_DUMP_MAGIC;
2b702b28	177	hdr.gid = gru->gs_gid;
9cc9b056 JS	178	hdr.ctxnum = ctxnum;
	179	hdr.cbrcnt = cbrcnt;
	180	hdr.dsrcnt = dsrcnt;
	181	hdr.cch_locked = cch_locked;
b6a83d92 DC	182	if (copy_to_user(uhdr, &hdr, sizeof(hdr)))
b6a83d92 DC	183	return -EFAULT;
9cc9b056	184
b6a83d92	185	return bytes;
9cc9b056 JS	186	}
	187
	188	int gru_dump_chiplet_request(unsigned long arg)
	189	{
	190	struct gru_state *gru;
	191	struct gru_dump_chiplet_state_req req;
	192	void __user *ubuf;
	193	void __user *ubufend;
	194	int ctxnum, ret, cnt = 0;
	195
	196	if (copy_from_user(&req, (void __user *)arg, sizeof(req)))
	197	return -EFAULT;
	198
	199	/* Currently, only dump by gid is implemented */
c2ed545c	200	if (req.gid >= gru_max_gids)
9cc9b056	201	return -EINVAL;
073fd29a	202	req.gid = array_index_nospec(req.gid, gru_max_gids);
9cc9b056 JS	203
	204	gru = GID_TO_GRU(req.gid);
	205	ubuf = req.buf;
	206	ubufend = req.buf + req.buflen;
	207
	208	ret = gru_dump_tfm(gru, ubuf, ubufend);
	209	if (ret < 0)
	210	goto fail;
	211	ubuf += ret;
	212
	213	ret = gru_dump_tgh(gru, ubuf, ubufend);
	214	if (ret < 0)
	215	goto fail;
	216	ubuf += ret;
	217
	218	for (ctxnum = 0; ctxnum < GRU_NUM_CCH; ctxnum++) {
	219	if (req.ctxnum == ctxnum \|\| req.ctxnum < 0) {
	220	ret = gru_dump_context(gru, ctxnum, ubuf, ubufend,
b8229bed JS	221	req.data_opt, req.lock_cch,
b8229bed JS	222	req.flush_cbrs);
9cc9b056 JS	223	if (ret < 0)
	224	goto fail;
	225	ubuf += ret;
	226	cnt++;
	227	}
	228	}
	229
	230	if (copy_to_user((void __user *)arg, &req, sizeof(req)))
	231	return -EFAULT;
	232	return cnt;
	233
	234	fail:
	235	return ret;
	236	}