134 lines
5.0 KiB
C++
134 lines
5.0 KiB
C++
//
|
|
// Copyright 2020 Electronic Arts Inc.
|
|
//
|
|
// TiberianDawn.DLL and RedAlert.dll and corresponding source code is free
|
|
// software: you can redistribute it and/or modify it under the terms of
|
|
// the GNU General Public License as published by the Free Software Foundation,
|
|
// either version 3 of the License, or (at your option) any later version.
|
|
|
|
// TiberianDawn.DLL and RedAlert.dll and corresponding source code is distributed
|
|
// in the hope that it will be useful, but with permitted additional restrictions
|
|
// under Section 7 of the GPL. See the GNU General Public License in LICENSE.TXT
|
|
// distributed with this program. You should have received a copy of the
|
|
// GNU General Public License along with permitted additional restrictions
|
|
// with this program. If not, see https://github.com/electronicarts/CnC_Remastered_Collection
|
|
|
|
/* $Header: /CounterStrike/PKPIPE.H 1 3/03/97 10:25a Joe_bostic $ */
|
|
/***********************************************************************************************
|
|
*** C O N F I D E N T I A L --- W E S T W O O D S T U D I O S ***
|
|
***********************************************************************************************
|
|
* *
|
|
* Project Name : Command & Conquer *
|
|
* *
|
|
* File Name : PKPIPE.H *
|
|
* *
|
|
* Programmer : Joe L. Bostic *
|
|
* *
|
|
* Start Date : 07/06/96 *
|
|
* *
|
|
* Last Update : July 6, 1996 [JLB] *
|
|
* *
|
|
*---------------------------------------------------------------------------------------------*
|
|
* Functions: *
|
|
* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
|
|
|
|
#ifndef PKPIPE_H
|
|
#define PKPIPE_H
|
|
|
|
#include "pipe.h"
|
|
#include "pk.h"
|
|
#include "rndstraw.h"
|
|
#include "blowpipe.h"
|
|
|
|
|
|
/*
|
|
** This pipe will encrypt/decrypt the data stream. The data is encrypted by generating a
|
|
** symetric key that is then encrypted using the public key system. This symetric key is then
|
|
** used to encrypt the remaining data.
|
|
*/
|
|
class PKPipe : public Pipe
|
|
{
|
|
public:
|
|
typedef enum CryptControl {
|
|
ENCRYPT,
|
|
DECRYPT
|
|
} CryptControl;
|
|
|
|
PKPipe(CryptControl control, RandomStraw & rnd);
|
|
|
|
virtual void Put_To(Pipe * pipe);
|
|
virtual void Put_To(Pipe & pipe) {Put_To(&pipe);}
|
|
|
|
// Feed data through for processing.
|
|
virtual int Put(void const * source, int length);
|
|
|
|
// Submit key for encryption/decryption.
|
|
void Key(PKey const * key);
|
|
|
|
private:
|
|
enum {
|
|
BLOWFISH_KEY_SIZE=BlowfishEngine::MAX_KEY_LENGTH,
|
|
MAX_KEY_BLOCK_SIZE=256 // Maximum size of pk encrypted blowfish key.
|
|
};
|
|
|
|
/*
|
|
** This flag indicates whether the PK (fetch blowfish key) phase is
|
|
** in progress or not.
|
|
*/
|
|
bool IsGettingKey;
|
|
|
|
/*
|
|
** This is the random straw that is needed to generate the
|
|
** blowfish key.
|
|
*/
|
|
RandomStraw & Rand;
|
|
|
|
/*
|
|
** This is the attached blowfish pipe. After the blowfish key has been
|
|
** decrypted, then the PK processor goes dormant and the blowfish processor
|
|
** takes over the data flow.
|
|
*/
|
|
BlowPipe BF;
|
|
|
|
/*
|
|
** Controls the method of processing the data stream.
|
|
*/
|
|
CryptControl Control;
|
|
|
|
/*
|
|
** Pointer to the key to use for encryption/decryption. The actual process
|
|
** performed is controlled by the Control member. A key can be used for
|
|
** either encryption or decryption -- it makes no difference. However, whichever
|
|
** process is performed, the opposite process must be performed using the
|
|
** other key.
|
|
*/
|
|
PKey const * CipherKey;
|
|
|
|
/*
|
|
** This is the staging buffer for the block of data. This block must be as large as
|
|
** the largest possible key size or the largest blowfish key (whichever is greater).
|
|
*/
|
|
char Buffer[MAX_KEY_BLOCK_SIZE];
|
|
|
|
/*
|
|
** The working counter that holds the number of bytes in the staging buffer.
|
|
*/
|
|
int Counter;
|
|
|
|
/*
|
|
** This records the number of bytes remaining in the current block. This
|
|
** will be the number of bytes left to accumulate before the block can be
|
|
** processed either for encryption or decryption.
|
|
*/
|
|
int BytesLeft;
|
|
|
|
int Encrypted_Key_Length(void) const;
|
|
int Plain_Key_Length(void) const;
|
|
|
|
PKPipe(PKPipe & rvalue);
|
|
PKPipe & operator = (PKPipe const & pipe);
|
|
};
|
|
|
|
|
|
#endif
|