main_restore.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "OptionParser.h"
 
#include "phoenix_mock.h"


OptionParser createOptionParser(){
        OptionParser parser(true, __PROGRAM_VERSION__);
        parser.setExampleLongOption("phoenix_mock_restore --input=file.pmock");
        parser.setExampleShortOption("phoenix_mock_restore -i file2.mock file2.pmock");
        
        parser.addOption("input", "i", OptionType::FILENAME, true, "List of input mock to get the info");
        return parser;
}


bool restoreMockFile(const PPath & inputFile){
        FILE* fp = fopen(inputFile.c_str(), "r+");
        if(fp == NULL){
                std::cerr << "restoreMockFile : cannot open file '"<<inputFile<<"'" << std::endl;
                return false;
        }
        size_t nbMessage(0lu);
        bool isReadFileOk = fseek(fp, sizeof(size_t), SEEK_SET) == 0;   //We can skip the possible broken number of messages in the mock
        while(isReadFileOk){
                size_t nbByteInMessage(0lu);
                isReadFileOk &= data_load(fp, nbByteInMessage);
                isReadFileOk &= fseek(fp, sizeof(size_t) + nbByteInMessage, SEEK_CUR) == 0;     //let's jump until the next message
                nbMessage += isReadFileOk;
        }
        fseek(fp, 0lu, SEEK_SET);
        bool b = data_save(fp, nbMessage);
        fseek(fp, 0l, SEEK_END);
        fclose(fp);
        return b;
}


bool restoreAllMock(const std::vector<PPath> & vecInputFile){
        bool b(true);
        for(std::vector<PPath>::const_iterator itFile(vecInputFile.begin()); itFile != vecInputFile.end(); ++itFile){
                b &= restoreMockFile(*itFile);
        }
        return b;
}
 
int main(int argc, char** argv){
        OptionParser parser = createOptionParser();
        parser.parseArgument(argc, argv);
        const OptionMode & defaultMode = parser.getDefaultMode();
        std::vector<PPath> vecInputFile;
        defaultMode.getValue(vecInputFile, "input");
        return restoreAllMock(vecInputFile) - 1;
}