| /* (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 <stdbool.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 *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() {} |