}
 
 static struct pm_ops sharpsl_pm_ops = {
-       .pm_disk_mode   = PM_DISK_FIRMWARE,
        .prepare        = pxa_pm_prepare,
        .enter          = corgi_pxa_pm_enter,
        .finish         = pxa_pm_finish,
 
 
 
 static struct pm_ops at91_pm_ops ={
-       .pm_disk_mode   = 0,
        .valid          = at91_pm_valid_state,
        .prepare        = at91_pm_prepare,
        .enter          = at91_pm_enter,
 
 
 
 static struct pm_ops omap_pm_ops ={
-       .pm_disk_mode   = 0,
        .prepare        = omap_pm_prepare,
        .enter          = omap_pm_enter,
        .finish         = omap_pm_finish,
 
 }
 
 static struct pm_ops omap_pm_ops = {
-       .pm_disk_mode   = 0,
        .prepare        = omap2_pm_prepare,
        .enter          = omap2_pm_enter,
        .finish         = omap2_pm_finish,
 
 
 EXPORT_SYMBOL_GPL(pxa_pm_finish);
 
-/*
- * Set to PM_DISK_FIRMWARE so we can quickly veto suspend-to-disk.
- */
 static struct pm_ops pxa_pm_ops = {
-       .pm_disk_mode   = PM_DISK_FIRMWARE,
        .prepare        = pxa_pm_prepare,
        .enter          = pxa_pm_enter,
        .finish         = pxa_pm_finish,
 
        unsigned long gpio, sleep_save[SLEEP_SAVE_SIZE];
        struct timespec delta, rtc;
 
-       if (state != PM_SUSPEND_MEM)
-               return -EINVAL;
-
        /* preserve current time */
        rtc.tv_sec = RCNR;
        rtc.tv_nsec = 0;
        return virt_to_phys(sp);
 }
 
-/*
- * Set to PM_DISK_FIRMWARE so we can quickly veto suspend-to-disk.
- */
 static struct pm_ops sa11x0_pm_ops = {
-       .pm_disk_mode   = PM_DISK_FIRMWARE,
        .enter          = sa11x0_pm_enter,
 };
 
 
                return -EINVAL;
        }
 
-       if (state != PM_SUSPEND_MEM) {
-               printk(KERN_ERR PFX "error: only PM_SUSPEND_MEM supported\n");
-               return -EINVAL;
-       }
-
        /* check if we have anything to wake-up with... bad things seem
         * to happen if you suspend with no wakeup (system will often
         * require a full power-cycle)
        return 0;
 }
 
-/*
- * Set to PM_DISK_FIRMWARE so we can quickly veto suspend-to-disk.
- */
 static struct pm_ops s3c2410_pm_ops = {
-       .pm_disk_mode   = PM_DISK_FIRMWARE,
        .prepare        = s3c2410_pm_prepare,
        .enter          = s3c2410_pm_enter,
        .finish         = s3c2410_pm_finish,
 
        u16 hd64461_stbcr;
 #endif
 
-       if (state != PM_SUSPEND_MEM)
-               return -EINVAL;
-
 #ifdef CONFIG_HD64461_ENABLER
        outb(0, HD64461_PCC1CSCIER);
 
        return 0;
 }
 
-/*
- * Set to PM_DISK_FIRMWARE so we can quickly veto suspend-to-disk.
- */
 static struct pm_ops hp6x0_pm_ops = {
-       .pm_disk_mode   = PM_DISK_FIRMWARE,
        .enter          = hp6x0_pm_enter,
 };
 
 
 
 typedef int __bitwise suspend_disk_method_t;
 
+/* invalid must be 0 so struct pm_ops initialisers can leave it out */
+#define PM_DISK_INVALID                ((__force suspend_disk_method_t) 0)
 #define        PM_DISK_FIRMWARE        ((__force suspend_disk_method_t) 1)
 #define        PM_DISK_PLATFORM        ((__force suspend_disk_method_t) 2)
 #define        PM_DISK_SHUTDOWN        ((__force suspend_disk_method_t) 3)
  * @finish: Called when the system has left the given state and all devices
  *     are resumed. The return value is ignored.
  *
- * @pm_disk_mode: Set to the disk method that the user should be able to
- *     configure for suspend-to-disk. Since %PM_DISK_SHUTDOWN,
- *     %PM_DISK_REBOOT, %PM_DISK_TEST and %PM_DISK_TESTPROC
- *     are always allowed, currently only %PM_DISK_PLATFORM
- *     makes sense. If the user then choses %PM_DISK_PLATFORM,
- *     the @prepare call will be called before suspending to disk
- *     (if present), the @enter call should be present and will
- *     be called after all state has been saved and the machine
- *     is ready to be shut down/suspended/..., and the @finish
- *     callback is called after state has been restored. All
- *     these calls are called with %PM_SUSPEND_DISK as the state.
+ * @pm_disk_mode: The generic code always allows one of the shutdown methods
+ *     %PM_DISK_SHUTDOWN, %PM_DISK_REBOOT, %PM_DISK_TEST and
+ *     %PM_DISK_TESTPROC. If this variable is set, the mode it is set
+ *     to is allowed in addition to those modes and is also made default.
+ *     When this mode is sent selected, the @prepare call will be called
+ *     before suspending to disk (if present), the @enter call should be
+ *     present and will be called after all state has been saved and the
+ *     machine is ready to be powered off; the @finish callback is called
+ *     after state has been restored. All these calls are called with
+ *     %PM_SUSPEND_DISK as the state.
  */
 struct pm_ops {
        int (*valid)(suspend_state_t state);
 
 {
        int error = 0;
 
-       if (pm_disk_mode == PM_DISK_PLATFORM) {
+       switch (pm_disk_mode) {
+       case PM_DISK_TEST:
+       case PM_DISK_TESTPROC:
+       case PM_DISK_SHUTDOWN:
+       case PM_DISK_REBOOT:
+               break;
+       default:
                if (pm_ops && pm_ops->prepare)
                        error = pm_ops->prepare(PM_SUSPEND_DISK);
        }
 
 /**
  *     power_down - Shut machine down for hibernate.
- *     @mode:          Suspend-to-disk mode
  *
- *     Use the platform driver, if configured so, and return gracefully if it
- *     fails.
- *     Otherwise, try to power off and reboot. If they fail, halt the machine,
- *     there ain't no turning back.
+ *     Use the platform driver, if configured so; otherwise try
+ *     to power off or reboot.
  */
 
-static void power_down(suspend_disk_method_t mode)
+static void power_down(void)
 {
-       switch(mode) {
-       case PM_DISK_PLATFORM:
-               if (pm_ops && pm_ops->enter) {
-                       kernel_shutdown_prepare(SYSTEM_SUSPEND_DISK);
-                       pm_ops->enter(PM_SUSPEND_DISK);
-                       break;
-               }
+       switch (pm_disk_mode) {
+       case PM_DISK_TEST:
+       case PM_DISK_TESTPROC:
+               break;
        case PM_DISK_SHUTDOWN:
                kernel_power_off();
                break;
        case PM_DISK_REBOOT:
                kernel_restart(NULL);
                break;
+       default:
+               if (pm_ops && pm_ops->enter) {
+                       kernel_shutdown_prepare(SYSTEM_SUSPEND_DISK);
+                       pm_ops->enter(PM_SUSPEND_DISK);
+                       break;
+               }
        }
        kernel_halt();
-       /* Valid image is on the disk, if we continue we risk serious data corruption
-          after resume. */
+       /*
+        * Valid image is on the disk, if we continue we risk serious data
+        * corruption after resume.
+        */
        printk(KERN_CRIT "Please power me down manually\n");
        while(1);
 }
 
 static inline void platform_finish(void)
 {
-       if (pm_disk_mode == PM_DISK_PLATFORM) {
+       switch (pm_disk_mode) {
+       case PM_DISK_TEST:
+       case PM_DISK_TESTPROC:
+       case PM_DISK_SHUTDOWN:
+       case PM_DISK_REBOOT:
+               break;
+       default:
                if (pm_ops && pm_ops->finish)
                        pm_ops->finish(PM_SUSPEND_DISK);
        }
                pr_debug("PM: writing image.\n");
                error = swsusp_write();
                if (!error)
-                       power_down(pm_disk_mode);
+                       power_down();
                else {
                        swsusp_free();
                        goto Thaw;
                }
        }
        if (mode) {
-               if (mode == PM_DISK_SHUTDOWN || mode == PM_DISK_REBOOT ||
-                    mode == PM_DISK_TEST || mode == PM_DISK_TESTPROC) {
+               switch (mode) {
+               case PM_DISK_SHUTDOWN:
+               case PM_DISK_REBOOT:
+               case PM_DISK_TEST:
+               case PM_DISK_TESTPROC:
                        pm_disk_mode = mode;
-               } else {
+                       break;
+               default:
                        if (pm_ops && pm_ops->enter &&
                            (mode == pm_ops->pm_disk_mode))
                                pm_disk_mode = mode;
 
 DEFINE_MUTEX(pm_mutex);
 
 struct pm_ops *pm_ops;
-suspend_disk_method_t pm_disk_mode = PM_DISK_PLATFORM;
+suspend_disk_method_t pm_disk_mode = PM_DISK_SHUTDOWN;
 
 /**
  *     pm_set_ops - Set the global power method table. 
 {
        mutex_lock(&pm_mutex);
        pm_ops = ops;
+       if (ops && ops->pm_disk_mode != PM_DISK_INVALID) {
+               pm_disk_mode = ops->pm_disk_mode;
+       } else
+               pm_disk_mode = PM_DISK_SHUTDOWN;
        mutex_unlock(&pm_mutex);
 }