cpupower: bench: parse.c: fix several resource leaks
[cascardo/linux.git] / tools / power / cpupower / bench / parse.c
1 /*  cpufreq-bench CPUFreq microbenchmark
2  *
3  *  Copyright (C) 2008 Christian Kornacker <ckornacker@suse.de>
4  *
5  *  This program is free software; you can redistribute it and/or modify
6  *  it under the terms of the GNU General Public License as published by
7  *  the Free Software Foundation; either version 2 of the License, or
8  *  (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU General Public License for more details.
14  *
15  *  You should have received a copy of the GNU General Public License
16  *  along with this program; if not, write to the Free Software
17  *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
18  */
19
20 #include <stdio.h>
21 #include <stdlib.h>
22 #include <stdarg.h>
23 #include <string.h>
24 #include <time.h>
25 #include <dirent.h>
26
27 #include <sys/utsname.h>
28 #include <sys/types.h>
29 #include <sys/stat.h>
30
31 #include "parse.h"
32 #include "config.h"
33
34 /**
35  * converts priority string to priority
36  *
37  * @param str string that represents a scheduler priority
38  *
39  * @retval priority
40  * @retval SCHED_ERR when the priority doesn't exit
41  **/
42
43 enum sched_prio string_to_prio(const char *str)
44 {
45         if (strncasecmp("high", str, strlen(str)) == 0)
46                 return  SCHED_HIGH;
47         else if (strncasecmp("default", str, strlen(str)) == 0)
48                 return SCHED_DEFAULT;
49         else if (strncasecmp("low", str, strlen(str)) == 0)
50                 return SCHED_LOW;
51         else
52                 return SCHED_ERR;
53 }
54
55 /**
56  * create and open logfile
57  *
58  * @param dir directory in which the logfile should be created
59  *
60  * @retval logfile on success
61  * @retval NULL when the file can't be created
62  **/
63
64 FILE *prepare_output(const char *dirname)
65 {
66         FILE *output = NULL;
67         int len;
68         char *filename;
69         struct utsname sysdata;
70         DIR *dir;
71
72         dir = opendir(dirname);
73         if (dir == NULL) {
74                 if (mkdir(dirname, 0755)) {
75                         perror("mkdir");
76                         fprintf(stderr, "error: Cannot create dir %s\n",
77                                 dirname);
78                         return NULL;
79                 }
80         }
81
82         len = strlen(dirname) + 30;
83         filename = malloc(sizeof(char) * len);
84         if (!filename) {
85                 perror("malloc");
86                 goto out_dir;
87         }
88
89         if (uname(&sysdata) == 0) {
90                 len += strlen(sysdata.nodename) + strlen(sysdata.release);
91                 filename = realloc(filename, sizeof(char) * len);
92
93                 if (!filename) {
94                         perror("realloc");
95                         goto out_dir;
96                 }
97
98                 snprintf(filename, len - 1, "%s/benchmark_%s_%s_%li.log",
99                         dirname, sysdata.nodename, sysdata.release, time(NULL));
100         } else {
101                 snprintf(filename, len - 1, "%s/benchmark_%li.log",
102                         dirname, time(NULL));
103         }
104
105         dprintf("logilename: %s\n", filename);
106
107         output = fopen(filename, "w+");
108         if (output == NULL) {
109                 perror("fopen");
110                 fprintf(stderr, "error: unable to open logfile\n");
111                 goto out;
112         }
113
114         fprintf(stdout, "Logfile: %s\n", filename);
115
116         fprintf(output, "#round load sleep performance powersave percentage\n");
117 out:
118         free(filename);
119 out_dir:
120         closedir(dir);
121         return output;
122 }
123
124 /**
125  * returns the default config
126  *
127  * @retval default config on success
128  * @retval NULL when the output file can't be created
129  **/
130
131 struct config *prepare_default_config()
132 {
133         struct config *config = malloc(sizeof(struct config));
134
135         dprintf("loading defaults\n");
136
137         config->sleep = 500000;
138         config->load = 500000;
139         config->sleep_step = 500000;
140         config->load_step = 500000;
141         config->cycles = 5;
142         config->rounds = 50;
143         config->cpu = 0;
144         config->prio = SCHED_HIGH;
145         config->verbose = 0;
146         strncpy(config->governor, "ondemand", 8);
147
148         config->output = stdout;
149
150 #ifdef DEFAULT_CONFIG_FILE
151         if (prepare_config(DEFAULT_CONFIG_FILE, config))
152                 return NULL;
153 #endif
154         return config;
155 }
156
157 /**
158  * parses config file and returns the config to the caller
159  *
160  * @param path config file name
161  *
162  * @retval 1 on error
163  * @retval 0 on success
164  **/
165
166 int prepare_config(const char *path, struct config *config)
167 {
168         size_t len = 0;
169         char opt[16], val[32], *line = NULL;
170         FILE *configfile;
171
172         if (config == NULL) {
173                 fprintf(stderr, "error: config is NULL\n");
174                 return 1;
175         }
176
177         configfile = fopen(path, "r");
178         if (configfile == NULL) {
179                 perror("fopen");
180                 fprintf(stderr, "error: unable to read configfile\n");
181                 free(config);
182                 return 1;
183         }
184
185         while (getline(&line, &len, configfile) != -1) {
186                 if (line[0] == '#' || line[0] == ' ' || line[0] == '\n')
187                         continue;
188
189                 if (sscanf(line, "%14s = %30s", opt, val) < 2)
190                         continue;
191
192                 dprintf("parsing: %s -> %s\n", opt, val);
193
194                 if (strcmp("sleep", opt) == 0)
195                         sscanf(val, "%li", &config->sleep);
196
197                 else if (strcmp("load", opt) == 0)
198                         sscanf(val, "%li", &config->load);
199
200                 else if (strcmp("load_step", opt) == 0)
201                         sscanf(val, "%li", &config->load_step);
202
203                 else if (strcmp("sleep_step", opt) == 0)
204                         sscanf(val, "%li", &config->sleep_step);
205
206                 else if (strcmp("cycles", opt) == 0)
207                         sscanf(val, "%u", &config->cycles);
208
209                 else if (strcmp("rounds", opt) == 0)
210                         sscanf(val, "%u", &config->rounds);
211
212                 else if (strcmp("verbose", opt) == 0)
213                         sscanf(val, "%u", &config->verbose);
214
215                 else if (strcmp("output", opt) == 0)
216                         config->output = prepare_output(val); 
217
218                 else if (strcmp("cpu", opt) == 0)
219                         sscanf(val, "%u", &config->cpu);
220
221                 else if (strcmp("governor", opt) == 0) {
222                         strncpy(config->governor, val,
223                                         sizeof(config->governor));
224                         config->governor[sizeof(config->governor) - 1] = '\0';
225                 }
226
227                 else if (strcmp("priority", opt) == 0) {
228                         if (string_to_prio(val) != SCHED_ERR)
229                                 config->prio = string_to_prio(val);
230                 }
231         }
232
233         free(line);
234
235         return 0;
236 }