FunctionIdDatabaseInterface.h

#ifndef FUNCTION_ID_DATABASE_INTERFACE_H
#define FUNCTION_ID_DATABASE_INTERFACE_H
#include <featureTests.h>
#ifdef ROSE_ENABLE_BINARY_ANALYSIS

#include "sqlite3x.h"
#include "LibraryInfo.h"
#include "FunctionInfo.h"
#include <vector>
#include <ctype.h>

namespace LibraryIdentification
{
    enum DUPLICATE_OPTION 
    {
        UNKNOWN,
        COMBINE,
        REPLACE,
        NO_ADD
    };
    
    enum DUPLICATE_OPTION duplicateOptionFromString(std::string option);


    class FunctionIdDatabaseInterface 
    {
    public:
        FunctionIdDatabaseInterface(const std::string& dbName);
        

        void createTables();
 

         // @brief Add an entry for a function to the database
         void addFunctionToDB(const FunctionInfo& fInfo, enum DUPLICATE_OPTION dupOption = COMBINE);

         bool matchOneFunction(FunctionInfo& fInfo);
         
         std::vector<FunctionInfo> matchFunction(const FunctionInfo& fInfo);


         bool exactMatchFunction(const FunctionInfo& fInfo);
         

         void removeFunctions(const std::string& funcHash);
         

         // @brief Add an entry for a library to the database
         void addLibraryToDB(const LibraryInfo& lInfo, bool replace = false);

         bool matchLibrary(LibraryInfo& fInfo);

 private:
     // @brief The name of the database
     std::string database_name;
     
     // @brief SQLite database handle
     sqlite3x::sqlite3_connection sqConnection;

    };   
}

#endif
#endif