6551cbf45057538ce81c2abcdffb9cd5dcfab003
[platform/upstream/libsolv.git] / src / policy.c
1 /*
2  * Copyright (c) 2007, Novell Inc.
3  *
4  * This program is licensed under the BSD license, read LICENSE.BSD
5  * for further information
6  */
7
8 /*
9  * Generic policy interface for SAT solver
10  *
11  */
12
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <unistd.h>
16 #include <string.h>
17
18 #include "solver.h"
19 #include "solver_private.h"
20 #include "evr.h"
21 #include "policy.h"
22 #include "poolvendor.h"
23 #include "poolarch.h"
24 #include "linkedpkg.h"
25 #include "cplxdeps.h"
26
27
28
29 /*-----------------------------------------------------------------*/
30
31 /*
32  * prep for prune_best_version
33  *   sort by name
34  */
35
36 static int
37 prune_to_best_version_sortcmp(const void *ap, const void *bp, void *dp)
38 {
39   Pool *pool = dp;
40   int r;
41   Id a = *(Id *)ap;
42   Id b = *(Id *)bp;
43   Solvable *sa, *sb;
44
45   sa = pool->solvables + a;
46   sb = pool->solvables + b;
47   r = sa->name - sb->name;
48   if (r)
49     {
50       const char *na, *nb;
51       /* different names. We use real strcmp here so that the result
52        * is not depending on some random solvable order */
53       na = pool_id2str(pool, sa->name);
54       nb = pool_id2str(pool, sb->name);
55       return strcmp(na, nb);
56     }
57   if (sa->arch != sb->arch)
58     {
59       unsigned int aa, ab;
60       aa = pool_arch2score(pool, sa->arch);
61       ab = pool_arch2score(pool, sb->arch);
62       if (aa != ab && aa > 1 && ab > 1)
63         return aa < ab ? -1 : 1;        /* lowest score first */
64     }
65
66   /* the same name, bring installed solvables to the front */
67   if (pool->installed)
68     {
69       if (sa->repo == pool->installed)
70         {
71           if (sb->repo != pool->installed)
72             return -1;
73         }
74       else if (sb->repo == pool->installed)
75         return 1;       
76     }
77   /* sort by repository sub-prio (installed repo handled above) */
78   r = (sb->repo ? sb->repo->subpriority : 0) - (sa->repo ? sa->repo->subpriority : 0);
79   if (r)
80     return r;
81   /* no idea about the order, sort by id */
82   return a - b;
83 }
84
85
86 /*
87  * prune to repository with highest priority.
88  * does not prune installed solvables.
89  */
90
91 static void
92 prune_to_highest_prio(Pool *pool, Queue *plist)
93 {
94   int i, j;
95   Solvable *s;
96   int bestprio = 0, bestprioset = 0;
97
98   /* prune to highest priority */
99   for (i = 0; i < plist->count; i++)  /* find highest prio in queue */
100     {
101       s = pool->solvables + plist->elements[i];
102       if (pool->installed && s->repo == pool->installed)
103         continue;
104       if (!bestprioset || s->repo->priority > bestprio)
105         {
106           bestprio = s->repo->priority;
107           bestprioset = 1;
108         }
109     }
110   if (!bestprioset)
111     return;
112   for (i = j = 0; i < plist->count; i++) /* remove all with lower prio */
113     {
114       s = pool->solvables + plist->elements[i];
115       if (s->repo->priority == bestprio || (pool->installed && s->repo == pool->installed))
116         plist->elements[j++] = plist->elements[i];
117     }
118   plist->count = j;
119 }
120
121
122 /* installed packages involed in a dup operation can only be kept
123  * if they are identical to a non-installed one */
124 static void
125 solver_prune_installed_dup_packages(Solver *solv, Queue *plist)
126 {
127   Pool *pool = solv->pool;
128   int i, j, bestprio = 0;
129
130   /* find bestprio (again) */
131   for (i = 0; i < plist->count; i++)
132     {
133       Solvable *s = pool->solvables + plist->elements[i];
134       if (s->repo != pool->installed)
135         {
136           bestprio = s->repo->priority;
137           break;
138         }
139     }
140   if (i == plist->count)
141     return;     /* only installed packages, could not find prio */
142   for (i = j = 0; i < plist->count; i++)
143     {
144       Id p = plist->elements[i];
145       Solvable *s = pool->solvables + p;
146       if (s->repo != pool->installed && s->repo->priority < bestprio)
147         continue;
148       if (s->repo == pool->installed && (solv->dupinvolvedmap_all || (solv->dupinvolvedmap.size && MAPTST(&solv->dupinvolvedmap, p))))
149         {
150           Id p2, pp2;
151           int keepit = 0;
152           FOR_PROVIDES(p2, pp2, s->name)
153             {
154               Solvable *s2 = pool->solvables + p2;
155               if (s2->repo == pool->installed || s2->evr != s->evr || s2->repo->priority < bestprio)
156                 continue;
157               if (!solvable_identical(s, s2))
158                 continue;
159               keepit = 1;
160               if (s2->repo->priority > bestprio)
161                 {
162                   /* new max prio! */
163                   bestprio = s2->repo->priority;
164                   j = 0;
165                 }
166             }
167           if (!keepit)
168             continue;   /* no identical package found, ignore installed package */
169         }
170       plist->elements[j++] = p;
171     }
172   if (j)
173     plist->count = j;
174 }
175
176 /*
177  * like prune_to_highest_prio, but calls solver prune_installed_dup_packages
178  * when there are dup packages
179  */
180 static inline void
181 solver_prune_to_highest_prio(Solver *solv, Queue *plist)
182 {
183   prune_to_highest_prio(solv->pool, plist);
184   if (plist->count > 1 && solv->pool->installed && (solv->dupinvolvedmap_all || solv->dupinvolvedmap.size))
185     solver_prune_installed_dup_packages(solv, plist);
186 }
187
188
189 static void
190 solver_prune_to_highest_prio_per_name(Solver *solv, Queue *plist)
191 {
192   Pool *pool = solv->pool;
193   Queue pq;
194   int i, j, k;
195   Id name;
196
197   queue_init(&pq);
198   solv_sort(plist->elements, plist->count, sizeof(Id), prune_to_best_version_sortcmp, pool);
199   queue_push(&pq, plist->elements[0]);
200   name = pool->solvables[pq.elements[0]].name;
201   for (i = 1, j = 0; i < plist->count; i++)
202     {
203       if (pool->solvables[plist->elements[i]].name != name)
204         {
205           name = pool->solvables[plist->elements[i]].name;
206           if (pq.count > 2)
207             solver_prune_to_highest_prio(solv, &pq);
208           for (k = 0; k < pq.count; k++)
209             plist->elements[j++] = pq.elements[k];
210           queue_empty(&pq);
211         }
212       queue_push(&pq, plist->elements[i]);
213     }
214   if (pq.count > 2)
215     solver_prune_to_highest_prio(solv, &pq);
216   for (k = 0; k < pq.count; k++)
217     plist->elements[j++] = pq.elements[k];
218   queue_free(&pq);
219   plist->count = j;
220 }
221
222
223 #ifdef ENABLE_COMPLEX_DEPS
224
225 /* simple fixed-size hash for package ids */
226 #define CPLXDEPHASH_EMPTY(elements) (memset(elements, 0, sizeof(Id) * 256))
227 #define CPLXDEPHASH_SET(elements, p) (elements[(p) & 255] |= (1 << ((p) >> 8 & 31)))
228 #define CPLXDEPHASH_TST(elements, p) (elements[(p) & 255] && (elements[(p) & 255] & (1 << ((p) >> 8 & 31))))
229
230 static void
231 check_complex_dep(Solver *solv, Id dep, Map *m, Queue **cqp)
232 {
233   Pool *pool = solv->pool;
234   Queue q;
235   queue_init(&q);
236   Id p;
237   int i, qcnt;
238
239 #if 0
240   printf("check_complex_dep %s\n", pool_dep2str(pool, dep));
241 #endif
242   i = pool_normalize_complex_dep(pool, dep, &q, CPLXDEPS_EXPAND);
243   if (i == 0 || i == 1)
244     {
245       queue_free(&q);
246       return;
247     }
248   qcnt = q.count;
249   for (i = 0; i < qcnt; i++)
250     {
251       /* we rely on the fact that blocks are ordered here.
252        * if we reach a positive element, we know that we
253        * saw all negative ones */
254       for (; (p = q.elements[i]) < 0; i++)
255         {
256           if (solv->decisionmap[-p] < 0)
257             break;
258           if (solv->decisionmap[-p] == 0)
259             queue_push(&q, -p);         /* undecided negative literal */
260         }
261       if (p <= 0)
262         {
263 #if 0
264           printf("complex dep block cannot be true or no pos literals\n");
265 #endif
266           while (q.elements[i])
267             i++;
268           if (qcnt != q.count)
269             queue_truncate(&q, qcnt);
270           continue;
271         }
272       if (qcnt == q.count)
273         {
274           /* all negative literals installed, add positive literals to map */
275           for (; (p = q.elements[i]) != 0; i++)
276             MAPSET(m, p);
277         }
278       else
279         {
280           /* at least one undecided negative literal, postpone */
281           int j, k;
282           Queue *cq;
283 #if 0
284           printf("add new complex dep block\n");
285           for (j = qcnt; j < q.count; j++)
286             printf("  - %s\n", pool_solvid2str(pool, q.elements[j]));
287 #endif
288           while (q.elements[i])
289             i++;
290           if (!(cq = *cqp))
291             {
292               cq = solv_calloc(1, sizeof(Queue));
293               queue_init(cq);
294               queue_insertn(cq, 0, 256, 0);     /* allocate hash area */
295               *cqp = cq;
296             }
297           for (j = qcnt; j < q.count; j++)
298             {
299               p = q.elements[j];
300               /* check if we already have this (dep, p) entry */
301               for (k = 256; k < cq->count; k += 2)
302                 if (cq->elements[k + 1] == dep && cq->elements[k] == p)
303                   break;
304               if (k == cq->count)
305                 {
306                   /* a new one. add to cq and hash */
307                   queue_push2(cq, p, dep);
308                   CPLXDEPHASH_SET(cq->elements, p);
309                 }
310             }
311           queue_truncate(&q, qcnt);
312         }
313     }
314   queue_free(&q);
315 }
316
317 static void
318 recheck_complex_deps(Solver *solv, Id p, Map *m, Queue **cqp)
319 {
320   Queue *cq = *cqp;
321   Id pp;
322   int i;
323 #if 0
324   printf("recheck_complex_deps for package %s\n", pool_solvid2str(solv->pool, p));
325 #endif
326   /* make sure that we don't have a false hit */
327   for (i = 256; i < cq->count; i += 2)
328     if (cq->elements[i] == p)
329       break;
330   if (i == cq->count)
331     return;     /* false alert */
332   if (solv->decisionmap[p] <= 0)
333     return;     /* just in case... */
334
335   /* rebuild the hash, call check_complex_dep for our package */
336   CPLXDEPHASH_EMPTY(cq->elements);
337   for (i = 256; i < cq->count; i += 2)
338     if ((pp = cq->elements[i]) == p)
339       {
340         Id dep = cq->elements[i + 1];
341         queue_deleten(cq, i, 2);
342         i -= 2;
343         check_complex_dep(solv, dep, m, &cq);
344       }
345     else
346       CPLXDEPHASH_SET(cq->elements, pp);
347 }
348
349 #endif
350
351
352 void
353 policy_update_recommendsmap(Solver *solv)
354 {
355   Pool *pool = solv->pool;
356   Solvable *s;
357   Id p, pp, rec, *recp, sug, *sugp;
358
359   if (solv->recommends_index < 0)
360     {
361       MAPZERO(&solv->recommendsmap);
362       MAPZERO(&solv->suggestsmap);
363 #ifdef ENABLE_COMPLEX_DEPS
364       if (solv->recommendscplxq)
365         {
366           queue_free(solv->recommendscplxq);
367           solv->recommendscplxq = solv_free(solv->recommendscplxq);
368         }
369       if (solv->suggestscplxq)
370         {
371           queue_free(solv->suggestscplxq);
372           solv->suggestscplxq = solv_free(solv->suggestscplxq);
373         }
374 #endif
375       solv->recommends_index = 0;
376     }
377   while (solv->recommends_index < solv->decisionq.count)
378     {
379       p = solv->decisionq.elements[solv->recommends_index++];
380       if (p < 0)
381         continue;
382       s = pool->solvables + p;
383 #ifdef ENABLE_COMPLEX_DEPS
384       /* re-check postponed complex blocks */
385       if (solv->recommendscplxq && CPLXDEPHASH_TST(solv->recommendscplxq->elements, p))
386         recheck_complex_deps(solv, p, &solv->recommendsmap, &solv->recommendscplxq);
387       if (solv->suggestscplxq && CPLXDEPHASH_TST(solv->suggestscplxq->elements, p))
388         recheck_complex_deps(solv, p, &solv->suggestsmap, &solv->suggestscplxq);
389 #endif
390       if (s->recommends)
391         {
392           recp = s->repo->idarraydata + s->recommends;
393           while ((rec = *recp++) != 0)
394             {
395 #ifdef ENABLE_COMPLEX_DEPS
396               if (pool_is_complex_dep(pool, rec))
397                 {
398                   check_complex_dep(solv, rec, &solv->recommendsmap, &solv->recommendscplxq);
399                   continue;
400                 }
401 #endif
402               FOR_PROVIDES(p, pp, rec)
403                 MAPSET(&solv->recommendsmap, p);
404             }
405         }
406       if (s->suggests)
407         {
408           sugp = s->repo->idarraydata + s->suggests;
409           while ((sug = *sugp++) != 0)
410             {
411 #ifdef ENABLE_COMPLEX_DEPS
412               if (pool_is_complex_dep(pool, sug))
413                 {
414                   check_complex_dep(solv, sug, &solv->suggestsmap, &solv->suggestscplxq);
415                   continue;
416                 }
417 #endif
418               FOR_PROVIDES(p, pp, sug)
419                 MAPSET(&solv->suggestsmap, p);
420             }
421         }
422     }
423 }
424
425 /* bring suggested/enhanced packages to front
426  * installed packages count as suggested */
427 static void
428 prefer_suggested(Solver *solv, Queue *plist)
429 {
430   Pool *pool = solv->pool;
431   int i, count;
432
433   /* update our recommendsmap/suggestsmap */
434   if (solv->recommends_index < solv->decisionq.count)
435     policy_update_recommendsmap(solv);
436
437   for (i = 0, count = plist->count; i < count; i++)
438     {
439       Id p = plist->elements[i];
440       Solvable *s = pool->solvables + p;
441       if ((pool->installed && s->repo == pool->installed) ||
442           MAPTST(&solv->suggestsmap, p) ||
443           solver_is_enhancing(solv, s))
444         continue;       /* good package */
445       /* bring to back */
446      if (i < plist->count - 1)
447         {
448           memmove(plist->elements + i, plist->elements + i + 1, (plist->count - 1 - i) * sizeof(Id));
449           plist->elements[plist->count - 1] = p;
450         }
451       i--;
452       count--;
453     }
454 }
455
456 static int
457 sort_by_favor_cmp(const void *ap, const void *bp, void *dp)
458 {
459   const Id *a = ap, *b = bp, *d = dp;
460   return d[b[0]] - d[a[0]];
461 }
462
463 /* bring favored packages to front and disfavored packages to back */
464 void
465 policy_prefer_favored(Solver *solv, Queue *plist)
466 {
467   if (solv->favormap && plist->count > 1)
468     solv_sort(plist->elements, plist->count, sizeof(Id), sort_by_favor_cmp, solv->favormap);
469 }
470
471 /*
472  * prune to recommended/suggested packages.
473  * does not prune installed packages (they are also somewhat recommended).
474  */
475 static void
476 prune_to_recommended(Solver *solv, Queue *plist)
477 {
478   Pool *pool = solv->pool;
479   int i, j, k, ninst;
480   Solvable *s;
481   Id p;
482
483   ninst = 0;
484   if (pool->installed)
485     {
486       for (i = 0; i < plist->count; i++)
487         {
488           p = plist->elements[i];
489           s = pool->solvables + p;
490           if (pool->installed && s->repo == pool->installed)
491             ninst++;
492         }
493     }
494   if (plist->count - ninst < 2)
495     return;
496
497   /* update our recommendsmap/suggestsmap */
498   if (solv->recommends_index < solv->decisionq.count)
499     policy_update_recommendsmap(solv);
500
501   /* prune to recommended/supplemented */
502   ninst = 0;
503   for (i = j = 0; i < plist->count; i++)
504     {
505       p = plist->elements[i];
506       s = pool->solvables + p;
507       if (pool->installed && s->repo == pool->installed)
508         {
509           ninst++;
510           if (j)
511             plist->elements[j++] = p;
512           continue;
513         }
514       if (!MAPTST(&solv->recommendsmap, p))
515         if (!solver_is_supplementing(solv, s))
516           continue;
517       if (!j && ninst)
518         {
519           for (k = 0; j < ninst; k++)
520             {
521               s = pool->solvables + plist->elements[k];
522               if (pool->installed && s->repo == pool->installed)
523                 plist->elements[j++] = plist->elements[k];
524             }
525         }
526       plist->elements[j++] = p;
527     }
528   if (j)
529     plist->count = j;
530
531 #if 0
532   /* anything left to prune? */
533   if (plist->count - ninst < 2)
534     return;
535
536   /* prune to suggested/enhanced */
537   ninst = 0;
538   for (i = j = 0; i < plist->count; i++)
539     {
540       p = plist->elements[i];
541       s = pool->solvables + p;
542       if (pool->installed && s->repo == pool->installed)
543         {
544           ninst++;
545           if (j)
546             plist->elements[j++] = p;
547           continue;
548         }
549       if (!MAPTST(&solv->suggestsmap, p))
550         if (!solver_is_enhancing(solv, s))
551           continue;
552       if (!j && ninst)
553         {
554           for (k = 0; j < ninst; k++)
555             {
556               s = pool->solvables + plist->elements[k];
557               if (pool->installed && s->repo == pool->installed)
558                 plist->elements[j++] = plist->elements[k];
559             }
560         }
561       plist->elements[j++] = p;
562     }
563   if (j)
564     plist->count = j;
565 #endif
566 }
567
568 static void
569 prune_to_best_arch(const Pool *pool, Queue *plist)
570 {
571   Id a, bestscore;
572   Solvable *s;
573   int i, j;
574
575   if (!pool->id2arch || plist->count < 2)
576     return;
577   bestscore = 0;
578   for (i = 0; i < plist->count; i++)
579     {
580       s = pool->solvables + plist->elements[i];
581       a = pool_arch2score(pool, s->arch);
582       if (a && a != 1 && (!bestscore || a < bestscore))
583         bestscore = a;
584     }
585   if (!bestscore)
586     return;
587   for (i = j = 0; i < plist->count; i++)
588     {
589       s = pool->solvables + plist->elements[i];
590       a = pool_arch2score(pool, s->arch);
591       if (!a)
592         continue;
593       /* a == 1 -> noarch */
594       if (a != 1 && ((a ^ bestscore) & 0xffff0000) != 0)
595         continue;
596       plist->elements[j++] = plist->elements[i];
597     }
598   if (j)
599     plist->count = j;
600 }
601
602
603 struct trj_data {
604   Pool *pool;
605   Queue *plist;
606   Id *stack;
607   Id nstack;
608   Id *low;
609   Id firstidx;
610   Id idx;
611 };
612
613 /* This is Tarjan's SCC algorithm, slightly modified */
614 static void
615 trj_visit(struct trj_data *trj, Id node)
616 {
617   Id *low = trj->low;
618   Pool *pool = trj->pool;
619   Queue *plist = trj->plist;
620   Id myidx, stackstart;
621   Solvable *s;
622   int i;
623   Id p, pp, obs, *obsp;
624
625   low[node] = myidx = trj->idx++;
626   trj->stack[(stackstart = trj->nstack++)] = node;
627
628   s = pool->solvables + plist->elements[node];
629   if (s->obsoletes)
630     {
631       obsp = s->repo->idarraydata + s->obsoletes;
632       while ((obs = *obsp++) != 0)
633         {
634           FOR_PROVIDES(p, pp, obs)
635             {
636               Solvable *ps = pool->solvables + p;
637               if (ps->name == s->name)
638                 continue;
639               if (!pool->obsoleteusesprovides && !pool_match_nevr(pool, ps, obs))
640                 continue;
641               if (pool->obsoleteusescolors && !pool_colormatch(pool, s, ps))
642                 continue;
643               /* hmm, expensive. should use hash if plist is big */
644               for (i = 0; i < plist->count; i++)
645                 {
646                   if (node != i && plist->elements[i] == p)
647                     {
648                       Id l = low[i];
649                       if (!l)
650                         {
651                           if (!ps->obsoletes)
652                             {
653                               /* don't bother */
654                               trj->idx++;
655                               low[i] = -1;
656                               continue;
657                             }
658                           trj_visit(trj, i);
659                           l = low[i];
660                         }
661                       if (l < 0)
662                         continue;
663                       if (l < trj->firstidx)
664                         {
665                           int k;
666                           /* this means we have reached an old SCC found earlier.
667                            * delete it as we obsolete it */
668                           for (k = l; ; k++)
669                             {
670                               if (low[trj->stack[k]] == l)
671                                 low[trj->stack[k]] = -1;
672                               else
673                                 break;
674                             }
675                         }
676                       else if (l < low[node])
677                         low[node] = l;
678                     }
679                 }
680             }
681         }
682     }
683   if (low[node] == myidx)       /* found a SCC? */
684     {
685       /* we're only interested in SCCs that contain the first node,
686        * as all others are "obsoleted" */
687       if (myidx != trj->firstidx)
688         myidx = -1;
689       for (i = stackstart; i < trj->nstack; i++)
690         low[trj->stack[i]] = myidx;
691       trj->nstack = stackstart; /* empty stack */
692     }
693 }
694
695 /*
696  * remove entries from plist that are obsoleted by other entries
697  * with different name.
698  */
699 static void
700 prune_obsoleted(Pool *pool, Queue *plist)
701 {
702   Id data_buf[2 * 16], *data;
703   struct trj_data trj;
704   int i, j;
705   Solvable *s;
706
707   if (plist->count <= 16)
708     {
709       memset(data_buf, 0, sizeof(data_buf));
710       data = data_buf;
711     }
712   else
713     data = solv_calloc(plist->count, 2 * sizeof(Id));
714   trj.pool = pool;
715   trj.plist = plist;
716   trj.low = data;
717   trj.idx = 1;
718   trj.stack = data + plist->count - 1;  /* -1 so we can index with idx (which starts with 1) */
719   for (i = 0; i < plist->count; i++)
720     {
721       if (trj.low[i])
722         continue;
723       s = pool->solvables + plist->elements[i];
724       if (s->obsoletes)
725         {
726           trj.firstidx = trj.nstack = trj.idx;
727           trj_visit(&trj, i);
728         }
729       else
730         {
731           Id myidx = trj.idx++;
732           trj.low[i] = myidx;
733           trj.stack[myidx] = i;
734         }
735     }
736   for (i = j = 0; i < plist->count; i++)
737     if (trj.low[i] >= 0)
738       plist->elements[j++] = plist->elements[i];
739   plist->count = j;
740   if (data != data_buf)
741     solv_free(data);
742 }
743
744 /* this is prune_obsoleted special-cased for two elements */
745 static void
746 prune_obsoleted_2(Pool *pool, Queue *plist)
747 {
748   int i;
749   Solvable *s;
750   Id p, pp, obs, *obsp;
751   Id other;
752   int obmap = 0;
753
754   for (i = 0; i < 2; i++)
755     {
756       s = pool->solvables + plist->elements[i];
757       other = plist->elements[1 - i];
758       if (s->obsoletes)
759         {
760           obsp = s->repo->idarraydata + s->obsoletes;
761           while ((obs = *obsp++) != 0)
762             {
763               FOR_PROVIDES(p, pp, obs)
764                 {
765                   Solvable *ps;
766                   if (p != other)
767                     continue;
768                   ps = pool->solvables + p;
769                   if (ps->name == s->name)
770                     continue;
771                   if (!pool->obsoleteusesprovides && !pool_match_nevr(pool, ps, obs))
772                     continue;
773                   if (pool->obsoleteusescolors && !pool_colormatch(pool, s, ps))
774                     continue;
775                   obmap |= 1 << i;
776                   break;
777                 }
778               if (p)
779                 break;
780             }
781         }
782     }
783   if (obmap == 0 || obmap == 3)
784     return;
785   if (obmap == 2)
786     plist->elements[0] = plist->elements[1];
787   plist->count = 1;
788 }
789
790 /*
791  * bring those elements to the front of the queue that
792  * have a installed solvable with the same name
793  */
794 static void
795 move_installed_to_front(Pool *pool, Queue *plist)
796 {
797   int i, j;
798   Solvable *s;
799   Id p, pp;
800
801   if (!pool->installed)
802     return;
803   for (i = j = 0; i < plist->count; i++)
804     {
805       s = pool->solvables + plist->elements[i];
806       if (s->repo != pool->installed)
807         {
808           FOR_PROVIDES(p, pp, s->name)
809             {
810               Solvable *ps = pool->solvables + p;
811               if (s->name == ps->name && ps->repo == pool->installed)
812                 {
813                   s = ps;
814                   break;
815                 }
816             }
817         }
818       if (s->repo == pool->installed)
819         {
820           if (i != j)
821             {
822               p = plist->elements[i];
823               if (i - j == 1)
824                 plist->elements[i] = plist->elements[j];
825               else
826                 memmove(plist->elements + j + 1, plist->elements + j, (i - j) * sizeof(Id));
827               plist->elements[j] = p;
828             }
829           else if (j + 2 == plist->count)
830             break;      /* no need to check last element if all prev ones are installed */
831           j++;
832         }
833     }
834 }
835
836 #ifdef ENABLE_CONDA
837 static int
838 pool_buildversioncmp(Pool *pool, Solvable *s1, Solvable *s2)
839 {
840   const char *bv1, *bv2;
841   unsigned int cnt1, cnt2;
842   cnt1 = solvable_lookup_count(s1, SOLVABLE_TRACK_FEATURES);
843   cnt2 = solvable_lookup_count(s2, SOLVABLE_TRACK_FEATURES);
844   if (cnt1 != cnt2)
845     return cnt1 > cnt2 ? -1 : 1;
846   bv1 = solvable_lookup_str(s1, SOLVABLE_BUILDVERSION);
847   bv2 = solvable_lookup_str(s2, SOLVABLE_BUILDVERSION);
848   if (!bv1 && !bv2)
849     return 0;
850   return pool_evrcmp_str(pool, bv1 ? bv1 : "" , bv2 ? bv2 : "", EVRCMP_COMPARE);
851 }
852
853 static int
854 pool_buildflavorcmp(Pool *pool, Solvable *s1, Solvable *s2)
855 {
856   const char *f1 = solvable_lookup_str(s1, SOLVABLE_BUILDFLAVOR);
857   const char *f2 = solvable_lookup_str(s2, SOLVABLE_BUILDFLAVOR);
858   if (!f1 && !f2)
859     return 0;
860   return pool_evrcmp_str(pool, f1 ? f1 : "" , f2 ? f2 : "", EVRCMP_COMPARE);
861 }
862 #endif
863
864 /*
865  * prune_to_best_version
866  *
867  * sort list of packages (given through plist) by name and evr
868  * return result through plist
869  */
870 void
871 prune_to_best_version(Pool *pool, Queue *plist)
872 {
873   int i, j, r;
874   Solvable *s, *best;
875
876   if (plist->count < 2)         /* no need to prune for a single entry */
877     return;
878   POOL_DEBUG(SOLV_DEBUG_POLICY, "prune_to_best_version %d\n", plist->count);
879
880   /* sort by name first, prefer installed */
881   solv_sort(plist->elements, plist->count, sizeof(Id), prune_to_best_version_sortcmp, pool);
882
883   /* now find best 'per name' */
884   best = 0;
885   for (i = j = 0; i < plist->count; i++)
886     {
887       s = pool->solvables + plist->elements[i];
888
889       POOL_DEBUG(SOLV_DEBUG_POLICY, "- %s [%d]%s\n",
890                  pool_solvable2str(pool, s), plist->elements[i], 
891                  (pool->installed && s->repo == pool->installed) ? "I" : "");
892
893       if (!best)                /* if no best yet, the current is best */
894         {
895           best = s;
896           continue;
897         }
898
899       /* name switch: finish group, re-init */
900       if (best->name != s->name)   /* new name */
901         {
902           plist->elements[j++] = best - pool->solvables; /* move old best to front */
903           best = s;             /* take current as new best */
904           continue;
905         }
906       r = best->evr != s->evr ? pool_evrcmp(pool, best->evr, s->evr, EVRCMP_COMPARE) : 0;
907 #ifdef ENABLE_LINKED_PKGS
908       if (r == 0 && has_package_link(pool, s))
909         r = pool_link_evrcmp(pool, best, s);
910 #endif
911 #ifdef ENABLE_CONDA
912       if (pool->disttype == DISTTYPE_CONDA)
913         {
914           if (r == 0)
915             r = (best->repo ? best->repo->subpriority : 0) - (s->repo ? s->repo->subpriority : 0);
916           if (r == 0)
917             r = pool_buildversioncmp(pool, best, s);
918           if (r == 0)
919             r = pool_buildflavorcmp(pool, best, s);
920         }
921 #endif
922       if (r < 0)
923         best = s;
924     }
925   plist->elements[j++] = best - pool->solvables;        /* finish last group */
926   plist->count = j;
927
928   /* we reduced the list to one package per name, now look at
929    * package obsoletes */
930   if (plist->count > 1)
931     {
932       if (plist->count == 2)
933         prune_obsoleted_2(pool, plist);
934       else
935         prune_obsoleted(pool, plist);
936     }
937 }
938
939
940 static int
941 sort_by_name_evr_sortcmp(const void *ap, const void *bp, void *dp)
942 {
943   Pool *pool = dp;
944   Id a, *aa = (Id *)ap;
945   Id b, *bb = (Id *)bp;
946   Id r = aa[1] - bb[1];
947   if (r)
948     return r < 0 ? -1 : 1;
949   if (aa[2] == bb[2])
950     return 0;
951   a = aa[2] < 0 ? -aa[2] : aa[2];
952   b = bb[2] < 0 ? -bb[2] : bb[2];
953   r = pool_evrcmp(pool, b, a, pool->disttype != DISTTYPE_DEB ? EVRCMP_MATCH_RELEASE : EVRCMP_COMPARE);
954   if (!r && (aa[2] < 0 || bb[2] < 0))
955     {
956       if (bb[2] >= 0)
957         return 1;
958       if (aa[2] >= 0)
959         return -1;
960     }
961   return r;
962 }
963
964 /* common end of sort_by_srcversion and sort_by_common_dep */
965 static void
966 sort_by_name_evr_array(Pool *pool, Queue *plist, int count, int ent)
967 {
968   Id lastname;
969   int i, j, bad, havebad;
970   Id *pp, *elements = plist->elements;
971
972   if (ent < 2)
973     {
974       queue_truncate(plist, count);
975       return;
976     }
977   solv_sort(elements + count * 2, ent, sizeof(Id) * 3, sort_by_name_evr_sortcmp, pool);
978   lastname = 0;
979   bad = havebad = 0;
980   for (i = 0, pp = elements + count * 2; i < ent; i++, pp += 3)
981     {
982       if (lastname && pp[1] == lastname)
983         {
984           if (pp[0] != pp[-3] && sort_by_name_evr_sortcmp(pp - 3, pp, pool) == -1)
985             {
986 #if 0
987               printf("%s - %s: bad %s %s - %s\n", pool_solvid2str(pool, elements[pp[-3]]), pool_solvid2str(pool, elements[pp[0]]), pool_dep2str(pool, lastname), pool_id2str(pool, pp[-1] < 0 ? -pp[-1] : pp[-1]), pool_id2str(pool, pp[2] < 0 ? -pp[2] : pp[2]));
988 #endif
989               bad++;
990               havebad = 1;
991             }
992         }
993       else
994         {
995           bad = 0;
996           lastname = pp[1];
997         }
998       elements[count + pp[0]] += bad;
999     }
1000
1001 #if 0
1002 for (i = 0; i < count; i++)
1003   printf("%s badness %d\n", pool_solvid2str(pool, elements[i]), elements[count + i]);
1004 #endif
1005
1006   if (havebad)
1007     {
1008       /* simple stable insertion sort */
1009       if (pool->installed)
1010         for (i = 0; i < count; i++)
1011           if (pool->solvables[elements[i]].repo == pool->installed)
1012             elements[i + count] = 0;
1013       for (i = 1; i < count; i++)
1014         for (j = i, pp = elements + count + j; j > 0; j--, pp--)
1015           if (pp[-1] > pp[0])
1016             {
1017               Id *pp2 = pp - count;
1018               Id p = pp[-1];
1019               pp[-1] = pp[0];
1020               pp[0] = p;
1021               p = pp2[-1];
1022               pp2[-1] = pp2[0];
1023               pp2[0] = p;
1024             }
1025           else
1026             break;
1027     }
1028   queue_truncate(plist, count);
1029 }
1030
1031 #if 0
1032 static void
1033 sort_by_srcversion(Pool *pool, Queue *plist)
1034 {
1035   int i, count = plist->count, ent = 0;
1036   queue_insertn(plist, count, count, 0);
1037   for (i = 0; i < count; i++)
1038     {
1039       Id name, evr, p = plist->elements[i];
1040       Solvable *s = pool->solvables + p;
1041       if (solvable_lookup_void(s, SOLVABLE_SOURCENAME))
1042         name = s->name;
1043       else
1044         name = solvable_lookup_id(s, SOLVABLE_SOURCENAME);
1045       if (solvable_lookup_void(s, SOLVABLE_SOURCEEVR))
1046         evr = s->evr;
1047       else
1048         evr = solvable_lookup_id(s, SOLVABLE_SOURCEEVR);
1049       if (!name || !evr || ISRELDEP(evr))
1050         continue;
1051       queue_push(plist, i);
1052       queue_push2(plist, name, evr);
1053       ent++;
1054     }
1055   sort_by_name_evr_array(pool, plist, count, ent);
1056 }
1057 #endif
1058
1059 static void
1060 sort_by_common_dep(Pool *pool, Queue *plist)
1061 {
1062   int i, count = plist->count, ent = 0;
1063   Id id, *dp;
1064   queue_insertn(plist, count, count, 0);
1065   for (i = 0; i < count; i++)
1066     {
1067       Id p = plist->elements[i];
1068       Solvable *s = pool->solvables + p;
1069       if (!s->provides)
1070         continue;
1071       for (dp = s->repo->idarraydata + s->provides; (id = *dp++) != 0; )
1072         {
1073           Reldep *rd;
1074           if (!ISRELDEP(id))
1075             continue;
1076           rd = GETRELDEP(pool, id);
1077           if ((rd->flags == REL_EQ || rd->flags == (REL_EQ | REL_LT) || rd->flags == REL_LT) && !ISRELDEP(rd->evr))
1078             {
1079               if (rd->flags == REL_EQ)
1080                 {
1081                   /* ignore hashes */
1082                   const char *s = pool_id2str(pool, rd->evr);
1083                   if (strlen(s) >= 4)
1084                     {
1085                       while ((*s >= 'a' && *s <= 'f') || (*s >= '0' && *s <= '9'))
1086                         s++;
1087                       if (!*s)
1088                         continue;
1089                     }
1090                 }
1091               queue_push(plist, i);
1092               queue_push2(plist, rd->name, rd->flags == REL_LT ? -rd->evr : rd->evr);
1093               ent++;
1094             }
1095         }
1096     }
1097   sort_by_name_evr_array(pool, plist, count, ent);
1098 }
1099
1100 /* check if we have an update candidate */
1101 static void
1102 dislike_old_versions(Pool *pool, Queue *plist)
1103 {
1104   int i, count;
1105
1106   for (i = 0, count = plist->count; i < count; i++)
1107     {
1108       Id p = plist->elements[i];
1109       Solvable *s = pool->solvables + p;
1110       Repo *repo = s->repo;
1111       Id q, qq;
1112       int bad = 0;
1113
1114       if (!repo || repo == pool->installed)
1115         continue;
1116       FOR_PROVIDES(q, qq, s->name)
1117         {
1118           Solvable *qs = pool->solvables + q;
1119           if (q == p)
1120             continue;
1121           if (s->name != qs->name || s->arch != qs->arch)
1122             continue;
1123           if (repo->priority != qs->repo->priority)
1124             {
1125               if (repo->priority > qs->repo->priority)
1126                 continue;
1127               bad = 1;
1128               break;
1129             }
1130           if (pool_evrcmp(pool, qs->evr, s->evr, EVRCMP_COMPARE) > 0)
1131             {
1132               bad = 1;
1133               break;
1134             }
1135         }
1136       if (!bad)
1137         continue;
1138       /* bring to back */
1139       if (i < plist->count - 1)
1140         {
1141           memmove(plist->elements + i, plist->elements + i + 1, (plist->count - 1 - i) * sizeof(Id));
1142           plist->elements[plist->count - 1] = p;
1143         }
1144       i--;
1145       count--;
1146     }
1147 }
1148
1149
1150 /* special lang package handling for urpm */
1151 /* see https://bugs.mageia.org/show_bug.cgi?id=18315 */
1152
1153 static int
1154 urpm_reorder_cmp(const void *ap, const void *bp, void *dp)
1155 {
1156   return ((Id *)bp)[1] - ((Id *)ap)[1];
1157 }
1158
1159 static void
1160 urpm_reorder(Solver *solv, Queue *plist)
1161 {
1162   Pool *pool = solv->pool;
1163   int i, count = plist->count;
1164   /* add locale score to packages */
1165   queue_insertn(plist, count, count, 0);
1166   for (i = count - 1; i >= 0; i--)
1167     {
1168       Solvable *s = pool->solvables + plist->elements[i];
1169       int score = 1;
1170       const char *sn = pool_id2str(pool, s->name);
1171
1172       if (!strncmp(sn, "kernel-", 7))
1173         {
1174           const char *devel = strstr(sn, "-devel-");
1175           if (devel && strlen(sn) < 256)
1176             {
1177               char kn[256];
1178               Id p, pp, knid;
1179               memcpy(kn, sn, devel - sn);
1180               strcpy(kn + (devel - sn), devel + 6);
1181               knid = pool_str2id(pool, kn, 0);
1182               if (knid)
1183                 {
1184                   FOR_PROVIDES(p, pp, knid)
1185                     {
1186                       if (solv->decisionmap[p] > 0)
1187                         {
1188                           score = 4;
1189                           break;
1190                         }
1191                       else if (pool->installed && pool->solvables[p].repo == pool->installed)
1192                         score = 3;
1193                     }
1194                 }
1195             }
1196         }
1197       else if ((sn = strstr(sn, "-kernel-")) != 0)
1198         {
1199           sn += 8;
1200           if (strlen(sn) < 256 - 8 && *sn >= '0' && *sn <= '9' && sn[1] == '.')
1201             {
1202               const char *flavor = strchr(sn, '-');
1203               if (flavor)
1204                 {
1205                   const char *release = strchr(flavor + 1, '-');
1206                   if (release)
1207                     {
1208                       char kn[256];
1209                       Id p, pp, knid;
1210                       memcpy(kn, "kernel", 7);
1211                       memcpy(kn + 6, flavor, release - flavor + 1);
1212                       memcpy(kn + 6 + (release - flavor) + 1, sn, flavor - sn);
1213                       strcpy(kn + 6 + (release + 1 - sn), release);
1214                       knid = pool_str2id(pool, kn, 0);
1215                       if (knid)
1216                         {
1217                           FOR_PROVIDES(p, pp, knid)
1218                             {
1219                               if (solv->decisionmap[p] > 0)
1220                                 {
1221                                   score = 4;
1222                                   break;
1223                                 }
1224                               if (pool->installed && pool->solvables[p].repo == pool->installed)
1225                                 score = 3;
1226                             }
1227                         }
1228                     }
1229                 }
1230             }
1231         }
1232       if (score == 1 && s->requires)
1233         {
1234           Id id, *idp, p, pp;
1235           const char *deps;
1236           for (idp = s->repo->idarraydata + s->requires; (id = *idp) != 0; idp++)
1237             {
1238               while (ISRELDEP(id))
1239                 {
1240                   Reldep *rd = GETRELDEP(pool, id);
1241                   id = rd->name;
1242                 }
1243               deps = strstr(pool_id2str(pool, id), "locales-");
1244               if (!deps)
1245                 continue;
1246               if (!strncmp(deps + 8, "en", 2))
1247                 score = 2;
1248               else
1249                 {
1250                   score = 0;
1251                   FOR_PROVIDES(p, pp, id)
1252                     {
1253                       if (solv->decisionmap[p] > 0)
1254                         {
1255                           score = 4;
1256                           break;
1257                         }
1258                       if (pool->installed && pool->solvables[p].repo == pool->installed)
1259                         score = 3;
1260                     }
1261                   break;
1262                 }
1263             }
1264         }
1265       plist->elements[i * 2] = plist->elements[i];
1266       plist->elements[i * 2 + 1] = score;
1267     }
1268   solv_sort(plist->elements, count, sizeof(Id) * 2, urpm_reorder_cmp, pool);
1269   for (i = 0; i < count; i++)
1270     plist->elements[i] = plist->elements[2 * i];
1271   queue_truncate(plist, count);
1272 }
1273
1274 /* support multiple favor groups by calling policy_filter_unwanted on
1275  * each of them and combining the result */
1276 static void
1277 policy_filter_unwanted_favored(Solver *solv, Queue *plist, int mode)
1278 {
1279   int i, j, f;
1280   Queue qin, qprune;
1281   queue_init_clone(&qin, plist);
1282   queue_empty(plist);
1283   /* sort by favor group */
1284   solv_sort(qin.elements, qin.count, sizeof(Id), sort_by_favor_cmp, solv->favormap);
1285   /* go over groups */
1286   queue_init(&qprune);
1287   for (i = 0; i < qin.count; i = j)
1288     {
1289       /* find end of group */
1290       f = solv->favormap[qin.elements[i]];
1291       for (j = i + 1; j < qin.count; j++)
1292         if (solv->favormap[qin.elements[j]] != f)
1293           break;
1294       /* prune this group */
1295       queue_empty(&qprune);
1296       queue_insertn(&qprune, 0, j, qin.elements);
1297       policy_filter_unwanted(solv, &qprune, mode | POLICY_MODE_FAVOR_REC);
1298       for (i = 0; i < qprune.count; i++)
1299         if (solv->favormap[qprune.elements[i]] == f)
1300           queue_push(plist, qprune.elements[i]);
1301     }
1302   queue_free(&qprune);
1303   queue_free(&qin);
1304 }
1305
1306 /*
1307  *  POLICY_MODE_CHOOSE:     default, do all pruning steps
1308  *  POLICY_MODE_RECOMMEND:  leave out prune_to_recommended
1309  *  POLICY_MODE_SUGGEST:    leave out prune_to_recommended, do prio pruning just per name
1310  */
1311 void
1312 policy_filter_unwanted(Solver *solv, Queue *plist, int mode)
1313 {
1314   Pool *pool = solv->pool;
1315   if (mode == POLICY_MODE_SUPPLEMENT)
1316     {
1317       /* reorder only */
1318       dislike_old_versions(pool, plist);
1319       sort_by_common_dep(pool, plist);
1320       if (solv->urpmreorder)
1321         urpm_reorder(solv, plist);
1322       prefer_suggested(solv, plist);
1323       policy_prefer_favored(solv, plist);
1324       return;
1325     }
1326   if (mode & POLICY_MODE_FAVOR_REC)
1327     mode ^= POLICY_MODE_FAVOR_REC;
1328   else if (solv->favormap && plist->count > 1)
1329     {
1330       /* check if we have multiple favor groups */
1331       int i, f = solv->favormap[plist->elements[0]];
1332       for (i = 1; i < plist->count; i++)
1333         if (solv->favormap[plist->elements[i]] != f)
1334           break;
1335       if (i < plist->count)
1336         {
1337           policy_filter_unwanted_favored(solv, plist, mode);
1338           return;
1339         }
1340     }
1341   if (plist->count > 1)
1342     {
1343       if (mode != POLICY_MODE_SUGGEST)
1344         solver_prune_to_highest_prio(solv, plist);
1345       else
1346         solver_prune_to_highest_prio_per_name(solv, plist);
1347     }
1348   if (plist->count > 1)
1349     prune_to_best_arch(pool, plist);
1350   if (plist->count > 1)
1351     prune_to_best_version(pool, plist);
1352   if (plist->count > 1 && (mode == POLICY_MODE_CHOOSE || mode == POLICY_MODE_CHOOSE_NOREORDER))
1353     {
1354       prune_to_recommended(solv, plist);
1355       if (plist->count > 1 && mode != POLICY_MODE_CHOOSE_NOREORDER)
1356         {
1357           /* do some fancy reordering */
1358 #if 0
1359           sort_by_srcversion(pool, plist);
1360 #endif
1361           dislike_old_versions(pool, plist);
1362           sort_by_common_dep(pool, plist);
1363           move_installed_to_front(pool, plist);
1364           if (solv->urpmreorder)
1365             urpm_reorder(solv, plist);
1366           prefer_suggested(solv, plist);
1367           policy_prefer_favored(solv, plist);
1368         }
1369     }
1370 }
1371
1372 void
1373 pool_best_solvables(Pool *pool, Queue *plist, int flags)
1374 {
1375   if (plist->count > 1)
1376     prune_to_highest_prio(pool, plist);
1377   if (plist->count > 1)
1378     prune_to_best_arch(pool, plist);
1379   if (plist->count > 1)
1380     prune_to_best_version(pool, plist);
1381   if (plist->count > 1)
1382     {
1383       dislike_old_versions(pool, plist);
1384       sort_by_common_dep(pool, plist);
1385       move_installed_to_front(pool, plist);
1386     }
1387 }
1388
1389
1390 /* check if there is an illegal architecture change if
1391  * installed solvable s1 is replaced by s2 */
1392 int
1393 policy_illegal_archchange(Solver *solv, Solvable *s1, Solvable *s2)
1394 {
1395   Pool *pool = solv->pool;
1396   Id a1 = s1->arch, a2 = s2->arch;
1397
1398   /* we allow changes to/from noarch */
1399   if (a1 == a2 || a1 == pool->noarchid || a2 == pool->noarchid)
1400     return 0;
1401   if (!pool->id2arch)
1402     return 0;
1403   a1 = pool_arch2score(pool, a1);
1404   a2 = pool_arch2score(pool, a2);
1405   if (((a1 ^ a2) & 0xffff0000) != 0)
1406     return 1;
1407   return 0;
1408 }
1409
1410 /* check if there is an illegal vendor change if
1411  * installed solvable s1 is replaced by s2 */
1412 int
1413 policy_illegal_vendorchange(Solver *solv, Solvable *s1, Solvable *s2)
1414 {
1415   Pool *pool = solv->pool;
1416   Id v1, v2;
1417   Id vendormask1, vendormask2;
1418
1419   if (pool->custom_vendorcheck)
1420      return pool->custom_vendorcheck(pool, s1, s2);
1421
1422   /* treat a missing vendor as empty string */
1423   v1 = s1->vendor ? s1->vendor : ID_EMPTY;
1424   v2 = s2->vendor ? s2->vendor : ID_EMPTY;
1425   if (v1 == v2)
1426     return 0;
1427   vendormask1 = pool_vendor2mask(pool, v1);
1428   if (!vendormask1)
1429     return 1;   /* can't match */
1430   vendormask2 = pool_vendor2mask(pool, v2);
1431   if ((vendormask1 & vendormask2) != 0)
1432     return 0;
1433   return 1;     /* no class matches */
1434 }
1435
1436 /* check if it is illegal to replace installed
1437  * package "is" with package "s" (which must obsolete "is")
1438  */
1439 int
1440 policy_is_illegal(Solver *solv, Solvable *is, Solvable *s, int ignore)
1441 {
1442   Pool *pool = solv->pool;
1443   int ret = 0;
1444   int duppkg = solv->dupinvolvedmap_all || (solv->dupinvolvedmap.size && MAPTST(&solv->dupinvolvedmap, is - pool->solvables));
1445   if (!(ignore & POLICY_ILLEGAL_DOWNGRADE) && !(duppkg ? solv->dup_allowdowngrade : solv->allowdowngrade))
1446     {
1447       if (is->name == s->name && pool_evrcmp(pool, is->evr, s->evr, EVRCMP_COMPARE) > 0)
1448         ret |= POLICY_ILLEGAL_DOWNGRADE;
1449     }
1450   if (!(ignore & POLICY_ILLEGAL_ARCHCHANGE) && !(duppkg ? solv->dup_allowarchchange : solv->allowarchchange))
1451     {
1452       if (is->arch != s->arch && policy_illegal_archchange(solv, is, s))
1453         ret |= POLICY_ILLEGAL_ARCHCHANGE;
1454     }
1455   if (!(ignore & POLICY_ILLEGAL_VENDORCHANGE) && !(duppkg ? solv->dup_allowvendorchange : solv->allowvendorchange))
1456     {
1457       if (is->vendor != s->vendor && policy_illegal_vendorchange(solv, is, s))
1458         ret |= POLICY_ILLEGAL_VENDORCHANGE;
1459     }
1460   if (!(ignore & POLICY_ILLEGAL_NAMECHANGE) && !(duppkg ? solv->dup_allownamechange : solv->allownamechange))
1461     {
1462       if (is->name != s->name)
1463         ret |= POLICY_ILLEGAL_NAMECHANGE;
1464     }
1465   return ret;
1466 }
1467
1468 /*-------------------------------------------------------------------
1469  *
1470  * create reverse obsoletes map for installed solvables
1471  *
1472  * For each installed solvable find which packages with *different* names
1473  * obsolete the solvable.
1474  * This index is used in policy_findupdatepackages() below.
1475  */
1476 void
1477 policy_create_obsolete_index(Solver *solv)
1478 {
1479   Pool *pool = solv->pool;
1480   Solvable *s;
1481   Repo *installed = solv->installed;
1482   Id p, pp, obs, *obsp, *obsoletes, *obsoletes_data;
1483   int i, n, cnt;
1484
1485   solv->obsoletes = solv_free(solv->obsoletes);
1486   solv->obsoletes_data = solv_free(solv->obsoletes_data);
1487   if (!installed || installed->start == installed->end)
1488     return;
1489   cnt = installed->end - installed->start;
1490   solv->obsoletes = obsoletes = solv_calloc(cnt, sizeof(Id));
1491   for (i = 1; i < pool->nsolvables; i++)
1492     {
1493       s = pool->solvables + i;
1494       if (!s->obsoletes)
1495         continue;
1496       if (!pool_installable(pool, s))
1497         continue;
1498       obsp = s->repo->idarraydata + s->obsoletes;
1499       while ((obs = *obsp++) != 0)
1500         {
1501           FOR_PROVIDES(p, pp, obs)
1502             {
1503               Solvable *ps = pool->solvables + p;;
1504               if (ps->repo != installed)
1505                 continue;
1506               if (ps->name == s->name)
1507                 continue;
1508               if (!pool->obsoleteusesprovides && !pool_match_nevr(pool, ps, obs))
1509                 continue;
1510               if (pool->obsoleteusescolors && !pool_colormatch(pool, s, ps))
1511                 continue;
1512               obsoletes[p - installed->start]++;
1513             }
1514         }
1515     }
1516   n = 0;
1517   for (i = 0; i < cnt; i++)
1518     if (obsoletes[i])
1519       {
1520         n += obsoletes[i] + 1;
1521         obsoletes[i] = n;
1522       }
1523   solv->obsoletes_data = obsoletes_data = solv_calloc(n + 1, sizeof(Id));
1524   POOL_DEBUG(SOLV_DEBUG_STATS, "obsoletes data: %d entries\n", n + 1);
1525   for (i = pool->nsolvables - 1; i > 0; i--)
1526     {
1527       s = pool->solvables + i;
1528       if (!s->obsoletes)
1529         continue;
1530       if (!pool_installable(pool, s))
1531         continue;
1532       obsp = s->repo->idarraydata + s->obsoletes;
1533       while ((obs = *obsp++) != 0)
1534         {
1535           FOR_PROVIDES(p, pp, obs)
1536             {
1537               Solvable *ps = pool->solvables + p;;
1538               if (ps->repo != installed)
1539                 continue;
1540               if (ps->name == s->name)
1541                 continue;
1542               if (!pool->obsoleteusesprovides && !pool_match_nevr(pool, ps, obs))
1543                 continue;
1544               if (pool->obsoleteusescolors && !pool_colormatch(pool, s, ps))
1545                 continue;
1546               if (obsoletes_data[obsoletes[p - installed->start]] != i)
1547                 obsoletes_data[--obsoletes[p - installed->start]] = i;
1548             }
1549         }
1550     }
1551 }
1552
1553
1554 /*
1555  * find update candidates
1556  *
1557  * s: installed solvable to be updated
1558  * qs: [out] queue to hold Ids of candidates
1559  * allow_all: 0 = dont allow downgrades, 1 = allow all candidates
1560  *            2 = dup mode
1561  *
1562  */
1563 void
1564 policy_findupdatepackages(Solver *solv, Solvable *s, Queue *qs, int allow_all)
1565 {
1566   /* installed packages get a special upgrade allowed rule */
1567   Pool *pool = solv->pool;
1568   Id p, pp, n, p2, pp2;
1569   Id obs, *obsp;
1570   Solvable *ps;
1571   int haveprovobs = 0;
1572   int allowdowngrade = allow_all ? 1 : solv->allowdowngrade;
1573   int allownamechange = allow_all ? 1 : solv->allownamechange;
1574   int allowarchchange = allow_all ? 1 : solv->allowarchchange;
1575   int allowvendorchange = allow_all ? 1 : solv->allowvendorchange;
1576   if (allow_all == 2)
1577     {
1578       allowdowngrade = solv->dup_allowdowngrade;
1579       allownamechange = solv->dup_allownamechange;
1580       allowarchchange = solv->dup_allowarchchange;
1581       allowvendorchange = solv->dup_allowvendorchange;
1582     }
1583
1584   queue_empty(qs);
1585
1586   n = s - pool->solvables;
1587
1588   /*
1589    * look for updates for s
1590    */
1591   FOR_PROVIDES(p, pp, s->name)  /* every provider of s' name */
1592     {
1593       if (p == n)               /* skip itself */
1594         continue;
1595
1596       ps = pool->solvables + p;
1597       if (pool->considered && pool->whatprovideswithdisabled && ps->repo != pool->installed && pool_disabled_solvable(pool, ps)) 
1598         continue;
1599       if (s->name == ps->name)  /* name match */
1600         {
1601           if (pool->implicitobsoleteusescolors && !pool_colormatch(pool, s, ps))
1602             continue;
1603           if (!allowdowngrade && pool_evrcmp(pool, s->evr, ps->evr, EVRCMP_COMPARE) > 0)
1604             continue;
1605         }
1606       else if (!allownamechange)
1607         continue;
1608       else if ((!solv->noupdateprovide || solv->needupdateprovide) && ps->obsoletes)   /* provides/obsoletes combination ? */
1609         {
1610           /* check if package ps obsoletes installed package s */
1611           /* implicitobsoleteusescolors is somewhat wrong here, but we nevertheless
1612            * use it to limit our update candidates */
1613           if ((pool->obsoleteusescolors || pool->implicitobsoleteusescolors) && !pool_colormatch(pool, s, ps))
1614             continue;
1615           obsp = ps->repo->idarraydata + ps->obsoletes;
1616           while ((obs = *obsp++) != 0)  /* for all obsoletes */
1617             {
1618               FOR_PROVIDES(p2, pp2, obs)   /* and all matching providers of the obsoletes */
1619                 {
1620                   Solvable *ps2 = pool->solvables + p2;
1621                   if (!pool->obsoleteusesprovides && !pool_match_nevr(pool, ps2, obs))
1622                     continue;
1623                   if (p2 == n)          /* match ! */
1624                     break;
1625                 }
1626               if (p2)                   /* match! */
1627                 break;
1628             }
1629           if (!obs)                     /* continue if no match */
1630             continue;
1631           /* here we have 'p' with a matching provides/obsoletes combination
1632            * thus flagging p as a valid update candidate for s
1633            */
1634           haveprovobs = 1;
1635         }
1636       else
1637         continue;
1638       if (!allowarchchange && s->arch != ps->arch && policy_illegal_archchange(solv, s, ps))
1639         continue;
1640       if (!allowvendorchange && s->vendor != ps->vendor && policy_illegal_vendorchange(solv, s, ps))
1641         continue;
1642       queue_push(qs, p);
1643     }
1644   if (!allownamechange)
1645     return;
1646   /* if we have found some valid candidates and noupdateprovide is not set, we're
1647      done. otherwise we fallback to all obsoletes */
1648   if (solv->needupdateprovide || (!solv->noupdateprovide && haveprovobs))
1649     return;
1650   if (solv->obsoletes && solv->obsoletes[n - solv->installed->start])
1651     {
1652       Id *opp;
1653       for (opp = solv->obsoletes_data + solv->obsoletes[n - solv->installed->start]; (p = *opp++) != 0;)
1654         {
1655           ps = pool->solvables + p;
1656           if (!allowarchchange && s->arch != ps->arch && policy_illegal_archchange(solv, s, ps))
1657             continue;
1658           if (!allowvendorchange && s->vendor != ps->vendor && policy_illegal_vendorchange(solv, s, ps))
1659             continue;
1660           /* implicitobsoleteusescolors is somewhat wrong here, but we nevertheless
1661            * use it to limit our update candidates */
1662           if (pool->implicitobsoleteusescolors && !pool_colormatch(pool, s, ps))
1663             continue;
1664           queue_push(qs, p);
1665         }
1666     }
1667 }
1668
1669 const char *
1670 policy_illegal2str(Solver *solv, int illegal, Solvable *s, Solvable *rs)
1671 {
1672   Pool *pool = solv->pool;
1673   const char *str;
1674   if (illegal == POLICY_ILLEGAL_DOWNGRADE)
1675     {
1676       str = pool_tmpjoin(pool, "downgrade of ", pool_solvable2str(pool, s), 0);
1677       return pool_tmpappend(pool, str, " to ", pool_solvable2str(pool, rs));
1678     }
1679   if (illegal == POLICY_ILLEGAL_NAMECHANGE)
1680     {
1681       str = pool_tmpjoin(pool, "name change of ", pool_solvable2str(pool, s), 0);
1682       return pool_tmpappend(pool, str, " to ", pool_solvable2str(pool, rs));
1683     }
1684   if (illegal == POLICY_ILLEGAL_ARCHCHANGE)
1685     {
1686       str = pool_tmpjoin(pool, "architecture change of ", pool_solvable2str(pool, s), 0);
1687       return pool_tmpappend(pool, str, " to ", pool_solvable2str(pool, rs));
1688     }
1689   if (illegal == POLICY_ILLEGAL_VENDORCHANGE)
1690     {
1691       str = pool_tmpjoin(pool, "vendor change from '", pool_id2str(pool, s->vendor), "' (");
1692       if (rs->vendor)
1693         {
1694           str = pool_tmpappend(pool, str, pool_solvable2str(pool, s), ") to '");
1695           str = pool_tmpappend(pool, str, pool_id2str(pool, rs->vendor), "' (");
1696         }
1697       else
1698         str = pool_tmpappend(pool, str, pool_solvable2str(pool, s), ") to no vendor (");
1699       return pool_tmpappend(pool, str, pool_solvable2str(pool, rs), ")");
1700     }
1701   return "unknown illegal change";
1702 }
1703