- Move header reading part from rpminstall to tryReadHeader function.
[platform/upstream/rpm.git] / lib / rpminstall.c
1 /** \ingroup rpmcli
2  * \file lib/rpminstall.c
3  */
4
5 #include "system.h"
6
7 #include <rpm/rpmcli.h>
8 #include <rpm/rpmtag.h>
9 #include <rpm/rpmlib.h>         /* rpmReadPackageFile, vercmp etc */
10 #include <rpm/rpmdb.h>
11 #include <rpm/rpmds.h>
12 #include <rpm/rpmts.h>
13 #include <rpm/rpmlog.h>
14 #include <rpm/rpmfileutil.h>
15 #include <rpm/rpmgi.h>
16
17 #include "lib/manifest.h"
18 #include "debug.h"
19
20 int rpmcliPackagesTotal = 0;
21 int rpmcliHashesCurrent = 0;
22 int rpmcliHashesTotal = 0;
23 int rpmcliProgressCurrent = 0;
24 int rpmcliProgressTotal = 0;
25
26 /**
27  * Print a CLI progress bar.
28  * @todo Unsnarl isatty(STDOUT_FILENO) from the control flow.
29  * @param amount        current
30  * @param total         final
31  */
32 static void printHash(const rpm_loff_t amount, const rpm_loff_t total)
33 {
34     int hashesNeeded;
35
36     rpmcliHashesTotal = (isatty (STDOUT_FILENO) ? 44 : 50);
37
38     if (rpmcliHashesCurrent != rpmcliHashesTotal) {
39         float pct = (total ? (((float) amount) / total) : 1.0);
40         hashesNeeded = (rpmcliHashesTotal * pct) + 0.5;
41         while (hashesNeeded > rpmcliHashesCurrent) {
42             if (isatty (STDOUT_FILENO)) {
43                 int i;
44                 for (i = 0; i < rpmcliHashesCurrent; i++)
45                     (void) putchar ('#');
46                 for (; i < rpmcliHashesTotal; i++)
47                     (void) putchar (' ');
48                 fprintf(stdout, "(%3d%%)", (int)((100 * pct) + 0.5));
49                 for (i = 0; i < (rpmcliHashesTotal + 6); i++)
50                     (void) putchar ('\b');
51             } else
52                 fprintf(stdout, "#");
53
54             rpmcliHashesCurrent++;
55         }
56         (void) fflush(stdout);
57
58         if (rpmcliHashesCurrent == rpmcliHashesTotal) {
59             int i;
60             rpmcliProgressCurrent++;
61             if (isatty(STDOUT_FILENO)) {
62                 for (i = 1; i < rpmcliHashesCurrent; i++)
63                     (void) putchar ('#');
64                 pct = (rpmcliProgressTotal
65                     ? (((float) rpmcliProgressCurrent) / rpmcliProgressTotal)
66                     : 1);
67                 fprintf(stdout, " [%3d%%]", (int)((100 * pct) + 0.5));
68             }
69             fprintf(stdout, "\n");
70         }
71         (void) fflush(stdout);
72     }
73 }
74
75 static rpmVSFlags setvsFlags(struct rpmInstallArguments_s * ia)
76 {
77     rpmVSFlags vsflags;
78
79     if (ia->installInterfaceFlags & (INSTALL_UPGRADE | INSTALL_ERASE))
80         vsflags = rpmExpandNumeric("%{?_vsflags_erase}");
81     else
82         vsflags = rpmExpandNumeric("%{?_vsflags_install}");
83
84     if (ia->qva_flags & VERIFY_DIGEST)
85         vsflags |= _RPMVSF_NODIGESTS;
86     if (ia->qva_flags & VERIFY_SIGNATURE)
87         vsflags |= _RPMVSF_NOSIGNATURES;
88     if (ia->qva_flags & VERIFY_HDRCHK)
89         vsflags |= RPMVSF_NOHDRCHK;
90
91     return vsflags;
92 }
93
94 void * rpmShowProgress(const void * arg,
95                         const rpmCallbackType what,
96                         const rpm_loff_t amount,
97                         const rpm_loff_t total,
98                         fnpyKey key,
99                         void * data)
100 {
101     Header h = (Header) arg;
102     char * s;
103     int flags = (int) ((long)data);
104     void * rc = NULL;
105     const char * filename = (const char *)key;
106     static FD_t fd = NULL;
107     int xx;
108
109     switch (what) {
110     case RPMCALLBACK_INST_OPEN_FILE:
111         if (filename == NULL || filename[0] == '\0')
112             return NULL;
113         fd = Fopen(filename, "r.ufdio");
114         /* FIX: still necessary? */
115         if (fd == NULL || Ferror(fd)) {
116             rpmlog(RPMLOG_ERR, _("open of %s failed: %s\n"), filename,
117                         Fstrerror(fd));
118             if (fd != NULL) {
119                 xx = Fclose(fd);
120                 fd = NULL;
121             }
122         } else
123             fd = fdLink(fd, RPMDBG_M("persist (showProgress)"));
124         return (void *)fd;
125         break;
126
127     case RPMCALLBACK_INST_CLOSE_FILE:
128         /* FIX: still necessary? */
129         fd = fdFree(fd, RPMDBG_M("persist (showProgress)"));
130         if (fd != NULL) {
131             xx = Fclose(fd);
132             fd = NULL;
133         }
134         break;
135
136     case RPMCALLBACK_INST_START:
137         rpmcliHashesCurrent = 0;
138         if (h == NULL || !(flags & INSTALL_LABEL))
139             break;
140         /* @todo Remove headerFormat() on a progress callback. */
141         if (flags & INSTALL_HASH) {
142             s = headerFormat(h, "%{NAME}", NULL);
143             if (isatty (STDOUT_FILENO))
144                 fprintf(stdout, "%4d:%-23.23s", rpmcliProgressCurrent + 1, s);
145             else
146                 fprintf(stdout, "%-28.28s", s);
147             (void) fflush(stdout);
148             s = _free(s);
149         } else {
150             s = headerFormat(h, "%{NAME}-%{VERSION}-%{RELEASE}", NULL);
151             fprintf(stdout, "%s\n", s);
152             (void) fflush(stdout);
153             s = _free(s);
154         }
155         break;
156
157     case RPMCALLBACK_TRANS_PROGRESS:
158     case RPMCALLBACK_INST_PROGRESS:
159         if (flags & INSTALL_PERCENT)
160             fprintf(stdout, "%%%% %f\n", (double) (total
161                                 ? ((((float) amount) / total) * 100)
162                                 : 100.0));
163         else if (flags & INSTALL_HASH)
164             printHash(amount, total);
165         (void) fflush(stdout);
166         break;
167
168     case RPMCALLBACK_TRANS_START:
169         rpmcliHashesCurrent = 0;
170         rpmcliProgressTotal = 1;
171         rpmcliProgressCurrent = 0;
172         if (!(flags & INSTALL_LABEL))
173             break;
174         if (flags & INSTALL_HASH)
175             fprintf(stdout, "%-28s", _("Preparing..."));
176         else
177             fprintf(stdout, "%s\n", _("Preparing packages for installation..."));
178         (void) fflush(stdout);
179         break;
180
181     case RPMCALLBACK_TRANS_STOP:
182         if (flags & INSTALL_HASH)
183             printHash(1, 1);    /* Fixes "preparing..." progress bar */
184         rpmcliProgressTotal = rpmcliPackagesTotal;
185         rpmcliProgressCurrent = 0;
186         break;
187
188     case RPMCALLBACK_UNINST_PROGRESS:
189         break;
190     case RPMCALLBACK_UNINST_START:
191         break;
192     case RPMCALLBACK_UNINST_STOP:
193         break;
194     case RPMCALLBACK_UNPACK_ERROR:
195         break;
196     case RPMCALLBACK_CPIO_ERROR:
197         break;
198     case RPMCALLBACK_SCRIPT_ERROR:
199         break;
200     case RPMCALLBACK_UNKNOWN:
201     default:
202         break;
203     }
204
205     return rc;
206 }       
207
208 static void setNotifyFlag(struct rpmInstallArguments_s * ia,
209                           rpmts ts)
210 {
211     int notifyFlags, xx;
212
213     notifyFlags = ia->installInterfaceFlags | (rpmIsVerbose() ? INSTALL_LABEL : 0 );
214     xx = rpmtsSetNotifyCallback(ts,
215                                 rpmShowProgress, (void *) ((long)notifyFlags));
216 }
217
218 struct rpmEIU {
219     Header h;
220     FD_t fd;
221     int numFailed;
222     int numPkgs;
223     char ** pkgURL;
224     char ** fnp;
225     char * pkgState;
226     int prevx;
227     int pkgx;
228     int numRPMS;
229     int numSRPMS;
230     char ** sourceURL;
231     int isSource;
232     int argc;
233     char ** argv;
234     rpmRelocation * relocations;
235     rpmRC rpmrc;
236 };
237
238 static int rpmcliTransaction(rpmts ts, struct rpmInstallArguments_s * ia,
239                       int numPackages)
240 {
241     rpmps ps;
242
243     int rc = 0;
244     int stop = 0;
245
246     int eflags = ia->installInterfaceFlags & INSTALL_ERASE;
247
248     if (!(ia->installInterfaceFlags & INSTALL_NODEPS)) {
249
250         if (rpmtsCheck(ts)) {
251             rc = numPackages;
252             stop = 1;
253         }
254
255         ps = rpmtsProblems(ts);
256         if (!stop && rpmpsNumProblems(ps) > 0) {
257             rpmlog(RPMLOG_ERR, _("Failed dependencies:\n"));
258             rpmpsPrint(NULL, ps);
259             rc = numPackages;
260             stop = 1;
261         }
262         ps = rpmpsFree(ps);
263     }
264
265     if ((eflags? 1 : (!stop)) && !(ia->installInterfaceFlags & INSTALL_NOORDER)) {
266         if (rpmtsOrder(ts)) {
267             rc = numPackages;
268             stop = 1;
269         }
270     }
271
272     if (numPackages && !stop) {
273
274         if (eflags) {
275             rpmlog(RPMLOG_DEBUG, "erasing packages\n");
276             rpmtsClean(ts);
277             rc = rpmtsRun(ts, NULL, ia->probFilter & (RPMPROB_FILTER_DISKSPACE|RPMPROB_FILTER_DISKNODES));
278         } else {
279             rpmlog(RPMLOG_DEBUG, "installing binary packages\n");
280             rpmtsClean(ts);
281             rc = rpmtsRun(ts, NULL, ia->probFilter);
282         }
283
284         ps = rpmtsProblems(ts);
285
286         if ((rpmpsNumProblems(ps) > 0) && (eflags? 1 : (rc > 0)))
287             rpmpsPrint((eflags? NULL : stderr), ps);
288         ps = rpmpsFree(ps);
289     }
290
291     return rc;
292 }
293
294 static int tryReadManifest(struct rpmEIU * eiu)
295 {
296     int rc, xx;
297
298     /* Try to read a package manifest. */
299     eiu->fd = Fopen(*eiu->fnp, "r.fpio");
300     if (eiu->fd == NULL || Ferror(eiu->fd)) {
301         rpmlog(RPMLOG_ERR, _("open of %s failed: %s\n"), *eiu->fnp,
302                Fstrerror(eiu->fd));
303         if (eiu->fd != NULL) {
304             xx = Fclose(eiu->fd);
305             eiu->fd = NULL;
306         }
307         eiu->numFailed++; *eiu->fnp = NULL;
308         return RPMRC_FAIL;
309     }
310
311     /* Read list of packages from manifest. */
312     rc = rpmReadPackageManifest(eiu->fd, &eiu->argc, &eiu->argv);
313     if (rc != RPMRC_OK)
314         rpmlog(RPMLOG_ERR, _("%s: not an rpm package (or package manifest): %s\n"),
315                *eiu->fnp, Fstrerror(eiu->fd));
316     xx = Fclose(eiu->fd);
317     eiu->fd = NULL;
318
319     if (rc != RPMRC_OK)
320         eiu->numFailed++; *eiu->fnp = NULL;
321
322     return rc;
323 }
324
325 static int tryReadHeader(rpmts ts, struct rpmEIU * eiu, rpmVSFlags vsflags)
326 {
327    rpmVSFlags tvsflags;
328
329    /* Try to read the header from a package file. */
330    eiu->fd = Fopen(*eiu->fnp, "r.ufdio");
331    if (eiu->fd == NULL || Ferror(eiu->fd)) {
332        rpmlog(RPMLOG_ERR, _("open of %s failed: %s\n"), *eiu->fnp,
333               Fstrerror(eiu->fd));
334        if (eiu->fd != NULL) {
335            Fclose(eiu->fd);
336            eiu->fd = NULL;
337        }
338        eiu->numFailed++; *eiu->fnp = NULL;
339        return RPMRC_FAIL;
340    }
341
342    /* Read the header, verifying signatures (if present). */
343    tvsflags = rpmtsSetVSFlags(ts, vsflags);
344    eiu->rpmrc = rpmReadPackageFile(ts, eiu->fd, *eiu->fnp, &eiu->h);
345    tvsflags = rpmtsSetVSFlags(ts, tvsflags);
346    Fclose(eiu->fd);
347    eiu->fd = NULL;
348    
349    /* Honor --nomanifest */
350    if (eiu->rpmrc == RPMRC_NOTFOUND && (giFlags & RPMGI_NOMANIFEST))
351        eiu->rpmrc = RPMRC_FAIL;
352
353    if(eiu->rpmrc == RPMRC_FAIL) {
354        rpmlog(RPMLOG_ERR, _("%s cannot be installed\n"), *eiu->fnp);
355        eiu->numFailed++; *eiu->fnp = NULL;
356    }
357
358    return RPMRC_OK;
359 }
360
361 /** @todo Generalize --freshen policies. */
362 int rpmInstall(rpmts ts, struct rpmInstallArguments_s * ia, ARGV_t fileArgv)
363 {
364     struct rpmEIU * eiu = xcalloc(1, sizeof(*eiu));
365     rpmRelocation * relocations;
366     char * fileURL = NULL;
367     rpmVSFlags vsflags, ovsflags;
368     int rc;
369     int xx;
370     int i;
371
372     if (fileArgv == NULL) goto exit;
373
374     rpmcliPackagesTotal = 0;
375
376     (void) rpmtsSetFlags(ts, ia->transFlags);
377
378     relocations = ia->relocations;
379
380     vsflags = setvsFlags(ia);
381     ovsflags = rpmtsSetVSFlags(ts, (vsflags | RPMVSF_NEEDPAYLOAD));
382
383     setNotifyFlag(ia, ts); 
384
385     if ((eiu->relocations = relocations) != NULL) {
386         while (eiu->relocations->oldPath)
387             eiu->relocations++;
388         if (eiu->relocations->newPath == NULL)
389             eiu->relocations = NULL;
390     }
391
392     /* Build fully globbed list of arguments in argv[argc]. */
393     for (eiu->fnp = fileArgv; *eiu->fnp != NULL; eiu->fnp++) {
394         ARGV_t av = NULL;
395         int ac = 0;
396         char * fn;
397
398         fn = rpmEscapeSpaces(*eiu->fnp);
399         rc = rpmGlob(fn, &ac, &av);
400         fn = _free(fn);
401         if (rc || ac == 0) {
402             rpmlog(RPMLOG_ERR, _("File not found by glob: %s\n"), *eiu->fnp);
403             eiu->numFailed++;
404             continue;
405         }
406
407         argvAppend(&(eiu->argv), av);
408         argvFree(av);
409         eiu->argc += ac;
410     }
411
412 restart:
413     /* Allocate sufficient storage for next set of args. */
414     if (eiu->pkgx >= eiu->numPkgs) {
415         eiu->numPkgs = eiu->pkgx + eiu->argc;
416         eiu->pkgURL = xrealloc(eiu->pkgURL,
417                         (eiu->numPkgs + 1) * sizeof(*eiu->pkgURL));
418         memset(eiu->pkgURL + eiu->pkgx, 0,
419                         ((eiu->argc + 1) * sizeof(*eiu->pkgURL)));
420         eiu->pkgState = xrealloc(eiu->pkgState,
421                         (eiu->numPkgs + 1) * sizeof(*eiu->pkgState));
422         memset(eiu->pkgState + eiu->pkgx, 0,
423                         ((eiu->argc + 1) * sizeof(*eiu->pkgState)));
424     }
425
426     /* Retrieve next set of args, cache on local storage. */
427     for (i = 0; i < eiu->argc; i++) {
428         fileURL = _free(fileURL);
429         fileURL = eiu->argv[i];
430         eiu->argv[i] = NULL;
431
432         switch (urlIsURL(fileURL)) {
433         case URL_IS_HTTPS:
434         case URL_IS_HTTP:
435         case URL_IS_FTP:
436         {   char *tfn;
437             FD_t tfd;
438
439             if (rpmIsVerbose())
440                 fprintf(stdout, _("Retrieving %s\n"), fileURL);
441
442             tfd = rpmMkTempFile(rpmtsRootDir(ts), &tfn);
443             if (tfd && tfn) {
444                 Fclose(tfd);
445                 rc = urlGetFile(fileURL, tfn);
446             } else {
447                 rc = -1;
448             }
449
450             if (rc != 0) {
451                 rpmlog(RPMLOG_ERR,
452                         _("skipping %s - transfer failed\n"), fileURL);
453                 eiu->numFailed++;
454                 eiu->pkgURL[eiu->pkgx] = NULL;
455                 tfn = _free(tfn);
456                 break;
457             }
458             eiu->pkgState[eiu->pkgx] = 1;
459             eiu->pkgURL[eiu->pkgx] = tfn;
460             eiu->pkgx++;
461         }   break;
462         case URL_IS_PATH:
463         case URL_IS_DASH:       /* WRONG WRONG WRONG */
464         case URL_IS_HKP:        /* WRONG WRONG WRONG */
465         default:
466             eiu->pkgURL[eiu->pkgx] = fileURL;
467             fileURL = NULL;
468             eiu->pkgx++;
469             break;
470         }
471     }
472     fileURL = _free(fileURL);
473
474     if (eiu->numFailed) goto exit;
475
476     /* Continue processing file arguments, building transaction set. */
477     for (eiu->fnp = eiu->pkgURL+eiu->prevx;
478          *eiu->fnp != NULL;
479          eiu->fnp++, eiu->prevx++)
480     {
481         const char * fileName;
482
483         rpmlog(RPMLOG_DEBUG, "============== %s\n", *eiu->fnp);
484         (void) urlPath(*eiu->fnp, &fileName);
485
486         if (tryReadHeader(ts, eiu, vsflags) == RPMRC_FAIL)
487             continue;
488
489         if (eiu->rpmrc == RPMRC_NOTFOUND) {
490             rc = tryReadManifest(eiu);
491             if (rc == RPMRC_OK) {
492                 eiu->prevx++;
493                 goto restart;
494             }
495         }
496
497         eiu->isSource = headerIsSource(eiu->h);
498
499         if (eiu->isSource) {
500             rpmlog(RPMLOG_DEBUG, "\tadded source package [%d]\n",
501                 eiu->numSRPMS);
502             eiu->sourceURL = xrealloc(eiu->sourceURL,
503                                 (eiu->numSRPMS + 2) * sizeof(*eiu->sourceURL));
504             eiu->sourceURL[eiu->numSRPMS] = *eiu->fnp;
505             *eiu->fnp = NULL;
506             eiu->numSRPMS++;
507             eiu->sourceURL[eiu->numSRPMS] = NULL;
508             continue;
509         }
510
511         if (eiu->relocations) {
512             struct rpmtd_s prefixes;
513
514             headerGet(eiu->h, RPMTAG_PREFIXES, &prefixes, HEADERGET_DEFAULT);
515             if (rpmtdCount(&prefixes) == 1) {
516                 eiu->relocations->oldPath = xstrdup(rpmtdGetString(&prefixes));
517                 rpmtdFreeData(&prefixes);
518             } else {
519                 const char * name;
520                 xx = headerNVR(eiu->h, &name, NULL, NULL);
521                 rpmlog(RPMLOG_ERR,
522                                _("package %s is not relocatable\n"), name);
523                 eiu->numFailed++;
524                 goto exit;
525             }
526         }
527
528         /* On --freshen, verify package is installed and newer */
529         if (ia->installInterfaceFlags & INSTALL_FRESHEN) {
530             rpmdbMatchIterator mi;
531             const char * name;
532             Header oldH;
533             int count;
534
535             xx = headerNVR(eiu->h, &name, NULL, NULL);
536             mi = rpmtsInitIterator(ts, RPMTAG_NAME, name, 0);
537             count = rpmdbGetIteratorCount(mi);
538             while ((oldH = rpmdbNextIterator(mi)) != NULL) {
539                 if (rpmVersionCompare(oldH, eiu->h) < 0)
540                     continue;
541                 /* same or newer package already installed */
542                 count = 0;
543                 break;
544             }
545             mi = rpmdbFreeIterator(mi);
546             if (count == 0) {
547                 eiu->h = headerFree(eiu->h);
548                 continue;
549             }
550             /* Package is newer than those currently installed. */
551         }
552
553         rc = rpmtsAddInstallElement(ts, eiu->h, (fnpyKey)fileName,
554                         (ia->installInterfaceFlags & INSTALL_UPGRADE) != 0,
555                         relocations);
556
557         /* XXX reference held by transaction set */
558         eiu->h = headerFree(eiu->h);
559         if (eiu->relocations)
560             eiu->relocations->oldPath = _free(eiu->relocations->oldPath);
561
562         switch(rc) {
563         case 0:
564             rpmlog(RPMLOG_DEBUG, "\tadded binary package [%d]\n",
565                         eiu->numRPMS);
566             break;
567         case 1:
568             rpmlog(RPMLOG_ERR,
569                             _("error reading from file %s\n"), *eiu->fnp);
570             eiu->numFailed++;
571             goto exit;
572             break;
573         case 2:
574             rpmlog(RPMLOG_ERR,
575                             _("file %s requires a newer version of RPM\n"),
576                             *eiu->fnp);
577             eiu->numFailed++;
578             goto exit;
579             break;
580         default:
581             eiu->numFailed++;
582             goto exit;
583             break;
584         }
585
586         eiu->numRPMS++;
587     }
588
589     rpmlog(RPMLOG_DEBUG, "found %d source and %d binary packages\n",
590                 eiu->numSRPMS, eiu->numRPMS);
591
592     if (eiu->numFailed) goto exit;
593
594     if (eiu->numRPMS) {
595         int rc = rpmcliTransaction(ts, ia, eiu->numPkgs);
596         if (rc < 0)
597             eiu->numFailed += eiu->numRPMS;
598         else if (rc > 0)
599             eiu->numFailed += rc;
600     }
601
602     if (eiu->numSRPMS && (eiu->sourceURL != NULL)) {
603         for (i = 0; i < eiu->numSRPMS; i++) {
604             rpmdbCheckSignals();
605             if (eiu->sourceURL[i] != NULL) {
606                 rc = RPMRC_OK;
607                 if (!(rpmtsFlags(ts) & RPMTRANS_FLAG_TEST))
608                     rc = rpmInstallSource(ts, eiu->sourceURL[i], NULL, NULL);
609                 if (rc != 0)
610                     eiu->numFailed++;
611             }
612         }
613     }
614
615 exit:
616     if (eiu->pkgURL != NULL) {
617         for (i = 0; i < eiu->numPkgs; i++) {
618             if (eiu->pkgURL[i] == NULL) continue;
619             if (eiu->pkgState[i] == 1)
620                 (void) unlink(eiu->pkgURL[i]);
621             eiu->pkgURL[i] = _free(eiu->pkgURL[i]);
622         }
623     }
624     eiu->pkgState = _free(eiu->pkgState);
625     eiu->pkgURL = _free(eiu->pkgURL);
626     eiu->argv = _free(eiu->argv);
627     rc = eiu->numFailed;
628     free(eiu);
629
630     rpmtsEmpty(ts);
631
632     return rc;
633 }
634
635 int rpmErase(rpmts ts, struct rpmInstallArguments_s * ia, ARGV_const_t argv)
636 {
637     char * const * arg;
638     char *qfmt = NULL;
639     int numFailed = 0;
640     int numPackages = 0;
641     rpmVSFlags vsflags, ovsflags;
642
643     if (argv == NULL) return 0;
644
645     vsflags = setvsFlags(ia);
646     ovsflags = rpmtsSetVSFlags(ts, vsflags);
647
648     /* XXX suggest mechanism only meaningful when installing */
649     ia->transFlags |= RPMTRANS_FLAG_NOSUGGEST;
650
651     (void) rpmtsSetFlags(ts, ia->transFlags);
652
653 #ifdef  NOTYET  /* XXX no callbacks on erase yet */
654     setNotifyFlag(ia, ts);
655 #endif
656
657     qfmt = rpmExpand("%{?_query_all_fmt}\n", NULL);
658     for (arg = argv; *arg; arg++) {
659         rpmdbMatchIterator mi;
660         int matches = 0;
661         int erasing = 1;
662
663         /* Iterator count isn't reliable with labels, count manually... */
664         mi = rpmtsInitIterator(ts, RPMDBI_LABEL, *arg, 0);
665         while (rpmdbNextIterator(mi) != NULL) {
666             matches++;
667         }
668         rpmdbFreeIterator(mi);
669
670         if (! matches) {
671             rpmlog(RPMLOG_ERR, _("package %s is not installed\n"), *arg);
672             numFailed++;
673         } else {
674             Header h;   /* XXX iterator owns the reference */
675
676             if (matches > 1 && 
677                 !(ia->installInterfaceFlags & UNINSTALL_ALLMATCHES)) {
678                 rpmlog(RPMLOG_ERR, _("\"%s\" specifies multiple packages:\n"),
679                         *arg);
680                 numFailed++;
681                 erasing = 0;
682             }
683
684             mi = rpmtsInitIterator(ts, RPMDBI_LABEL, *arg, 0);
685             while ((h = rpmdbNextIterator(mi)) != NULL) {
686                 if (erasing) {
687                     (void) rpmtsAddEraseElement(ts, h, -1);
688                     numPackages++;
689                 } else {
690                     char *nevra = headerFormat(h, qfmt, NULL);
691                     rpmlog(RPMLOG_NOTICE, "  %s", nevra);
692                     free(nevra);
693                 }
694             }
695             mi = rpmdbFreeIterator(mi);
696         }
697     }
698     free(qfmt);
699
700     if (numFailed) goto exit;
701     numFailed = rpmcliTransaction(ts, ia, numPackages);
702 exit:
703     rpmtsEmpty(ts);
704
705     return numFailed;
706 }
707
708 int rpmInstallSource(rpmts ts, const char * arg,
709                 char ** specFilePtr, char ** cookie)
710 {
711     FD_t fd;
712     int rc;
713
714
715     fd = Fopen(arg, "r.ufdio");
716     if (fd == NULL || Ferror(fd)) {
717         rpmlog(RPMLOG_ERR, _("cannot open %s: %s\n"), arg, Fstrerror(fd));
718         if (fd != NULL) (void) Fclose(fd);
719         return 1;
720     }
721
722     if (rpmIsVerbose() && specFilePtr != NULL)
723         fprintf(stdout, _("Installing %s\n"), arg);
724
725     {
726         rpmVSFlags ovsflags =
727                 rpmtsSetVSFlags(ts, (specFilePtr) ? (rpmtsVSFlags(ts) | RPMVSF_NEEDPAYLOAD) : rpmtsVSFlags(ts));
728         rpmRC rpmrc = rpmInstallSourcePackage(ts, fd, specFilePtr, cookie);
729         rc = (rpmrc == RPMRC_OK ? 0 : 1);
730         ovsflags = rpmtsSetVSFlags(ts, ovsflags);
731     }
732     if (rc != 0) {
733         rpmlog(RPMLOG_ERR, _("%s cannot be installed\n"), arg);
734         if (specFilePtr && *specFilePtr)
735             *specFilePtr = _free(*specFilePtr);
736         if (cookie && *cookie)
737             *cookie = _free(*cookie);
738     }
739
740     (void) Fclose(fd);
741
742     return rc;
743 }
744