Move options files from <build_dir>/build_options to build_options/<build_dir>.

[gem5.git] / dev / disk_image.cc

diff --git a/dev/disk_image.cc b/dev/disk_image.cc
index 78599f8d72ee0c4d308432c31da48e06b39e41ea..106723c554d50ecacdd0f817d14d5b306fd3967b 100644 (file)
--- a/dev/disk_image.cc
+++ b/dev/disk_image.cc
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2003 The Regents of The University of Michigan
+ * Copyright (c) 2001-2005 The Regents of The University of Michigan
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -26,7 +26,7 @@
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
-/* @file
+/** @file
  * Disk Image Definitions
  */
 
@@ -46,6 +46,7 @@
 #include "dev/disk_image.hh"
 #include "sim/builder.hh"
 #include "sim/sim_exit.hh"
+#include "targetarch/byte_swap.hh"
 
 using namespace std;
 
@@ -109,7 +110,7 @@ RawDiskImage::read(uint8_t *data, off_t offset) const
     if (stream.seekg(offset * SectorSize, ios::beg) < 0)
         panic("Could not seek to location in file");
 
-    off_t pos = stream.tellg();
+    streampos pos = stream.tellg();
     stream.read((char *)data, SectorSize);
 
     DPRINTF(DiskImageRead, "read: offset=%d\n", (uint64_t)offset);
@@ -136,7 +137,7 @@ RawDiskImage::write(const uint8_t *data, off_t offset)
     DPRINTF(DiskImageWrite, "write: offset=%d\n", (uint64_t)offset);
     DDUMP(DiskImageWrite, data, SectorSize);
 
-    off_t pos = stream.tellp();
+    streampos pos = stream.tellp();
     stream.write((const char *)data, SectorSize);
     return stream.tellp() - pos;
 }
@@ -190,7 +191,7 @@ CowDiskImage::CowDiskImage(const string &name, DiskImage *kid, int hash_size,
                            const string &file, bool read_only)
     : DiskImage(name), filename(file), child(kid), table(NULL)
 {
-    if (!open()) {
+    if (!open(filename)) {
         assert(!read_only && "why have a non-existent read only file?");
         init(hash_size);
     }
@@ -227,40 +228,50 @@ SafeRead(ifstream &stream, void *data, int count)
 template<class T>
 void
 SafeRead(ifstream &stream, T &data)
-{ SafeRead(stream, &data, sizeof(data)); }
+{
+    SafeRead(stream, &data, sizeof(data));
+}
+
+template<class T>
+void
+SafeReadSwap(ifstream &stream, T &data)
+{
+    SafeRead(stream, &data, sizeof(data));
+    data = letoh(data); //is this the proper byte order conversion?
+}
 
 bool
-CowDiskImage::open()
+CowDiskImage::open(const string &file)
 {
-    ifstream stream(filename.c_str());
+    ifstream stream(file.c_str());
     if (!stream.is_open())
         return false;
 
     if (stream.fail() || stream.bad())
-        panic("Error opening %s", filename);
+        panic("Error opening %s", file);
 
     uint64_t magic;
     SafeRead(stream, magic);
 
     if (memcmp(&magic, "COWDISK!", sizeof(magic)) != 0)
-        panic("Could not open %s: Invalid magic", filename);
+        panic("Could not open %s: Invalid magic", file);
 
     uint32_t major, minor;
-    SafeRead(stream, major);
-    SafeRead(stream, minor);
+    SafeReadSwap(stream, major);
+    SafeReadSwap(stream, minor);
 
     if (major != VersionMajor && minor != VersionMinor)
         panic("Could not open %s: invalid version %d.%d != %d.%d",
-              filename, major, minor, VersionMajor, VersionMinor);
+              file, major, minor, VersionMajor, VersionMinor);
 
     uint64_t sector_count;
-    SafeRead(stream, sector_count);
+    SafeReadSwap(stream, sector_count);
     table = new SectorTable(sector_count);
 
 
     for (uint64_t i = 0; i < sector_count; i++) {
         uint64_t offset;
-        SafeRead(stream, offset);
+        SafeReadSwap(stream, offset);
 
         Sector *sector = new Sector;
         SafeRead(stream, sector, sizeof(Sector));
@@ -300,25 +311,40 @@ SafeWrite(ofstream &stream, const void *data, int count)
 template<class T>
 void
 SafeWrite(ofstream &stream, const T &data)
-{ SafeWrite(stream, &data, sizeof(data)); }
+{
+    SafeWrite(stream, &data, sizeof(data));
+}
 
+template<class T>
+void
+SafeWriteSwap(ofstream &stream, const T &data)
+{
+    T swappeddata = letoh(data); //is this the proper byte order conversion?
+    SafeWrite(stream, &swappeddata, sizeof(data));
+}
 void
 CowDiskImage::save()
+{
+    save(filename);
+}
+
+void
+CowDiskImage::save(const string &file)
 {
     if (!initialized)
         panic("RawDiskImage not initialized");
 
-    ofstream stream(filename.c_str());
+    ofstream stream(file.c_str());
     if (!stream.is_open() || stream.fail() || stream.bad())
-        panic("Error opening %s", filename);
+        panic("Error opening %s", file);
 
     uint64_t magic;
     memcpy(&magic, "COWDISK!", sizeof(magic));
     SafeWrite(stream, magic);
 
-    SafeWrite(stream, (uint32_t)VersionMajor);
-    SafeWrite(stream, (uint32_t)VersionMinor);
-    SafeWrite(stream, (uint64_t)table->size());
+    SafeWriteSwap(stream, (uint32_t)VersionMajor);
+    SafeWriteSwap(stream, (uint32_t)VersionMinor);
+    SafeWriteSwap(stream, (uint64_t)table->size());
 
     uint64_t size = table->size();
     SectorTable::iterator iter = table->begin();
@@ -328,7 +354,7 @@ CowDiskImage::save()
         if (iter == end)
             panic("Incorrect Table Size during save of COW disk image");
 
-        SafeWrite(stream, (uint64_t)(*iter).first);
+        SafeWriteSwap(stream, (uint64_t)(*iter).first);
         SafeWrite(stream, (*iter).second->data, sizeof(Sector));
         ++iter;
     }
@@ -396,6 +422,23 @@ CowDiskImage::write(const uint8_t *data, off_t offset)
     return SectorSize;
 }
 
+void
+CowDiskImage::serialize(ostream &os)
+{
+    string cowFilename = name() + ".cow";
+    SERIALIZE_SCALAR(cowFilename);
+    save(Checkpoint::dir() + "/" + cowFilename);
+}
+
+void
+CowDiskImage::unserialize(Checkpoint *cp, const string &section)
+{
+    string cowFilename;
+    UNSERIALIZE_SCALAR(cowFilename);
+    cowFilename = cp->cptDir + "/" + cowFilename;
+    open(cowFilename);
+}
+
 BEGIN_DECLARE_SIM_OBJECT_PARAMS(CowDiskImage)
 
     SimObjectParam<DiskImage *> child;