src/repo_solv.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  * repo_solv.c
  10  *
  11  * Read the binary dump of a Repo and create a Repo * from it
  12  *
  13  *  See
  14  *   Repo *pool_addrepo_solv(Pool *pool, FILE *fp)
  15  * below
  16  *
  17  */
  18
  19
  20
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <unistd.h>
  24 #include <string.h>
  25
  26 #include "repo_solv.h"
  27 #include "util.h"
  28
  29 #define INTERESTED_START        SOLVABLE_NAME
  30 #define INTERESTED_END          SOLVABLE_FRESHENS
  31
  32 #define SOLV_ERROR_NOT_SOLV     1
  33 #define SOLV_ERROR_UNSUPPORTED  2
  34 #define SOLV_ERROR_EOF          3
  35 #define SOLV_ERROR_ID_RANGE     4
  36 #define SOLV_ERROR_OVERFLOW     5
  37 #define SOLV_ERROR_CORRUPT      6
  38
  39 static Pool *mypool;            /* for pool_debug... */
  40
  41 /*-----------------------------------------------------------------*/
  42 /* .solv read functions */
  43
  44 /*
  45  * read u32
  46  */
  47
  48 static unsigned int
  49 read_u32(Repodata *data)
  50 {
  51   int c, i;
  52   unsigned int x = 0;
  53
  54   if (data->error)
  55     return 0;
  56   for (i = 0; i < 4; i++)
  57     {
  58       c = getc(data->fp);
  59       if (c == EOF)
  60         {
  61           pool_debug(mypool, SAT_ERROR, "unexpected EOF\n");
  62           data->error = SOLV_ERROR_EOF;
  63           return 0;
  64         }
  65       x = (x << 8) | c;
  66     }
  67   return x;
  68 }
  69
  70
  71 /*
  72  * read u8
  73  */
  74
  75 static unsigned int
  76 read_u8(Repodata *data)
  77 {
  78   int c;
  79
  80   if (data->error)
  81     return 0;
  82   c = getc(data->fp);
  83   if (c == EOF)
  84     {
  85       pool_debug(mypool, SAT_ERROR, "unexpected EOF\n");
  86       data->error = SOLV_ERROR_EOF;
  87       return 0;
  88     }
  89   return c;
  90 }
  91
  92
  93 /*
  94  * read Id
  95  */
  96
  97 static Id
  98 read_id(Repodata *data, Id max)
  99 {
 100   unsigned int x = 0;
 101   int c, i;
 102
 103   if (data->error)
 104     return 0;
 105   for (i = 0; i < 5; i++)
 106     {
 107       c = getc(data->fp);
 108       if (c == EOF)
 109         {
 110           pool_debug(mypool, SAT_ERROR, "unexpected EOF\n");
 111           data->error = SOLV_ERROR_EOF;
 112           return 0;
 113         }
 114       if (!(c & 128))
 115         {
 116           x = (x << 7) | c;
 117           if (max && x >= max)
 118             {
 119               pool_debug(mypool, SAT_ERROR, "read_id: id too large (%u/%u)\n", x, max);
 120               data->error = SOLV_ERROR_ID_RANGE;
 121               return 0;
 122             }
 123           return x;
 124         }
 125       x = (x << 7) ^ c ^ 128;
 126     }
 127   pool_debug(mypool, SAT_ERROR, "read_id: id too long\n");
 128   data->error = SOLV_ERROR_CORRUPT;
 129   return 0;
 130 }
 131
 132
 133 /*
 134  * read array of Ids
 135  */
 136
 137 static Id *
 138 read_rel_idarray(Repodata *data, Id max, Id *map, Id *store, Id *end, Id marker)
 139 {
 140   unsigned int x = 0;
 141   int c;
 142   Id old = 0;
 143
 144   if (data->error)
 145     return 0;
 146   for (;;)
 147     {
 148       c = getc(data->fp);
 149       if (c == EOF)
 150         {
 151           pool_debug(mypool, SAT_ERROR, "unexpected EOF\n");
 152           data->error = SOLV_ERROR_EOF;
 153           return 0;
 154         }
 155       if ((c & 128) != 0)
 156         {
 157           x = (x << 7) ^ c ^ 128;
 158           continue;
 159         }
 160       x = (x << 6) | (c & 63);
 161       if (x == 0 && c == 0x40)
 162         {
 163           /* marker hack */
 164           if (store == end)
 165             {
 166               pool_debug(mypool, SAT_ERROR, "read_rel_idarray: array overflow\n");
 167               data->error = SOLV_ERROR_OVERFLOW;
 168               return 0;
 169             }
 170           *store++ = marker;    /* do not map! */
 171           old = 0;
 172           x = 0;
 173           continue;
 174         }
 175       x = (x - 1) + old;
 176       old = x;
 177       if (max && x >= max)
 178         {
 179           pool_debug(mypool, SAT_ERROR, "read_rel_idarray: id too large (%u/%u)\n", x, max);
 180           data->error = SOLV_ERROR_ID_RANGE;
 181           return 0;
 182         }
 183       if (map)
 184         x = map[x];
 185       if (store == end)
 186         {
 187           pool_debug(mypool, SAT_ERROR, "read_rel_idarray: array overflow\n");
 188           return 0;
 189         }
 190       *store++ = x;
 191       if ((c & 64) == 0)
 192         {
 193           if (x == 0)   /* already have trailing zero? */
 194             return store;
 195           if (store == end)
 196             {
 197               pool_debug(mypool, SAT_ERROR, "read_rel_idarray: array overflow\n");
 198               data->error = SOLV_ERROR_OVERFLOW;
 199               return 0;
 200             }
 201           *store++ = 0;
 202           return store;
 203         }
 204       x = 0;
 205     }
 206 }
 207
 208 static Id *
 209 read_idarray(Repodata *data, Id max, Id *map, Id *store, Id *end)
 210 {
 211   unsigned int x = 0;
 212   int c;
 213
 214   if (data->error)
 215     return 0;
 216   for (;;)
 217     {
 218       c = getc(data->fp);
 219       if (c == EOF)
 220         {
 221           pool_debug(mypool, SAT_ERROR, "unexpected EOF\n");
 222           data->error = SOLV_ERROR_EOF;
 223           return 0;
 224         }
 225       if ((c & 128) != 0)
 226         {
 227           x = (x << 7) ^ c ^ 128;
 228           continue;
 229         }
 230       x = (x << 6) | (c & 63);
 231       if (max && x >= max)
 232         {
 233           pool_debug(mypool, SAT_ERROR, "read_idarray: id too large (%u/%u)\n", x, max);
 234           data->error = SOLV_ERROR_ID_RANGE;
 235           return 0;
 236         }
 237       if (map)
 238         x = map[x];
 239       if (store == end)
 240         {
 241           pool_debug(mypool, SAT_ERROR, "read_idarray: array overflow\n");
 242           return 0;
 243         }
 244       *store++ = x;
 245       if ((c & 64) == 0)
 246         {
 247           if (x == 0)   /* already have trailing zero? */
 248             return store;
 249           if (store == end)
 250             {
 251               pool_debug(mypool, SAT_ERROR, "read_idarray: array overflow\n");
 252               data->error = SOLV_ERROR_OVERFLOW;
 253               return 0;
 254             }
 255           *store++ = 0;
 256           return store;
 257         }
 258       x = 0;
 259     }
 260 }
 261
 262 static void
 263 read_str(Repodata *data, char **inbuf, unsigned *len)
 264 {
 265   unsigned char *buf = (unsigned char*)*inbuf;
 266   if (!buf)
 267     {
 268       buf = sat_malloc(1024);
 269       *len = 1024;
 270     }
 271   int c;
 272   unsigned ofs = 0;
 273   while((c = getc(data->fp)) != 0)
 274     {
 275       if (c == EOF)
 276         {
 277           pool_debug (mypool, SAT_ERROR, "unexpected EOF\n");
 278           data->error = SOLV_ERROR_EOF;
 279           return;
 280         }
 281       /* Plus 1 as we also want to add the 0.  */
 282       if (ofs + 1 >= *len)
 283         {
 284           *len += 256;
 285           /* Don't realloc on the inbuf, it might be on the stack.  */
 286           if (buf == (unsigned char*)*inbuf)
 287             {
 288               buf = sat_malloc(*len);
 289               memcpy(buf, *inbuf, *len - 256);
 290             }
 291           else
 292             buf = sat_realloc(buf, *len);
 293         }
 294       buf[ofs++] = c;
 295     }
 296   buf[ofs++] = 0;
 297   *inbuf = (char*)buf;
 298 }
 299
 300 static void
 301 skip_item (Repodata *data, unsigned type, unsigned numid, unsigned numrel)
 302 {
 303   switch (type)
 304     {
 305       case TYPE_VOID:
 306       case TYPE_CONSTANT:
 307         break;
 308       case TYPE_ID:
 309         read_id(data, numid + numrel);          /* just check Id */
 310         break;
 311       case TYPE_DIR:
 312         read_id(data, numid + data->dirpool.ndirs);     /* just check Id */
 313         break;
 314       case TYPE_NUM:
 315         read_id(data, 0);
 316         break;
 317       case TYPE_U32:
 318         read_u32(data);
 319         break;
 320       case TYPE_ATTR_STRING:
 321       case TYPE_STR:
 322         while (read_u8(data) != 0)
 323           ;
 324         break;
 325       case TYPE_IDARRAY:
 326       case TYPE_REL_IDARRAY:
 327       case TYPE_ATTR_INTLIST:
 328         while ((read_u8(data) & 0xc0) != 0)
 329           ;
 330         break;
 331       case TYPE_DIRNUMNUMARRAY:
 332         for (;;)
 333           {
 334             read_id(data, numid + data->dirpool.ndirs); /* just check Id */
 335             read_id(data, 0);
 336             if (!(read_id(data, 0) & 0x40))
 337               break;
 338           }
 339         break;
 340       case TYPE_DIRSTRARRAY:
 341         for (;;)
 342           {
 343             Id id = read_id(data, 0);
 344             while (read_u8(data) != 0)
 345               ;
 346             if (!(id & 0x40))
 347               break;
 348           }
 349         break;
 350       case TYPE_COUNT_NAMED:
 351         {
 352           unsigned count = read_id(data, 0);
 353           while (count--)
 354             {
 355               read_id(data, numid);    /* Name */
 356               unsigned t = read_id(data, TYPE_ATTR_TYPE_MAX + 1);
 357               skip_item(data, t, numid, numrel);
 358             }
 359         }
 360         break;
 361       case TYPE_COUNTED:
 362         {
 363           unsigned count = read_id(data, 0);
 364           unsigned t = read_id(data, TYPE_ATTR_TYPE_MAX + 1);
 365           while (count--)
 366             skip_item(data, t, numid, numrel);
 367         }
 368         break;
 369       case TYPE_ATTR_CHUNK:
 370         read_id(data, 0);
 371         /* Fallthrough.  */
 372       case TYPE_ATTR_INT:
 373         read_id(data, 0);
 374         break;
 375       case TYPE_ATTR_LOCALIDS:
 376         while (read_id(data, 0) != 0)
 377           ;
 378         break;
 379       default:
 380         pool_debug(mypool, SAT_ERROR, "unknown type %d\n", type);
 381         data->error = SOLV_ERROR_CORRUPT;
 382         break;
 383     }
 384 }
 385
 386 static int
 387 key_cmp (const void *pa, const void *pb)
 388 {
 389   Repokey *a = (Repokey *)pa;
 390   Repokey *b = (Repokey *)pb;
 391   return a->name - b->name;
 392 }
 393
 394 static void repodata_load_solv(Repodata *data);
 395
 396 static void
 397 parse_repodata(Repodata *maindata, Id *keyp, Repokey *keys, Id *idmap, unsigned numid, unsigned numrel, Repo *repo)
 398 {
 399   Id key, id;
 400   Id *ida, *ide;
 401   Repodata *data;
 402   int i, n;
 403
 404   repo->repodata = sat_realloc2(repo->repodata, repo->nrepodata + 1, sizeof (*data));
 405   data = repo->repodata + repo->nrepodata++;
 406   memset(data, 0, sizeof(*data));
 407   data->repo = repo;
 408   data->state = REPODATA_STUB;
 409   data->loadcallback = repodata_load_solv;
 410
 411   while ((key = *keyp++) != 0)
 412     {
 413       id = keys[key].name;
 414       switch (keys[key].type)
 415         {
 416         case TYPE_IDVALUEARRAY:
 417           if (id != REPODATA_KEYS)
 418             {
 419               skip_item(maindata, TYPE_IDVALUEARRAY, numid, numrel);
 420               break;
 421             }
 422           /* read_idarray writes a terminating 0, that's why the + 1 */
 423           ida = sat_calloc(keys[key].size + 1, sizeof(Id));
 424           ide = read_idarray(maindata, 0, 0, ida, ida + keys[key].size + 1);
 425           n = ide - ida - 1;
 426           if (n & 1)
 427             {
 428               pool_debug (mypool, SAT_ERROR, "invalid attribute data\n");
 429               data->error = SOLV_ERROR_CORRUPT;
 430               return;
 431             }
 432           data->nkeys = 1 + (n >> 1);
 433           data->keys = sat_malloc2(data->nkeys, sizeof(data->keys[0]));
 434           memset(data->keys, 0, sizeof(Repokey));
 435           for (i = 1, ide = ida; i < data->nkeys; i++)
 436             {
 437               if (*ide >= numid)
 438                 {
 439                   pool_debug (mypool, SAT_ERROR, "invalid attribute data\n");
 440                   data->error = SOLV_ERROR_CORRUPT;
 441                   return;
 442                 }
 443               data->keys[i].name = idmap ? idmap[*ide++] : *ide++;
 444               data->keys[i].type = *ide++;
 445               data->keys[i].size = 0;
 446               data->keys[i].storage = 0;
 447             }
 448           sat_free(ida);
 449           if (data->nkeys > 2)
 450             qsort(data->keys + 1, data->nkeys - 1, sizeof(data->keys[0]), key_cmp);
 451           break;
 452         case TYPE_STR:
 453           if (id != REPODATA_LOCATION)
 454             skip_item(maindata, TYPE_STR, numid, numrel);
 455           else
 456             {
 457               char buf[1024];
 458               unsigned len = sizeof (buf);
 459               char *filename = buf;
 460               read_str(maindata, &filename, &len);
 461               data->location = strdup(filename);
 462               if (filename != buf)
 463                 free(filename);
 464             }
 465           break;
 466         default:
 467           skip_item(maindata, keys[key].type, numid, numrel);
 468           break;
 469         }
 470     }
 471 }
 472
 473 /*-----------------------------------------------------------------*/
 474
 475
 476 static void
 477 skip_schema(Repodata *data, Id *keyp, Repokey *keys, unsigned int numid, unsigned int numrel)
 478 {
 479   Id key;
 480   while ((key = *keyp++) != 0)
 481     skip_item(data, keys[key].type, numid, numrel);
 482 }
 483
 484 /*-----------------------------------------------------------------*/
 485
 486 static void
 487 incore_add_id(Repodata *data, Id x)
 488 {
 489   unsigned char *dp;
 490   /* make sure we have at least 5 bytes free */
 491   if (data->incoredatafree < 5)
 492     {
 493       data->incoredata = sat_realloc(data->incoredata, data->incoredatalen + 1024);
 494       data->incoredatafree = 1024;
 495     }
 496   dp = data->incoredata + data->incoredatalen;
 497   if (x < 0)
 498     abort();
 499   if (x >= (1 << 14))
 500     {
 501       if (x >= (1 << 28))
 502         *dp++ = (x >> 28) | 128;
 503       if (x >= (1 << 21))
 504         *dp++ = (x >> 21) | 128;
 505       *dp++ = (x >> 14) | 128;
 506     }
 507   if (x >= (1 << 7))
 508     *dp++ = (x >> 7) | 128;
 509   *dp++ = x & 127;
 510   data->incoredatafree -= dp - (data->incoredata + data->incoredatalen);
 511   data->incoredatalen = dp - data->incoredata;
 512 }
 513
 514 static void
 515 incore_add_u32(Repodata *data, unsigned int x)
 516 {
 517   unsigned char *dp;
 518   /* make sure we have at least 4 bytes free */
 519   if (data->incoredatafree < 4)
 520     {
 521       data->incoredata = sat_realloc(data->incoredata, data->incoredatalen + 1024);
 522       data->incoredatafree = 1024;
 523     }
 524   dp = data->incoredata + data->incoredatalen;
 525   *dp++ = x >> 24;
 526   *dp++ = x >> 16;
 527   *dp++ = x >> 8;
 528   *dp++ = x;
 529   data->incoredatafree -= 4;
 530   data->incoredatalen += 4;
 531 }
 532
 533 static void
 534 incore_add_u8(Repodata *data, unsigned int x)
 535 {
 536   unsigned char *dp;
 537   /* make sure we have at least 1 byte free */
 538   if (data->incoredatafree < 1)
 539     {
 540       data->incoredata = sat_realloc(data->incoredata, data->incoredatalen + 1024);
 541       data->incoredatafree = 1024;
 542     }
 543   dp = data->incoredata + data->incoredatalen;
 544   *dp++ = x;
 545   data->incoredatafree--;
 546   data->incoredatalen++;
 547 }
 548
 549
 550
 551 // ----------------------------------------------
 552
 553
 554 /*
 555  * read repo from .solv file
 556  *  and add it to pool
 557  */
 558
 559 static int
 560 repo_add_solv_parent(Repo *repo, FILE *fp, Repodata *parent)
 561 {
 562   Pool *pool = repo->pool;
 563   int i, l;
 564   unsigned int numid, numrel, numdir, numsolv;
 565   unsigned int numkeys, numschemata, numinfo;
 566
 567   Offset sizeid;
 568   Offset *str;                         /* map Id -> Offset into string space */
 569   char *strsp;                         /* repo string space */
 570   char *sp;                            /* pointer into string space */
 571   Id *idmap;                           /* map of repo Ids to pool Ids */
 572   Id id;
 573   unsigned int hashmask, h;
 574   int hh;
 575   Id *hashtbl;
 576   Id name, evr, did;
 577   int flags;
 578   Reldep *ran;
 579   unsigned int size_idarray;
 580   Id *idarraydatap, *idarraydataend;
 581   Offset ido;
 582   Solvable *s;
 583   unsigned int solvflags;
 584   unsigned int solvversion;
 585   Repokey *keys;
 586   Id *schemadata, *schemadatap, *schemadataend;
 587   Id *schemata, key;
 588   int have_xdata;
 589   unsigned oldnrepodata;
 590
 591   struct _Stringpool *spool;
 592
 593   Repodata data;
 594
 595   memset(&data, 0, sizeof(data));
 596   data.repo = repo;
 597   data.fp = fp;
 598
 599   mypool = pool;
 600
 601   if (read_u32(&data) != ('S' << 24 | 'O' << 16 | 'L' << 8 | 'V'))
 602     {
 603       pool_debug(pool, SAT_ERROR, "not a SOLV file\n");
 604       return SOLV_ERROR_NOT_SOLV;
 605     }
 606   solvversion = read_u32(&data);
 607   switch (solvversion)
 608     {
 609       case SOLV_VERSION_1:
 610       case SOLV_VERSION_2:
 611       case SOLV_VERSION_3:
 612       case SOLV_VERSION_4:
 613       case SOLV_VERSION_5:
 614       /* Version 6 existed only intermittantly.  It's equivalent to
 615          version 5.  */
 616       case 6:
 617         break;
 618       default:
 619         pool_debug(pool, SAT_ERROR, "unsupported SOLV version\n");
 620         return SOLV_ERROR_UNSUPPORTED;
 621     }
 622
 623   pool_freeidhashes(pool);
 624
 625   numid = read_u32(&data);
 626   numrel = read_u32(&data);
 627   if (solvversion >= SOLV_VERSION_4)
 628     numdir = read_u32(&data);
 629   else
 630     numdir = 0;
 631   numsolv = read_u32(&data);
 632   numkeys = read_u32(&data);
 633   numschemata = read_u32(&data);
 634   numinfo = read_u32(&data);
 635   solvflags = read_u32(&data);
 636
 637   if (solvversion < SOLV_VERSION_5)
 638     numschemata++;
 639
 640   if (numdir && numdir < 2)
 641     {
 642       pool_debug(pool, SAT_ERROR, "bad number of dirs\n");
 643       return SOLV_ERROR_CORRUPT;
 644     }
 645   if (numinfo && solvversion < SOLV_VERSION_3)
 646     {
 647       pool_debug(pool, SAT_ERROR, "unsupported SOLV format (has info)\n");
 648       return SOLV_ERROR_UNSUPPORTED;
 649     }
 650
 651   if (parent)
 652     {
 653       if (numrel)
 654         {
 655           pool_debug(pool, SAT_ERROR, "relations are forbidden in a store\n");
 656           return SOLV_ERROR_CORRUPT;
 657         }
 658       if (parent->end - parent->start != numsolv)
 659         {
 660           pool_debug(pool, SAT_ERROR, "unequal number of solvables in a store\n");
 661           return SOLV_ERROR_CORRUPT;
 662         }
 663       if (numinfo)
 664         {
 665           pool_debug(pool, SAT_ERROR, "info blocks are forbidden in a store\n");
 666           return SOLV_ERROR_CORRUPT;
 667         }
 668     }
 669
 670   /*******  Part 1: string IDs  *****************************************/
 671
 672   sizeid = read_u32(&data);            /* size of string+Id space */
 673
 674   /*
 675    * read strings and Ids
 676    *
 677    */
 678
 679
 680   /*
 681    * alloc buffers
 682    */
 683
 684   if (!parent)
 685     spool = &pool->ss;
 686   else
 687     {
 688       data.localpool = 1;
 689       spool = &data.spool;
 690       spool->stringspace = sat_malloc(7);
 691       strcpy(spool->stringspace, "<NULL>");
 692       spool->sstrings = 7;
 693       spool->nstrings = 0;
 694     }
 695
 696   /* alloc string buffer */
 697   spool->stringspace = sat_realloc(spool->stringspace, spool->sstrings + sizeid + 1);
 698   /* alloc string offsets (Id -> Offset into string space) */
 699   spool->strings = sat_realloc2(spool->strings, spool->nstrings + numid, sizeof(Offset));
 700
 701   strsp = spool->stringspace;
 702   str = spool->strings;                /* array of offsets into strsp, indexed by Id */
 703
 704   /* point to _BEHIND_ already allocated string/Id space */
 705   strsp += spool->sstrings;
 706
 707
 708   /*
 709    * read new repo at end of pool
 710    */
 711
 712   if ((solvflags & SOLV_FLAG_PREFIX_POOL) == 0)
 713     {
 714       if (sizeid && fread(strsp, sizeid, 1, fp) != 1)
 715         {
 716           pool_debug(pool, SAT_ERROR, "read error while reading strings\n");
 717           return SOLV_ERROR_EOF;
 718         }
 719     }
 720   else
 721     {
 722       unsigned int pfsize = read_u32(&data);
 723       char *prefix = sat_malloc(pfsize);
 724       char *pp = prefix;
 725       char *old_str = 0;
 726       char *dest = strsp;
 727       if (pfsize && fread(prefix, pfsize, 1, fp) != 1)
 728         {
 729           pool_debug(pool, SAT_ERROR, "read error while reading strings\n");
 730           sat_free(prefix);
 731           return SOLV_ERROR_EOF;
 732         }
 733       for (i = 1; i < numid; i++)
 734         {
 735           int same = (unsigned char)*pp++;
 736           size_t len = strlen (pp) + 1;
 737           if (same)
 738             memcpy(dest, old_str, same);
 739           memcpy(dest + same, pp, len);
 740           pp += len;
 741           old_str = dest;
 742           dest += same + len;
 743         }
 744       sat_free(prefix);
 745     }
 746   strsp[sizeid] = 0;                   /* make string space \0 terminated */
 747   sp = strsp;
 748
 749   if (parent)
 750     {
 751       /* no shared pool, thus no idmap and no unification */
 752       idmap = 0;
 753       spool->nstrings = numid;
 754       str[0] = 0;
 755       for (i = 1; i < spool->nstrings; i++)
 756         {
 757           if (sp >= strsp + sizeid)
 758             {
 759               pool_debug(pool, SAT_ERROR, "not enough strings\n");
 760               return SOLV_ERROR_OVERFLOW;
 761             }
 762           str[i] = sp - spool->stringspace;
 763           sp += strlen(sp) + 1;
 764         }
 765       spool->sstrings = sp - spool->stringspace;
 766     }
 767   else
 768     {
 769
 770       /* alloc id map for name and rel Ids. this maps ids in the solv files
 771        * to the ids in our pool */
 772       idmap = sat_calloc(numid + numrel, sizeof(Id));
 773
 774       /*
 775        * build hashes for all read strings
 776        *
 777        */
 778
 779       hashmask = mkmask(spool->nstrings + numid);
 780
 781 #if 0
 782       POOL_DEBUG(SAT_DEBUG_STATS, "read %d strings\n", numid);
 783       POOL_DEBUG(SAT_DEBUG_STATS, "string hash buckets: %d\n", hashmask + 1);
 784 #endif
 785
 786       /*
 787        * create hashtable with strings already in pool
 788        */
 789
 790       hashtbl = sat_calloc(hashmask + 1, sizeof(Id));
 791       for (i = 1; i < spool->nstrings; i++)  /* leave out our dummy zero id */
 792         {
 793           h = strhash(spool->stringspace + spool->strings[i]) & hashmask;
 794           hh = HASHCHAIN_START;
 795           while (hashtbl[h])
 796             h = HASHCHAIN_NEXT(h, hh, hashmask);
 797           hashtbl[h] = i;
 798         }
 799
 800       /*
 801        * run over string space, calculate offsets
 802        *
 803        * build id map (maps solv Id -> pool Id)
 804        */
 805
 806       for (i = 1; i < numid; i++)
 807         {
 808           if (sp >= strsp + sizeid)
 809             {
 810               sat_free(hashtbl);
 811               sat_free(idmap);
 812               pool_debug(pool, SAT_ERROR, "not enough strings\n");
 813               return SOLV_ERROR_OVERFLOW;
 814             }
 815           if (!*sp)                            /* empty string */
 816             {
 817               idmap[i] = ID_EMPTY;
 818               sp++;
 819               continue;
 820             }
 821
 822           /* find hash slot */
 823           h = strhash(sp) & hashmask;
 824           hh = HASHCHAIN_START;
 825           for (;;)
 826             {
 827               id = hashtbl[h];
 828               if (id == 0)
 829                 break;
 830               if (!strcmp(spool->stringspace + spool->strings[id], sp))
 831                 break;                 /* existing string */
 832               h = HASHCHAIN_NEXT(h, hh, hashmask);
 833             }
 834
 835           /* length == offset to next string */
 836           l = strlen(sp) + 1;
 837           if (id == ID_NULL)           /* end of hash chain -> new string */
 838             {
 839               id = spool->nstrings++;
 840               hashtbl[h] = id;
 841               str[id] = spool->sstrings;    /* save Offset */
 842               if (sp != spool->stringspace + spool->sstrings)   /* not at end-of-buffer */
 843                 memmove(spool->stringspace + spool->sstrings, sp, l);   /* append to pool buffer */
 844               spool->sstrings += l;
 845             }
 846           idmap[i] = id;                       /* repo relative -> pool relative */
 847           sp += l;                             /* next string */
 848         }
 849       sat_free(hashtbl);
 850     }
 851   pool_shrink_strings(pool);           /* vacuum */
 852
 853
 854   /*******  Part 2: Relation IDs  ***************************************/
 855
 856   /*
 857    * read RelDeps
 858    *
 859    */
 860
 861   if (numrel)
 862     {
 863       /* extend rels */
 864       pool->rels = sat_realloc2(pool->rels, pool->nrels + numrel, sizeof(Reldep));
 865       ran = pool->rels;
 866
 867       hashmask = mkmask(pool->nrels + numrel);
 868 #if 0
 869       POOL_DEBUG(SAT_DEBUG_STATS, "read %d rels\n", numrel);
 870       POOL_DEBUG(SAT_DEBUG_STATS, "rel hash buckets: %d\n", hashmask + 1);
 871 #endif
 872       /*
 873        * prep hash table with already existing RelDeps
 874        */
 875
 876       hashtbl = sat_calloc(hashmask + 1, sizeof(Id));
 877       for (i = 1; i < pool->nrels; i++)
 878         {
 879           h = relhash(ran[i].name, ran[i].evr, ran[i].flags) & hashmask;
 880           hh = HASHCHAIN_START;
 881           while (hashtbl[h])
 882             h = HASHCHAIN_NEXT(h, hh, hashmask);
 883           hashtbl[h] = i;
 884         }
 885
 886       /*
 887        * read RelDeps from repo
 888        */
 889
 890       for (i = 0; i < numrel; i++)
 891         {
 892           name = read_id(&data, i + numid);     /* read (repo relative) Ids */
 893           evr = read_id(&data, i + numid);
 894           flags = read_u8(&data);
 895           name = idmap[name];           /* map to (pool relative) Ids */
 896           evr = idmap[evr];
 897           h = relhash(name, evr, flags) & hashmask;
 898           hh = HASHCHAIN_START;
 899           for (;;)
 900             {
 901               id = hashtbl[h];
 902               if (id == ID_NULL)        /* end of hash chain */
 903                 break;
 904               if (ran[id].name == name && ran[id].evr == evr && ran[id].flags == flags)
 905                 break;
 906               h = HASHCHAIN_NEXT(h, hh, hashmask);
 907             }
 908           if (id == ID_NULL)            /* new RelDep */
 909             {
 910               id = pool->nrels++;
 911               hashtbl[h] = id;
 912               ran[id].name = name;
 913               ran[id].evr = evr;
 914               ran[id].flags = flags;
 915             }
 916           idmap[i + numid] = MAKERELDEP(id);   /* fill Id map */
 917         }
 918       sat_free(hashtbl);
 919       pool_shrink_rels(pool);           /* vacuum */
 920     }
 921
 922
 923   /*******  Part 3: Dirs  ***********************************************/
 924   if (numdir)
 925     {
 926       data.dirpool.dirs = sat_malloc2(numdir, sizeof(Id));
 927       data.dirpool.ndirs = numdir;
 928       data.dirpool.dirs[0] = 0;         /* dir 0: virtual root */
 929       data.dirpool.dirs[1] = 1;         /* dir 1: / */
 930       for (i = 2; i < numdir; i++)
 931         {
 932           id = read_id(&data, i + numid);
 933           if (id >= numid)
 934             data.dirpool.dirs[i] = -(id - numid);
 935           else if (idmap)
 936             data.dirpool.dirs[i] = idmap[id];
 937           else
 938             data.dirpool.dirs[i] = id;
 939         }
 940     }
 941
 942   /*******  Part 4: Keys  ***********************************************/
 943
 944   keys = sat_calloc(numkeys, sizeof(*keys));
 945   /* keys start at 1 */
 946   for (i = 1; i < numkeys; i++)
 947     {
 948       id = read_id(&data, numid);
 949       if (idmap)
 950         id = idmap[id];
 951       else if (parent)
 952         id = str2id(pool, stringpool_id2str(spool, id), 1);
 953       keys[i].name = id;
 954       keys[i].type = read_id(&data, 0);
 955       keys[i].size = read_id(&data, 0);
 956 #if 0
 957       fprintf (stderr, "key %d %s %d %d\n", i, id2str(pool,id), keys[i].type,
 958                keys[i].size);
 959 #endif
 960       if (solvversion >= SOLV_VERSION_5)
 961         {
 962           keys[i].storage = read_id(&data, 0);
 963           continue;
 964         }
 965       keys[i].storage = KEY_STORAGE_DROPPED;
 966       if (parent)
 967         {
 968           keys[i].storage = KEY_STORAGE_INCORE;
 969           continue;
 970         }
 971       switch (keys[i].type)
 972         {
 973         case TYPE_VOID:
 974         case TYPE_CONSTANT:
 975         case TYPE_STR:
 976         case TYPE_NUM:
 977         case TYPE_DIRNUMNUMARRAY:
 978           keys[i].storage = KEY_STORAGE_INCORE;
 979           break;
 980         case TYPE_ID:
 981           switch(id)
 982             {
 983             case SOLVABLE_NAME:
 984             case SOLVABLE_ARCH:
 985             case SOLVABLE_EVR:
 986             case SOLVABLE_VENDOR:
 987               keys[i].storage = KEY_STORAGE_SOLVABLE;
 988               break;
 989             default:
 990               keys[i].storage = KEY_STORAGE_INCORE;
 991               break;
 992             }
 993           break;
 994         case TYPE_IDARRAY:
 995         case TYPE_REL_IDARRAY:
 996           if (id >= INTERESTED_START && id <= INTERESTED_END)
 997             keys[i].storage = KEY_STORAGE_SOLVABLE;
 998           else
 999             keys[i].storage = KEY_STORAGE_INCORE;
1000           break;
1001         case TYPE_U32:
1002           if (id == RPM_RPMDBID)
1003             keys[i].storage = KEY_STORAGE_SOLVABLE;
1004           else
1005             keys[i].storage = KEY_STORAGE_INCORE;
1006           break;
1007         default:
1008           break;
1009         }
1010     }
1011
1012   have_xdata = parent ? 1 : 0;
1013   for (i = 1; i < numkeys; i++)
1014     if (keys[i].storage == KEY_STORAGE_INCORE || keys[i].storage == KEY_STORAGE_VERTICAL_OFFSET)
1015       have_xdata = 1;
1016
1017   data.keys = keys;
1018   data.nkeys = numkeys;
1019
1020   /*******  Part 5: Schemata ********************************************/
1021
1022   id = read_id(&data, 0);
1023   schemadata = sat_calloc(id + 1, sizeof(Id));
1024   schemadatap = schemadata + 1;
1025   schemadataend = schemadatap + id;
1026   schemata = sat_calloc(numschemata, sizeof(Id));
1027   for (i = 1; i < numschemata; i++)
1028     {
1029       schemata[i] = schemadatap - schemadata;
1030       schemadatap = read_idarray(&data, numid, 0, schemadatap, schemadataend);
1031 #if 0
1032       Id *sp = schemadata + schemata[i];
1033       fprintf (stderr, "schema %d:", i);
1034       for (; *sp; sp++)
1035         fprintf (stderr, " %d", *sp);
1036       fprintf (stderr, "\n");
1037 #endif
1038     }
1039   data.schemata = schemata;
1040   data.nschemata = numschemata;
1041   data.schemadata = schemadata;
1042   data.schemadatalen = schemadataend - data.schemadata;
1043
1044
1045   /*******  Part 6: Info  ***********************************************/
1046   oldnrepodata = repo->nrepodata;
1047   for (i = 0; i < numinfo; i++)
1048     {
1049       /* for now we're just interested in data that starts with
1050        * the repodata_external id
1051        */
1052       Id *keyp;
1053       id = read_id(&data, numschemata);
1054       if (solvversion < SOLV_VERSION_5)
1055         id++;
1056       keyp = schemadata + schemata[id];
1057       key = *keyp;
1058       if (keys[key].name == REPODATA_EXTERNAL && keys[key].type == TYPE_VOID)
1059         {
1060           /* external data for some ids */
1061           parse_repodata(&data, keyp, keys, idmap, numid, numrel, repo);
1062         }
1063       else
1064         skip_schema(&data, keyp, keys, numid, numrel);
1065     }
1066
1067
1068   /*******  Part 7: packed sizes (optional)  ****************************/
1069   char *exists = 0;
1070   if ((solvflags & SOLV_FLAG_PACKEDSIZES) != 0)
1071     {
1072       exists = sat_malloc (numsolv);
1073       for (i = 0; i < numsolv; i++)
1074         exists[i] = read_id(&data, 0) != 0;
1075     }
1076
1077
1078   /*******  Part 8: item data *******************************************/
1079
1080   /* calculate idarray size */
1081   size_idarray = 0;
1082   for (i = 1; i < numkeys; i++)
1083     {
1084       id = keys[i].name;
1085       if ((keys[i].type == TYPE_IDARRAY || keys[i].type == TYPE_REL_IDARRAY)
1086           && id >= INTERESTED_START && id <= INTERESTED_END)
1087         size_idarray += keys[i].size;
1088     }
1089
1090   /* allocate needed space in repo */
1091   if (size_idarray)
1092     {
1093       repo_reserve_ids(repo, 0, size_idarray);
1094       idarraydatap = repo->idarraydata + repo->idarraysize;
1095       repo->idarraysize += size_idarray;
1096       idarraydataend = idarraydatap + size_idarray;
1097       repo->lastoff = 0;
1098     }
1099   else
1100     {
1101       idarraydatap = 0;
1102       idarraydataend = 0;
1103     }
1104
1105   /* read solvables */
1106   if (parent)
1107     {
1108       data.start = parent->start;
1109       data.end = parent->end;
1110       s = pool_id2solvable(pool, data.start);
1111     }
1112   else if (numsolv)
1113     {
1114       s = pool_id2solvable(pool, repo_add_solvable_block(repo, numsolv));
1115       /* store start and end of our id block */
1116       data.start = s - pool->solvables;
1117       data.end = data.start + numsolv;
1118       /* In case we have subfiles, make them refer to our part of the
1119          repository now.  */
1120       for (i = oldnrepodata; i < repo->nrepodata; i++)
1121         {
1122           repo->repodata[i].start = data.start;
1123           repo->repodata[i].end = data.end;
1124         }
1125     }
1126   else
1127     s = 0;
1128
1129   if (have_xdata)
1130     data.incoreoffset = sat_calloc(numsolv, sizeof(Id));
1131   for (i = 0; i < numsolv; i++, s++)
1132     {
1133       Id *keyp;
1134       if (data.error)
1135         break;
1136       if (exists && !exists[i])
1137         continue;
1138       id = read_id(&data, numschemata);
1139       if (solvversion < SOLV_VERSION_5)
1140         id++;
1141       if (have_xdata)
1142         {
1143           data.incoreoffset[i] = data.incoredatalen;
1144           incore_add_id(&data, id);
1145         }
1146       keyp = schemadata + schemata[id];
1147       while ((key = *keyp++) != 0)
1148         {
1149           id = keys[key].name;
1150 #if 0
1151 fprintf(stderr, "solv %d name %d type %d class %d\n", i, id, keys[key].type, keys[key].storage);
1152 #endif
1153           if (keys[key].storage == KEY_STORAGE_VERTICAL_OFFSET)
1154             {
1155               /* copy offset/length into incore */
1156               did = read_id(&data, 0);
1157               incore_add_id(&data, did);
1158               did = read_id(&data, 0);
1159               incore_add_id(&data, did);
1160               continue;
1161             }
1162           switch (keys[key].type)
1163             {
1164             case TYPE_VOID:
1165             case TYPE_CONSTANT:
1166               break;
1167             case TYPE_ID:
1168               did = read_id(&data, numid + numrel);
1169               if (idmap)
1170                 did = idmap[did];
1171               if (id == SOLVABLE_NAME)
1172                 s->name = did;
1173               else if (id == SOLVABLE_ARCH)
1174                 s->arch = did;
1175               else if (id == SOLVABLE_EVR)
1176                 s->evr = did;
1177               else if (id == SOLVABLE_VENDOR)
1178                 s->vendor = did;
1179               else if (keys[key].storage == KEY_STORAGE_INCORE)
1180                 incore_add_id(&data, did);
1181 #if 0
1182               POOL_DEBUG(SAT_DEBUG_STATS, "%s -> %s\n", id2str(pool, id), id2str(pool, did));
1183 #endif
1184               break;
1185             case TYPE_NUM:
1186               did = read_id(&data, 0);
1187               if (keys[key].storage == KEY_STORAGE_INCORE)
1188                 incore_add_id(&data, did);
1189 #if 0
1190               POOL_DEBUG(SAT_DEBUG_STATS, "%s -> %d\n", id2str(pool, id), did);
1191 #endif
1192               break;
1193             case TYPE_U32:
1194               h = read_u32(&data);
1195 #if 0
1196               POOL_DEBUG(SAT_DEBUG_STATS, "%s -> %u\n", id2str(pool, id), h);
1197 #endif
1198               if (id == RPM_RPMDBID)
1199                 {
1200                   if (!repo->rpmdbid)
1201                     repo->rpmdbid = sat_calloc(numsolv, sizeof(Id));
1202                   repo->rpmdbid[i] = h;
1203                 }
1204               else if (keys[key].storage == KEY_STORAGE_INCORE)
1205                 incore_add_u32(&data, h);
1206               break;
1207             case TYPE_STR:
1208               if (keys[key].storage == KEY_STORAGE_INCORE)
1209                 {
1210                   while ((h = read_u8(&data)) != 0)
1211                     incore_add_u8(&data, h);
1212                   incore_add_u8(&data, 0);
1213                 }
1214               else
1215                 {
1216                   while (read_u8(&data) != 0)
1217                     ;
1218                 }
1219               break;
1220             case TYPE_IDARRAY:
1221             case TYPE_REL_IDARRAY:
1222               if (id < INTERESTED_START || id > INTERESTED_END)
1223                 {
1224                   if (keys[key].storage == KEY_STORAGE_INCORE)
1225                     {
1226                       if (idmap)
1227                         {
1228                           abort();      /* implement me! */
1229                         }
1230                       else
1231                         {
1232                           while (((h = read_u8(&data)) & 0xc0) != 0)
1233                             incore_add_u8(&data, h);
1234                           break;
1235                         }
1236                     }
1237                   else
1238                     {
1239                       while ((read_u8(&data) & 0xc0) != 0)
1240                         ;
1241                       break;
1242                     }
1243                   break;
1244                 }
1245               ido = idarraydatap - repo->idarraydata;
1246               if (keys[key].type == TYPE_IDARRAY)
1247                 idarraydatap = read_idarray(&data, numid + numrel, idmap, idarraydatap, idarraydataend);
1248               else if (id == SOLVABLE_REQUIRES)
1249                 idarraydatap = read_rel_idarray(&data, numid + numrel, idmap, idarraydatap, idarraydataend, SOLVABLE_PREREQMARKER);
1250               else if (id == SOLVABLE_PROVIDES)
1251                 idarraydatap = read_rel_idarray(&data, numid + numrel, idmap, idarraydatap, idarraydataend, SOLVABLE_FILEMARKER);
1252               else
1253                 idarraydatap = read_rel_idarray(&data, numid + numrel, idmap, idarraydatap, idarraydataend, 0);
1254               if (id == SOLVABLE_PROVIDES)
1255                 s->provides = ido;
1256               else if (id == SOLVABLE_OBSOLETES)
1257                 s->obsoletes = ido;
1258               else if (id == SOLVABLE_CONFLICTS)
1259                 s->conflicts = ido;
1260               else if (id == SOLVABLE_REQUIRES)
1261                 s->requires = ido;
1262               else if (id == SOLVABLE_RECOMMENDS)
1263                 s->recommends= ido;
1264               else if (id == SOLVABLE_SUPPLEMENTS)
1265                 s->supplements = ido;
1266               else if (id == SOLVABLE_SUGGESTS)
1267                 s->suggests = ido;
1268               else if (id == SOLVABLE_ENHANCES)
1269                 s->enhances = ido;
1270               else if (id == SOLVABLE_FRESHENS)
1271                 s->freshens = ido;
1272 #if 0
1273               POOL_DEBUG(SAT_DEBUG_STATS, "%s ->\n", id2str(pool, id));
1274               for (; repo->idarraydata[ido]; ido++)
1275                 POOL_DEBUG(SAT_DEBUG_STATS,"  %s\n", dep2str(pool, repo->idarraydata[ido]));
1276 #endif
1277               break;
1278             case TYPE_DIRNUMNUMARRAY:
1279               for (;;)
1280                 {
1281                   Id num, num2;
1282                   did = read_id(&data, numdir);
1283                   num = read_id(&data, 0);
1284                   num2 = read_id(&data, 0);
1285                   if (keys[key].storage == KEY_STORAGE_INCORE)
1286                     {
1287 #if 0
1288                       POOL_DEBUG(SAT_DEBUG_STATS, "%s -> %d %d %d\n", id2str(pool, id), did, num, num2);
1289 #endif
1290                       incore_add_id(&data, did);
1291                       incore_add_id(&data, num);
1292                       incore_add_id(&data, num2);
1293                     }
1294                   if (!(num2 & 0x40))
1295                     break;
1296                 }
1297               break;
1298             case TYPE_DIRSTRARRAY:
1299               for (;;)
1300                 {
1301                   did = read_id(&data, 0);
1302                   if (keys[key].storage == KEY_STORAGE_INCORE)
1303                     {
1304                       incore_add_id(&data, did);
1305                       while ((h = read_u8(&data)) != 0)
1306                         incore_add_u8(&data, h);
1307                       incore_add_u8(&data, 0);
1308                     }
1309                   else
1310                     {
1311                       while (read_u8(&data) != 0)
1312                         ;
1313                     }
1314                   if (!(did & 0x40))
1315                     break;
1316                 }
1317               break;
1318             default:
1319               skip_item(&data, keys[key].type, numid, numrel);
1320             }
1321         }
1322     }
1323
1324   if (data.error)
1325     {
1326       /* free solvables */
1327       repo_free_solvable_block(repo, data.start, data.end - data.start, 1);
1328       /* free id array */
1329       repo->idarraysize -= size_idarray;
1330       /* free incore data */
1331       data.incoredata = sat_free(data.incoredata);
1332       data.incoredatalen = data.incoredatafree = 0;
1333     }
1334
1335   if (data.incoredatafree)
1336     {
1337       /* shrink excess size */
1338       data.incoredata = sat_realloc(data.incoredata, data.incoredatalen);
1339       data.incoredatafree = 0;
1340     }
1341
1342   for (i = 1; i < numkeys; i++)
1343     if (keys[i].storage == KEY_STORAGE_VERTICAL_OFFSET)
1344       break;
1345   if (i < numkeys && !data.error)
1346     {
1347       Id fileoffset = 0;
1348       unsigned int pagesize;
1349
1350       /* we have vertical data, make it available */
1351       data.verticaloffset = sat_calloc(numkeys, sizeof(Id));
1352       for (i = 1; i < numkeys; i++)
1353         if (keys[i].storage == KEY_STORAGE_VERTICAL_OFFSET)
1354           {
1355             data.verticaloffset[i] = fileoffset;
1356             fileoffset += keys[i].size;
1357           }
1358       data.lastverticaloffset = fileoffset;
1359       pagesize = read_u32(&data);
1360       repodata_read_or_setup_pages(&data, pagesize, fileoffset);
1361     }
1362   else
1363     {
1364       /* no longer needed */
1365       data.fp = 0;
1366     }
1367
1368   if (parent)
1369     {
1370       /* we're a store */
1371       sat_free(parent->schemata);
1372       sat_free(parent->schemadata);
1373       sat_free(parent->keys);
1374       *parent = data;
1375     }
1376   else if (data.incoredatalen || data.fp)
1377     {
1378       /* we got some data, make it available */
1379       repo->repodata = sat_realloc2(repo->repodata, repo->nrepodata + 1, sizeof(data));
1380       repo->repodata[repo->nrepodata++] = data;
1381     }
1382   else
1383     {
1384       /* discard data */
1385       sat_free(data.dirpool.dirs);
1386       sat_free(data.incoreoffset);
1387       sat_free(schemata);
1388       sat_free(schemadata);
1389       sat_free(keys);
1390     }
1391
1392   sat_free(exists);
1393   sat_free(idmap);
1394   mypool = 0;
1395   return data.error;
1396 }
1397
1398 int
1399 repo_add_solv(Repo *repo, FILE *fp)
1400 {
1401   return repo_add_solv_parent(repo, fp, 0);
1402 }
1403
1404 static void
1405 repodata_load_solv(Repodata *data)
1406 {
1407   FILE *fp;
1408   Pool *pool = data->repo->pool;
1409   if (!pool->loadcallback)
1410     {
1411       data->state = REPODATA_ERROR;
1412       return;
1413     }
1414   fp = pool->loadcallback(pool, data, pool->loadcallbackdata);
1415
1416   if (!fp)
1417     {
1418       data->state = REPODATA_ERROR;
1419       return;
1420     }
1421   if (repo_add_solv_parent(data->repo, fp, data))
1422     data->state = REPODATA_ERROR;
1423   else
1424     data->state = REPODATA_AVAILABLE;
1425 }