Commit | Line | Data |
---|---|---|
25903407 ACM |
1 | /* |
2 | * (c) 2009 Arnaldo Carvalho de Melo <acme@redhat.com> | |
3 | * | |
4 | * Licensed under the GPLv2. | |
5 | */ | |
6 | ||
7 | #include "strlist.h" | |
74cf249d | 8 | #include "util.h" |
25903407 ACM |
9 | #include <errno.h> |
10 | #include <stdio.h> | |
11 | #include <stdlib.h> | |
12 | #include <string.h> | |
13 | ||
ee8dd3ca DA |
14 | static |
15 | struct rb_node *strlist__node_new(struct rblist *rblist, const void *entry) | |
25903407 | 16 | { |
ee8dd3ca DA |
17 | const char *s = entry; |
18 | struct rb_node *rc = NULL; | |
19 | struct strlist *strlist = container_of(rblist, struct strlist, rblist); | |
20 | struct str_node *snode = malloc(sizeof(*snode)); | |
25903407 | 21 | |
ee8dd3ca DA |
22 | if (snode != NULL) { |
23 | if (strlist->dupstr) { | |
25903407 ACM |
24 | s = strdup(s); |
25 | if (s == NULL) | |
26 | goto out_delete; | |
27 | } | |
ee8dd3ca DA |
28 | snode->s = s; |
29 | rc = &snode->rb_node; | |
25903407 ACM |
30 | } |
31 | ||
ee8dd3ca | 32 | return rc; |
25903407 ACM |
33 | |
34 | out_delete: | |
ee8dd3ca | 35 | free(snode); |
25903407 ACM |
36 | return NULL; |
37 | } | |
38 | ||
d8639f06 | 39 | static void str_node__delete(struct str_node *snode, bool dupstr) |
25903407 ACM |
40 | { |
41 | if (dupstr) | |
7d16c634 | 42 | zfree((char **)&snode->s); |
d8639f06 | 43 | free(snode); |
25903407 ACM |
44 | } |
45 | ||
ee8dd3ca DA |
46 | static |
47 | void strlist__node_delete(struct rblist *rblist, struct rb_node *rb_node) | |
25903407 | 48 | { |
ee8dd3ca DA |
49 | struct strlist *slist = container_of(rblist, struct strlist, rblist); |
50 | struct str_node *snode = container_of(rb_node, struct str_node, rb_node); | |
25903407 | 51 | |
ee8dd3ca DA |
52 | str_node__delete(snode, slist->dupstr); |
53 | } | |
25903407 | 54 | |
ee8dd3ca DA |
55 | static int strlist__node_cmp(struct rb_node *rb_node, const void *entry) |
56 | { | |
57 | const char *str = entry; | |
58 | struct str_node *snode = container_of(rb_node, struct str_node, rb_node); | |
59 | ||
60 | return strcmp(snode->s, str); | |
61 | } | |
25903407 | 62 | |
d8639f06 | 63 | int strlist__add(struct strlist *slist, const char *new_entry) |
ee8dd3ca | 64 | { |
d8639f06 | 65 | return rblist__add_node(&slist->rblist, new_entry); |
25903407 ACM |
66 | } |
67 | ||
d8639f06 | 68 | int strlist__load(struct strlist *slist, const char *filename) |
25903407 ACM |
69 | { |
70 | char entry[1024]; | |
71 | int err; | |
72 | FILE *fp = fopen(filename, "r"); | |
73 | ||
74 | if (fp == NULL) | |
ab7322af | 75 | return -errno; |
25903407 ACM |
76 | |
77 | while (fgets(entry, sizeof(entry), fp) != NULL) { | |
78 | const size_t len = strlen(entry); | |
79 | ||
80 | if (len == 0) | |
81 | continue; | |
82 | entry[len - 1] = '\0'; | |
83 | ||
d8639f06 | 84 | err = strlist__add(slist, entry); |
25903407 ACM |
85 | if (err != 0) |
86 | goto out; | |
87 | } | |
88 | ||
89 | err = 0; | |
90 | out: | |
91 | fclose(fp); | |
92 | return err; | |
93 | } | |
94 | ||
ee8dd3ca | 95 | void strlist__remove(struct strlist *slist, struct str_node *snode) |
25903407 | 96 | { |
45922814 | 97 | rblist__remove_node(&slist->rblist, &snode->rb_node); |
25903407 ACM |
98 | } |
99 | ||
ee8dd3ca | 100 | struct str_node *strlist__find(struct strlist *slist, const char *entry) |
25903407 | 101 | { |
ee8dd3ca DA |
102 | struct str_node *snode = NULL; |
103 | struct rb_node *rb_node = rblist__find(&slist->rblist, entry); | |
25903407 | 104 | |
ee8dd3ca DA |
105 | if (rb_node) |
106 | snode = container_of(rb_node, struct str_node, rb_node); | |
107 | ||
108 | return snode; | |
25903407 ACM |
109 | } |
110 | ||
8ff9daf3 ACM |
111 | static int strlist__parse_list_entry(struct strlist *slist, const char *s, |
112 | const char *subst_dir) | |
25903407 | 113 | { |
8ff9daf3 ACM |
114 | int err; |
115 | char *subst = NULL; | |
116 | ||
25903407 | 117 | if (strncmp(s, "file://", 7) == 0) |
d8639f06 | 118 | return strlist__load(slist, s + 7); |
25903407 | 119 | |
8ff9daf3 ACM |
120 | if (subst_dir) { |
121 | err = -ENOMEM; | |
122 | if (asprintf(&subst, "%s/%s", subst_dir, s) < 0) | |
123 | goto out; | |
124 | ||
125 | if (access(subst, F_OK) == 0) { | |
126 | err = strlist__load(slist, subst); | |
127 | goto out; | |
128 | } | |
dd8232bc NK |
129 | |
130 | if (slist->file_only) { | |
131 | err = -ENOENT; | |
132 | goto out; | |
133 | } | |
8ff9daf3 ACM |
134 | } |
135 | ||
136 | err = strlist__add(slist, s); | |
137 | out: | |
138 | free(subst); | |
139 | return err; | |
25903407 ACM |
140 | } |
141 | ||
8816d38d | 142 | static int strlist__parse_list(struct strlist *slist, const char *s, const char *subst_dir) |
25903407 ACM |
143 | { |
144 | char *sep; | |
145 | int err; | |
146 | ||
147 | while ((sep = strchr(s, ',')) != NULL) { | |
148 | *sep = '\0'; | |
8ff9daf3 | 149 | err = strlist__parse_list_entry(slist, s, subst_dir); |
25903407 ACM |
150 | *sep = ','; |
151 | if (err != 0) | |
152 | return err; | |
153 | s = sep + 1; | |
154 | } | |
155 | ||
8ff9daf3 | 156 | return *s ? strlist__parse_list_entry(slist, s, subst_dir) : 0; |
25903407 ACM |
157 | } |
158 | ||
4a77e218 | 159 | struct strlist *strlist__new(const char *list, const struct strlist_config *config) |
25903407 | 160 | { |
d8639f06 | 161 | struct strlist *slist = malloc(sizeof(*slist)); |
25903407 | 162 | |
d8639f06 | 163 | if (slist != NULL) { |
8ff9daf3 | 164 | bool dupstr = true; |
dd8232bc | 165 | bool file_only = false; |
8ff9daf3 ACM |
166 | const char *dirname = NULL; |
167 | ||
168 | if (config) { | |
169 | dupstr = !config->dont_dupstr; | |
170 | dirname = config->dirname; | |
dd8232bc | 171 | file_only = config->file_only; |
8ff9daf3 ACM |
172 | } |
173 | ||
d8639f06 ACM |
174 | rblist__init(&slist->rblist); |
175 | slist->rblist.node_cmp = strlist__node_cmp; | |
176 | slist->rblist.node_new = strlist__node_new; | |
177 | slist->rblist.node_delete = strlist__node_delete; | |
ee8dd3ca | 178 | |
8ff9daf3 | 179 | slist->dupstr = dupstr; |
dd8232bc | 180 | slist->file_only = file_only; |
8ff9daf3 ACM |
181 | |
182 | if (list && strlist__parse_list(slist, list, dirname) != 0) | |
25903407 ACM |
183 | goto out_error; |
184 | } | |
185 | ||
d8639f06 | 186 | return slist; |
25903407 | 187 | out_error: |
d8639f06 | 188 | free(slist); |
25903407 ACM |
189 | return NULL; |
190 | } | |
191 | ||
d8639f06 | 192 | void strlist__delete(struct strlist *slist) |
25903407 | 193 | { |
d8639f06 ACM |
194 | if (slist != NULL) |
195 | rblist__delete(&slist->rblist); | |
25903407 | 196 | } |
27d0fd41 | 197 | |
ee8dd3ca | 198 | struct str_node *strlist__entry(const struct strlist *slist, unsigned int idx) |
27d0fd41 | 199 | { |
ee8dd3ca DA |
200 | struct str_node *snode = NULL; |
201 | struct rb_node *rb_node; | |
27d0fd41 | 202 | |
ee8dd3ca DA |
203 | rb_node = rblist__entry(&slist->rblist, idx); |
204 | if (rb_node) | |
205 | snode = container_of(rb_node, struct str_node, rb_node); | |
27d0fd41 | 206 | |
ee8dd3ca | 207 | return snode; |
27d0fd41 | 208 | } |