client.h

#ifndef CLIENT_H
#define CLIENT_H

#include <iostream>
#include <string>
#include <afxwin.h>

#include "polarssl/aes.h"
#include "polarssl/entropy.h"
#include "polarssl/ctr_drbg.h"

#include "socket.h"
#include "struct.h"
//#include "minunit.h"

struct adress{
	std::string IP;

};

class Client{
public:
	SocketClient* activePartnerSocket;
	SocketClient* activeServerSocket;
	std::string partnerName;
	int port;
	bool incomingConnection;
	unsigned char key[16];
	unsigned char encBuffer[1000];
	unsigned char decBuffer[1000];
	int getPointerEnc;
	int putPointerEnc;
	int counterEnc;
	int getPointerDec;
	int putPointerDec;
	int counterDec;
	bool stop;
private:
	unsigned char publicKey[128];
	unsigned char privateKey[128];
	personInfo PI;
	std::string login;
	cert myCert;
	cert partnerCert;
	adress myAdress;
	adress partnerAdress;
public:
	Client(_In_ string login);
	/**
	* Encrypts/decrypts given data with AES - 128.
	*
	* @param key 128 bit long key
	* @param iv initialisation vector
	* @param data input data
	* @param outData output data
	* @param mode 0 - encryption ; 1 - decryption
	*
	* @return returns zero when succesful, nonzero value otherwise
	*/
	_Check_return_ int cryptoSym(std::string, unsigned char iv[16], std::string data, std::string& outData, int mode);

	/**
	* Encrypts/decrypts given data with RSA - 1024.
	*
	* @param key 1024 bit long key
	* @param data input data
	* @param outData output data
	* @param mode 0 - encryption ; 1 - decryption
	*
	* @return returns zero when succesful, nonzero value otherwise
	*/
	_Check_return_ int cryptoAsym(unsigned char key[128] , unsigned char* data , unsigned char* outData , int mode);

	/**
	* Generates random pair of RSA keys.
	*
	* @param publicKey generated public key
	* @param privateKey generated private key
	*
	* @return returns zero when succesful, nonzero value otherwise
	*/
	_Check_return_ int randGenRSA(unsigned char* publicKey , unsigned char* privateKey);

	/**
	* Requests login to server.
	*
	* @param requestType type of request
	* @param login user's login
	* @param password user's password
	*
	* @return returns zero when succesful, nonzero value otherwise
	*/
	_Check_return_ int loginRequest(_In_ std::string password);

	/**
	* Requests logout from server.
	*
	* @param requestType type of request
	* @param login user's login
	*
	* @return returns zero when succesful, nonzero value otherwise
	*/
	_Check_return_ int logoutRequest(requestType rT , unsigned char* login);

	/**
	* Requests registration on server.
	*
	* @param requestType type of request
	* @param myCert user's certificate
	*
	* @return returns zero when succesful, nonzero value otherwise
	*/
	_Check_return_ int registrationRequest();

	/**
	* Requests partner's adress from server, based on partner's login.
	*
	* @param requestType type of request
	* @param partnerLogin partner's login
	*
	* @return returns zero when succesful, nonzero value otherwise
	*/
	_Check_return_ int adressRequest(requestType rt , unsigned char* partnerLogin);

	/**
	* Request signed certificate from CA.
	*
	* @param requestType type of request
	* @param PI user's personal info
	*
	* @return returns zero when succesful, nonzero value otherwise
	*/
	_Check_return_ int certificateRequest(requestType rt , personInfo PI);

	/*
	* Metody pre obsluhu prikazov uzivatela(Lubo)
	*/
private:
	_Check_return_ int commandParse(_Out_opt_ std::string& , _In_ std::string);
	_Check_return_ int listRequest();
	_Check_return_ int communicationRequest(_In_ std::string);
	_Check_return_ int logoutRequest();
	_Check_return_ int connectToPartner(_In_ std::string value);
	_Check_return_ int acceptComm();
	_Check_return_ int declineComm();
	_Check_return_ int endComm();
	_Check_return_ int disconnect();
	_Check_return_ int quit();
	void help();
public:
	_Check_return_ bool command(_In_ std::string);
	_Check_return_ int sendMessage(_In_ std::string);

	/**
	* XOR given plaintext to pre-generated encryption buffer
	* 
	* @param text plaintext to be enciphered
	*
	* @return ciphertext
	*/
	_Check_return_ _Ret_ std::string encipher(_In_ std::string text);

	/**
	* XOR given ciphertext to pre-generated decryption buffer
	*
	* @param text ciphertext to be decrypted
	*
	* @return plaintext
	*/
	_Check_return_ _Ret_ std::string decipher(_In_ std::string text);
};


#endif //CLIENT