1
From: Geliang Tang <tanggeliang@kylinos.cn>
1
From: Geliang Tang <tanggeliang@kylinos.cn>
2
3
v4:
4
- include a new patch "define BPF path manager type".
5
6
- add new interfaces:
7
    created established closed
8
    listerner_created listener_closed
9
10
- rename interfaces as:
11
    address_announced address_removed
12
    subflow_established subflow_closed
13
    get_priority set_priority
14
15
- rename functions as:
16
    mptcp_pm_validate
17
    mptcp_pm_register
18
    mptcp_pm_unregister
19
    mptcp_pm_initialize
20
    mptcp_pm_release
21
22
v3:
23
- rename the 2nd parameter of get_local_id() from 'local' to 'skc'.
24
- keep the 'msk_sport' check in mptcp_userspace_pm_get_local_id().
25
- return 'err' instead of '0' in userspace_pm_subflow_create().
26
- drop 'ret' variable inmptcp_pm_data_reset().
27
- fix typos in commit log.
28
29
v2:
30
- update get_local_id interface in patch 2.
2
31
3
get_addr() and dump_addr() interfaces of BPF userspace pm are dropped
32
get_addr() and dump_addr() interfaces of BPF userspace pm are dropped
4
as Matt suggested.
33
as Matt suggested.
5
6
Based-on: <cover.1736924314.git.tanggeliang@kylinos.cn>
7
34
8
In order to implement BPF userspace path manager, it is necessary to
35
In order to implement BPF userspace path manager, it is necessary to
9
unify the interfaces of the path manager. This set contains some
36
unify the interfaces of the path manager. This set contains some
10
cleanups and refactoring to unify the interfaces in kernel space.
37
cleanups and refactoring to unify the interfaces in kernel space.
11
Finally, define a struct mptcp_pm_ops for a userspace path manager
38
Finally, define a struct mptcp_pm_ops for a userspace path manager
12
like this:
39
like this:
13
40
14
struct mptcp_pm_ops {
41
struct mptcp_pm_ops {
15
    int (*address_announce)(struct mptcp_sock *msk,
42
int (*created)(struct mptcp_sock *msk);
16
                struct mptcp_pm_addr_entry *local);
43
int (*established)(struct mptcp_sock *msk);
17
    int (*address_remove)(struct mptcp_sock *msk, u8 id);
44
int (*closed)(struct mptcp_sock *msk);
18
    int (*subflow_create)(struct mptcp_sock *msk,
45
int (*address_announced)(struct mptcp_sock *msk,
19
             struct mptcp_pm_addr_entry *local,
46
struct mptcp_pm_addr_entry *local);
20
             struct mptcp_addr_info *remote);
47
int (*address_removed)(struct mptcp_sock *msk, u8 id);
21
    int (*subflow_destroy)(struct mptcp_sock *msk,
48
int (*subflow_established)(struct mptcp_sock *msk,
22
             struct mptcp_pm_addr_entry *local,
49
struct mptcp_pm_addr_entry *local,
23
             struct mptcp_addr_info *remote);
50
struct mptcp_addr_info *remote);
24
    int (*get_local_id)(struct mptcp_sock *msk,
51
int (*subflow_closed)(struct mptcp_sock *msk,
25
             struct mptcp_pm_addr_entry *local);
52
struct mptcp_pm_addr_entry *local,
26
    u8 (*get_flags)(struct mptcp_sock *msk,
53
struct mptcp_addr_info *remote);
27
            struct mptcp_addr_info *skc);
54
int (*get_local_id)(struct mptcp_sock *msk,
28
    int (*set_flags)(struct mptcp_sock *msk,
55
struct mptcp_pm_addr_entry *skc);
29
             struct mptcp_pm_addr_entry *local,
56
bool (*get_priority)(struct mptcp_sock *msk,
30
             struct mptcp_addr_info *remote);
57
struct mptcp_addr_info *skc);
58
int (*set_priority)(struct mptcp_sock *msk,
59
struct mptcp_pm_addr_entry *local,
60
struct mptcp_addr_info *remote);
61
int (*listener_created)(struct mptcp_sock *msk);
62
int (*listener_closed)(struct mptcp_sock *msk);
31
63
32
    u8            type;
64
u8 type;
33
    struct module        *owner;
65
struct module *owner;
34
    struct list_head    list;
66
struct list_head list;
35
67
36
    void (*init)(struct mptcp_sock *msk);
68
void (*init)(struct mptcp_sock *msk);
37
    void (*release)(struct mptcp_sock *msk);
69
void (*release)(struct mptcp_sock *msk);
38
} ____cacheline_aligned_in_smp;
70
} ____cacheline_aligned_in_smp;
39
71
40
Closes: https://github.com/multipath-tcp/mptcp_net-next/issues/74
72
Geliang Tang (4):
73
mptcp: define struct mptcp_pm_ops
74
mptcp: define BPF path manager type
75
mptcp: register default userspace pm
76
mptcp: initialize and release mptcp_pm_ops
41
77
42
Geliang Tang (3):
78
include/net/mptcp.h | 32 +++++
43
mptcp: define struct mptcp_pm_ops
79
net/mptcp/pm.c | 109 ++++++++++++++-
44
mptcp: register default userspace pm
80
net/mptcp/pm_netlink.c | 11 +-
45
mptcp: init and release mptcp_pm_ops
81
net/mptcp/pm_userspace.c | 294 ++++++++++++++++++++++++---------------
46
82
net/mptcp/protocol.c | 10 +-
47
include/net/mptcp.h | 27 +++
83
net/mptcp/protocol.h | 15 +-
48
net/mptcp/pm.c | 8 +
84
6 files changed, 355 insertions(+), 116 deletions(-)
49
net/mptcp/pm_userspace.c | 362 ++++++++++++++++++++++++++++-----------
50
net/mptcp/protocol.c | 1 +
51
net/mptcp/protocol.h | 9 +
52
5 files changed, 309 insertions(+), 98 deletions(-)
53
85
54
--
86
--
55
2.43.0
87
2.43.0
diff view generated by jsdifflib
...
...
7
Add a set of functions to register, unregister, find and validate a
7
Add a set of functions to register, unregister, find and validate a
8
given struct ops.
8
given struct ops.
9
9
10
Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
10
Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
11
---
11
---
12
include/net/mptcp.h | 27 +++++++++++++++++++
12
include/net/mptcp.h | 32 ++++++++++++++++++++++++
13
net/mptcp/pm_userspace.c | 57 ++++++++++++++++++++++++++++++++++++++++
13
net/mptcp/pm.c | 59 ++++++++++++++++++++++++++++++++++++++++++++
14
net/mptcp/protocol.h | 5 ++++
14
net/mptcp/protocol.h | 5 ++++
15
3 files changed, 89 insertions(+)
15
3 files changed, 96 insertions(+)
16
16
17
diff --git a/include/net/mptcp.h b/include/net/mptcp.h
17
diff --git a/include/net/mptcp.h b/include/net/mptcp.h
18
index XXXXXXX..XXXXXXX 100644
18
index XXXXXXX..XXXXXXX 100644
19
--- a/include/net/mptcp.h
19
--- a/include/net/mptcp.h
20
+++ b/include/net/mptcp.h
20
+++ b/include/net/mptcp.h
...
...
29
@@ -XXX,XX +XXX,XX @@ struct mptcp_sched_ops {
29
@@ -XXX,XX +XXX,XX @@ struct mptcp_sched_ops {
30
    void (*release)(struct mptcp_sock *msk);
30
    void (*release)(struct mptcp_sock *msk);
31
} ____cacheline_aligned_in_smp;
31
} ____cacheline_aligned_in_smp;
32
32
33
+struct mptcp_pm_ops {
33
+struct mptcp_pm_ops {
34
+    int (*address_announce)(struct mptcp_sock *msk,
34
+    int (*created)(struct mptcp_sock *msk);
35
+                struct mptcp_pm_addr_entry *local);
35
+    int (*established)(struct mptcp_sock *msk);
36
+    int (*address_remove)(struct mptcp_sock *msk, u8 id);
36
+    int (*closed)(struct mptcp_sock *msk);
37
+    int (*subflow_create)(struct mptcp_sock *msk,
37
+    int (*address_announced)(struct mptcp_sock *msk,
38
+                 struct mptcp_pm_addr_entry *local);
39
+    int (*address_removed)(struct mptcp_sock *msk, u8 id);
40
+    int (*subflow_established)(struct mptcp_sock *msk,
41
+                 struct mptcp_pm_addr_entry *local,
42
+                 struct mptcp_addr_info *remote);
43
+    int (*subflow_closed)(struct mptcp_sock *msk,
38
+             struct mptcp_pm_addr_entry *local,
44
+             struct mptcp_pm_addr_entry *local,
39
+             struct mptcp_addr_info *remote);
45
+             struct mptcp_addr_info *remote);
40
+    int (*subflow_destroy)(struct mptcp_sock *msk,
41
+             struct mptcp_pm_addr_entry *local,
42
+             struct mptcp_addr_info *remote);
43
+    int (*get_local_id)(struct mptcp_sock *msk,
46
+    int (*get_local_id)(struct mptcp_sock *msk,
44
+             struct mptcp_pm_addr_entry *local);
47
+             struct mptcp_pm_addr_entry *skc);
45
+    u8 (*get_flags)(struct mptcp_sock *msk,
48
+    bool (*get_priority)(struct mptcp_sock *msk,
46
+            struct mptcp_addr_info *skc);
49
+             struct mptcp_addr_info *skc);
47
+    int (*set_flags)(struct mptcp_sock *msk,
50
+    int (*set_priority)(struct mptcp_sock *msk,
48
+             struct mptcp_pm_addr_entry *local,
51
+             struct mptcp_pm_addr_entry *local,
49
+             struct mptcp_addr_info *remote);
52
+             struct mptcp_addr_info *remote);
53
+    int (*listener_created)(struct mptcp_sock *msk);
54
+    int (*listener_closed)(struct mptcp_sock *msk);
50
+
55
+
51
+    u8            type;
56
+    u8            type;
52
+    struct module        *owner;
57
+    struct module        *owner;
53
+    struct list_head    list;
58
+    struct list_head    list;
54
+
59
+
...
...
57
+} ____cacheline_aligned_in_smp;
62
+} ____cacheline_aligned_in_smp;
58
+
63
+
59
#ifdef CONFIG_MPTCP
64
#ifdef CONFIG_MPTCP
60
void mptcp_init(void);
65
void mptcp_init(void);
61
66
62
diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c
67
diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c
63
index XXXXXXX..XXXXXXX 100644
68
index XXXXXXX..XXXXXXX 100644
64
--- a/net/mptcp/pm_userspace.c
69
--- a/net/mptcp/pm.c
65
+++ b/net/mptcp/pm_userspace.c
70
+++ b/net/mptcp/pm.c
66
@@ -XXX,XX +XXX,XX @@
71
@@ -XXX,XX +XXX,XX @@
67
* Copyright (c) 2022, Intel Corporation.
72
#define pr_fmt(fmt) "MPTCP: " fmt
68
*/
73
69
74
#include <linux/kernel.h>
70
+#include <linux/rculist.h>
75
+#include <linux/rculist.h>
71
+#include <linux/spinlock.h>
76
+#include <linux/spinlock.h>
77
#include <net/mptcp.h>
72
#include "protocol.h"
78
#include "protocol.h"
79
73
#include "mib.h"
80
#include "mib.h"
74
#include "mptcp_pm_gen.h"
81
#include "mptcp_pm_gen.h"
75
@@ -XXX,XX +XXX,XX @@
76
    list_for_each_entry(__entry,                        \
77
             &((__msk)->pm.userspace_pm_local_addr_list), list)
78
82
79
+static DEFINE_SPINLOCK(mptcp_pm_list_lock);
83
+static DEFINE_SPINLOCK(mptcp_pm_list_lock);
80
+static LIST_HEAD(mptcp_pm_list);
84
+static LIST_HEAD(mptcp_pm_list);
81
+
85
+
82
void mptcp_free_local_addr_list(struct mptcp_sock *msk)
86
/* path manager command handlers */
87
88
int mptcp_pm_announce_addr(struct mptcp_sock *msk,
89
@@ -XXX,XX +XXX,XX @@ void __init mptcp_pm_init(void)
83
{
90
{
84
    struct mptcp_pm_addr_entry *entry, *tmp;
91
    mptcp_pm_nl_init();
85
@@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_addr(u8 id, struct mptcp_pm_addr_entry *addr,
86
    sock_put(sk);
87
    return ret;
88
}
92
}
89
+
93
+
90
+/* Must be called with rcu read lock held */
94
+/* Must be called with rcu read lock held */
91
+struct mptcp_pm_ops *mptcp_pm_find(enum mptcp_pm_type type)
95
+struct mptcp_pm_ops *mptcp_pm_find(enum mptcp_pm_type type)
92
+{
96
+{
...
...
98
+    }
102
+    }
99
+
103
+
100
+    return NULL;
104
+    return NULL;
101
+}
105
+}
102
+
106
+
103
+int mptcp_validate_path_manager(struct mptcp_pm_ops *pm)
107
+int mptcp_pm_validate(struct mptcp_pm_ops *pm)
104
+{
108
+{
105
+    if (!pm->address_announce && !pm->address_remove &&
109
+    if (!pm->created && !pm->established && !pm->closed &&
106
+     !pm->subflow_create && !pm->subflow_destroy &&
110
+     !pm->address_announced && !pm->address_removed &&
107
+     !pm->get_local_id && !pm->get_flags && !pm->set_flags) {
111
+     !pm->subflow_established && !pm->subflow_closed &&
112
+     !pm->get_local_id && !pm->get_priority && !pm->set_priority &&
113
+     !pm->listener_created && !pm->listener_closed) {
108
+        pr_err("%u does not implement required ops\n", pm->type);
114
+        pr_err("%u does not implement required ops\n", pm->type);
109
+        return -EINVAL;
115
+        return -EINVAL;
110
+    }
116
+    }
111
+
117
+
112
+    return 0;
118
+    return 0;
113
+}
119
+}
114
+
120
+
115
+int mptcp_register_path_manager(struct mptcp_pm_ops *pm)
121
+int mptcp_pm_register(struct mptcp_pm_ops *pm)
116
+{
122
+{
117
+    int ret;
123
+    int ret;
118
+
124
+
119
+    ret = mptcp_validate_path_manager(pm);
125
+    ret = mptcp_pm_validate(pm);
120
+    if (ret)
126
+    if (ret)
121
+        return ret;
127
+        return ret;
122
+
128
+
123
+    spin_lock(&mptcp_pm_list_lock);
129
+    spin_lock(&mptcp_pm_list_lock);
124
+    if (mptcp_pm_find(pm->type)) {
130
+    if (mptcp_pm_find(pm->type)) {
...
...
130
+
136
+
131
+    pr_debug("userspace_pm type %u registered\n", pm->type);
137
+    pr_debug("userspace_pm type %u registered\n", pm->type);
132
+    return 0;
138
+    return 0;
133
+}
139
+}
134
+
140
+
135
+void mptcp_unregister_path_manager(struct mptcp_pm_ops *pm)
141
+void mptcp_pm_unregister(struct mptcp_pm_ops *pm)
136
+{
142
+{
137
+    spin_lock(&mptcp_pm_list_lock);
143
+    spin_lock(&mptcp_pm_list_lock);
138
+    list_del_rcu(&pm->list);
144
+    list_del_rcu(&pm->list);
139
+    spin_unlock(&mptcp_pm_list_lock);
145
+    spin_unlock(&mptcp_pm_list_lock);
140
+}
146
+}
...
...
145
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_remove_addr(struct mptcp_sock *msk, const struct mptcp_rm_list *rm_
151
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_remove_addr(struct mptcp_sock *msk, const struct mptcp_rm_list *rm_
146
void mptcp_pm_remove_addr_entry(struct mptcp_sock *msk,
152
void mptcp_pm_remove_addr_entry(struct mptcp_sock *msk,
147
                struct mptcp_pm_addr_entry *entry);
153
                struct mptcp_pm_addr_entry *entry);
148
154
149
+struct mptcp_pm_ops *mptcp_pm_find(enum mptcp_pm_type type);
155
+struct mptcp_pm_ops *mptcp_pm_find(enum mptcp_pm_type type);
150
+int mptcp_validate_path_manager(struct mptcp_pm_ops *pm);
156
+int mptcp_pm_validate(struct mptcp_pm_ops *pm);
151
+int mptcp_register_path_manager(struct mptcp_pm_ops *pm);
157
+int mptcp_pm_register(struct mptcp_pm_ops *pm);
152
+void mptcp_unregister_path_manager(struct mptcp_pm_ops *pm);
158
+void mptcp_pm_unregister(struct mptcp_pm_ops *pm);
153
+
159
+
154
void mptcp_free_local_addr_list(struct mptcp_sock *msk);
160
void mptcp_free_local_addr_list(struct mptcp_sock *msk);
155
161
156
void mptcp_event(enum mptcp_event_type type, const struct mptcp_sock *msk,
162
void mptcp_event(enum mptcp_event_type type, const struct mptcp_sock *msk,
157
--
163
--
158
2.43.0
164
2.43.0
diff view generated by jsdifflib
New patch
1
From: Geliang Tang <tanggeliang@kylinos.cn>
1
2
3
This patch defines a new path manager type: MPTCP_PM_TYPE_BPF_USERSPACE,
4
which is used for the userspace path manager implemented in BPF. This
5
path manager is also a userspace path manager, so add this new type in
6
mptcp_pm_is_userspace() too.
7
8
Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
9
---
10
net/mptcp/protocol.h | 6 +++++-
11
1 file changed, 5 insertions(+), 1 deletion(-)
12
13
diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h
14
index XXXXXXX..XXXXXXX 100644
15
--- a/net/mptcp/protocol.h
16
+++ b/net/mptcp/protocol.h
17
@@ -XXX,XX +XXX,XX @@ enum mptcp_pm_status {
18
enum mptcp_pm_type {
19
    MPTCP_PM_TYPE_KERNEL = 0,
20
    MPTCP_PM_TYPE_USERSPACE,
21
+    MPTCP_PM_TYPE_BPF_USERSPACE,
22
23
    __MPTCP_PM_TYPE_NR,
24
    __MPTCP_PM_TYPE_MAX = __MPTCP_PM_TYPE_NR - 1,
25
@@ -XXX,XX +XXX,XX @@ static inline bool mptcp_pm_should_rm_signal(struct mptcp_sock *msk)
26
27
static inline bool mptcp_pm_is_userspace(const struct mptcp_sock *msk)
28
{
29
-    return READ_ONCE(msk->pm.pm_type) == MPTCP_PM_TYPE_USERSPACE;
30
+    enum mptcp_pm_type pm_type = READ_ONCE(msk->pm.pm_type);
31
+
32
+    return (pm_type == MPTCP_PM_TYPE_USERSPACE ||
33
+        pm_type == MPTCP_PM_TYPE_BPF_USERSPACE);
34
}
35
36
static inline bool mptcp_pm_is_kernel(const struct mptcp_sock *msk)
37
--
38
2.43.0
diff view generated by jsdifflib
1
From: Geliang Tang <tanggeliang@kylinos.cn>
1
From: Geliang Tang <tanggeliang@kylinos.cn>
2
2
3
This patch defines the original userspace pm as the default path manager,
3
This patch defines the original userspace pm as the default path manager,
4
named mptcp_userspace_pm, and register it in mptcp_pm_data_init().
4
named mptcp_userspace_pm, and register it in mptcp_pm_data_init().
5
5
6
Extract address_announce() interface of the userspace PM from the handler
6
Extract address_announce() interface of the userspace PM from the handler
7
of netlink commond MPTCP_PM_CMD_ANNOUNCE mptcp_pm_nl_announce_doit(), only
7
of netlink command MPTCP_PM_CMD_ANNOUNCE mptcp_pm_nl_announce_doit(), only
8
leave the code for obtaining msk through "info" and parsing address entry
8
leave the code for obtaining msk through "info" and parsing address entry
9
in the handler.
9
in the handler.
10
10
11
Extract address_remove() interface of the userspace PM from the handler
11
Extract address_remove() interface of the userspace PM from the handler
12
of netlink commond MPTCP_PM_CMD_REMOVE mptcp_pm_nl_remove_doit(), only
12
of netlink command MPTCP_PM_CMD_REMOVE mptcp_pm_nl_remove_doit(), only
13
leave the code for parsing address id and obtaining msk through "info"
13
leave the code for parsing address id and obtaining msk through "info"
14
in the handler.
14
in the handler.
15
15
16
Extract subflow_create() interface of the userspace PM from the handler of
16
Extract subflow_create() interface of the userspace PM from the handler of
17
netlink commond MPTCP_PM_CMD_SUBFLOW_CREATE
17
netlink command MPTCP_PM_CMD_SUBFLOW_CREATE
18
18
19
    mptcp_pm_nl_subflow_create_doit(),
19
    mptcp_pm_nl_subflow_create_doit(),
20
20
21
only leave the code for obtaining msk through "info", parsing local address
21
only leave the code for obtaining msk through "info", parsing local address
22
entry and parsing remote address info in the handler.
22
entry and parsing remote address info in the handler.
23
23
24
Extract subflow_destroy() interface of the userspace PM from the handler of
24
Extract subflow_destroy() interface of the userspace PM from the handler of
25
netlink commond MPTCP_PM_CMD_SUBFLOW_DESTROY
25
netlink command MPTCP_PM_CMD_SUBFLOW_DESTROY
26
26
27
    mptcp_pm_nl_subflow_destroy_doit(),
27
    mptcp_pm_nl_subflow_destroy_doit(),
28
28
29
only leave the code for obtaining msk through "info", parsing local address
29
only leave the code for obtaining msk through "info", parsing local address
30
entry and parsing remote address info in the handler.
30
entry and parsing remote address info in the handler.
...
...
33
mptcp_userspace_pm_set_flags(), only leave the code for obtaining
33
mptcp_userspace_pm_set_flags(), only leave the code for obtaining
34
msk through "info" in this function.
34
msk through "info" in this function.
35
35
36
All fives interfaces are invoked under holding the msk socket lock.
36
All fives interfaces are invoked under holding the msk socket lock.
37
37
38
Other interfaces get_local_id() and get_flags() are only defined as
38
Extract get_local_id() interface of the userspace PM from function
39
wrappers.
39
mptcp_userspace_pm_get_local_id(), only leave the code for setting
40
new address entry in this function.
41
42
get_flags() interface of the userspace PM is defined as the same as
43
the function mptcp_userspace_pm_get_flags(), then this function
44
becomes a wrapper.
40
45
41
Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
46
Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
42
---
47
---
43
net/mptcp/pm.c | 1 +
48
net/mptcp/pm.c | 1 +
44
net/mptcp/pm_userspace.c | 261 ++++++++++++++++++++++++---------------
49
net/mptcp/pm_userspace.c | 280 +++++++++++++++++++++++----------------
45
net/mptcp/protocol.h | 1 +
50
net/mptcp/protocol.h | 1 +
46
3 files changed, 165 insertions(+), 98 deletions(-)
51
3 files changed, 171 insertions(+), 111 deletions(-)
47
52
48
diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c
53
diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c
49
index XXXXXXX..XXXXXXX 100644
54
index XXXXXXX..XXXXXXX 100644
50
--- a/net/mptcp/pm.c
55
--- a/net/mptcp/pm.c
51
+++ b/net/mptcp/pm.c
56
+++ b/net/mptcp/pm.c
52
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_data_init(struct mptcp_sock *msk)
57
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_data_init(struct mptcp_sock *msk)
53
void __init mptcp_pm_init(void)
58
void __init mptcp_pm_init(void)
54
{
59
{
55
    mptcp_pm_nl_init();
60
    mptcp_pm_nl_init();
56
+    mptcp_userspace_pm_init();
61
+    mptcp_userspace_pm_init();
57
}
62
}
63
64
/* Must be called with rcu read lock held */
58
diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c
65
diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c
59
index XXXXXXX..XXXXXXX 100644
66
index XXXXXXX..XXXXXXX 100644
60
--- a/net/mptcp/pm_userspace.c
67
--- a/net/mptcp/pm_userspace.c
61
+++ b/net/mptcp/pm_userspace.c
68
+++ b/net/mptcp/pm_userspace.c
62
@@ -XXX,XX +XXX,XX @@ mptcp_userspace_pm_lookup_addr_by_id(struct mptcp_sock *msk, unsigned int id)
69
@@ -XXX,XX +XXX,XX @@ mptcp_userspace_pm_lookup_addr_by_id(struct mptcp_sock *msk, unsigned int id)
63
    return NULL;
70
    return NULL;
64
}
71
}
65
72
66
-int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk,
73
-int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk,
67
-                 struct mptcp_pm_addr_entry *local)
74
-                 struct mptcp_addr_info *skc)
68
+static int userspace_pm_get_local_id(struct mptcp_sock *msk,
75
+static int userspace_pm_get_local_id(struct mptcp_sock *msk,
69
+                 struct mptcp_pm_addr_entry *local)
76
+                 struct mptcp_pm_addr_entry *skc)
70
{
77
{
71
    __be16 msk_sport = ((struct inet_sock *)
78
-    struct mptcp_pm_addr_entry *entry = NULL, new_entry;
72
             inet_sk((struct sock *)msk))->inet_sport;
79
-    __be16 msk_sport = ((struct inet_sock *)
80
-             inet_sk((struct sock *)msk))->inet_sport;
81
+    struct mptcp_pm_addr_entry *entry;
82
83
    spin_lock_bh(&msk->pm.lock);
84
-    entry = mptcp_userspace_pm_lookup_addr(msk, skc);
85
+    entry = mptcp_userspace_pm_lookup_addr(msk, &skc->addr);
86
    spin_unlock_bh(&msk->pm.lock);
87
    if (entry)
88
        return entry->addr.id;
89
90
+    return mptcp_userspace_pm_append_new_local_addr(msk, skc, true);
91
+}
92
+
93
+int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk,
94
+                 struct mptcp_addr_info *skc)
95
+{
96
+    __be16 msk_sport = ((struct inet_sock *)
97
+             inet_sk((struct sock *)msk))->inet_sport;
98
+    struct mptcp_pm_addr_entry new_entry;
99
+
100
    memset(&new_entry, 0, sizeof(struct mptcp_pm_addr_entry));
101
    new_entry.addr = *skc;
102
    new_entry.addr.id = 0;
73
@@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk,
103
@@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk,
74
    return mptcp_userspace_pm_append_new_local_addr(msk, local, true);
104
    if (new_entry.addr.port == msk_sport)
75
}
105
        new_entry.addr.port = 0;
76
106
77
-u8 mptcp_userspace_pm_get_flags(struct mptcp_sock *msk,
107
-    return mptcp_userspace_pm_append_new_local_addr(msk, &new_entry, true);
78
-                struct mptcp_addr_info *skc)
108
+    return userspace_pm_get_local_id(msk, &new_entry);
79
+int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk,
109
}
80
+                 struct mptcp_pm_addr_entry *local)
110
81
+{
111
-bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk,
82
+    return userspace_pm_get_local_id(msk, local);
112
-                 struct mptcp_addr_info *skc)
83
+}
113
+static bool userspace_pm_get_priority(struct mptcp_sock *msk,
84
+
114
+                 struct mptcp_addr_info *skc)
85
+static u8 userspace_pm_get_flags(struct mptcp_sock *msk,
86
+                 struct mptcp_addr_info *skc)
87
{
115
{
88
    struct mptcp_pm_addr_entry *entry;
116
    struct mptcp_pm_addr_entry *entry;
89
    u8 flags;
117
    bool backup;
90
@@ -XXX,XX +XXX,XX @@ u8 mptcp_userspace_pm_get_flags(struct mptcp_sock *msk,
118
@@ -XXX,XX +XXX,XX @@ bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk,
91
    return flags;
119
    return backup;
92
}
120
}
93
121
94
+u8 mptcp_userspace_pm_get_flags(struct mptcp_sock *msk,
122
+bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk,
95
+                struct mptcp_addr_info *skc)
123
+                 struct mptcp_addr_info *skc)
96
+{
124
+{
97
+    return userspace_pm_get_flags(msk, skc);
125
+    return userspace_pm_get_priority(msk, skc);
98
+}
126
+}
99
+
127
+
100
static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *info)
128
static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *info)
101
{
129
{
102
    struct mptcp_sock *msk;
130
    struct mptcp_sock *msk;
103
@@ -XXX,XX +XXX,XX @@ static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *in
131
@@ -XXX,XX +XXX,XX @@ static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *in
104
    return msk;
132
    return msk;
105
}
133
}
106
134
107
+static int userspace_pm_address_announce(struct mptcp_sock *msk,
135
+static int userspace_pm_address_announced(struct mptcp_sock *msk,
108
+                     struct mptcp_pm_addr_entry *local)
136
+                     struct mptcp_pm_addr_entry *local)
109
+{
137
+{
110
+    int err;
138
+    int err;
111
+
139
+
112
+    err = mptcp_userspace_pm_append_new_local_addr(msk, local, false);
140
+    err = mptcp_userspace_pm_append_new_local_addr(msk, local, false);
113
+    if (err < 0)
141
+    if (err < 0)
...
...
148
-        mptcp_pm_announce_addr(msk, &addr_val.addr, false);
176
-        mptcp_pm_announce_addr(msk, &addr_val.addr, false);
149
-        mptcp_pm_nl_addr_send_ack(msk);
177
-        mptcp_pm_nl_addr_send_ack(msk);
150
-    }
178
-    }
151
-
179
-
152
-    spin_unlock_bh(&msk->pm.lock);
180
-    spin_unlock_bh(&msk->pm.lock);
153
+    err = userspace_pm_address_announce(msk, &addr_val);
181
+    err = userspace_pm_address_announced(msk, &addr_val);
154
    release_sock(sk);
182
    release_sock(sk);
155
+    if (err)
183
+    if (err)
156
+        NL_SET_ERR_MSG_ATTR(info->extack, addr,
184
+        NL_SET_ERR_MSG_ATTR(info->extack, addr,
157
+                 "did not match address and id");
185
+                 "did not match address and id");
158
186
159
-    err = 0;
187
-    err = 0;
160
announce_err:
188
announce_err:
161
    sock_put(sk);
189
    sock_put(sk);
162
    return err;
190
    return err;
191
@@ -XXX,XX +XXX,XX @@ static int mptcp_userspace_pm_remove_id_zero_address(struct mptcp_sock *msk)
192
{
193
    struct mptcp_rm_list list = { .nr = 0 };
194
    struct mptcp_subflow_context *subflow;
195
-    struct sock *sk = (struct sock *)msk;
196
    bool has_id_0 = false;
197
-    int err = -EINVAL;
198
199
-    lock_sock(sk);
200
    mptcp_for_each_subflow(msk, subflow) {
201
        if (READ_ONCE(subflow->local_id) == 0) {
202
            has_id_0 = true;
203
@@ -XXX,XX +XXX,XX @@ static int mptcp_userspace_pm_remove_id_zero_address(struct mptcp_sock *msk)
204
        }
205
    }
206
    if (!has_id_0)
207
-        goto remove_err;
208
+        return -EINVAL;
209
210
    list.ids[list.nr++] = 0;
211
212
@@ -XXX,XX +XXX,XX @@ static int mptcp_userspace_pm_remove_id_zero_address(struct mptcp_sock *msk)
213
    mptcp_pm_remove_addr(msk, &list);
214
    spin_unlock_bh(&msk->pm.lock);
215
216
-    err = 0;
217
-
218
-remove_err:
219
-    release_sock(sk);
220
-    return err;
221
+    return 0;
222
}
223
224
void mptcp_pm_remove_addr_entry(struct mptcp_sock *msk,
163
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_remove_addr_entry(struct mptcp_sock *msk,
225
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_remove_addr_entry(struct mptcp_sock *msk,
164
    spin_unlock_bh(&msk->pm.lock);
226
    spin_unlock_bh(&msk->pm.lock);
165
}
227
}
166
228
167
+static int userspace_pm_address_remove(struct mptcp_sock *msk, u8 id)
229
+static int userspace_pm_address_removed(struct mptcp_sock *msk, u8 id)
168
+{
230
+{
169
+    struct mptcp_pm_addr_entry *entry;
231
+    struct mptcp_pm_addr_entry *entry;
170
+
232
+
171
+    if (id == 0)
233
+    if (id == 0)
172
+        return mptcp_userspace_pm_remove_id_zero_address(msk);
234
+        return mptcp_userspace_pm_remove_id_zero_address(msk);
...
...
193
-    struct mptcp_pm_addr_entry *match;
255
-    struct mptcp_pm_addr_entry *match;
194
    struct mptcp_sock *msk;
256
    struct mptcp_sock *msk;
195
    struct nlattr *id;
257
    struct nlattr *id;
196
    int err = -EINVAL;
258
    int err = -EINVAL;
197
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info)
259
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info)
260
198
    sk = (struct sock *)msk;
261
    sk = (struct sock *)msk;
199
262
200
    lock_sock(sk);
201
-
202
-    if (id_val == 0) {
263
-    if (id_val == 0) {
203
-        err = mptcp_userspace_pm_remove_id_zero_address(msk);
264
-        err = mptcp_userspace_pm_remove_id_zero_address(msk);
204
-        release_sock(sk);
205
-        goto out;
265
-        goto out;
206
-    }
266
-    }
267
-
268
    lock_sock(sk);
207
-
269
-
208
-    spin_lock_bh(&msk->pm.lock);
270
-    spin_lock_bh(&msk->pm.lock);
209
-    match = mptcp_userspace_pm_lookup_addr_by_id(msk, id_val);
271
-    match = mptcp_userspace_pm_lookup_addr_by_id(msk, id_val);
210
-    if (!match) {
272
-    if (!match) {
211
-        spin_unlock_bh(&msk->pm.lock);
273
-        spin_unlock_bh(&msk->pm.lock);
...
...
216
-    list_del_rcu(&match->list);
278
-    list_del_rcu(&match->list);
217
-    spin_unlock_bh(&msk->pm.lock);
279
-    spin_unlock_bh(&msk->pm.lock);
218
-
280
-
219
-    mptcp_pm_remove_addr_entry(msk, match);
281
-    mptcp_pm_remove_addr_entry(msk, match);
220
-
282
-
221
+    err = userspace_pm_address_remove(msk, id_val);
283
+    err = userspace_pm_address_removed(msk, id_val);
222
    release_sock(sk);
284
    release_sock(sk);
223
-
285
-
224
-    sock_kfree_s(sk, match, sizeof(*match));
286
-    sock_kfree_s(sk, match, sizeof(*match));
225
-
287
-
226
-    err = 0;
288
-    err = 0;
...
...
230
                    "address with id %u not found",
292
                    "address with id %u not found",
231
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info)
293
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info)
232
    return err;
294
    return err;
233
}
295
}
234
296
235
+static int userspace_pm_subflow_create(struct mptcp_sock *msk,
297
+static int userspace_pm_subflow_established(struct mptcp_sock *msk,
236
+                 struct mptcp_pm_addr_entry *entry,
298
+                     struct mptcp_pm_addr_entry *entry,
237
+                 struct mptcp_addr_info *remote)
299
+                     struct mptcp_addr_info *remote)
238
+{
300
+{
239
+    struct sock *sk = (struct sock *)msk;
301
+    struct sock *sk = (struct sock *)msk;
240
+    struct mptcp_pm_local local;
302
+    struct mptcp_pm_local local;
241
+    int err;
303
+    int err;
242
+
304
+
...
...
254
+        mptcp_userspace_pm_delete_local_addr(msk, entry);
316
+        mptcp_userspace_pm_delete_local_addr(msk, entry);
255
+    else
317
+    else
256
+        msk->pm.subflows++;
318
+        msk->pm.subflows++;
257
+    spin_unlock_bh(&msk->pm.lock);
319
+    spin_unlock_bh(&msk->pm.lock);
258
+
320
+
259
+    return 0;
321
+    return err;
260
+}
322
+}
261
+
323
+
262
int mptcp_pm_nl_subflow_create_doit(struct sk_buff *skb, struct genl_info *info)
324
int mptcp_pm_nl_subflow_create_doit(struct sk_buff *skb, struct genl_info *info)
263
{
325
{
264
    struct mptcp_pm_addr_entry entry = { 0 };
326
    struct mptcp_pm_addr_entry entry = { 0 };
...
...
283
-    local.flags = entry.flags;
345
-    local.flags = entry.flags;
284
-    local.ifindex = entry.ifindex;
346
-    local.ifindex = entry.ifindex;
285
-
347
-
286
    lock_sock(sk);
348
    lock_sock(sk);
287
-    err = __mptcp_subflow_connect(sk, &local, &addr_r);
349
-    err = __mptcp_subflow_connect(sk, &local, &addr_r);
288
+    err = userspace_pm_subflow_create(msk, &entry, &addr_r);
350
+    err = userspace_pm_subflow_established(msk, &entry, &addr_r);
289
    release_sock(sk);
351
    release_sock(sk);
290
352
291
    if (err)
353
    if (err)
292
        GENL_SET_ERR_MSG_FMT(info, "connect error: %d", err);
354
        GENL_SET_ERR_MSG_FMT(info, "connect error: %d", err);
293
355
...
...
303
    return err;
365
    return err;
304
@@ -XXX,XX +XXX,XX @@ static struct sock *mptcp_nl_find_ssk(struct mptcp_sock *msk,
366
@@ -XXX,XX +XXX,XX @@ static struct sock *mptcp_nl_find_ssk(struct mptcp_sock *msk,
305
    return NULL;
367
    return NULL;
306
}
368
}
307
369
308
+static int userspace_pm_subflow_destroy(struct mptcp_sock *msk,
370
+static int userspace_pm_subflow_closed(struct mptcp_sock *msk,
309
+                    struct mptcp_pm_addr_entry *local,
371
+                 struct mptcp_pm_addr_entry *local,
310
+                    struct mptcp_addr_info *remote)
372
+                 struct mptcp_addr_info *remote)
311
+{
373
+{
312
+    struct sock *ssk, *sk = (struct sock *)msk;
374
+    struct sock *ssk, *sk = (struct sock *)msk;
313
+
375
+
314
+    ssk = mptcp_nl_find_ssk(msk, &local->addr, remote);
376
+    ssk = mptcp_nl_find_ssk(msk, &local->addr, remote);
315
+    if (!ssk)
377
+    if (!ssk)
...
...
353
-    spin_unlock_bh(&msk->pm.lock);
415
-    spin_unlock_bh(&msk->pm.lock);
354
-    mptcp_subflow_shutdown(sk, ssk, RCV_SHUTDOWN | SEND_SHUTDOWN);
416
-    mptcp_subflow_shutdown(sk, ssk, RCV_SHUTDOWN | SEND_SHUTDOWN);
355
-    mptcp_close_ssk(sk, ssk, mptcp_subflow_ctx(ssk));
417
-    mptcp_close_ssk(sk, ssk, mptcp_subflow_ctx(ssk));
356
-    MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_RMSUBFLOW);
418
-    MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_RMSUBFLOW);
357
-release_sock:
419
-release_sock:
358
+    err = userspace_pm_subflow_destroy(msk, &addr_l, &addr_r);
420
+    err = userspace_pm_subflow_closed(msk, &addr_l, &addr_r);
359
    release_sock(sk);
421
    release_sock(sk);
360
+    if (err)
422
+    if (err)
361
+        GENL_SET_ERR_MSG(info, "subflow not found");
423
+        GENL_SET_ERR_MSG(info, "subflow not found");
362
424
363
destroy_err:
425
destroy_err:
364
    sock_put(sk);
426
    sock_put(sk);
365
    return err;
427
    return err;
366
}
428
}
367
429
368
+static int userspace_pm_set_flags(struct mptcp_sock *msk,
430
+static int userspace_pm_set_priority(struct mptcp_sock *msk,
369
+                 struct mptcp_pm_addr_entry *local,
431
+                 struct mptcp_pm_addr_entry *local,
370
+                 struct mptcp_addr_info *remote)
432
+                 struct mptcp_addr_info *remote)
371
+{
433
+{
372
+    struct mptcp_pm_addr_entry *entry;
434
+    struct mptcp_pm_addr_entry *entry;
373
+    u8 bkup = 0;
435
+    u8 bkup = 0;
374
+
436
+
375
+    if (local->flags & MPTCP_PM_ADDR_FLAG_BACKUP)
437
+    if (local->flags & MPTCP_PM_ADDR_FLAG_BACKUP)
...
...
418
-    }
480
-    }
419
-    spin_unlock_bh(&msk->pm.lock);
481
-    spin_unlock_bh(&msk->pm.lock);
420
-
482
-
421
    lock_sock(sk);
483
    lock_sock(sk);
422
-    ret = mptcp_pm_nl_mp_prio_send_ack(msk, &local->addr, &rem, bkup);
484
-    ret = mptcp_pm_nl_mp_prio_send_ack(msk, &local->addr, &rem, bkup);
423
+    ret = userspace_pm_set_flags(msk, local, &rem);
485
+    ret = userspace_pm_set_priority(msk, local, &rem);
424
    release_sock(sk);
486
    release_sock(sk);
425
487
426
    /* mptcp_pm_nl_mp_prio_send_ack() only fails in one case */
488
    /* mptcp_pm_nl_mp_prio_send_ack() only fails in one case */
427
@@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_addr(u8 id, struct mptcp_pm_addr_entry *addr,
489
@@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_addr(u8 id, struct mptcp_pm_addr_entry *addr,
490
    sock_put(sk);
428
    return ret;
491
    return ret;
429
}
492
}
430
493
+
431
+static struct mptcp_pm_ops mptcp_userspace_pm = {
494
+static struct mptcp_pm_ops mptcp_userspace_pm = {
432
+    .address_announce    = userspace_pm_address_announce,
495
+    .address_announced    = userspace_pm_address_announced,
433
+    .address_remove        = userspace_pm_address_remove,
496
+    .address_removed    = userspace_pm_address_removed,
434
+    .subflow_create        = userspace_pm_subflow_create,
497
+    .subflow_established    = userspace_pm_subflow_established,
435
+    .subflow_destroy    = userspace_pm_subflow_destroy,
498
+    .subflow_closed        = userspace_pm_subflow_closed,
436
+    .get_local_id        = userspace_pm_get_local_id,
499
+    .get_local_id        = userspace_pm_get_local_id,
437
+    .get_flags        = userspace_pm_get_flags,
500
+    .get_priority        = userspace_pm_get_priority,
438
+    .set_flags        = userspace_pm_set_flags,
501
+    .set_priority        = userspace_pm_set_priority,
439
+    .type            = MPTCP_PM_TYPE_USERSPACE,
502
+    .type            = MPTCP_PM_TYPE_USERSPACE,
440
+    .owner            = THIS_MODULE,
503
+    .owner            = THIS_MODULE,
441
+};
504
+};
442
+
505
+
443
/* Must be called with rcu read lock held */
444
struct mptcp_pm_ops *mptcp_pm_find(enum mptcp_pm_type type)
445
{
446
@@ -XXX,XX +XXX,XX @@ int mptcp_register_path_manager(struct mptcp_pm_ops *pm)
447
448
void mptcp_unregister_path_manager(struct mptcp_pm_ops *pm)
449
{
450
+    if (pm == &mptcp_userspace_pm)
451
+        return;
452
+
453
    spin_lock(&mptcp_pm_list_lock);
454
    list_del_rcu(&pm->list);
455
    spin_unlock(&mptcp_pm_list_lock);
456
}
457
+
458
+void __init mptcp_userspace_pm_init(void)
506
+void __init mptcp_userspace_pm_init(void)
459
+{
507
+{
460
+    mptcp_register_path_manager(&mptcp_userspace_pm);
508
+    mptcp_pm_register(&mptcp_userspace_pm);
461
+}
509
+}
462
diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h
510
diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h
463
index XXXXXXX..XXXXXXX 100644
511
index XXXXXXX..XXXXXXX 100644
464
--- a/net/mptcp/protocol.h
512
--- a/net/mptcp/protocol.h
465
+++ b/net/mptcp/protocol.h
513
+++ b/net/mptcp/protocol.h
...
...
diff view generated by jsdifflib
1
From: Geliang Tang <tanggeliang@kylinos.cn>
1
From: Geliang Tang <tanggeliang@kylinos.cn>
2
2
3
Add a struct mptcp_pm_ops pointer "ops" in struct mptcp_pm_data, and two
3
Add a struct mptcp_pm_ops pointer "ops" in struct mptcp_pm_data, and two
4
functions mptcp_init_pm() and mptcp_release_pm(), to set and release this
4
functions mptcp_pm_initialize() and mptcp_pm_release(), to set and release
5
pointer. mptcp_init_pm() is invoked in mptcp_pm_data_reset(), while
5
this pointer. mptcp_pm_initialize() is invoked in mptcp_pm_data_reset(),
6
mptcp_release_pm() is invoked in __mptcp_destroy_sock().
6
while mptcp_pm_release() is invoked in __mptcp_destroy_sock().
7
7
8
In this way, different userspace path managers can be initialized through
8
In this way, different userspace path managers can be initialized through
9
the pm_type sysctl, and then called into their respective interfaces
9
the pm_type sysctl, and then called into their respective interfaces
10
through "ops" of "msk->pm".
10
through "ops" of "msk->pm".
11
11
12
Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
12
Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
13
---
13
---
14
net/mptcp/pm.c | 7 +++++
14
net/mptcp/pm.c | 49 ++++++++++++++++++++++++++++++++++++++--
15
net/mptcp/pm_userspace.c | 58 +++++++++++++++++++++++++++++++++++-----
15
net/mptcp/pm_netlink.c | 11 ++++++++-
16
net/mptcp/protocol.c | 1 +
16
net/mptcp/pm_userspace.c | 28 +++++++++++++++++------
17
net/mptcp/protocol.c | 10 +++++++-
17
net/mptcp/protocol.h | 3 +++
18
net/mptcp/protocol.h | 3 +++
18
4 files changed, 62 insertions(+), 7 deletions(-)
19
5 files changed, 90 insertions(+), 11 deletions(-)
19
20
20
diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c
21
diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c
21
index XXXXXXX..XXXXXXX 100644
22
index XXXXXXX..XXXXXXX 100644
22
--- a/net/mptcp/pm.c
23
--- a/net/mptcp/pm.c
23
+++ b/net/mptcp/pm.c
24
+++ b/net/mptcp/pm.c
24
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_data_reset(struct mptcp_sock *msk)
25
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_new_connection(struct mptcp_sock *msk, const struct sock *ssk, int
26
27
    WRITE_ONCE(pm->server_side, server_side);
28
    mptcp_event(MPTCP_EVENT_CREATED, msk, ssk, GFP_ATOMIC);
29
+
30
+    if (pm->ops && pm->ops->created)
31
+        pm->ops->created(msk);
32
}
33
34
bool mptcp_pm_allow_new_subflow(struct mptcp_sock *msk)
35
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_fully_established(struct mptcp_sock *msk, const struct sock *ssk)
36
    msk->pm.status |= BIT(MPTCP_PM_ALREADY_ESTABLISHED);
37
    spin_unlock_bh(&pm->lock);
38
39
-    if (announce)
40
+    if (announce) {
41
        mptcp_event(MPTCP_EVENT_ESTABLISHED, msk, ssk, GFP_ATOMIC);
42
+
43
+        if (pm->ops && pm->ops->established)
44
+            pm->ops->established(msk);
45
+    }
46
}
47
48
void mptcp_pm_connection_closed(struct mptcp_sock *msk)
25
{
49
{
26
    u8 pm_type = mptcp_get_pm_type(sock_net((struct sock *)msk));
50
    pr_debug("msk=%p\n", msk);
27
    struct mptcp_pm_data *pm = &msk->pm;
51
28
+    int ret;
52
-    if (msk->token)
29
53
+    if (msk->token) {
30
    pm->add_addr_signaled = 0;
54
        mptcp_event(MPTCP_EVENT_CLOSED, msk, NULL, GFP_KERNEL);
31
    pm->add_addr_accepted = 0;
55
+
56
+        if (msk->pm.ops && msk->pm.ops->closed)
57
+            msk->pm.ops->closed(msk);
58
+    }
59
}
60
61
void mptcp_pm_subflow_established(struct mptcp_sock *msk)
32
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_data_reset(struct mptcp_sock *msk)
62
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_data_reset(struct mptcp_sock *msk)
33
        WRITE_ONCE(pm->work_pending, 0);
63
        WRITE_ONCE(pm->work_pending, 0);
34
        WRITE_ONCE(pm->accept_addr, 0);
64
        WRITE_ONCE(pm->accept_addr, 0);
35
        WRITE_ONCE(pm->accept_subflow, 0);
65
        WRITE_ONCE(pm->accept_subflow, 0);
36
+
66
+
37
+        rcu_read_lock();
67
+        rcu_read_lock();
38
+        ret = mptcp_init_pm(msk, mptcp_pm_find(pm_type));
68
+        mptcp_pm_initialize(msk, mptcp_pm_find(pm_type));
39
+        rcu_read_unlock();
69
+        rcu_read_unlock();
40
+        if (ret)
70
    }
41
+            return;
42
    }
43
71
44
    WRITE_ONCE(pm->addr_signal, 0);
72
    WRITE_ONCE(pm->addr_signal, 0);
73
@@ -XXX,XX +XXX,XX @@ void mptcp_pm_unregister(struct mptcp_pm_ops *pm)
74
    list_del_rcu(&pm->list);
75
    spin_unlock(&mptcp_pm_list_lock);
76
}
77
+
78
+int mptcp_pm_initialize(struct mptcp_sock *msk, struct mptcp_pm_ops *pm)
79
+{
80
+    if (!pm)
81
+        return -EINVAL;
82
+
83
+    if (!bpf_try_module_get(pm, pm->owner))
84
+        return -EBUSY;
85
+
86
+    msk->pm.ops = pm;
87
+    if (msk->pm.ops->init)
88
+        msk->pm.ops->init(msk);
89
+
90
+    pr_debug("userspace_pm type %u initialized\n", msk->pm.ops->type);
91
+    return 0;
92
+}
93
+
94
+void mptcp_pm_release(struct mptcp_sock *msk)
95
+{
96
+    struct mptcp_pm_ops *pm = msk->pm.ops;
97
+
98
+    if (!pm)
99
+        return;
100
+
101
+    msk->pm.ops = NULL;
102
+    if (pm->release)
103
+        pm->release(msk);
104
+
105
+    bpf_module_put(pm, pm->owner);
106
+}
107
diff --git a/net/mptcp/pm_netlink.c b/net/mptcp/pm_netlink.c
108
index XXXXXXX..XXXXXXX 100644
109
--- a/net/mptcp/pm_netlink.c
110
+++ b/net/mptcp/pm_netlink.c
111
@@ -XXX,XX +XXX,XX @@ static int mptcp_pm_nl_create_listen_socket(struct sock *sk,
112
    int addrlen = sizeof(struct sockaddr_in);
113
    struct sockaddr_storage addr;
114
    struct sock *newsk, *ssk;
115
+    struct mptcp_sock *msk;
116
    int backlog = 1024;
117
    int err;
118
119
@@ -XXX,XX +XXX,XX @@ static int mptcp_pm_nl_create_listen_socket(struct sock *sk,
120
                 is_ipv6 ? "msk_lock-AF_INET6" : "msk_lock-AF_INET",
121
                 &mptcp_keys[is_ipv6]);
122
123
+    msk = mptcp_sk(newsk);
124
    lock_sock(newsk);
125
-    ssk = __mptcp_nmpc_sk(mptcp_sk(newsk));
126
+    ssk = __mptcp_nmpc_sk(msk);
127
    release_sock(newsk);
128
    if (IS_ERR(ssk))
129
        return PTR_ERR(ssk);
130
@@ -XXX,XX +XXX,XX @@ static int mptcp_pm_nl_create_listen_socket(struct sock *sk,
131
    if (!err)
132
        mptcp_event_pm_listener(ssk, MPTCP_EVENT_LISTENER_CREATED);
133
    release_sock(ssk);
134
+
135
+    if (!err) {
136
+        lock_sock(newsk);
137
+        if (msk->pm.ops && msk->pm.ops->listener_created)
138
+            msk->pm.ops->listener_created(msk);
139
+        release_sock(newsk);
140
+    }
141
    return err;
142
}
143
45
diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c
144
diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c
46
index XXXXXXX..XXXXXXX 100644
145
index XXXXXXX..XXXXXXX 100644
47
--- a/net/mptcp/pm_userspace.c
146
--- a/net/mptcp/pm_userspace.c
48
+++ b/net/mptcp/pm_userspace.c
147
+++ b/net/mptcp/pm_userspace.c
49
@@ -XXX,XX +XXX,XX @@ static int userspace_pm_get_local_id(struct mptcp_sock *msk,
148
@@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk,
50
int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk,
149
    if (new_entry.addr.port == msk_sport)
51
                 struct mptcp_pm_addr_entry *local)
150
        new_entry.addr.port = 0;
151
152
-    return userspace_pm_get_local_id(msk, &new_entry);
153
+    return msk->pm.ops && msk->pm.ops->get_local_id ?
154
+     msk->pm.ops->get_local_id(msk, &new_entry) :
155
+     userspace_pm_get_local_id(msk, &new_entry);
156
}
157
158
static bool userspace_pm_get_priority(struct mptcp_sock *msk,
159
@@ -XXX,XX +XXX,XX @@ static bool userspace_pm_get_priority(struct mptcp_sock *msk,
160
bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk,
161
                 struct mptcp_addr_info *skc)
52
{
162
{
53
-    return userspace_pm_get_local_id(msk, local);
163
-    return userspace_pm_get_priority(msk, skc);
54
+    return msk->pm.ops->get_local_id ?
164
+    return msk->pm.ops && msk->pm.ops->get_priority ?
55
+     msk->pm.ops->get_local_id(msk, local) :
165
+     msk->pm.ops->get_priority(msk, skc) :
56
+     userspace_pm_get_local_id(msk, local);
166
+     userspace_pm_get_priority(msk, skc);
57
}
167
}
58
59
static u8 userspace_pm_get_flags(struct mptcp_sock *msk,
60
@@ -XXX,XX +XXX,XX @@ static u8 userspace_pm_get_flags(struct mptcp_sock *msk,
61
u8 mptcp_userspace_pm_get_flags(struct mptcp_sock *msk,
62
                struct mptcp_addr_info *skc)
63
{
64
-    return userspace_pm_get_flags(msk, skc);
65
+    return msk->pm.ops->get_flags ?
66
+     msk->pm.ops->get_flags(msk, skc) :
67
+     userspace_pm_get_flags(msk, skc);
68
}
69
168
70
static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *info)
169
static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *info)
71
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_announce_doit(struct sk_buff *skb, struct genl_info *info)
170
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_announce_doit(struct sk_buff *skb, struct genl_info *info)
72
    }
171
    }
73
172
74
    lock_sock(sk);
173
    lock_sock(sk);
75
-    err = userspace_pm_address_announce(msk, &addr_val);
174
-    err = userspace_pm_address_announced(msk, &addr_val);
76
+    err = msk->pm.ops->address_announce ?
175
+    err = msk->pm.ops && msk->pm.ops->address_announced ?
77
+     msk->pm.ops->address_announce(msk, &addr_val) :
176
+     msk->pm.ops->address_announced(msk, &addr_val) :
78
+     userspace_pm_address_announce(msk, &addr_val);
177
+     userspace_pm_address_announced(msk, &addr_val);
79
    release_sock(sk);
178
    release_sock(sk);
80
    if (err)
179
    if (err)
81
        NL_SET_ERR_MSG_ATTR(info->extack, addr,
180
        NL_SET_ERR_MSG_ATTR(info->extack, addr,
82
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info)
181
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info)
83
    sk = (struct sock *)msk;
182
    sk = (struct sock *)msk;
84
183
85
    lock_sock(sk);
184
    lock_sock(sk);
86
-    err = userspace_pm_address_remove(msk, id_val);
185
-    err = userspace_pm_address_removed(msk, id_val);
87
+    err = msk->pm.ops->address_remove ?
186
+    err = msk->pm.ops && msk->pm.ops->address_removed ?
88
+     msk->pm.ops->address_remove(msk, id_val) :
187
+     msk->pm.ops->address_removed(msk, id_val) :
89
+     userspace_pm_address_remove(msk, id_val);
188
+     userspace_pm_address_removed(msk, id_val);
90
    release_sock(sk);
189
    release_sock(sk);
91
    if (err)
190
    if (err)
92
        NL_SET_ERR_MSG_ATTR_FMT(info->extack, id,
191
        NL_SET_ERR_MSG_ATTR_FMT(info->extack, id,
93
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_subflow_create_doit(struct sk_buff *skb, struct genl_info *info)
192
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_subflow_create_doit(struct sk_buff *skb, struct genl_info *info)
94
    }
193
    }
95
194
96
    lock_sock(sk);
195
    lock_sock(sk);
97
-    err = userspace_pm_subflow_create(msk, &entry, &addr_r);
196
-    err = userspace_pm_subflow_established(msk, &entry, &addr_r);
98
+    err = msk->pm.ops->subflow_create ?
197
+    err = msk->pm.ops && msk->pm.ops->subflow_established ?
99
+     msk->pm.ops->subflow_create(msk, &entry, &addr_r) :
198
+     msk->pm.ops->subflow_established(msk, &entry, &addr_r) :
100
+     userspace_pm_subflow_create(msk, &entry, &addr_r);
199
+     userspace_pm_subflow_established(msk, &entry, &addr_r);
101
    release_sock(sk);
200
    release_sock(sk);
102
201
103
    if (err)
202
    if (err)
104
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_subflow_destroy_doit(struct sk_buff *skb, struct genl_info *info
203
@@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_subflow_destroy_doit(struct sk_buff *skb, struct genl_info *info
105
    }
204
    }
106
205
107
    lock_sock(sk);
206
    lock_sock(sk);
108
-    err = userspace_pm_subflow_destroy(msk, &addr_l, &addr_r);
207
-    err = userspace_pm_subflow_closed(msk, &addr_l, &addr_r);
109
+    err = msk->pm.ops->subflow_destroy ?
208
+    err = msk->pm.ops && msk->pm.ops->subflow_closed ?
110
+     msk->pm.ops->subflow_destroy(msk, &addr_l, &addr_r) :
209
+     msk->pm.ops->subflow_closed(msk, &addr_l, &addr_r) :
111
+     userspace_pm_subflow_destroy(msk, &addr_l, &addr_r);
210
+     userspace_pm_subflow_closed(msk, &addr_l, &addr_r);
112
    release_sock(sk);
211
    release_sock(sk);
113
    if (err)
212
    if (err)
114
        GENL_SET_ERR_MSG(info, "subflow not found");
213
        GENL_SET_ERR_MSG(info, "subflow not found");
115
@@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_set_flags(struct mptcp_pm_addr_entry *local,
214
@@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_set_flags(struct mptcp_pm_addr_entry *local,
116
    }
215
    }
117
216
118
    lock_sock(sk);
217
    lock_sock(sk);
119
-    ret = userspace_pm_set_flags(msk, local, &rem);
218
-    ret = userspace_pm_set_priority(msk, local, &rem);
120
+    ret = msk->pm.ops->set_flags ?
219
+    ret = msk->pm.ops && msk->pm.ops->set_priority ?
121
+     msk->pm.ops->set_flags(msk, local, &rem) :
220
+     msk->pm.ops->set_priority(msk, local, &rem) :
122
+     userspace_pm_set_flags(msk, local, &rem);
221
+     userspace_pm_set_priority(msk, local, &rem);
123
    release_sock(sk);
222
    release_sock(sk);
124
223
125
    /* mptcp_pm_nl_mp_prio_send_ack() only fails in one case */
224
    /* mptcp_pm_nl_mp_prio_send_ack() only fails in one case */
126
@@ -XXX,XX +XXX,XX @@ void mptcp_unregister_path_manager(struct mptcp_pm_ops *pm)
127
    spin_unlock(&mptcp_pm_list_lock);
128
}
129
130
+int mptcp_init_pm(struct mptcp_sock *msk, struct mptcp_pm_ops *pm)
131
+{
132
+    if (!pm)
133
+        pm = &mptcp_userspace_pm;
134
+
135
+    if (!bpf_try_module_get(pm, pm->owner))
136
+        return -EBUSY;
137
+
138
+    msk->pm.ops = pm;
139
+    if (msk->pm.ops->init)
140
+        msk->pm.ops->init(msk);
141
+
142
+    pr_debug("userspace_pm type %u initialized\n", msk->pm.ops->type);
143
+    return 0;
144
+}
145
+
146
+void mptcp_release_pm(struct mptcp_sock *msk)
147
+{
148
+    struct mptcp_pm_ops *pm = msk->pm.ops;
149
+
150
+    if (!pm)
151
+        return;
152
+
153
+    msk->pm.ops = NULL;
154
+    if (pm->release)
155
+        pm->release(msk);
156
+
157
+    bpf_module_put(pm, pm->owner);
158
+}
159
+
160
void __init mptcp_userspace_pm_init(void)
161
{
162
    mptcp_register_path_manager(&mptcp_userspace_pm);
163
diff --git a/net/mptcp/protocol.c b/net/mptcp/protocol.c
225
diff --git a/net/mptcp/protocol.c b/net/mptcp/protocol.c
164
index XXXXXXX..XXXXXXX 100644
226
index XXXXXXX..XXXXXXX 100644
165
--- a/net/mptcp/protocol.c
227
--- a/net/mptcp/protocol.c
166
+++ b/net/mptcp/protocol.c
228
+++ b/net/mptcp/protocol.c
167
@@ -XXX,XX +XXX,XX @@ static void __mptcp_destroy_sock(struct sock *sk)
229
@@ -XXX,XX +XXX,XX @@ static void __mptcp_destroy_sock(struct sock *sk)
168
    sk_stop_timer(sk, &sk->sk_timer);
230
    sk_stop_timer(sk, &sk->sk_timer);
169
    msk->pm.status = 0;
231
    msk->pm.status = 0;
170
    mptcp_release_sched(msk);
232
    mptcp_release_sched(msk);
171
+    mptcp_release_pm(msk);
233
+    mptcp_pm_release(msk);
172
234
173
    sk->sk_prot->destroy(sk);
235
    sk->sk_prot->destroy(sk);
174
236
237
@@ -XXX,XX +XXX,XX @@ static __poll_t mptcp_check_readable(struct sock *sk)
238
239
static void mptcp_check_listen_stop(struct sock *sk)
240
{
241
+    struct mptcp_sock *msk = mptcp_sk(sk);
242
    struct sock *ssk;
243
244
    if (inet_sk_state_load(sk) != TCP_LISTEN)
245
        return;
246
247
    sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
248
-    ssk = mptcp_sk(sk)->first;
249
+    ssk = msk->first;
250
    if (WARN_ON_ONCE(!ssk || inet_sk_state_load(ssk) != TCP_LISTEN))
251
        return;
252
253
@@ -XXX,XX +XXX,XX @@ static void mptcp_check_listen_stop(struct sock *sk)
254
    inet_csk_listen_stop(ssk);
255
    mptcp_event_pm_listener(ssk, MPTCP_EVENT_LISTENER_CLOSED);
256
    release_sock(ssk);
257
+
258
+    if (msk->pm.ops && msk->pm.ops->listener_closed)
259
+        msk->pm.ops->listener_closed(msk);
260
}
261
262
bool __mptcp_close(struct sock *sk, long timeout)
263
@@ -XXX,XX +XXX,XX @@ static int mptcp_listen(struct socket *sock, int backlog)
264
        sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
265
        mptcp_copy_inaddrs(sk, ssk);
266
        mptcp_event_pm_listener(ssk, MPTCP_EVENT_LISTENER_CREATED);
267
+
268
+        if (msk->pm.ops && msk->pm.ops->listener_created)
269
+            msk->pm.ops->listener_created(msk);
270
    }
271
272
unlock:
175
diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h
273
diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h
176
index XXXXXXX..XXXXXXX 100644
274
index XXXXXXX..XXXXXXX 100644
177
--- a/net/mptcp/protocol.h
275
--- a/net/mptcp/protocol.h
178
+++ b/net/mptcp/protocol.h
276
+++ b/net/mptcp/protocol.h
179
@@ -XXX,XX +XXX,XX @@ struct mptcp_pm_data {
277
@@ -XXX,XX +XXX,XX @@ struct mptcp_pm_data {
...
...
183
+    struct mptcp_pm_ops *ops;
281
+    struct mptcp_pm_ops *ops;
184
282
185
    spinlock_t    lock;        /*protects the whole PM data */
283
    spinlock_t    lock;        /*protects the whole PM data */
186
284
187
@@ -XXX,XX +XXX,XX @@ struct mptcp_pm_ops *mptcp_pm_find(enum mptcp_pm_type type);
285
@@ -XXX,XX +XXX,XX @@ struct mptcp_pm_ops *mptcp_pm_find(enum mptcp_pm_type type);
188
int mptcp_validate_path_manager(struct mptcp_pm_ops *pm);
286
int mptcp_pm_validate(struct mptcp_pm_ops *pm);
189
int mptcp_register_path_manager(struct mptcp_pm_ops *pm);
287
int mptcp_pm_register(struct mptcp_pm_ops *pm);
190
void mptcp_unregister_path_manager(struct mptcp_pm_ops *pm);
288
void mptcp_pm_unregister(struct mptcp_pm_ops *pm);
191
+int mptcp_init_pm(struct mptcp_sock *msk, struct mptcp_pm_ops *pm);
289
+int mptcp_pm_initialize(struct mptcp_sock *msk, struct mptcp_pm_ops *pm);
192
+void mptcp_release_pm(struct mptcp_sock *msk);
290
+void mptcp_pm_release(struct mptcp_sock *msk);
193
291
194
void mptcp_free_local_addr_list(struct mptcp_sock *msk);
292
void mptcp_free_local_addr_list(struct mptcp_sock *msk);
195
293
196
--
294
--
197
2.43.0
295
2.43.0
diff view generated by jsdifflib