*** empty log message ***

1 files changed, 429 insertions, 0 deletions

diff --git a/rpmtools.xs b/rpmtools.xs
new file mode 100644
index 0000000..b7621e0
--- /dev/null
+++ b/rpmtools.xs
@@ -0,0 +1,429 @@
+#include "EXTERN.h"
+#include "perl.h"
+#include "XSUB.h"
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+
+#undef Fflush
+#undef Mkdir
+#undef Stat
+#include <rpm/rpmlib.h>
+#include <rpm/header.h>
+
+char *get_name(Header header, int_32 tag) {
+  int_32 type, count;
+  char *name;
+
+  headerGetEntry(header, tag, &type, (void **) &name, &count);
+  return name;
+}
+
+int get_int(Header header, int_32 tag) {
+  int_32 type, count;
+  int *i;
+
+  headerGetEntry(header, tag, &type, (void **) &i, &count);
+  return *i;
+}
+
+HV* get_info(Header header) {
+  HV* info = newHV();
+  char *name = get_name(header, RPMTAG_NAME);
+  STRLEN len = strlen(name);
+
+  if (info != 0) {
+    SV* sv_name = newSVpv(name, len);
+    SV* sv_version = newSVpv(get_name(header, RPMTAG_VERSION), 0);
+    SV* sv_release = newSVpv(get_name(header, RPMTAG_RELEASE), 0);
+
+    hv_store(info, "name", 4, sv_name, 0);
+    hv_store(info, "version", 7, sv_version, 0);
+    hv_store(info, "release", 7, sv_release, 0);
+  }
+
+  return info;
+}
+
+SV *get_table_sense(Header header, int_32 tag_name, int_32 tag_flags, int_32 tag_version, HV* iprovides) {
+  AV* table_sense;
+  int_32 type, count;
+  char **list;
+  int_32 *flags;
+  char **list_evr;
+  int i;
+
+  char buff[4096];
+  char *p;
+  int len;
+
+  headerGetEntry(header, tag_name, &type, (void **) &list, &count);
+  if (tag_flags) headerGetEntry(header, tag_flags, &type, (void **) &flags, &count);
+  else flags = 0;
+  if (tag_version) headerGetEntry(header, tag_version, &type, (void **) &list_evr, &count);
+  else list_evr = 0;
+
+  if (list) {
+    table_sense = newAV();
+    if (!table_sense) return &PL_sv_undef;
+
+    for(i = 0; i < count; i++) {
+      len = strlen(list[i]); if (len >= sizeof(buff)) continue;
+      memcpy(p = buff, list[i], len + 1); p+= len;
+
+      if (flags) {
+	if (flags[i] & RPMSENSE_PREREQ) {
+	  if (p - buff + 3 >= sizeof(buff)) continue;
+	  memcpy(p, "[*]", 4); p += 3;
+	}
+	if (list_evr) {
+	  if (list_evr[i]) {
+	    len = strlen(list_evr[i]);
+	    if (len > 0) {
+	      if (p - buff + 6 + len >= sizeof(buff)) continue;
+	      *p++ = '[';
+	      if (flags[i] & RPMSENSE_LESS) *p++ = '<';
+	      if (flags[i] & RPMSENSE_GREATER) *p++ = '>';
+	      if (flags[i] & RPMSENSE_EQUAL) *p++ = '=';
+	      if ((flags[i] & (RPMSENSE_LESS|RPMSENSE_EQUAL|RPMSENSE_GREATER)) == RPMSENSE_EQUAL) *p++ = '=';
+	      *p++ = ' ';
+	      memcpy(p, list_evr[i], len); p+= len;
+	      *p++ = ']';
+	      *p = '\0';
+	    }
+	  }
+	}
+      }
+
+      /* for getting provides about required files */
+      if (iprovides && buff[0] == '/')
+	hv_fetch(iprovides, buff, p - buff, 1);
+
+      av_push(table_sense, newSVpv(buff, p - buff));
+    }
+
+    return newRV_noinc((SV*)table_sense);
+  }
+  return &PL_sv_undef;
+}
+
+#define HDFLAGS_NAME          0x00000001
+#define HDFLAGS_VERSION       0x00000002
+#define HDFLAGS_RELEASE       0x00000004
+#define HDFLAGS_ARCH          0x00000008
+#define HDFLAGS_GROUP         0x00000010
+#define HDFLAGS_SIZE          0x00000020
+#define HDFLAGS_SENSE         0x00080000
+#define HDFLAGS_REQUIRES      0x00100000
+#define HDFLAGS_PROVIDES      0x00200000
+#define HDFLAGS_OBSOLETES     0x00400000
+#define HDFLAGS_CONFLICTS     0x00800000
+#define HDFLAGS_FILES         0x01000000
+#define HDFLAGS_DIRSIND       0x02000000
+#define HDFLAGS_FILESIND      0x04000000
+
+
+MODULE = rpmtools			PACKAGE = rpmtools
+
+
+int
+get_packages_installed(prefix, packages, lnames)
+  char* prefix
+  SV* packages
+  SV* lnames
+  PREINIT:
+  int count = 0;
+  CODE:
+  if (SvROK(packages) && SvTYPE(SvRV(packages)) == SVt_PVAV &&
+      SvROK(lnames) && SvTYPE(SvRV(lnames)) == SVt_PVAV) {
+    AV* pkgs = (AV*)SvRV(packages);
+    AV* names = (AV*)SvRV(lnames);
+    HV* info;
+    SV** isv;
+    rpmdb db;
+    dbiIndexSet matches;
+    int num, i, j, rc, len;
+    char *name;
+    Header header;
+    
+    if (rpmReadConfigFiles(NULL, NULL) == 0) {
+      if (rpmdbOpen(prefix, &db, O_RDONLY, 0644) == 0) {
+	len = av_len(names);
+	for (j = 0; j <= len; ++j) {
+	  isv = av_fetch(names, j, 0);
+	  name = SvPV_nolen(*isv);
+	  rc = rpmdbFindPackage(db, name, &matches);
+	  if (rc == 0) {
+	    count += matches.count;
+	    for (i = 0; i < matches.count; ++i) {
+	      header = rpmdbGetRecord(db, matches.recs[i].recOffset);
+	      info = get_info(header);
+
+	      if (info != 0) av_push(pkgs, newRV_noinc((SV*)info));
+
+	      headerFree(header);
+	    }
+	  }
+	}
+	rpmdbClose(db);
+      } else croak("unable to open database");
+    } else croak("cannot read rpm config files");
+  } else croak("bad arguments list");
+  RETVAL = count;
+  OUTPUT:
+  RETVAL
+
+
+int
+get_all_packages_installed(prefix, packages)
+  char* prefix
+  SV* packages
+  PREINIT:
+  int count = 0;
+  CODE:
+  if (SvROK(packages) && SvTYPE(SvRV(packages)) == SVt_PVAV) {
+    AV* pkgs = (AV*)SvRV(packages);
+    HV* info;
+    rpmdb db;
+    int num;
+    Header header;
+    
+    if (rpmReadConfigFiles(NULL, NULL) == 0) {
+      if (rpmdbOpen(prefix, &db, O_RDONLY, 0644) == 0) {
+	num = rpmdbFirstRecNum(db);
+	while (num > 0) {
+	  header = rpmdbGetRecord(db, num);
+	  info = get_info(header);
+
+	  if (info != 0) av_push(pkgs, newRV_noinc((SV*)info));
+
+	  headerFree(header);
+	  num = rpmdbNextRecNum(db, num);
+	  ++count;
+	}
+	rpmdbClose(db);
+      } else croak("unable to open database");
+    } else croak("cannot read rpm config files");
+  } else croak("bad arguments list");
+  RETVAL = count;
+  OUTPUT:
+  RETVAL
+
+
+void
+_parse_(fileno_or_rpmfile, flag, info, ...)
+  SV* fileno_or_rpmfile
+  SV* flag
+  SV* info
+  PREINIT:
+  SV* provides = &PL_sv_undef;
+  CODE:
+  if (items > 3)
+    provides = ST(3);
+  if (SvROK(flag) && SvROK(info) && (provides == &PL_sv_undef || SvROK(provides))) {
+    FD_t fd;
+    int fd_is_hdlist;
+    Header header;
+    int_32 type, count;
+    char **list;
+    int_32 *flags;
+
+    int bflag = 0;
+    AV* iflag;
+    HV* iinfo;
+    HV* iprovides;
+    SV** ret;
+    I32 flag_len;
+    STRLEN len;
+    char* str;
+    int i;
+
+    if (SvIOK(fileno_or_rpmfile)) {
+      fd = fdDup(SvIV(fileno_or_rpmfile));
+      fd_is_hdlist = 1;
+    } else {
+      fd = fdOpen(SvPV_nolen(fileno_or_rpmfile), O_RDONLY, 0666);
+      if (fd < 0) croak("unable to open rpm file %s", SvPV_nolen(fileno_or_rpmfile));
+      fd_is_hdlist = 0;
+    }
+
+    if ((SvTYPE(SvRV(flag)) != SVt_PVAV) ||
+	(SvTYPE(SvRV(info)) != SVt_PVHV) ||
+	provides != &PL_sv_undef && (SvTYPE(SvRV(provides)) != SVt_PVHV))
+      croak("bad arguments list");
+
+    iflag = (AV*)SvRV(flag);
+    iinfo = (HV*)SvRV(info);
+    iprovides = (HV*)(provides != &PL_sv_undef ? SvRV(provides) : 0);
+
+    /* examine flag and set up iflag, which is faster to fecth out */
+    flag_len = av_len(iflag);
+    for (i = 0; i <= flag_len; ++i) {
+      ret = av_fetch(iflag, i, 0); if (!ret) continue;
+      str = SvPV(*ret, len);
+
+      switch (len) {
+      case 4:
+	if (!strncmp(str, "name", 4))      bflag |= HDFLAGS_NAME;
+	else if (!strncmp(str, "arch", 4)) bflag |= HDFLAGS_ARCH;
+	else if (!strncmp(str, "size", 4)) bflag |= HDFLAGS_SIZE;
+	break;
+      case 5:
+	if (!strncmp(str, "group", 5))      bflag |= HDFLAGS_GROUP;
+	else if (!strncmp(str, "sense", 5)) bflag |= HDFLAGS_SENSE;
+	else if (!strncmp(str, "files", 5)) bflag |= HDFLAGS_FILES;
+	break;
+      case 7:
+	if (!strncmp(str, "version", 7))      bflag |= HDFLAGS_VERSION;
+	else if (!strncmp(str, "release", 7)) bflag |= HDFLAGS_RELEASE;
+	else if (!strncmp(str, "dirsind", 7)) bflag |= HDFLAGS_DIRSIND;
+	break;
+      case 8:
+	if (!strncmp(str, "requires", 8))      bflag |= HDFLAGS_REQUIRES;
+	else if (!strncmp(str, "provides", 8)) bflag |= HDFLAGS_PROVIDES;
+	else if (!strncmp(str, "filesind", 8)) bflag |= HDFLAGS_FILESIND;
+	break;
+      case 9:
+	if (!strncmp(str, "obsoletes", 9))      bflag |= HDFLAGS_OBSOLETES;
+	else if (!strncmp(str, "conflicts", 9)) bflag |= HDFLAGS_CONFLICTS;
+	break;
+      }
+    }
+    bflag |= HDFLAGS_NAME; /* this one should always be used */
+    if (iprovides) bflag |= HDFLAGS_REQUIRES; /* not really usefull else */
+
+    /* start the big loop,
+       parse all header from fileno, then extract information to store into iinfo and iprovides. */
+    while (fd_is_hdlist >= 0 ? (fd_is_hdlist > 0 ?
+				((header=headerRead(fd, HEADER_MAGIC_YES)) != 0) :
+				((fd_is_hdlist = -1), rpmReadPackageHeader(fd, &header, &i, NULL, NULL) == 0)) : 0) {
+      char *name = get_name(header, RPMTAG_NAME);
+      SV* sv_name = newSVpv(name, 0);
+      HV* header_info = newHV();
+
+      if (bflag & HDFLAGS_NAME)
+	hv_store(header_info, "name", 4, SvREFCNT_inc(sv_name), 0);
+      if (bflag & HDFLAGS_VERSION)
+	hv_store(header_info, "version", 7, newSVpv(get_name(header, RPMTAG_VERSION), 0), 0);
+      if (bflag & HDFLAGS_RELEASE)
+	hv_store(header_info, "release", 7, newSVpv(get_name(header, RPMTAG_RELEASE), 0), 0);
+      if (bflag & HDFLAGS_ARCH)
+	hv_store(header_info, "arch", 4, newSVpv(get_name(header, RPMTAG_ARCH), 0), 0);
+      if (bflag & HDFLAGS_GROUP)
+	hv_store(header_info, "group", 5, newSVpv(get_name(header, RPMTAG_GROUP), 0), 0);
+      if (bflag & HDFLAGS_SIZE)
+	hv_store(header_info, "size", 4, newSViv(get_int(header, RPMTAG_SIZE)), 0);
+      if (bflag & HDFLAGS_REQUIRES)
+	hv_store(header_info, "requires", 8, get_table_sense(header,                 RPMTAG_REQUIRENAME,
+							     bflag & HDFLAGS_SENSE ? RPMTAG_REQUIREFLAGS : 0,
+							     bflag & HDFLAGS_SENSE ? RPMTAG_REQUIREVERSION : 0, iprovides), 0);
+      if (bflag & HDFLAGS_PROVIDES)
+	hv_store(header_info, "provides", 8, get_table_sense(header,                 RPMTAG_PROVIDENAME,
+							     bflag & HDFLAGS_SENSE ? RPMTAG_PROVIDEFLAGS : 0,
+							     bflag & HDFLAGS_SENSE ? RPMTAG_PROVIDEVERSION : 0, 0), 0);
+      if (bflag & HDFLAGS_OBSOLETES)
+	hv_store(header_info, "obsoletes", 9, get_table_sense(header,                 RPMTAG_OBSOLETENAME,
+							      bflag & HDFLAGS_SENSE ? RPMTAG_OBSOLETEFLAGS : 0,
+							      bflag & HDFLAGS_SENSE ? RPMTAG_OBSOLETEVERSION : 0, 0), 0);
+      if (bflag & HDFLAGS_CONFLICTS)
+	hv_store(header_info, "conflicts", 9, get_table_sense(header,                 RPMTAG_CONFLICTNAME,
+							      bflag & HDFLAGS_SENSE ? RPMTAG_CONFLICTFLAGS : 0,
+							      bflag & HDFLAGS_SENSE ? RPMTAG_CONFLICTVERSION : 0, 0), 0);
+      if (iprovides || (bflag & HDFLAGS_FILES)) {
+	/* at this point, there is a need to parse all files to update provides of needed files,
+	   or to store them. */
+	AV* table_files = bflag & HDFLAGS_FILES ? newAV() : 0;
+	char ** baseNames, ** dirNames;
+	int_32 * dirIndexes;
+
+	headerGetEntry(header, RPMTAG_OLDFILENAMES, &type, (void **) &list, &count);
+
+	if (list) {
+	  for (i = 0; i < count; i++) {
+	    SV** isv;
+
+	    len = strlen(list[i]);
+
+	    if (iprovides && (isv = hv_fetch(iprovides, list[i], len, 0)) != 0) {
+	      if (!SvROK(*isv) || SvTYPE(SvRV(*isv)) != SVt_PVAV) {
+		SV* choice_table = (SV*)newAV();
+		SvREFCNT_dec(*isv); /* drop the old as we are changing it */
+		*isv = choice_table ? newRV_noinc(choice_table) : &PL_sv_undef;
+	      }
+	      if (*isv != &PL_sv_undef) av_push((AV*)SvRV(*isv), SvREFCNT_inc(sv_name));
+	    }
+	    /*	    if (iprovides && hv_exists(iprovides, list[i], len))
+		    hv_store(iprovides, list[i], len, newSVpv(name, 0), 0); */
+	    if (table_files)
+	      av_push(table_files, newSVpv(list[i], len));
+	  }
+	}
+
+	headerGetEntry(header, RPMTAG_BASENAMES, &type, (void **) &baseNames, 
+		       &count);
+	headerGetEntry(header, RPMTAG_DIRINDEXES, &type, (void **) &dirIndexes, 
+		       NULL);
+	headerGetEntry(header, RPMTAG_DIRNAMES, &type, (void **) &dirNames, NULL);
+
+	if (baseNames && dirNames && dirIndexes) {
+	  char buff[4096];
+	  char *p;
+
+	  for(i = 0; i < count; i++) {
+	    SV** isv;
+
+	    len = strlen(dirNames[dirIndexes[i]]);
+	    if (len >= sizeof(buff)) continue;
+	    memcpy(p = buff, dirNames[dirIndexes[i]], len + 1); p += len;
+	    len = strlen(baseNames[i]);
+	    if (p - buff + len >= sizeof(buff)) continue;
+	    memcpy(p, baseNames[i], len + 1); p += len;
+
+	    if (iprovides && (isv = hv_fetch(iprovides, buff, p - buff, 0)) != 0) {
+	      if (!SvROK(*isv) || SvTYPE(SvRV(*isv)) != SVt_PVAV) {
+		SV* choice_table = (SV*)newAV();
+		SvREFCNT_dec(*isv); /* drop the old as we are changing it */
+		*isv = choice_table ? newRV_noinc(choice_table) : &PL_sv_undef;
+	      }
+	      if (*isv != &PL_sv_undef) av_push((AV*)SvRV(*isv), SvREFCNT_inc(sv_name));
+	    }
+	    if (table_files)
+	      av_push(table_files, newSVpv(buff, p - buff));
+	  }
+	}
+
+	if (table_files)
+	  hv_store(header_info, "files", 5, newRV_noinc((SV*)table_files), 0);
+      }
+      if (iprovides) {
+	/* we have to examine provides to update the hash here. */
+	headerGetEntry(header, RPMTAG_PROVIDENAME, &type, (void **) &list, &count);
+
+	if (list) {
+	  for (i = 0; i < count; i++) {
+	    SV** isv;
+
+	    len = strlen(list[i]);
+
+	    isv = hv_fetch(iprovides, list[i], len, 1);
+	    if (!SvROK(*isv) || SvTYPE(SvRV(*isv)) != SVt_PVAV) {
+	      SV* choice_table = (SV*)newAV();
+	      SvREFCNT_dec(*isv); /* drop the old as we are changing it */
+	      *isv = choice_table ? newRV_noinc(choice_table) : &PL_sv_undef;
+	    }
+	    if (*isv != &PL_sv_undef) av_push((AV*)SvRV(*isv), SvREFCNT_inc(sv_name));
+	  }
+	}
+      }
+
+      /* once the hash header_info is built, store a reference to it
+	 in iinfo.
+	 note sv_name is not incremented here, it has the default value of before. */
+      hv_store_ent(iinfo, sv_name, newRV_noinc((SV*)header_info), 0);
+
+      /* dispose of some memory */
+      headerFree(header);
+    }
+    fdClose(fd);
+  } else croak("bad arguments list");