int main()
{
printf("Testing subscriber database code.\n");
osmo_init_logging(&log_info);
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(NULL, alice_imsi);
db_sync_subscriber(alice);
alice_db = db_get_subscriber(NULL, GSM_SUBSCRIBER_IMSI, alice->imsi);
COMPARE(alice, alice_db);
SUBSCR_PUT(alice_db);
SUBSCR_PUT(alice);
alice_imsi = "3693245423445";
alice = db_create_subscriber(NULL, alice_imsi);
db_subscriber_assoc_imei(alice, "1234567890");
db_subscriber_alloc_tmsi(alice);
alice->lac=42;
db_sync_subscriber(alice);
alice_db = db_get_subscriber(NULL, GSM_SUBSCRIBER_IMSI, alice_imsi);
COMPARE(alice, alice_db);
SUBSCR_PUT(alice);
SUBSCR_PUT(alice_db);
alice_imsi = "9993245423445";
alice = db_create_subscriber(NULL, alice_imsi);
db_subscriber_alloc_tmsi(alice);
alice->lac=42;
db_sync_subscriber(alice);
db_subscriber_assoc_imei(alice, "1234567890");
db_subscriber_assoc_imei(alice, "6543560920");
alice_db = db_get_subscriber(NULL, GSM_SUBSCRIBER_IMSI, alice_imsi);
COMPARE(alice, alice_db);
SUBSCR_PUT(alice);
SUBSCR_PUT(alice_db);
db_fini();
printf("Done\n");
return 0;
}
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 *imsi, char *imei1, char *imei2, bool make_ext)
{
struct gsm_subscriber *alice = NULL, *alice_db;
char scratch_str[256];
alice = db_create_subscriber(imsi, GSM_MIN_EXTEN, GSM_MAX_EXTEN,
make_ext);
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, 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);
if (alice_db) {
if (!make_ext)
printf("FAIL: bogus extension created for IMSI %s\n",
imsi);
COMPARE(alice, alice_db);
SUBSCR_PUT(alice_db);
} else if (make_ext)
printf("FAIL: no subscriber extension for IMSI %s\n", imsi);
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,
true);
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, true);
test_subs("9993245423445", "1234567890", "6543560920", true);
test_subs("3123122223445", "1234567890", NULL, false);
test_subs("9123121223445", "1234567890", "6543560920", false);
/* create it again and see it fails */
alice = db_create_subscriber(alice_imsi, GSM_MIN_EXTEN, GSM_MAX_EXTEN,
true);
OSMO_ASSERT(!alice);
test_sms();
test_sms_migrate();
db_fini();
printf("Done\n");
return 0;
}
/*
* 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);
}