- make POLICY_MODE_SUGGEST obey repo priorities
[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 "evr.h"
20 #include "policy.h"
21 #include "poolvendor.h"
22 #include "poolarch.h"
23
24
25 /*-----------------------------------------------------------------*/
26
27 /*
28  * prep for prune_best_version
29  *   sort by name
30  */
31
32 static int
33 prune_to_best_version_sortcmp(const void *ap, const void *bp, void *dp)
34 {
35   Pool *pool = dp;
36   int r;
37   Id a = *(Id *)ap;
38   Id b = *(Id *)bp;
39   Solvable *sa, *sb;
40
41   sa = pool->solvables + a;
42   sb = pool->solvables + b;
43   r = sa->name - sb->name;
44   if (r)
45     {
46       const char *na, *nb;
47       /* different names. We use real strcmp here so that the result
48        * is not depending on some random solvable order */
49       na = id2str(pool, sa->name);
50       nb = id2str(pool, sb->name);
51       return strcmp(na, nb);
52     }
53   /* the same name, bring installed solvables to the front */
54   if (pool->installed)
55     {
56       if (sa->repo == pool->installed)
57         {
58           if (sb->repo != pool->installed)
59             return -1;
60         }
61       else if (sb->repo == pool->installed)
62         return 1;       
63     }
64   /* sort by repository sub-prio (installed repo handled above) */
65   r = (sb->repo ? sb->repo->subpriority : 0) - (sa->repo ? sa->repo->subpriority : 0);
66   if (r)
67     return r;
68   /* no idea about the order, sort by id */
69   return a - b;
70 }
71
72
73 /*
74  * prune to repository with highest priority.
75  * does not prune installed solvables.
76  */
77
78 static void
79 prune_to_highest_prio(Pool *pool, Queue *plist)
80 {
81   int i, j;
82   Solvable *s;
83   int bestprio = 0, bestprioset = 0;
84
85   /* prune to highest priority */
86   for (i = 0; i < plist->count; i++)  /* find highest prio in queue */
87     {
88       s = pool->solvables + plist->elements[i];
89       if (pool->installed && s->repo == pool->installed)
90         continue;
91       if (!bestprioset || s->repo->priority > bestprio)
92         {
93           bestprio = s->repo->priority;
94           bestprioset = 1;
95         }
96     }
97   if (!bestprioset)
98     return;
99   for (i = j = 0; i < plist->count; i++) /* remove all with lower prio */
100     {
101       s = pool->solvables + plist->elements[i];
102       if (s->repo->priority == bestprio || (pool->installed && s->repo == pool->installed))
103         plist->elements[j++] = plist->elements[i];
104     }
105   plist->count = j;
106 }
107
108 static void
109 prune_to_highest_prio_per_name(Pool *pool, Queue *plist)
110 {
111   Queue pq;
112   int i, j, k;
113   Id name;
114
115   queue_init(&pq);
116   sat_sort(plist->elements, plist->count, sizeof(Id), prune_to_best_version_sortcmp, pool);
117   queue_push(&pq, plist->elements[0]);
118   name = pool->solvables[pq.elements[0]].name;
119   for (i = 1, j = 0; i < plist->count; i++)
120     {
121       if (pool->solvables[plist->elements[i]].name != name)
122         {
123           if (pq.count > 2)
124             prune_to_highest_prio(pool, &pq);
125           for (k = 0; k < pq.count; k++)
126             plist->elements[j++] = pq.elements[k];
127           queue_empty(&pq);
128           queue_push(&pq, plist->elements[i]);
129           name = pool->solvables[pq.elements[0]].name;
130         }
131     }
132   if (pq.count > 2)
133     prune_to_highest_prio(pool, &pq);
134   for (k = 0; k < pq.count; k++)
135     plist->elements[j++] = pq.elements[k];
136   queue_free(&pq);
137   plist->count = j;
138 }
139
140
141 /*
142  * prune to recommended/suggested packages.
143  * does not prune installed packages (they are also somewhat recommended).
144  */
145
146 static void
147 prune_to_recommended(Solver *solv, Queue *plist)
148 {
149   Pool *pool = solv->pool;
150   int i, j, k, ninst;
151   Solvable *s;
152   Id p, pp, rec, *recp, sug, *sugp;
153
154   ninst = 0;
155   if (pool->installed)
156     {
157       for (i = 0; i < plist->count; i++)
158         {
159           p = plist->elements[i];
160           s = pool->solvables + p;
161           if (pool->installed && s->repo == pool->installed)
162             ninst++;
163         }
164     }
165   if (plist->count - ninst < 2)
166     return;
167
168   /* update our recommendsmap/suggestsmap */
169   if (solv->recommends_index < 0)
170     {
171       MAPZERO(&solv->recommendsmap);
172       MAPZERO(&solv->suggestsmap);
173       solv->recommends_index = 0;
174     }
175   while (solv->recommends_index < solv->decisionq.count)
176     {
177       p = solv->decisionq.elements[solv->recommends_index++];
178       if (p < 0)
179         continue;
180       s = pool->solvables + p;
181       if (s->recommends)
182         {
183           recp = s->repo->idarraydata + s->recommends;
184           while ((rec = *recp++) != 0)
185             FOR_PROVIDES(p, pp, rec)
186               MAPSET(&solv->recommendsmap, p);
187         }
188       if (s->suggests)
189         {
190           sugp = s->repo->idarraydata + s->suggests;
191           while ((sug = *sugp++) != 0)
192             FOR_PROVIDES(p, pp, sug)
193               MAPSET(&solv->suggestsmap, p);
194         }
195     }
196
197   /* prune to recommended/supplemented */
198   ninst = 0;
199   for (i = j = 0; i < plist->count; i++)
200     {
201       p = plist->elements[i];
202       s = pool->solvables + p;
203       if (pool->installed && s->repo == pool->installed)
204         {
205           ninst++;
206           if (j)
207             plist->elements[j++] = p;
208           continue;
209         }
210       if (!MAPTST(&solv->recommendsmap, p))
211         if (!solver_is_supplementing(solv, s))
212           continue;
213       if (!j && ninst)
214         {
215           for (k = 0; j < ninst; k++)
216             {
217               s = pool->solvables + plist->elements[k];
218               if (pool->installed && s->repo == pool->installed)
219                 plist->elements[j++] = plist->elements[k];
220             }
221         }
222       plist->elements[j++] = p;
223     }
224   if (j)
225     plist->count = j;
226
227   /* anything left to prune? */
228   if (plist->count - ninst < 2)
229     return;
230
231   /* prune to suggested/enhanced*/
232   ninst = 0;
233   for (i = j = 0; i < plist->count; i++)
234     {
235       p = plist->elements[i];
236       s = pool->solvables + p;
237       if (pool->installed && s->repo == pool->installed)
238         {
239           ninst++;
240           if (j)
241             plist->elements[j++] = p;
242           continue;
243         }
244       if (!MAPTST(&solv->suggestsmap, p))
245         if (!solver_is_enhancing(solv, s))
246           continue;
247       if (!j && ninst)
248         {
249           for (k = 0; j < ninst; k++)
250             {
251               s = pool->solvables + plist->elements[k];
252               if (pool->installed && s->repo == pool->installed)
253                 plist->elements[j++] = plist->elements[k];
254             }
255         }
256       plist->elements[j++] = p;
257     }
258   if (j)
259     plist->count = j;
260 }
261
262 void
263 prune_to_best_arch(const Pool *pool, Queue *plist)
264 {
265   Id a, bestscore;
266   Solvable *s;
267   int i, j;
268
269   if (!pool->id2arch || plist->count < 2)
270     return;
271   bestscore = 0;
272   for (i = 0; i < plist->count; i++)
273     {
274       s = pool->solvables + plist->elements[i];
275       a = s->arch;
276       a = (a <= pool->lastarch) ? pool->id2arch[a] : 0;
277       if (a && a != 1 && (!bestscore || a < bestscore))
278         bestscore = a;
279     }
280   for (i = j = 0; i < plist->count; i++)
281     {
282       s = pool->solvables + plist->elements[i];
283       a = s->arch;
284       if (a > pool->lastarch)
285         continue;
286       a = pool->id2arch[a];
287       /* a == 1 -> noarch */
288       if (a != 1 && ((a ^ bestscore) & 0xffff0000) != 0)
289         continue;
290       plist->elements[j++] = plist->elements[i];
291     }
292   if (j)
293     plist->count = j;
294 }
295
296 /*
297  * prune_to_best_version
298  *
299  * sort list of packages (given through plist) by name and evr
300  * return result through plist
301  */
302 void
303 prune_to_best_version(Pool *pool, Queue *plist)
304 {
305   int i, j;
306   Solvable *s, *best;
307
308   if (plist->count < 2)         /* no need to prune for a single entry */
309     return;
310   POOL_DEBUG(SAT_DEBUG_POLICY, "prune_to_best_version %d\n", plist->count);
311
312   /* sort by name first, prefer installed */
313   sat_sort(plist->elements, plist->count, sizeof(Id), prune_to_best_version_sortcmp, pool);
314
315   /* delete obsoleted. hmm, looks expensive! */
316   /* FIXME maybe also check provides depending on noupdateprovide? */
317   /* FIXME do not prune cycles */
318   for (i = 0; i < plist->count; i++)
319     {
320       Id p, pp, obs, *obsp;
321       s = pool->solvables + plist->elements[i];
322       if (!s->obsoletes)
323         continue;
324       obsp = s->repo->idarraydata + s->obsoletes;
325       while ((obs = *obsp++) != 0)
326         {
327           FOR_PROVIDES(p, pp, obs)
328             {
329               Solvable *ps = pool->solvables + p;
330               if (ps->name == s->name)
331                 continue;
332               if (!pool->obsoleteusesprovides && !pool_match_nevr(pool, ps, obs))
333                 continue;
334               if (pool->obsoleteusescolors && !pool_colormatch(pool, s, ps))
335                 continue;
336               for (j = 0; j < plist->count; j++)
337                 {
338                   if (i == j)
339                     continue;
340                   if (plist->elements[j] == p)
341                     plist->elements[j] = 0;
342                 }
343             }
344         }
345     }
346   /* delete zeroed out queue entries */
347   for (i = j = 0; i < plist->count; i++)
348     if (plist->elements[i])
349       plist->elements[j++] = plist->elements[i];
350   plist->count = j;
351
352   if (plist->count < 2)
353     return;
354
355   /* now find best 'per name' */
356   best = 0;
357   for (i = j = 0; i < plist->count; i++)
358     {
359       s = pool->solvables + plist->elements[i];
360
361       POOL_DEBUG(SAT_DEBUG_POLICY, "- %s[%s]\n",
362                  solvable2str(pool, s),
363                  (pool->installed && s->repo == pool->installed) ? "installed" : "not installed");
364
365       if (!best)                /* if no best yet, the current is best */
366         {
367           best = s;
368           continue;
369         }
370
371       /* name switch: finish group, re-init */
372       if (best->name != s->name)   /* new name */
373         {
374           plist->elements[j++] = best - pool->solvables; /* move old best to front */
375           best = s;             /* take current as new best */
376           continue;
377         }
378
379       if (best->evr != s->evr)  /* compare evr */
380         {
381           if (evrcmp(pool, best->evr, s->evr, EVRCMP_COMPARE) < 0)
382             best = s;
383         }
384     }
385   plist->elements[j++] = best - pool->solvables;        /* finish last group */
386   plist->count = j;
387 }
388
389
390 /* legacy, do not use anymore!
391  * (rates arch higher than version, but thats a policy)
392  */
393
394 void
395 prune_best_arch_name_version(const Solver *solv, Pool *pool, Queue *plist)
396 {
397   if (solv && solv->bestSolvableCb)
398     { /* The application is responsible for */
399       return solv->bestSolvableCb(solv->pool, plist);
400     }
401
402   if (plist->count > 1)
403     prune_to_best_arch(pool, plist);
404   if (plist->count > 1)
405     prune_to_best_version(pool, plist);
406 }
407
408 /* installed packages involed in a dup operation can only be kept
409 * if they are identical to a non-installed one */
410 static void
411 prune_installed_dup_packages(Solver *solv, Queue *plist)
412 {
413   Pool *pool = solv->pool;
414   int i, j, k;
415
416   for (i = j = 0; i < plist->count; i++)
417     {
418       Id p = plist->elements[i];
419       Solvable *s = pool->solvables + p;
420       if (s->repo == pool->installed && (solv->dupmap_all || (solv->dupinvolvedmap.size && MAPTST(&solv->dupinvolvedmap, p))))
421         {
422           for (k = 0; k < plist->count; k++)
423             {
424               Solvable *s2 = pool->solvables + plist->elements[k];
425               if (s2->repo != pool->installed && solvable_identical(s, s2))
426                 break;
427             }
428           if (k == plist->count)
429             continue;   /* no identical package found, ignore installed package */
430         }
431       plist->elements[j++] = p;
432     }
433   if (j)
434     plist->count = j;
435 }
436
437 /*
438  *  POLICY_MODE_CHOOSE:     default, do all pruning steps
439  *  POLICY_MODE_RECOMMEND:  leave out prune_to_recommended
440  *  POLICY_MODE_SUGGEST:    leave out prune_to_recommended, do prio pruning just per name
441  */
442 void
443 policy_filter_unwanted(Solver *solv, Queue *plist, int mode)
444 {
445   Pool *pool = solv->pool;
446   if (plist->count > 1)
447     {
448       if (mode != POLICY_MODE_SUGGEST)
449         prune_to_highest_prio(pool, plist);
450       else
451         prune_to_highest_prio_per_name(pool, plist);
452       /* installed dup packages need special treatment as prio pruning
453        * does not prune installed packages */
454       if (plist->count > 1 && pool->installed && (solv->dupmap_all || solv->dupinvolvedmap.size))
455         prune_installed_dup_packages(solv, plist);
456     }
457   if (plist->count > 1 && mode == POLICY_MODE_CHOOSE)
458     prune_to_recommended(solv, plist);
459   prune_best_arch_name_version(solv, pool, plist);
460 }
461
462
463 int
464 policy_illegal_archchange(Solver *solv, Solvable *s1, Solvable *s2)
465 {
466   Pool *pool = solv->pool;
467   Id a1 = s1->arch, a2 = s2->arch;
468
469   if (solv && solv->archCheckCb)
470     { /* The application is responsible for */
471       return solv->archCheckCb(solv->pool, s1, s2);
472     }
473
474   /* we allow changes to/from noarch */
475 #ifndef DEBIAN_SEMANTICS
476   if (a1 == a2 || a1 == ARCH_NOARCH || a2 == ARCH_NOARCH)
477     return 0;
478 #else
479   if (a1 == a2 || a1 == ARCH_ALL || a2 == ARCH_ALL)
480     return 0;
481 #endif
482   if (!pool->id2arch)
483     return 0;
484   a1 = a1 <= pool->lastarch ? pool->id2arch[a1] : 0;
485   a2 = a2 <= pool->lastarch ? pool->id2arch[a2] : 0;
486   if (((a1 ^ a2) & 0xffff0000) != 0)
487     return 1;
488   return 0;
489 }
490
491 int
492 policy_illegal_vendorchange(Solver *solv, Solvable *s1, Solvable *s2)
493 {
494   Pool *pool = solv->pool;
495   Id v1, v2;
496   Id vendormask1, vendormask2;
497
498   if (solv->vendorCheckCb)
499    {   /* The application is responsible for */
500      return solv->vendorCheckCb(pool, s1, s2);
501    }
502   /* treat a missing vendor as empty string */
503   v1 = s1->vendor ? s1->vendor : ID_EMPTY;
504   v2 = s2->vendor ? s2->vendor : ID_EMPTY;
505   if (v1 == v2)
506     return 0;
507   vendormask1 = pool_vendor2mask(pool, v1);
508   if (!vendormask1)
509     return 1;   /* can't match */
510   vendormask2 = pool_vendor2mask(pool, v2);
511   if ((vendormask1 & vendormask2) != 0)
512     return 0;
513   return 1;     /* no class matches */
514 }
515
516 /* check if it is illegal to replace installed
517  * package "is" with package "s" (which must obsolete "is")
518  */
519 int
520 policy_is_illegal(Solver *solv, Solvable *is, Solvable *s, int ignore)
521 {
522   Pool *pool = solv->pool;
523   int ret = 0;
524   if (!(ignore & POLICY_ILLEGAL_DOWNGRADE) && !solv->allowdowngrade)
525     {
526       if (is->name == s->name && evrcmp(pool, is->evr, s->evr, EVRCMP_COMPARE) > 0)
527         ret |= POLICY_ILLEGAL_DOWNGRADE;
528     }
529   if (!(ignore & POLICY_ILLEGAL_ARCHCHANGE) && !solv->allowarchchange)
530     {
531       if (is->arch != s->arch && policy_illegal_archchange(solv, s, is))
532         ret |= POLICY_ILLEGAL_ARCHCHANGE;
533     }
534   if (!(ignore & POLICY_ILLEGAL_VENDORCHANGE) && !solv->allowvendorchange)
535     {
536       if (is->vendor != s->vendor && policy_illegal_vendorchange(solv, s, is))
537         ret |= POLICY_ILLEGAL_VENDORCHANGE;
538     }
539   return ret;
540 }
541
542 /*-------------------------------------------------------------------
543  * 
544  * create reverse obsoletes map for installed solvables
545  *
546  * For each installed solvable find which packages with *different* names
547  * obsolete the solvable.
548  * This index is used in policy_findupdatepackages() below.
549  */
550 void
551 policy_create_obsolete_index(Solver *solv)
552 {
553   Pool *pool = solv->pool;
554   Solvable *s;
555   Repo *installed = solv->installed;
556   Id p, pp, obs, *obsp, *obsoletes, *obsoletes_data;
557   int i, n, cnt;
558
559   if (!installed || installed->start == installed->end)
560     return;
561   cnt = installed->end - installed->start;
562   solv->obsoletes = obsoletes = sat_calloc(cnt, sizeof(Id));
563   for (i = 1; i < pool->nsolvables; i++)
564     {
565       s = pool->solvables + i;
566       if (!s->obsoletes)
567         continue;
568       if (!pool_installable(pool, s))
569         continue;
570       obsp = s->repo->idarraydata + s->obsoletes;
571       while ((obs = *obsp++) != 0)
572         {
573           FOR_PROVIDES(p, pp, obs)
574             {
575               Solvable *ps = pool->solvables + p;;
576               if (ps->repo != installed)
577                 continue;
578               if (ps->name == s->name)
579                 continue;
580               if (!pool->obsoleteusesprovides && !pool_match_nevr(pool, ps, obs))
581                 continue;
582               if (pool->obsoleteusescolors && !pool_colormatch(pool, s, ps))
583                 continue;
584               obsoletes[p - installed->start]++;
585             }
586         }
587     }
588   n = 0;
589   for (i = 0; i < cnt; i++)
590     if (obsoletes[i])
591       {
592         n += obsoletes[i] + 1;
593         obsoletes[i] = n;
594       }
595   solv->obsoletes_data = obsoletes_data = sat_calloc(n + 1, sizeof(Id));
596   POOL_DEBUG(SAT_DEBUG_STATS, "obsoletes data: %d entries\n", n + 1);
597   for (i = pool->nsolvables - 1; i > 0; i--)
598     {
599       s = pool->solvables + i;
600       if (!s->obsoletes)
601         continue;
602       if (!pool_installable(pool, s))
603         continue;
604       obsp = s->repo->idarraydata + s->obsoletes;
605       while ((obs = *obsp++) != 0)
606         {
607           FOR_PROVIDES(p, pp, obs)
608             {
609               Solvable *ps = pool->solvables + p;;
610               if (ps->repo != installed)
611                 continue;
612               if (ps->name == s->name)
613                 continue;
614               if (!pool->obsoleteusesprovides && !pool_match_nevr(pool, ps, obs))
615                 continue;
616               if (pool->obsoleteusescolors && !pool_colormatch(pool, s, ps))
617                 continue;
618               if (obsoletes_data[obsoletes[p - installed->start]] != i)
619                 obsoletes_data[--obsoletes[p - installed->start]] = i;
620             }
621         }
622     }
623 }
624
625
626 /*
627  * find update candidates
628  * 
629  * s: solvable to be updated
630  * qs: [out] queue to hold Ids of candidates
631  * allow_all: 0 = dont allow downgrades, 1 = allow all candidates
632  * 
633  */
634 void
635 policy_findupdatepackages(Solver *solv, Solvable *s, Queue *qs, int allow_all)
636 {
637   /* installed packages get a special upgrade allowed rule */
638   Pool *pool = solv->pool;
639   Id p, pp, n, p2, pp2;
640   Id obs, *obsp;
641   Solvable *ps;
642   int haveprovobs = 0;
643
644   queue_empty(qs);
645
646   if (solv && solv->updateCandidateCb)
647     { /* The application is responsible for */
648       return solv->updateCandidateCb(solv->pool, s, qs);
649     }
650
651   /*
652    * s = solvable ptr
653    * n = solvable Id
654    */
655   n = s - pool->solvables;
656
657   /*
658    * look for updates for s
659    */
660   FOR_PROVIDES(p, pp, s->name)  /* every provider of s' name */
661     {
662       if (p == n)               /* skip itself */
663         continue;
664
665       ps = pool->solvables + p;
666       if (s->name == ps->name)  /* name match */
667         {
668           if (!allow_all && !solv->allowdowngrade && evrcmp(pool, s->evr, ps->evr, EVRCMP_COMPARE) > 0)
669             continue;
670         }
671       else if (!solv->noupdateprovide && ps->obsoletes)   /* provides/obsoletes combination ? */
672         {
673           obsp = ps->repo->idarraydata + ps->obsoletes;
674           while ((obs = *obsp++) != 0)  /* for all obsoletes */
675             {
676               FOR_PROVIDES(p2, pp2, obs)   /* and all matching providers of the obsoletes */
677                 {
678                   Solvable *ps2 = pool->solvables + p2;
679                   if (!pool->obsoleteusesprovides && !pool_match_nevr(pool, ps2, obs))
680                     continue;
681                   if (pool->obsoleteusescolors && !pool_colormatch(pool, s, ps2))
682                     continue;
683                   if (p2 == n)          /* match ! */
684                     break;
685                 }
686               if (p2)                   /* match! */
687                 break;
688             }
689           if (!obs)                     /* continue if no match */
690             continue;
691           /* here we have 'p' with a matching provides/obsoletes combination
692            * thus flagging p as a valid update candidate for s
693            */
694           haveprovobs = 1;
695         }
696       else
697         continue;
698       if (!allow_all && !solv->allowarchchange && s->arch != ps->arch && policy_illegal_archchange(solv, s, ps))
699         continue;
700       if (!allow_all && !solv->allowvendorchange && s->vendor != ps->vendor && policy_illegal_vendorchange(solv, s, ps))
701         continue;
702       queue_push(qs, p);
703     }
704   /* if we have found some valid candidates and noupdateprovide is not set, we're
705      done. otherwise we fallback to all obsoletes */
706   if (!solv->noupdateprovide && haveprovobs)
707     return;
708   if (solv->obsoletes && solv->obsoletes[n - solv->installed->start])
709     {
710       Id *opp;
711       for (opp = solv->obsoletes_data + solv->obsoletes[n - solv->installed->start]; (p = *opp++) != 0;)
712         {
713           ps = pool->solvables + p;
714           if (!allow_all && !solv->allowarchchange && s->arch != ps->arch && policy_illegal_archchange(solv, s, ps))
715             continue;
716           if (!allow_all && !solv->allowvendorchange && s->vendor != ps->vendor && policy_illegal_vendorchange(solv, s, ps))
717             continue;
718           queue_push(qs, p);
719         }
720     }
721 }
722