Update unit_tests for testing short messages and connection loss

storage-manager/src/unit_tests.cpp

@@ -189,9 +189,10 @@ void makeConnection()
     t.join();
 }
     
-bool opentask()
+bool opentask(bool connectionTest=false)
 {
     // going to rely on msgs being smaller than the buffer here
+	int err=0;
     uint8_t buf[1024];
     sm_msg_header *hdr = (sm_msg_header *) buf;
     open_cmd *cmd = (open_cmd *) &hdr[1];
@@ -206,6 +207,7 @@ bool opentask()
     cmd->flen = 19;
     strcpy((char *) cmd->filename, filename);
     
+    cout << "open file " << filename << endl;
     ::unlink(filename);
     ssize_t result = ::write(sessionSock, cmd, hdr->payloadLen);
     assert(result==(hdr->payloadLen));
@@ -213,8 +215,16 @@ bool opentask()
     OpenTask o(clientSock, hdr->payloadLen);
     o.run();
     
+    if (connectionTest)
+    {
+    	close(sessionSock);
+    	err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+    	assert(err == -1);
+    }
+    else
+    {
 		// read the response
-    int err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+		err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
 		sm_response *resp = (sm_response *) buf;
 		assert(err == sizeof(struct stat) + sizeof(sm_response));
 		assert(resp->header.type == SM_MSG_START);
@@ -227,7 +237,7 @@ bool opentask()
 		assert(_stat->st_uid == getuid());
 		assert(_stat->st_gid == getgid());
 		assert(_stat->st_size == 0);
-    
+    }
     /* verify the file is there */
     string metaPath = Config::get()->getValue("ObjectStorage", "metadata_path");
     assert(!metaPath.empty());
@@ -476,8 +486,9 @@ bool appendtask()
     return true;
 }
 
-void unlinktask()
+void unlinktask(bool connectionTest=false)
 {
+	int err=0;
     // make a meta file and delete it
     bf::path fullPath = homepath / prefix / "unlinktest1";
     string pathMeta = prefix + "/unlinktest1";
@@ -503,14 +514,23 @@ void unlinktask()
     assert(result==(sizeof(unlink_cmd) + cmd->flen));
     u.run();
     
-    // verify response
+    if (connectionTest)
-    int err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+    {
+    	close(sessionSock);
+    	err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+    	assert(err == -1);
+    }
+    else
+    {
+		// read the response
+		err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
 	    sm_response *resp = (sm_response *) buf;
 	    assert(err == sizeof(*resp));
 	    assert(resp->header.type == SM_MSG_START);
 	    assert(resp->header.payloadLen == sizeof(ssize_t));
 	    assert(resp->header.flags == 0);
 	    assert(resp->returnCode == 0);
+    }
     
     // confirm it no longer exists
     assert(!bf::exists(fullPathMeta));
@@ -547,8 +567,9 @@ void unlinktask()
     cout << "unlink task OK" << endl;
 }
 
-bool stattask()
+bool stattask(bool connectionTest=false)
 {
+	int err=0;
     bf::path fullPath = homepath / prefix / "stattest1";
     string filename = fullPath.string();
     string Metafilename = prefix + "/stattest1";
@@ -571,8 +592,16 @@ bool stattask()
     StatTask s(clientSock, sizeof(*cmd) + cmd->flen);
     s.run();
     
+    if (connectionTest)
+    {
+    	close(sessionSock);
+    	err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+    	assert(err == -1);
+    }
+    else
+    {
         // read the response
-    int err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+        err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
         sm_response *resp = (sm_response *) buf;
         assert(err == sizeof(struct stat) + sizeof(sm_response));
         assert(resp->header.type == SM_MSG_START);
@@ -585,6 +614,7 @@ bool stattask()
         assert(_stat->st_uid == getuid());
         assert(_stat->st_gid == getgid());
         assert(_stat->st_size == 8192);
+    }
 
     ::unlink(fullFilename.c_str());
     cout << "stattask OK" << endl;
@@ -732,7 +762,7 @@ bool IOCTruncate()
 }
 
 
-bool truncatetask()
+bool truncatetask(bool connectionTest=false)
 {
     IOCoordinator *ioc = IOCoordinator::get();
     Cache *cache = Cache::get();
@@ -742,7 +772,7 @@ bool truncatetask()
     string metaStr = prefix + "/trunctest1";
     const char *filename = fullPath.string().c_str();
     const char *Metafilename = metaStr.c_str();
-
+    int err=0;
     // get the metafile created
     string metaFullName = (metaPath/Metafilename).string() + ".meta";
     ::unlink(metaFullName.c_str());
@@ -762,14 +792,23 @@ bool truncatetask()
     TruncateTask t(clientSock, sizeof(*cmd) + cmd->flen);
     t.run();
     
+    if (connectionTest)
+    {
+    	close(sessionSock);
+    	err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+    	assert(err == -1);
+    }
+    else
+    {
         // read the response
-    int err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+        err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
         sm_response *resp = (sm_response *) buf;
         assert(err == sizeof(sm_response));
         assert(resp->header.type == SM_MSG_START);
         assert(resp->header.flags == 0);
         assert(resp->header.payloadLen == sizeof(ssize_t));
         assert(resp->returnCode == 0);
+    }
     
     // reload the metadata, check that it is 1000 bytes
     meta = MetadataFile(Metafilename);
@@ -781,7 +820,7 @@ bool truncatetask()
     return true;
 }
 
-bool listdirtask()
+bool listdirtask(bool connectionTest=false)
 {
     IOCoordinator *ioc = IOCoordinator::get();
     const bf::path metaPath = ioc->getMetadataPath();
@@ -821,6 +860,14 @@ bool listdirtask()
     ListDirectoryTask l(clientSock, sizeof(*cmd) + cmd->plen);
     l.run();
 
+    if (connectionTest)
+    {
+    	close(sessionSock);
+    	err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+    	assert(err == -1);
+    }
+    else
+    {
         /* going to keep this simple. Don't run this in a big dir. */
         /* maybe later I'll make a dir, put a file in it, and etc.  For now run it in a small dir. */
         err = ::recv(sessionSock, buf, 8192, MSG_DONTWAIT);
@@ -845,12 +892,15 @@ bool listdirtask()
             off += e->flen + sizeof(listdir_resp_entry);
         }
         assert(fileCounter == r->elements);
+    }
+
     bf::remove_all(tmpPath);
     return true;
 }
 
-void pingtask()
+void pingtask(bool connectionTest=false)
 {
+	int err=0;
     uint8_t buf[1024];
     ping_cmd *cmd = (ping_cmd *) buf;
     cmd->opcode = PING;
@@ -861,18 +911,28 @@ void pingtask()
     PingTask p(clientSock, sizeof(*cmd));
     p.run();
     
+    if (connectionTest)
+    {
+    	close(sessionSock);
+    	err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+    	assert(err == -1);
+    }
+    else
+    {
         // read the response
-    int err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+        err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
         sm_response *resp = (sm_response *) buf;
         assert(err == sizeof(sm_response));
         assert(resp->header.type == SM_MSG_START);
         assert(resp->header.payloadLen == sizeof(ssize_t));
         assert(resp->header.flags == 0);
         assert(resp->returnCode == 0);
+    }
+
     cout << "pingtask OK" << endl;
 }
 
-bool copytask()
+bool copytask(bool connectionTest=false)
 {
     /*  
         make a file
@@ -907,15 +967,25 @@ bool copytask()
 
     CopyTask c(clientSock, len);
     c.run();
-    
+    int err=0;
+    if (connectionTest)
+    {
+    	close(sessionSock);
+    	err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+    	assert(err == -1);
+    }
+    else
+    {
         // read the response
-    int err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+        err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
         sm_response *resp = (sm_response *) buf;
         assert(err == sizeof(sm_response));
         assert(resp->header.type == SM_MSG_START);
         assert(resp->header.payloadLen == sizeof(ssize_t));
         assert(resp->header.flags == 0);
         assert(resp->returnCode == 0);
+    }
+
     
     // verify copytest2 is there
     MetadataFile meta2(Metadest, MetadataFile::no_create_t(),true);
@@ -1585,54 +1655,79 @@ void bigMergeJournal1()
     
 // This should write an incomplete msg(s) to make sure SM does the right thing.  Not
 // done yet, handing this off to Ben.
-void shortWriteMsg()
+void shortMsg()
 {
-    // copy/modified/pasted from writetask().
+    IOCoordinator *ioc = IOCoordinator::get();
+
+    struct stat _stat;
     bf::path fullPath = homepath / prefix / "writetest1";
     const char *filename = fullPath.string().c_str();
     ::unlink(filename);
-    int fd = ::open(filename, O_CREAT | O_RDWR, 0666);
+    ioc->open(filename,O_WRONLY | O_CREAT,&_stat);
-    assert(fd > 0);
-    scoped_closer f(fd);
 
-    uint8_t buf[1024];
+    size_t size = 27;
-    sm_msg_header *hdr = (sm_msg_header *) buf;
+    uint8_t bufWrite[(sizeof(write_cmd)+std::strlen(filename)+size)];
-    write_cmd *cmd = (write_cmd *) &hdr[1];
-    uint8_t *data;
  
-    cmd->opcode = WRITE;
+    sm_msg_header *hdrWrite = (sm_msg_header *) bufWrite;
-    cmd->offset = 0;
+    write_cmd *cmdWrite = (write_cmd *) &hdrWrite[1];
-    cmd->count = 9;
+    uint8_t *dataWrite;
-    cmd->flen = 10;
-    memcpy(&cmd->filename, filename, cmd->flen);
-    data = (uint8_t *) &cmd->filename[cmd->flen];
-    memcpy(data, "123456789", cmd->count);
 
-    hdr->type = SM_MSG_START;
+    cmdWrite->opcode = WRITE;
-    hdr->payloadLen = sizeof(*cmd) + cmd->flen + cmd->count;
+    cmdWrite->offset = 0;
+    cmdWrite->count = size;
+    cmdWrite->flen = std::strlen(filename);
+    memcpy(&cmdWrite->filename, filename, cmdWrite->flen);
+    dataWrite = (uint8_t *) &cmdWrite->filename[cmdWrite->flen];
+    memcpy(dataWrite, "123456789123456789123456789", cmdWrite->count);
 
-    WriteTask w(clientSock, hdr->payloadLen);
+    hdrWrite->type = SM_MSG_START;
-    ssize_t result = ::write(sessionSock, cmd, hdr->payloadLen);
+    hdrWrite->payloadLen = sizeof(*cmdWrite) + cmdWrite->flen + 9;
-    assert(result==(hdr->payloadLen));
+
+    WriteTask w(clientSock, hdrWrite->payloadLen);
+    ssize_t result = ::write(sessionSock, cmdWrite, hdrWrite->payloadLen);
+    assert(result==(hdrWrite->payloadLen));
 
     w.run();
     
     // verify response
-    int err = ::recv(sessionSock, buf, 1024, MSG_DONTWAIT);
+    int err = ::recv(sessionSock, bufWrite, 1024, MSG_DONTWAIT);
-    sm_response *resp = (sm_response *) buf;
+    sm_response *resp = (sm_response *) bufWrite;
     assert(err == sizeof(*resp));
     assert(resp->header.type == SM_MSG_START);
     assert(resp->header.payloadLen == sizeof(ssize_t));
     assert(resp->header.flags == 0);
     assert(resp->returnCode == 9);
     
-    //check file contents
+
-    err = ::read(fd, buf, 1024);
+	uint8_t bufAppend[(sizeof(append_cmd)+std::strlen(filename)+size)];
-    assert(err == 9);
+    uint8_t *dataAppend;
-    buf[9] = 0;
+
-    assert(!strcmp("123456789", (const char *) buf));
+	sm_msg_header *hdrAppend = (sm_msg_header *) bufAppend;
-    ::unlink(filename);
+	append_cmd *cmdAppend = (append_cmd *) &hdrAppend[1];
-    cout << "write task OK" << endl;
+	cmdAppend->opcode = APPEND;
+	cmdAppend->count = size;
+	cmdAppend->flen = std::strlen(filename);
+	memcpy(&cmdAppend->filename, filename, cmdAppend->flen);
+	dataAppend = (uint8_t *) &cmdAppend->filename[cmdAppend->flen];
+    memcpy(dataAppend, "123456789123456789123456789", cmdAppend->count);
+    hdrAppend->type = SM_MSG_START;
+    hdrAppend->payloadLen = sizeof(*cmdAppend) + cmdAppend->flen + 9;
+
+	AppendTask a(clientSock, hdrAppend->payloadLen);
+	err = ::write(sessionSock, cmdAppend, hdrAppend->payloadLen);
+
+	a.run();
+
+	// verify response
+	err = ::recv(sessionSock, bufAppend, 1024, MSG_DONTWAIT);
+	resp = (sm_response *) bufAppend;
+	assert(err == sizeof(*resp));
+	assert(resp->header.type == SM_MSG_START);
+	assert(resp->header.payloadLen == sizeof(ssize_t));
+	assert(resp->header.flags == 0);
+	assert(resp->returnCode == 9);
+    ioc->unlink(fullPath.string().c_str());
+    cout << "shortWriteMsg Test OK" << endl;
 }
 
 // write and append are the biggest vulnerabilities here b/c those msgs could be sent in multiple
@@ -1642,7 +1737,7 @@ void shortWriteMsg()
 // tested as well.
 void shortMsgTests()
 {
-    shortWriteMsg();
+    shortMsg();
 }
 
 int main(int argc, char* argv[])
@@ -1732,7 +1827,7 @@ int main(int argc, char* argv[])
     IOCTruncate();
     IOCUnlink();
     IOCCopyFile();
-    //shortMsgTests();
+    shortMsgTests();
     
     // For the moment, this next one just verifies no error happens as reported by the fcns called.
     // It doesn't verify the result yet.
@@ -1747,12 +1842,43 @@ int main(int argc, char* argv[])
         cout << "To run the S3Storage unit tests, configure the S3 section of test-data/storagemanager.cnf" 
             << endl;
 
-    
+    cout << "Cleanup";
     metadataJournalTestCleanup();
+    cout << " DONE" << endl;
+
+    cout << "Testing connection loss..." << endl;
+
+    //opentask();
+    opentask(true);
+    cout << "connecting" << endl;
+    makeConnection();
+    cout << "connected" << endl;
+    unlinktask(true);
+    cout << "connecting" << endl;
+    makeConnection();
+    cout << "connected" << endl;
+    stattask(true);
+    cout << "connecting" << endl;
+    makeConnection();
+    cout << "connected" << endl;
+    truncatetask(true);
+    cout << "connecting" << endl;
+    makeConnection();
+    cout << "connected" << endl;
+    listdirtask(true);
+    cout << "connecting" << endl;
+    makeConnection();
+    cout << "connected" << endl;
+    pingtask(true);
+    cout << "connecting" << endl;
+    makeConnection();
+    cout << "connected" << endl;
+    copytask(true);
 
     (Cache::get())->shutdown();
-    delete (IOCoordinator::get());
+    delete (Synchronizer::get());
     delete (Cache::get());
+    delete (IOCoordinator::get());
 
     return 0;
 }