1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
// SPDX-License-Identifier: GPL-2.0-or-later
/*
* Copyright (c) 2018 Jan Stancek <jstancek@redhat.com>
*/
#include <limits.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#define TST_NO_DEFAULT_MAIN
#include "tst_test.h"
#include "tst_sys_conf.h"
struct tst_sys_conf {
char path[PATH_MAX];
char value[PATH_MAX];
struct tst_sys_conf *next;
};
static struct tst_sys_conf *save_restore_data;
static void print_error(const int lineno, int info_only, const char *err,
const char *path)
{
if (info_only)
tst_res_(__FILE__, lineno, TINFO | TERRNO, err, path);
else
tst_brk_(__FILE__, lineno, TBROK | TERRNO, err, path);
}
void tst_sys_conf_dump(void)
{
struct tst_sys_conf *i;
for (i = save_restore_data; i; i = i->next)
tst_res(TINFO, "%s = %s", i->path, i->value);
}
void tst_sys_conf_save_str(const char *path, const char *value)
{
struct tst_sys_conf *n = SAFE_MALLOC(sizeof(*n));
strncpy(n->path, path, sizeof(n->path)-1);
strncpy(n->value, value, sizeof(n->value)-1);
n->path[sizeof(n->path) - 1] = 0;
n->value[sizeof(n->value) - 1] = 0;
n->next = save_restore_data;
save_restore_data = n;
}
int tst_sys_conf_save(const struct tst_path_val *conf)
{
char line[PATH_MAX];
int ttype, iret;
FILE *fp;
void *ret;
if (!conf || !conf->path)
tst_brk(TBROK, "path is empty");
if (access(conf->path, F_OK) != 0) {
if (conf->flags & TST_SR_SKIP_MISSING) {
tst_res(TINFO | TERRNO, "Path not found: %s",
conf->path);
return 1;
}
ttype = (conf->flags & TST_SR_TBROK_MISSING) ? TBROK : TCONF;
tst_brk(ttype | TERRNO, "Path not found: %s", conf->path);
}
if (access(conf->path, W_OK) != 0) {
if (conf->flags & TST_SR_SKIP_RO) {
tst_res(TINFO | TERRNO, "Path is not writable: %s",
conf->path);
return 1;
}
ttype = (conf->flags & TST_SR_TBROK_RO) ? TBROK : TCONF;
tst_brk(ttype | TERRNO, "Path is not writable: %s", conf->path);
}
fp = fopen(conf->path, "r");
if (fp == NULL) {
print_error(__LINE__, conf->flags & TST_SR_IGNORE_ERR,
"Failed to open '%s' for reading", conf->path);
return 1;
}
ret = fgets(line, sizeof(line), fp);
fclose(fp);
if (ret == NULL) {
if (conf->flags & TST_SR_IGNORE_ERR)
return 1;
tst_brk(TBROK | TERRNO, "Failed to read anything from '%s'",
conf->path);
}
tst_sys_conf_save_str(conf->path, line);
if (!conf->val)
return 0;
fp = fopen(conf->path, "w");
if (fp == NULL) {
print_error(__LINE__, conf->flags & TST_SR_IGNORE_ERR,
"Failed to open '%s' for writing", conf->path);
return 0;
}
iret = fputs(conf->val, fp);
if (iret < 0) {
print_error(__LINE__, conf->flags & TST_SR_IGNORE_ERR,
"Failed to write into '%s'", conf->path);
}
iret = fclose(fp);
if (iret < 0) {
print_error(__LINE__, conf->flags & TST_SR_IGNORE_ERR,
"Failed to close '%s'", conf->path);
}
return 0;
}
void tst_sys_conf_restore(int verbose)
{
struct tst_sys_conf *i;
for (i = save_restore_data; i; i = i->next) {
if (verbose) {
tst_res(TINFO, "Restoring conf.: %s -> %s\n",
i->path, i->value);
}
FILE_PRINTF(i->path, "%s", i->value);
}
}
|
