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 struct CodeEntry* GetNextCodeEntry (CodeSeg* S, unsigned Index)
 546 /* Get the code entry following the one with the index Index. If there is no
 547  * following code entry, return NULL.
 548  */
 549 {
 550     if (Index >= CollCount (&S->Entries)-1) {
 551         /* This is the last entry */
 552         return 0;
 553     } else {
 554         /* Code entries left */
 555         return CollAtUnchecked (&S->Entries, Index+1);
 556     }
 557 }
 558
 559
 560
 561 int GetCodeEntries (CodeSeg* S, struct CodeEntry** List,
 562                     unsigned Start, unsigned Count)
 563 /* Get Count code entries into List starting at index start. Return true if
 564  * we got the lines, return false if not enough lines were available.
 565  */
 566 {
 567     /* Check if enough entries are available */
 568     if (Start + Count > CollCount (&S->Entries)) {
 569         return 0;
 570     }
 571
 572     /* Copy the entries */
 573     while (Count--) {
 574         *List++ = CollAtUnchecked (&S->Entries, Start++);
 575     }
 576
 577     /* We have the entries */
 578     return 1;
 579 }
 580
 581
 582
 583 unsigned GetCodeEntryIndex (CodeSeg* S, struct CodeEntry* E)
 584 /* Return the index of a code entry */
 585 {
 586     int Index = CollIndex (&S->Entries, E);
 587     CHECK (Index >= 0);
 588     return Index;
 589 }
 590
 591
 592
 593 void AddCodeLabel (CodeSeg* S, const char* Name)
 594 /* Add a code label for the next instruction to follow */
 595 {
 596     /* Calculate the hash from the name */
 597     unsigned Hash = HashStr (Name) % CS_LABEL_HASH_SIZE;
 598
 599     /* Try to find the code label if it does already exist */
 600     CodeLabel* L = FindCodeLabel (S, Name, Hash);
 601
 602     /* Did we find it? */
 603     if (L) {
 604         /* We found it - be sure it does not already have an owner */
 605         CHECK (L->Owner == 0);
 606     } else {
 607         /* Not found - create a new one */
 608         L = NewCodeSegLabel (S, Name, Hash);
 609     }
 610
 611     /* Safety. This call is quite costly, but safety is better */
 612     if (CollIndex (&S->Labels, L) >= 0) {
 613         Internal ("AddCodeLabel: Label `%s' already defined", Name);
 614     }
 615
 616     /* We do now have a valid label. Remember it for later */
 617     CollAppend (&S->Labels, L);
 618 }
 619
 620
 621
 622 CodeLabel* GenCodeLabel (CodeSeg* S, struct CodeEntry* E)
 623 /* If the code entry E does already have a label, return it. Otherwise
 624  * create a new label, attach it to E and return it.
 625  */
 626 {
 627     CodeLabel* L;
 628
 629     if (CodeEntryHasLabel (E)) {
 630
 631         /* Get the label from this entry */
 632         L = GetCodeLabel (E, 0);
 633
 634     } else {
 635
 636         /* Get a new name */
 637         const char* Name = LocalLabelName (GetLocalLabel ());
 638
 639         /* Generate the hash over the name */
 640         unsigned Hash = HashStr (Name) % CS_LABEL_HASH_SIZE;
 641
 642         /* Create a new label */
 643         L = NewCodeSegLabel (S, Name, Hash);
 644
 645         /* Attach this label to the code entry */
 646         AttachCodeLabel (E, L);
 647
 648     }
 649
 650     /* Return the label */
 651     return L;
 652 }
 653
 654
 655
 656 void DelCodeLabel (CodeSeg* S, CodeLabel* L)
 657 /* Remove references from this label and delete it. */
 658 {
 659     unsigned Count, I;
 660
 661     /* First, remove the label from the hash chain */
 662     RemoveLabelFromHash (S, L);
 663
 664     /* Remove references from insns jumping to this label */
 665     Count = CollCount (&L->JumpFrom);
 666     for (I = 0; I < Count; ++I) {
 667         /* Get the insn referencing this label */
 668         CodeEntry* E = CollAt (&L->JumpFrom, I);
 669         /* Remove the reference */
 670         E->JumpTo = 0;
 671     }
 672     CollDeleteAll (&L->JumpFrom);
 673
 674     /* Remove the reference to the owning instruction if it has one. The
 675      * function may be called for a label without an owner when deleting
 676      * unfinished parts of the code. This is unfortunate since it allows
 677      * errors to slip through.
 678      */
 679     if (L->Owner) {
 680         CollDeleteItem (&L->Owner->Labels, L);
 681     }
 682
 683     /* All references removed, delete the label itself */
 684     FreeCodeLabel (L);
 685 }
 686
 687
 688
 689 void MergeCodeLabels (CodeSeg* S)
 690 /* Merge code labels. That means: For each instruction, remove all labels but
 691  * one and adjust references accordingly.
 692  */
 693 {
 694     unsigned I;
 695
 696     /* Walk over all code entries */
 697     unsigned EntryCount = GetCodeEntryCount (S);
 698     for (I = 0; I < EntryCount; ++I) {
 699
 700         CodeLabel* RefLab;
 701         unsigned   J;
 702
 703         /* Get a pointer to the next entry */
 704         CodeEntry* E = GetCodeEntry (S, I);
 705
 706         /* If this entry has zero labels, continue with the next one */
 707         unsigned LabelCount = GetCodeLabelCount (E);
 708         if (LabelCount == 0) {
 709             continue;
 710         }
 711
 712         /* We have at least one label. Use the first one as reference label. */
 713         RefLab = GetCodeLabel (E, 0);
 714
 715         /* Walk through the remaining labels and change references to these
 716          * labels to a reference to the one and only label. Delete the labels
 717          * that are no longer used. To increase performance, walk backwards
 718          * through the list.
 719          */
 720         for (J = LabelCount-1; J >= 1; --J) {
 721
 722             /* Get the next label */
 723             CodeLabel* L = GetCodeLabel (E, J);
 724
 725             /* Move all references from this label to the reference label */
 726             MoveLabelRefs (L, RefLab);
 727
 728             /* Remove the label completely. */
 729             DelCodeLabel (S, L);
 730         }
 731
 732         /* The reference label is the only remaining label. Check if there
 733          * are any references to this label, and delete it if this is not
 734          * the case.
 735          */
 736         if (CollCount (&RefLab->JumpFrom) == 0) {
 737             /* Delete the label */
 738             DelCodeLabel (S, RefLab);
 739         }
 740     }
 741 }
 742
 743
 744
 745 void MoveCodeLabels (CodeSeg* S, struct CodeEntry* Old, struct CodeEntry* New)
 746 /* Move all labels from Old to New. The routine will move the labels itself
 747  * if New does not have any labels, and move references if there is at least
 748  * a label for new. If references are moved, the old label is deleted
 749  * afterwards.
 750  */
 751 {
 752     /* Get the number of labels to move */
 753     unsigned OldLabelCount = GetCodeLabelCount (Old);
 754
 755     /* Does the new entry have itself a label? */
 756     if (CodeEntryHasLabel (New)) {
 757
 758         /* The new entry does already have a label - move references */
 759         CodeLabel* NewLabel = GetCodeLabel (New, 0);
 760         while (OldLabelCount--) {
 761
 762             /* Get the next label */
 763             CodeLabel* OldLabel = GetCodeLabel (Old, OldLabelCount);
 764
 765             /* Move references */
 766             MoveLabelRefs (OldLabel, NewLabel);
 767
 768             /* Delete the label */
 769             DelCodeLabel (S, OldLabel);
 770
 771         }
 772
 773     } else {
 774
 775         /* The new entry does not have a label, just move them */
 776         while (OldLabelCount--) {
 777
 778             /* Move the label to the new entry */
 779             MoveCodeLabel (GetCodeLabel (Old, OldLabelCount), New);
 780
 781         }
 782
 783     }
 784 }
 785
 786
 787
 788 void RemoveCodeLabelRef (CodeSeg* S, struct CodeEntry* E)
 789 /* Remove the reference between E and the label it jumps to. The reference
 790  * will be removed on both sides and E->JumpTo will be 0 after that. If
 791  * the reference was the only one for the label, the label will get
 792  * deleted.
 793  */
 794 {
 795     /* Get a pointer to the label and make sure it exists */
 796     CodeLabel* L = E->JumpTo;
 797     CHECK (L != 0);
 798
 799     /* Delete the entry from the label */
 800     CollDeleteItem (&L->JumpFrom, E);
 801
 802     /* The entry jumps no longer to L */
 803     E->JumpTo = 0;
 804
 805     /* If there are no more references, delete the label */
 806     if (CollCount (&L->JumpFrom) == 0) {
 807         DelCodeLabel (S, L);
 808     }
 809 }
 810
 811
 812
 813 void MoveCodeLabelRef (CodeSeg* S, struct CodeEntry* E, CodeLabel* L)
 814 /* Change the reference of E to L instead of the current one. If this
 815  * was the only reference to the old label, the old label will get
 816  * deleted.
 817  */
 818 {
 819     /* Get the old label */
 820     CodeLabel* OldLabel = E->JumpTo;
 821
 822     /* Be sure that code entry references a label */
 823     PRECONDITION (OldLabel != 0);
 824
 825     /* Remove the reference to our label */
 826     RemoveCodeLabelRef (S, E);
 827
 828     /* Use the new label */
 829     AddLabelRef (L, E);
 830 }
 831
 832
 833
 834 void AddCodeSegHint (CodeSeg* S, unsigned Hint)
 835 /* Add a hint for the preceeding instruction */
 836 {
 837     CodeEntry* E;
 838
 839     /* Get the number of entries in this segment */
 840     unsigned EntryCount = GetCodeEntryCount (S);
 841
 842     /* Must have at least one entry */
 843     CHECK (EntryCount > 0);
 844
 845     /* Get the last entry */
 846     E = GetCodeEntry (S, EntryCount-1);
 847
 848     /* Add the hint */
 849     E->Hints |= Hint;
 850 }
 851
 852
 853
 854 void DelCodeSegAfter (CodeSeg* S, unsigned Last)
 855 /* Delete all entries including the given one */
 856 {
 857     /* Get the number of entries in this segment */
 858     unsigned Count = GetCodeEntryCount (S);
 859
 860     /* First pass: Delete all references to labels. If the reference count
 861      * for a label drops to zero, delete it.
 862      */
 863     unsigned C = Count;
 864     while (Last < C--) {
 865
 866         /* Get the next entry */
 867         CodeEntry* E = GetCodeEntry (S, C);
 868
 869         /* Check if this entry has a label reference */
 870         if (E->JumpTo) {
 871             /* Remove the reference to the label */
 872             RemoveCodeLabelRef (S, E);
 873         }
 874
 875     }
 876
 877     /* Second pass: Delete the instructions. If a label attached to an
 878      * instruction still has references, it must be references from outside
 879      * the deleted area. Don't delete the label in this case, just make it
 880      * ownerless and move it to the label pool.
 881      */
 882     C = Count;
 883     while (Last < C--) {
 884
 885         /* Get the next entry */
 886         CodeEntry* E = GetCodeEntry (S, C);
 887
 888         /* Check if this entry has a label attached */
 889         if (CodeEntryHasLabel (E)) {
 890             /* Move the labels to the pool and clear the owner pointer */
 891             MoveLabelsToPool (S, E);
 892         }
 893
 894         /* Delete the pointer to the entry */
 895         CollDelete (&S->Entries, C);
 896
 897         /* Delete the entry itself */
 898         FreeCodeEntry (E);
 899     }
 900 }
 901
 902
 903
 904 void OutputCodeSeg (const CodeSeg* S, FILE* F)
 905 /* Output the code segment data to a file */
 906 {
 907     unsigned I;
 908
 909     /* Get the number of entries in this segment */
 910     unsigned Count = GetCodeEntryCount (S);
 911
 912     /* If the code segment is empty, bail out here */
 913     if (Count == 0) {
 914         return;
 915     }
 916
 917     /* Output the segment directive */
 918     fprintf (F, ".segment\t\"%s\"\n\n", S->SegName);
 919
 920     /* If this is a segment for a function, enter a function */
 921     if (S->Func) {
 922         fprintf (F, ".proc\t_%s\n\n", S->Func->Name);
 923     }
 924
 925     /* Output all entries */
 926     for (I = 0; I < Count; ++I) {
 927
 928         unsigned char Use;
 929
 930         OutputCodeEntry (CollConstAt (&S->Entries, I), F);
 931
 932         /* Print usage info */
 933         Use = GetRegInfo ((CodeSeg*) S, I+1);
 934         fprintf (F,
 935                  "  Use: %c%c%c\n",
 936                  (Use & REG_A)? 'A' : '_',
 937                  (Use & REG_X)? 'X' : '_',
 938                  (Use & REG_Y)? 'Y' : '_');
 939     }
 940
 941     /* If this is a segment for a function, leave the function */
 942     if (S->Func) {
 943         fprintf (F, "\n.endproc\n\n");
 944     }
 945 }
 946
 947
 948
 949
 950
 951
 952