|
krh@27
|
1 |
#define _GNU_SOURCE
|
|
krh@27
|
2 |
|
|
krh@27
|
3 |
#include <string.h>
|
|
krh@27
|
4 |
#include <stdio.h>
|
|
krh@27
|
5 |
#include <sys/stat.h>
|
|
krh@27
|
6 |
#include <sys/mman.h>
|
|
krh@27
|
7 |
#include <unistd.h>
|
|
krh@27
|
8 |
#include <fcntl.h>
|
|
krh@27
|
9 |
#include <errno.h>
|
|
krh@27
|
10 |
|
|
krh@27
|
11 |
#include <expat.h>
|
|
krh@28
|
12 |
#include <rpm/rpmlib.h>
|
|
krh@28
|
13 |
#include <rpm/rpmdb.h>
|
|
krh@27
|
14 |
#include "sha1.h"
|
|
krh@27
|
15 |
#include "razor.h"
|
|
krh@27
|
16 |
|
|
krh@27
|
17 |
static void
|
|
krh@30
|
18 |
parse_package(struct razor_importer *importer, const char **atts, void *data)
|
|
krh@27
|
19 |
{
|
|
krh@27
|
20 |
const char *name = NULL, *version = NULL;
|
|
krh@27
|
21 |
int i;
|
|
krh@27
|
22 |
|
|
krh@27
|
23 |
for (i = 0; atts[i]; i += 2) {
|
|
krh@27
|
24 |
if (strcmp(atts[i], "name") == 0)
|
|
krh@27
|
25 |
name = atts[i + 1];
|
|
krh@27
|
26 |
else if (strcmp(atts[i], "version") == 0)
|
|
krh@27
|
27 |
version = atts[i + 1];
|
|
krh@27
|
28 |
}
|
|
krh@27
|
29 |
|
|
krh@27
|
30 |
if (name == NULL || version == NULL) {
|
|
krh@27
|
31 |
fprintf(stderr, "invalid package tag, "
|
|
krh@27
|
32 |
"missing name or version attributes\n");
|
|
krh@27
|
33 |
return;
|
|
krh@27
|
34 |
}
|
|
krh@27
|
35 |
|
|
krh@30
|
36 |
razor_importer_begin_package(importer, name, version);
|
|
krh@27
|
37 |
}
|
|
krh@27
|
38 |
|
|
krh@30
|
39 |
enum {
|
|
krh@30
|
40 |
RZR_REQUIRES, RZR_PROVIDES
|
|
krh@30
|
41 |
};
|
|
krh@30
|
42 |
|
|
krh@27
|
43 |
static void
|
|
krh@30
|
44 |
parse_property(struct razor_importer *importer, const char **atts, void *data)
|
|
krh@27
|
45 |
{
|
|
krh@27
|
46 |
const char *name = NULL, *version = NULL;
|
|
krh@27
|
47 |
int i;
|
|
krh@27
|
48 |
|
|
krh@27
|
49 |
for (i = 0; atts[i]; i += 2) {
|
|
krh@27
|
50 |
if (strcmp(atts[i], "name") == 0)
|
|
krh@27
|
51 |
name = atts[i + 1];
|
|
krh@27
|
52 |
if (strcmp(atts[i], "version") == 0)
|
|
krh@27
|
53 |
version = atts[i + 1];
|
|
krh@27
|
54 |
}
|
|
krh@27
|
55 |
|
|
krh@27
|
56 |
if (name == NULL) {
|
|
krh@27
|
57 |
fprintf(stderr, "invalid tag, missing name attribute\n");
|
|
krh@27
|
58 |
return;
|
|
krh@27
|
59 |
}
|
|
krh@27
|
60 |
|
|
krh@30
|
61 |
switch ((int) data) {
|
|
krh@30
|
62 |
case RZR_REQUIRES:
|
|
krh@30
|
63 |
razor_importer_add_requires(importer, name, version);
|
|
krh@30
|
64 |
break;
|
|
krh@30
|
65 |
case RZR_PROVIDES:
|
|
krh@30
|
66 |
razor_importer_add_provides(importer, name, version);
|
|
krh@30
|
67 |
break;
|
|
krh@30
|
68 |
}
|
|
krh@27
|
69 |
}
|
|
krh@27
|
70 |
|
|
krh@27
|
71 |
static void
|
|
krh@27
|
72 |
start_element(void *data, const char *name, const char **atts)
|
|
krh@27
|
73 |
{
|
|
krh@30
|
74 |
struct razor_importer *importer = data;
|
|
krh@27
|
75 |
|
|
krh@27
|
76 |
if (strcmp(name, "package") == 0)
|
|
krh@30
|
77 |
parse_package(importer, atts, NULL);
|
|
krh@27
|
78 |
else if (strcmp(name, "requires") == 0)
|
|
krh@30
|
79 |
parse_property(importer, atts, (void *) RZR_REQUIRES);
|
|
krh@27
|
80 |
else if (strcmp(name, "provides") == 0)
|
|
krh@30
|
81 |
parse_property(importer, atts, (void*) RZR_PROVIDES);
|
|
krh@27
|
82 |
}
|
|
krh@27
|
83 |
|
|
krh@27
|
84 |
static void
|
|
krh@27
|
85 |
end_element (void *data, const char *name)
|
|
krh@27
|
86 |
{
|
|
krh@30
|
87 |
struct razor_importer *importer = data;
|
|
krh@27
|
88 |
|
|
krh@27
|
89 |
if (strcmp(name, "package") == 0)
|
|
krh@30
|
90 |
razor_importer_finish_package(importer);
|
|
krh@27
|
91 |
}
|
|
krh@27
|
92 |
|
|
krh@27
|
93 |
static int
|
|
krh@30
|
94 |
import_rzr_file(struct razor_importer *importer, const char *filename)
|
|
krh@27
|
95 |
{
|
|
krh@27
|
96 |
SHA_CTX sha1;
|
|
krh@27
|
97 |
XML_Parser parser;
|
|
krh@27
|
98 |
int fd;
|
|
krh@27
|
99 |
void *p;
|
|
krh@27
|
100 |
struct stat stat;
|
|
krh@27
|
101 |
unsigned char hash[20];
|
|
krh@27
|
102 |
|
|
krh@27
|
103 |
fd = open(filename, O_RDONLY);
|
|
krh@27
|
104 |
if (fstat(fd, &stat) < 0)
|
|
krh@27
|
105 |
return -1;
|
|
krh@27
|
106 |
p = mmap(NULL, stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
|
|
krh@27
|
107 |
if (p == MAP_FAILED)
|
|
krh@27
|
108 |
return -1;
|
|
krh@27
|
109 |
|
|
krh@27
|
110 |
parser = XML_ParserCreate(NULL);
|
|
krh@30
|
111 |
XML_SetUserData(parser, importer);
|
|
krh@27
|
112 |
XML_SetElementHandler(parser, start_element, end_element);
|
|
krh@27
|
113 |
if (XML_Parse(parser, p, stat.st_size, 1) == XML_STATUS_ERROR) {
|
|
krh@27
|
114 |
fprintf(stderr,
|
|
krh@46
|
115 |
"%s at line %ld, %s\n",
|
|
krh@27
|
116 |
XML_ErrorString(XML_GetErrorCode(parser)),
|
|
krh@27
|
117 |
XML_GetCurrentLineNumber(parser),
|
|
krh@27
|
118 |
filename);
|
|
krh@27
|
119 |
return 1;
|
|
krh@27
|
120 |
}
|
|
krh@27
|
121 |
|
|
krh@27
|
122 |
XML_ParserFree(parser);
|
|
krh@27
|
123 |
|
|
krh@27
|
124 |
SHA1_Init(&sha1);
|
|
krh@27
|
125 |
SHA1_Update(&sha1, p, stat.st_size);
|
|
krh@27
|
126 |
SHA1_Final(hash, &sha1);
|
|
krh@27
|
127 |
|
|
krh@27
|
128 |
close(fd);
|
|
krh@27
|
129 |
|
|
krh@27
|
130 |
munmap(p, stat.st_size);
|
|
krh@27
|
131 |
|
|
krh@27
|
132 |
return 0;
|
|
krh@27
|
133 |
}
|
|
krh@27
|
134 |
|
|
krh@27
|
135 |
struct razor_set *
|
|
krh@27
|
136 |
razor_import_rzr_files(int count, const char *files[])
|
|
krh@27
|
137 |
{
|
|
krh@30
|
138 |
struct razor_importer *importer;
|
|
krh@27
|
139 |
int i;
|
|
krh@27
|
140 |
|
|
krh@30
|
141 |
importer = razor_importer_new();
|
|
krh@27
|
142 |
|
|
krh@27
|
143 |
for (i = 0; i < count; i++) {
|
|
krh@30
|
144 |
if (import_rzr_file(importer, files[i]) < 0) {
|
|
krh@27
|
145 |
fprintf(stderr, "failed to import %s\n", files[i]);
|
|
krh@27
|
146 |
exit(-1);
|
|
krh@27
|
147 |
}
|
|
krh@27
|
148 |
}
|
|
krh@27
|
149 |
|
|
krh@30
|
150 |
return razor_importer_finish(importer);
|
|
krh@27
|
151 |
}
|
|
krh@27
|
152 |
|
|
krh@27
|
153 |
/* Import a yum filelist as a razor package set. */
|
|
krh@27
|
154 |
|
|
krh@27
|
155 |
enum {
|
|
krh@27
|
156 |
YUM_STATE_BEGIN,
|
|
krh@30
|
157 |
YUM_STATE_PACKAGE_NAME,
|
|
krh@30
|
158 |
YUM_STATE_REQUIRES,
|
|
krh@46
|
159 |
YUM_STATE_PROVIDES,
|
|
krh@46
|
160 |
YUM_STATE_FILE
|
|
krh@27
|
161 |
};
|
|
krh@27
|
162 |
|
|
krh@27
|
163 |
struct yum_context {
|
|
krh@30
|
164 |
struct razor_importer *importer;
|
|
krh@27
|
165 |
struct import_property_context *current_property_context;
|
|
krh@48
|
166 |
char name[256], buffer[512], *p;
|
|
krh@27
|
167 |
int state;
|
|
krh@27
|
168 |
};
|
|
krh@27
|
169 |
|
|
krh@27
|
170 |
static void
|
|
krh@27
|
171 |
yum_start_element(void *data, const char *name, const char **atts)
|
|
krh@27
|
172 |
{
|
|
krh@27
|
173 |
struct yum_context *ctx = data;
|
|
krh@27
|
174 |
const char *n, *version;
|
|
krh@27
|
175 |
int i;
|
|
krh@27
|
176 |
|
|
krh@27
|
177 |
if (strcmp(name, "name") == 0) {
|
|
krh@27
|
178 |
ctx->state = YUM_STATE_PACKAGE_NAME;
|
|
krh@48
|
179 |
ctx->p = ctx->name;
|
|
krh@27
|
180 |
} else if (strcmp(name, "version") == 0) {
|
|
krh@27
|
181 |
version = NULL;
|
|
krh@27
|
182 |
for (i = 0; atts[i]; i += 2) {
|
|
krh@27
|
183 |
if (strcmp(atts[i], "ver") == 0)
|
|
krh@27
|
184 |
version = atts[i + 1];
|
|
krh@27
|
185 |
}
|
|
krh@30
|
186 |
razor_importer_begin_package(ctx->importer, ctx->name, version);
|
|
krh@27
|
187 |
} else if (strcmp(name, "rpm:requires") == 0) {
|
|
krh@30
|
188 |
ctx->state = YUM_STATE_REQUIRES;
|
|
krh@27
|
189 |
} else if (strcmp(name, "rpm:provides") == 0) {
|
|
krh@30
|
190 |
ctx->state = YUM_STATE_PROVIDES;
|
|
krh@27
|
191 |
} else if (strcmp(name, "rpm:entry") == 0 &&
|
|
krh@30
|
192 |
ctx->state != YUM_STATE_BEGIN) {
|
|
krh@27
|
193 |
n = NULL;
|
|
krh@27
|
194 |
version = NULL;
|
|
krh@27
|
195 |
for (i = 0; atts[i]; i += 2) {
|
|
krh@27
|
196 |
if (strcmp(atts[i], "name") == 0)
|
|
krh@27
|
197 |
n = atts[i + 1];
|
|
krh@27
|
198 |
else if (strcmp(atts[i], "ver") == 0)
|
|
krh@27
|
199 |
version = atts[i + 1];
|
|
krh@27
|
200 |
}
|
|
krh@27
|
201 |
|
|
krh@27
|
202 |
if (n == NULL) {
|
|
krh@27
|
203 |
fprintf(stderr, "invalid rpm:entry, "
|
|
krh@27
|
204 |
"missing name or version attributes\n");
|
|
krh@27
|
205 |
return;
|
|
krh@27
|
206 |
}
|
|
krh@27
|
207 |
|
|
krh@30
|
208 |
switch (ctx->state) {
|
|
krh@30
|
209 |
case YUM_STATE_REQUIRES:
|
|
krh@30
|
210 |
razor_importer_add_requires(ctx->importer, n, version);
|
|
krh@30
|
211 |
break;
|
|
krh@30
|
212 |
case YUM_STATE_PROVIDES:
|
|
krh@30
|
213 |
razor_importer_add_provides(ctx->importer, n, version);
|
|
krh@30
|
214 |
break;
|
|
krh@30
|
215 |
}
|
|
krh@46
|
216 |
} else if (strcmp(name, "file") == 0) {
|
|
krh@46
|
217 |
ctx->state = YUM_STATE_FILE;
|
|
krh@48
|
218 |
ctx->p = ctx->buffer;
|
|
krh@27
|
219 |
}
|
|
krh@27
|
220 |
}
|
|
krh@27
|
221 |
|
|
krh@27
|
222 |
static void
|
|
krh@27
|
223 |
yum_end_element (void *data, const char *name)
|
|
krh@27
|
224 |
{
|
|
krh@27
|
225 |
struct yum_context *ctx = data;
|
|
krh@27
|
226 |
|
|
krh@46
|
227 |
ctx->state = YUM_STATE_BEGIN;
|
|
krh@48
|
228 |
if (strcmp(name, "package") == 0)
|
|
krh@30
|
229 |
razor_importer_finish_package(ctx->importer);
|
|
krh@48
|
230 |
else if (strcmp(name, "file") == 0)
|
|
krh@48
|
231 |
razor_importer_add_file(ctx->importer, ctx->buffer);
|
|
krh@27
|
232 |
}
|
|
krh@27
|
233 |
|
|
krh@27
|
234 |
static void
|
|
krh@27
|
235 |
yum_character_data (void *data, const XML_Char *s, int len)
|
|
krh@27
|
236 |
{
|
|
krh@27
|
237 |
struct yum_context *ctx = data;
|
|
krh@27
|
238 |
|
|
krh@48
|
239 |
switch (ctx->state) {
|
|
krh@48
|
240 |
case YUM_STATE_PACKAGE_NAME:
|
|
krh@48
|
241 |
case YUM_STATE_FILE:
|
|
krh@48
|
242 |
memcpy(ctx->p, s, len);
|
|
krh@48
|
243 |
ctx->p += len;
|
|
krh@48
|
244 |
*ctx->p = '\0';
|
|
krh@48
|
245 |
break;
|
|
krh@46
|
246 |
}
|
|
krh@27
|
247 |
}
|
|
krh@27
|
248 |
|
|
krh@27
|
249 |
struct razor_set *
|
|
krh@27
|
250 |
razor_set_create_from_yum_filelist(int fd)
|
|
krh@27
|
251 |
{
|
|
krh@27
|
252 |
struct yum_context ctx;
|
|
krh@27
|
253 |
XML_Parser parser;
|
|
krh@27
|
254 |
char buf[4096];
|
|
krh@27
|
255 |
int len;
|
|
krh@27
|
256 |
|
|
krh@30
|
257 |
ctx.importer = razor_importer_new();
|
|
krh@27
|
258 |
|
|
krh@27
|
259 |
parser = XML_ParserCreate(NULL);
|
|
krh@27
|
260 |
XML_SetUserData(parser, &ctx);
|
|
krh@27
|
261 |
XML_SetElementHandler(parser, yum_start_element, yum_end_element);
|
|
krh@27
|
262 |
XML_SetCharacterDataHandler(parser, yum_character_data);
|
|
krh@27
|
263 |
|
|
krh@27
|
264 |
while (1) {
|
|
krh@27
|
265 |
len = read(fd, buf, sizeof buf);
|
|
krh@27
|
266 |
if (len < 0) {
|
|
krh@27
|
267 |
fprintf(stderr,
|
|
krh@27
|
268 |
"couldn't read input: %s\n", strerror(errno));
|
|
krh@27
|
269 |
return NULL;
|
|
krh@27
|
270 |
} else if (len == 0)
|
|
krh@27
|
271 |
break;
|
|
krh@27
|
272 |
|
|
krh@27
|
273 |
if (XML_Parse(parser, buf, len, 0) == XML_STATUS_ERROR) {
|
|
krh@27
|
274 |
fprintf(stderr,
|
|
krh@46
|
275 |
"%s at line %ld\n",
|
|
krh@27
|
276 |
XML_ErrorString(XML_GetErrorCode(parser)),
|
|
krh@27
|
277 |
XML_GetCurrentLineNumber(parser));
|
|
krh@27
|
278 |
return NULL;
|
|
krh@27
|
279 |
}
|
|
krh@27
|
280 |
}
|
|
krh@27
|
281 |
|
|
krh@27
|
282 |
XML_ParserFree(parser);
|
|
krh@27
|
283 |
|
|
krh@30
|
284 |
return razor_importer_finish(ctx.importer);
|
|
krh@27
|
285 |
}
|
|
krh@28
|
286 |
|
|
krh@31
|
287 |
union rpm_entry {
|
|
krh@31
|
288 |
void *p;
|
|
krh@31
|
289 |
char *string;
|
|
krh@31
|
290 |
char **list;
|
|
krh@42
|
291 |
unsigned int *flags;
|
|
krh@31
|
292 |
};
|
|
krh@31
|
293 |
|
|
krh@28
|
294 |
struct razor_set *
|
|
krh@28
|
295 |
razor_set_create_from_rpmdb(void)
|
|
krh@28
|
296 |
{
|
|
krh@30
|
297 |
struct razor_importer *importer;
|
|
krh@28
|
298 |
rpmdbMatchIterator iter;
|
|
krh@28
|
299 |
Header h;
|
|
krh@28
|
300 |
int_32 type, count, i;
|
|
krh@42
|
301 |
union rpm_entry name, version, release;
|
|
krh@42
|
302 |
union rpm_entry property_names, property_versions, property_flags;
|
|
krh@46
|
303 |
union rpm_entry basenames, dirnames, dirindexes;
|
|
krh@46
|
304 |
char filename[PATH_MAX];
|
|
krh@28
|
305 |
rpmdb db;
|
|
krh@28
|
306 |
|
|
krh@28
|
307 |
rpmReadConfigFiles(NULL, NULL);
|
|
krh@28
|
308 |
|
|
krh@28
|
309 |
if (rpmdbOpen("", &db, O_RDONLY, 0644) != 0) {
|
|
krh@28
|
310 |
fprintf(stderr, "cannot open rpm database\n");
|
|
krh@28
|
311 |
exit(1);
|
|
krh@28
|
312 |
}
|
|
krh@28
|
313 |
|
|
krh@30
|
314 |
importer = razor_importer_new();
|
|
krh@28
|
315 |
|
|
krh@28
|
316 |
iter = rpmdbInitIterator(db, 0, NULL, 0);
|
|
krh@28
|
317 |
while (h = rpmdbNextIterator(iter), h != NULL) {
|
|
krh@31
|
318 |
headerGetEntry(h, RPMTAG_NAME, &type, &name.p, &count);
|
|
krh@42
|
319 |
headerGetEntry(h, RPMTAG_VERSION, &type, &version.p, &count);
|
|
krh@31
|
320 |
headerGetEntry(h, RPMTAG_RELEASE, &type, &release.p, &count);
|
|
krh@31
|
321 |
razor_importer_begin_package(importer,
|
|
krh@31
|
322 |
name.string, version.string);
|
|
krh@28
|
323 |
|
|
krh@42
|
324 |
headerGetEntry(h, RPMTAG_REQUIRENAME, &type,
|
|
krh@42
|
325 |
&property_names.p, &count);
|
|
krh@28
|
326 |
headerGetEntry(h, RPMTAG_REQUIREVERSION, &type,
|
|
krh@31
|
327 |
&property_versions.p, &count);
|
|
krh@42
|
328 |
headerGetEntry(h, RPMTAG_REQUIREFLAGS, &type,
|
|
krh@42
|
329 |
&property_flags.p, &count);
|
|
krh@28
|
330 |
for (i = 0; i < count; i++)
|
|
krh@30
|
331 |
razor_importer_add_requires(importer,
|
|
krh@42
|
332 |
property_names.list[i],
|
|
krh@31
|
333 |
property_versions.list[i]);
|
|
krh@28
|
334 |
|
|
krh@42
|
335 |
headerGetEntry(h, RPMTAG_PROVIDENAME, &type,
|
|
krh@42
|
336 |
&property_names.p, &count);
|
|
krh@28
|
337 |
headerGetEntry(h, RPMTAG_PROVIDEVERSION, &type,
|
|
krh@31
|
338 |
&property_versions.p, &count);
|
|
krh@28
|
339 |
for (i = 0; i < count; i++)
|
|
krh@30
|
340 |
razor_importer_add_provides(importer,
|
|
krh@42
|
341 |
property_names.list[i],
|
|
krh@31
|
342 |
property_versions.list[i]);
|
|
krh@28
|
343 |
|
|
krh@46
|
344 |
headerGetEntry(h, RPMTAG_BASENAMES, &type,
|
|
krh@46
|
345 |
&basenames.p, &count);
|
|
krh@46
|
346 |
headerGetEntry(h, RPMTAG_DIRNAMES, &type,
|
|
krh@46
|
347 |
&dirnames.p, &count);
|
|
krh@46
|
348 |
headerGetEntry(h, RPMTAG_DIRINDEXES, &type,
|
|
krh@46
|
349 |
&dirindexes.p, &count);
|
|
krh@46
|
350 |
for (i = 0; i < count; i++) {
|
|
krh@46
|
351 |
snprintf(filename, sizeof filename, "%s%s",
|
|
krh@46
|
352 |
dirnames.list[dirindexes.flags[i]],
|
|
krh@46
|
353 |
basenames.list[i]);
|
|
krh@46
|
354 |
razor_importer_add_file(importer, filename);
|
|
krh@46
|
355 |
}
|
|
krh@46
|
356 |
|
|
krh@30
|
357 |
razor_importer_finish_package(importer);
|
|
krh@28
|
358 |
}
|
|
krh@28
|
359 |
|
|
krh@28
|
360 |
rpmdbClose(db);
|
|
krh@28
|
361 |
|
|
krh@30
|
362 |
return razor_importer_finish(importer);
|
|
krh@28
|
363 |
}
|