#include "stdafx.h"

INT_PTR SVC_OTRSendMessage(WPARAM wParam,LPARAM lParam){
	if (!lParam) return 0;

	CCSDATA *ccs = (CCSDATA *) lParam;

	DEBUGOUTW(L"SENDING\n");
		DEBUGOUTA("'");
		DEBUGOUTA((const char*)ccs->lParam);
		DEBUGOUTA("'\n");

	if (ccs->wParam & PREF_BYPASS_OTR) // bypass for OTR-messages
		return Proto_ChainSend(wParam, ccs);

	char *proto = Proto_GetBaseAccountName(ccs->hContact);
	if(proto && mir_strcmp(proto, META_PROTO) == 0) // bypass for metacontacts
		return Proto_ChainSend(wParam, ccs);
	
	if (!proto || !ccs->hContact)
		return 1; // error

	const char *oldmessage = (const char *)ccs->lParam;
	if (!oldmessage)
		return 1;

	// don't filter OTR messages being sent (OTR messages should only happen *after* the otrl_message_sending call below)
	if(strncmp(oldmessage, "?OTR", 4) == 0) {
		DEBUGOUTW(L"OTR message without PREF_BYPASS_OTR\n");
		return Proto_ChainSend(wParam, ccs);
	}

	ptrA tmpencode;
	ConnContext *context = otrl_context_find_miranda(otr_user_state, ccs->hContact);
	if (db_get_b(ccs->hContact, MODULENAME, "HTMLConv", 0) && otr_context_get_trust(context) >= TRUST_UNVERIFIED) {
		tmpencode = encode_html_entities_utf8(oldmessage);
		if (tmpencode != NULL)
			oldmessage = tmpencode;
	}
	
	char *newmessage = nullptr;
	gcry_error_t err = otrl_message_sending(otr_user_state, &ops, (void*)ccs->hContact, proto, proto, 
		ptrA(contact_get_id(ccs->hContact)), OTRL_INSTAG_BEST, oldmessage, nullptr, &newmessage, 
		OTRL_FRAGMENT_SEND_ALL_BUT_LAST, nullptr, add_appdata, (void*)ccs->hContact);
	
	if (err) { /* Be *sure* not to send out plaintext */
		DEBUGOUTA("otrl_message_sending err");
		ShowError(TranslateW(LANG_ENCRYPTION_ERROR));
		otrl_message_free(newmessage);
		return 1;
	}
	
	if (newmessage == nullptr)
		return Proto_ChainSend(wParam, ccs);
	
	if(!newmessage[0]){
		otrl_message_free(newmessage);
		return 1; // skip empty messages (OTR might prevent us sending unencrypted messages by replacing them with empty ones)
	}
	WPARAM oldflags = ccs->wParam;
	ccs->lParam = (LPARAM)newmessage;

	INT_PTR ret = Proto_ChainSend(wParam, ccs);

	DEBUGOUTA("OTR - sending raw message: '");
	DEBUGOUTA((const char*)ccs->lParam);
	DEBUGOUTA("'\n");
	otrl_message_free(newmessage);
		
	// reset to original values
	ccs->lParam = (LPARAM)oldmessage;
	ccs->wParam = oldflags;
	return ret;
}

INT_PTR SVC_OTRRecvMessage(WPARAM wParam,LPARAM lParam)
{
	CCSDATA *ccs = (CCSDATA *) lParam;
	PROTORECVEVENT *pre = (PROTORECVEVENT *) ccs->lParam;

	DEBUGOUTA("OTR - receiving message: '");
	DEBUGOUTA(pre->szMessage);
	DEBUGOUTA("'\n");

	if (pre->flags & PREF_BYPASS_OTR)  // bypass for our inline messages
		return Proto_ChainRecv(wParam, ccs);

	char *proto = Proto_GetBaseAccountName(ccs->hContact);
	if (!proto || !ccs->hContact)
		return 1; //error
	else if(proto && mir_strcmp(proto, META_PROTO) == 0) // bypass for metacontacts
		return Proto_ChainRecv(wParam, ccs);

	char *oldmessage = pre->szMessage;
	// convert oldmessage to utf-8
	if (!oldmessage)
		return 1;

	ConnContext* context=nullptr;
	char *newmessage = nullptr;
	OtrlTLV *tlvs = nullptr;
	
	int ignore_msg;
	{
		mir_cslock lck(lib_cs);
		ignore_msg = otrl_message_receiving(otr_user_state, &ops, (void*)ccs->hContact, proto, proto, ptrA(contact_get_id(ccs->hContact)),
			oldmessage, &newmessage, &tlvs, &context, add_appdata, (void*)ccs->hContact);
	}
	
	OtrlTLV *tlv = otrl_tlv_find(tlvs, OTRL_TLV_DISCONNECTED);
	if (tlv && !Miranda_IsTerminated()) {
		/* Notify the user that the other side disconnected. */
		wchar_t buff[256];
		mir_snwprintf(buff, TranslateW(LANG_SESSION_TERMINATED_BY_OTR), contact_get_nameT(ccs->hContact));
		SetEncryptionStatus(ccs->hContact, otr_context_get_trust(context)); // required since libotr 4!?
		ShowMessage(ccs->hContact, buff);
	}
	otrl_tlv_free(tlvs);

	if (ignore_msg) {
		if (newmessage)
			otrl_message_free(newmessage);
		return 1; // discard internal protocol messages
	}
	if (newmessage == nullptr)
		return Proto_ChainRecv(wParam, ccs);
	
	DWORD oldflags = pre->flags;
		
	typedef void (*msg_free_t)(void*);
	msg_free_t msg_free = (msg_free_t)otrl_message_free;
	if (db_get_b(ccs->hContact, MODULENAME, "HTMLConv", 0)) {
		char* tmp = striphtml(newmessage);
		msg_free(newmessage);
		newmessage = tmp;
		msg_free = mir_free;
	}
	if (options.prefix_messages) {
		CMStringA tmp = options.prefix;
		tmp.Append(newmessage);
		msg_free(newmessage);
		replaceStr(newmessage, tmp.Detach());
		msg_free = mir_free;
	}
	pre->szMessage = newmessage;
	BOOL ret = Proto_ChainRecv(wParam, ccs);
/// @todo (White-Tiger#1#03/23/15): why are we doing this?
	pre->flags = oldflags;
	pre->szMessage = oldmessage;
	msg_free(newmessage);
	return ret;
}

/* Abort the SMP protocol.  Used when malformed or unexpected messages
 * are received. */
void otr_abort_smp(ConnContext *context)
{
	otrl_message_abort_smp(otr_user_state, &ops, context->app_data, context);
}

/* Start the Socialist Millionaires' Protocol over the current connection,
 * using the given initial secret, and optionally a question to pass to
 * the buddy. */
void otr_start_smp(ConnContext *context, const char *question,
	const unsigned char *secret, size_t secretlen)
{
	otrl_message_initiate_smp_q(otr_user_state, &ops, context->app_data,
		context, question, secret, secretlen);
}

/* Continue the Socialist Millionaires' Protocol over the current connection,
 * using the given initial secret (ie finish step 2). */
void otr_continue_smp(ConnContext *context,
	const unsigned char *secret, size_t secretlen)
{
	otrl_message_respond_smp(otr_user_state, &ops, context->app_data,
		context, secret, secretlen);
}