yaosharing.h

Go to the documentation of this file.

#ifndef __YAOSHARING_H__
#define __YAOSHARING_H__

#include "sharing.h"
#include "../util/yaokey.h"
#include "../circuit/booleancircuits.h"
#include "../util/constants.h"

#define FIXED_KEY_GARBLING
//#define MAXSHAREBUFSIZE 1000000

//TODO the garbled table is addressed as uint32_t and might overflow if sufficient (>4mio AND gates) are required. Change to LONG


#define KEYS_PER_GATE_IN_TABLE 2

class YaoSharing: public Sharing {

public:

    YaoSharing(e_role role, uint32_t sharebitlen, ABYCircuit* circuit, crypto* crypt) :
            Sharing(role, sharebitlen, circuit, crypt) {
        Init();
    }
    ;
    ~YaoSharing() {
    }
    ;

    // METHODS FROM SUPER CLASS SHARING...
    virtual void Reset() = 0;
    virtual void PrepareSetupPhase(ABYSetup* setup) = 0;
    virtual void PerformSetupPhase(ABYSetup* setup) = 0;
    virtual void FinishSetupPhase(ABYSetup* setup) = 0;
    virtual void EvaluateLocalOperations(uint32_t gateid) = 0;

    virtual void FinishCircuitLayer() = 0;

    virtual void PrepareOnlinePhase() = 0;

    virtual void InstantiateGate(GATE* gate) = 0;
    virtual void UsedGate(uint32_t gateid) = 0;

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

    virtual uint32_t AssignInput(CBitVector& input) = 0;
    virtual uint32_t GetOutput(CBitVector& out) = 0;

    uint32_t GetMaxCommunicationRounds() {
        return m_cBoolCircuit->GetMaxDepth();
    }
    ;
    uint32_t GetNumNonLinearOperations() {
        return m_nANDGates;
    }
    ;

    Circuit* GetCircuitBuildRoutine() {
        return m_cBoolCircuit;
    }
    ;

    void PrintPerformanceStatistics();
    //SUPER CLASS METHODS END HERE...

    void EvaluateSIMDGate(uint32_t gateid);

protected:
    /* A variable that points to inline functions for key xor */
    YaoKey *m_pKeyOps; 
    uint32_t m_nANDGates; 
    uint32_t m_nXORGates; 
    XORMasking *fMaskFct; 
    vector<GATE*> m_vANDGates; 
    vector<GATE*> m_vOutputShareGates; 
    uint32_t m_nInputShareSndSize; 
    uint32_t m_nOutputShareSndSize; 
    uint32_t m_nInputShareRcvSize; 
    uint32_t m_nOutputShareRcvSize; 
    uint32_t m_nClientInputBits; 
    CBitVector m_vClientInputKeys; 
    uint32_t m_nConversionInputBits; 
    //CBitVector            m_vConversionInputKeys;

    uint32_t m_nServerInputBits; 
    CBitVector m_vServerInputKeys; 
    CBitVector m_vGarbledCircuit; 
    uint64_t m_nGarbledTableCtr; 
    BYTE* m_bZeroBuf; 
    BYTE* m_bTempKeyBuf; 
    BooleanCircuit* m_cBoolCircuit; 
    uint32_t m_nSecParamIters; 
    uint64_t m_nANDWindowCtr; 
    uint64_t m_nRemANDGates; 
#ifdef FIXED_KEY_GARBLING
    BYTE* m_bResKeyBuf; 
    AES_KEY_CTX* m_kGarble; 
#endif

    void Init();

    BOOL EncryptWire(BYTE* c, BYTE* p, uint32_t id);

    void PrintKey(BYTE* key);
};

#endif /* __YAOSHARING_H__ */