optflags: set _FORTIFY_SOURCE for preproc

[platform/upstream/rpm.git] / rpm2cpio.c

diff --git a/rpm2cpio.c b/rpm2cpio.c
index a7cf175..89ebdfa 100644 (file)
--- a/rpm2cpio.c
+++ b/rpm2cpio.c
@@ -1,84 +1,92 @@
 /* rpmarchive: spit out the main archive portion of a package */
 
 #include "system.h"
+const char *__progname;
 
-#include "rpmlib.h"
+#include <rpm/rpmlib.h>                /* rpmReadPackageFile .. */
+#include <rpm/rpmtag.h>
+#include <rpm/rpmio.h>
+#include <rpm/rpmpgp.h>
 
-int main(int argc, char **argv)
+#include <rpm/rpmts.h>
+
+#include "debug.h"
+
+int main(int argc, char *argv[])
 {
     FD_t fdi, fdo;
     Header h;
-    char * rpmio_flags;
-    int rc, isSource;
+    char * rpmio_flags = NULL;
+    int rc;
+    off_t payload_size;
     FD_t gzdi;
     
     setprogname(argv[0]);      /* Retrofit glibc __progname */
-    if (argc == 1) {
+    rpmReadConfigFiles(NULL, NULL);
+    if (argc == 1)
        fdi = fdDup(STDIN_FILENO);
-    } else {
+    else {
+       if (rstreq(argv[1], "-h") || rstreq(argv[1], "--help")) {
+           fprintf(stderr, "Usage: rpm2cpio file.rpm\n");
+           exit(EXIT_FAILURE);
+       }
        fdi = Fopen(argv[1], "r.ufdio");
     }
 
-    if (fdi == NULL || Ferror(fdi)) {
-       fprintf(stderr, _("cannot open package: %s\n"), Fstrerror(fdi));
+    if (Ferror(fdi)) {
+       fprintf(stderr, "%s: %s: %s\n", argv[0],
+               (argc == 1 ? "<stdin>" : argv[1]), Fstrerror(fdi));
        exit(EXIT_FAILURE);
     }
     fdo = fdDup(STDOUT_FILENO);
 
-    rc = rpmReadPackageHeader(fdi, &h, &isSource, NULL, NULL);
+    {  rpmts ts = rpmtsCreate();
+       rpmVSFlags vsflags = 0;
+
+       /* XXX retain the ageless behavior of rpm2cpio */
+        vsflags |= _RPMVSF_NODIGESTS;
+        vsflags |= _RPMVSF_NOSIGNATURES;
+        vsflags |= RPMVSF_NOHDRCHK;
+       (void) rpmtsSetVSFlags(ts, vsflags);
+
+       rc = rpmReadPackageFile(ts, fdi, "rpm2cpio", &h);
+
+       ts = rpmtsFree(ts);
+    }
+
     switch (rc) {
-    case 0:
+    case RPMRC_OK:
+    case RPMRC_NOKEY:
+    case RPMRC_NOTTRUSTED:
        break;
-    case 1:
+    case RPMRC_NOTFOUND:
        fprintf(stderr, _("argument is not an RPM package\n"));
        exit(EXIT_FAILURE);
        break;
+    case RPMRC_FAIL:
     default:
        fprintf(stderr, _("error reading header from package\n"));
        exit(EXIT_FAILURE);
        break;
     }
 
-    /* Retrieve type of payload compression. */
-    {  const char * payload_compressor = NULL;
-       char * t;
-
-       if (!headerGetEntry(h, RPMTAG_PAYLOADCOMPRESSOR, NULL,
-                           (void **) &payload_compressor, NULL))
-           payload_compressor = "gzip";
-       rpmio_flags = t = alloca(sizeof("r.gzdio"));
-       *t++ = 'r';
-       if (!strcmp(payload_compressor, "gzip"))
-           t = stpcpy(t, ".gzdio");
-       if (!strcmp(payload_compressor, "bzip2"))
-           t = stpcpy(t, ".bzdio");
+    /* Retrieve payload size and compression type. */
+    {  const char *compr = headerGetString(h, RPMTAG_PAYLOADCOMPRESSOR);
+       rpmio_flags = rstrscat(NULL, "r.", compr ? compr : "gzip", NULL);
+       payload_size = headerGetNumber(h, RPMTAG_LONGARCHIVESIZE);
     }
 
     gzdi = Fdopen(fdi, rpmio_flags);   /* XXX gzdi == fdi */
+    free(rpmio_flags);
+
     if (gzdi == NULL) {
        fprintf(stderr, _("cannot re-open payload: %s\n"), Fstrerror(gzdi));
        exit(EXIT_FAILURE);
     }
 
-#ifdef DYING
-  { char buffer[BUFSIZ];
-    int ct;
-    while ((ct = Fread(buffer, sizeof(buffer[0]), sizeof(buffer), gzdi)) > 0) {
-       Fwrite(buffer, sizeof(buffer[0]), ct, fdo);
-    }
+    rc = (ufdCopy(gzdi, fdo) == payload_size) ? EXIT_SUCCESS : EXIT_FAILURE;
 
-    if (ct < 0) {
-        fprintf (stderr, "rpm2cpio: zlib: %s\n", Fstrerror(gzdi));
-       rc = EXIT_FAILURE;
-    } else {
-       rc = EXIT_SUCCESS;
-    }
-  }
-#else
-    rc = ufdCopy(gzdi, fdo);
-    rc = (rc <= 0) ? EXIT_FAILURE : EXIT_SUCCESS;
     Fclose(fdo);
-#endif
 
     Fclose(gzdi);      /* XXX gzdi == fdi */