brintos

brintos / linux-shallow public Read only

0
0
Text · 6.3 KiB · 70e8bdf Raw
230 lines · c
1// SPDX-License-Identifier: GPL-2.02/* Copyright (c) 2019 Mellanox Technologies. All rights reserved */3 4#include <linux/debugfs.h>5#include <linux/err.h>6#include <linux/kernel.h>7#include <linux/slab.h>8 9#include "netdevsim.h"10 11static int12nsim_dev_empty_reporter_dump(struct devlink_health_reporter *reporter,13			     struct devlink_fmsg *fmsg, void *priv_ctx,14			     struct netlink_ext_ack *extack)15{16	return 0;17}18 19static int20nsim_dev_empty_reporter_diagnose(struct devlink_health_reporter *reporter,21				 struct devlink_fmsg *fmsg,22				 struct netlink_ext_ack *extack)23{24	return 0;25}26 27static const28struct devlink_health_reporter_ops nsim_dev_empty_reporter_ops = {29	.name = "empty",30	.dump = nsim_dev_empty_reporter_dump,31	.diagnose = nsim_dev_empty_reporter_diagnose,32};33 34struct nsim_dev_dummy_reporter_ctx {35	char *break_msg;36};37 38static int39nsim_dev_dummy_reporter_recover(struct devlink_health_reporter *reporter,40				void *priv_ctx,41				struct netlink_ext_ack *extack)42{43	struct nsim_dev_health *health = devlink_health_reporter_priv(reporter);44	struct nsim_dev_dummy_reporter_ctx *ctx = priv_ctx;45 46	if (health->fail_recover) {47		/* For testing purposes, user set debugfs fail_recover48		 * value to true. Fail right away.49		 */50		NL_SET_ERR_MSG_MOD(extack, "User setup the recover to fail for testing purposes");51		return -EINVAL;52	}53	if (ctx) {54		kfree(health->recovered_break_msg);55		health->recovered_break_msg = kstrdup(ctx->break_msg,56						      GFP_KERNEL);57		if (!health->recovered_break_msg)58			return -ENOMEM;59	}60	return 0;61}62 63static int nsim_dev_dummy_fmsg_put(struct devlink_fmsg *fmsg, u32 binary_len)64{65	char *binary;66	int i;67 68	devlink_fmsg_bool_pair_put(fmsg, "test_bool", true);69	devlink_fmsg_u8_pair_put(fmsg, "test_u8", 1);70	devlink_fmsg_u32_pair_put(fmsg, "test_u32", 3);71	devlink_fmsg_u64_pair_put(fmsg, "test_u64", 4);72	devlink_fmsg_string_pair_put(fmsg, "test_string", "somestring");73 74	binary = kmalloc(binary_len, GFP_KERNEL | __GFP_NOWARN);75	if (!binary)76		return -ENOMEM;77	get_random_bytes(binary, binary_len);78	devlink_fmsg_binary_pair_put(fmsg, "test_binary", binary, binary_len);79	kfree(binary);80 81	devlink_fmsg_pair_nest_start(fmsg, "test_nest");82	devlink_fmsg_obj_nest_start(fmsg);83	devlink_fmsg_bool_pair_put(fmsg, "nested_test_bool", false);84	devlink_fmsg_u8_pair_put(fmsg, "nested_test_u8", false);85	devlink_fmsg_obj_nest_end(fmsg);86	devlink_fmsg_pair_nest_end(fmsg);87	devlink_fmsg_arr_pair_nest_end(fmsg);88	devlink_fmsg_arr_pair_nest_start(fmsg, "test_u32_array");89 90	for (i = 0; i < 10; i++)91		devlink_fmsg_u32_put(fmsg, i);92	devlink_fmsg_arr_pair_nest_end(fmsg);93	devlink_fmsg_arr_pair_nest_start(fmsg, "test_array_of_objects");94 95	for (i = 0; i < 10; i++) {96		devlink_fmsg_obj_nest_start(fmsg);97		devlink_fmsg_bool_pair_put(fmsg, "in_array_nested_test_bool",98					   false);99		devlink_fmsg_u8_pair_put(fmsg, "in_array_nested_test_u8", i);100		devlink_fmsg_obj_nest_end(fmsg);101	}102	devlink_fmsg_arr_pair_nest_end(fmsg);103 104	return 0;105}106 107static int108nsim_dev_dummy_reporter_dump(struct devlink_health_reporter *reporter,109			     struct devlink_fmsg *fmsg, void *priv_ctx,110			     struct netlink_ext_ack *extack)111{112	struct nsim_dev_health *health = devlink_health_reporter_priv(reporter);113	struct nsim_dev_dummy_reporter_ctx *ctx = priv_ctx;114 115	if (ctx)116		devlink_fmsg_string_pair_put(fmsg, "break_message", ctx->break_msg);117 118	return nsim_dev_dummy_fmsg_put(fmsg, health->binary_len);119}120 121static int122nsim_dev_dummy_reporter_diagnose(struct devlink_health_reporter *reporter,123				 struct devlink_fmsg *fmsg,124				 struct netlink_ext_ack *extack)125{126	struct nsim_dev_health *health = devlink_health_reporter_priv(reporter);127 128	if (health->recovered_break_msg)129		devlink_fmsg_string_pair_put(fmsg, "recovered_break_message",130					     health->recovered_break_msg);131 132	return nsim_dev_dummy_fmsg_put(fmsg, health->binary_len);133}134 135static const136struct devlink_health_reporter_ops nsim_dev_dummy_reporter_ops = {137	.name = "dummy",138	.recover = nsim_dev_dummy_reporter_recover,139	.dump = nsim_dev_dummy_reporter_dump,140	.diagnose = nsim_dev_dummy_reporter_diagnose,141};142 143static ssize_t nsim_dev_health_break_write(struct file *file,144					   const char __user *data,145					   size_t count, loff_t *ppos)146{147	struct nsim_dev_health *health = file->private_data;148	struct nsim_dev_dummy_reporter_ctx ctx;149	char *break_msg;150	int err;151 152	break_msg = memdup_user_nul(data, count);153	if (IS_ERR(break_msg))154		return PTR_ERR(break_msg);155 156	if (break_msg[count - 1] == '\n')157		break_msg[count - 1] = '\0';158 159	ctx.break_msg = break_msg;160	err = devlink_health_report(health->dummy_reporter, break_msg, &ctx);161	if (err)162		goto out;163 164out:165	kfree(break_msg);166	return err ?: count;167}168 169static const struct file_operations nsim_dev_health_break_fops = {170	.open = simple_open,171	.write = nsim_dev_health_break_write,172	.llseek = generic_file_llseek,173	.owner = THIS_MODULE,174};175 176int nsim_dev_health_init(struct nsim_dev *nsim_dev, struct devlink *devlink)177{178	struct nsim_dev_health *health = &nsim_dev->health;179	int err;180 181	health->empty_reporter =182		devl_health_reporter_create(devlink,183					    &nsim_dev_empty_reporter_ops,184					    0, health);185	if (IS_ERR(health->empty_reporter))186		return PTR_ERR(health->empty_reporter);187 188	health->dummy_reporter =189		devl_health_reporter_create(devlink,190					    &nsim_dev_dummy_reporter_ops,191					    0, health);192	if (IS_ERR(health->dummy_reporter)) {193		err = PTR_ERR(health->dummy_reporter);194		goto err_empty_reporter_destroy;195	}196 197	health->ddir = debugfs_create_dir("health", nsim_dev->ddir);198	if (IS_ERR(health->ddir)) {199		err = PTR_ERR(health->ddir);200		goto err_dummy_reporter_destroy;201	}202 203	health->recovered_break_msg = NULL;204	debugfs_create_file("break_health", 0200, health->ddir, health,205			    &nsim_dev_health_break_fops);206	health->binary_len = 16;207	debugfs_create_u32("binary_len", 0600, health->ddir,208			   &health->binary_len);209	health->fail_recover = false;210	debugfs_create_bool("fail_recover", 0600, health->ddir,211			    &health->fail_recover);212	return 0;213 214err_dummy_reporter_destroy:215	devl_health_reporter_destroy(health->dummy_reporter);216err_empty_reporter_destroy:217	devl_health_reporter_destroy(health->empty_reporter);218	return err;219}220 221void nsim_dev_health_exit(struct nsim_dev *nsim_dev)222{223	struct nsim_dev_health *health = &nsim_dev->health;224 225	debugfs_remove_recursive(health->ddir);226	kfree(health->recovered_break_msg);227	devl_health_reporter_destroy(health->dummy_reporter);228	devl_health_reporter_destroy(health->empty_reporter);229}230