[PATCH net-next 02/11] mptcp: unify ADD_ADDR and ADD_ADDR6 suboptions writing

[Mat Martineau]

From: Geliang Tang <geliangt...@gmail.com>

The length of ADD_ADDR6 is 12 octets longer than ADD_ADDR. That's the
only difference between them.

This patch dropped the duplicate code between ADD_ADDR and ADD_ADDR6
suboptions writing, and unify them into one.

Signed-off-by: Geliang Tang <geliangt...@gmail.com>
Signed-off-by: Mat Martineau <mathew.j.martin...@linux.intel.com>
---
 net/mptcp/options.c | 37 +++++++++++++++++--------------------
 1 file changed, 17 insertions(+), 20 deletions(-)

diff --git a/net/mptcp/options.c b/net/mptcp/options.c
index a061b2106cfe..639d47e6e2d0 100644
--- a/net/mptcp/options.c
+++ b/net/mptcp/options.c
@@ -1075,10 +1075,19 @@ void mptcp_write_options(__be32 *ptr, const struct 
tcp_sock *tp,
        }
 
 mp_capable_done:
-       if (OPTION_MPTCP_ADD_ADDR & opts->suboptions) {
+       if ((OPTION_MPTCP_ADD_ADDR
+#if IS_ENABLED(CONFIG_MPTCP_IPV6)
+            | OPTION_MPTCP_ADD_ADDR6
+#endif
+           ) & opts->suboptions) {
                u8 len = TCPOLEN_MPTCP_ADD_ADDR_BASE;
                u8 echo = MPTCP_ADDR_ECHO;
 
+#if IS_ENABLED(CONFIG_MPTCP_IPV6)
+               if (OPTION_MPTCP_ADD_ADDR6 & opts->suboptions)
+                       len = TCPOLEN_MPTCP_ADD_ADDR6_BASE;
+#endif
+
                if (opts->ahmac) {
                        len += sizeof(opts->ahmac);
                        echo = 0;
@@ -1086,33 +1095,21 @@ void mptcp_write_options(__be32 *ptr, const struct 
tcp_sock *tp,
 
                *ptr++ = mptcp_option(MPTCPOPT_ADD_ADDR,
                                      len, echo, opts->addr_id);
-               memcpy((u8 *)ptr, (u8 *)&opts->addr.s_addr, 4);
-               ptr += 1;
-               if (opts->ahmac) {
-                       put_unaligned_be64(opts->ahmac, ptr);
-                       ptr += 2;
+               if (OPTION_MPTCP_ADD_ADDR & opts->suboptions) {
+                       memcpy((u8 *)ptr, (u8 *)&opts->addr.s_addr, 4);
+                       ptr += 1;
                }
-       }
-
 #if IS_ENABLED(CONFIG_MPTCP_IPV6)
-       if (OPTION_MPTCP_ADD_ADDR6 & opts->suboptions) {
-               u8 len = TCPOLEN_MPTCP_ADD_ADDR6_BASE;
-               u8 echo = MPTCP_ADDR_ECHO;
-
-               if (opts->ahmac) {
-                       len += sizeof(opts->ahmac);
-                       echo = 0;
+               else if (OPTION_MPTCP_ADD_ADDR6 & opts->suboptions) {
+                       memcpy((u8 *)ptr, opts->addr6.s6_addr, 16);
+                       ptr += 4;
                }
-               *ptr++ = mptcp_option(MPTCPOPT_ADD_ADDR,
-                                     len, echo, opts->addr_id);
-               memcpy((u8 *)ptr, opts->addr6.s6_addr, 16);
-               ptr += 4;
+#endif
                if (opts->ahmac) {
                        put_unaligned_be64(opts->ahmac, ptr);
                        ptr += 2;
                }
        }
-#endif
 
        if (OPTION_MPTCP_RM_ADDR & opts->suboptions) {
                *ptr++ = mptcp_option(MPTCPOPT_RM_ADDR,
-- 
2.29.2