hub.h

Go to the documentation of this file.

/*
SeComLib
Copyright 2012-2013 TU Delft, Information Security & Privacy Lab (http://isplab.tudelft.nl/)

Contributors:
Inald Lagendijk (R.L.Lagendijk@TUDelft.nl)
Mihai Todor (todormihai@gmail.com)
Thijs Veugen (P.J.M.Veugen@tudelft.nl)
Zekeriya Erkin (z.erkin@tudelft.nl)

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
#ifndef HUB_HEADER_GUARD
#define HUB_HEADER_GUARD

//include our headers
#include "utils/config.h"
#include "utils/filesystem.h"
#include "utils/math.h"
#include "utils/date_time.h"
#include "utils/cpu_timer.h"
#include "core/big_integer.h"
#include "core/random_provider.h"
#include "core/paillier.h"

#include "server.h"
#include "test_data_row.h"

//include C++ libraries
#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
#include <numeric>
#include <vector>
#include <map>
#include <stdexcept>

namespace SeComLib {
using namespace Core;

namespace SecureRecommendations {
    //forward-declare required classes
    class Server;

    class Hub {
    public:
        Hub ();

        ~Hub () {}

        const PaillierPublicKey &GetPublicKey () const;

        void SetServer (const std::shared_ptr<const Server> &server);

        void EvaluateSign (SecureSvm::EncryptedVector &input) const;

        void EvaluateMaximum (SecureSvm::EncryptedVector &input) const;

        void EvaluateDivision (const BigInteger &numerator, SecureSvm::EncryptedVector &input) const;

        void DoAccuracyAnalysis ();

        void DoPerformanceAnalysis ();

        void DoTrafficAnalysis ();

        void DebugValue (const Paillier::Ciphertext &value) const;
    private:
        Paillier cryptoProvider;

        std::shared_ptr<const Server> server;

        unsigned long testVectorCount;

        unsigned int attributeCount;

        SecureSvm::KernelTypes kernel;

        BigInteger featureScalingFactor;

        BigInteger svWeightScaling;

        std::vector<TestDataRow> medicalRelevanceTestData;

        std::vector<std::map<std::string, TestDataRow>> safetyTestData;

        std::vector<unsigned short> predictedClusters;

        std::vector<std::vector<unsigned short>> predictedQualityOfLifeMatrix;

        std::map<std::string, unsigned int> safetyBlockFalsePositives;

        std::map<std::string, unsigned int> safetyBlockFalseNegatives;

        static bool measureTraffic;

        static BigInteger bitsSent;

        static BigInteger bitsReceived;

        void loadSafetyTestData (const std::string &testFilesDirectory, const std::string &safetyTestFilesPrefix);

        TestDataRow parseTestDataRow (const std::string &line) const;

        unsigned short getVotedCluster (const std::vector<Paillier::Ciphertext> &encryptedClusterVotes) const;

        std::vector<unsigned short> processSafetyPredictions (const std::vector<Paillier::Ciphertext> &encryptedSafetyPredictions) const;

        unsigned int getMedicalRelevanceBlockCorrectPredictionCount () const;

        void computeSafetyBlockFalsePredictions ();

        static void resetTrafficCounters ();

        std::string getKernelName () const;

        Hub (Hub const &);

        Hub operator= (Hub const &);
    };
}//namespace SecureRecommendations
}//namespace SeComLib

#endif//HUB_HEADER_GUARD