[core] seperate ioctl setup from module init

pull/1/head
Geoffrey McRae 4 years ago
parent c1ab03e74a
commit dd8eadc6e7

@ -1,4 +1,5 @@
vendor-reset-y += \ vendor-reset-y += \
src/module.o \
src/vendor-reset-dev.o \ src/vendor-reset-dev.o \
src/ioctl.o \ src/ioctl.o \
src/ftrace.o \ src/ftrace.o \

@ -17,24 +17,12 @@ this program; if not, write to the Free Software Foundation, Inc., 59 Temple
Place, Suite 330, Boston, MA 02111-1307 USA Place, Suite 330, Boston, MA 02111-1307 USA
*/ */
#include <linux/module.h>
#include <linux/miscdevice.h>
#include <linux/fs.h>
#include <linux/highmem.h>
#include <linux/pci.h>
#include "vendor-reset-dev.h" #include "vendor-reset-dev.h"
#include "vendor-reset-ioctl.h" #include "vendor-reset-ioctl.h"
#include "ftrace.h" #include <linux/miscdevice.h>
#include "hooks.h"
#define VENDOR_RESET_DEVNAME "vendor_reset"
#define vr_info(fmt, args...) pr_info("vendor-reset: " fmt, ##args)
static bool install_hook = true; #define VENDOR_RESET_IOCTL_DEVNAME "vendor_reset"
module_param(install_hook, bool, 0);
static long vendor_reset_ioctl_reset(struct file * filp, unsigned long arg) static long vendor_reset_ioctl_reset(struct file * filp, unsigned long arg)
{ {
@ -83,7 +71,7 @@ static long vendor_reset_ioctl(struct file * filp, unsigned int ioctl,
return ret; return ret;
} }
static const struct file_operations vendor_reset_fops = static const struct file_operations vendor_reset_ioctl_fops =
{ {
.owner = THIS_MODULE, .owner = THIS_MODULE,
.unlocked_ioctl = vendor_reset_ioctl, .unlocked_ioctl = vendor_reset_ioctl,
@ -92,48 +80,19 @@ static const struct file_operations vendor_reset_fops =
#endif #endif
}; };
static struct miscdevice vendor_reset_misc = static struct miscdevice vendor_reset_ioctl_misc =
{ {
.minor = MISC_DYNAMIC_MINOR, .minor = MISC_DYNAMIC_MINOR,
.name = VENDOR_RESET_DEVNAME, .name = VENDOR_RESET_IOCTL_DEVNAME,
.fops = &vendor_reset_fops .fops = &vendor_reset_ioctl_fops
}; };
static int __init vendor_reset_init(void) int vendor_reset_ioctl_init(void)
{ {
int ret; return misc_register(&vendor_reset_ioctl_misc);
ret = misc_register(&vendor_reset_misc);
if (ret)
return ret;
if (install_hook)
{
ret = fh_install_hooks(fh_hooks);
if (ret)
goto err;
vr_info("Hooks installed successfully\n");
}
return 0;
err:
misc_deregister(&vendor_reset_misc);
return ret;
} }
static void __exit vendor_reset_exit(void) void vendor_reset_ioctl_exit(void)
{ {
if (install_hook) misc_deregister(&vendor_reset_ioctl_misc);
fh_remove_hooks(fh_hooks);
misc_deregister(&vendor_reset_misc);
} }
module_init(vendor_reset_init);
module_exit(vendor_reset_exit);
MODULE_LICENSE("GPL v2");
MODULE_AUTHOR("Geoffrey McRae <geoff@hostfission.com>");
MODULE_AUTHOR("Adam Madsen <adam@ajmadsen.com>");

@ -0,0 +1,25 @@
/*
Vendor Reset - Vendor Specific Reset
Copyright (C) 2020 Geoffrey McRae <geoff@hostfission.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
*/
#ifndef _H_VENDOR_RESET_IOCTL
#define _H_VENDOR_RESET_IOCTL
int vendor_reset_ioctl_init(void);
void vendor_reset_ioctl_exit(void);
#endif

@ -0,0 +1,68 @@
/*
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/module.h>
#include "ioctl.h"
#include "ftrace.h"
#include "hooks.h"
#define vr_info(fmt, args...) pr_info("vendor-reset: " fmt, ##args)
static bool install_hook = true;
static int __init vendor_reset_init(void)
{
int ret;
ret = vendor_reset_ioctl_init();
if (ret)
return ret;
if (install_hook)
{
ret = fh_install_hooks(fh_hooks);
if (ret)
goto err;
vr_info("hooks installed\n");
}
return 0;
err:
vendor_reset_ioctl_exit();
return ret;
}
static void __exit vendor_reset_exit(void)
{
if (install_hook)
fh_remove_hooks(fh_hooks);
vendor_reset_ioctl_exit();
}
module_init(vendor_reset_init);
module_exit(vendor_reset_exit);
module_param(install_hook, bool, 0);
MODULE_LICENSE("GPL v2");
MODULE_AUTHOR("Geoffrey McRae <geoff@hostfission.com>");
MODULE_AUTHOR("Adam Madsen <adam@ajmadsen.com>");
Loading…
Cancel
Save