/* Web Polygraph       http://www.web-polygraph.org/
 * (C) 2003-2006 The Measurement Factory
 * Licensed under the Apache License, Version 2.0 */

#include "pgl/pgl.h"

#include "xstd/String.h"
#include "xstd/TblDistr.h"
#include "pgl/PglRec.h"
#include "pgl/PglNumSym.h"
#include "pgl/PglStringSym.h"
#include "pgl/PglArraySym.h"
#include "pgl/PglSizeSym.h"
#include "pgl/PglIntSym.h"
#include "pgl/SslWrapSym.h"



String SslWrapSym::TheType = "SslWrap";

static String strStringArr = "string[]";
static String strProtocols = "protocols";
static String strRootCertificate = "root_certificate";
static String strSizeArr = "size[]";
static String strRsaKeySizes = "rsa_key_sizes";
static String strCiphers = "ciphers";
static String strSslCiphers = "ssl-ciphers";	// a global name for the distr
static String strSessionResump = "session_resumption";
static String strSessionCache = "session_cache";


SslWrapSym::SslWrapSym(): RecSym(TheType, new PglRec) {
	theRec->bAdd(strStringArr, strProtocols, 0);
	theRec->bAdd(StringSym::TheType, strRootCertificate, 0);
	theRec->bAdd(strSizeArr, strRsaKeySizes, 0);
	theRec->bAdd(strStringArr, strCiphers, 0);
	theRec->bAdd(NumSym::TheType, strSessionResump, 0);
	theRec->bAdd(IntSym::TheType, strSessionCache, 0);
}

SslWrapSym::SslWrapSym(const String &aType, PglRec *aRec): RecSym(aType, aRec) {
}

bool SslWrapSym::isA(const String &type) const {
	return RecSym::isA(type) || type == TheType;
}

SynSym *SslWrapSym::dupe(const String &type) const {
	if (isA(type))
		return new SslWrapSym(this->type(), theRec->clone());
	return RecSym::dupe(type);
}

String SslWrapSym::rootCertificate() const {
        return getString(strRootCertificate);
}

RndDistr *SslWrapSym::protocols(const StrIdentifier &protocolNames) const {
	return namesToDistr(strProtocols, protocolNames);
}

bool SslWrapSym::rsaKeySizes(Array<Size> &sizes, RndDistr *&sel) const {
	SynSymTblItem *wi = 0;
	Assert(theRec->find(strRsaKeySizes, wi));
	if (!wi->sym())
		return false; // undefined

	ArraySym &a = (ArraySym&)wi->sym()->cast(ArraySym::TheType);
	Array<SizeSym*> syms;
	ArraySymExportM(SizeSym, a, SizeSym::TheType, syms);

	sizes.stretch(syms.count());
	for (int i = 0; i < syms.count(); ++i) {
		const Size len = Size::Byte(syms[i]->val().byte());
		const Size minLen = Size::Bit(384);
		if (len < minLen) {
			cerr << loc() << "RSA key size must be at least " << minLen <<
				"; got: " << len << endl;
			exit(1);
		}
		sizes.append(len);
	}

	Array<double> probs;
	a.copyProbs(probs);
	sel = TblDistr::FromDistrTable(type() + "-" + strRsaKeySizes, probs);

	return true;
}

bool SslWrapSym::ciphers(Array<String*> &ciphers, RndDistr *&selector) const {
	if (ArraySym *as = getArraySym(strCiphers)) {
		selector = as->makeSelector(strSslCiphers);
		Assert(getStrings(strCiphers, ciphers));
		return true;
	}
	return false;
}

bool SslWrapSym::sessionResumpt(double &prob) const {
	return getDouble(strSessionResump, prob);
}

bool SslWrapSym::sessionCacheSize(int &count) const {
	return getInt(strSessionCache, count);
}