forked from rrcarlosr/Jetpack
65 lines
1.9 KiB
C
65 lines
1.9 KiB
C
/*
|
|
* Copyright (c) 2015-2016, NVIDIA CORPORATION. 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 version 2 as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* 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.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#include <linux/bitops.h>
|
|
#include <linux/psci.h>
|
|
|
|
#include <asm/io.h>
|
|
|
|
#include "iomap.h"
|
|
#include <soc/tegra/pmc.h>
|
|
|
|
static void program_reboot_reason(const char *cmd)
|
|
{
|
|
u32 reboot_reason;
|
|
|
|
/* clean up */
|
|
reboot_reason = (BOOTLOADER_MODE | RECOVERY_MODE | FORCED_RECOVERY_MODE
|
|
| UPDATE_MODE);
|
|
tegra_pmc_clear_reboot_reason(reboot_reason);
|
|
|
|
/* valid command? */
|
|
if (!cmd || (strlen(cmd) == 0))
|
|
return;
|
|
|
|
/* set reset_reboot_reason to 0, so that only 'cmd' can be set
|
|
* to register.
|
|
*/
|
|
reboot_reason = 0;
|
|
/* Writing recovery kernel or Bootloader mode or update kernel in
|
|
* SCRATCH0 31:30:29:1. */
|
|
if (!strcmp(cmd, "recovery"))
|
|
reboot_reason |= RECOVERY_MODE;
|
|
else if (!strcmp(cmd, "bootloader"))
|
|
reboot_reason |= BOOTLOADER_MODE;
|
|
else if (!strcmp(cmd, "forced-recovery"))
|
|
reboot_reason |= FORCED_RECOVERY_MODE;
|
|
else if (!strcmp(cmd, "update"))
|
|
reboot_reason |= UPDATE_MODE;
|
|
|
|
/* write the restart command */
|
|
tegra_pmc_set_reboot_reason(reboot_reason);
|
|
}
|
|
|
|
static __init int tegra_register_reboot_handler(void)
|
|
{
|
|
psci_handle_reboot_cmd = program_reboot_reason;
|
|
pr_info("Tegra reboot handler registered.\n");
|
|
return 0;
|
|
}
|
|
arch_initcall(tegra_register_reboot_handler);
|