msm: restart: bring back Sony's changes

Fix issues with power down and rebooting