rpmal rpmtsCreateAl(rpmts ts, rpmElementTypes types)
{
rpmal al = NULL;
if (ts) {
rpmte p;
rpmtsi pi;
rpmstrPool tspool = rpmtsPool(ts);
al = rpmalCreate(tspool, (rpmtsNElements(ts) / 4) + 1, rpmtsFlags(ts),
rpmtsColor(ts), rpmtsPrefColor(ts));
pi = rpmtsiInit(ts);
while ((p = rpmtsiNext(pi, types)))
rpmalAdd(al, p);
rpmtsiFree(pi);
}
return al;
}
static PyObject *
rpmal_Add(rpmalObject * s, PyObject * args, PyObject * kwds)
{
rpmdsObject * dso;
rpmfiObject * fio;
PyObject * key;
rpmalKey pkgKey;
char * kwlist[] = {"packageKey", "key", "dso", "fileInfo", NULL};
if (!PyArg_ParseTupleAndKeywords(args, kwds, "iOO!O!:Add", kwlist,
&pkgKey, &key, &rpmds_Type, &dso, &rpmfi_Type, &fio))
return NULL;
/* XXX errors */
/* XXX transaction colors */
pkgKey = rpmalAdd(&s->al, pkgKey, key, dso->ds, fio->fi, 0);
return Py_BuildValue("i", pkgKey);
}
static int addPackage(rpmts ts, Header h,
fnpyKey key, int op, rpmRelocation * relocs)
{
tsMembers tsmem = rpmtsMembers(ts);
rpm_color_t tscolor = rpmtsColor(ts);
rpmte p = NULL;
int isSource = headerIsSource(h);
int ec = 0;
int oc = tsmem->orderCount;
/* Check for supported payload format if it's a package */
if (key && headerCheckPayloadFormat(h) != RPMRC_OK) {
ec = 1;
goto exit;
}
/* Source packages are never "upgraded" */
if (isSource)
op = RPMTE_INSTALL;
/* Do lazy (readonly?) open of rpm database for upgrades. */
if (op != RPMTE_INSTALL && rpmtsGetRdb(ts) == NULL && rpmtsGetDBMode(ts) != -1) {
if ((ec = rpmtsOpenDB(ts, rpmtsGetDBMode(ts))) != 0)
goto exit;
}
p = rpmteNew(ts, h, TR_ADDED, key, relocs);
if (p == NULL) {
ec = 1;
goto exit;
}
/* Check binary packages for redundancies in the set */
if (!isSource) {
oc = findPos(ts, tscolor, p, (op == RPMTE_UPGRADE));
/* If we're replacing a previously added element, free the old one */
if (oc >= 0 && oc < tsmem->orderCount) {
rpmalDel(tsmem->addedPackages, tsmem->order[oc]);
tsmem->order[oc] = rpmteFree(tsmem->order[oc]);
/* If newer NEVR was already added, we're done */
} else if (oc < 0) {
p = rpmteFree(p);
goto exit;
}
}
if (oc >= tsmem->orderAlloced) {
tsmem->orderAlloced += (oc - tsmem->orderAlloced) + tsmem->delta;
tsmem->order = xrealloc(tsmem->order,
tsmem->orderAlloced * sizeof(*tsmem->order));
}
tsmem->order[oc] = p;
if (oc == tsmem->orderCount) {
tsmem->orderCount++;
}
if (tsmem->addedPackages == NULL) {
tsmem->addedPackages = rpmalCreate(rpmtsPool(ts), 5, rpmtsFlags(ts),
tscolor, rpmtsPrefColor(ts));
}
rpmalAdd(tsmem->addedPackages, p);
/* Add erasure elements for old versions and obsoletions on upgrades */
/* XXX TODO: If either of these fails, we'd need to undo all additions */
if (op != RPMTE_INSTALL)
addSelfErasures(ts, tscolor, op, p, rpmteColor(p), h);
if (op == RPMTE_UPGRADE)
addObsoleteErasures(ts, tscolor, p);
exit:
return ec;
}
