Changeset 223 in dev for trunk/abdev/BasicCompiler_Common/PESchedule.cpp

Timestamp:

Jul 19, 2007, 3:38:13 AM (17 years ago)

Author:

dai_9181

Message:

 

File:

• trunk/abdev/BasicCompiler_Common/PESchedule.cpp (modified) (6 diffs)

trunk/abdev/BasicCompiler_Common/PESchedule.cpp

@@ -25 +25 @@
     length=0;
     NowPageRVAToReloc=0;
-
-    pdwData_CodeSection=(DWORD *)HeapAlloc(hHeap,0,1);
-    iNum_CodeSection=0;
-    pdwData_DataSection=(DWORD *)HeapAlloc(hHeap,0,1);
-    iNum_DataSection=0;
 }
 CReloc::~CReloc(){
@@ -39 +34 @@
         buffer=0;
     }
-
-    HeapDefaultFree(pdwData_CodeSection);
-    HeapDefaultFree(pdwData_DataSection);
 }
 void CReloc::copy(CReloc *por){
@@ -57 +49 @@
     NowCountAddrToReloc=por->NowCountAddrToReloc;
 
-    iNum_CodeSection=por->iNum_CodeSection;
-    pdwData_CodeSection=(DWORD *)HeapAlloc(hHeap,0,iNum_CodeSection*sizeof(DWORD));
-    memcpy(pdwData_CodeSection,por->pdwData_CodeSection,iNum_CodeSection*sizeof(DWORD));
-
-    iNum_DataSection=por->iNum_DataSection;
-    pdwData_DataSection=(DWORD *)HeapAlloc(hHeap,0,iNum_DataSection*sizeof(DWORD));
-    memcpy(pdwData_DataSection,por->pdwData_DataSection,iNum_DataSection*sizeof(DWORD));
+    codeSectionAddresses = por->codeSectionAddresses;
+
+    dataSectionAddresses = por->dataSectionAddresses;
 }
 
@@ -70 +58 @@
     if(!bDll) return;
 
-    pdwData_CodeSection=(DWORD *)HeapReAlloc(hHeap,0,pdwData_CodeSection,(iNum_CodeSection+1)*sizeof(DWORD));
-    pdwData_CodeSection[iNum_CodeSection]=addr;
-    iNum_CodeSection++;
+    codeSectionAddresses.push_back( addr );
 }
 void CReloc::AddSchedule_DataSection(DWORD addr){
@@ -78 +64 @@
     if(!bDll) return;
 
-    pdwData_DataSection=(DWORD *)HeapReAlloc(hHeap,0,pdwData_DataSection,(iNum_DataSection+1)*sizeof(DWORD));
-    pdwData_DataSection[iNum_DataSection]=addr;
-    iNum_DataSection++;
+    dataSectionAddresses.push_back( addr );
 }
 
@@ -136 +120 @@
     NowPageRVAToReloc=0;
 
-    int i;
-    for(i=0;i<iNum_CodeSection;i++){
+    BOOST_FOREACH( DWORD addr, codeSectionAddresses )
+    {
         extern int MemPos_CodeSection;
-        __add(MemPos_CodeSection + pdwData_CodeSection[i]);
-    }
-
-    for(i=0;i<iNum_DataSection;i++){
+        __add(MemPos_CodeSection + addr);
+    }
+
+    BOOST_FOREACH( DWORD addr, dataSectionAddresses )
+    {
         extern int MemPos_DataSection;
-        __add(MemPos_DataSection + pdwData_DataSection[i]);
+        __add(MemPos_DataSection + addr);
     }
 }