input: keyreset: Allow reset function to be overriden

Change-Id: Ibb3dda05772b2e89d7b2646689944d309cb1f74e
Signed-off-by: Colin Cross <ccross@android.com>
diff --git a/drivers/input/keyreset.c b/drivers/input/keyreset.c
index b513506..a5091ac 100644
--- a/drivers/input/keyreset.c
+++ b/drivers/input/keyreset.c
@@ -33,6 +33,7 @@
 	int key_down;
 	int key_up;
 	int restart_disabled;
+	int (*reset_fn)(void);
 };
 
 int restart_requested;
@@ -88,8 +89,12 @@
 		if (restart_requested)
 			panic("keyboard reset failed, %d", restart_requested);
 		pr_info("keyboard reset\n");
-		schedule_work(&restart_work);
-		restart_requested = 1;
+		if (state->reset_fn) {
+			restart_requested = state->reset_fn();
+		} else {
+			schedule_work(&restart_work);
+			restart_requested = 1;
+		}
 	}
 done:
 	spin_unlock_irqrestore(&state->lock, flags);
@@ -187,6 +192,10 @@
 			__set_bit(key, state->upbit);
 		}
 	}
+
+	if (pdata->reset_fn)
+		state->reset_fn = pdata->reset_fn;
+
 	state->input_handler.event = keyreset_event;
 	state->input_handler.connect = keyreset_connect;
 	state->input_handler.disconnect = keyreset_disconnect;