aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/gbproxy/gb_proxy_peer.c8
-rw-r--r--src/gbproxy/gb_proxy_vty.c32
2 files changed, 39 insertions, 1 deletions
diff --git a/src/gbproxy/gb_proxy_peer.c b/src/gbproxy/gb_proxy_peer.c
index 3958de99..2c32587d 100644
--- a/src/gbproxy/gb_proxy_peer.c
+++ b/src/gbproxy/gb_proxy_peer.c
@@ -507,9 +507,15 @@ struct gbproxy_sgsn *gbproxy_sgsn_by_tlli(struct gbproxy_config *cfg, struct gbp
{
uint32_t i = 0;
uint32_t index, num_sgsns;
- struct gbproxy_sgsn *sgsn;
OSMO_ASSERT(cfg);
+ struct gbproxy_sgsn *sgsn = cfg->pool.nsf_override;
+
+ if (sgsn) {
+ LOGPSGSN(sgsn, LOGL_ERROR, "Node select function is overridden by config\n");
+ return sgsn;
+ }
+
// TODO: We should keep track of count in cfg
num_sgsns = llist_count(&cfg->sgsns);
diff --git a/src/gbproxy/gb_proxy_vty.c b/src/gbproxy/gb_proxy_vty.c
index adcdf706..bdf85e74 100644
--- a/src/gbproxy/gb_proxy_vty.c
+++ b/src/gbproxy/gb_proxy_vty.c
@@ -649,6 +649,36 @@ DEFUN(delete_gb_nsei, delete_gb_nsei_cmd,
return CMD_SUCCESS;
}
+/* Only for ttcn3 testing */
+DEFUN_HIDDEN(sgsn_pool_nsf_fixed, sgsn_pool_nsf_fixed_cmd,
+ "sgsn-pool nsf fixed NAME",
+ "SGSN pooling: load balancing across multiple SGSNs.\n"
+ "Customize the Network Selection Function.\n"
+ "Set a fixed SGSN to use (for testing).\n"
+ "The name of the SGSN to use.\n")
+{
+ const char *name = argv[0];
+ struct gbproxy_sgsn *sgsn = gbproxy_sgsn_by_name(g_cfg, name);
+
+ if (!sgsn) {
+ vty_out(vty, "%% Could not find SGSN with name %s%s", name, VTY_NEWLINE);
+ return CMD_WARNING;
+ }
+
+ g_cfg->pool.nsf_override = sgsn;
+ return CMD_SUCCESS;
+}
+
+DEFUN_HIDDEN(sgsn_pool_nsf_normal, sgsn_pool_nsf_normal_cmd,
+ "sgsn-pool nsf normal",
+ "SGSN pooling: load balancing across multiple SGSNs.\n"
+ "Customize the Network Selection Function.\n"
+ "Reset the NSF back to regular operation (for testing).\n")
+{
+ g_cfg->pool.nsf_override = NULL;
+ return CMD_SUCCESS;
+}
+
int gbproxy_vty_init(void)
{
install_element_ve(&show_gbproxy_bvc_cmd);
@@ -660,6 +690,8 @@ int gbproxy_vty_init(void)
install_element(ENABLE_NODE, &delete_gb_bvci_cmd);
install_element(ENABLE_NODE, &delete_gb_nsei_cmd);
+ install_element(ENABLE_NODE, &sgsn_pool_nsf_fixed_cmd);
+ install_element(ENABLE_NODE, &sgsn_pool_nsf_normal_cmd);
install_element(CONFIG_NODE, &cfg_gbproxy_cmd);
install_node(&gbproxy_node, config_write_gbproxy);