245 lines
7.4 KiB
C
245 lines
7.4 KiB
C
/* (C) 2008 by Jan Luebbe <jluebbe@debian.org>
|
|
* (C) 2009-2016 by Holger Hans Peter Freyther <zecke@selfish.org>
|
|
* (C) 2014 by Alexander Chemeris <Alexander.Chemeris@fairwaves.co>
|
|
* All Rights Reserved
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation; either version 3 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 Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
#include <openbsc/debug.h>
|
|
#include <openbsc/db.h>
|
|
#include <openbsc/gsm_subscriber.h>
|
|
#include <openbsc/gsm_04_11.h>
|
|
|
|
#include <osmocom/core/application.h>
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <inttypes.h>
|
|
|
|
static struct gsm_network dummy_net;
|
|
static struct gsm_subscriber_group dummy_sgrp;
|
|
|
|
#define SUBSCR_PUT(sub) \
|
|
sub->group = &dummy_sgrp; \
|
|
subscr_put(sub);
|
|
|
|
#define COMPARE(original, copy) \
|
|
if (original->id != copy->id) \
|
|
printf("Ids do not match in %s:%d %llu %llu\n", \
|
|
__FUNCTION__, __LINE__, original->id, copy->id); \
|
|
if (original->lac != copy->lac) \
|
|
printf("LAC do not match in %s:%d %d %d\n", \
|
|
__FUNCTION__, __LINE__, original->lac, copy->lac); \
|
|
if (original->authorized != copy->authorized) \
|
|
printf("Authorize do not match in %s:%d %d %d\n", \
|
|
__FUNCTION__, __LINE__, original->authorized, \
|
|
copy->authorized); \
|
|
if (strcmp(original->imsi, copy->imsi) != 0) \
|
|
printf("IMSIs do not match in %s:%d '%s' '%s'\n", \
|
|
__FUNCTION__, __LINE__, original->imsi, copy->imsi); \
|
|
if (original->tmsi != copy->tmsi) \
|
|
printf("TMSIs do not match in %s:%d '%u' '%u'\n", \
|
|
__FUNCTION__, __LINE__, original->tmsi, copy->tmsi); \
|
|
if (strcmp(original->name, copy->name) != 0) \
|
|
printf("names do not match in %s:%d '%s' '%s'\n", \
|
|
__FUNCTION__, __LINE__, original->name, copy->name); \
|
|
if (strcmp(original->extension, copy->extension) != 0) \
|
|
printf("Extensions do not match in %s:%d '%s' '%s'\n", \
|
|
__FUNCTION__, __LINE__, original->extension, copy->extension); \
|
|
|
|
/*
|
|
* Create/Store a SMS and then try to load it.
|
|
*/
|
|
static void test_sms(void)
|
|
{
|
|
int rc;
|
|
struct gsm_sms *sms;
|
|
struct gsm_subscriber *subscr;
|
|
subscr = db_get_subscriber(GSM_SUBSCRIBER_IMSI, "9993245423445");
|
|
OSMO_ASSERT(subscr);
|
|
subscr->group = &dummy_sgrp;
|
|
|
|
sms = sms_alloc();
|
|
sms->receiver = subscr_get(subscr);
|
|
|
|
sms->src.ton = 0x23;
|
|
sms->src.npi = 0x24;
|
|
memcpy(sms->src.addr, "1234", strlen("1234") + 1);
|
|
|
|
sms->dst.ton = 0x32;
|
|
sms->dst.npi = 0x42;
|
|
memcpy(sms->dst.addr, subscr->extension, sizeof(subscr->extension));
|
|
|
|
memcpy(sms->text, "Text123", strlen("Text123") + 1);
|
|
memcpy(sms->user_data, "UserData123", strlen("UserData123") + 1);
|
|
sms->user_data_len = strlen("UserData123");
|
|
|
|
/* random values */
|
|
sms->reply_path_req = 1;
|
|
sms->status_rep_req = 2;
|
|
sms->ud_hdr_ind = 3;
|
|
sms->protocol_id = 4;
|
|
sms->data_coding_scheme = 5;
|
|
|
|
rc = db_sms_store(sms);
|
|
sms_free(sms);
|
|
OSMO_ASSERT(rc == 0);
|
|
|
|
/* now query */
|
|
sms = db_sms_get_unsent_for_subscr(subscr);
|
|
OSMO_ASSERT(sms);
|
|
OSMO_ASSERT(sms->receiver == subscr);
|
|
OSMO_ASSERT(sms->reply_path_req == 1);
|
|
OSMO_ASSERT(sms->status_rep_req == 2);
|
|
OSMO_ASSERT(sms->ud_hdr_ind == 3);
|
|
OSMO_ASSERT(sms->protocol_id == 4);
|
|
OSMO_ASSERT(sms->data_coding_scheme == 5);
|
|
OSMO_ASSERT(sms->src.ton == 0x23);
|
|
OSMO_ASSERT(sms->src.npi == 0x24);
|
|
OSMO_ASSERT(sms->dst.ton == 0x32);
|
|
OSMO_ASSERT(sms->dst.npi == 0x42);
|
|
OSMO_ASSERT(strcmp((char *) sms->text, "Text123") == 0);
|
|
OSMO_ASSERT(sms->user_data_len == strlen("UserData123"));
|
|
OSMO_ASSERT(strcmp((char *) sms->user_data, "UserData123") == 0);
|
|
|
|
/* Mark the SMS as delivered */
|
|
db_sms_mark_delivered(sms);
|
|
sms_free(sms);
|
|
|
|
sms = db_sms_get_unsent_for_subscr(subscr);
|
|
OSMO_ASSERT(!sms);
|
|
|
|
subscr_put(subscr);
|
|
}
|
|
|
|
static void test_sms_migrate(void)
|
|
{
|
|
struct gsm_subscriber *rcv_subscr;
|
|
struct gsm_sms *sms;
|
|
static const uint8_t user_data_1[] = {
|
|
0x41, 0xf1, 0xd8, 0x05, 0x22, 0x96, 0xcd, 0x2e,
|
|
0x90, 0xf1, 0xfd, 0x06, 0x00 };
|
|
static const uint8_t user_data_2[] = {
|
|
0x41, 0xf1, 0xd8, 0x05, 0x22, 0x96, 0xcd, 0x2e,
|
|
0xd0, 0xf1, 0xfd, 0x06, 0x00 };
|
|
|
|
rcv_subscr = db_get_subscriber(GSM_SUBSCRIBER_IMSI, "901010000001111");
|
|
rcv_subscr->group = &dummy_sgrp;
|
|
|
|
sms = db_sms_get(&dummy_net, 1);
|
|
OSMO_ASSERT(sms->id == 1);
|
|
OSMO_ASSERT(sms->receiver == rcv_subscr);
|
|
OSMO_ASSERT(strcmp(sms->text, "Abc. Def. Foo") == 0);
|
|
OSMO_ASSERT(sms->user_data_len == ARRAY_SIZE(user_data_1));
|
|
OSMO_ASSERT(memcmp(sms->user_data, user_data_1, ARRAY_SIZE(user_data_1)) == 0);
|
|
sms_free(sms);
|
|
|
|
sms = db_sms_get(&dummy_net, 2);
|
|
OSMO_ASSERT(sms->id == 2);
|
|
OSMO_ASSERT(sms->receiver == rcv_subscr);
|
|
OSMO_ASSERT(strcmp(sms->text, "Abc. Def. Goo") == 0);
|
|
OSMO_ASSERT(sms->user_data_len == ARRAY_SIZE(user_data_2));
|
|
OSMO_ASSERT(memcmp(sms->user_data, user_data_2, ARRAY_SIZE(user_data_2)) == 0);
|
|
sms_free(sms);
|
|
|
|
subscr_put(rcv_subscr);
|
|
}
|
|
|
|
static void test_subs(const char *alice_imsi, char *imei1, char *imei2)
|
|
{
|
|
struct gsm_subscriber *alice = NULL, *alice_db;
|
|
char scratch_str[256];
|
|
|
|
alice = db_create_subscriber(alice_imsi, GSM_MIN_EXTEN, GSM_MAX_EXTEN);
|
|
db_subscriber_assoc_imei(alice, imei1);
|
|
if (imei2)
|
|
db_subscriber_assoc_imei(alice, imei2);
|
|
db_subscriber_alloc_tmsi(alice);
|
|
alice->lac=42;
|
|
db_sync_subscriber(alice);
|
|
/* Get by TMSI */
|
|
snprintf(scratch_str, sizeof(scratch_str), "%"PRIu32, alice->tmsi);
|
|
alice_db = db_get_subscriber(GSM_SUBSCRIBER_TMSI, scratch_str);
|
|
COMPARE(alice, alice_db);
|
|
SUBSCR_PUT(alice_db);
|
|
/* Get by IMSI */
|
|
alice_db = db_get_subscriber(GSM_SUBSCRIBER_IMSI, alice_imsi);
|
|
COMPARE(alice, alice_db);
|
|
SUBSCR_PUT(alice_db);
|
|
/* Get by id */
|
|
snprintf(scratch_str, sizeof(scratch_str), "%llu", alice->id);
|
|
alice_db = db_get_subscriber(GSM_SUBSCRIBER_ID, scratch_str);
|
|
COMPARE(alice, alice_db);
|
|
SUBSCR_PUT(alice_db);
|
|
/* Get by extension */
|
|
alice_db = db_get_subscriber(GSM_SUBSCRIBER_EXTENSION, alice->extension);
|
|
COMPARE(alice, alice_db);
|
|
SUBSCR_PUT(alice_db);
|
|
SUBSCR_PUT(alice);
|
|
}
|
|
|
|
int main()
|
|
{
|
|
printf("Testing subscriber database code.\n");
|
|
osmo_init_logging(&log_info);
|
|
log_set_print_filename(osmo_stderr_target, 0);
|
|
|
|
dummy_net.subscr_group = &dummy_sgrp;
|
|
dummy_sgrp.net = &dummy_net;
|
|
|
|
if (db_init("hlr.sqlite3")) {
|
|
printf("DB: Failed to init database. Please check the option settings.\n");
|
|
return 1;
|
|
}
|
|
printf("DB: Database initialized.\n");
|
|
|
|
if (db_prepare()) {
|
|
printf("DB: Failed to prepare database.\n");
|
|
return 1;
|
|
}
|
|
printf("DB: Database prepared.\n");
|
|
|
|
struct gsm_subscriber *alice = NULL;
|
|
struct gsm_subscriber *alice_db;
|
|
|
|
char *alice_imsi = "3243245432345";
|
|
alice = db_create_subscriber(alice_imsi, GSM_MIN_EXTEN, GSM_MAX_EXTEN);
|
|
db_sync_subscriber(alice);
|
|
alice_db = db_get_subscriber(GSM_SUBSCRIBER_IMSI, alice->imsi);
|
|
COMPARE(alice, alice_db);
|
|
SUBSCR_PUT(alice_db);
|
|
SUBSCR_PUT(alice);
|
|
|
|
test_subs("3693245423445", "1234567890", NULL);
|
|
test_subs("9993245423445", "1234567890", "6543560920");
|
|
|
|
/* create it again and see it fails */
|
|
alice = db_create_subscriber(alice_imsi, GSM_MIN_EXTEN, GSM_MAX_EXTEN);
|
|
OSMO_ASSERT(!alice);
|
|
|
|
test_sms();
|
|
test_sms_migrate();
|
|
|
|
db_fini();
|
|
|
|
printf("Done\n");
|
|
return 0;
|
|
}
|
|
|
|
/* stubs */
|
|
void vty_out() {}
|