#include <u.h>
#include <libc.h>
#include <mp.h>
#include <fcall.h>
#include <thread.h>
#include <9p.h>
#include <libsec.h>
#include "sshtun.h"

struct CipherState
{
	BFstate state;
};

static CipherState*
initblowfish(Conn *c, int dir)
{
	CipherState *cs;

int i;
fprint(2, "initblowfish dir:%d\ns2cek: ", dir);
for(i = 0; i < 16; ++i) fprint(2, "%02x", c->s2cek[i]);
fprint(2, "\nc2sek: ");
for(i = 0; i < 16; ++i) fprint(2, "%02x", c->c2sek[i]);
fprint(2, "\ns2civ: ");
for(i = 0; i < 8; ++i) fprint(2, "%02x", c->s2civ[i]);
fprint(2, "\nc2siv: ");
for(i = 0; i < 8; ++i) fprint(2, "%02x", c->c2siv[i]);
fprint(2, "\n");

	cs = emalloc9p(sizeof(CipherState));
memset(cs, '\0', sizeof(CipherState));
fprint(2,"cs:%p\n", cs);
	if(dir)
		setupBFstate(&cs->state, c->s2cek, 16, c->s2civ);
	else
		setupBFstate(&cs->state, c->c2sek, 16, c->c2siv);
	return cs;
}

static void
encryptblowfish(CipherState *cs, uchar *buf, int nbuf)
{
	bfCBCencrypt(buf, nbuf, &cs->state);
}

static void
decryptblowfish(CipherState *cs, uchar *buf, int nbuf)
{
fprint(2,"cs:%p, nb:%d\n", cs, nbuf);
fprint(2, "before decrypt: %02ux %02ux %02ux %02ux\n", buf[0], buf[1], buf[2], buf[3]);
	bfCBCdecrypt(buf, nbuf, &cs->state);
fprint(2, "after decrypt: %02ux %02ux %02ux %02ux\n", buf[0], buf[1], buf[2], buf[3]);
}

Cipher cipherblowfish = 
{
	"blowfish-cbc",
	8,
	initblowfish,
	encryptblowfish,
	decryptblowfish,
};