add hdparm

Signed-off-by: Mattias Andrée <maandree@operamail.com>

1 files changed, 66 insertions, 0 deletions

diff --git a/src/restricted-hdparm.c b/src/restricted-hdparm.c
new file mode 100644
index 0000000..bb4d923
--- /dev/null
+++ b/src/restricted-hdparm.c
@@ -0,0 +1,66 @@
+/**
+ * xpybar – xmobar replacement written in python
+ * Copyright © 2014  Mattias Andrée (maandree@member.fsf.org)
+ * 
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * 
+ * 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, see <http://www.gnu.org/licenses/>.
+ */
+#include <unistd.h>
+#include <stdio.h>
+#include <string.h>
+
+
+int main(int argc, char** argv)
+{
+  char* act = argv[1];
+  int i;
+  
+  setuid(0);
+  
+  for (i = 2; i < argc; i++)
+    if (argv[i][0] == '/')
+      return 1;
+  
+  if (*act++ != '-')
+    return 1;
+  
+  if      (!strcmp(act, "a"))  ;
+  else if (!strcmp(act, "A"))  ;
+  else if ( strstr(act, "B") == act)  ;
+  else if (!strcmp(act, "c"))  ;
+  else if (!strcmp(act, "C"))  ;
+  else if (!strcmp(act, "g"))  ;
+  else if (!strcmp(act, "H"))  ;
+  else if (!strcmp(act, "i"))  ;
+  else if (!strcmp(act, "I"))  ;
+  else if ( strstr(act, "J") == act)  ;
+  else if (!strcmp(act, "m"))  ;
+  else if ( strstr(act, "M") == act)  ;
+  else if (!strcmp(act, "N"))  ;
+  else if (!strcmp(act, "Q"))  ;
+  else if (!strcmp(act, "r"))  ;
+  else if (!strcmp(act, "R"))  ;
+  else if ( strstr(act, "S") == act)  ;
+  else if (!strcmp(act, "W"))  ;
+  else if (!strcmp(act, "y"))  ;
+  else if (!strcmp(act, "Y"))  ;
+  else if (!strcmp(act, "Z"))  ;
+  else
+    return 1;
+  
+  execv("hdparm", argv);
+  perror(*argv);
+  
+  return 1;
+}
+