oscar64/oscar64/Linker.cpp

758 lines
17 KiB
C++

#include "Linker.h"
#include <string.h>
#include <stdio.h>
#include "CompilerTypes.h"
LinkerRegion::LinkerRegion(void)
: mSections(nullptr), mFreeChunks(FreeChunk{ 0, 0 } )
{}
LinkerSection::LinkerSection(void)
: mObjects(nullptr), mSections(nullptr)
{}
LinkerObject::LinkerObject(void)
: mReferences(nullptr), mNumTemporaries(0), mSize(0), mAlignment(1), mStackSection(nullptr)
{}
LinkerObject::~LinkerObject(void)
{
}
void LinkerObject::AddReference(const LinkerReference& ref)
{
LinkerReference* nref = new LinkerReference(ref);
mReferences.Push(nref);
}
void LinkerObject::AddData(const uint8* data, int size)
{
mSize = size;
mData = new uint8[size];
memcpy(mData, data, size);
}
uint8* LinkerObject::AddSpace(int size)
{
mSize = size;
mData = new uint8[size];
memset(mData, 0, size);
return mData;
}
Linker::Linker(Errors* errors)
: mErrors(errors), mSections(nullptr), mReferences(nullptr), mObjects(nullptr), mRegions(nullptr), mCompilerOptions(COPT_DEFAULT)
{
for (int i = 0; i < 64; i++)
{
mCartridgeBankUsed[i] = 0;
memset(mCartridge[i], 0, 0x4000);
}
memset(mMemory, 0, 0x10000);
}
Linker::~Linker(void)
{
}
LinkerRegion* Linker::AddRegion(const Ident* region, int start, int end)
{
LinkerRegion* lrgn = new LinkerRegion();
lrgn->mIdent = region;
lrgn->mStart = start;
lrgn->mReloc = 0;
lrgn->mEnd = end;
lrgn->mUsed = 0;
lrgn->mNonzero = 0;
lrgn->mCartridgeBanks = 0;
lrgn->mFlags = 0;
mRegions.Push(lrgn);
return lrgn;
}
LinkerRegion* Linker::FindRegion(const Ident* region)
{
for (int i = 0; i < mRegions.Size(); i++)
{
if (mRegions[i]->mIdent == region)
return mRegions[i];
}
return nullptr;
}
LinkerSection* Linker::AddSection(const Ident* section, LinkerSectionType type)
{
LinkerSection* lsec = new LinkerSection;
lsec->mIdent = section;
lsec->mType = type;
mSections.Push(lsec);
return lsec;
}
LinkerSection* Linker::FindSection(const Ident* section)
{
for (int i = 0; i < mSections.Size(); i++)
{
if (mSections[i]->mIdent == section)
return mSections[i];
}
return nullptr;
}
bool Linker::IsSectionPlaced(LinkerSection* section)
{
for (int i = 0; i < mRegions.Size(); i++)
{
LinkerRegion* rgn = mRegions[i];
for (int j = 0; j < rgn->mSections.Size(); j++)
if (section == rgn->mSections[j])
return true;
}
return false;
}
LinkerObject* Linker::FindObjectByAddr(int addr)
{
for (int i = 0; i < mObjects.Size(); i++)
{
LinkerObject* lobj = mObjects[i];
if (lobj->mFlags & LOBJF_PLACED)
{
if (addr >= lobj->mAddress && addr < lobj->mAddress + lobj->mSize)
return lobj;
}
}
return nullptr;
}
LinkerObject * Linker::AddObject(const Location& location, const Ident* ident, LinkerSection * section, LinkerObjectType type, int alignment)
{
LinkerObject* obj = new LinkerObject;
obj->mLocation = location;
obj->mID = mObjects.Size();
obj->mType = type;
obj->mData = nullptr;
obj->mSize = 0;
obj->mIdent = ident;
obj->mSection = section;
obj->mRegion = nullptr;
obj->mProc = nullptr;
obj->mFlags = 0;
obj->mAlignment = alignment;
section->mObjects.Push(obj);
mObjects.Push(obj);
return obj;
}
void Linker::CollectReferences(void)
{
for (int i = 0; i < mObjects.Size(); i++)
{
LinkerObject* lobj(mObjects[i]);
for (int j = 0; j < lobj->mReferences.Size(); j++)
mReferences.Push(lobj->mReferences[j]);
}
}
void Linker::ReferenceObject(LinkerObject* obj)
{
if (!(obj->mFlags & LOBJF_REFERENCED))
{
obj->mFlags |= LOBJF_REFERENCED;
for (int i = 0; i < mReferences.Size(); i++)
{
LinkerReference* ref = mReferences[i];
if (ref->mObject == obj)
ReferenceObject(ref->mRefObject);
}
}
}
bool LinkerRegion::Allocate(Linker * linker, LinkerObject* lobj)
{
int i = 0;
while (i < mFreeChunks.Size())
{
int start = (mFreeChunks[i].mStart + lobj->mAlignment - 1) & ~(lobj->mAlignment - 1);
int end = start + lobj->mSize;
if (!(linker->mCompilerOptions & COPT_OPTIMIZE_CODE_SIZE) && (lobj->mFlags & LOBJF_NO_CROSS) && lobj->mSize <= 256 && (start & 0xff00) != ((end - 1) & 0xff00))
;
else if (end <= mFreeChunks[i].mEnd)
{
lobj->mFlags |= LOBJF_PLACED;
lobj->mAddress = start;
lobj->mRefAddress = start + mReloc;
lobj->mRegion = this;
if (start == mFreeChunks[i].mStart)
{
if (end == mFreeChunks[i].mEnd)
mFreeChunks.Remove(i);
else
mFreeChunks[i].mStart = end;
}
else if (end == mFreeChunks[i].mEnd)
{
mFreeChunks[i].mEnd = start;
}
else
{
mFreeChunks.Insert(i + 1, FreeChunk{ end, mFreeChunks[i].mEnd } );
mFreeChunks[i].mEnd = start;
}
return true;
}
i++;
}
int start = (mStart + mUsed + lobj->mAlignment - 1) & ~(lobj->mAlignment - 1);
int end = start + lobj->mSize;
if (!(linker->mCompilerOptions & COPT_OPTIMIZE_CODE_SIZE) && (lobj->mFlags & LOBJF_NO_CROSS) && lobj->mSize <= 256 && (start & 0xff00) != ((end - 1) & 0xff00))
{
start = (start + 0x00ff) & 0xff00;
end = start + lobj->mSize;
}
if (end <= mEnd)
{
lobj->mFlags |= LOBJF_PLACED;
lobj->mAddress = start;
lobj->mRefAddress = start + mReloc;
lobj->mRegion = this;
#if 1
if (start != mStart + mUsed)
mFreeChunks.Push( FreeChunk{ mStart + mUsed, start } );
#endif
mUsed = end - mStart;
return true;
}
return false;
}
void LinkerRegion::PlaceStackSection(LinkerSection* stackSection, LinkerSection* section)
{
if (!section->mEnd)
{
int start = stackSection->mEnd;
for(int i=0; i<section->mSections.Size(); i++)
{
PlaceStackSection(stackSection, section->mSections[i]);
if (section->mSections[i]->mStart < start)
start = section->mSections[i]->mStart;
}
section->mStart = start;
section->mEnd = start;
for (int i = 0; i < section->mObjects.Size(); i++)
{
LinkerObject* lobj = section->mObjects[i];
if (lobj->mFlags & LOBJF_REFERENCED)
{
section->mStart -= lobj->mSize;
section->mSize += lobj->mSize;
lobj->mFlags |= LOBJF_PLACED;
lobj->mAddress = section->mStart;
lobj->mRefAddress = section->mStart + mReloc;
lobj->mRegion = this;
}
}
if (stackSection->mStart > section->mStart)
stackSection->mStart = section->mStart;
}
}
void Linker::Link(void)
{
if (mErrors->mErrorCount == 0)
{
for (int i = 0; i < mSections.Size(); i++)
{
LinkerSection* lsec = mSections[i];
lsec->mStart = 0x10000;
lsec->mEnd = 0x0000;
}
// Move objects into regions
for (int i = 0; i < mRegions.Size(); i++)
{
LinkerRegion* lrgn = mRegions[i];
for (int j = 0; j < lrgn->mSections.Size(); j++)
{
LinkerSection* lsec = lrgn->mSections[j];
for (int k = 0; k < lsec->mObjects.Size(); k++)
{
LinkerObject* lobj = lsec->mObjects[k];
if ((lobj->mFlags & LOBJF_REFERENCED) && !(lobj->mFlags & LOBJF_PLACED) && lrgn->Allocate(this, lobj))
{
if (lobj->mIdent && lobj->mIdent->mString && (mCompilerOptions & COPT_VERBOSE2))
printf("Placed object <%s> $%04x - $%04x\n", lobj->mIdent->mString, lobj->mAddress, lobj->mAddress + lobj->mSize);
if (lobj->mAddress < lsec->mStart)
lsec->mStart = lobj->mAddress;
if (lobj->mAddress + lobj->mSize > lsec->mEnd)
lsec->mEnd = lobj->mAddress + lobj->mSize;
if (lsec->mType == LST_DATA && lsec->mEnd > lrgn->mNonzero)
lrgn->mNonzero = lsec->mEnd;
}
}
}
for (int j = 0; j < lrgn->mSections.Size(); j++)
{
LinkerSection* lsec = lrgn->mSections[j];
if (lsec->mType == LST_BSS && lsec->mStart < lrgn->mNonzero)
lsec->mStart = lrgn->mNonzero;
if (lsec->mEnd < lsec->mStart)
lsec->mEnd = lsec->mStart;
}
}
mProgramStart = 0xffff;
mProgramEnd = 0x0000;
for (int i = 0; i < mRegions.Size(); i++)
{
LinkerRegion* lrgn = mRegions[i];
if (lrgn->mNonzero)
{
if (lrgn->mStart < mProgramStart)
mProgramStart = lrgn->mStart;
if (lrgn->mNonzero > mProgramEnd)
mProgramEnd = lrgn->mNonzero;
}
}
// Place stack segment
for (int i = 0; i < mRegions.Size(); i++)
{
LinkerRegion* lrgn = mRegions[i];
for (int j = 0; j < lrgn->mSections.Size(); j++)
{
LinkerSection* lsec = lrgn->mSections[j];
if (lsec->mType == LST_STACK)
{
lsec->mStart = lsec->mEnd = lrgn->mEnd;
lrgn->mEnd = lsec->mStart - lsec->mSize;
for(int i=0; i<lsec->mSections.Size(); i++)
lrgn->PlaceStackSection(lsec, lsec->mSections[i]);
lsec->mEnd = lsec->mStart;
lsec->mStart = lrgn->mEnd;
}
}
}
// Now expand the heap section to cover the remainder of the region
for (int i = 0; i < mRegions.Size(); i++)
{
LinkerRegion* lrgn = mRegions[i];
for (int j = 0; j < lrgn->mSections.Size(); j++)
{
LinkerSection* lsec = lrgn->mSections[j];
if (lsec->mType == LST_HEAP)
{
lsec->mStart = lrgn->mStart + lrgn->mUsed;
lsec->mEnd = lrgn->mEnd;
}
}
}
for (int i = 0; i < mObjects.Size(); i++)
{
LinkerObject* obj = mObjects[i];
if (obj->mType == LOT_SECTION_START)
{
obj->mAddress = obj->mSection->mStart;
obj->mRefAddress = obj->mAddress + (obj->mRegion ? obj->mRegion->mReloc : 0);
}
else if (obj->mType == LOT_SECTION_END)
{
obj->mAddress = obj->mSection->mEnd;
obj->mRefAddress = obj->mAddress + (obj->mRegion ? obj->mRegion->mReloc : 0);
}
else if (obj->mFlags & LOBJF_REFERENCED)
{
if (!obj->mRegion)
mErrors->Error(obj->mLocation, ERRR_INSUFFICIENT_MEMORY, "Could not place object", obj->mIdent->mString);
else if (obj->mRegion->mCartridgeBanks != 0)
{
for (int i = 0; i < 64; i++)
{
if (obj->mRegion->mCartridgeBanks & (1ULL << i))
{
mCartridgeBankUsed[i] = true;
memcpy(mCartridge[i] + obj->mAddress - 0x8000, obj->mData, obj->mSize);
}
}
}
else
{
memcpy(mMemory + obj->mAddress, obj->mData, obj->mSize);
}
}
}
for (int i = 0; i < mReferences.Size(); i++)
{
LinkerReference* ref = mReferences[i];
LinkerObject* obj = ref->mObject;
if (obj->mFlags & LOBJF_REFERENCED)
{
if (obj->mRegion)
{
LinkerObject* robj = ref->mRefObject;
int raddr = robj->mRefAddress + ref->mRefOffset;
uint8* dp;
if (obj->mRegion->mCartridgeBanks)
{
for (int i = 0; i < 64; i++)
{
if (obj->mRegion->mCartridgeBanks & (1ULL << i))
{
dp = mCartridge[i] + obj->mAddress - 0x8000 + ref->mOffset;
if (ref->mFlags & LREF_LOWBYTE)
{
*dp++ = raddr & 0xff;
}
if (ref->mFlags & LREF_HIGHBYTE)
{
*dp++ = (raddr >> 8) & 0xff;
}
if (ref->mFlags & LREF_TEMPORARY)
*dp += obj->mTemporaries[ref->mRefOffset];
}
}
}
else
{
dp = mMemory + obj->mAddress + ref->mOffset;
if (ref->mFlags & LREF_LOWBYTE)
{
*dp++ = raddr & 0xff;
}
if (ref->mFlags & LREF_HIGHBYTE)
{
*dp++ = (raddr >> 8) & 0xff;
}
if (ref->mFlags & LREF_TEMPORARY)
*dp += obj->mTemporaries[ref->mRefOffset];
}
}
}
}
}
}
static const char * LinkerObjectTypeNames[] =
{
"NONE",
"PAD",
"BASIC",
"BYTE_CODE",
"NATIVE_CODE",
"RUNTIME",
"DATA",
"BSS",
"HEAP",
"STACK",
"START",
"END"
};
static const char* LinkerSectionTypeNames[] = {
"NONE",
"DATA",
"BSS",
"HEAP",
"STACK",
"SSTACK",
"ZEROPAGE"
};
bool Linker::WriteBinFile(const char* filename)
{
FILE* file;
fopen_s(&file, filename, "wb");
if (file)
{
int done = fwrite(mMemory + mProgramStart, 1, mProgramEnd - mProgramStart, file);
fclose(file);
return done == mProgramEnd - mProgramStart;
}
else
return false;
}
bool Linker::WritePrgFile(const char* filename)
{
FILE* file;
fopen_s(&file, filename, "wb");
if (file)
{
mMemory[mProgramStart - 2] = mProgramStart & 0xff;
mMemory[mProgramStart - 1] = mProgramStart >> 8;
int done = fwrite(mMemory + mProgramStart - 2, 1, mProgramEnd - mProgramStart + 2, file);
fclose(file);
return done == mProgramEnd - mProgramStart + 2;
}
else
return false;
}
bool Linker::WriteCrtFile(const char* filename)
{
FILE* file;
fopen_s(&file, filename, "wb");
if (file)
{
struct CRIHeader
{
char mSignature[16];
uint32 mHeaderLength;
uint16 mVersion, mHardware;
uint8 mExrom, mGameLine;
uint8 mPad[6];
char mName[32];
} criHeader = { 0 };
memcpy(criHeader.mSignature, "C64 CARTRIDGE ", 16);
criHeader.mHeaderLength = 0x40000000;
criHeader.mVersion = 0x0001;
criHeader.mHardware = 0x2000;
criHeader.mExrom = 0;
criHeader.mGameLine = 0;
memset(criHeader.mName, 0, 32);
strcpy_s(criHeader.mName, "OSCAR");
fwrite(&criHeader, sizeof(CRIHeader), 1, file);
struct CHIPHeader
{
char mSignature[4];
uint32 mPacketLength;
uint16 mChipType, mBankNumber, mLoadAddress, mImageSize;
} chipHeader = { 0 };
memcpy(chipHeader.mSignature, "CHIP", 4);
chipHeader.mPacketLength = 0x10200000;
chipHeader.mChipType = 0;
chipHeader.mBankNumber = 0;
chipHeader.mImageSize = 0x0020;
uint8 bootmem[8192];
memset(bootmem, 0, 0x2000);
chipHeader.mLoadAddress = 0x0080;
fwrite(&chipHeader, sizeof(chipHeader), 1, file);
fwrite(mMemory + 0x0800, 1, 0x2000, file);
memcpy(bootmem, mMemory + 0x2800, 0x1800);
bootmem[0x1ffc] = 0x00;
bootmem[0x1ffd] = 0xff;
uint8 bootcode[] = {
0xa9, 0x87,
0x8d, 0x02, 0xde,
0xa9, 0x00,
0x8d, 0x00, 0xde,
0x6c, 0xfc, 0xff
};
int j = 0x1f00;
for (int i = 0; i < sizeof(bootcode); i++)
{
bootmem[j++] = 0xa9;
bootmem[j++] = bootcode[i];
bootmem[j++] = 0x8d;
bootmem[j++] = i;
bootmem[j++] = 0x04;
}
bootmem[j++] = 0x4c;
bootmem[j++] = 0x00;
bootmem[j++] = 0x04;
chipHeader.mLoadAddress = 0x00e0;
fwrite(&chipHeader, sizeof(chipHeader), 1, file);
fwrite(bootmem, 1, 0x2000, file);
for (int i = 1; i < 64; i++)
{
if (mCartridgeBankUsed[i])
{
chipHeader.mBankNumber = i << 8;
chipHeader.mLoadAddress = 0x0080;
fwrite(&chipHeader, sizeof(chipHeader), 1, file);
fwrite(mCartridge[i] + 0x0000, 1, 0x2000, file);
chipHeader.mLoadAddress = 0x00a0;
fwrite(&chipHeader, sizeof(chipHeader), 1, file);
fwrite(mCartridge[i] + 0x2000, 1, 0x2000, file);
}
}
fclose(file);
return true;
}
else
return false;
}
bool Linker::WriteMapFile(const char* filename)
{
FILE* file;
fopen_s(&file, filename, "wb");
if (file)
{
fprintf(file, "sections\n");
for (int i = 0; i < mSections.Size(); i++)
{
LinkerSection* lsec = mSections[i];
fprintf(file, "%04x - %04x : %s, %s\n", lsec->mStart, lsec->mEnd, LinkerSectionTypeNames[lsec->mType], lsec->mIdent->mString);
}
fprintf(file, "\nregions\n");
for (int i = 0; i < mRegions.Size(); i++)
{
LinkerRegion * lrgn = mRegions[i];
fprintf(file, "%04x - %04x : %04x, %04x, %s\n", lrgn->mStart, lrgn->mEnd, lrgn->mNonzero, lrgn->mUsed, lrgn->mIdent->mString);
}
fprintf(file, "\nobjects\n");
for (int i = 0; i < mObjects.Size(); i++)
{
LinkerObject* obj = mObjects[i];
if (obj->mFlags & LOBJF_REFERENCED)
{
if (obj->mIdent)
fprintf(file, "%04x - %04x : %s, %s:%s\n", obj->mAddress, obj->mAddress + obj->mSize, obj->mIdent->mString, LinkerObjectTypeNames[obj->mType], obj->mSection->mIdent->mString);
else
fprintf(file, "%04x - %04x : *, %s:%s\n", obj->mAddress, obj->mAddress + obj->mSize, LinkerObjectTypeNames[obj->mType], obj->mSection->mIdent->mString);
}
}
fclose(file);
return true;
}
else
return false;
}
bool Linker::WriteLblFile(const char* filename)
{
FILE* file;
fopen_s(&file, filename, "wb");
if (file)
{
for (int i = 0; i < mObjects.Size(); i++)
{
LinkerObject* obj = mObjects[i];
if (obj->mFlags & LOBJF_REFERENCED)
{
if (obj->mIdent)
fprintf(file, "al %04x .%s\n", obj->mAddress, obj->mIdent->mString);
}
}
fclose(file);
return true;
}
else
return false;
}
bool Linker::WriteAsmFile(const char* filename)
{
FILE* file;
fopen_s(&file, filename, "wb");
if (file)
{
for (int i = 0; i < mObjects.Size(); i++)
{
LinkerObject* obj = mObjects[i];
if (obj->mFlags & LOBJF_REFERENCED)
{
switch (obj->mType)
{
case LOT_BYTE_CODE:
mByteCodeDisassembler.Disassemble(file, mMemory, 0, obj->mAddress, obj->mSize, obj->mProc, obj->mIdent, this);
break;
case LOT_NATIVE_CODE:
if (obj->mRegion->mCartridgeBanks)
{
int i = 0;
while (!(obj->mRegion->mCartridgeBanks & (1ULL << i)))
i++;
mNativeDisassembler.Disassemble(file, mCartridge[i] - 0x8000, i, obj->mAddress, obj->mSize, obj->mProc, obj->mIdent, this);
}
else
mNativeDisassembler.Disassemble(file, mMemory, 0, obj->mAddress, obj->mSize, obj->mProc, obj->mIdent, this);
break;
case LOT_DATA:
if (obj->mRegion->mCartridgeBanks)
{
int i = 0;
while (!(obj->mRegion->mCartridgeBanks & (1ULL << i)))
i++;
mNativeDisassembler.DumpMemory(file, mCartridge[i] - 0x8000, i, obj->mAddress, obj->mSize, obj->mProc, obj->mIdent, this, obj);
}
else
mNativeDisassembler.DumpMemory(file, mMemory, 0, obj->mAddress, obj->mSize, obj->mProc, obj->mIdent, this, obj);
break;
}
}
}
fclose(file);
return true;
}
else
return false;
}