/*
 * Copyright (C) 2009 Mamadou Diop.
 *
 * Contact: Mamadou Diop <diopmamadou@doubango.org>
 *	
 * This file is part of Open Source Doubango Framework.
 *
 * DOUBANGO is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *	
 * DOUBANGO is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *	
 * You should have received a copy of the GNU General Public License
 * along with DOUBANGO.
 *
 */
 #include "SMSEncoder.h"
 
 // Short description: http://betelco.blogspot.com/2009/10/sms-over-3gpp-ims-network.html
 
 
 RPMessage::RPMessage(twrap_rpmessage_type_t _type, tsms_rpdu_message_t* _rp_message)
 {
 	this->rp_message = (tsms_rpdu_message_t*)tsk_object_ref(_rp_message);
 	this->type = _type;
 	this->tmpBuffer = tsk_null;
 }
 
 RPMessage::RPMessage() : 
 	rp_message(tsk_null), 
 	type(twrap_rpmessage_type_sms_none),
 	tmpBuffer(tsk_null)
 {
 }
 
 twrap_rpmessage_type_t RPMessage::getType()
 {
 	return this->type;
 }
 
 unsigned RPMessage::getPayloadLength()
 {
 	if(!this->tmpBuffer){
 		if((this->tmpBuffer = tsk_buffer_create_null())){
 			tsms_rpdu_data_serialize(this->rp_message, this->tmpBuffer);
 		}
 	}
 	return this->tmpBuffer ? this->tmpBuffer->size : 0;
 }
 
 unsigned RPMessage::getPayload(void* output, unsigned maxsize)
 {
 	unsigned retsize = 0;
 
 	if(!this->tmpBuffer){
 		if((this->tmpBuffer = tsk_buffer_create_null())){
 			tsms_rpdu_message_serialize(this->rp_message, this->tmpBuffer);
 		}
 	}
 	
 	if(output && maxsize && this->tmpBuffer && this->tmpBuffer->data){
 		retsize = (this->tmpBuffer->size > maxsize) ? maxsize : this->tmpBuffer->size;
 		memcpy(output, this->tmpBuffer->data, retsize);
 	}
 	return retsize;
 }
 
 RPMessage::~RPMessage()
 {
 	TSK_OBJECT_SAFE_FREE(this->rp_message);
 	TSK_OBJECT_SAFE_FREE(this->tmpBuffer);
 }
 
 
 SMSData::SMSData(twrap_sms_type_t _type, int _mr, const void* _ascii, tsk_size_t _size): oa(tsk_null), da(tsk_null)
 {
 	this->type = _type;
 	this->mr = _mr;
 	if((this->size = _size)){
 		#if HAVE_CRT //Debug memory
 		if((this->ascii = calloc(size+1, 1))){
 		
 	#else
 			if((this->ascii = tsk_calloc(size+1, 1))){
 	
 	#endif //HAVE_CRT
 			memcpy(this->ascii, _ascii, _size);
 		}
 	}
 	else{
 		this->ascii = tsk_null;
 	}
 }
 
 SMSData::SMSData(): oa(tsk_null), da(tsk_null)
 {
 	this->type = twrap_sms_type_none;
 	this->mr = 0;
 	this->size = 0;
 	this->ascii = tsk_null;
 }
 
 SMSData::~SMSData()
 {
 	TSK_FREE(this->ascii);
 	TSK_FREE(this->oa);
 	TSK_FREE(this->da);
 }
 
 twrap_sms_type_t SMSData::getType()
 {
 	return this->type;
 }
 
 int SMSData::getMR()
 {
 	return this->mr;
 }
 
 unsigned SMSData::getPayloadLength()
 {
 	return this->size;
 }
 
 unsigned SMSData::getPayload(void* output, unsigned maxsize)
 {
 	unsigned retsize = 0;
 	
 	if(output && maxsize && this->ascii){
 		retsize = (this->size > maxsize) ? maxsize : this->size;
 		memcpy(output, this->ascii, retsize);
 	}
 	return retsize;
 }
 
 const char* SMSData::getOA()
 {
 	return this->oa;
 }
 
 const char* SMSData::getDA()
 {
 	return this->da;
 }
 
 void SMSData::setOA(const char* _oa)
 {
 	TSK_FREE(this->oa);
 	this->oa = tsk_strdup(_oa);
 }
 
 void SMSData::setDA(const char* _da)
 {
 	TSK_FREE(this->da);
 	this->da = tsk_strdup(_da);
 }
 
 
 // More information about RP-DATA: http://www.doubango.org/API/tinySMS/group__tsms__rpdu__group.html#tsms_rpdu_group_DATA
 RPMessage* SMSEncoder::encodeSubmit(int mr, const char *smsc, const char *destination, const char *ascii)
 {
 	int ret;
 	tsk_buffer_t* buffer = tsk_null;
 	tsms_tpdu_submit_t* sms_submit = tsk_null;
 	tsms_rpdu_data_t* rp_data = tsk_null;
 
 	RPMessage* encodedData = tsk_null;
 	
 	if(!smsc || ! destination || !ascii){
 		TSK_DEBUG_ERROR("Invalid parameter");
 		return tsk_null;
 	}
 
 	if(mr<0 || mr>0xFF){
 		TSK_DEBUG_WARN("Invalid Message Reference");
 		mr &= 0xFF;
 	}
 
 	// create SMS-SUBMIT message
 	if(!(sms_submit = tsms_tpdu_submit_create(mr, (const uint8_t*)smsc, (const uint8_t*)destination))){
 		TSK_DEBUG_ERROR("Failed to create the TPDU SMS-SUBMIT message");
 		goto bail;
 	}
 	// Set content for SMS-SUBMIT
 	if((buffer = tsms_pack_to_7bit(ascii))){
 		ret = tsms_tpdu_submit_set_userdata(sms_submit, buffer, tsms_alpha_7bit);
 		TSK_OBJECT_SAFE_FREE(buffer);
 	}
 	else{
 		TSK_DEBUG_ERROR("Failed to encode the TPDU SMS-SUBMIT message");
 		goto bail;
 	}
 
 	// create RP-DATA(SMS-SUBMIT)
 	if((rp_data = tsms_rpdu_data_create_mo(mr, (const uint8_t*)smsc, TSMS_TPDU_MESSAGE(sms_submit)))){
 		encodedData = new RPMessage(twrap_rpmessage_type_sms_submit, TSMS_RPDU_MESSAGE(rp_data));
 	}
 	else{
 		TSK_DEBUG_ERROR("Failed to create the RP-DATA(SMS-SUBMIT) message");
 		goto bail;
 	}
 	
 bail:
 	TSK_OBJECT_SAFE_FREE(buffer);
 	TSK_OBJECT_SAFE_FREE(sms_submit);
 	TSK_OBJECT_SAFE_FREE(rp_data);
 
 	return encodedData;
 }
 
 // More information about RP-DATA: http://www.doubango.org/API/tinySMS/group__tsms__rpdu__group.html#tsms_rpdu_group_DATA
 RPMessage* SMSEncoder::encodeDeliver(int mr, const char* smsc, const char* originator, const char* ascii)
 {
 	int ret;
 	tsk_buffer_t* buffer = tsk_null;
 	tsms_tpdu_deliver_t* sms_deliver = tsk_null;
 	tsms_rpdu_data_t* rp_data = tsk_null;
 	
 	RPMessage* encodedData = tsk_null;
 
 	if(!smsc || ! originator || !ascii){
 		TSK_DEBUG_ERROR("Invalid parameter");
 		return tsk_null;
 	}
 
 	if(mr<0 || mr>0xFF){
 		TSK_DEBUG_WARN("Invalid Message Reference");
 		mr &= 0xFF;
 	}
 
 	// create SMS-DELIVER message
 	sms_deliver = tsms_tpdu_deliver_create((const uint8_t*)smsc, (const uint8_t*)originator);
 	// Set content for SMS-DELIVER
 	if((buffer = tsms_pack_to_7bit(ascii))){
 		ret = tsms_tpdu_deliver_set_userdata(sms_deliver, buffer, tsms_alpha_7bit);
 		TSK_OBJECT_SAFE_FREE(buffer);
 	}
 	// create RP-DATA message
 	if((rp_data = tsms_rpdu_data_create_mt(mr, (const uint8_t*)smsc, TSMS_TPDU_MESSAGE(sms_deliver)))){
 		encodedData = new RPMessage(twrap_rpmessage_type_sms_deliver, TSMS_RPDU_MESSAGE(rp_data));
 	}
 	else{
 		TSK_DEBUG_ERROR("Failed to create the RP-DATA(SMS-DELIVER) message");
 		goto bail;
 	}
 
 bail:
 	TSK_OBJECT_SAFE_FREE(buffer);
 	TSK_OBJECT_SAFE_FREE(sms_deliver);
 	TSK_OBJECT_SAFE_FREE(rp_data);
 
 	return encodedData;
 }
 
 RPMessage* SMSEncoder::encodeACK(int mr, const char* smsc, const char* destination, bool forSUBMIT)
 {
 	tsms_tpdu_report_t* sms_report = tsk_null;
 	tsms_rpdu_ack_t* rp_ack = tsk_null;
 	tsk_bool_t isSUBMIT = forSUBMIT ? tsk_true : tsk_false;
 	tsk_bool_t isERROR = tsk_false;
 
 	RPMessage* encodedData = tsk_null;
 
 	// create SMS-DELIVER(or SUBMIT)-REPORT message
 	sms_report = tsms_tpdu_report_create((const uint8_t*)smsc, isSUBMIT, isERROR);
 	// create RP-ACK message (From MS to SC)
 	if((rp_ack = tsms_rpdu_ack_create_mo(mr, TSMS_TPDU_MESSAGE(sms_report)))){
 		encodedData = new RPMessage(twrap_rpmessage_type_sms_ack, TSMS_RPDU_MESSAGE(rp_ack));
 	}
 	
 	TSK_OBJECT_SAFE_FREE(sms_report);
 	TSK_OBJECT_SAFE_FREE(rp_ack);
 
 	return encodedData;
 }
 
 RPMessage* SMSEncoder::encodeError(int mr, const char* smsc, const char* destination, bool forSUBMIT)
 {
 	tsms_tpdu_report_t* sms_report = tsk_null;
 	tsms_rpdu_error_t* rp_error= tsk_null;
 	tsk_bool_t isSUBMIT = forSUBMIT ? tsk_true : tsk_false;
 	tsk_bool_t isERROR = tsk_true;
 
 	RPMessage* encodedData = tsk_null;
 
 	// create SMS-DELIVER-REPORT message
 	sms_report = tsms_tpdu_report_create((const uint8_t*)smsc, isSUBMIT, isERROR);
 	// create RP-ERROR message
 	if((rp_error = tsms_rpdu_error_create_mo(mr, TSMS_TPDU_MESSAGE(sms_report), 0x0A/*call barred*/))){
 		encodedData = new RPMessage(twrap_rpmessage_type_sms_error, TSMS_RPDU_MESSAGE(rp_error));
 	}
 
 	TSK_OBJECT_SAFE_FREE(sms_report);
 	TSK_OBJECT_SAFE_FREE(rp_error);
 
 	return encodedData;
 }
 
 SMSData* SMSEncoder::decode(const void* data, unsigned size, bool MobOrig)
 {
 	tsms_rpdu_message_t* rp_message = tsk_null;
 	tsms_tpdu_message_t* tpdu = tsk_null;
 
 	SMSData* decodedData = tsk_null;
 
 	if(!(rp_message = tsms_rpdu_message_deserialize(data, size))){
 		TSK_DEBUG_ERROR("Failed to deserialize the RP-MESSAGE");
 		goto bail;
 	}
 	
 	switch(rp_message->mti){
 		case tsms_rpdu_type_data_mo:
 		case tsms_rpdu_type_data_mt:
 			{
 				char* ascii = tsk_null;
 				tsms_rpdu_data_t* rp_data = TSMS_RPDU_DATA(rp_message);
 				if((tpdu = tsms_tpdu_message_deserialize(rp_data->udata->data, rp_data->udata->size, MobOrig))){
 					if(tpdu->mti == tsms_tpdu_mti_deliver_mt || tpdu->mti == tsms_tpdu_mti_submit_mo){ /* SMS-SUBMIT or SMS-DELIVER? */
 						ascii = tsms_tpdu_message_get_payload(tpdu);
 						decodedData = new SMSData(twrap_sms_type_rpdata, rp_message->mr, ascii, (tsk_size_t)tsk_strlen(ascii));
 
 						if(tpdu->mti == tsms_tpdu_mti_deliver_mt){
 							tsms_tpdu_deliver_t* tpdu_deliver = (tsms_tpdu_deliver_t*)tpdu;
 							decodedData->setOA(tpdu_deliver->oa ? tpdu_deliver->oa->digits : tsk_null);
 						}
 						/* IMPORTANT: to not uncomment
 						else if(TSK_OBJECT_DEF(tpdu) == tsms_tpdu_submit_def_t){
 							tsms_tpdu_submit_t* tpdu_submit = (tsms_tpdu_submit_t*)tpdu;
 							decodedData->setDA(tpdu_submit->da ? tpdu_submit->da->digits : tsk_null);
 						}*/
 						TSK_FREE(ascii);
 					}
 				}
 				break;
 			}
 		case tsms_rpdu_type_ack_mo:
 		case tsms_rpdu_type_ack_mt:
 			 {
 				tsms_rpdu_ack_t* rp_ack = TSMS_RPDU_ACK(rp_message);
 				// ...do whatever you want
 				if(rp_ack->udata && (tpdu = tsms_tpdu_message_deserialize(rp_ack->udata->data, rp_ack->udata->size, MobOrig))){
 					// ...do whatever you want
 				}
 				decodedData = new SMSData(twrap_sms_type_ack, rp_message->mr, tsk_null, 0);
 				break;
 			}
 		case tsms_rpdu_type_error_mo:
 		case tsms_rpdu_type_error_mt:
 			 {
 				tsms_rpdu_error_t* rp_error = TSMS_RPDU_ERROR(rp_message);
 				// ...do whatever you want
 				if(rp_error->udata && (tpdu = tsms_tpdu_message_deserialize(rp_error->udata->data, rp_error->udata->size, MobOrig))){
 					// ...do whatever you want
 				}
 				decodedData = new SMSData(twrap_sms_type_error, rp_message->mr, tsk_null, 0);
 				break;
 			}
 		case tsms_rpdu_type_smma_mo:
 			{
 				// tsms_rpdu_smma_t* rp_smma = TSMS_RPDU_SMMA(rp_message);
 				// ...do whatever you want
 				decodedData = new SMSData(twrap_sms_type_smma, rp_message->mr, tsk_null, 0);
 				break;
 			}
 		default:
 			{
 				TSK_DEBUG_INFO("Unknown RP-Message type (%u).", rp_message->mti);
 				break;
 			}
 	}
 
 bail:
 	TSK_OBJECT_SAFE_FREE(rp_message);
 	TSK_OBJECT_SAFE_FREE(tpdu);
 
 	return decodedData;
 }