/*****************************************************************************/ /* */ /* library.c */ /* */ /* Library data structures and helpers for the ar65 archiver */ /* */ /* */ /* */ /* (C) 1998-2013, Ullrich von Bassewitz */ /* Roemerstrasse 52 */ /* D-70794 Filderstadt */ /* EMail: uz@cc65.org */ /* */ /* */ /* This software is provided 'as-is', without any expressed or implied */ /* warranty. In no event will the authors be held liable for any damages */ /* arising from the use of this software. */ /* */ /* Permission is granted to anyone to use this software for any purpose, */ /* including commercial applications, and to alter it and redistribute it */ /* freely, subject to the following restrictions: */ /* */ /* 1. The origin of this software must not be misrepresented; you must not */ /* claim that you wrote the original software. If you use this software */ /* in a product, an acknowledgment in the product documentation would be */ /* appreciated but is not required. */ /* 2. Altered source versions must be plainly marked as such, and must not */ /* be misrepresented as being the original software. */ /* 3. This notice may not be removed or altered from any source */ /* distribution. */ /* */ /*****************************************************************************/ #include #include #include #if defined(_WIN32) # include #else # include # include #endif /* common */ #include "cmdline.h" #include "exprdefs.h" #include "libdefs.h" #include "print.h" #include "symdefs.h" #include "xmalloc.h" /* ar65 */ #include "error.h" #include "exports.h" #include "fileio.h" #include "global.h" #include "library.h" #include "objdata.h" #include "objfile.h" /*****************************************************************************/ /* Data */ /*****************************************************************************/ /* Name of the library file */ const char* LibName = 0; static char* NewLibName = 0; /* File descriptor for the library file */ static FILE* Lib = 0; static FILE* NewLib = 0; /* The library header */ static LibHeader Header = { LIB_MAGIC, LIB_VERSION, 0, 0 }; /*****************************************************************************/ /* Writing file data structures */ /*****************************************************************************/ static void ReadHeader (void) /* Read the header of a library file */ { /* Seek to position zero */ fseek (Lib, 0, SEEK_SET); /* Read the header fields, checking magic and version */ Header.Magic = Read32 (Lib); if (Header.Magic != LIB_MAGIC) { Error ("'%s' is not a valid library file", LibName); } Header.Version = Read16 (Lib); if (Header.Version != LIB_VERSION) { Error ("Wrong data version in '%s'", LibName); } Header.Flags = Read16 (Lib); Header.IndexOffs = Read32 (Lib); } static void ReadIndexEntry (void) /* Read one entry in the index */ { /* Create a new entry and insert it into the list */ ObjData* O = NewObjData (); /* Module name/flags/MTime/Start/Size */ O->Name = ReadStr (Lib); O->Flags = Read16 (Lib); O->MTime = Read32 (Lib); O->Start = Read32 (Lib); O->Size = Read32 (Lib); } static void ReadIndex (void) /* Read the index of a library file */ { unsigned Count, I; /* Seek to the start of the index */ fseek (Lib, Header.IndexOffs, SEEK_SET); /* Read the object file count and calculate the cross ref size */ Count = ReadVar (Lib); /* Read all entries in the index */ while (Count--) { ReadIndexEntry (); } /* Read basic object file data from the actual entries */ for (I = 0; I < CollCount (&ObjPool); ++I) { /* Get the object file entry */ ObjData* O = CollAtUnchecked (&ObjPool, I); /* Read data */ ObjReadData (Lib, O); } } /*****************************************************************************/ /* Writing file data structures */ /*****************************************************************************/ static void WriteHeader (void) /* Write the header to the library file */ { /* Seek to position zero */ fseek (NewLib, 0, SEEK_SET); /* Write the header fields */ Write32 (NewLib, Header.Magic); Write16 (NewLib, Header.Version); Write16 (NewLib, Header.Flags); Write32 (NewLib, Header.IndexOffs); } static void WriteIndexEntry (const ObjData* O) /* Write one index entry */ { /* Module name/flags/MTime/start/size */ WriteStr (NewLib, O->Name); Write16 (NewLib, O->Flags & ~OBJ_HAVEDATA); Write32 (NewLib, O->MTime); Write32 (NewLib, O->Start); Write32 (NewLib, O->Size); } static void WriteIndex (void) /* Write the index of a library file */ { unsigned I; /* Sync I/O in case the last operation was a read */ fseek (NewLib, 0, SEEK_CUR); /* Remember the current offset in the header */ Header.IndexOffs = ftell (NewLib); /* Write the object file count */ WriteVar (NewLib, CollCount (&ObjPool)); /* Write the object files */ for (I = 0; I < CollCount (&ObjPool); ++I) { WriteIndexEntry (CollConstAt (&ObjPool, I)); } } /*****************************************************************************/ /* High level stuff */ /*****************************************************************************/ void LibOpen (const char* Name, int MustExist, int NeedTemp) /* Open an existing library and a temporary copy. If MustExist is true, the ** old library is expected to exist. If NeedTemp is true, a temporary library ** is created. */ { /* Remember the name */ LibName = xstrdup (Name); /* Open the existing library for reading */ Lib = fopen (Name, "rb"); if (Lib == 0) { /* File does not exist */ if (MustExist) { Error ("Library '%s' does not exist", Name); } else { /* Announce the library's creation if ar65 is verbose. */ Print (stdout, 1, "%s: Library '%s' will be created.\n", ProgName, Name); } } else { /* We have an existing file: Read the header */ ReadHeader (); /* Now read the existing index */ ReadIndex (); } if (NeedTemp) { /* Create the temporary library name */ NewLibName = xmalloc (strlen (Name) + (sizeof (".temp-") - 1) + 2 * sizeof (unsigned int) + 1); sprintf (NewLibName, "%s.temp-%X", Name, (unsigned int)getpid ()); /* Create the temporary library */ NewLib = fopen (NewLibName, "w+b"); if (NewLib == 0) { Error ("Cannot create temporary library file: %s", strerror (errno)); } /* Write a dummy header to the temp file */ WriteHeader (); } } unsigned long LibCopyTo (FILE* F, unsigned long Bytes) /* Copy data from F to the temp library file, return the start position in ** the temporary library file. */ { unsigned char Buf [4096]; /* Remember the position */ unsigned long Pos = ftell (NewLib); /* Copy loop */ while (Bytes) { unsigned Count = (Bytes > sizeof (Buf))? sizeof (Buf) : Bytes; ReadData (F, Buf, Count); WriteData (NewLib, Buf, Count); Bytes -= Count; } /* Return the start position */ return Pos; } void LibCopyFrom (unsigned long Pos, unsigned long Bytes, FILE* F) /* Copy data from the library file into another file */ { unsigned char Buf [4096]; /* Seek to the correct position */ fseek (Lib, Pos, SEEK_SET); /* Copy loop */ while (Bytes) { unsigned Count = (Bytes > sizeof (Buf))? sizeof (Buf) : Bytes; ReadData (Lib, Buf, Count); WriteData (F, Buf, Count); Bytes -= Count; } } static void LibCheckExports (ObjData* O) /* Insert all exports from the given object file into the global list ** checking for duplicates. */ { unsigned I; /* Let the user know what we do */ Print (stdout, 2, "Module '%s' (%u exports):\n", O->Name, CollCount (&O->Exports)); /* Insert the exports into the global table */ for (I = 0; I < CollCount (&O->Exports); ++I) { /* Get the name of the export */ const char* Name = CollConstAt (&O->Exports, I); /* Insert the name into the hash table */ Print (stdout, 2, " %s\n", Name); ExpInsert (Name, O); } } void LibClose (void) /* Write remaining data, close both files and copy the temp file to the old ** filename */ { /* Do we have a temporary library? */ if (NewLib) { unsigned I; unsigned char Buf [4096]; size_t Count; /* Walk through the object file list, inserting exports into the ** export list checking for duplicates. Copy any data that is still ** in the old library into the new one. */ for (I = 0; I < CollCount (&ObjPool); ++I) { /* Get a pointer to the object */ ObjData* O = CollAtUnchecked (&ObjPool, I); /* Check exports, make global export table */ LibCheckExports (O); /* Copy data if needed */ if ((O->Flags & OBJ_HAVEDATA) == 0) { /* Data is still in the old library */ fseek (Lib, O->Start, SEEK_SET); O->Start = ftell (NewLib); LibCopyTo (Lib, O->Size); O->Flags |= OBJ_HAVEDATA; } } /* Write the index */ WriteIndex (); /* Write the updated header */ WriteHeader (); /* Close the file */ if (Lib && fclose (Lib) != 0) { Error ("Error closing library: %s", strerror (errno)); } /* Reopen the library and truncate it */ Lib = fopen (LibName, "wb"); if (Lib == 0) { Error ("Cannot open library '%s' for writing: %s", LibName, strerror (errno)); } /* Copy the temporary library to the new one */ fseek (NewLib, 0, SEEK_SET); while ((Count = fread (Buf, 1, sizeof (Buf), NewLib)) != 0) { if (fwrite (Buf, 1, Count, Lib) != Count) { Error ("Cannot write to '%s': %s", LibName, strerror (errno)); } } } /* Close both files */ if (Lib && fclose (Lib) != 0) { Error ("Problem closing '%s': %s", LibName, strerror (errno)); } if (NewLib && fclose (NewLib) != 0) { Error ("Problem closing temporary library file '%s': %s", NewLibName, strerror (errno)); } if (NewLibName && remove (NewLibName) != 0) { Error ("Problem deleting temporary library file '%s': %s", NewLibName, strerror (errno)); } }