git.sur5r.net Git - cc65/blob - src/cc65/codeseg.c

   1 /*****************************************************************************/
   2 /*                                                                           */
   3 /*                                 codeseg.c                                 */
   4 /*                                                                           */
   5 /*                          Code segment structure                           */
   6 /*                                                                           */
   7 /*                                                                           */
   8 /*                                                                           */
   9 /* (C) 2001      Ullrich von Bassewitz                                       */
  10 /*               Wacholderweg 14                                             */
  11 /*               D-70597 Stuttgart                                           */
  12 /* EMail:        uz@cc65.org                                                 */
  13 /*                                                                           */
  14 /*                                                                           */
  15 /* This software is provided 'as-is', without any expressed or implied       */
  16 /* warranty.  In no event will the authors be held liable for any damages    */
  17 /* arising from the use of this software.                                    */
  18 /*                                                                           */
  19 /* Permission is granted to anyone to use this software for any purpose,     */
  20 /* including commercial applications, and to alter it and redistribute it    */
  21 /* freely, subject to the following restrictions:                            */
  22 /*                                                                           */
  23 /* 1. The origin of this software must not be misrepresented; you must not   */
  24 /*    claim that you wrote the original software. If you use this software   */
  25 /*    in a product, an acknowledgment in the product documentation would be  */
  26 /*    appreciated but is not required.                                       */
  27 /* 2. Altered source versions must be plainly marked as such, and must not   */
  28 /*    be misrepresented as being the original software.                      */
  29 /* 3. This notice may not be removed or altered from any source              */
  30 /*    distribution.                                                          */
  31 /*                                                                           */
  32 /*****************************************************************************/
  33
  34
  35
  36 #include <string.h>
  37 #include <ctype.h>
  38
  39 /* common */
  40 #include "chartype.h"
  41 #include "check.h"
  42 #include "hashstr.h"
  43 #include "strutil.h"
  44 #include "xmalloc.h"
  45 #include "xsprintf.h"
  46
  47 /* cc65 */
  48 #include "asmlabel.h"
  49 #include "codeent.h"
  50 #include "codeinfo.h"
  51 #include "datatype.h"
  52 #include "error.h"
  53 #include "symentry.h"
  54 #include "codeseg.h"
  55
  56
  57
  58 /*****************************************************************************/
  59 /*                             Helper functions                              */
  60 /*****************************************************************************/
  61
  62
  63
  64 static void MoveLabelsToPool (CodeSeg* S, CodeEntry* E)
  65 /* Move the labels of the code entry E to the label pool of the code segment */
  66 {
  67     unsigned LabelCount = GetCodeLabelCount (E);
  68     while (LabelCount--) {
  69         CodeLabel* L = GetCodeLabel (E, LabelCount);
  70         L->Flags &= ~LF_DEF;
  71         L->Owner = 0;
  72         CollAppend (&S->Labels, L);
  73     }
  74     CollDeleteAll (&E->Labels);
  75 }
  76
  77
  78
  79 static CodeLabel* FindCodeLabel (CodeSeg* S, const char* Name, unsigned Hash)
  80 /* Find the label with the given name. Return the label or NULL if not found */
  81 {
  82     /* Get the first hash chain entry */
  83     CodeLabel* L = S->LabelHash[Hash];
  84
  85     /* Search the list */
  86     while (L) {
  87         if (strcmp (Name, L->Name) == 0) {
  88             /* Found */
  89             break;
  90         }
  91         L = L->Next;
  92     }
  93     return L;
  94 }
  95
  96
  97
  98 static CodeLabel* NewCodeSegLabel (CodeSeg* S, const char* Name, unsigned Hash)
  99 /* Create a new label and insert it into the label hash table */
 100 {
 101     /* Not found - create a new one */
 102     CodeLabel* L = NewCodeLabel (Name, Hash);
 103
 104     /* Enter the label into the hash table */
 105     L->Next = S->LabelHash[L->Hash];
 106     S->LabelHash[L->Hash] = L;
 107
 108     /* Return the new label */
 109     return L;
 110 }
 111
 112
 113
 114 static void RemoveLabelFromHash (CodeSeg* S, CodeLabel* L)
 115 /* Remove the given code label from the hash list */
 116 {
 117     /* Get the first entry in the hash chain */
 118     CodeLabel* List = S->LabelHash[L->Hash];
 119     CHECK (List != 0);
 120
 121     /* First, remove the label from the hash chain */
 122     if (List == L) {
 123         /* First entry in hash chain */
 124         S->LabelHash[L->Hash] = L->Next;
 125     } else {
 126         /* Must search through the chain */
 127         while (List->Next != L) {
 128             /* If we've reached the end of the chain, something is *really* wrong */
 129             CHECK (List->Next != 0);
 130             /* Next entry */
 131             List = List->Next;
 132         }
 133         /* The next entry is the one, we have been searching for */
 134         List->Next = L->Next;
 135     }
 136 }
 137
 138
 139
 140 /*****************************************************************************/
 141 /*                    Functions for parsing instructions                     */
 142 /*****************************************************************************/
 143
 144
 145
 146 static const char* SkipSpace (const char* S)
 147 /* Skip white space and return an updated pointer */
 148 {
 149     while (IsSpace (*S)) {
 150         ++S;
 151     }
 152     return S;
 153 }
 154
 155
 156
 157 static const char* ReadToken (const char* L, const char* Term,
 158                               char* Buf, unsigned BufSize)
 159 /* Read the next token into Buf, return the updated line pointer. The
 160  * token is terminated by one of the characters given in term.
 161  */
 162 {
 163     /* Read/copy the token */
 164     unsigned I = 0;
 165     unsigned ParenCount = 0;
 166     while (*L && (ParenCount > 0 || strchr (Term, *L) == 0)) {
 167         if (I < BufSize-1) {
 168             Buf[I++] = *L;
 169         }
 170         if (*L == ')') {
 171             --ParenCount;
 172         } else if (*L == '(') {
 173             ++ParenCount;
 174         }
 175         ++L;
 176     }
 177
 178     /* Terminate the buffer contents */
 179     Buf[I] = '\0';
 180
 181     /* Return the updated line pointer */
 182     return L;
 183 }
 184
 185
 186
 187 static CodeEntry* ParseInsn (CodeSeg* S, const char* L)
 188 /* Parse an instruction nnd generate a code entry from it. If the line contains
 189  * errors, output an error message and return NULL.
 190  * For simplicity, we don't accept the broad range of input a "real" assembler
 191  * does. The instruction and the argument are expected to be separated by
 192  * white space, for example.
 193  */
 194 {
 195     char                Mnemo[16];
 196     const OPCDesc*      OPC;
 197     am_t                AM = 0;         /* Initialize to keep gcc silent */
 198     char                Arg[64];
 199     char                Reg;
 200     CodeEntry*          E;
 201     CodeLabel*          Label;
 202
 203     /* Mnemonic */
 204     L = ReadToken (L, " \t", Mnemo, sizeof (Mnemo));
 205
 206     /* Try to find the opcode description for the mnemonic */
 207     OPC = FindOpcode (Mnemo);
 208
 209     /* If we didn't find the opcode, print an error and bail out */
 210     if (OPC == 0) {
 211         Error ("ASM code error: %s is not a valid mnemonic", Mnemo);
 212         return 0;
 213     }
 214
 215     /* Skip separator white space */
 216     L = SkipSpace (L);
 217
 218     /* Get the addressing mode */
 219     Arg[0] = '\0';
 220     switch (*L) {
 221
 222         case '\0':
 223             /* Implicit */
 224             AM = AM_IMP;
 225             break;
 226
 227         case '#':
 228             /* Immidiate */
 229             StrCopy (Arg, sizeof (Arg), L+1);
 230             AM = AM_IMM;
 231             break;
 232
 233         case '(':
 234             /* Indirect */
 235             L = ReadToken (L+1, ",)", Arg, sizeof (Arg));
 236
 237             /* Check for errors */
 238             if (*L == '\0') {
 239                 Error ("ASM code error: syntax error");
 240                 return 0;
 241             }
 242
 243             /* Check the different indirect modes */
 244             if (*L == ',') {
 245                 /* Expect zp x indirect */
 246                 L = SkipSpace (L+1);
 247                 if (toupper (*L) != 'X') {
 248                     Error ("ASM code error: `X' expected");
 249                     return 0;
 250                 }
 251                 L = SkipSpace (L+1);
 252                 if (*L != ')') {
 253                     Error ("ASM code error: `)' expected");
 254                     return 0;
 255                 }
 256                 L = SkipSpace (L+1);
 257                 if (*L != '\0') {
 258                     Error ("ASM code error: syntax error");
 259                     return 0;
 260                 }
 261                 AM = AM_ZPX_IND;
 262             } else if (*L == ')') {
 263                 /* zp indirect or zp indirect, y */
 264                 L = SkipSpace (L+1);
 265                 if (*L == ',') {
 266                     L = SkipSpace (L+1);
 267                     if (toupper (*L) != 'Y') {
 268                         Error ("ASM code error: `Y' expected");
 269                         return 0;
 270                     }
 271                     L = SkipSpace (L+1);
 272                     if (*L != '\0') {
 273                         Error ("ASM code error: syntax error");
 274                         return 0;
 275                     }
 276                     AM = AM_ZP_INDY;
 277                 } else if (*L == '\0') {
 278                     AM = AM_ZP_IND;
 279                 } else {
 280                     Error ("ASM code error: syntax error");
 281                     return 0;
 282                 }
 283             }
 284             break;
 285
 286         case 'a':
 287         case 'A':
 288             /* Accumulator? */
 289             if (L[1] == '\0') {
 290                 AM = AM_ACC;
 291                 break;
 292             }
 293             /* FALLTHROUGH */
 294
 295         default:
 296             /* Absolute, maybe indexed */
 297             L = ReadToken (L, ",", Arg, sizeof (Arg));
 298             if (*L == '\0') {
 299                 /* Assume absolute */
 300                 AM = AM_ABS;
 301             } else if (*L == ',') {
 302                 /* Indexed */
 303                 L = SkipSpace (L+1);
 304                 if (*L == '\0') {
 305                     Error ("ASM code error: syntax error");
 306                     return 0;
 307                 } else {
 308                     Reg = toupper (*L);
 309                     L = SkipSpace (L+1);
 310                     if (Reg == 'X') {
 311                         AM = AM_ABSX;
 312                     } else if (Reg == 'Y') {
 313                         AM = AM_ABSY;
 314                     } else {
 315                         Error ("ASM code error: syntax error");
 316                         return 0;
 317                     }
 318                     if (*L != '\0') {
 319                         Error ("ASM code error: syntax error");
 320                         return 0;
 321                     }
 322                 }
 323             }
 324             break;
 325
 326     }
 327
 328     /* If the instruction is a branch, check for the label and generate it
 329      * if it does not exist. Ignore anything but local labels here.
 330      */
 331     Label = 0;
 332     if ((OPC->Info & OF_BRA) != 0 && Arg[0] == 'L') {
 333
 334         unsigned Hash;
 335
 336         /* Addressing mode must be alsobute or something is really wrong */
 337         CHECK (AM == AM_ABS);
 338
 339         /* Addressing mode is a branch/jump */
 340         AM = AM_BRA;
 341
 342         /* Generate the hash over the label, then search for the label */
 343         Hash = HashStr (Arg) % CS_LABEL_HASH_SIZE;
 344         Label = FindCodeLabel (S, Arg, Hash);
 345
 346         /* If we don't have the label, it's a forward ref - create it */
 347         if (Label == 0) {
 348             /* Generate a new label */
 349             Label = NewCodeSegLabel (S, Arg, Hash);
 350         }
 351     }
 352
 353     /* We do now have the addressing mode in AM. Allocate a new CodeEntry
 354      * structure and initialize it.
 355      */
 356     E = NewCodeEntry (OPC->OPC, AM, Arg, Label);
 357
 358     /* Return the new code entry */
 359     return E;
 360 }
 361
 362
 363
 364 /*****************************************************************************/
 365 /*                                   Code                                    */
 366 /*****************************************************************************/
 367
 368
 369
 370 CodeSeg* NewCodeSeg (const char* SegName, SymEntry* Func)
 371 /* Create a new code segment, initialize and return it */
 372 {
 373     unsigned I;
 374
 375     /* Allocate memory */
 376     CodeSeg* S = xmalloc (sizeof (CodeSeg));
 377
 378     /* Initialize the fields */
 379     S->SegName  = xstrdup (SegName);
 380     S->Func     = Func;
 381     InitCollection (&S->Entries);
 382     InitCollection (&S->Labels);
 383     for (I = 0; I < sizeof(S->LabelHash) / sizeof(S->LabelHash[0]); ++I) {
 384         S->LabelHash[I] = 0;
 385     }
 386
 387     /* If we have a function given, get the return type of the function.
 388      * Assume ANY return type besides void will use the A and X registers.
 389      */
 390     if (S->Func && !IsTypeVoid (GetFuncReturn (Func->Type))) {
 391         S->ExitRegs = REG_AX;
 392     } else {
 393         S->ExitRegs = REG_NONE;
 394     }
 395
 396     /* Return the new struct */
 397     return S;
 398 }
 399
 400
 401
 402 void AddCodeEntry (CodeSeg* S, const char* Format, va_list ap)
 403 /* Add a line to the given code segment */
 404 {
 405     const char* L;
 406     CodeEntry*  E;
 407     char        Token[64];
 408
 409     /* Format the line */
 410     char Buf [256];
 411     xvsprintf (Buf, sizeof (Buf), Format, ap);
 412
 413     /* Skip whitespace */
 414     L = SkipSpace (Buf);
 415
 416     /* Check which type of instruction we have */
 417     E = 0;      /* Assume no insn created */
 418     switch (*L) {
 419
 420         case '\0':
 421             /* Empty line, just ignore it */
 422             break;
 423
 424         case ';':
 425             /* Comment or hint, ignore it for now */
 426             break;
 427
 428         case '.':
 429             /* Control instruction */
 430             ReadToken (L, " \t", Token, sizeof (Token));
 431             Error ("ASM code error: Pseudo instruction `%s' not supported", Token);
 432             break;
 433
 434         default:
 435             E = ParseInsn (S, L);
 436             break;
 437     }
 438
 439     /* If we have a code entry, transfer the labels and insert it */
 440     if (E) {
 441
 442         /* Transfer the labels if we have any */
 443         unsigned I;
 444         unsigned LabelCount = CollCount (&S->Labels);
 445         for (I = 0; I < LabelCount; ++I) {
 446
 447             /* Get the label */
 448             CodeLabel* L = CollAt (&S->Labels, I);
 449
 450             /* Attach it to the entry */
 451             AttachCodeLabel (E, L);
 452         }
 453
 454         /* Delete the transfered labels */
 455         CollDeleteAll (&S->Labels);
 456
 457         /* Add the entry to the list of code entries in this segment */
 458         CollAppend (&S->Entries, E);
 459
 460     }
 461 }
 462
 463
 464
 465 void InsertCodeEntry (CodeSeg* S, struct CodeEntry* E, unsigned Index)
 466 /* Insert the code entry at the index given. Following code entries will be
 467  * moved to slots with higher indices.
 468  */
 469 {
 470     /* Insert the entry into the collection */
 471     CollInsert (&S->Entries, E, Index);
 472 }
 473
 474
 475
 476 void DelCodeEntry (CodeSeg* S, unsigned Index)
 477 /* Delete an entry from the code segment. This includes moving any associated
 478  * labels, removing references to labels and even removing the referenced labels
 479  * if the reference count drops to zero.
 480  */
 481 {
 482     /* Get the code entry for the given index */
 483     CodeEntry* E = GetCodeEntry (S, Index);
 484
 485     /* If the entry has a labels, we have to move this label to the next insn.
 486      * If there is no next insn, move the label into the code segement label
 487      * pool. The operation is further complicated by the fact that the next
 488      * insn may already have a label. In that case change all reference to
 489      * this label and delete the label instead of moving it.
 490      */
 491     unsigned Count = GetCodeLabelCount (E);
 492     if (Count > 0) {
 493
 494         /* The instruction has labels attached. Check if there is a next
 495          * instruction.
 496          */
 497         if (Index == GetCodeEntryCount (S)-1) {
 498
 499             /* No next instruction, move to the codeseg label pool */
 500             MoveLabelsToPool (S, E);
 501
 502         } else {
 503
 504             /* There is a next insn, get it */
 505             CodeEntry* N = GetCodeEntry (S, Index+1);
 506
 507             /* Move labels to the next entry */
 508             MoveCodeLabels (S, E, N);
 509
 510         }
 511     }
 512
 513     /* If this insn references a label, remove the reference. And, if the
 514      * the reference count for this label drops to zero, remove this label.
 515      */
 516     if (E->JumpTo) {
 517         /* Remove the reference */
 518         RemoveCodeLabelRef (S, E);
 519     }
 520
 521     /* Delete the pointer to the insn */
 522     CollDelete (&S->Entries, Index);
 523
 524     /* Delete the instruction itself */
 525     FreeCodeEntry (E);
 526 }
 527
 528
 529
 530 void DelCodeEntries (CodeSeg* S, unsigned Start, unsigned Count)
 531 /* Delete a range of code entries. This includes removing references to labels,
 532  * labels attached to the entries and so on.
 533  */
 534 {
 535     /* Start deleting the entries from the rear, because this involves less
 536      * memory moving.
 537      */
 538     while (Count--) {
 539         DelCodeEntry (S, Start + Count);
 540     }
 541 }
 542
 543
 544
 545 void MoveCodeEntry (CodeSeg* S, unsigned OldPos, unsigned NewPos)
 546 /* Move an entry from one position to another. OldPos is the current position
 547  * of the entry, NewPos is the new position of the entry.
 548  */
 549 {
 550     /* Get the code entry and remove it from the collection */
 551     CodeEntry* E = GetCodeEntry (S, OldPos);
 552     CollDelete (&S->Entries, OldPos);
 553
 554     /* Correct NewPos if needed */
 555     if (NewPos >= OldPos) {
 556         /* Position has changed with removal */
 557         --NewPos;
 558     }
 559
 560     /* Now insert it at the new position */
 561     CollInsert (&S->Entries, E, NewPos);
 562 }
 563
 564
 565
 566 struct CodeEntry* GetNextCodeEntry (CodeSeg* S, unsigned Index)
 567 /* Get the code entry following the one with the index Index. If there is no
 568  * following code entry, return NULL.
 569  */
 570 {
 571     if (Index >= CollCount (&S->Entries)-1) {
 572         /* This is the last entry */
 573         return 0;
 574     } else {
 575         /* Code entries left */
 576         return CollAtUnchecked (&S->Entries, Index+1);
 577     }
 578 }
 579
 580
 581
 582 int GetCodeEntries (CodeSeg* S, struct CodeEntry** List,
 583                     unsigned Start, unsigned Count)
 584 /* Get Count code entries into List starting at index start. Return true if
 585  * we got the lines, return false if not enough lines were available.
 586  */
 587 {
 588     /* Check if enough entries are available */
 589     if (Start + Count > CollCount (&S->Entries)) {
 590         return 0;
 591     }
 592
 593     /* Copy the entries */
 594     while (Count--) {
 595         *List++ = CollAtUnchecked (&S->Entries, Start++);
 596     }
 597
 598     /* We have the entries */
 599     return 1;
 600 }
 601
 602
 603
 604 unsigned GetCodeEntryIndex (CodeSeg* S, struct CodeEntry* E)
 605 /* Return the index of a code entry */
 606 {
 607     int Index = CollIndex (&S->Entries, E);
 608     CHECK (Index >= 0);
 609     return Index;
 610 }
 611
 612
 613
 614 void AddCodeLabel (CodeSeg* S, const char* Name)
 615 /* Add a code label for the next instruction to follow */
 616 {
 617     /* Calculate the hash from the name */
 618     unsigned Hash = HashStr (Name) % CS_LABEL_HASH_SIZE;
 619
 620     /* Try to find the code label if it does already exist */
 621     CodeLabel* L = FindCodeLabel (S, Name, Hash);
 622
 623     /* Did we find it? */
 624     if (L) {
 625         /* We found it - be sure it does not already have an owner */
 626         CHECK (L->Owner == 0);
 627     } else {
 628         /* Not found - create a new one */
 629         L = NewCodeSegLabel (S, Name, Hash);
 630     }
 631
 632     /* Safety. This call is quite costly, but safety is better */
 633     if (CollIndex (&S->Labels, L) >= 0) {
 634         Internal ("AddCodeLabel: Label `%s' already defined", Name);
 635     }
 636
 637     /* We do now have a valid label. Remember it for later */
 638     CollAppend (&S->Labels, L);
 639 }
 640
 641
 642
 643 CodeLabel* GenCodeLabel (CodeSeg* S, struct CodeEntry* E)
 644 /* If the code entry E does already have a label, return it. Otherwise
 645  * create a new label, attach it to E and return it.
 646  */
 647 {
 648     CodeLabel* L;
 649
 650     if (CodeEntryHasLabel (E)) {
 651
 652         /* Get the label from this entry */
 653         L = GetCodeLabel (E, 0);
 654
 655     } else {
 656
 657         /* Get a new name */
 658         const char* Name = LocalLabelName (GetLocalLabel ());
 659
 660         /* Generate the hash over the name */
 661         unsigned Hash = HashStr (Name) % CS_LABEL_HASH_SIZE;
 662
 663         /* Create a new label */
 664         L = NewCodeSegLabel (S, Name, Hash);
 665
 666         /* Attach this label to the code entry */
 667         AttachCodeLabel (E, L);
 668
 669     }
 670
 671     /* Return the label */
 672     return L;
 673 }
 674
 675
 676
 677 void DelCodeLabel (CodeSeg* S, CodeLabel* L)
 678 /* Remove references from this label and delete it. */
 679 {
 680     unsigned Count, I;
 681
 682     /* First, remove the label from the hash chain */
 683     RemoveLabelFromHash (S, L);
 684
 685     /* Remove references from insns jumping to this label */
 686     Count = CollCount (&L->JumpFrom);
 687     for (I = 0; I < Count; ++I) {
 688         /* Get the insn referencing this label */
 689         CodeEntry* E = CollAt (&L->JumpFrom, I);
 690         /* Remove the reference */
 691         E->JumpTo = 0;
 692     }
 693     CollDeleteAll (&L->JumpFrom);
 694
 695     /* Remove the reference to the owning instruction if it has one. The
 696      * function may be called for a label without an owner when deleting
 697      * unfinished parts of the code. This is unfortunate since it allows
 698      * errors to slip through.
 699      */
 700     if (L->Owner) {
 701         CollDeleteItem (&L->Owner->Labels, L);
 702     }
 703
 704     /* All references removed, delete the label itself */
 705     FreeCodeLabel (L);
 706 }
 707
 708
 709
 710 void MergeCodeLabels (CodeSeg* S)
 711 /* Merge code labels. That means: For each instruction, remove all labels but
 712  * one and adjust references accordingly.
 713  */
 714 {
 715     unsigned I;
 716
 717     /* Walk over all code entries */
 718     unsigned EntryCount = GetCodeEntryCount (S);
 719     for (I = 0; I < EntryCount; ++I) {
 720
 721         CodeLabel* RefLab;
 722         unsigned   J;
 723
 724         /* Get a pointer to the next entry */
 725         CodeEntry* E = GetCodeEntry (S, I);
 726
 727         /* If this entry has zero labels, continue with the next one */
 728         unsigned LabelCount = GetCodeLabelCount (E);
 729         if (LabelCount == 0) {
 730             continue;
 731         }
 732
 733         /* We have at least one label. Use the first one as reference label. */
 734         RefLab = GetCodeLabel (E, 0);
 735
 736         /* Walk through the remaining labels and change references to these
 737          * labels to a reference to the one and only label. Delete the labels
 738          * that are no longer used. To increase performance, walk backwards
 739          * through the list.
 740          */
 741         for (J = LabelCount-1; J >= 1; --J) {
 742
 743             /* Get the next label */
 744             CodeLabel* L = GetCodeLabel (E, J);
 745
 746             /* Move all references from this label to the reference label */
 747             MoveLabelRefs (L, RefLab);
 748
 749             /* Remove the label completely. */
 750             DelCodeLabel (S, L);
 751         }
 752
 753         /* The reference label is the only remaining label. Check if there
 754          * are any references to this label, and delete it if this is not
 755          * the case.
 756          */
 757         if (CollCount (&RefLab->JumpFrom) == 0) {
 758             /* Delete the label */
 759             DelCodeLabel (S, RefLab);
 760         }
 761     }
 762 }
 763
 764
 765
 766 void MoveCodeLabels (CodeSeg* S, struct CodeEntry* Old, struct CodeEntry* New)
 767 /* Move all labels from Old to New. The routine will move the labels itself
 768  * if New does not have any labels, and move references if there is at least
 769  * a label for new. If references are moved, the old label is deleted
 770  * afterwards.
 771  */
 772 {
 773     /* Get the number of labels to move */
 774     unsigned OldLabelCount = GetCodeLabelCount (Old);
 775
 776     /* Does the new entry have itself a label? */
 777     if (CodeEntryHasLabel (New)) {
 778
 779         /* The new entry does already have a label - move references */
 780         CodeLabel* NewLabel = GetCodeLabel (New, 0);
 781         while (OldLabelCount--) {
 782
 783             /* Get the next label */
 784             CodeLabel* OldLabel = GetCodeLabel (Old, OldLabelCount);
 785
 786             /* Move references */
 787             MoveLabelRefs (OldLabel, NewLabel);
 788
 789             /* Delete the label */
 790             DelCodeLabel (S, OldLabel);
 791
 792         }
 793
 794     } else {
 795
 796         /* The new entry does not have a label, just move them */
 797         while (OldLabelCount--) {
 798
 799             /* Move the label to the new entry */
 800             MoveCodeLabel (GetCodeLabel (Old, OldLabelCount), New);
 801
 802         }
 803
 804     }
 805 }
 806
 807
 808
 809 void RemoveCodeLabelRef (CodeSeg* S, struct CodeEntry* E)
 810 /* Remove the reference between E and the label it jumps to. The reference
 811  * will be removed on both sides and E->JumpTo will be 0 after that. If
 812  * the reference was the only one for the label, the label will get
 813  * deleted.
 814  */
 815 {
 816     /* Get a pointer to the label and make sure it exists */
 817     CodeLabel* L = E->JumpTo;
 818     CHECK (L != 0);
 819
 820     /* Delete the entry from the label */
 821     CollDeleteItem (&L->JumpFrom, E);
 822
 823     /* The entry jumps no longer to L */
 824     E->JumpTo = 0;
 825
 826     /* If there are no more references, delete the label */
 827     if (CollCount (&L->JumpFrom) == 0) {
 828         DelCodeLabel (S, L);
 829     }
 830 }
 831
 832
 833
 834 void MoveCodeLabelRef (CodeSeg* S, struct CodeEntry* E, CodeLabel* L)
 835 /* Change the reference of E to L instead of the current one. If this
 836  * was the only reference to the old label, the old label will get
 837  * deleted.
 838  */
 839 {
 840     /* Get the old label */
 841     CodeLabel* OldLabel = E->JumpTo;
 842
 843     /* Be sure that code entry references a label */
 844     PRECONDITION (OldLabel != 0);
 845
 846     /* Remove the reference to our label */
 847     RemoveCodeLabelRef (S, E);
 848
 849     /* Use the new label */
 850     AddLabelRef (L, E);
 851 }
 852
 853
 854
 855 void AddCodeSegHint (CodeSeg* S, unsigned Hint)
 856 /* Add a hint for the preceeding instruction */
 857 {
 858     CodeEntry* E;
 859
 860     /* Get the number of entries in this segment */
 861     unsigned EntryCount = GetCodeEntryCount (S);
 862
 863     /* Must have at least one entry */
 864     CHECK (EntryCount > 0);
 865
 866     /* Get the last entry */
 867     E = GetCodeEntry (S, EntryCount-1);
 868
 869     /* Add the hint */
 870     E->Hints |= Hint;
 871 }
 872
 873
 874
 875 void DelCodeSegAfter (CodeSeg* S, unsigned Last)
 876 /* Delete all entries including the given one */
 877 {
 878     /* Get the number of entries in this segment */
 879     unsigned Count = GetCodeEntryCount (S);
 880
 881     /* First pass: Delete all references to labels. If the reference count
 882      * for a label drops to zero, delete it.
 883      */
 884     unsigned C = Count;
 885     while (Last < C--) {
 886
 887         /* Get the next entry */
 888         CodeEntry* E = GetCodeEntry (S, C);
 889
 890         /* Check if this entry has a label reference */
 891         if (E->JumpTo) {
 892             /* Remove the reference to the label */
 893             RemoveCodeLabelRef (S, E);
 894         }
 895
 896     }
 897
 898     /* Second pass: Delete the instructions. If a label attached to an
 899      * instruction still has references, it must be references from outside
 900      * the deleted area. Don't delete the label in this case, just make it
 901      * ownerless and move it to the label pool.
 902      */
 903     C = Count;
 904     while (Last < C--) {
 905
 906         /* Get the next entry */
 907         CodeEntry* E = GetCodeEntry (S, C);
 908
 909         /* Check if this entry has a label attached */
 910         if (CodeEntryHasLabel (E)) {
 911             /* Move the labels to the pool and clear the owner pointer */
 912             MoveLabelsToPool (S, E);
 913         }
 914
 915         /* Delete the pointer to the entry */
 916         CollDelete (&S->Entries, C);
 917
 918         /* Delete the entry itself */
 919         FreeCodeEntry (E);
 920     }
 921 }
 922
 923
 924
 925 void OutputCodeSeg (const CodeSeg* S, FILE* F)
 926 /* Output the code segment data to a file */
 927 {
 928     unsigned I;
 929
 930     /* Get the number of entries in this segment */
 931     unsigned Count = GetCodeEntryCount (S);
 932
 933     /* If the code segment is empty, bail out here */
 934     if (Count == 0) {
 935         return;
 936     }
 937
 938     /* Output the segment directive */
 939     fprintf (F, ".segment\t\"%s\"\n\n", S->SegName);
 940
 941     /* If this is a segment for a function, enter a function */
 942     if (S->Func) {
 943         fprintf (F, ".proc\t_%s\n\n", S->Func->Name);
 944     }
 945
 946     /* Output all entries */
 947     for (I = 0; I < Count; ++I) {
 948
 949         unsigned char Use;
 950
 951         OutputCodeEntry (CollConstAt (&S->Entries, I), F);
 952
 953         /* Print usage info */
 954         Use = GetRegInfo ((CodeSeg*) S, I+1);
 955         fprintf (F,
 956                  "  Use: %c%c%c\n",
 957                  (Use & REG_A)? 'A' : '_',
 958                  (Use & REG_X)? 'X' : '_',
 959                  (Use & REG_Y)? 'Y' : '_');
 960     }
 961
 962     /* If this is a segment for a function, leave the function */
 963     if (S->Func) {
 964         fprintf (F, "\n.endproc\n\n");
 965     }
 966 }
 967
 968
 969
 970
 971
 972
 973