/*
 *	PROGRAM:		Firebird authentication
 *	MODULE:			Auth.cpp
 *	DESCRIPTION:	Implementation of interfaces, passed to plugins
 *					Plugins loader
 *
 *  The contents of this file are subject to the Initial
 *  Developer's Public License Version 1.0 (the "License");
 *  you may not use this file except in compliance with the
 *  License. You may obtain a copy of the License at
 *  http://www.ibphoenix.com/main.nfs?a=ibphoenix&page=ibp_idpl.
 *
 *  Software distributed under the License is distributed AS IS,
 *  WITHOUT WARRANTY OF ANY KIND, either express or implied.
 *  See the License for the specific language governing rights
 *  and limitations under the License.
 *
 *  The Original Code was created by Alex Peshkov
 *  for the Firebird Open Source RDBMS project.
 *
 *  Copyright (c) 2010 Alex Peshkov <peshkoff at mail.ru>
 *  and all contributors signed below.
 *
 *  All Rights Reserved.
 *  Contributor(s): ______________________________________.
 */

#include "firebird.h"
#include "../auth/AuthDbg.h"
#include "../jrd/ibase.h"

#ifdef AUTH_DEBUG

//#define AUTH_VERBOSE

// register plugin
static Firebird::SimpleFactory<Auth::DebugClient> clientFactory;
static Firebird::SimpleFactory<Auth::DebugServer> serverFactory;

extern "C" void FB_PLUGIN_ENTRY_POINT(Firebird::IMaster* master)
{
	const char* name = "Auth_Debug";

	Firebird::PluginInterface iPlugin(master);

	iPlugin->registerPlugin(Firebird::PluginType::AuthClient, name, &clientFactory);
	iPlugin->registerPlugin(Firebird::PluginType::AuthServer, name, &serverFactory);
}


namespace Auth {

DebugServer::DebugServer(Firebird::IFactoryParameter*)
	: str(getPool())
{ }

Result FB_CARG DebugServer::startAuthentication(Firebird::Status* status, bool isService, const char* dbName,
										const unsigned char* dpb, unsigned int dpbSize,
										WriterInterface* writerInterface)
{
	try
	{
		str.erase();
		Firebird::ClumpletReader rdr(isService ?
			Firebird::ClumpletReader::spbList :
			Firebird::ClumpletReader::dpbList, dpb, dpbSize);

		if (rdr.find(isService ? isc_spb_trusted_auth : isc_dpb_trusted_auth))
		{
			str.assign(rdr.getBytes(), rdr.getClumpLength());
		}

		str += '_';
		return AUTH_MORE_DATA;
	}
	catch (const Firebird::Exception& ex)
	{
		ex.stuffException(status);
		return AUTH_FAILED;
	}
}

Result FB_CARG DebugServer::contAuthentication(Firebird::Status* status, WriterInterface* writerInterface,
									   const unsigned char* data, unsigned int size)
{
	try
	{
#ifdef AUTH_VERBOSE
		fprintf(stderr, "DebugServerInstance::contAuthentication: %.*s\n", size, data);
#endif
		writerInterface->add(Firebird::string((const char*) data, size).c_str(), "DEBUG", "");
		return AUTH_SUCCESS;
	}
	catch (const Firebird::Exception& ex)
	{
		ex.stuffException(status);
		return AUTH_FAILED;
	}
}

void FB_CARG DebugServer::getData(const unsigned char** data, unsigned short* dataSize)
{
	*data = reinterpret_cast<const unsigned char*>(str.c_str());
	*dataSize = str.length();
#ifdef AUTH_VERBOSE
	fprintf(stderr, "DebugServerInstance::getData: %.*s\n", *dataSize, *data);
#endif
}

int FB_CARG DebugServer::release()
{
	if (--refCounter == 0)
	{
		delete this;
		return 0;
	}

	return 1;
}

DebugClient::DebugClient(Firebird::IFactoryParameter*)
	: str(getPool())
{ }

Result FB_CARG DebugClient::startAuthentication(Firebird::Status* status, bool isService, const char*, DpbInterface* dpb)
{
	try
	{
		str = "HAND";
		if (dpb)
		{
			dpb->add((isService ? isc_spb_trusted_auth : isc_dpb_trusted_auth),
					 str.c_str(), str.length());
			return AUTH_SUCCESS;
		}
		return AUTH_MORE_DATA;
	}
	catch (const Firebird::Exception& ex)
	{
		ex.stuffException(status);
		return AUTH_FAILED;
	}
}

Result FB_CARG DebugClient::contAuthentication(Firebird::Status* status, const unsigned char* data, unsigned int size)
{
	try
	{
#ifdef AUTH_VERBOSE
		fprintf(stderr, "DebugClientInstance::contAuthentication: %.*s\n", size, data);
#endif
		str.assign(data, size);
		str += "SHAKE";
		return AUTH_CONTINUE;
	}
	catch (const Firebird::Exception& ex)
	{
		ex.stuffException(status);
		return AUTH_FAILED;
	}
}

void FB_CARG DebugClient::getData(const unsigned char** data, unsigned short* dataSize)
{
	*data = reinterpret_cast<const unsigned char*>(str.c_str());
	*dataSize = str.length();
#ifdef AUTH_VERBOSE
	fprintf(stderr, "DebugClientInstance::getData: %.*s\n", *dataSize, *data);
#endif
}

int FB_CARG DebugClient::release()
{
	if (--refCounter == 0)
	{
		delete this;
		return 0;
	}

	return 1;
}

} // namespace Auth

#endif // AUTH_DEBUG