136 lines
4.0 KiB
C
136 lines
4.0 KiB
C
#include <pthread.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <error.h>
|
|
|
|
#include <mysql/mysql.h>
|
|
|
|
#include "hss_config.h"
|
|
#include "db_proto.h"
|
|
|
|
int hss_mysql_query_pdns(const char *imsi,
|
|
mysql_pdn_t **pdns_p,
|
|
uint8_t *nb_pdns)
|
|
{
|
|
int ret;
|
|
MYSQL_RES *res;
|
|
MYSQL_ROW row;
|
|
char query[255];
|
|
mysql_pdn_t *pdn_array = NULL;
|
|
|
|
if (db_desc->db_conn == NULL) {
|
|
return EINVAL;
|
|
}
|
|
if (nb_pdns == NULL || pdns_p == NULL) {
|
|
return EINVAL;
|
|
}
|
|
|
|
sprintf(query, "SELECT * FROM `pdn` WHERE "
|
|
"`pdn`.`users_imsi`=%s LIMIT 10; ", imsi);
|
|
|
|
printf("Query: %s\n", query);
|
|
|
|
pthread_mutex_lock(&db_desc->db_cs_mutex);
|
|
|
|
if (mysql_query(db_desc->db_conn, query))
|
|
{
|
|
pthread_mutex_unlock(&db_desc->db_cs_mutex);
|
|
fprintf(stderr, "Query execution failed: %s\n",
|
|
mysql_error(db_desc->db_conn));
|
|
mysql_thread_end();
|
|
return EINVAL;
|
|
}
|
|
|
|
res = mysql_store_result(db_desc->db_conn);
|
|
|
|
pthread_mutex_unlock(&db_desc->db_cs_mutex);
|
|
|
|
*nb_pdns = 0;
|
|
|
|
while ((row = mysql_fetch_row(res)) != NULL)
|
|
{
|
|
mysql_pdn_t *pdn_elm; /* Local PDN element in array */
|
|
unsigned long *lengths;
|
|
|
|
lengths = mysql_fetch_lengths(res);
|
|
|
|
*nb_pdns += 1;
|
|
if (*nb_pdns == 1) {
|
|
/* First row, must malloc */
|
|
pdn_array = malloc(sizeof(mysql_pdn_t));
|
|
} else {
|
|
pdn_array = realloc(pdn_array, *nb_pdns * sizeof(mysql_pdn_t));
|
|
}
|
|
if (pdn_array == NULL) {
|
|
/* Error on malloc */
|
|
ret = ENOMEM;
|
|
goto err;
|
|
}
|
|
pdn_elm = &pdn_array[*nb_pdns - 1];
|
|
/* Copying the APN */
|
|
memset(pdn_elm, 0, sizeof(mysql_pdn_t));
|
|
memcpy(pdn_elm->apn, row[1], lengths[1]);
|
|
/* PDN Type + PDN address */
|
|
if (strcmp(row[2], "IPv6") == 0) {
|
|
pdn_elm->pdn_type = IPV6;
|
|
memcpy(pdn_elm->pdn_address.ipv6_address, row[4], lengths[4]);
|
|
pdn_elm->pdn_address.ipv6_address[lengths[4]] = '\0';
|
|
} else if (strcmp(row[2], "IPv4v6") == 0) {
|
|
pdn_elm->pdn_type = IPV4V6;
|
|
memcpy(pdn_elm->pdn_address.ipv4_address, row[3], lengths[3]);
|
|
pdn_elm->pdn_address.ipv4_address[lengths[3]] = '\0';
|
|
memcpy(pdn_elm->pdn_address.ipv6_address, row[4], lengths[4]);
|
|
pdn_elm->pdn_address.ipv6_address[lengths[4]] = '\0';
|
|
} else if (strcmp(row[2], "IPv4_or_IPv6") == 0) {
|
|
pdn_elm->pdn_type = IPV4_OR_IPV6;
|
|
memcpy(pdn_elm->pdn_address.ipv4_address, row[3], lengths[3]);
|
|
pdn_elm->pdn_address.ipv4_address[lengths[3]] = '\0';
|
|
memcpy(pdn_elm->pdn_address.ipv6_address, row[4], lengths[4]);
|
|
pdn_elm->pdn_address.ipv6_address[lengths[4]] = '\0';
|
|
} else {
|
|
pdn_elm->pdn_type = IPV4;
|
|
memcpy(pdn_elm->pdn_address.ipv4_address, row[3], lengths[3]);
|
|
pdn_elm->pdn_address.ipv4_address[lengths[3]] = '\0';
|
|
}
|
|
pdn_elm->aggr_ul = atoi(row[5]);
|
|
pdn_elm->aggr_dl = atoi(row[6]);
|
|
pdn_elm->qci = atoi(row[9]);
|
|
pdn_elm->priority_level = atoi(row[10]);
|
|
if (strcmp(row[11], "ENABLED") == 0) {
|
|
pdn_elm->pre_emp_cap = 0;
|
|
} else {
|
|
pdn_elm->pre_emp_cap = 1;
|
|
}
|
|
if (strcmp(row[12], "DISABLED") == 0) {
|
|
pdn_elm->pre_emp_vul = 1;
|
|
} else {
|
|
pdn_elm->pre_emp_vul = 0;
|
|
}
|
|
}
|
|
|
|
mysql_free_result(res);
|
|
mysql_thread_end();
|
|
|
|
/* We did not find any APN for the requested IMSI */
|
|
if (*nb_pdns == 0) {
|
|
return EINVAL;
|
|
} else {
|
|
*pdns_p = pdn_array;
|
|
return 0;
|
|
}
|
|
|
|
err:
|
|
if (pdn_array) {
|
|
free(pdn_array);
|
|
}
|
|
pdn_array = NULL;
|
|
*pdns_p = pdn_array;
|
|
*nb_pdns = 0;
|
|
mysql_free_result(res);
|
|
mysql_thread_end();
|
|
return ret;
|
|
}
|