diff options
Diffstat (limited to 'trayfreq_set.c')
| -rw-r--r-- | trayfreq_set.c | 160 |
1 files changed, 160 insertions, 0 deletions
diff --git a/trayfreq_set.c b/trayfreq_set.c new file mode 100644 index 0000000..02b0eef --- /dev/null +++ b/trayfreq_set.c @@ -0,0 +1,160 @@ +/************************************************************************ + * This file is part of trayfreq-archlinux. * + * * + * trayfreq-archlinux 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. * + * * + * trayfreq-archlinux 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 trayfreq-archlinux. If not, see * + * <http://www.gnu.org/licenses/>. * + ************************************************************************/ + + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> +#include <libintl.h> +#include <locale.h> + +#include "getfreq.h" +#include "getcore.h" +#include "common.h" + +#define FILE_PATH_STRING_SIZE 100 + +#define ARG_CORE 0x1 +#define ARG_GOV 0x2 +#define ARG_FREQ 0x4 +typedef struct { + char present; + char *core; + char *governor; + char *frequency; +} argument_summary; + +char write_str_to_file(const char *file, const char *data, const char *core) +{ + FILE *fd; + char file_path[FILE_PATH_STRING_SIZE]; + + // Prepare file path + memset(file_path, '\0', sizeof(file_path)); + sprintf(file_path, "/sys/devices/system/cpu/cpu%d/cpufreq/%s", atoi(core), file ); + + // Try to open file and write data to it + if ( (fd = fopen(file_path, "w")) != NULL ) + { + debug("Writing '%s' to '%s'\n",data,file_path); + fprintf(fd, data); + fclose(fd); + return 1; + } + + // Fallthrough: File couldn't be opened for writing + fprintf(stderr, _("FAILED: Couldn't open %s for writing\n") , file_path); + return 0; +} + + +#define set_freq_max(freq,core) write_str_to_file("scaling_max_freq",freq,core) +#define set_freq_min(freq,core) write_str_to_file("scaling_min_freq",freq,core) +#define set_speed(freq,core) write_str_to_file("scaling_setspeed",freq,core) +#define set_gov(gov,core) write_str_to_file("scaling_governor",gov,core) + +void get_argument_summary(int argc, char **argv, argument_summary *argsum) +{ + int arg; + + // Check for -{c,g,f} xyz + for ( arg = 1; arg < argc-1; arg+=2 ) + { + // Can't have empty argument + if ( strlen(argv[arg+1]) != 0 ) + { + if ( strcmp(argv[arg], "-c") == 0 ) + { + // Found -c with an arg + debug("Found -c with arg '%s'\n",argv[arg+1]); + argsum->present |= ARG_CORE; + argsum->core = (char*)(argv[arg+1]); + continue; + } + + if ( strcmp(argv[arg], "-f") == 0 ) + { + // Found -f with an arg + debug("Found -f with arg '%s'\n",argv[arg+1]); + argsum->present |= ARG_FREQ; + argsum->frequency = (char*)(argv[arg+1]); + continue; + } + + if ( strcmp(argv[arg], "-g") == 0 ) + { + // Found -g with an arg + debug("Found -g with arg '%s'\n",argv[arg+1]); + argsum->present |= ARG_GOV; + argsum->governor = (char*)(argv[arg+1]); + //continue; + } + } + } +} + +int main(int argc, char *argv[]) +{ + setlocale(LC_ALL,""); + bindtextdomain("trayfreq",LOCALEDIR); + textdomain("trayfreq"); + + debug("Set gettext up\n"); + + gc_init(); + gf_init(); + + argument_summary args; + memset(&args, 0, sizeof(args)); + + // If unusual number of args, give up now + if (argc == 5) + { + + debug ("Checking UID\n"); + if (getuid() != 0) + fprintf(stderr,"Warning: running as UID %d, not 0\n",getuid() ); + + get_argument_summary(argc, argv, &args); + + debug("Correct number of command line arguments\n"); + debug("-c: %s -g: %s -f: %s\n", (args.present | ARG_CORE )? "Yes":"No", + (args.present | ARG_GOV )? "Yes":"No", + (args.present | ARG_FREQ )? "Yes":"No" ); + debug("Core: %s\n",args.core); + debug("Gov : %s\n",args.governor); + debug("Freq: %s\n",args.frequency); + + if ( args.present == ( ARG_CORE | ARG_GOV ) ) + { + debug("Changing governor\n"); + return set_gov(args.governor , args.core); + } + + if ( args.present == ( ARG_CORE | ARG_FREQ ) ) + { + debug("Changing frequency\n"); + return set_gov("userspace", args.core) | set_speed(args.frequency, args.core); + } + } + // Fall through to here if no valid argument combination + debug("Fell through, showing command usage\n"); + fprintf(stderr, _("%s {-f frequency|-g governor} -c core\n"), argv[0] ); + return 1; +} |