From c54fb59a87f258358c68c1d8e188b1ddfae50a7d Mon Sep 17 00:00:00 2001
From: Alexander Couzens <lynxis@fe80.eu>
Date: Mon, 22 Aug 2022 23:59:34 +0200
Subject: [PATCH] [dbi]: make all db functions optional

Allow database driver to only implement parts of the database driver.
In preprataion of the json interface which will only implement
session_data as might only be used by the PCRF.
---
 lib/dbi/dbi.c | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/lib/dbi/dbi.c b/lib/dbi/dbi.c
index e26ee7293..b828f3325 100644
--- a/lib/dbi/dbi.c
+++ b/lib/dbi/dbi.c
@@ -76,12 +76,16 @@ int ogs_dbi_msisdn_data(
         char *imsi_or_msisdn_bcd, ogs_msisdn_data_t *msisdn_data)
 {
     ogs_assert(dbi_selected);
+    if (!dbi_selected->msisdn_data)
+        return OGS_ERROR;
     return dbi_selected->msisdn_data(imsi_or_msisdn_bcd, msisdn_data);
 }
 
 int ogs_dbi_ims_data(char *supi, ogs_ims_data_t *ims_data)
 {
     ogs_assert(dbi_selected);
+    if (!dbi_selected->ims_data)
+        return OGS_ERROR;
     return dbi_selected->ims_data(supi, ims_data);
 }
 
@@ -90,6 +94,8 @@ int ogs_dbi_session_data(char *supi, ogs_s_nssai_t *s_nssai, char *dnn,
         ogs_session_data_t *session_data)
 {
     ogs_assert(dbi_selected);
+    if (!dbi_selected->session_data)
+        return OGS_ERROR;
     return dbi_selected->session_data(supi, s_nssai, dnn, session_data);
 }
 
@@ -97,24 +103,32 @@ int ogs_dbi_session_data(char *supi, ogs_s_nssai_t *s_nssai, char *dnn,
 int ogs_dbi_auth_info(char *supi, ogs_dbi_auth_info_t *auth_info)
 {
     ogs_assert(dbi_selected);
+    if (!dbi_selected->auth_info)
+        return OGS_ERROR;
     return dbi_selected->auth_info(supi, auth_info);
 }
 
 int ogs_dbi_update_sqn(char *supi, uint64_t sqn)
 {
     ogs_assert(dbi_selected);
+    if (!dbi_selected->update_sqn)
+        return OGS_ERROR;
     return dbi_selected->update_sqn(supi, sqn);
 }
 
 int ogs_dbi_increment_sqn(char *supi)
 {
     ogs_assert(dbi_selected);
+    if (!dbi_selected->increment_sqn)
+        return OGS_ERROR;
     return dbi_selected->increment_sqn(supi);
 }
 
 int ogs_dbi_update_imeisv(char *supi, char *imeisv)
 {
     ogs_assert(dbi_selected);
+    if (!dbi_selected->update_imeisv)
+        return OGS_ERROR;
     return dbi_selected->update_imeisv(supi, imeisv);
 }
 
@@ -122,6 +136,7 @@ int ogs_dbi_subscription_data(char *supi,
         ogs_subscription_data_t *subscription_data)
 {
     ogs_assert(dbi_selected);
+    if (!dbi_selected->subscription_data)
+        return OGS_ERROR;
     return dbi_selected->subscription_data(supi, subscription_data);
 }
-