commit 2a6fa2baf022a63527e2e5bc8aaec85793fe3693
author Alexander Afanasyev <alexander.afanasyev@ucla.edu> Wed Jan 23 16:51:55 2013 -0800
committer Alexander Afanasyev <alexander.afanasyev@ucla.edu> Wed Jan 23 16:51:55 2013 -0800
tree 05a7883df68822a2e81898b8eedfc6604af5cb5a
parent f1122c3458bf6243dea26e54f0b6938aa1284eec

Solve small bug in object manager

src/object-manager.cc

diff --git a/src/object-manager.cc b/src/object-manager.cc
index f822120..848929e 100644
--- a/src/object-manager.cc
+++ b/src/object-manager.cc
@@ -61,15 +61,21 @@
 
   fs::ifstream iff (file, std::ios::in | std::ios::binary);
   sqlite3_int64 segment = 0;
-  while (iff.good ())
+  while (iff.good () && !iff.eof ())
     {
       char buf[MAX_FILE_SEGMENT_SIZE];
       iff.read (buf, MAX_FILE_SEGMENT_SIZE);
+      if (iff.gcount () == 0)
+        {
+          // stupid streams...
+          break;
+        }
 
       Name name = Name (deviceName)("file")(fileHash->GetHash (), fileHash->GetHashBytes ())(segment);
 
       // cout << *fileHash << endl;
       // cout << name << endl;
+      _LOG_DEBUG ("Read " << iff.gcount () << " from " << file << " for segment " << segment);
 
       Bytes data = m_ccnx->createContentObject (name, buf, iff.gcount ());
       fileDb.saveContentObject (deviceName, segment, data);

test/test-serve-and-fetch.cc

diff --git a/test/test-serve-and-fetch.cc b/test/test-serve-and-fetch.cc
index 0c53fa0..6c9e877 100644
--- a/test/test-serve-and-fetch.cc
+++ b/test/test-serve-and-fetch.cc
@@ -40,7 +40,7 @@
 using namespace boost;
 using namespace boost::filesystem;
 
-BOOST_AUTO_TEST_SUITE(TestFetchManager)
+BOOST_AUTO_TEST_SUITE(TestServeAndFetch)
 
 path root("test-server-and-fetch");
 path filePath = root / "random-file";