Settings.c 17.7 KB
Newer Older
Hisham Muhammad's avatar
Hisham Muhammad committed
1
2
/*
htop - Settings.c
Hisham Muhammad's avatar
Hisham Muhammad committed
3
(C) 2004-2011 Hisham H. Muhammad
Hisham Muhammad's avatar
Hisham Muhammad committed
4
5
6
7
8
Released under the GNU GPL, see the COPYING file
in the source distribution for its full text.
*/

#include "Settings.h"
9
#include "Platform.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
10

David Hunt's avatar
David Hunt committed
11
#include "StringUtils.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
12
#include "Vector.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
13
#include "CRT.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
14

Hisham Muhammad's avatar
Hisham Muhammad committed
15
16
17
18
19
#include <sys/stat.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

Hisham Muhammad's avatar
Hisham Muhammad committed
20
21
22
#define DEFAULT_DELAY 15

/*{
Hisham Muhammad's avatar
Hisham Muhammad committed
23
#include "Process.h"
Hisham Muhammad's avatar
Hisham Muhammad committed
24
#include <stdbool.h>
Hisham Muhammad's avatar
Hisham Muhammad committed
25

Hisham Muhammad's avatar
Hisham Muhammad committed
26
27
28
29
30
typedef struct {
   int len;
   char** names;
   int* modes;
} MeterColumnSettings;
31

32
33
34
35
36
37
38
39
40
typedef struct {
   char* name;
   ProcessField* fields;
   int flags;
   int direction;
   ProcessField sortKey;
   bool treeView;
} ScreenSettings;

Hisham Muhammad's avatar
Hisham Muhammad committed
41
typedef struct Settings_ {
Hisham Muhammad's avatar
Hisham Muhammad committed
42
43
   char* filename;
   
44
   MeterColumnSettings meterColumns[2];
Hisham Muhammad's avatar
Hisham Muhammad committed
45

46
47
48
49
   ScreenSettings** screens;
   unsigned int nScreens;
   unsigned int ssIndex;
   ScreenSettings* ss;
50

Hisham Muhammad's avatar
Hisham Muhammad committed
51
52
   int colorScheme;
   int delay;
Hisham Muhammad's avatar
Hisham Muhammad committed
53

54
   int cpuCount;
Hisham Muhammad's avatar
Hisham Muhammad committed
55
56
57

   bool countCPUsFromZero;
   bool detailedCPUTime;
58
   bool showProgramPath;
Hisham Muhammad's avatar
Hisham Muhammad committed
59
60
61
62
63
64
65
66
67
68
69
70
   bool hideThreads;
   bool shadowOtherUsers;
   bool showThreadNames;
   bool hideKernelThreads;
   bool hideUserlandThreads;
   bool highlightBaseName;
   bool highlightMegabytes;
   bool highlightThreads;
   bool updateProcessNames;
   bool accountGuestInCPUMeter;
   bool headerMargin;

Hisham Muhammad's avatar
Hisham Muhammad committed
71
   bool changed;
Hisham Muhammad's avatar
Hisham Muhammad committed
72
73
} Settings;

Hisham Muhammad's avatar
Hisham Muhammad committed
74
75
76
77
#ifndef Settings_cpuId
#define Settings_cpuId(settings, cpu) ((settings)->countCPUsFromZero ? (cpu) : (cpu)+1)
#endif

Hisham Muhammad's avatar
Hisham Muhammad committed
78
79
}*/

80
81
82
83
84
85
86
87
88
static void writeList(FILE* fd, char** list, int len) {
   const char* sep = "";
   for (int i = 0; i < len; i++) {
      fprintf(fd, "%s%s", sep, list[i]);
      sep = " ";
   }
   fprintf(fd, "\n");
}

89
90
91
92
/*

static char** readQuotedList(char* line) {
   int n = 0;
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
   char** list = xCalloc(sizeof(char*), 1);
   int start = 0;
   for (;;) {
      while (line[start] && line[start] == ' ') {
         start++;
      }
      if (line[start] != '"') {
         break;
      }
      start++;
      int close = start;
      while (line[close] && line[close] != '"') {
         close++;
      }
      int len = close - start;
      char* item = xMalloc(len + 1);
      strncpy(item, line + start, len);
      item[len] = '\0';
111
112
113
      list[n] = item;
      n++;
      list = xRealloc(list, sizeof(char*) * (n + 1));
114
115
      start = close + 1;
   }
116
   list[n] = NULL;
117
118
119
   return list;
}

120
static void writeQuotedList(FILE* fd, char** list) {
121
   const char* sep = "";
122
   for (int i = 0; list[i]; i++) {
123
124
125
126
127
128
      fprintf(fd, "%s\"%s\"", sep, list[i]);
      sep = " ";
   }
   fprintf(fd, "\n");
}

129
130
*/

Hisham Muhammad's avatar
Hisham Muhammad committed
131
void Settings_delete(Settings* this) {
Hisham Muhammad's avatar
Hisham Muhammad committed
132
   free(this->filename);
133
134
135
136
137
138
139
140
141
142
   for (unsigned int i = 0; i < (sizeof(this->meterColumns)/sizeof(MeterColumnSettings)); i++) {
      String_freeArray(this->meterColumns[i].names);
      free(this->meterColumns[i].modes);
   }
   if (this->screens) {
      for (unsigned int i = 0; this->screens[i]; i++) {
         free(this->screens[i]->name);
         free(this->screens[i]->fields);
      }
      free(this->screens);
Hisham Muhammad's avatar
Hisham Muhammad committed
143
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
144
145
146
   free(this);
}

147
static void Settings_readMeters(Settings* this, char* line, int side) {
Hisham Muhammad's avatar
Hisham Muhammad committed
148
   char* trim = String_trim(line);
149
150
   int nIds;
   char** ids = String_split(trim, ' ', &nIds);
Hisham Muhammad's avatar
Hisham Muhammad committed
151
   free(trim);
152
   this->meterColumns[side].names = ids;
Hisham Muhammad's avatar
Hisham Muhammad committed
153
154
}

155
static void Settings_readMeterModes(Settings* this, char* line, int side) {
Hisham Muhammad's avatar
Hisham Muhammad committed
156
   char* trim = String_trim(line);
157
158
   int nIds;
   char** ids = String_split(trim, ' ', &nIds);
Hisham Muhammad's avatar
Hisham Muhammad committed
159
   free(trim);
Hisham Muhammad's avatar
Hisham Muhammad committed
160
   int len = 0;
161
   for (int i = 0; ids[i]; i++) {
Hisham Muhammad's avatar
Hisham Muhammad committed
162
163
      len++;
   }
164
   this->meterColumns[side].len = len;
Hisham's avatar
Hisham committed
165
   int* modes = xCalloc(len, sizeof(int));
Hisham Muhammad's avatar
Hisham Muhammad committed
166
167
   for (int i = 0; i < len; i++) {
      modes[i] = atoi(ids[i]);
Hisham Muhammad's avatar
Hisham Muhammad committed
168
169
   }
   String_freeArray(ids);
170
   this->meterColumns[side].modes = modes;
Hisham Muhammad's avatar
Hisham Muhammad committed
171
172
}

173
static void Settings_defaultMeters(Settings* this) {
Hisham Muhammad's avatar
Hisham Muhammad committed
174
   int sizes[] = { 3, 3 };
175
   if (this->cpuCount > 4) {
Hisham Muhammad's avatar
Hisham Muhammad committed
176
177
178
      sizes[1]++;
   }
   for (int i = 0; i < 2; i++) {
179
180
181
      this->meterColumns[i].names = xCalloc(sizes[i] + 1, sizeof(char*));
      this->meterColumns[i].modes = xCalloc(sizes[i], sizeof(int));
      this->meterColumns[i].len = sizes[i];
Hisham Muhammad's avatar
Hisham Muhammad committed
182
183
184
   }
   
   int r = 0;
185
   if (this->cpuCount > 8) {
186
187
188
189
      this->meterColumns[0].names[0] = xStrdup("LeftCPUs2");
      this->meterColumns[0].modes[0] = BAR_METERMODE;
      this->meterColumns[1].names[r] = xStrdup("RightCPUs2");
      this->meterColumns[1].modes[r++] = BAR_METERMODE;
190
   } else if (this->cpuCount > 4) {
191
192
193
194
      this->meterColumns[0].names[0] = xStrdup("LeftCPUs");
      this->meterColumns[0].modes[0] = BAR_METERMODE;
      this->meterColumns[1].names[r] = xStrdup("RightCPUs");
      this->meterColumns[1].modes[r++] = BAR_METERMODE;
195
   } else {
196
197
      this->meterColumns[0].names[0] = xStrdup("AllCPUs");
      this->meterColumns[0].modes[0] = BAR_METERMODE;
Hisham Muhammad's avatar
Hisham Muhammad committed
198
   }
199
200
201
202
   this->meterColumns[0].names[1] = xStrdup("Memory");
   this->meterColumns[0].modes[1] = BAR_METERMODE;
   this->meterColumns[0].names[2] = xStrdup("Swap");
   this->meterColumns[0].modes[2] = BAR_METERMODE;
Hisham Muhammad's avatar
Hisham Muhammad committed
203
   
204
205
206
207
208
209
   this->meterColumns[1].names[r] = xStrdup("Tasks");
   this->meterColumns[1].modes[r++] = TEXT_METERMODE;
   this->meterColumns[1].names[r] = xStrdup("LoadAverage");
   this->meterColumns[1].modes[r++] = TEXT_METERMODE;
   this->meterColumns[1].names[r] = xStrdup("Uptime");
   this->meterColumns[1].modes[r++] = TEXT_METERMODE;
Hisham Muhammad's avatar
Hisham Muhammad committed
210
211
}

212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
static int toFieldIndex(const char* str) {
   if (isdigit(str[0])) {
      // This "+1" is for compatibility with the older enum format.
      int id = atoi(str) + 1;
      if (Process_fields[id].name && id < Platform_numberOfFields) {
         return id;
      }
   } else {
      for (int p = 1; p < LAST_PROCESSFIELD; p++) {
         if (Process_fields[p].name && strcmp(Process_fields[p].name, str) == 0) {
            return p;
         }
      }
   }
   return -1;
227
228
}

Hisham Muhammad's avatar
Hisham Muhammad committed
229
230
231
232
233
234
235
static void readFields(ProcessField* fields, int* flags, const char* line) {
   char* trim = String_trim(line);
   int nIds;
   char** ids = String_split(trim, ' ', &nIds);
   free(trim);
   int i, j;
   *flags = 0;
236
   for (j = 0, i = 0; i < Platform_numberOfFields && ids[i]; i++) {
237
238
239
240
      int idx = toFieldIndex(ids[i]);
      if (idx != -1) {
         fields[j] = idx;
         *flags |= Process_fields[idx].flags;
Hisham Muhammad's avatar
Hisham Muhammad committed
241
242
         j++;
      }
243
   }
Hisham's avatar
Hisham committed
244
   fields[j] = NULL_PROCESSFIELD;
Hisham Muhammad's avatar
Hisham Muhammad committed
245
   String_freeArray(ids);
246
247
}

248
ScreenSettings* Settings_newScreen(Settings* this, const char* name, const char* line) {
249
250
251
252
253
254
255
256
257
258
259
   ScreenSettings* ss = xCalloc(sizeof(ScreenSettings), 1);
   ss->name = xStrdup(name);
   ss->fields = xCalloc(Platform_numberOfFields+1, sizeof(ProcessField));
   ss->flags = 0;
   ss->direction = 1;
   ss->treeView = 0;
   readFields(ss->fields, &(ss->flags), line);
   this->screens[this->nScreens] = ss;
   this->nScreens++;
   this->screens = xRealloc(this->screens, sizeof(ScreenSettings*) * (this->nScreens + 1));
   this->screens[this->nScreens] = NULL;
260
   return ss;
261
262
263
}

static void Settings_defaultScreens(Settings* this) {
264
265
   Settings_newScreen(this, "Default", "PID USER PRIORITY NICE M_SIZE M_RESIDENT M_SHARE STATE PERCENT_CPU PERCENT_MEM TIME Command");
   Settings_newScreen(this, "I/O", "PID IO_PRIORITY USER IO_READ_RATE IO_WRITE_RATE Command");
266
267
}

268
static bool Settings_read(Settings* this, const char* fileName) {
269
   FILE* fd;
270
271
   
   CRT_dropPrivileges();
272
   fd = fopen(fileName, "r");
273
   CRT_restorePrivileges();
274
   if (!fd)
Hisham Muhammad's avatar
Hisham Muhammad committed
275
      return false;
276
   
277
278
   bool didReadMeters = false;
   bool didReadFields = false;
279
280
281
   ProcessField* legacyFields = xCalloc(Platform_numberOfFields+1, sizeof(ProcessField));
   int legacyFlags;
   bool legacyFieldsRead = false;
282
283
284
285
286
   for (;;) {
      char* line = String_readLine(fd);
      if (!line) {
         break;
      }
287
      int nOptions;
288
289
      char** option = String_split(line, '=', &nOptions);
      free (line);
290
291
292
293
      if (nOptions < 2) {
         String_freeArray(option);
         continue;
      }
Hisham Muhammad's avatar
Hisham Muhammad committed
294
      if (String_eq(option[0], "fields")) {
295
296
         readFields(legacyFields, &legacyFlags, option[1]);
         legacyFieldsRead = true;
Hisham Muhammad's avatar
Hisham Muhammad committed
297
      } else if (String_eq(option[0], "hide_threads")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
298
         this->hideThreads = atoi(option[1]);
Hisham Muhammad's avatar
Hisham Muhammad committed
299
      } else if (String_eq(option[0], "hide_kernel_threads")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
300
         this->hideKernelThreads = atoi(option[1]);
Hisham Muhammad's avatar
Hisham Muhammad committed
301
      } else if (String_eq(option[0], "hide_userland_threads")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
302
         this->hideUserlandThreads = atoi(option[1]);
Hisham Muhammad's avatar
Hisham Muhammad committed
303
      } else if (String_eq(option[0], "shadow_other_users")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
304
         this->shadowOtherUsers = atoi(option[1]);
Hisham Muhammad's avatar
Hisham Muhammad committed
305
      } else if (String_eq(option[0], "show_thread_names")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
306
         this->showThreadNames = atoi(option[1]);
307
308
      } else if (String_eq(option[0], "show_program_path")) {
         this->showProgramPath = atoi(option[1]);
Hisham Muhammad's avatar
Hisham Muhammad committed
309
      } else if (String_eq(option[0], "highlight_base_name")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
310
         this->highlightBaseName = atoi(option[1]);
Hisham Muhammad's avatar
Hisham Muhammad committed
311
      } else if (String_eq(option[0], "highlight_megabytes")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
312
         this->highlightMegabytes = atoi(option[1]);
313
      } else if (String_eq(option[0], "highlight_threads")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
314
         this->highlightThreads = atoi(option[1]);
Hisham Muhammad's avatar
Hisham Muhammad committed
315
      } else if (String_eq(option[0], "header_margin")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
316
         this->headerMargin = atoi(option[1]);
317
      } else if (String_eq(option[0], "expand_system_time")) {
318
         // Compatibility option.
Hisham Muhammad's avatar
Hisham Muhammad committed
319
         this->detailedCPUTime = atoi(option[1]);
320
      } else if (String_eq(option[0], "detailed_cpu_time")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
321
         this->detailedCPUTime = atoi(option[1]);
322
      } else if (String_eq(option[0], "cpu_count_from_zero")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
323
         this->countCPUsFromZero = atoi(option[1]);
324
      } else if (String_eq(option[0], "update_process_names")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
325
         this->updateProcessNames = atoi(option[1]);
326
      } else if (String_eq(option[0], "account_guest_in_cpu_meter")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
327
         this->accountGuestInCPUMeter = atoi(option[1]);
Hisham Muhammad's avatar
Hisham Muhammad committed
328
329
330
331
      } else if (String_eq(option[0], "delay")) {
         this->delay = atoi(option[1]);
      } else if (String_eq(option[0], "color_scheme")) {
         this->colorScheme = atoi(option[1]);
332
         if (this->colorScheme < 0 || this->colorScheme >= LAST_COLORSCHEME) this->colorScheme = 0;
Hisham Muhammad's avatar
Hisham Muhammad committed
333
      } else if (String_eq(option[0], "left_meters")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
334
         Settings_readMeters(this, option[1], 0);
335
         didReadMeters = true;
Hisham Muhammad's avatar
Hisham Muhammad committed
336
      } else if (String_eq(option[0], "right_meters")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
337
         Settings_readMeters(this, option[1], 1);
338
         didReadMeters = true;
Hisham Muhammad's avatar
Hisham Muhammad committed
339
      } else if (String_eq(option[0], "left_meter_modes")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
340
         Settings_readMeterModes(this, option[1], 0);
341
         didReadMeters = true;
Hisham Muhammad's avatar
Hisham Muhammad committed
342
      } else if (String_eq(option[0], "right_meter_modes")) {
Hisham Muhammad's avatar
Hisham Muhammad committed
343
         Settings_readMeterModes(this, option[1], 1);
344
         didReadMeters = true;
345
      } else if (strncmp(option[0], "screen:", 7) == 0) {
346
         Settings_newScreen(this, option[0] + 7, option[1]);
347
348
349
350
351
352
353
354
355
356
357
358
      } else if (String_eq(option[0], ".tree_view")) {
         if (this->nScreens > 0) {
            this->screens[this->nScreens - 1]->treeView = atoi(option[1]);
         }
      } else if (String_eq(option[0], ".sort_direction")) {
         if (this->nScreens > 0) {
            this->screens[this->nScreens - 1]->direction = atoi(option[1]);
         }
      } else if (String_eq(option[0], ".sort_key")) {
         if (this->nScreens > 0) {
            this->screens[this->nScreens - 1]->sortKey = toFieldIndex(option[1]);
         }
Hisham Muhammad's avatar
Hisham Muhammad committed
359
360
361
362
      }
      String_freeArray(option);
   }
   fclose(fd);
363
   if (this->nScreens == 0) {
364
      Settings_defaultScreens(this);
365
366
367
368
369
      if (legacyFieldsRead) {
         free(this->screens[0]->fields);
         this->screens[0]->fields = legacyFields;
         this->screens[0]->flags = legacyFlags;
      }
370
   }
371
   if (!didReadMeters) {
372
      Settings_defaultMeters(this);
Hisham Muhammad's avatar
Hisham Muhammad committed
373
   }
374
   return didReadFields;
Hisham Muhammad's avatar
Hisham Muhammad committed
375
376
}

377
static void writeFields(FILE* fd, ProcessField* fields, bool byName) {
378
   const char* sep = "";
Hisham Muhammad's avatar
Hisham Muhammad committed
379
   for (int i = 0; fields[i]; i++) {
380
381
382
383
384
385
      if (byName) {
         fprintf(fd, "%s%s", sep, Process_fields[fields[i]].name);
      } else {
         // This " - 1" is for compatibility with the older enum format.
         fprintf(fd, "%s%d", sep, (int) fields[i] - 1);
      }
386
      sep = " ";
Hisham Muhammad's avatar
Hisham Muhammad committed
387
388
389
390
   }
   fprintf(fd, "\n");
}

391
392
static void writeMeters(Settings* this, FILE* fd, int side) {
   writeList(fd, this->meterColumns[side].names, this->meterColumns[side].len);
Hisham Muhammad's avatar
Hisham Muhammad committed
393
394
}

395
static void writeMeterModes(Settings* this, FILE* fd, int side) {
396
   const char* sep = "";
397
398
   for (int i = 0; i < this->meterColumns[side].len; i++) {
      fprintf(fd, "%s%d", sep, this->meterColumns[side].modes[i]);
399
      sep = " ";
Hisham Muhammad's avatar
Hisham Muhammad committed
400
401
402
403
   }
   fprintf(fd, "\n");
}

Hisham Muhammad's avatar
Hisham Muhammad committed
404
405
bool Settings_write(Settings* this) {
   FILE* fd;
406

407
   CRT_dropPrivileges();
Hisham Muhammad's avatar
Hisham Muhammad committed
408
   fd = fopen(this->filename, "w");
409
410
   CRT_restorePrivileges();

Hisham Muhammad's avatar
Hisham Muhammad committed
411
412
413
   if (fd == NULL) {
      return false;
   }
414
   fprintf(fd, "# Beware! This file is rewritten by htop when settings are changed in the interface.\n");
Hisham Muhammad's avatar
Hisham Muhammad committed
415
   fprintf(fd, "# The parser is also very primitive, and not human-friendly.\n");
416
   fprintf(fd, "fields="); writeFields(fd, this->screens[0]->fields, false);
Hisham Muhammad's avatar
Hisham Muhammad committed
417
418
419
420
421
   fprintf(fd, "hide_threads=%d\n", (int) this->hideThreads);
   fprintf(fd, "hide_kernel_threads=%d\n", (int) this->hideKernelThreads);
   fprintf(fd, "hide_userland_threads=%d\n", (int) this->hideUserlandThreads);
   fprintf(fd, "shadow_other_users=%d\n", (int) this->shadowOtherUsers);
   fprintf(fd, "show_thread_names=%d\n", (int) this->showThreadNames);
422
   fprintf(fd, "show_program_path=%d\n", (int) this->showProgramPath);
Hisham Muhammad's avatar
Hisham Muhammad committed
423
424
425
426
427
428
429
430
   fprintf(fd, "highlight_base_name=%d\n", (int) this->highlightBaseName);
   fprintf(fd, "highlight_megabytes=%d\n", (int) this->highlightMegabytes);
   fprintf(fd, "highlight_threads=%d\n", (int) this->highlightThreads);
   fprintf(fd, "header_margin=%d\n", (int) this->headerMargin);
   fprintf(fd, "detailed_cpu_time=%d\n", (int) this->detailedCPUTime);
   fprintf(fd, "cpu_count_from_zero=%d\n", (int) this->countCPUsFromZero);
   fprintf(fd, "update_process_names=%d\n", (int) this->updateProcessNames);
   fprintf(fd, "account_guest_in_cpu_meter=%d\n", (int) this->accountGuestInCPUMeter);
Hisham Muhammad's avatar
Hisham Muhammad committed
431
432
   fprintf(fd, "color_scheme=%d\n", (int) this->colorScheme);
   fprintf(fd, "delay=%d\n", (int) this->delay);
Hisham Muhammad's avatar
Hisham Muhammad committed
433
434
435
436
   fprintf(fd, "left_meters="); writeMeters(this, fd, 0);
   fprintf(fd, "left_meter_modes="); writeMeterModes(this, fd, 0);
   fprintf(fd, "right_meters="); writeMeters(this, fd, 1);
   fprintf(fd, "right_meter_modes="); writeMeterModes(this, fd, 1);
437
438
439
440
441
442
443

   // Legacy compatibility with older versions of htop
   fprintf(fd, "tree_view=%d\n", (int) this->screens[0]->treeView);
   // This "-1" is for compatibility with the older enum format.
   fprintf(fd, "sort_key=%d\n", (int) this->screens[0]->sortKey-1);
   fprintf(fd, "sort_direction=%d\n", (int) this->screens[0]->direction);

444
445
446
447
448
449
450
451
452
453
   if (this->screens && this->screens[0]) {
      for (unsigned int i = 0; i < this->nScreens; i++) {
         ScreenSettings* ss = this->screens[i];
         fprintf(fd, "screen:%s=", ss->name);
         writeFields(fd, ss->fields, true);
         fprintf(fd, ".tree_view=%d\n", (int) ss->treeView);
         // This "-1" is for compatibility with the older enum format.
         fprintf(fd, ".sort_key=%d\n", (int) ss->sortKey-1);
         fprintf(fd, ".sort_direction=%d\n", (int) ss->direction);
      }
454
   }
Hisham Muhammad's avatar
Hisham Muhammad committed
455
456
457
   fclose(fd);
   return true;
}
458

Hisham Muhammad's avatar
Hisham Muhammad committed
459
460
Settings* Settings_new(int cpuCount) {
  
Hisham's avatar
Hisham committed
461
   Settings* this = xCalloc(1, sizeof(Settings));
Hisham Muhammad's avatar
Hisham Muhammad committed
462
463
464
465
466
467
468
469
470
471
472

   this->hideThreads = false;
   this->shadowOtherUsers = false;
   this->showThreadNames = false;
   this->hideKernelThreads = false;
   this->hideUserlandThreads = false;
   this->highlightBaseName = false;
   this->highlightMegabytes = false;
   this->detailedCPUTime = false;
   this->countCPUsFromZero = false;
   this->updateProcessNames = false;
473
   this->cpuCount = cpuCount;
474
   this->showProgramPath = true;
Hisham's avatar
Hisham committed
475
   this->highlightThreads = true;
Hisham Muhammad's avatar
Hisham Muhammad committed
476
   
477
478
   this->screens = xCalloc(sizeof(ScreenSettings*), 1);
   this->nScreens = 0;
Hisham Muhammad's avatar
Hisham Muhammad committed
479

480
481
482
   char* legacyDotfile = NULL;
   char* rcfile = getenv("HTOPRC");
   if (rcfile) {
Hisham's avatar
Hisham committed
483
      this->filename = xStrdup(rcfile);
484
485
486
487
488
489
490
   } else {
      const char* home = getenv("HOME");
      if (!home) home = "";
      const char* xdgConfigHome = getenv("XDG_CONFIG_HOME");
      char* configDir = NULL;
      char* htopDir = NULL;
      if (xdgConfigHome) {
Hisham Muhammad's avatar
Hisham Muhammad committed
491
         this->filename = String_cat(xdgConfigHome, "/htop/htoprc");
Hisham's avatar
Hisham committed
492
         configDir = xStrdup(xdgConfigHome);
493
494
         htopDir = String_cat(xdgConfigHome, "/htop");
      } else {
Hisham Muhammad's avatar
Hisham Muhammad committed
495
         this->filename = String_cat(home, "/.config/htop/htoprc");
496
497
498
499
         configDir = String_cat(home, "/.config");
         htopDir = String_cat(home, "/.config/htop");
      }
      legacyDotfile = String_cat(home, "/.htoprc");
500
501
      
      CRT_dropPrivileges();
502
503
      (void) mkdir(configDir, 0700);
      (void) mkdir(htopDir, 0700);
504
505
      free(htopDir);
      free(configDir);
506
      struct stat st;
507
508
      int err = lstat(legacyDotfile, &st);
      if (err || S_ISLNK(st.st_mode)) {
509
510
511
         free(legacyDotfile);
         legacyDotfile = NULL;
      }
512
      CRT_restorePrivileges();
513
   }
514
515
516
   this->colorScheme = 0;
   this->changed = false;
   this->delay = DEFAULT_DELAY;
517
518
519
520
   bool ok = false;
   if (legacyDotfile) {
      ok = Settings_read(this, legacyDotfile);
      if (ok) {
521
522
523
524
         // Transition to new location and delete old configuration file
         if (Settings_write(this))
            unlink(legacyDotfile);
      }
525
526
527
528
529
530
      free(legacyDotfile);
   }
   if (!ok) {
      ok = Settings_read(this, this->filename);
   }
   if (!ok) {
531
532
533
      this->changed = true;
      // TODO: how to get SYSCONFDIR correctly through Autoconf?
      char* systemSettings = String_cat(SYSCONFDIR, "/htoprc");
534
      ok = Settings_read(this, systemSettings);
535
536
      free(systemSettings);
   }
537
538
   if (!ok) {
      Settings_defaultMeters(this);
539
      Settings_defaultScreens(this);
540
541
542
543
544
      this->hideKernelThreads = true;
      this->highlightMegabytes = true;
      this->highlightThreads = true;
      this->headerMargin = true;
   }
545
546
547
548

   this->ssIndex = 0;
   this->ss = this->screens[this->ssIndex];

549
550
   return this;
}
Hisham Muhammad's avatar
Hisham Muhammad committed
551

552
void ScreenSettings_invertSortOrder(ScreenSettings* this) {
Hisham Muhammad's avatar
Hisham Muhammad committed
553
554
555
556
557
   if (this->direction == 1)
      this->direction = -1;
   else
      this->direction = 1;
}