From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Thu, 21 Aug 2014 15:20:40 +0000 (+0100)
Subject: Add function rspamd_inet_address_set_port.
X-Git-Tag: 0.7.0~116
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=763bc588d221eecdbd607c88d4950ec5cf5d04da;p=thirdparty%2Frspamd.git

Add function rspamd_inet_address_set_port.
---

diff --git a/src/libutil/util.c b/src/libutil/util.c
index 3c696c5791..4562a90fe7 100644
--- a/src/libutil/util.c
+++ b/src/libutil/util.c
@@ -2371,6 +2371,19 @@ rspamd_inet_address_get_port (rspamd_inet_addr_t *addr)
 	return 0;
 }
 
+void
+rspamd_inet_address_set_port (rspamd_inet_addr_t *addr, uint16_t port)
+{
+	switch (addr->af) {
+	case AF_INET:
+		addr->addr.s4.sin_port = htons (port);
+		break;
+	case AF_INET6:
+		addr->addr.s6.sin6_port = htons (port);
+		break;
+	}
+}
+
 int
 rspamd_inet_address_connect (rspamd_inet_addr_t *addr, gint type,
 		gboolean async)
diff --git a/src/libutil/util.h b/src/libutil/util.h
index 7cc73f53a6..9bef49a7e8 100644
--- a/src/libutil/util.h
+++ b/src/libutil/util.h
@@ -510,6 +510,11 @@ const char * rspamd_inet_address_to_string (rspamd_inet_addr_t *addr);
  */
 uint16_t rspamd_inet_address_get_port (rspamd_inet_addr_t *addr);
 
+/**
+ * Set port for inet address
+ */
+void rspamd_inet_address_set_port (rspamd_inet_addr_t *addr, uint16_t port);
+
 /**
  * Connect to inet_addr address
  * @param addr