From 49768177cfb52e9bf9ea3c5376d26d1247c73b3a Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Sun, 7 Nov 2021 12:49:50 +0000
Subject: [Minor] Allow to get a port for a specific upstream

---
 src/libutil/upstream.c | 9 +++++++++
 src/libutil/upstream.h | 7 +++++++
 2 files changed, 16 insertions(+)

(limited to 'src/libutil')

diff --git a/src/libutil/upstream.c b/src/libutil/upstream.c
index 8c194eb79..578efd0f4 100644
--- a/src/libutil/upstream.c
+++ b/src/libutil/upstream.c
@@ -1021,6 +1021,15 @@ rspamd_upstream_name (struct upstream *up)
 	return up->name;
 }
 
+gint
+rspamd_upstream_port (struct upstream *up)
+{
+	struct upstream_addr_elt *elt;
+
+	elt = g_ptr_array_index (up->addrs.addr, up->addrs.cur);
+	return rspamd_inet_address_get_port (elt->addr);
+}
+
 gboolean
 rspamd_upstreams_add_upstream (struct upstream_list *ups, const gchar *str,
 		guint16 def_port, enum rspamd_upstream_parse_type parse_type,
diff --git a/src/libutil/upstream.h b/src/libutil/upstream.h
index 72a768892..e81fa2ed7 100644
--- a/src/libutil/upstream.h
+++ b/src/libutil/upstream.h
@@ -248,6 +248,13 @@ gboolean rspamd_upstream_add_addr (struct upstream *up,
  */
 const gchar *rspamd_upstream_name (struct upstream *up);
 
+/**
+ * Returns the port of the current addres for the upstream
+ * @param up
+ * @return
+ */
+gint rspamd_upstream_port (struct upstream *up);
+
 /**
  * Sets opaque user data associated with this upstream
  * @param up
-- 
cgit v1.2.3