manip.c 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143
  1. /* manip.c: Routines to manipulate the internal data
  2. Copyright (C) 2003 Sebastian Reichelt
  3. This program is free software; you can redistribute it and/or modify
  4. it under the terms of the GNU General Public License as published by
  5. the Free Software Foundation; either version 2, or (at your option)
  6. any later version.
  7. This program is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU General Public License for more details.
  11. You should have received a copy of the GNU General Public License
  12. along with this program; if not, write to the Free Software Foundation,
  13. Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */
  14. #include "manip.h"
  15. #include <stdlib.h>
  16. #include <string.h>
  17. // Free the archive tree.
  18. void FreeArchive (ARCHIVE *Archive)
  19. {
  20. OBJECT_FILE *ObjectFile, *NextObjectFile;
  21. // Free symbol table, if one exists.
  22. if (Archive->SymbolTable)
  23. free (Archive->SymbolTable);
  24. // Free all object files.
  25. for (ObjectFile = GetLast (Archive->ObjectFiles); ObjectFile; ObjectFile = NextObjectFile)
  26. {
  27. SYMBOL *Symbol, *NextSymbol;
  28. NextObjectFile = GetPrev (ObjectFile);
  29. if (ObjectFile->Data)
  30. free (ObjectFile->Data);
  31. // Free all symbols.
  32. for (Symbol = GetLast (ObjectFile->Symbols); Symbol; Symbol = NextSymbol)
  33. {
  34. NextSymbol = GetPrev (Symbol);
  35. free (Symbol);
  36. }
  37. free (ObjectFile);
  38. }
  39. // Set all data to 0.
  40. memset (Archive, 0, sizeof (ARCHIVE));
  41. }
  42. // Create a symbol table for an archive.
  43. BOOLEAN CreateSymbolTable (ARCHIVE *Archive)
  44. {
  45. if (Archive->SymbolTable)
  46. {
  47. free (Archive->SymbolTable);
  48. Archive->SymbolTable = NULL;
  49. }
  50. // Only create a symbol table if the file contains symbols.
  51. if (Archive->SymbolCount > 0)
  52. {
  53. SYMBOL_TABLE *SymbolTable = calloc (1, sizeof (SYMBOL_TABLE) + (Archive->SymbolCount * sizeof (SYMBOL *)));
  54. if (SymbolTable)
  55. {
  56. OBJECT_FILE *ObjectFile;
  57. SymbolTable->Parent = Archive;
  58. SymbolTable->Size = 4; // The first 4 bytes represent the number of symbols.
  59. // Add all symbols to the symbol table.
  60. for_each (ObjectFile, Archive->ObjectFiles)
  61. {
  62. SYMBOL *Symbol;
  63. for_each (Symbol, ObjectFile->Symbols)
  64. {
  65. if (SymbolTable->SymbolCount < Archive->SymbolCount)
  66. {
  67. SymbolTable->Symbols [SymbolTable->SymbolCount++] = Symbol;
  68. SymbolTable->Size += 4 + Symbol->NameLength + 1; // 4 bytes for the file offset, 1 byte for the terminating null character.
  69. }
  70. else
  71. {
  72. Error (NULL, "Internal symbol table inconsistency.");
  73. goto BreakSymbolLoop;
  74. }
  75. }
  76. }
  77. BreakSymbolLoop:
  78. Archive->SymbolTable = SymbolTable;
  79. return TRUE;
  80. }
  81. else
  82. Error (NULL, "Not enough memory to create symbol table.");
  83. }
  84. return FALSE;
  85. }
  86. // Fill the "export help" fields of the archive.
  87. BOOLEAN FillExportHelpFields (ARCHIVE *Archive)
  88. {
  89. // The symbol table (if it exists) is always the first member.
  90. if (Archive->SymbolTable)
  91. {
  92. Archive->SymbolTable->ArMemberOffset = GetArStartOffset ();
  93. Archive->SymbolTable->ArMemberSize = GetArMemberSize (Archive->SymbolTable->Size);
  94. }
  95. // Get the dimensions of each archive member.
  96. {
  97. OBJECT_FILE *ObjectFile, *PrevObjectFile = NULL;
  98. for_each (ObjectFile, Archive->ObjectFiles)
  99. {
  100. if (PrevObjectFile)
  101. ObjectFile->ArMemberOffset = PrevObjectFile->ArMemberOffset + PrevObjectFile->ArMemberSize;
  102. else if (Archive->SymbolTable)
  103. ObjectFile->ArMemberOffset = Archive->SymbolTable->ArMemberOffset + Archive->SymbolTable->ArMemberSize;
  104. else
  105. ObjectFile->ArMemberOffset = GetArStartOffset ();
  106. ObjectFile->ArMemberSize = GetArMemberSize (ObjectFile->Size);
  107. PrevObjectFile = ObjectFile;
  108. }
  109. // Update the size of the archive.
  110. if (PrevObjectFile)
  111. Archive->ArFileSize = PrevObjectFile->ArMemberOffset + PrevObjectFile->ArMemberSize;
  112. }
  113. return TRUE;
  114. }