137 lines
4.9 KiB
C++
137 lines
4.9 KiB
C++
/*
|
|
** Command & Conquer Red Alert(tm)
|
|
** Copyright 2025 Electronic Arts Inc.
|
|
**
|
|
** This program 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.
|
|
**
|
|
** This program is distributed in the hope that it will be useful,
|
|
** but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
** GNU General Public License for more details.
|
|
**
|
|
** You should have received a copy of the GNU General Public License
|
|
** along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
/* $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
|