src/ir.c

   1 /*!@file
   2   \brief   IR Memory Implementation
   3   \details Intermediary memory management
   4   \author  Jordan Lavatai
   5   \date    Aug 2016
   6   ----------------------------------------------------------------------------*/
   7 /* Standard */
   8 #include <stdlib.h> //exit, malloc
   9 #include <stdio.h>  //print
  10 #include <stdarg.h> //va_args
  11 #include <stdint.h> //uint64_t
  12 #include <string.h> //memset, str*
  13 /* Unicode */
  14 #include <unistd.h>   //u8_* functions
  15 #include <unitypes.h> //uint8_t as a char
  16 #include <unistr.h>   //u32_cpy
  17 #include <unistdio.h> //ulc_fprintf
  18 /* Local */
  19 #include "print.h"
  20 #include "apc.h"
  21 #include "ir.h"
  22 #undef  do_error
  23 #define do_error(...) exit(-1)
  24 /* Public */
  25 int      ir_init(void);
  26 void     ir_quit(void);
  27 int      ir_linker(void);
  28 int      ir_condenser(void);
  29 /* Memory allocation structures */
  30 struct pagenode_t;
  31 struct pagenode_header_t {
  32   struct pagenode_t* next;
  33   char*              head;
  34 };
  35 struct pagenode_t {
  36   struct pagenode_header_t header;
  37   char                     root[];
  38 };
  39 struct pagelist_t {
  40   struct pagenode_t* root, * head;
  41   size_t             pagesize;
  42 };
  43 #define DATA_PAGESIZE    (sys_pagesize)
  44 #define NAME_PAGESIZE    (APC_NAME_MAX * 1024)
  45 #define PL_HEADERSIZE    (sizeof(struct pagenode_header_t))
  46 #define PL_HEADSIZE(_PL) (_PL.head->header.head - _PL.head->root)
  47 #define PL_HEADMEM(_PL)  (_PL.pagesize - PL_HEADERSIZE - PL_HEADSIZE(_PL))
  48 /* Set data mem */
  49 enum dtype { FSDAT, MSDAT, ADAT, LDAT, FBDAT };
  50 struct ir_namelist_t;
  51 struct ir_namelist_t
  52 { struct ir_namelist_t* nextsib;
  53   uint8_t*              name;
  54 };
  55 struct ir_classld_t
  56 { struct ir_class_t*    root_class;
  57   struct ir_namelist_t* namelist, * namelist_head;
  58 };
  59 struct ir_setld_t
  60 { struct ir_classld_t*  classld;
  61   long long             ref;
  62   struct ir_namelist_t* namelist, * namelist_head;
  63 };
  64 struct ir_setdata_header_t
  65 { enum dtype          type;
  66   uint8_t*            src_filename, * data_name;
  67   union ir_setdata_t* nextsib;
  68 };
  69 struct ir_frameinfo_t
  70 { int facing, w, h; };
  71 struct ir_framedata_t
  72 { struct ir_setdata_header_t header;
  73   struct ir_frameinfo_t      frameinfo;
  74 };
  75 struct ir_framebox_t
  76 { struct ir_setdata_header_t header;
  77   struct ir_framedata_t      framesheets[FACING_MAX];
  78   struct ir_framedata_t      mapsheets[FACING_MAX];
  79 };
  80 struct ir_simplex_t { struct ir_setdata_header_t header; };
  81 struct ir_link_t
  82 { struct ir_setdata_header_t header;
  83   struct ir_classld_t*       classld;
  84   struct ir_setld_t*         setld;
  85   enum ltype                 type;
  86 };
  87 union ir_setdata_t
  88 { struct ir_setdata_header_t  header;
  89   struct ir_framebox_t        framebox;
  90   struct ir_framedata_t       framesheet;
  91   struct ir_framedata_t       mapsheet;
  92   struct ir_simplex_t         audio;
  93   struct ir_link_t            link;
  94 };
  95 struct ir_class_t
  96 { struct ir_class_t* nextchild, * nextsib;
  97   struct ir_set_t*   root_set;
  98   uint8_t*           name;
  99 };
 100 struct ir_set_t
 101 { struct ir_set_t*      nextchild, * nextsib;
 102   struct ir_class_t*    class;
 103   long long             ref;
 104   uint8_t*              name;
 105   struct ir_framebox_t* frameboxes;
 106   struct ir_simplex_t*  audio;
 107   struct ir_link_t*     links;
 108 };
 109 /* Functions */
 110 static inline
 111 struct ir_framebox_t* ir_set_add_framebox(struct ir_set_t*,const uint8_t*);
 112 static inline
 113 union ir_setdata_t*   ir_framedata (enum dtype,const uint8_t*,apc_facing,int,int);
 114 static inline
 115 int      init_pagelist(struct pagelist_t*,size_t);
 116 static
 117 void     ir_free_pagenodes(struct pagenode_t*);
 118 static inline
 119 int      bytes_identical(const uint8_t*,const uint8_t*);
 120 static inline
 121 int      classnames_identical(const uint8_t*,const uint8_t*);
 122 static
 123 void*    stack_alloc(size_t);
 124 #define  struct_alloc(_T) ((struct _T*) stack_alloc(sizeof(struct _T)))
 125 static
 126 uint8_t* name_alloc(const uint8_t*);
 127 static
 128 uint8_t* classname_alloc(const uint8_t*);
 129 extern //apc.c
 130 long     sys_pagesize;
 131 static
 132 struct pagelist_t datapages, namepages;
 133 static
 134 struct ir_class_t root_class = { .name = (uint8_t*)"." };
 135
 136 /* Init */
 137 int ir_init
 138 ( void )
 139 { if (init_pagelist(&datapages, (size_t)DATA_PAGESIZE))
 140     eprintf("Memory allocation error\n");
 141   if (init_pagelist(&namepages, (size_t)NAME_PAGESIZE))
 142     eprintf("Memory allocation error\n");
 143   return 0;
 144 }
 145
 146 static inline
 147 int init_pagelist
 148 ( struct pagelist_t* pl,
 149   size_t size
 150 )
 151 { pl->pagesize = size;
 152   pl->root = (struct pagenode_t*) calloc(size,1);
 153   if (pl->root == NULL)
 154     return -1;
 155   pl->root->header.head = pl->root->root;
 156   pl->head = pl->root;
 157   return 0;
 158 }
 159
 160 /* Quit/Cleanup
 161    Recursively clean pagenode linked list
 162 */
 163 void ir_quit
 164 ( void )
 165 { ir_free_pagenodes(datapages.root);
 166   ir_free_pagenodes(namepages.root);
 167 }
 168
 169 static
 170 void ir_free_pagenodes
 171 ( struct pagenode_t* pagenode )
 172 { if (pagenode->header.next != NULL)
 173     ir_free_pagenodes(pagenode->header.next);
 174   free(pagenode);
 175 }
 176
 177 /* Link
 178 */
 179 int ir_linker
 180 ( void )
 181 { return 0; }
 182
 183 /* Condense
 184 */
 185 int ir_condenser
 186 ( void )
 187 { return 0; }
 188
 189 /* Return the class's name string */
 190 uint8_t* ir_class_name
 191 ( struct ir_class_t* class )
 192 { return class->name; }
 193
 194 /* Return a pointer to the root class */
 195 struct ir_class_t* ir_class_root
 196 ( void )
 197 { return &root_class; }
 198
 199 /* Add a subclass to a class
 200    Attempts to create a new subclass in the provided class, returning
 201    the class if it already exists
 202 */
 203 struct ir_class_t* ir_class_addchild
 204 ( struct ir_class_t* class,
 205   const uint8_t*     name
 206 )
 207 { struct ir_class_t* iter;
 208   if (class->nextchild == NULL)
 209     goto alloc;
 210   iter = class->nextchild;
 211   if (iter->name == NULL)
 212     eprintf("Null name pointer in class %p\n", iter);
 213   if (name == NULL)
 214     eprintf("Null child added to class %s\n", iter->name);
 215  check:
 216   if (classnames_identical(iter->name, name))
 217     return iter;
 218   if (iter->nextsib != NULL)
 219     { iter = iter->nextsib;
 220       goto check;
 221     }
 222  alloc:
 223   iter = struct_alloc(ir_class_t);
 224   iter->nextsib = class->nextchild;
 225   iter->name = classname_alloc(name);
 226   return class->nextchild = iter;
 227 }
 228
 229 /* Add a set to a class
 230    Attempts to create a new root set in the specified class, returning
 231    the set if it already exists
 232 */
 233 struct ir_set_t* ir_class_addset
 234 ( struct ir_class_t* class,
 235   const uint8_t*     name
 236 )
 237 { struct ir_set_t* iter;
 238   if (class->root_set == NULL)
 239     goto alloc;
 240   iter = class->root_set;
 241   if (iter->name == NULL)
 242     eprintf("Null name pointer in class %p\n", iter);
 243   if (name == NULL)
 244     eprintf("Null set added to class %U\n", iter->name);
 245  check:
 246   if (bytes_identical(iter->name, name))
 247     return iter;
 248   if (iter->nextsib != NULL)
 249     { iter = iter->nextsib;
 250       goto check;
 251     }
 252  alloc:
 253   iter = struct_alloc(ir_set_t);
 254   iter->nextsib = class->root_set;
 255   iter->name = name_alloc(name);
 256   return class->root_set = iter;
 257 }
 258
 259 /* Add a set to a set
 260    Attempts to create a new subset of the specified set, returning the
 261    child if it already exists
 262 */
 263 struct ir_set_t* ir_set_addchild
 264 ( struct ir_set_t* set,
 265   const uint8_t*   name
 266 )
 267 { struct ir_set_t* iter;
 268   if (set->nextchild == NULL)
 269     goto alloc;
 270   iter = set->nextchild;
 271   if (iter->name == NULL)
 272     eprintf("Null name pointer in set %p\n", iter);
 273   if (name == NULL)
 274     eprintf("Null child added to set %s\n", iter->name);
 275  check:
 276   if (bytes_identical(iter->name, name))
 277     return iter;
 278   if (iter->nextsib != NULL)
 279     { iter = iter->nextsib;
 280       goto check;
 281     }
 282  alloc:
 283   iter = struct_alloc(ir_set_t);
 284   iter->nextsib = set->nextchild;
 285   iter->name = name_alloc(name);
 286   return set->nextchild = iter;
 287 }
 288
 289 /* Add a framebox to a set
 290    Attempts to create a new framebox of the specified set, returning
 291    the framebox if it already exists
 292 */
 293 static inline
 294 struct ir_framebox_t* ir_set_add_framebox
 295 ( struct ir_set_t* set,
 296   const uint8_t*   name
 297 )
 298 { struct ir_framebox_t* iter;
 299   if (set->frameboxes == NULL)
 300     goto alloc;
 301   iter = set->frameboxes;
 302  check:
 303   if (bytes_identical(iter->header.data_name, name))
 304     return iter;
 305   if (iter->header.nextsib != NULL)
 306     { iter = (struct ir_framebox_t*) iter->header.nextsib;
 307       goto check;
 308     }
 309  alloc:
 310   iter = struct_alloc(ir_framebox_t);
 311   iter->header.nextsib = (union ir_setdata_t*) set->frameboxes;
 312   iter->header.data_name = name_alloc(name);
 313   return set->frameboxes = iter;
 314 }
 315
 316 /* Match two null-terminated bytestrings
 317    Return 1 if the two bytestrings are identical, else 0
 318 */
 319 static inline
 320 int bytes_identical
 321 ( const uint8_t* stra,
 322   const uint8_t* strb
 323 )
 324 { int ca, cb;
 325   do {
 326     ca = *stra++;
 327     cb = *strb++;
 328   } while (ca && ca != '_' && ca == cb);
 329   return (ca == cb);
 330 }
 331
 332 static inline
 333 int classnames_identical
 334 ( const uint8_t* stra,
 335   const uint8_t* strb
 336 )
 337 { int ca, cb;
 338   do {
 339     ca = *stra++;
 340     cb = *strb++;
 341   } while (ca && ca == cb);
 342   return (ca == cb);
 343 }
 344
 345 /* Assign Setdata to Set
 346
 347  */
 348 void ir_set_assign_data
 349 ( struct ir_set_t*      set,
 350   union  ir_setdata_t*  setdata
 351 )
 352 { struct ir_framebox_t* framebox;
 353   struct ir_simplex_t*  simplex;
 354   switch (setdata->header.type)
 355     { case FSDAT:
 356         framebox = ir_set_add_framebox(set, setdata->header.data_name);
 357         if (framebox->framesheets[setdata->framesheet.frameinfo.facing].header.data_name != NULL)
 358           wprintf("Duplicate framesheet [%i] %s\n",
 359                   setdata->framesheet.frameinfo.facing, setdata->header.data_name);
 360         framebox->framesheets[setdata->framesheet.frameinfo.facing] = setdata->framesheet;
 361         break;
 362       case MSDAT:
 363         framebox = ir_set_add_framebox(set, setdata->header.data_name);
 364         if (framebox->mapsheets[setdata->mapsheet.frameinfo.facing].header.data_name != NULL)
 365           wprintf("Duplicate mapsheet [%i] %s\n",
 366                   setdata->mapsheet.frameinfo.facing, setdata->header.data_name);
 367         framebox->mapsheets[setdata->mapsheet.frameinfo.facing] = setdata->mapsheet;
 368         break;
 369       case ADAT:
 370         if (set->audio == NULL)
 371           { set->audio = (struct ir_simplex_t*) setdata;
 372             return;
 373           }
 374         simplex = set->audio;
 375         while (simplex->header.nextsib != NULL)
 376           if (bytes_identical(simplex->header.data_name, setdata->header.data_name))
 377             { wprintf("Duplicate audio %s\n", setdata->header.data_name);
 378               *simplex = setdata->audio;
 379               //setdata is now a pointer to redundant, unused memory.
 380               return;
 381             }
 382           else
 383             simplex = (struct ir_simplex_t*) simplex->header.nextsib;
 384         setdata->audio.header.nextsib = (union ir_setdata_t*) set->audio;
 385         set->audio = (struct ir_simplex_t*) setdata;
 386         break;
 387       case LDAT:
 388         setdata->link.header.nextsib = (union ir_setdata_t*) set->links;
 389         set->links = (struct ir_link_t*) setdata;
 390         break;
 391       default:
 392         fprintf(stderr, "Unknown setdata type %x\n", setdata->header.type);
 393         exit(-1);
 394     }
 395 }
 396
 397 void ir_set_assign_ref
 398 ( struct ir_set_t* set,
 399   long long        ref
 400 )
 401 { if (set->ref != 0)
 402     wprintf("Ref override: 0x%lx -> 0x%lx for set %s\n",
 403             (long unsigned) set->ref, (long unsigned) ref, set->name);
 404   set->ref = ref;
 405   //TODO: reflist_add(set);
 406 }
 407
 408 void ir_data_assign_path
 409 ( union ir_setdata_t* setdata,
 410   const uint8_t*      path
 411 )
 412 { if (path == NULL)
 413     eprintf("Null path in data %s\n", setdata->header.data_name);
 414   if (setdata->header.src_filename != NULL)
 415     wprintf("Path override: %s -> %s for setdata %s\n",
 416             setdata->header.src_filename, path, setdata->header.data_name);
 417   setdata->header.src_filename = name_alloc(path);
 418 }
 419
 420 union ir_setdata_t* ir_framesheet
 421 ( const uint8_t* name,
 422   apc_facing     d,
 423   int            width,
 424   int            height
 425 )
 426 { return ir_framedata(FSDAT, name, d, width, height); }
 427
 428 union ir_setdata_t* ir_mapsheet
 429 ( const uint8_t* name,
 430   apc_facing     d,
 431   int            width,
 432   int            height
 433 )
 434 { return ir_framedata(MSDAT, name, d, width, height); }
 435
 436 static inline
 437 union ir_setdata_t* ir_framedata
 438 ( enum dtype     type,
 439   const uint8_t* name,
 440   apc_facing     d,
 441   int            width,
 442   int            height
 443 )
 444 { struct ir_framedata_t* framedata = struct_alloc(ir_framedata_t);
 445   if (name == NULL)
 446     eprintf("Null name in set allocation\n");
 447   framedata->header.type = type;
 448   framedata->header.data_name = name_alloc(name);
 449   framedata->frameinfo.facing = d;
 450   framedata->frameinfo.w = width;
 451   framedata->frameinfo.h = height;
 452   return (union ir_setdata_t*) framedata;
 453 }
 454
 455 union ir_setdata_t* ir_audio
 456 ( const uint8_t* name )
 457 { struct ir_simplex_t* audio = struct_alloc(ir_simplex_t);
 458   if (name == NULL)
 459     eprintf("Null audio\n");
 460   audio->header.type = ADAT;
 461   audio->header.data_name = name_alloc(name);
 462   return (union ir_setdata_t*) audio;
 463 }
 464
 465
 466 /* Create classld that points to a class */
 467 struct ir_classld_t* ir_classld_from_class
 468 ( struct ir_class_t* class )
 469 { struct ir_classld_t* classld;
 470   if (class == NULL)
 471     eprintf("Null class in classld\n");
 472   classld = struct_alloc(ir_classld_t);
 473   classld->root_class = class;
 474   return classld;
 475 }
 476
 477 struct ir_setld_t* ir_setld_from_ref
 478 ( long long ref )
 479 { struct ir_setld_t* setld;
 480   setld = struct_alloc(ir_setld_t);
 481   setld->ref = ref;
 482   return setld;
 483 }
 484
 485 struct ir_setld_t* ir_setld_from_classld
 486 ( struct ir_classld_t* classld,
 487   const uint8_t*       name
 488 )
 489 { struct ir_setld_t* setld;
 490   setld = struct_alloc(ir_setld_t);
 491   setld->namelist = struct_alloc(ir_namelist_t);
 492   setld->namelist_head = setld->namelist;
 493   setld->namelist_head->name = name_alloc(name);
 494   setld->classld = classld;
 495   return setld;
 496 }
 497
 498 struct ir_setld_t* ir_setld_addchild
 499 ( struct ir_setld_t* setld,
 500   const uint8_t*     name
 501 )
 502 { if (setld->namelist == NULL)
 503     { setld->namelist = struct_alloc(ir_namelist_t);
 504       setld->namelist_head = setld->namelist;
 505     }
 506   else
 507     { setld->namelist_head->nextsib = struct_alloc(ir_namelist_t);
 508       setld->namelist_head = setld->namelist_head->nextsib;
 509     }
 510   setld->namelist_head->name = name_alloc(name);
 511   return setld;
 512 }
 513
 514 union ir_setdata_t* ir_link
 515 ( enum ltype         link_type,
 516   struct ir_setld_t* setld,
 517   const uint8_t*     name
 518 )
 519 { struct ir_link_t* link;
 520   link = struct_alloc(ir_link_t);
 521   link->header.type = LDAT;
 522   link->type = link_type;
 523   link->classld = setld->classld;
 524   link->setld = setld;
 525   if (link_type != OLINK && name != NULL)
 526     link->header.data_name = name_alloc(name);
 527   return (union ir_setdata_t*) link;
 528 }
 529
 530
 531 static
 532 void* stack_alloc
 533 ( size_t bytes )
 534 { if (!bytes) //valid behavior to attain current head
 535     return datapages.head->header.head;
 536   if (PL_HEADMEM(datapages) < bytes)
 537     { datapages.head->header.next = (struct pagenode_t*) calloc(datapages.pagesize,1);
 538       if (datapages.head->header.next == NULL)
 539         eprintf("Memory allocation error \n");
 540       datapages.head = datapages.head->header.next;
 541       datapages.head->header.head = datapages.head->root;
 542     }
 543   datapages.head->header.head += bytes;
 544   return (void*) datapages.head->header.head - bytes;
 545 }
 546
 547 static
 548 uint8_t* name_alloc
 549 ( const uint8_t* name_src )
 550 { const uint8_t* iter;
 551   uint8_t*       name;
 552   int            head_mem;
 553  copy:
 554   name = (uint8_t*)namepages.head->header.head;
 555   iter = name_src;
 556   for (head_mem = PL_HEADMEM(namepages); *iter && *iter != '_' && head_mem; head_mem--)
 557     *(namepages.head->header.head)++ = *iter++;
 558   if (head_mem == 0) //not enough room
 559     { namepages.head->header.next = (struct pagenode_t*) calloc(namepages.pagesize,1);
 560       if (namepages.head->header.next == NULL)
 561         eprintf("Memory allocation error\n");
 562       namepages.head = namepages.head->header.next;
 563       namepages.head->header.head = namepages.head->root;
 564       goto copy;
 565     }
 566   *(namepages.head->header.head)++ = '\0';
 567   return name;
 568 }
 569
 570 static
 571 uint8_t* classname_alloc
 572 ( const uint8_t* name_src )
 573 { const uint8_t* iter;
 574   uint8_t*       name;
 575   int            head_mem;
 576  copy:
 577   name = (uint8_t*)namepages.head->header.head;
 578   iter = name_src;
 579   for (head_mem = PL_HEADMEM(namepages); *iter && head_mem; head_mem--)
 580     *(namepages.head->header.head)++ = *iter++;
 581   if (head_mem == 0) //not enough room
 582     { namepages.head->header.next = (struct pagenode_t*) calloc(namepages.pagesize,1);
 583       if (namepages.head->header.next == NULL)
 584         eprintf("Memory allocation error\n");
 585       namepages.head = namepages.head->header.next;
 586       namepages.head->header.head = namepages.head->root;
 587       goto copy;
 588     }
 589   *(namepages.head->header.head)++ = '\0';
 590   return name;
 591 }