mirror of https://github.com/gnif/vendor-reset
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
2.0 KiB
C
67 lines
2.0 KiB
C
/*
|
|
Vendor Reset - Vendor Specific Reset
|
|
Copyright (C) 2020 Geoffrey McRae <geoff@hostfission.com>
|
|
Copyright (C) 2020 Adam Madsen <adam@ajmadsen.com>
|
|
|
|
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/pci.h>
|
|
#include "vendor-reset-dev.h"
|
|
#include "device-db.h"
|
|
#include "ftrace.h"
|
|
#include "hooks.h"
|
|
|
|
static int (*orig_pci_dev_specific_reset)(struct pci_dev *dev, int probe);
|
|
|
|
static int hooked_pci_dev_specific_reset(struct pci_dev *dev, int probe)
|
|
{
|
|
int ret;
|
|
struct vendor_reset_cfg *cfg;
|
|
struct vendor_reset_dev vdev;
|
|
|
|
ret = orig_pci_dev_specific_reset(dev, probe);
|
|
if (!ret || ret != -ENOTTY)
|
|
return ret;
|
|
|
|
for (cfg = vendor_reset_devices; cfg->vendor; ++cfg)
|
|
if ((cfg->vendor == dev->vendor || cfg->vendor == PCI_ANY_ID) && (cfg->device == dev->device || cfg->device == PCI_ANY_ID))
|
|
break;
|
|
|
|
if (cfg->vendor)
|
|
{
|
|
vdev.pdev = dev;
|
|
vdev.info = cfg->info;
|
|
|
|
if (cfg->ops->pre_reset && (ret = cfg->ops->pre_reset(&vdev)) && ret)
|
|
return ret;
|
|
|
|
ret = vdev.reset_ret = cfg->ops->reset(&vdev);
|
|
if (ret)
|
|
pr_warn("failed to reset device: %d\n", ret);
|
|
|
|
if (cfg->ops->post_reset)
|
|
ret = cfg->ops->post_reset(&vdev);
|
|
|
|
return ret;
|
|
}
|
|
|
|
return -ENOTTY;
|
|
}
|
|
|
|
struct ftrace_hook fh_hooks[] = {
|
|
HOOK("pci_dev_specific_reset", &orig_pci_dev_specific_reset, hooked_pci_dev_specific_reset),
|
|
{0},
|
|
}; |