Sanitize python object -> tag number exception handling

[platform/upstream/rpm.git] / lib / rpmlead.c

diff --git a/lib/rpmlead.c b/lib/rpmlead.c
index d5b10be..22c29c8 100644 (file)
--- a/lib/rpmlead.c
+++ b/lib/rpmlead.c
@@ -1,56 +1,142 @@
-#ifdef HAVE_MACHINE_TYPES_H
-# include <machine/types.h>
-#endif
+/** \ingroup lead
+ * \file lib/rpmlead.c
+ */
+
+#include "system.h"
 
 #include <netinet/in.h>
-#include <stdlib.h>
-#include <stdio.h>
-#include <unistd.h>
-#include <string.h>
-#include <errno.h>
 
-#include "rpmerr.h"
-#include "rpmlead.h"
-#include "tread.h"
+#include <rpm/rpmlib.h>                /* rpmGetOs/ArchInfo() */
+#include <rpm/rpmlog.h>
+#include <rpm/rpmstring.h>
 
-/* The lead needs to be 8 byte aligned */
+#include "lib/signature.h"
+#include "lib/rpmlead.h"
+
+#include "debug.h"
+
+int _noDirTokens = 0;
+
+static unsigned char const lead_magic[] = {
+    RPMLEAD_MAGIC0, RPMLEAD_MAGIC1, RPMLEAD_MAGIC2, RPMLEAD_MAGIC3
+};
+
+/** \ingroup lead
+ * The lead data structure.
+ * The lead needs to be 8 byte aligned.
+ * @deprecated The lead (except for signature_type) is legacy.
+ * @todo Don't use any information from lead.
+ */
+struct rpmlead_s {
+    unsigned char magic[4];
+    unsigned char major;
+    unsigned char minor;
+    short type;
+    short archnum;
+    char name[66];
+    short osnum;
+    short signature_type;       /*!< Signature header type (RPMSIG_HEADERSIG) */
+    char reserved[16];      /*!< Pad to 96 bytes -- 8 byte aligned! */
+};
+
+rpmlead rpmLeadNew(void)
+{
+    int archnum, osnum;
+    rpmlead l = xcalloc(1, sizeof(*l));
+
+    rpmGetArchInfo(NULL, &archnum);
+    rpmGetOsInfo(NULL, &osnum);
+
+    l->major = (_noDirTokens ? 4: 3);
+    l->minor = 0;
+    l->archnum = archnum;
+    l->osnum = osnum;
+    l->signature_type = RPMSIGTYPE_HEADERSIG;
+    return l;
+}
+
+rpmlead rpmLeadFromHeader(Header h)
+{
+    char * nevr;
+    assert(h != NULL);
+    rpmlead l = rpmLeadNew();
+
+    l->type = (headerIsSource(h) ? 1 : 0);
+    nevr = headerGetAsString(h, RPMTAG_NEVR);
+    rstrlcpy(l->name, nevr, sizeof(l->name));
+    free(nevr);
 
-int writeLead(int fd, struct rpmlead *lead)
+    return l;
+}
+
+rpmlead rpmLeadFree(rpmlead lead)
 {
-    struct rpmlead l;
+    assert(lead != NULL);
+    free(lead);
+    return NULL;
+}
 
-    memcpy(&l, lead, sizeof(*lead));
+/* The lead needs to be 8 byte aligned */
+rpmRC rpmLeadWrite(FD_t fd, rpmlead lead)
+{
+    struct rpmlead_s l;
+    assert(lead != NULL);
+
+    memcpy(&l, lead, sizeof(l));
     
-    l.magic[0] = RPMLEAD_MAGIC0;
-    l.magic[1] = RPMLEAD_MAGIC1;
-    l.magic[2] = RPMLEAD_MAGIC2;
-    l.magic[3] = RPMLEAD_MAGIC3;
-
-    l.type = htons(l.type);
-    l.archnum = htons(l.archnum);
-    l.osnum = htons(l.osnum);
-    l.signature_type = htons(l.signature_type);
+    memcpy(&l.magic, lead_magic, sizeof(l.magic));
+    l.type = htons(lead->type);
+    l.archnum = htons(lead->archnum);
+    l.osnum = htons(lead->osnum);
+    l.signature_type = htons(lead->signature_type);
        
-    write(fd, &l, sizeof(l));
+    if (Fwrite(&l, 1, sizeof(l), fd) != sizeof(l))
+       return RPMRC_FAIL;
 
-    return 0;
+    return RPMRC_OK;
 }
 
-int readLead(int fd, struct rpmlead *lead)
+rpmRC rpmLeadCheck(rpmlead lead, const char **msg)
 {
-    if (timedRead(fd, lead, sizeof(*lead)) != sizeof(*lead)) {
-       error(RPMERR_READERROR, "read failed: %s (%d)", strerror(errno), 
-             errno);
-       return 1;
+    if (memcmp(lead->magic, lead_magic, sizeof(lead_magic))) {
+       if (msg) *msg = _("not an rpm package");
+       return RPMRC_NOTFOUND;
+    }
+    if (lead->signature_type != RPMSIGTYPE_HEADERSIG) {
+       if (msg) *msg = _("illegal signature type");
+       return RPMRC_FAIL;
     }
+    if (lead->major < 3 || lead->major > 4) {
+       if (msg) *msg = _("unsupported RPM package version");
+       return RPMRC_FAIL;
+    }
+    return RPMRC_OK;
+}
 
+rpmRC rpmLeadRead(FD_t fd, rpmlead lead)
+{
+    assert(lead != NULL);
+    memset(lead, 0, sizeof(*lead));
+    /* FIX: remove timed read */
+    if (timedRead(fd, (char *)lead, sizeof(*lead)) != sizeof(*lead)) {
+       if (Ferror(fd)) {
+           rpmlog(RPMLOG_ERR, _("read failed: %s (%d)\n"),
+                       Fstrerror(fd), errno);
+           return RPMRC_FAIL;
+       } else {
+           rpmlog(RPMLOG_ERR, _("not an rpm package\n"));
+           return RPMRC_NOTFOUND;
+       }
+    }
     lead->type = ntohs(lead->type);
     lead->archnum = ntohs(lead->archnum);
     lead->osnum = ntohs(lead->osnum);
+    lead->signature_type = ntohs(lead->signature_type);
 
-    if (lead->major >= 2)
-       lead->signature_type = ntohs(lead->signature_type);
-
-    return 0;
+    return RPMRC_OK;
 }
 
+int rpmLeadType(rpmlead lead)
+{
+    return lead ? lead->type : -1;
+}