]> git.ipfire.org Git - people/ms/pakfire.git/blame - src/libpakfire/packager.c
archive: Add flags argument to walk helper function
[people/ms/pakfire.git] / src / libpakfire / packager.c
CommitLineData
6aeb48e6
MT
1/*#############################################################################
2# #
3# Pakfire - The IPFire package management system #
4# Copyright (C) 2021 Pakfire development team #
5# #
6# This program is free software: you can redistribute it and/or modify #
7# it under the terms of the GNU General Public License as published by #
8# the Free Software Foundation, either version 3 of the License, or #
9# (at your option) any later version. #
10# #
11# This program is distributed in the hope that it will be useful, #
12# but WITHOUT ANY WARRANTY; without even the implied warranty of #
13# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
14# GNU General Public License for more details. #
15# #
16# You should have received a copy of the GNU General Public License #
17# along with this program. If not, see <http://www.gnu.org/licenses/>. #
18# #
19#############################################################################*/
20
21#include <errno.h>
1ea1f35b
MT
22#include <fcntl.h>
23#include <linux/limits.h>
6aeb48e6 24#include <stdlib.h>
da08f989
MT
25#include <sys/stat.h>
26#include <sys/types.h>
05cfc2c9 27#include <time.h>
da08f989 28#include <unistd.h>
6aeb48e6 29
1ea1f35b 30#include <archive.h>
da08f989 31#include <archive_entry.h>
1ea1f35b 32
2adc4a4a 33#include <pakfire/archive.h>
436677a3 34#include <pakfire/constants.h>
1ea1f35b 35#include <pakfire/logging.h>
6aeb48e6
MT
36#include <pakfire/package.h>
37#include <pakfire/packager.h>
38#include <pakfire/pakfire.h>
677e9e5b 39#include <pakfire/pwd.h>
f57ad279 40#include <pakfire/util.h>
6aeb48e6 41
da08f989
MT
42#define BUFFER_SIZE 64 * 1024
43
6aeb48e6 44struct pakfire_packager {
ac4c607b 45 struct pakfire* pakfire;
6aeb48e6 46 int nrefs;
05cfc2c9 47 time_t time_created;
6aeb48e6 48
31480bee 49 struct pakfire_package* pkg;
bb06d548 50 char filename[PATH_MAX];
1ea1f35b 51
738b3582
MT
52 // Reader
53 struct archive* reader;
54
55 // Payload
1ea1f35b 56 struct archive* payload;
7836e21b 57 FILE* fpayload;
4eab5079
MT
58 struct archive* mtree;
59 FILE* fmtree;
e282863a 60 size_t installsize;
106d2edd
MT
61
62 struct pakfire_scriptlet** scriptlets;
63 unsigned int num_scriptlets;
6aeb48e6
MT
64};
65
ac4c607b 66static int pakfire_packager_create_mtree(struct pakfire* pakfire, struct archive** mtree,
265a7974 67 FILE** f, const char** fields) {
96de0b5d 68 char path[] = PAKFIRE_PRIVATE_DIR "/tmp/.pakfire-mtree.XXXXXX";
265a7974
MT
69 int r;
70
71 // Create an mtree
72 *mtree = archive_write_new();
73 if (!*mtree) {
74 ERROR(pakfire, "archive_write_new() failed\n");
75 goto ERROR;
76 }
77
78 // Use mtree format
79 r = archive_write_set_format_mtree(*mtree);
80 if (r) {
81 ERROR(pakfire, "Could not set format to mtree: %s\n",
82 archive_error_string(*mtree));
83 goto ERROR;
84 }
85
86 // Always compress using Zstd
87 r = archive_write_add_filter_zstd(*mtree);
88 if (r) {
89 ERROR(pakfire, "Could not enable Zstandard compression: %s\n",
90 archive_error_string(*mtree));
91 goto ERROR;
92 }
93
94 // Enable mtree fields
95 if (fields) {
96 for (const char** field = fields; *field; field++) {
97 r = archive_write_set_options(*mtree, *field);
98 if (r) {
99 ERROR(pakfire, "Could not set mtree options %s: %s\n",
100 *field, archive_error_string(*mtree));
101 goto ERROR;
102 }
103 }
104 }
105
106 // Create a new temporary file
107 *f = pakfire_mktemp(path);
108 if (!*f)
109 goto ERROR;
110
111 // Unlink the file straight away
112 unlink(path);
113
114 // Write mtree to file
115 r = archive_write_open_FILE(*mtree, *f);
116 if (r)
117 goto ERROR;
118
119 return 0;
120
121ERROR:
122 if (*mtree) {
123 archive_write_free(*mtree);
124 *mtree = NULL;
125 }
126
127 if (*f) {
128 fclose(*f);
129 *f = NULL;
130 }
131
132 return 1;
133}
134
af14aefb 135static int pakfire_packager_create_payload(struct pakfire_packager* p) {
96de0b5d 136 char path[] = PAKFIRE_PRIVATE_DIR "/tmp/.pakfire-payload.XXXXXX";
7836e21b 137
af14aefb
MT
138 // Do not compress source packages
139 const int compress = !pakfire_package_is_source(p->pkg);
140
1ea1f35b 141 p->payload = archive_write_new();
427fdd80
MT
142 if (!p->payload) {
143 ERROR(p->pakfire, "archive_write_new() failed\n");
144 return 1;
145 }
1ea1f35b
MT
146
147 // Use the PAX format
148 int r = archive_write_set_format_pax(p->payload);
149 if (r) {
150 ERROR(p->pakfire, "Could not set format to PAX: %s\n",
151 archive_error_string(p->payload));
152 return r;
153 }
154
155 // Add filters to compress the payload
156 if (compress) {
157 // Enable Zstd
158 r = archive_write_add_filter_zstd(p->payload);
159 if (r) {
160 ERROR(p->pakfire, "Could not enable Zstandard compression: %s\n",
161 archive_error_string(p->payload));
162 return r;
163 }
164
165 // Set compression level to highest
be69fe38 166 r = archive_write_set_filter_option(p->payload, NULL, "compression-level", "22");
1ea1f35b
MT
167 if (r) {
168 ERROR(p->pakfire, "Could not set Zstandard compression level: %s\n",
169 archive_error_string(p->payload));
170 return r;
171 }
ae8f5905
MT
172
173 // Add feature marker
174 pakfire_package_add_requires(p->pkg, "pakfire(Compress-Zstandard)");
0dd87cbe
MT
175
176 // Do not pad the last block
177 archive_write_set_bytes_in_last_block(p->payload, 1);
1ea1f35b
MT
178 }
179
7836e21b 180 // Create a new temporary file
265a7974 181 p->fpayload = pakfire_mktemp(path);
f57ad279 182 if (!p->fpayload)
1ea1f35b 183 return 1;
1ea1f35b 184
7836e21b 185 // Unlink the file straight away
265a7974 186 unlink(path);
7836e21b 187
7836e21b
MT
188 // Write archive to file
189 r = archive_write_open_FILE(p->payload, p->fpayload);
1ea1f35b
MT
190 if (r)
191 return r;
192
4eab5079
MT
193 static const char* mtree_fields[] = {
194 // Disable the default
195 "!all",
196
197 // Enable standard file fields
198 "device",
199 "gname",
200 "link",
201 "mode",
202 "nlink",
203 "size",
204 "time",
205 "type",
206 "uname",
207
208 // Enable cryptographic checksums
209 "sha512",
210 "sha256",
211
212 NULL,
213 };
214
265a7974
MT
215 // Create an mtree
216 r = pakfire_packager_create_mtree(p->pakfire, &p->mtree, &p->fmtree, mtree_fields);
217 if (r) {
218 ERROR(p->pakfire, "Could not create mtree\n");
4eab5079 219 return r;
265a7974 220 }
4eab5079 221
1ea1f35b
MT
222 return 0;
223}
224
225static void pakfire_packager_free(struct pakfire_packager* packager) {
106d2edd
MT
226 // Scriptlets
227 if (packager->scriptlets) {
228 for (unsigned int i = 0; i < packager->num_scriptlets; i++)
9dd7b1aa 229 pakfire_scriptlet_unref(packager->scriptlets[i]);
106d2edd
MT
230 free(packager->scriptlets);
231 }
232
4eab5079 233 // Payload
1ea1f35b
MT
234 if (packager->payload)
235 archive_write_free(packager->payload);
236
7836e21b
MT
237 if (packager->fpayload)
238 fclose(packager->fpayload);
239
4eab5079
MT
240 // mtree
241 if (packager->mtree)
242 archive_write_free(packager->mtree);
243
244 if (packager->fmtree)
245 fclose(packager->fmtree);
246
738b3582 247 if (packager->reader)
66817062 248 archive_read_free(packager->reader);
738b3582 249
1ea1f35b
MT
250 pakfire_package_unref(packager->pkg);
251 pakfire_unref(packager->pakfire);
51105783 252 free(packager);
1ea1f35b
MT
253}
254
22b8e37b 255int pakfire_packager_create(struct pakfire_packager** packager,
31480bee 256 struct pakfire_package* pkg) {
6aeb48e6
MT
257 struct pakfire_packager* p = calloc(1, sizeof(*p));
258 if (!p)
259 return ENOMEM;
260
e282863a
MT
261 int r = 1;
262
05cfc2c9
MT
263 // Save creation time
264 p->time_created = time(NULL);
265
6aeb48e6
MT
266 // Initialize reference counting
267 p->nrefs = 1;
268
269 // Store a reference to Pakfire
270 p->pakfire = pakfire_package_get_pakfire(pkg);
271
272 // Store a reference to the package
273 p->pkg = pakfire_package_ref(pkg);
274
0682aeb3 275 // Create reader
66817062
MT
276 p->reader = pakfire_make_archive_disk_reader(p->pakfire, 1);
277 if (!p->reader)
1ba1869e 278 goto ERROR;
1ea1f35b 279
0682aeb3 280 // Start payload
af14aefb 281 r = pakfire_packager_create_payload(p);
0682aeb3 282 if (r)
738b3582 283 goto ERROR;
738b3582 284
6aeb48e6
MT
285 *packager = p;
286
287 return 0;
1ba1869e
MT
288
289ERROR:
290 pakfire_packager_free(p);
291
292 return r;
6aeb48e6
MT
293}
294
22b8e37b 295struct pakfire_packager* pakfire_packager_ref(
6aeb48e6
MT
296 struct pakfire_packager* packager) {
297 ++packager->nrefs;
298
299 return packager;
300}
301
22b8e37b 302struct pakfire_packager* pakfire_packager_unref(
6aeb48e6
MT
303 struct pakfire_packager* packager) {
304 if (--packager->nrefs > 0)
305 return packager;
306
307 pakfire_packager_free(packager);
308
309 return NULL;
310}
da08f989 311
22b8e37b 312const char* pakfire_packager_filename(struct pakfire_packager* packager) {
bb06d548
MT
313 if (!*packager->filename) {
314 const char* filename = pakfire_package_get_filename(packager->pkg);
315
316 // Add arch
317 if (pakfire_package_is_source(packager->pkg))
318 pakfire_string_set(packager->filename, filename);
319 else {
320 const char* arch = pakfire_package_get_arch(packager->pkg);
321
322 pakfire_string_format(packager->filename, "%s/%s", arch, filename);
323 }
324 }
325
326 return packager->filename;
96d2c7bc
MT
327}
328
7836e21b
MT
329static int pakfire_packager_copy_data(struct pakfire_packager* packager,
330 struct archive* a, FILE* f) {
331 char buffer[BUFFER_SIZE];
332
265a7974
MT
333 rewind(f);
334
7836e21b
MT
335 while (!feof(f)) {
336 // Read a block from file
337 size_t bytes_read = fread(buffer, 1, sizeof(buffer), f);
338
339 // Check if any error occured
340 if (ferror(f)) {
b1772bfb 341 ERROR(packager->pakfire, "Error reading from file: %m\n");
7836e21b
MT
342 return 1;
343 }
344
345 // Write the block to the archive
346 ssize_t bytes_written = archive_write_data(a, buffer, bytes_read);
347 if (bytes_written < 0) {
348 ERROR(packager->pakfire, "Error writing data to archive: %s\n",
349 archive_error_string(a));
350 return 1;
351 }
352 }
353
354 return 0;
355}
356
28700a5b 357static struct archive_entry* pakfire_packager_create_file(
95b81a98 358 struct pakfire_packager* packager, const char* filename, size_t size, mode_t mode) {
98e85f1c
MT
359 // Create a new file entry
360 struct archive_entry* entry = archive_entry_new();
361 if (!entry)
28700a5b 362 return NULL;
98e85f1c
MT
363
364 // Set filename
365 archive_entry_set_pathname(entry, filename);
366
367 // This is a regular file
368 archive_entry_set_filetype(entry, AE_IFREG);
95b81a98 369 archive_entry_set_perm(entry, mode);
98e85f1c 370
28700a5b
MT
371 // Set size
372 archive_entry_set_size(entry, size);
373
05cfc2c9
MT
374 // Set ownership
375 archive_entry_set_uname(entry, "root");
376 archive_entry_set_uid(entry, 0);
377 archive_entry_set_gname(entry, "root");
378 archive_entry_set_gid(entry, 0);
379
380 // Set times
381 archive_entry_set_birthtime(entry, packager->time_created, 0);
382 archive_entry_set_ctime(entry, packager->time_created, 0);
383 archive_entry_set_mtime(entry, packager->time_created, 0);
384 archive_entry_set_atime(entry, packager->time_created, 0);
385
28700a5b
MT
386 return entry;
387}
388
389static int pakfire_packager_write_file_from_buffer(struct pakfire_packager* packager,
7e346194 390 struct archive* a, struct archive* mtree, const char* filename, mode_t mode, const char* buffer) {
28700a5b
MT
391 size_t size = strlen(buffer);
392
393 // Create a new file
7e346194
MT
394 struct archive_entry* entry = pakfire_packager_create_file(packager, filename, size, mode);
395 if (!entry) {
396 ERROR(packager->pakfire, "Could not create file '%s'\n", filename);
28700a5b 397 return 1;
7e346194 398 }
98e85f1c
MT
399
400 // This is the end of the header
401 int r = archive_write_header(a, entry);
402 if (r) {
403 ERROR(packager->pakfire, "Error writing header: %s\n", archive_error_string(a));
265a7974 404 goto ERROR;
98e85f1c
MT
405 }
406
407 // Write content
408 r = archive_write_data(a, buffer, strlen(buffer));
409 if (r < 0) {
410 ERROR(packager->pakfire, "Error writing data: %s\n", archive_error_string(a));
265a7974 411 goto ERROR;
98e85f1c
MT
412 }
413
265a7974
MT
414 // Add this file to the mtree
415 if (mtree) {
416 r = archive_write_header(mtree, entry);
417 if (r) {
418 ERROR(packager->pakfire, "Error adding file to mtree: %s\n",
419 archive_error_string(mtree));
420 goto ERROR;
421 }
422 }
423
424 // Success
425 r = 0;
426
427ERROR:
98e85f1c
MT
428 archive_entry_free(entry);
429
265a7974 430 return r;
98e85f1c
MT
431}
432
436677a3
MT
433static int pakfire_packager_write_format(struct pakfire_packager* packager,
434 struct archive* a) {
435 const char buffer[] = TO_STRING(PACKAGE_FORMAT) "\n";
436
7e346194
MT
437 DEBUG(packager->pakfire, "Writing package format\n");
438
265a7974 439 int r = pakfire_packager_write_file_from_buffer(packager, a, NULL,
95b81a98 440 PAKFIRE_ARCHIVE_FN_FORMAT, 0444, buffer);
a6dd151e
MT
441 if (r)
442 return r;
443
444 // Add package format marker
445 pakfire_package_add_requires(packager->pkg,
446 "pakfire(PackageFormat-" TO_STRING(PACKAGE_FORMAT) ")");
447
448 return 0;
436677a3
MT
449}
450
98e85f1c 451static char* pakfire_package_make_metadata(struct pakfire_packager* packager) {
31480bee 452 struct pakfire_package* pkg = packager->pkg;
98e85f1c
MT
453
454 char* buffer = NULL;
455 int r;
456
457 // Print version information
458 r = asprintf(&buffer, "# Pakfire %s\n\n", PACKAGE_VERSION);
459 if (r < 0)
460 goto ERROR;
461
462 // Write package information
463 r = asprintf(&buffer, "%s# Package information\npackage\n", buffer);
464 if (r < 0)
465 goto ERROR;
466
467 // Write package name
468 r = asprintf(&buffer, "%s\tname = %s\n", buffer, pakfire_package_get_name(pkg));
469 if (r < 0)
470 goto ERROR;
471
6ed66687
MT
472 // Write package EVR
473 r = asprintf(&buffer, "%s\tevr = %s\n", buffer, pakfire_package_get_evr(pkg));
98e85f1c
MT
474 if (r < 0)
475 goto ERROR;
476
477 // Write package arch
478 r = asprintf(&buffer, "%s\tarch = %s\n", buffer, pakfire_package_get_arch(pkg));
479 if (r < 0)
480 goto ERROR;
481
482 // Write package UUID
483 r = asprintf(&buffer, "%s\tuuid = %s\n", buffer, pakfire_package_get_uuid(pkg));
484 if (r < 0)
485 goto ERROR;
486
487 // Write package groups
488 const char* groups = pakfire_package_get_groups(pkg);
489 if (groups) {
490 r = asprintf(&buffer, "%s\tgroups = %s\n", buffer, groups);
491 if (r < 0)
492 goto ERROR;
493 }
494
495 // Write package maintainer
496 const char* maintainer = pakfire_package_get_maintainer(pkg);
497 if (maintainer) {
498 r = asprintf(&buffer, "%s\tmaintainer = %s\n", buffer, maintainer);
499 if (r < 0)
500 goto ERROR;
501 }
502
503 // Write package url
e3c279d5 504 const char* url = pakfire_package_get_url(pkg);
98e85f1c
MT
505 if (url) {
506 r = asprintf(&buffer, "%s\turl = %s\n", buffer, url);
507 if (r < 0)
508 goto ERROR;
509 }
510
511 // Write package license
512 const char* license = pakfire_package_get_license(pkg);
513 if (license) {
514 r = asprintf(&buffer, "%s\tlicense = %s\n", buffer, license);
515 if (r < 0)
516 goto ERROR;
517 }
518
519 // Write package summary
520 const char* summary = pakfire_package_get_summary(pkg);
521 if (summary) {
522 r = asprintf(&buffer, "%s\tsummary = %s\n", buffer, summary);
523 if (r < 0)
524 goto ERROR;
525 }
526
527 // XXX description
528
529 size_t size = pakfire_package_get_installsize(pkg);
530 r = asprintf(&buffer, "%s\tsize = %zu\n", buffer, size);
531 if (r < 0)
532 goto ERROR;
533
534 // End package block
535 r = asprintf(&buffer, "%send\n\n", buffer);
536 if (r < 0)
537 goto ERROR;
538
539 // Write build information
540 r = asprintf(&buffer, "%s# Build information\nbuild\n", buffer);
541 if (r < 0)
542 goto ERROR;
543
544 // Write build host
545 const char* build_host = pakfire_package_get_build_host(pkg);
546 if (build_host) {
547 r = asprintf(&buffer, "%s\thost = %s\n", buffer, build_host);
548 if (r < 0)
549 goto ERROR;
550 }
551
98e85f1c
MT
552 // Write build id
553 const char* build_id = pakfire_package_get_build_id(pkg);
554 if (build_id) {
555 r = asprintf(&buffer, "%s\tid = %s\n", buffer, build_id);
556 if (r < 0)
557 goto ERROR;
558 }
98e85f1c
MT
559
560 // Write build host
561 time_t build_time = pakfire_package_get_build_time(pkg);
562 if (build_host) {
563 r = asprintf(&buffer, "%s\ttime = %lu\n", buffer, build_time);
564 if (r < 0)
565 goto ERROR;
566 }
567
568 // End build block
569 r = asprintf(&buffer, "%send\n\n", buffer);
570 if (r < 0)
571 goto ERROR;
572
573#if 0
574 // Write distribution information
575 r = asprintf(&buffer, "%s# Distribution information\ndistribution\n", buffer);
576 if (r < 0)
577 goto ERROR;
578
579 // End distribution block
580 r = asprintf(&buffer, "%send\n\n", buffer);
581 if (r < 0)
582 goto ERROR;
583#endif
584
585 // Write dependency information
586 r = asprintf(&buffer, "%s# Dependency information\ndependencies\n", buffer);
587 if (r < 0)
588 goto ERROR;
589
590 const struct dependencies {
591 const char* type;
31480bee 592 char** (*func)(struct pakfire_package* pkg);
98e85f1c
MT
593 } dependencies[] = {
594 { "prerequires", pakfire_package_get_prerequires },
595 { "requires", pakfire_package_get_requires },
596 { "provides", pakfire_package_get_provides },
597 { "conflicts", pakfire_package_get_conflicts },
598 { "obsoletes", pakfire_package_get_obsoletes },
599 { "recommends", pakfire_package_get_recommends },
600 { "suggests", pakfire_package_get_suggests },
601 { "supplements", pakfire_package_get_supplements },
602 { "enhances", pakfire_package_get_enhances },
603 { NULL },
604 };
605
606 for (const struct dependencies* d = dependencies; d->type; d++) {
452d3833 607 char** list = d->func(pkg);
98e85f1c
MT
608 if (!list)
609 continue;
610
98e85f1c
MT
611 // Write header
612 r = asprintf(&buffer, "%s\t%s\n", buffer, d->type);
613 if (r < 0) {
98e85f1c
MT
614 goto ERROR;
615 }
616
452d3833
MT
617 for (char** dep = list; *dep; dep++) {
618 asprintf(&buffer, "%s\t\t%s\n", buffer, *dep);
619 free(*dep);
98e85f1c 620 }
452d3833 621 free(list);
98e85f1c
MT
622
623 // End block
624 r = asprintf(&buffer, "%s\tend\n", buffer);
625 if (r < 0)
626 goto ERROR;
98e85f1c
MT
627 }
628
629 // End dependencies block
630 r = asprintf(&buffer, "%send\n\n", buffer);
631 if (r < 0)
632 goto ERROR;
633
634 // EOF
635 r = asprintf(&buffer, "%s# EOF\n", buffer);
636 if (r < 0)
637 goto ERROR;
638
639 return buffer;
640
641ERROR:
642 if (buffer)
643 free(buffer);
644
645 return NULL;
646}
647
648static int pakfire_packager_write_metadata(struct pakfire_packager* packager,
265a7974 649 struct archive* a, struct archive* mtree) {
98e85f1c
MT
650 // Make metadata
651 char* buffer = pakfire_package_make_metadata(packager);
652 if (!buffer)
653 return 1;
654
655 DEBUG(packager->pakfire, "Generated package metadata:\n%s", buffer);
656
657 // Write buffer
265a7974 658 int r = pakfire_packager_write_file_from_buffer(packager, a, mtree,
95b81a98 659 PAKFIRE_ARCHIVE_FN_METADATA, 0444, buffer);
98e85f1c
MT
660
661 free(buffer);
662
663 return r;
664}
665
4eab5079 666static int pakfire_packager_write_archive(struct pakfire_packager* packager,
265a7974
MT
667 struct archive* a, struct archive* mtree, const char* filename,
668 struct archive** payload, FILE* f) {
2adc4a4a
MT
669 struct stat st;
670
7e346194
MT
671 DEBUG(packager->pakfire, "Writing '%s' to package\n", filename);
672
2adc4a4a 673 // Close the payload
4eab5079
MT
674 if (*payload) {
675 archive_write_free(*payload);
676 *payload = NULL;
2adc4a4a
MT
677 }
678
679 // Reset fd to beginning of the file
4eab5079 680 rewind(f);
2adc4a4a 681
4eab5079 682 int fd = fileno(f);
2adc4a4a
MT
683
684 // Stat the payload file
685 int r = fstat(fd, &st);
686 if (r) {
b1772bfb 687 ERROR(packager->pakfire, "stat() on fd %d failed: %m\n", fd);
2adc4a4a
MT
688 return 1;
689 }
690
28700a5b
MT
691 // Create a new file
692 struct archive_entry* entry = pakfire_packager_create_file(packager,
95b81a98 693 filename, st.st_size, 0444);
2adc4a4a
MT
694 if (!entry)
695 return 1;
696
2adc4a4a
MT
697 // This is the end of the header
698 r = archive_write_header(a, entry);
699 if (r) {
700 ERROR(packager->pakfire, "Error writing header: %s\n", archive_error_string(a));
701 goto ERROR;
702 }
703
704 // Copy data
4eab5079 705 r = pakfire_packager_copy_data(packager, a, f);
7e346194
MT
706 if (r) {
707 const char* error = archive_error_string(a);
708 if (!error)
709 error = strerror(errno);
710
711 ERROR(packager->pakfire, "Could not copy payload: %s\n", error);
2adc4a4a 712 goto ERROR;
7e346194 713 }
2adc4a4a 714
265a7974
MT
715 // Add this file to the mtree
716 if (mtree) {
717 r = archive_write_header(mtree, entry);
718 if (r) {
719 ERROR(packager->pakfire, "Error adding file to mtree: %s\n",
720 archive_error_string(mtree));
721 goto ERROR;
722 }
723
724 r = pakfire_packager_copy_data(packager, mtree, f);
725 if (r) {
726 ERROR(packager->pakfire, "Error copying data to mtree: %s\n",
727 archive_error_string(mtree));
728 goto ERROR;
729 }
730 }
731
2adc4a4a
MT
732 // Success
733 r = 0;
734
735ERROR:
736 archive_entry_free(entry);
737
738 return r;
739}
740
106d2edd
MT
741static int pakfire_packager_write_scriptlet(struct pakfire_packager* packager,
742 struct archive* a, struct archive* mtree, struct pakfire_scriptlet* scriptlet) {
743 char filename[PATH_MAX];
744 size_t size;
745 int r;
746
747 // Fetch type
748 const char* type = pakfire_scriptlet_get_type(scriptlet);
749
750 DEBUG(packager->pakfire, "Writing scriptlet '%s' to package\n", type);
751
752 // Make filename
753 r = pakfire_string_format(filename, "scriptlet/%s", type);
754 if (r < 0)
755 return r;
756
757 // Fetch scriptlet
758 const char* data = pakfire_scriptlet_get_data(scriptlet, &size);
759
760 // Write file
95b81a98 761 return pakfire_packager_write_file_from_buffer(packager, a, mtree, filename, 0544, data);
106d2edd
MT
762}
763
436677a3
MT
764/*
765 This function is being called at the end when all data has been added to the package.
766
767 It will create a new archive and write the package to the given file descriptor.
768*/
22b8e37b 769int pakfire_packager_finish(struct pakfire_packager* packager, FILE* f) {
265a7974
MT
770 struct archive* mtree = NULL;
771 FILE* fmtree = NULL;
96d2c7bc 772 int r = 1;
436677a3 773
fafe383d 774 // Store total instal size
e282863a 775 pakfire_package_set_installsize(packager->pkg, packager->installsize);
fafe383d 776
436677a3
MT
777 // Open a new archive
778 struct archive* a = archive_write_new();
779 if (!a) {
780 ERROR(packager->pakfire, "archive_write_new() failed\n");
781 goto ERROR;
782 }
783
784 // Use the PAX format
96d2c7bc 785 r = archive_write_set_format_pax(a);
436677a3
MT
786 if (r) {
787 ERROR(packager->pakfire, "Could not set format to PAX: %s\n",
788 archive_error_string(a));
789 goto ERROR;
790 }
791
792 // Write archive to f
793 r = archive_write_open_FILE(a, f);
794 if (r) {
795 ERROR(packager->pakfire, "archive_write_open_FILE() failed: %s\n",
796 archive_error_string(a));
797 goto ERROR;
798 }
799
265a7974
MT
800 static const char* mtree_fields[] = {
801 // Disable everything
802 "!all",
803
7afc8d8f
MT
804 // File type is mandatory
805 "type",
806
265a7974
MT
807 // Include the file size
808 "size",
809
810 // Add the checksums
811 "sha512",
812 "sha256",
813
814 NULL,
815 };
816
817 // Create an mtree for the checksums
818 r = pakfire_packager_create_mtree(packager->pakfire, &mtree, &fmtree, mtree_fields);
819 if (r) {
820 ERROR(packager->pakfire, "Could not create mtree\n");
821 goto ERROR;
822 }
823
436677a3
MT
824 // Start with the format file
825 r = pakfire_packager_write_format(packager, a);
265a7974
MT
826 if (r) {
827 ERROR(packager->pakfire, "Could not add format file to archive: %s\n",
828 archive_error_string(a));
436677a3 829 goto ERROR;
265a7974 830 }
436677a3 831
98e85f1c 832 // Write the metadata
265a7974
MT
833 r = pakfire_packager_write_metadata(packager, a, mtree);
834 if (r) {
835 ERROR(packager->pakfire, "Could not add metadata file to archive: %s\n",
836 archive_error_string(a));
98e85f1c 837 goto ERROR;
265a7974 838 }
98e85f1c 839
4eab5079 840 // Write the filelist in mtree format
265a7974 841 r = pakfire_packager_write_archive(packager, a, mtree, PAKFIRE_ARCHIVE_FN_FILELIST,
4eab5079 842 &packager->mtree, packager->fmtree);
265a7974
MT
843 if (r) {
844 ERROR(packager->pakfire, "Could not add filelist to archive: %s\n",
845 archive_error_string(a));
4eab5079 846 goto ERROR;
265a7974 847 }
4eab5079 848
2adc4a4a 849 // Write the payload
265a7974 850 r = pakfire_packager_write_archive(packager, a, mtree, PAKFIRE_ARCHIVE_FN_PAYLOAD,
4eab5079 851 &packager->payload, packager->fpayload);
265a7974
MT
852 if (r) {
853 ERROR(packager->pakfire, "Could not add payload to archive: %s\n",
854 archive_error_string(a));
2adc4a4a 855 goto ERROR;
265a7974
MT
856 }
857
106d2edd
MT
858 // Write scriptlets
859 for (unsigned int i = 0; i < packager->num_scriptlets; i++) {
860 r = pakfire_packager_write_scriptlet(packager, a, mtree, packager->scriptlets[i]);
861 if (r) {
b1772bfb 862 ERROR(packager->pakfire, "Could not add scriptlet to the archive: %m\n");
106d2edd
MT
863 goto ERROR;
864 }
865 }
866
265a7974
MT
867 // Finish checksums
868 r = archive_write_finish_entry(mtree);
869 if (r) {
870 ERROR(packager->pakfire, "Could not finish mtree: %s\n",
871 archive_error_string(mtree));
872 goto ERROR;
873 }
874
bb06d548 875 // Write checksums
265a7974
MT
876 r = pakfire_packager_write_archive(packager, a, NULL, PAKFIRE_ARCHIVE_FN_CHECKSUMS,
877 &mtree, fmtree);
878 if (r) {
879 ERROR(packager->pakfire, "Could not add checksums to archive: %s\n",
880 archive_error_string(a));
881 goto ERROR;
882 }
2adc4a4a 883
96d2c7bc
MT
884 // Success
885 r = 0;
436677a3
MT
886
887ERROR:
888 if (a)
265a7974
MT
889 archive_write_free(a);
890 if (mtree)
891 archive_write_free(mtree);
892 if (fmtree)
893 fclose(fmtree);
436677a3 894
96d2c7bc 895 return r;
436677a3
MT
896}
897
22b8e37b 898int pakfire_packager_finish_to_directory(struct pakfire_packager* packager,
48c6f2e7 899 const char* target) {
48c6f2e7 900 char path[PATH_MAX];
bb06d548 901 char tmppath[PATH_MAX];
48c6f2e7
MT
902 int r = 1;
903
904 // target cannot be empty
905 if (!target) {
906 errno = EINVAL;
907 return 1;
908 }
909
bb06d548
MT
910 // Get the filename of the package
911 const char* filename = pakfire_packager_filename(packager);
912 if (!filename) {
b1772bfb 913 ERROR(packager->pakfire, "Could not generate filename for package: %m\n");
bb06d548
MT
914 r = 1;
915 goto ERROR;
916 }
917
918 // Make the package path
919 r = pakfire_string_format(path, "%s/%s", target, filename);
920 if (r < 0)
921 goto ERROR;
922
923 // Create the parent directory
924 r = pakfire_mkparentdir(path, 0);
925 if (r)
926 goto ERROR;
927
48c6f2e7 928 // Create a temporary file in the target directory
bb06d548
MT
929 r = pakfire_string_format(tmppath, "%s.XXXXXX", path);
930 if (r < 0)
931 goto ERROR;
48c6f2e7
MT
932
933 // Create a temporary result file
bb06d548 934 FILE* f = pakfire_mktemp(tmppath);
48c6f2e7
MT
935 if (!f)
936 goto ERROR;
937
938 // Write the finished package
939 r = pakfire_packager_finish(packager, f);
bb06d548
MT
940 fclose(f);
941
48c6f2e7 942 if (r) {
b1772bfb 943 ERROR(packager->pakfire, "pakfire_packager_finish() failed: %m\n");
48c6f2e7
MT
944 goto ERROR;
945 }
946
48c6f2e7 947 // Move the temporary file to destination
bb06d548
MT
948 r = rename(tmppath, path);
949 if (r) {
b1772bfb 950 ERROR(packager->pakfire, "Could not move %s to %s: %m\n", tmppath, path);
48c6f2e7 951 goto ERROR;
bb06d548 952 }
48c6f2e7 953
bb06d548 954 INFO(packager->pakfire, "Package written to %s\n", path);
48c6f2e7
MT
955
956 // Success
957 r = 0;
958
959ERROR:
960 // Remove temporary file
bb06d548
MT
961 if (r && *tmppath)
962 unlink(tmppath);
48c6f2e7 963
bb06d548 964 return r;
48c6f2e7
MT
965}
966
22b8e37b 967int pakfire_packager_add(struct pakfire_packager* packager,
809606fe 968 const char* sourcepath, const char* path) {
da08f989 969 FILE* f = NULL;
da08f989
MT
970
971 // Check if path is set
809606fe 972 if (!sourcepath)
da08f989
MT
973 return EINVAL;
974
809606fe
MT
975 // Use basename if path isn't set
976 if (!path) {
977 path = strrchr(sourcepath, '/');
978 if (path)
979 path++;
980 }
981
7836e21b
MT
982 // Payload has already been closed
983 if (!packager->payload)
984 return EINVAL;
985
da08f989
MT
986 // Create a new file entry
987 struct archive_entry* entry = archive_entry_new();
988 if (!entry)
989 return ENOMEM;
990
7e346194
MT
991 DEBUG(packager->pakfire, "Adding '%s' to archive (from %s)\n", path, sourcepath);
992
809606fe
MT
993 // Set the source path
994 archive_entry_copy_sourcepath(entry, sourcepath);
995
da08f989 996 // Set path in archive
809606fe
MT
997 if (path)
998 archive_entry_set_pathname(entry, path);
da08f989 999
738b3582
MT
1000 // Read all attributes from file
1001 int r = archive_read_disk_entry_from_file(packager->reader, entry, -1, NULL);
1002 if (r) {
b1772bfb 1003 ERROR(packager->pakfire, "Could not read attributes from %s: %m\n", path);
738b3582
MT
1004 goto ERROR;
1005 }
da08f989
MT
1006
1007 // Write the header
1008 r = archive_write_header(packager->payload, entry);
1009 if (r) {
1010 ERROR(packager->pakfire, "Error writing file header: %s\n",
1011 archive_error_string(packager->payload));
1012 goto ERROR;
1013 }
1014
1015 // Copy the data of regular files
1016 if (archive_entry_filetype(entry) == AE_IFREG) {
809606fe 1017 f = fopen(sourcepath, "r");
da08f989 1018 if (!f) {
b1772bfb 1019 ERROR(packager->pakfire, "Could not open %s: %m\n", sourcepath);
da08f989
MT
1020 r = errno;
1021 goto ERROR;
1022 }
1023
7836e21b
MT
1024 r = pakfire_packager_copy_data(packager, packager->payload, f);
1025 if (r)
1026 goto ERROR;
da08f989
MT
1027 }
1028
4eab5079
MT
1029 // Write to mtree
1030 r = archive_write_header(packager->mtree, entry);
1031 if (r) {
1032 ERROR(packager->pakfire, "Adding file to mtree failed: %s\n",
1033 archive_error_string(packager->mtree));
1034 goto ERROR;
1035 }
1036
e282863a
MT
1037 // Increment installsize
1038 packager->installsize += archive_entry_size(entry);
1ba1869e 1039
da08f989
MT
1040 // Successful
1041 r = 0;
1042
1043ERROR:
1044 if (entry)
1045 archive_entry_free(entry);
1046
1047 if (f)
1048 fclose(f);
1049
1050 return r;
1051}
106d2edd
MT
1052
1053int pakfire_packager_add_scriptlet(struct pakfire_packager* packager,
1054 struct pakfire_scriptlet* scriptlet) {
1055 if (!scriptlet) {
1056 errno = EINVAL;
1057 return 1;
1058 }
1059
1060 // Extend array
1061 packager->scriptlets = reallocarray(packager->scriptlets,
1062 packager->num_scriptlets + 1, sizeof(*packager->scriptlets));
1063 if (!packager->scriptlets)
1064 return 1;
1065
1066 // Append scriptlet
1067 packager->scriptlets[packager->num_scriptlets++] = pakfire_scriptlet_ref(scriptlet);
1068
1069 return 0;
1070}