yaoclientsharing.h

Go to the documentation of this file.

#ifndef __YAOCLIENTSHARING_H__
#define __YAOCLIENTSHARING_H__

#include "sharing.h"
#include <algorithm>
#include "yaosharing.h"

//#define DEBUGYAOCLIENT

class YaoClientSharing: public YaoSharing {

public:
    YaoClientSharing(e_role role, uint32_t sharebitlen, ABYCircuit* circuit, crypto* crypt) :
            YaoSharing(role, sharebitlen, circuit, crypt) {
        InitClient();
    }
    ;
    ~YaoClientSharing() {
        Reset();
    }
    ;

    //MEMBER FUNCTIONS FROM SUPER CLASS YAO SHARING
    void Reset();
    void PrepareSetupPhase(ABYSetup* setup);
    void PerformSetupPhase(ABYSetup* setup);
    void FinishSetupPhase(ABYSetup* setup);
    void EvaluateLocalOperations(uint32_t gateid);
    void EvaluateInteractiveOperations(uint32_t gateid);
    void EvaluateConversionGate(uint32_t gateid);

    void FinishCircuitLayer();

    void PrepareOnlinePhase();

    void InstantiateGate(GATE* gate);
    void UsedGate(uint32_t gateid);

    void GetDataToSend(vector<BYTE*>& sendbuf, vector<uint32_t>& bytesize);
    void GetBuffersToReceive(vector<BYTE*>& rcvbuf, vector<uint32_t>& rcvbytes);

    uint32_t AssignInput(CBitVector& input);
    uint32_t GetOutput(CBitVector& out);

    const char* sharing_type() {
        return "Yao client";
    }
    ;
    //ENDS HERE..

private:

    CBitVector m_vROTMasks; 
    uint32_t m_nChoiceBitCtr; 
    CBitVector m_vChoiceBits; 
    uint32_t m_nKeyInputRcvIdx; 
    uint32_t m_nServerInBitCtr; 
    uint32_t m_nClientSndOTCtr; 
    uint32_t m_nClientRcvKeyCtr; 
    uint32_t m_nClientOutputShareCtr;
    uint32_t m_nServerOutputShareCtr;
    uint32_t m_nClientOUTBitCtr; 
    CBitVector m_vServerKeyRcvBuf; 
    vector<CBitVector> m_vClientKeyRcvBuf; 
    uint32_t m_nGarbledCircuitRcvCtr;
    CBitVector m_vOutputShareRcvBuf;
    CBitVector m_vOutputShareSndBuf;
    vector<uint32_t> m_vClientSendCorrectionGates; 
    vector<uint32_t> m_vClientRcvInputKeyGates; 
    vector<uint32_t> m_vServerInputGates; 
    CBitVector m_vROTSndBuf;
    uint32_t m_vROTCtr;
    uint8_t**           m_vTmpEncBuf;

    void ReceiveServerKeys(uint32_t gateid);
    void ReceiveClientKeys(uint32_t gateid);

    void AssignServerInputKeys();
    void AssignClientInputKeys();

    void EvaluateXORGate(GATE* gate);
    void EvaluateANDGate(GATE* gate);
    BOOL EvaluateGarbledTable(GATE* gate, uint32_t pos, GATE* gleft, GATE* gright);
    void EvaluateServerOutputGate(GATE* gate);
    void EvaluateSIMDGate(uint32_t gateid);

    void InitClient();
    void InitNewLayer();

    void EvaluateClientOutputGate(uint32_t gateid);

    void ReceiveGarbledCircuitAndOutputShares(ABYSetup* setup);
};

#endif /* __YAOCLIENTSHARING_H__ */