---

yaml --- r: 312729 b: refs/heads/master c: 11388c8 h: refs/heads/master i: 312727: eab5610 v: v3
git-mirror · Jul 18, 2012 · 24e606c · 24e606c
1 parent 7abf142
commit 24e606c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/[refs] b/[refs]
@@ -1,2 +1,2 @@
 ---
-refs/heads/master: 823d93640927d739f7c918b73fe5ec494fb27e17
+refs/heads/master: 11388c87d2abca1f01975ced28ce9eacea239104
diff --git a/trunk/kernel/power/wakelock.c b/trunk/kernel/power/wakelock.c
@@ -9,6 +9,7 @@
  * manipulate wakelocks on Android.
  */
 
+#include <linux/capability.h>
 #include <linux/ctype.h>
 #include <linux/device.h>
 #include <linux/err.h>
@@ -188,6 +189,9 @@ int pm_wake_lock(const char *buf)
 	size_t len;
 	int ret = 0;
 
+	if (!capable(CAP_BLOCK_SUSPEND))
+		return -EPERM;
+
 	while (*str && !isspace(*str))
 		str++;
 
@@ -231,6 +235,9 @@ int pm_wake_unlock(const char *buf)
 	size_t len;
 	int ret = 0;
 
+	if (!capable(CAP_BLOCK_SUSPEND))
+		return -EPERM;
+
 	len = strlen(buf);
 	if (!len)
 		return -EINVAL;