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_arch
  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   Solver *solv = dp;
  36   Pool *pool = solv->pool;
  37   int r;
  38   Id a = *(Id *)ap;
  39   Id b = *(Id *)bp;
  40   Solvable *sa, *sb;
  41
  42   sa = pool->solvables + a;
  43   sb = pool->solvables + b;
  44   r = sa->name - sb->name;
  45   if (r)
  46     {
  47       const char *na, *nb;
  48       /* different names. We use real strcmp here so that the result
  49        * is not depending on some random solvable order */
  50       na = id2str(pool, sa->name);
  51       nb = id2str(pool, sb->name);
  52       return strcmp(na, nb);
  53     }
  54   /* the same name, bring installed solvables to the front */
  55   if (solv->installed)
  56     {
  57       if (sa->repo == solv->installed)
  58         {
  59           if (sb->repo != solv->installed)
  60             return -1;
  61         }
  62       else if (sb->repo == solv->installed)
  63         return 1;
  64     }
  65   /* sort by repository sub-prio (installed repo handled above) */
  66   r = (sb->repo ? sb->repo->subpriority : 0) - (sa->repo ? sa->repo->subpriority : 0);
  67   if (r)
  68     return r;
  69   /* no idea about the order, sort by id */
  70   return a - b;
  71 }
  72
  73
  74 /*
  75  * prune to repository with highest priority.
  76  * does not prune installed solvables.
  77  */
  78
  79 static void
  80 prune_to_highest_prio(Pool *pool, Queue *plist)
  81 {
  82   int i, j;
  83   Solvable *s;
  84   int bestprio = 0, bestprioset = 0;
  85
  86   /* prune to highest priority */
  87   for (i = 0; i < plist->count; i++)  /* find highest prio in queue */
  88     {
  89       s = pool->solvables + plist->elements[i];
  90       if (pool->installed && s->repo == pool->installed)
  91         continue;
  92       if (!bestprioset || s->repo->priority > bestprio)
  93         {
  94           bestprio = s->repo->priority;
  95           bestprioset = 1;
  96         }
  97     }
  98   if (!bestprioset)
  99     return;
 100   for (i = j = 0; i < plist->count; i++) /* remove all with lower prio */
 101     {
 102       s = pool->solvables + plist->elements[i];
 103       if (s->repo->priority == bestprio || (pool->installed && s->repo == pool->installed))
 104         plist->elements[j++] = plist->elements[i];
 105     }
 106   plist->count = j;
 107 }
 108
 109
 110 /*
 111  * prune to recommended/suggested packages.
 112  * does not prune installed packages (they are also somewhat recommended).
 113  */
 114
 115 static void
 116 prune_to_recommended(Solver *solv, Queue *plist)
 117 {
 118   Pool *pool = solv->pool;
 119   int i, j, k, ninst;
 120   Solvable *s;
 121   Id p, pp, rec, *recp, sug, *sugp;
 122
 123   ninst = 0;
 124   if (pool->installed)
 125     {
 126       for (i = 0; i < plist->count; i++)
 127         {
 128           p = plist->elements[i];
 129           s = pool->solvables + p;
 130           if (pool->installed && s->repo == pool->installed)
 131             ninst++;
 132         }
 133     }
 134   if (plist->count - ninst < 2)
 135     return;
 136
 137   /* update our recommendsmap/suggestsmap */
 138   if (solv->recommends_index < 0)
 139     {
 140       MAPZERO(&solv->recommendsmap);
 141       MAPZERO(&solv->suggestsmap);
 142       solv->recommends_index = 0;
 143     }
 144   while (solv->recommends_index < solv->decisionq.count)
 145     {
 146       p = solv->decisionq.elements[solv->recommends_index++];
 147       if (p < 0)
 148         continue;
 149       s = pool->solvables + p;
 150       if (s->recommends)
 151         {
 152           recp = s->repo->idarraydata + s->recommends;
 153           while ((rec = *recp++) != 0)
 154             FOR_PROVIDES(p, pp, rec)
 155               MAPSET(&solv->recommendsmap, p);
 156         }
 157       if (s->suggests)
 158         {
 159           sugp = s->repo->idarraydata + s->suggests;
 160           while ((sug = *sugp++) != 0)
 161             FOR_PROVIDES(p, pp, sug)
 162               MAPSET(&solv->suggestsmap, p);
 163         }
 164     }
 165
 166   /* prune to recommended/supplemented */
 167   ninst = 0;
 168   for (i = j = 0; i < plist->count; i++)
 169     {
 170       p = plist->elements[i];
 171       s = pool->solvables + p;
 172       if (pool->installed && s->repo == pool->installed)
 173         {
 174           ninst++;
 175           if (j)
 176             plist->elements[j++] = p;
 177           continue;
 178         }
 179       if (!MAPTST(&solv->recommendsmap, p))
 180         if (!solver_is_supplementing(solv, s))
 181           continue;
 182       if (!j && ninst)
 183         {
 184           for (k = 0; j < ninst; k++)
 185             {
 186               s = pool->solvables + plist->elements[k];
 187               if (pool->installed && s->repo == pool->installed)
 188                 plist->elements[j++] = plist->elements[k];
 189             }
 190         }
 191       plist->elements[j++] = p;
 192     }
 193   if (j)
 194     plist->count = j;
 195
 196   /* anything left to prune? */
 197   if (plist->count - ninst < 2)
 198     return;
 199
 200   /* prune to suggested/enhanced*/
 201   ninst = 0;
 202   for (i = j = 0; i < plist->count; i++)
 203     {
 204       p = plist->elements[i];
 205       s = pool->solvables + p;
 206       if (pool->installed && s->repo == pool->installed)
 207         {
 208           ninst++;
 209           if (j)
 210             plist->elements[j++] = p;
 211           continue;
 212         }
 213       if (!MAPTST(&solv->suggestsmap, p))
 214         if (!solver_is_enhancing(solv, s))
 215           continue;
 216       if (!j && ninst)
 217         {
 218           for (k = 0; j < ninst; k++)
 219             {
 220               s = pool->solvables + plist->elements[k];
 221               if (pool->installed && s->repo == pool->installed)
 222                 plist->elements[j++] = plist->elements[k];
 223             }
 224         }
 225       plist->elements[j++] = p;
 226     }
 227   if (j)
 228     plist->count = j;
 229 }
 230
 231 void
 232 prune_to_best_arch(Pool *pool, Queue *plist)
 233 {
 234   Id a, bestscore;
 235   Solvable *s;
 236   int i, j;
 237
 238   if (!pool->id2arch || plist->count < 2)
 239     return;
 240   bestscore = 0;
 241   for (i = 0; i < plist->count; i++)
 242     {
 243       s = pool->solvables + plist->elements[i];
 244       a = s->arch;
 245       a = (a <= pool->lastarch) ? pool->id2arch[a] : 0;
 246       if (a && a != 1 && (!bestscore || a < bestscore))
 247         bestscore = a;
 248     }
 249   for (i = j = 0; i < plist->count; i++)
 250     {
 251       s = pool->solvables + plist->elements[i];
 252       a = s->arch;
 253       if (a > pool->lastarch)
 254         continue;
 255       a = pool->id2arch[a];
 256       /* a == 1 -> noarch */
 257       if (a != 1 && ((a ^ bestscore) & 0xffff0000) != 0)
 258         continue;
 259       plist->elements[j++] = plist->elements[i];
 260     }
 261   if (j)
 262     plist->count = j;
 263 }
 264
 265 /*
 266  * prune_to_best_version
 267  *
 268  * sort list of packages (given through plist) by name and evr
 269  * return result through plist
 270  */
 271 void
 272 prune_to_best_version(Solver *solv, Queue *plist)
 273 {
 274   Pool *pool = solv->pool;
 275   Id best;
 276   int i, j;
 277   Solvable *s;
 278
 279   if (plist->count < 2)         /* no need to prune for a single entry */
 280     return;
 281   POOL_DEBUG(SAT_DEBUG_POLICY, "prune_to_best_version %d\n", plist->count);
 282
 283   /* sort by name first, prefer installed */
 284   sat_sort(plist->elements, plist->count, sizeof(Id), prune_to_best_version_sortcmp, solv);
 285
 286   /* delete obsoleted. hmm, looks expensive! */
 287   /* FIXME maybe also check provides depending on noupdateprovide? */
 288   /* FIXME do not prune cycles */
 289   for (i = 0; i < plist->count; i++)
 290     {
 291       Id p, pp, obs, *obsp;
 292       s = pool->solvables + plist->elements[i];
 293       if (!s->obsoletes)
 294         continue;
 295       obsp = s->repo->idarraydata + s->obsoletes;
 296       while ((obs = *obsp++) != 0)
 297         {
 298           FOR_PROVIDES(p, pp, obs)
 299             {
 300               if (pool->solvables[p].name == s->name)
 301                 continue;
 302               if (!solv->obsoleteusesprovides && !pool_match_nevr(pool, pool->solvables + p, obs))
 303                 continue;
 304               for (j = 0; j < plist->count; j++)
 305                 {
 306                   if (i == j)
 307                     continue;
 308                   if (plist->elements[j] == p)
 309                     plist->elements[j] = 0;
 310                 }
 311             }
 312         }
 313     }
 314   /* delete zeroed out queue entries */
 315   for (i = j = 0; i < plist->count; i++)
 316     if (plist->elements[i])
 317       plist->elements[j++] = plist->elements[i];
 318   plist->count = j;
 319
 320   /* now find best 'per name' */
 321   best = 0;
 322   for (i = j = 0; i < plist->count; i++)
 323     {
 324       s = pool->solvables + plist->elements[i];
 325
 326       POOL_DEBUG(SAT_DEBUG_POLICY, "- %s[%s]\n",
 327                  solvable2str(pool, s),
 328                  (solv->installed && s->repo == solv->installed) ? "installed" : "not installed");
 329
 330       if (!best)                       /* if no best yet, the current is best */
 331         {
 332           best = plist->elements[i];
 333           continue;
 334         }
 335
 336       /* name switch: re-init */
 337       if (pool->solvables[best].name != s->name)   /* new name */
 338         {
 339           plist->elements[j++] = best; /* move old best to front */
 340           best = plist->elements[i];   /* take current as new best */
 341           continue;
 342         }
 343
 344       if (pool->solvables[best].evr != s->evr)   /* compare evr */
 345         {
 346           if (evrcmp(pool, pool->solvables[best].evr, s->evr, EVRCMP_COMPARE) < 0)
 347             best = plist->elements[i];
 348         }
 349     }
 350
 351   if (!best)
 352     best = plist->elements[0];
 353
 354   plist->elements[j++] = best;
 355   plist->count = j;
 356 }
 357
 358
 359 /* legacy, do not use anymore!
 360  * (rates arch higher than version, but thats a policy)
 361  */
 362
 363 void
 364 prune_best_arch_name_version(Solver *solv, Pool *pool, Queue *plist)
 365 {
 366   if (solv && solv->bestSolvableCb)
 367     { /* The application is responsible for */
 368       return solv->bestSolvableCb(solv->pool, plist);
 369     }
 370
 371   if (plist->count > 1)
 372     prune_to_best_arch(pool, plist);
 373   if (plist->count > 1)
 374     prune_to_best_version(solv, plist);
 375 }
 376
 377
 378 void
 379 policy_filter_unwanted(Solver *solv, Queue *plist, int mode)
 380 {
 381   Pool *pool = solv->pool;
 382   if (plist->count > 1 && mode != POLICY_MODE_SUGGEST)
 383     prune_to_highest_prio(pool, plist);
 384   if (plist->count > 1 && mode == POLICY_MODE_CHOOSE)
 385     prune_to_recommended(solv, plist);
 386   prune_best_arch_name_version(solv, pool, plist);
 387 }
 388
 389
 390 int
 391 policy_illegal_archchange(Solver *solv, Solvable *s1, Solvable *s2)
 392 {
 393   Pool *pool = solv->pool;
 394   Id a1 = s1->arch, a2 = s2->arch;
 395
 396   if (solv && solv->archCheckCb)
 397     { /* The application is responsible for */
 398       return solv->archCheckCb(solv->pool, s1, s2);
 399     }
 400
 401   /* we allow changes to/from noarch */
 402   if (a1 == a2 || a1 == ARCH_NOARCH || a2 == ARCH_NOARCH)
 403     return 0;
 404   if (!pool->id2arch)
 405     return 0;
 406   a1 = a1 <= pool->lastarch ? pool->id2arch[a1] : 0;
 407   a2 = a2 <= pool->lastarch ? pool->id2arch[a2] : 0;
 408   if (((a1 ^ a2) & 0xffff0000) != 0)
 409     return 1;
 410   return 0;
 411 }
 412
 413 int
 414 policy_illegal_vendorchange(Solver *solv, Solvable *s1, Solvable *s2)
 415 {
 416   Pool *pool = solv->pool;
 417   Id vendormask1, vendormask2;
 418
 419   if (solv && solv->vendorCheckCb)
 420    {   /* The application is responsible for */
 421      return solv->vendorCheckCb(solv->pool, s1, s2);
 422    }
 423
 424   if (s1->vendor == s2->vendor)
 425     return 0;
 426   vendormask1 = pool_vendor2mask(pool, s1->vendor);
 427   if (!vendormask1)
 428     return 0;
 429   vendormask2 = pool_vendor2mask(pool, s2->vendor);
 430   if ((vendormask1 & vendormask2) != 0)
 431     return 0;
 432   return 1;
 433 }
 434
 435
 436 /*
 437  * find update candidates
 438  *
 439  * s: solvable to be updated
 440  * qs: [out] queue to hold Ids of candidates
 441  * allow_all: 0 = dont allow downgrades, 1 = allow all candidates
 442  *
 443  */
 444 void
 445 policy_findupdatepackages(Solver *solv, Solvable *s, Queue *qs, int allow_all)
 446 {
 447   /* installed packages get a special upgrade allowed rule */
 448   Pool *pool = solv->pool;
 449   Id p, pp, n, p2, pp2;
 450   Id obs, *obsp;
 451   Solvable *ps;
 452
 453   queue_empty(qs);
 454
 455   if (solv && solv->updateCandidateCb)
 456     { /* The application is responsible for */
 457       return solv->updateCandidateCb(solv->pool, s, qs);
 458     }
 459
 460   /*
 461    * s = solvable ptr
 462    * n = solvable Id
 463    */
 464   n = s - pool->solvables;
 465
 466   /*
 467    * look for updates for s
 468    */
 469   FOR_PROVIDES(p, pp, s->name)  /* every provider of s' name */
 470     {
 471       if (p == n)               /* skip itself */
 472         continue;
 473
 474       ps = pool->solvables + p;
 475       if (s->name == ps->name)  /* name match */
 476         {
 477           if (!allow_all && !solv->allowdowngrade && evrcmp(pool, s->evr, ps->evr, EVRCMP_COMPARE) > 0)
 478             continue;
 479         }
 480       else if (!solv->noupdateprovide && ps->obsoletes)   /* provides/obsoletes combination ? */
 481         {
 482           obsp = ps->repo->idarraydata + ps->obsoletes;
 483           while ((obs = *obsp++) != 0)  /* for all obsoletes */
 484             {
 485               FOR_PROVIDES(p2, pp2, obs)   /* and all matching providers of the obsoletes */
 486                 {
 487                   if (!solv->obsoleteusesprovides && !pool_match_nevr(pool, pool->solvables + p2, obs))
 488                     continue;
 489                   if (p2 == n)          /* match ! */
 490                     break;
 491                 }
 492               if (p2)                   /* match! */
 493                 break;
 494             }
 495           if (!obs)                     /* continue if no match */
 496             continue;
 497           /* here we have 'p' with a matching provides/obsoletes combination
 498            * thus flagging p as a valid update candidate for s
 499            */
 500         }
 501       else
 502         continue;
 503       if (!allow_all && !solv->allowarchchange && s->arch != ps->arch && policy_illegal_archchange(solv, s, ps))
 504         continue;
 505       if (!allow_all && !solv->allowvendorchange && s->vendor != ps->vendor && policy_illegal_vendorchange(solv, s, ps))
 506         continue;
 507       queue_push(qs, p);
 508     }
 509   /* if we have found some valid candidates and noupdateprovide is not set, we're
 510      done. otherwise we fallback to all obsoletes */
 511   if (!solv->noupdateprovide && qs->count)
 512     return;
 513   if (solv->obsoletes && solv->obsoletes[n - solv->installed->start])
 514     {
 515       Id *opp;
 516       for (opp = solv->obsoletes_data + solv->obsoletes[n - solv->installed->start]; (p = *opp++) != 0;)
 517         {
 518           ps = pool->solvables + p;
 519           if (!allow_all && !solv->allowarchchange && s->arch != ps->arch && policy_illegal_archchange(solv, s, ps))
 520             continue;
 521           if (!allow_all && !solv->allowvendorchange && s->vendor != ps->vendor && policy_illegal_vendorchange(solv, s, ps))
 522             continue;
 523           queue_push(qs, p);
 524         }
 525     }
 526 }
 527