/* gprs_ms_storage.cpp
 *
 * Copyright (C) 2015 by Sysmocom s.f.m.c. GmbH
 * Author: Jacob Erlbeck <jerlbeck@sysmocom.de>
 *
 * This program 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 2
 * of the License, or (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */


#include "gprs_ms_storage.h"

#include "tbf.h"
#include "gprs_debug.h"

GprsMsStorage::GprsMsStorage(BTS *bts) :
	m_bts(bts)
{
}

GprsMsStorage::~GprsMsStorage()
{
	LListHead<GprsMs> *pos, *tmp;

	llist_for_each_safe(pos, tmp, &m_list) {
		GprsMs *ms = pos->entry();
		ms->set_callback(NULL);
		ms_idle(ms);
	}
}

void GprsMsStorage::ms_idle(class GprsMs *ms)
{
	llist_del(&ms->list());
	if (ms->is_idle())
		delete ms;
}

void GprsMsStorage::ms_active(class GprsMs *ms)
{
	/* Nothing to do */
}

GprsMs *GprsMsStorage::get_ms(uint32_t tlli, uint32_t old_tlli, const char *imsi) const
{
	GprsMs *ms;
	LListHead<GprsMs> *pos;

	if (tlli || old_tlli) {
		llist_for_each(pos, &m_list) {
			ms = pos->entry();
			if (ms->check_tlli(tlli))
				return ms;
			if (ms->check_tlli(old_tlli))
				return ms;
		}
	}

	/* not found by TLLI */

	if (imsi && imsi[0]) {
		llist_for_each(pos, &m_list) {
			ms = pos->entry();
			if (strcmp(imsi, ms->imsi()) == 0)
				return ms;
		}
	}

	return NULL;
}

GprsMs *GprsMsStorage::create_ms()
{
	GprsMs *ms;

	ms = new GprsMs(m_bts, 0);

	ms->set_callback(this);
	llist_add(&ms->list(), &m_list);

	return ms;
}

GprsMs *GprsMsStorage::create_ms(uint32_t tlli, enum gprs_rlcmac_tbf_direction dir)
{
	GprsMs *ms = get_ms(tlli);

	if (ms)
		return ms;

	ms = create_ms();

	if (dir == GPRS_RLCMAC_UL_TBF)
		ms->set_tlli(tlli);
	else
		ms->confirm_tlli(tlli);

	return ms;
}
