yaoserversharing.h

Go to the documentation of this file.

#ifndef __YAOSERVERSHARING_H__
#define __YAOSERVERSHARING_H__

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

typedef struct {
    uint32_t gateid;
    UGATE_T* inval;
} input_gate_val_t;

typedef struct {
    uint32_t gateid;
    uint32_t pos;
} a2y_gate_pos_t;

//#define DEBUGYAOSERVER
class YaoServerSharing: public YaoSharing {

public:
    YaoServerSharing(e_role role, uint32_t sharebitlen, ABYCircuit* circuit, crypto* crypt) :
            YaoSharing(role, sharebitlen, circuit, crypt) {
        InitServer();
    }
    ;
    ~YaoServerSharing() {
        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 SendConversionValues(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 server";
    }
    ;
    //ENDS HERE..

private:
    //Global constant key
    CBitVector m_vR; 
    //Permutation bits for the servers input keys
    CBitVector m_vPermBits; 
    //Random values from output of ot extension
    vector<CBitVector> m_vROTMasks; 
    uint32_t m_nClientInputKexIdx; 
    uint32_t m_nClientInputKeyCtr; 
    CBitVector m_vServerKeySndBuf; 
    vector<CBitVector> m_vClientKeySndBuf; 
    CBitVector m_vClientROTRcvBuf; 
    //vector<CBitVector> m_vClientConversionKeySndBuf;
    //CBitVector            m_vClientCOnversionROTRcvBuf;

    CBitVector m_vOutputShareSndBuf; 
    CBitVector m_vOutputShareRcvBuf; 
    vector<GATE*> m_vServerOutputGates; 
    uint32_t m_nOutputShareRcvCtr; 
    uint64_t m_nPermBitCtr; 
    uint32_t m_nServerInBitCtr; 
    uint32_t m_nServerKeyCtr; 
    uint32_t m_nClientInBitCtr; 
    uint8_t* m_bLMaskBuf[2]; 
    uint8_t* m_bRMaskBuf[2]; 
    uint8_t* m_bLKeyBuf; 
    uint8_t* m_bOKeyBuf[2]; 
    uint8_t* m_bTmpBuf;
    //CBitVector

    vector<uint32_t> m_vClientInputGate; 
    deque<input_gate_val_t> m_vPreSetInputGates;
    deque<a2y_gate_pos_t> m_vPreSetA2YPositions;
    e_role* m_vOutputDestionations; 
    uint32_t m_nOutputDestionationsCtr;


    //deque<uint32_t>           m_vClientInputGate;

    void InitServer();
    void InitNewLayer();

    void CreateRandomWireKeys(CBitVector& vec, uint32_t numkeys);
    void CreateAndSendGarbledCircuit(ABYSetup* setup);
    void ReceiveGarbledCircuit();
    void EvaluateInputGate(uint32_t gateid);
    void EvaluateXORGate(GATE* gate);
    void EvaluateANDGate(GATE* gate);
    void EvaluateSIMDGate(uint32_t gateid);
    void EvaluateInversionGate(GATE* gate);
    void EvaluateConversionGate(uint32_t gateid);
    void CreateGarbledTable(GATE* ggate, uint32_t pos, GATE* gleft, GATE* gright);
    void PrecomputeGC(deque<uint32_t>& queue);

    //void EvaluateClientOutputGate(GATE* gate);
    void CollectClientOutputShares();
    void EvaluateOutputGate(GATE* gate);

    void SendServerInputKey(uint32_t gateid);
    void SendClientInputKey(uint32_t gateid);
    void AssignOutputShares();
};

#endif /* __YAOSERVERSHARING_H__ */