debian/libwvstreams-dev/usr/include/wvstreams/wvblowfish.h

/* -*- Mode: C++ -*-
 * Worldvisions Tunnel Vision Software:
 *   Copyright (C) 1997-2002 Net Integration Technologies, Inc.
 *
 * Blowfish cryptography abstractions.
 */
#ifndef __WVBLOWFISH_H
#define __WVBLOWFISH_H
 
#include "wvencoder.h"
#include "wvencoderstream.h"
 
struct bf_key_st;

class WvBlowfishEncoder : public WvEncoder
{
public:
    enum Mode {
        ECBEncrypt, 
        ECBDecrypt, 
        CFBEncrypt, 
        CFBDecrypt  
    };

    WvBlowfishEncoder(Mode mode, const void *key, size_t keysize);
    virtual ~WvBlowfishEncoder();

    void setkey(const void *key, size_t keysize);
    
    void setiv(const void *iv);

    bool is_encrypting() const {
        return (mode == ECBEncrypt || mode == CFBEncrypt);
    }
    
protected:
    virtual bool _encode(WvBuf &in, WvBuf &out, bool flush);
    virtual bool _reset(); // supported: restores most recently set
        // key and initialization vector
 
    Mode mode;
    size_t keysize;
    unsigned char *key;
    struct bf_key_st *bfkey;
    unsigned char ivec[8]; // initialization vector
    int ivecoff; // current offset into initvec
 
    void preparekey();
};
 

class WvBlowfishStream : public WvEncoderStream
{
public:
    WvBlowfishStream(WvStream *_cloned,
        const void *key, size_t _keysize,
        WvBlowfishEncoder::Mode readmode = WvBlowfishEncoder::CFBDecrypt,
        WvBlowfishEncoder::Mode writemode = WvBlowfishEncoder::CFBEncrypt);
    virtual ~WvBlowfishStream() { }
};
 
#endif // __WVBLOWFISH_H