cellular-infrastructure
/
osmo-msc
mirror of https://gerrit.osmocom.org/osmo-msc
9
0
Fork 0
Code Issues Releases Wiki Activity

token support in the DB

This commit is contained in:
Jan Luebbe 2009-08-12 14:31:14 +02:00 committed by Harald Welte
parent c70979a337
commit 31bef49be2
2 changed files with 55 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
openbsc/include/openbsc/db.h
View File

@ -38,6 +38,7 @@ struct gsm_subscriber* db_get_subscriber(struct gsm_network *net,
const char *subscr);
int db_sync_subscriber(struct gsm_subscriber* subscriber);
int db_subscriber_alloc_tmsi(struct gsm_subscriber* subscriber);
int db_subscriber_alloc_token(struct gsm_subscriber* subscriber, u_int32_t* token);
int db_subscriber_assoc_imei(struct gsm_subscriber* subscriber, char *imei);
int db_sync_equipment(struct gsm_equipment *equip);

54
openbsc/src/db.c
View File

@ -57,6 +57,12 @@ static char *create_stmts[] = {
"tmsi TEXT UNIQUE, "
"lac INTEGER NOT NULL DEFAULT 0"
")",
"CREATE TABLE IF NOT EXISTS AuthToken ("
"id INTEGER PRIMARY KEY AUTOINCREMENT, "
"subscriber_id INTEGER UNIQUE NOT NULL, "
"created TIMESTAMP NOT NULL, "
"token TEXT UNIQUE NOT NULL"
")",
"CREATE TABLE IF NOT EXISTS Equipment ("
"id INTEGER PRIMARY KEY AUTOINCREMENT, "
"created TIMESTAMP NOT NULL, "
@ -413,6 +419,54 @@ int db_subscriber_alloc_tmsi(struct gsm_subscriber* subscriber) {
return 0;
}
/*
* try to allocate a new unique token for this subscriber and return it
* via a parameter. if the subscriber already has a token, return
* an error.
*/
int db_subscriber_alloc_token(struct gsm_subscriber* subscriber, u_int32_t* token) {
dbi_result result=NULL;
u_int32_t try;
for (;;) {
try = rand();
if (!try) /* 0 is an invalid token */
continue;
result = dbi_conn_queryf(conn,
"SELECT * FROM AuthToken "
"WHERE subscriber_id = %llu OR token = %08x ",
subscriber->id, try
);
if (result==NULL) {
printf("DB: Failed to query AuthToken while allocating new token.\n");
return 1;
}
if (dbi_result_get_numrows(result)){
dbi_result_free(result);
continue;
}
if (!dbi_result_next_row(result)) {
dbi_result_free(result);
break;
}
dbi_result_free(result);
}
result = dbi_conn_queryf(conn,
"INSERT INTO AuthToken "
"(subscriber_id, created, token) "
"VALUES "
"(%llu, datetime('now'), %08x)) ",
subscriber->id, try
);
if (result==NULL) {
printf("DB: Failed to create token %08x for IMSI %s.\n", try, subscriber->imsi);
return 1;
}
*token = try;
printf("DB: Allocated token %08x for IMSI %s.\n", try, subscriber->imsi);
return 0;
}
int db_subscriber_assoc_imei(struct gsm_subscriber* subscriber, char imei[GSM_IMEI_LENGTH]) {
u_int64_t equipment_id, watch_id;
dbi_result result;