/* (C) 2008 by Jan Luebbe * (C) 2009-2016 by Holger Hans Peter Freyther * (C) 2014 by Alexander Chemeris * 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 . * */ #include #include #include #include #include #include #include #include #include #include #include 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 *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; } /* stubs */ void vty_out() {} unsigned int mgcpgw_client_next_endpoint(struct mgcpgw_client *client) { return 0; } int mgcpgw_client_tx_crcx(struct mgcpgw_client *client, uint16_t rtp_endpoint, unsigned int call_id, enum mgcp_connection_mode mode) { return -ENOTSUP; } const char *mgcpgw_client_remote_addr_str(struct mgcpgw_client *mgcp) { return "0.0.0.0"; } uint32_t mgcpgw_client_remote_addr_n(struct mgcpgw_client *mgcp) { return 0; }