1 | From: Geliang Tang <tanggeliang@kylinos.cn> | 1 | From: Geliang Tang <tanggeliang@kylinos.cn> |
---|---|---|---|
2 | |||
3 | v5: | ||
4 | - use "struct mptcp_pm_param *param" as unified parameters for all | ||
5 | interfaces. | ||
6 | - register in-kernel mptcp_pm_ops too. | ||
7 | - only implement two interfaces "get_local_id" and "get_priority" in | ||
8 | this set. | ||
2 | 9 | ||
3 | v4: | 10 | v4: |
4 | - include a new patch "define BPF path manager type". | 11 | - include a new patch "define BPF path manager type". |
5 | 12 | ||
6 | - add new interfaces: | 13 | - add new interfaces: |
... | ... | ||
30 | - update get_local_id interface in patch 2. | 37 | - update get_local_id interface in patch 2. |
31 | 38 | ||
32 | get_addr() and dump_addr() interfaces of BPF userspace pm are dropped | 39 | get_addr() and dump_addr() interfaces of BPF userspace pm are dropped |
33 | as Matt suggested. | 40 | as Matt suggested. |
34 | 41 | ||
35 | In order to implement BPF userspace path manager, it is necessary to | 42 | In order to implement BPF path manager, it's necessary to unify the |
36 | unify the interfaces of the path manager. This set contains some | 43 | interfaces of the path manager. This set contains some cleanups and |
37 | cleanups and refactoring to unify the interfaces in kernel space. | 44 | refactoring to unify the interfaces in kernel space. Finally, define |
38 | Finally, define a struct mptcp_pm_ops for a userspace path manager | 45 | a struct mptcp_pm_ops for a path manager. |
39 | like this: | ||
40 | 46 | ||
41 | struct mptcp_pm_ops { | 47 | Geliang Tang (9): |
42 | int (*created)(struct mptcp_sock *msk); | 48 | mptcp: pm: use addr entry for get_local_id |
43 | int (*established)(struct mptcp_sock *msk); | 49 | mptcp: pm: add struct mptcp_pm_param |
44 | int (*closed)(struct mptcp_sock *msk); | 50 | mptcp: pm: pass pm_param to get_local_id |
45 | int (*address_announced)(struct mptcp_sock *msk, | 51 | mptcp: pm: define struct mptcp_pm_ops |
46 | struct mptcp_pm_addr_entry *local); | 52 | mptcp: pm: in-kernel: register mptcp_netlink_pm |
47 | int (*address_removed)(struct mptcp_sock *msk, u8 id); | 53 | mptcp: pm: userspace: register mptcp_userspace_pm |
48 | int (*subflow_established)(struct mptcp_sock *msk, | 54 | mptcp: pm: initialize and release mptcp_pm_ops |
49 | struct mptcp_pm_addr_entry *local, | 55 | mptcp: pm: drop get_local_id helpers |
50 | struct mptcp_addr_info *remote); | 56 | mptcp: pm: drop is_backup helpers |
51 | int (*subflow_closed)(struct mptcp_sock *msk, | ||
52 | struct mptcp_pm_addr_entry *local, | ||
53 | struct mptcp_addr_info *remote); | ||
54 | int (*get_local_id)(struct mptcp_sock *msk, | ||
55 | struct mptcp_pm_addr_entry *skc); | ||
56 | bool (*get_priority)(struct mptcp_sock *msk, | ||
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); | ||
63 | 57 | ||
64 | u8 type; | 58 | include/net/mptcp.h | 42 +++++++++++++ |
65 | struct module *owner; | 59 | net/mptcp/pm.c | 131 ++++++++++++++++++++++++++++++++++++--- |
66 | struct list_head list; | 60 | net/mptcp/pm_netlink.c | 23 +++++-- |
67 | 61 | net/mptcp/pm_userspace.c | 37 ++++++----- | |
68 | void (*init)(struct mptcp_sock *msk); | 62 | net/mptcp/protocol.c | 1 + |
69 | void (*release)(struct mptcp_sock *msk); | 63 | net/mptcp/protocol.h | 24 +++---- |
70 | } ____cacheline_aligned_in_smp; | 64 | 6 files changed, 216 insertions(+), 42 deletions(-) |
71 | |||
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 | ||
77 | |||
78 | include/net/mptcp.h | 32 +++++ | ||
79 | net/mptcp/pm.c | 109 ++++++++++++++- | ||
80 | net/mptcp/pm_netlink.c | 11 +- | ||
81 | net/mptcp/pm_userspace.c | 294 ++++++++++++++++++++++++--------------- | ||
82 | net/mptcp/protocol.c | 10 +- | ||
83 | net/mptcp/protocol.h | 15 +- | ||
84 | 6 files changed, 355 insertions(+), 116 deletions(-) | ||
85 | 65 | ||
86 | -- | 66 | -- |
87 | 2.43.0 | 67 | 2.43.0 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Geliang Tang <tanggeliang@kylinos.cn> | ||
1 | 2 | ||
3 | The following code in mptcp_userspace_pm_get_local_id() that assigns "skc" | ||
4 | to "new_entry" is not allowed in BPF if we use the same code to implement | ||
5 | the get_local_id() interface of a BFP path manager: | ||
6 | |||
7 | memset(&new_entry, 0, sizeof(struct mptcp_pm_addr_entry)); | ||
8 | new_entry.addr = *skc; | ||
9 | new_entry.addr.id = 0; | ||
10 | new_entry.flags = MPTCP_PM_ADDR_FLAG_IMPLICIT; | ||
11 | |||
12 | To solve the issue, this patch moves this assignment to "new_entry" forward | ||
13 | to mptcp_pm_get_local_id(), and then passing "new_entry" as a parameter to | ||
14 | both mptcp_pm_nl_get_local_id() and mptcp_userspace_pm_get_local_id(). | ||
15 | |||
16 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> | ||
17 | --- | ||
18 | net/mptcp/pm.c | 11 ++++++++--- | ||
19 | net/mptcp/pm_netlink.c | 9 ++++----- | ||
20 | net/mptcp/pm_userspace.c | 17 ++++++----------- | ||
21 | net/mptcp/protocol.h | 6 ++++-- | ||
22 | 4 files changed, 22 insertions(+), 21 deletions(-) | ||
23 | |||
24 | diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c | ||
25 | index XXXXXXX..XXXXXXX 100644 | ||
26 | --- a/net/mptcp/pm.c | ||
27 | +++ b/net/mptcp/pm.c | ||
28 | @@ -XXX,XX +XXX,XX @@ bool mptcp_pm_rm_addr_signal(struct mptcp_sock *msk, unsigned int remaining, | ||
29 | |||
30 | int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc) | ||
31 | { | ||
32 | - struct mptcp_addr_info skc_local; | ||
33 | + struct mptcp_pm_addr_entry skc_local; | ||
34 | struct mptcp_addr_info msk_local; | ||
35 | |||
36 | if (WARN_ON_ONCE(!msk)) | ||
37 | return -1; | ||
38 | |||
39 | + memset(&skc_local, 0, sizeof(struct mptcp_pm_addr_entry)); | ||
40 | + | ||
41 | /* The 0 ID mapping is defined by the first subflow, copied into the msk | ||
42 | * addr | ||
43 | */ | ||
44 | mptcp_local_address((struct sock_common *)msk, &msk_local); | ||
45 | - mptcp_local_address((struct sock_common *)skc, &skc_local); | ||
46 | - if (mptcp_addresses_equal(&msk_local, &skc_local, false)) | ||
47 | + mptcp_local_address((struct sock_common *)skc, &skc_local.addr); | ||
48 | + if (mptcp_addresses_equal(&msk_local, &skc_local.addr, false)) | ||
49 | return 0; | ||
50 | |||
51 | + skc_local.addr.id = 0; | ||
52 | + skc_local.flags = MPTCP_PM_ADDR_FLAG_IMPLICIT; | ||
53 | + | ||
54 | if (mptcp_pm_is_userspace(msk)) | ||
55 | return mptcp_userspace_pm_get_local_id(msk, &skc_local); | ||
56 | return mptcp_pm_nl_get_local_id(msk, &skc_local); | ||
57 | diff --git a/net/mptcp/pm_netlink.c b/net/mptcp/pm_netlink.c | ||
58 | index XXXXXXX..XXXXXXX 100644 | ||
59 | --- a/net/mptcp/pm_netlink.c | ||
60 | +++ b/net/mptcp/pm_netlink.c | ||
61 | @@ -XXX,XX +XXX,XX @@ static int mptcp_pm_nl_create_listen_socket(struct sock *sk, | ||
62 | return err; | ||
63 | } | ||
64 | |||
65 | -int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, struct mptcp_addr_info *skc) | ||
66 | +int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, | ||
67 | + struct mptcp_pm_addr_entry *skc) | ||
68 | { | ||
69 | struct mptcp_pm_addr_entry *entry; | ||
70 | struct pm_nl_pernet *pernet; | ||
71 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, struct mptcp_addr_info *skc | ||
72 | pernet = pm_nl_get_pernet_from_msk(msk); | ||
73 | |||
74 | rcu_read_lock(); | ||
75 | - entry = __lookup_addr(pernet, skc); | ||
76 | + entry = __lookup_addr(pernet, &skc->addr); | ||
77 | ret = entry ? entry->addr.id : -1; | ||
78 | rcu_read_unlock(); | ||
79 | if (ret >= 0) | ||
80 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, struct mptcp_addr_info *skc | ||
81 | if (!entry) | ||
82 | return -ENOMEM; | ||
83 | |||
84 | - entry->addr = *skc; | ||
85 | - entry->addr.id = 0; | ||
86 | + *entry = *skc; | ||
87 | entry->addr.port = 0; | ||
88 | entry->ifindex = 0; | ||
89 | - entry->flags = MPTCP_PM_ADDR_FLAG_IMPLICIT; | ||
90 | entry->lsk = NULL; | ||
91 | ret = mptcp_pm_nl_append_new_local_addr(pernet, entry, true); | ||
92 | if (ret < 0) | ||
93 | diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c | ||
94 | index XXXXXXX..XXXXXXX 100644 | ||
95 | --- a/net/mptcp/pm_userspace.c | ||
96 | +++ b/net/mptcp/pm_userspace.c | ||
97 | @@ -XXX,XX +XXX,XX @@ mptcp_userspace_pm_lookup_addr_by_id(struct mptcp_sock *msk, unsigned int id) | ||
98 | } | ||
99 | |||
100 | int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
101 | - struct mptcp_addr_info *skc) | ||
102 | + struct mptcp_pm_addr_entry *skc) | ||
103 | { | ||
104 | - struct mptcp_pm_addr_entry *entry = NULL, new_entry; | ||
105 | __be16 msk_sport = ((struct inet_sock *) | ||
106 | inet_sk((struct sock *)msk))->inet_sport; | ||
107 | + struct mptcp_pm_addr_entry *entry; | ||
108 | |||
109 | spin_lock_bh(&msk->pm.lock); | ||
110 | - entry = mptcp_userspace_pm_lookup_addr(msk, skc); | ||
111 | + entry = mptcp_userspace_pm_lookup_addr(msk, &skc->addr); | ||
112 | spin_unlock_bh(&msk->pm.lock); | ||
113 | if (entry) | ||
114 | return entry->addr.id; | ||
115 | |||
116 | - memset(&new_entry, 0, sizeof(struct mptcp_pm_addr_entry)); | ||
117 | - new_entry.addr = *skc; | ||
118 | - new_entry.addr.id = 0; | ||
119 | - new_entry.flags = MPTCP_PM_ADDR_FLAG_IMPLICIT; | ||
120 | - | ||
121 | - if (new_entry.addr.port == msk_sport) | ||
122 | - new_entry.addr.port = 0; | ||
123 | + if (skc->addr.port == msk_sport) | ||
124 | + skc->addr.port = 0; | ||
125 | |||
126 | - return mptcp_userspace_pm_append_new_local_addr(msk, &new_entry, true); | ||
127 | + return mptcp_userspace_pm_append_new_local_addr(msk, skc, true); | ||
128 | } | ||
129 | |||
130 | bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, | ||
131 | diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h | ||
132 | index XXXXXXX..XXXXXXX 100644 | ||
133 | --- a/net/mptcp/protocol.h | ||
134 | +++ b/net/mptcp/protocol.h | ||
135 | @@ -XXX,XX +XXX,XX @@ bool mptcp_pm_add_addr_signal(struct mptcp_sock *msk, const struct sk_buff *skb, | ||
136 | bool mptcp_pm_rm_addr_signal(struct mptcp_sock *msk, unsigned int remaining, | ||
137 | struct mptcp_rm_list *rm_list); | ||
138 | int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc); | ||
139 | -int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, struct mptcp_addr_info *skc); | ||
140 | -int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, struct mptcp_addr_info *skc); | ||
141 | +int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, | ||
142 | + struct mptcp_pm_addr_entry *skc); | ||
143 | +int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
144 | + struct mptcp_pm_addr_entry *skc); | ||
145 | bool mptcp_pm_is_backup(struct mptcp_sock *msk, struct sock_common *skc); | ||
146 | bool mptcp_pm_nl_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc); | ||
147 | bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc); | ||
148 | -- | ||
149 | 2.43.0 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Geliang Tang <tanggeliang@kylinos.cn> | ||
1 | 2 | ||
3 | Generally, in the path manager interfaces, the local address is defined | ||
4 | as an mptcp_pm_addr_entry type address, while the remote address is | ||
5 | defined as an mptcp_addr_info type one: | ||
6 | |||
7 | (struct mptcp_pm_addr_entry *local, struct mptcp_addr_info *remote) | ||
8 | |||
9 | In order to make these interfaces more flexible and extensible, a struct | ||
10 | mptcp_pm_param is defined here to pass parameters. "entry" can be used | ||
11 | as the local address entry, and "addr" can be used as the remote address. | ||
12 | |||
13 | Also add a new helper mptcp_pm_param_set_contexts() to set a struct | ||
14 | mptcp_pm_param type parameter. | ||
15 | |||
16 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> | ||
17 | --- | ||
18 | include/net/mptcp.h | 13 +++++++++++++ | ||
19 | net/mptcp/pm.c | 10 ++++++++++ | ||
20 | net/mptcp/protocol.h | 11 +++-------- | ||
21 | 3 files changed, 26 insertions(+), 8 deletions(-) | ||
22 | |||
23 | diff --git a/include/net/mptcp.h b/include/net/mptcp.h | ||
24 | index XXXXXXX..XXXXXXX 100644 | ||
25 | --- a/include/net/mptcp.h | ||
26 | +++ b/include/net/mptcp.h | ||
27 | @@ -XXX,XX +XXX,XX @@ struct mptcp_sched_ops { | ||
28 | void (*release)(struct mptcp_sock *msk); | ||
29 | } ____cacheline_aligned_in_smp; | ||
30 | |||
31 | +struct mptcp_pm_addr_entry { | ||
32 | + struct list_head list; | ||
33 | + struct mptcp_addr_info addr; | ||
34 | + u8 flags; | ||
35 | + int ifindex; | ||
36 | + struct socket *lsk; | ||
37 | +}; | ||
38 | + | ||
39 | +struct mptcp_pm_param { | ||
40 | + struct mptcp_pm_addr_entry entry; | ||
41 | + struct mptcp_addr_info addr; | ||
42 | +}; | ||
43 | + | ||
44 | #ifdef CONFIG_MPTCP | ||
45 | void mptcp_init(void); | ||
46 | |||
47 | diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c | ||
48 | index XXXXXXX..XXXXXXX 100644 | ||
49 | --- a/net/mptcp/pm.c | ||
50 | +++ b/net/mptcp/pm.c | ||
51 | @@ -XXX,XX +XXX,XX @@ bool mptcp_pm_rm_addr_signal(struct mptcp_sock *msk, unsigned int remaining, | ||
52 | return ret; | ||
53 | } | ||
54 | |||
55 | +void mptcp_pm_param_set_contexts(struct mptcp_pm_param *param, | ||
56 | + const struct mptcp_pm_addr_entry *entry, | ||
57 | + const struct mptcp_addr_info *addr) | ||
58 | +{ | ||
59 | + if (entry) | ||
60 | + param->entry = *entry; | ||
61 | + if (addr) | ||
62 | + param->addr = *addr; | ||
63 | +} | ||
64 | + | ||
65 | int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc) | ||
66 | { | ||
67 | struct mptcp_pm_addr_entry skc_local; | ||
68 | diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h | ||
69 | index XXXXXXX..XXXXXXX 100644 | ||
70 | --- a/net/mptcp/protocol.h | ||
71 | +++ b/net/mptcp/protocol.h | ||
72 | @@ -XXX,XX +XXX,XX @@ struct mptcp_pm_local { | ||
73 | int ifindex; | ||
74 | }; | ||
75 | |||
76 | -struct mptcp_pm_addr_entry { | ||
77 | - struct list_head list; | ||
78 | - struct mptcp_addr_info addr; | ||
79 | - u8 flags; | ||
80 | - int ifindex; | ||
81 | - struct socket *lsk; | ||
82 | -}; | ||
83 | - | ||
84 | struct mptcp_data_frag { | ||
85 | struct list_head list; | ||
86 | u64 data_seq; | ||
87 | @@ -XXX,XX +XXX,XX @@ bool mptcp_pm_add_addr_signal(struct mptcp_sock *msk, const struct sk_buff *skb, | ||
88 | bool *drop_other_suboptions); | ||
89 | bool mptcp_pm_rm_addr_signal(struct mptcp_sock *msk, unsigned int remaining, | ||
90 | struct mptcp_rm_list *rm_list); | ||
91 | +void mptcp_pm_param_set_contexts(struct mptcp_pm_param *param, | ||
92 | + const struct mptcp_pm_addr_entry *entry, | ||
93 | + const struct mptcp_addr_info *addr); | ||
94 | int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc); | ||
95 | int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, | ||
96 | struct mptcp_pm_addr_entry *skc); | ||
97 | -- | ||
98 | 2.43.0 | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Geliang Tang <tanggeliang@kylinos.cn> | ||
1 | 2 | ||
3 | This patch changes the 2nd parameter of get_local_id() interface as | ||
4 | "struct mptcp_pm_param" type. Only "entry" member of this struct is | ||
5 | used. | ||
6 | |||
7 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> | ||
8 | --- | ||
9 | net/mptcp/pm.c | 6 ++++-- | ||
10 | net/mptcp/pm_netlink.c | 3 ++- | ||
11 | net/mptcp/pm_userspace.c | 3 ++- | ||
12 | net/mptcp/protocol.h | 4 ++-- | ||
13 | 4 files changed, 10 insertions(+), 6 deletions(-) | ||
14 | |||
15 | diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c | ||
16 | index XXXXXXX..XXXXXXX 100644 | ||
17 | --- a/net/mptcp/pm.c | ||
18 | +++ b/net/mptcp/pm.c | ||
19 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc) | ||
20 | { | ||
21 | struct mptcp_pm_addr_entry skc_local; | ||
22 | struct mptcp_addr_info msk_local; | ||
23 | + struct mptcp_pm_param param; | ||
24 | |||
25 | if (WARN_ON_ONCE(!msk)) | ||
26 | return -1; | ||
27 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc) | ||
28 | skc_local.addr.id = 0; | ||
29 | skc_local.flags = MPTCP_PM_ADDR_FLAG_IMPLICIT; | ||
30 | |||
31 | + mptcp_pm_param_set_contexts(¶m, &skc_local, NULL); | ||
32 | if (mptcp_pm_is_userspace(msk)) | ||
33 | - return mptcp_userspace_pm_get_local_id(msk, &skc_local); | ||
34 | - return mptcp_pm_nl_get_local_id(msk, &skc_local); | ||
35 | + return mptcp_userspace_pm_get_local_id(msk, ¶m); | ||
36 | + return mptcp_pm_nl_get_local_id(msk, ¶m); | ||
37 | } | ||
38 | |||
39 | bool mptcp_pm_is_backup(struct mptcp_sock *msk, struct sock_common *skc) | ||
40 | diff --git a/net/mptcp/pm_netlink.c b/net/mptcp/pm_netlink.c | ||
41 | index XXXXXXX..XXXXXXX 100644 | ||
42 | --- a/net/mptcp/pm_netlink.c | ||
43 | +++ b/net/mptcp/pm_netlink.c | ||
44 | @@ -XXX,XX +XXX,XX @@ static int mptcp_pm_nl_create_listen_socket(struct sock *sk, | ||
45 | } | ||
46 | |||
47 | int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, | ||
48 | - struct mptcp_pm_addr_entry *skc) | ||
49 | + struct mptcp_pm_param *param) | ||
50 | { | ||
51 | + struct mptcp_pm_addr_entry *skc = ¶m->entry; | ||
52 | struct mptcp_pm_addr_entry *entry; | ||
53 | struct pm_nl_pernet *pernet; | ||
54 | int ret; | ||
55 | diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c | ||
56 | index XXXXXXX..XXXXXXX 100644 | ||
57 | --- a/net/mptcp/pm_userspace.c | ||
58 | +++ b/net/mptcp/pm_userspace.c | ||
59 | @@ -XXX,XX +XXX,XX @@ mptcp_userspace_pm_lookup_addr_by_id(struct mptcp_sock *msk, unsigned int id) | ||
60 | } | ||
61 | |||
62 | int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
63 | - struct mptcp_pm_addr_entry *skc) | ||
64 | + struct mptcp_pm_param *param) | ||
65 | { | ||
66 | __be16 msk_sport = ((struct inet_sock *) | ||
67 | inet_sk((struct sock *)msk))->inet_sport; | ||
68 | + struct mptcp_pm_addr_entry *skc = ¶m->entry; | ||
69 | struct mptcp_pm_addr_entry *entry; | ||
70 | |||
71 | spin_lock_bh(&msk->pm.lock); | ||
72 | diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h | ||
73 | index XXXXXXX..XXXXXXX 100644 | ||
74 | --- a/net/mptcp/protocol.h | ||
75 | +++ b/net/mptcp/protocol.h | ||
76 | @@ -XXX,XX +XXX,XX @@ void mptcp_pm_param_set_contexts(struct mptcp_pm_param *param, | ||
77 | const struct mptcp_addr_info *addr); | ||
78 | int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc); | ||
79 | int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, | ||
80 | - struct mptcp_pm_addr_entry *skc); | ||
81 | + struct mptcp_pm_param *param); | ||
82 | int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
83 | - struct mptcp_pm_addr_entry *skc); | ||
84 | + struct mptcp_pm_param *param); | ||
85 | bool mptcp_pm_is_backup(struct mptcp_sock *msk, struct sock_common *skc); | ||
86 | bool mptcp_pm_nl_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc); | ||
87 | bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc); | ||
88 | -- | ||
89 | 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 | 32 ++++++++++++++++++++++++ | 12 | include/net/mptcp.h | 29 ++++++++++++++++++++++ |
13 | net/mptcp/pm.c | 59 ++++++++++++++++++++++++++++++++++++++++++++ | 13 | net/mptcp/pm.c | 59 ++++++++++++++++++++++++++++++++++++++++++++ |
14 | net/mptcp/protocol.h | 5 ++++ | 14 | net/mptcp/protocol.h | 5 ++++ |
15 | 3 files changed, 96 insertions(+) | 15 | 3 files changed, 93 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 |
21 | @@ -XXX,XX +XXX,XX @@ | 21 | @@ -XXX,XX +XXX,XX @@ struct mptcp_pm_param { |
22 | 22 | struct mptcp_addr_info addr; | |
23 | struct mptcp_info; | 23 | }; |
24 | struct mptcp_sock; | ||
25 | +struct mptcp_pm_addr_entry; | ||
26 | struct seq_file; | ||
27 | |||
28 | /* MPTCP sk_buff extension data */ | ||
29 | @@ -XXX,XX +XXX,XX @@ struct mptcp_sched_ops { | ||
30 | void (*release)(struct mptcp_sock *msk); | ||
31 | } ____cacheline_aligned_in_smp; | ||
32 | 24 | ||
33 | +struct mptcp_pm_ops { | 25 | +struct mptcp_pm_ops { |
34 | + int (*created)(struct mptcp_sock *msk); | 26 | + int (*created)(struct mptcp_sock *msk); |
35 | + int (*established)(struct mptcp_sock *msk); | 27 | + int (*established)(struct mptcp_sock *msk); |
36 | + int (*closed)(struct mptcp_sock *msk); | 28 | + int (*closed)(struct mptcp_sock *msk); |
37 | + int (*address_announced)(struct mptcp_sock *msk, | 29 | + int (*address_announced)(struct mptcp_sock *msk, |
38 | + struct mptcp_pm_addr_entry *local); | 30 | + struct mptcp_pm_param *param); |
39 | + int (*address_removed)(struct mptcp_sock *msk, u8 id); | 31 | + int (*address_removed)(struct mptcp_sock *msk, |
32 | + struct mptcp_pm_param *param); | ||
40 | + int (*subflow_established)(struct mptcp_sock *msk, | 33 | + int (*subflow_established)(struct mptcp_sock *msk, |
41 | + struct mptcp_pm_addr_entry *local, | 34 | + struct mptcp_pm_param *param); |
42 | + struct mptcp_addr_info *remote); | ||
43 | + int (*subflow_closed)(struct mptcp_sock *msk, | 35 | + int (*subflow_closed)(struct mptcp_sock *msk, |
44 | + struct mptcp_pm_addr_entry *local, | 36 | + struct mptcp_pm_param *param); |
45 | + struct mptcp_addr_info *remote); | ||
46 | + int (*get_local_id)(struct mptcp_sock *msk, | 37 | + int (*get_local_id)(struct mptcp_sock *msk, |
47 | + struct mptcp_pm_addr_entry *skc); | 38 | + struct mptcp_pm_param *param); |
48 | + bool (*get_priority)(struct mptcp_sock *msk, | 39 | + bool (*get_priority)(struct mptcp_sock *msk, |
49 | + struct mptcp_addr_info *skc); | 40 | + struct mptcp_pm_param *param); |
50 | + int (*set_priority)(struct mptcp_sock *msk, | 41 | + int (*set_priority)(struct mptcp_sock *msk, |
51 | + struct mptcp_pm_addr_entry *local, | 42 | + struct mptcp_pm_param *param); |
52 | + struct mptcp_addr_info *remote); | ||
53 | + int (*listener_created)(struct mptcp_sock *msk); | 43 | + int (*listener_created)(struct mptcp_sock *msk); |
54 | + int (*listener_closed)(struct mptcp_sock *msk); | 44 | + int (*listener_closed)(struct mptcp_sock *msk); |
55 | + | 45 | + |
56 | + u8 type; | 46 | + u8 type; |
57 | + struct module *owner; | 47 | + struct module *owner; |
... | ... | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Geliang Tang <tanggeliang@kylinos.cn> | ||
1 | 2 | ||
3 | This patch defines the original in-kernel netlink path manager as a new | ||
4 | struct mptcp_pm_ops named "mptcp_netlink_pm", and register it in | ||
5 | mptcp_pm_nl_init(). | ||
6 | |||
7 | Only get_local_id() and get_priority() interfaces are implemented here. | ||
8 | mptcp_pm_nl_is_backup() becomes a wrapper of get_priority(). | ||
9 | |||
10 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> | ||
11 | --- | ||
12 | net/mptcp/pm_netlink.c | 21 ++++++++++++++++++++- | ||
13 | 1 file changed, 20 insertions(+), 1 deletion(-) | ||
14 | |||
15 | diff --git a/net/mptcp/pm_netlink.c b/net/mptcp/pm_netlink.c | ||
16 | index XXXXXXX..XXXXXXX 100644 | ||
17 | --- a/net/mptcp/pm_netlink.c | ||
18 | +++ b/net/mptcp/pm_netlink.c | ||
19 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, | ||
20 | return ret; | ||
21 | } | ||
22 | |||
23 | -bool mptcp_pm_nl_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc) | ||
24 | +static bool mptcp_pm_nl_get_priority(struct mptcp_sock *msk, | ||
25 | + struct mptcp_pm_param *param) | ||
26 | { | ||
27 | struct pm_nl_pernet *pernet = pm_nl_get_pernet_from_msk(msk); | ||
28 | + struct mptcp_addr_info *skc = ¶m->addr; | ||
29 | struct mptcp_pm_addr_entry *entry; | ||
30 | bool backup; | ||
31 | |||
32 | @@ -XXX,XX +XXX,XX @@ bool mptcp_pm_nl_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc) | ||
33 | return backup; | ||
34 | } | ||
35 | |||
36 | +bool mptcp_pm_nl_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc) | ||
37 | +{ | ||
38 | + struct mptcp_pm_param param; | ||
39 | + | ||
40 | + mptcp_pm_param_set_contexts(¶m, NULL, skc); | ||
41 | + return mptcp_pm_nl_get_priority(msk, ¶m); | ||
42 | +} | ||
43 | + | ||
44 | #define MPTCP_PM_CMD_GRP_OFFSET 0 | ||
45 | #define MPTCP_PM_EV_GRP_OFFSET 1 | ||
46 | |||
47 | @@ -XXX,XX +XXX,XX @@ static struct pernet_operations mptcp_pm_pernet_ops = { | ||
48 | .size = sizeof(struct pm_nl_pernet), | ||
49 | }; | ||
50 | |||
51 | +static struct mptcp_pm_ops mptcp_netlink_pm = { | ||
52 | + .get_local_id = mptcp_pm_nl_get_local_id, | ||
53 | + .get_priority = mptcp_pm_nl_get_priority, | ||
54 | + .type = MPTCP_PM_TYPE_KERNEL, | ||
55 | + .owner = THIS_MODULE, | ||
56 | +}; | ||
57 | + | ||
58 | void __init mptcp_pm_nl_init(void) | ||
59 | { | ||
60 | if (register_pernet_subsys(&mptcp_pm_pernet_ops) < 0) | ||
61 | @@ -XXX,XX +XXX,XX @@ void __init mptcp_pm_nl_init(void) | ||
62 | |||
63 | if (genl_register_family(&mptcp_genl_family)) | ||
64 | panic("Failed to register MPTCP PM netlink family\n"); | ||
65 | + | ||
66 | + mptcp_pm_register(&mptcp_netlink_pm); | ||
67 | } | ||
68 | -- | ||
69 | 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 path manager as a new struct |
4 | named mptcp_userspace_pm, and register it in mptcp_pm_data_init(). | 4 | mptcp_pm_ops named "mptcp_userspace_pm", and register it in |
5 | mptcp_pm_data_init(). | ||
5 | 6 | ||
6 | Extract address_announce() interface of the userspace PM from the handler | 7 | Only get_local_id() and get_priority() interfaces are implemented here. |
7 | of netlink command MPTCP_PM_CMD_ANNOUNCE mptcp_pm_nl_announce_doit(), only | 8 | mptcp_userspace_pm_is_backup() becomes a wrapper of get_priority(). |
8 | leave the code for obtaining msk through "info" and parsing address entry | ||
9 | in the handler. | ||
10 | |||
11 | Extract address_remove() interface of the userspace PM from the handler | ||
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" | ||
14 | in the handler. | ||
15 | |||
16 | Extract subflow_create() interface of the userspace PM from the handler of | ||
17 | netlink command MPTCP_PM_CMD_SUBFLOW_CREATE | ||
18 | |||
19 | mptcp_pm_nl_subflow_create_doit(), | ||
20 | |||
21 | only leave the code for obtaining msk through "info", parsing local address | ||
22 | entry and parsing remote address info in the handler. | ||
23 | |||
24 | Extract subflow_destroy() interface of the userspace PM from the handler of | ||
25 | netlink command MPTCP_PM_CMD_SUBFLOW_DESTROY | ||
26 | |||
27 | mptcp_pm_nl_subflow_destroy_doit(), | ||
28 | |||
29 | only leave the code for obtaining msk through "info", parsing local address | ||
30 | entry and parsing remote address info in the handler. | ||
31 | |||
32 | Extract set_flags() interface of the userspace PM from function | ||
33 | mptcp_userspace_pm_set_flags(), only leave the code for obtaining | ||
34 | msk through "info" in this function. | ||
35 | |||
36 | All fives interfaces are invoked under holding the msk socket lock. | ||
37 | |||
38 | Extract get_local_id() interface of the userspace PM from function | ||
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. | ||
45 | 9 | ||
46 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> | 10 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> |
47 | --- | 11 | --- |
48 | net/mptcp/pm.c | 1 + | 12 | net/mptcp/pm.c | 1 + |
49 | net/mptcp/pm_userspace.c | 280 +++++++++++++++++++++++---------------- | 13 | net/mptcp/pm_userspace.c | 26 ++++++++++++++++++++++++-- |
50 | net/mptcp/protocol.h | 1 + | 14 | net/mptcp/protocol.h | 1 + |
51 | 3 files changed, 171 insertions(+), 111 deletions(-) | 15 | 3 files changed, 26 insertions(+), 2 deletions(-) |
52 | 16 | ||
53 | diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c | 17 | diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c |
54 | index XXXXXXX..XXXXXXX 100644 | 18 | index XXXXXXX..XXXXXXX 100644 |
55 | --- a/net/mptcp/pm.c | 19 | --- a/net/mptcp/pm.c |
56 | +++ b/net/mptcp/pm.c | 20 | +++ b/net/mptcp/pm.c |
... | ... | ||
64 | /* Must be called with rcu read lock held */ | 28 | /* Must be called with rcu read lock held */ |
65 | diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c | 29 | diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c |
66 | index XXXXXXX..XXXXXXX 100644 | 30 | index XXXXXXX..XXXXXXX 100644 |
67 | --- a/net/mptcp/pm_userspace.c | 31 | --- a/net/mptcp/pm_userspace.c |
68 | +++ b/net/mptcp/pm_userspace.c | 32 | +++ b/net/mptcp/pm_userspace.c |
69 | @@ -XXX,XX +XXX,XX @@ mptcp_userspace_pm_lookup_addr_by_id(struct mptcp_sock *msk, unsigned int id) | 33 | @@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, |
70 | return NULL; | 34 | return mptcp_userspace_pm_append_new_local_addr(msk, skc, true); |
71 | } | 35 | } |
72 | |||
73 | -int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
74 | - struct mptcp_addr_info *skc) | ||
75 | +static int userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
76 | + struct mptcp_pm_addr_entry *skc) | ||
77 | { | ||
78 | - struct mptcp_pm_addr_entry *entry = NULL, new_entry; | ||
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; | ||
103 | @@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
104 | if (new_entry.addr.port == msk_sport) | ||
105 | new_entry.addr.port = 0; | ||
106 | |||
107 | - return mptcp_userspace_pm_append_new_local_addr(msk, &new_entry, true); | ||
108 | + return userspace_pm_get_local_id(msk, &new_entry); | ||
109 | } | ||
110 | 36 | ||
111 | -bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, | 37 | -bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, |
112 | - struct mptcp_addr_info *skc) | 38 | - struct mptcp_addr_info *skc) |
113 | +static bool userspace_pm_get_priority(struct mptcp_sock *msk, | 39 | +static bool mptcp_userspace_pm_get_priority(struct mptcp_sock *msk, |
114 | + struct mptcp_addr_info *skc) | 40 | + struct mptcp_pm_param *param) |
115 | { | 41 | { |
42 | + struct mptcp_addr_info *skc = ¶m->addr; | ||
116 | struct mptcp_pm_addr_entry *entry; | 43 | struct mptcp_pm_addr_entry *entry; |
117 | bool backup; | 44 | bool backup; |
45 | |||
118 | @@ -XXX,XX +XXX,XX @@ bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, | 46 | @@ -XXX,XX +XXX,XX @@ bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, |
119 | return backup; | 47 | return backup; |
120 | } | 48 | } |
121 | 49 | ||
122 | +bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, | 50 | +bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, |
123 | + struct mptcp_addr_info *skc) | 51 | + struct mptcp_addr_info *skc) |
124 | +{ | 52 | +{ |
125 | + return userspace_pm_get_priority(msk, skc); | 53 | + struct mptcp_pm_param param; |
54 | + | ||
55 | + mptcp_pm_param_set_contexts(¶m, NULL, skc); | ||
56 | + return mptcp_userspace_pm_get_priority(msk, ¶m); | ||
126 | +} | 57 | +} |
127 | + | 58 | + |
128 | static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *info) | 59 | static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *info) |
129 | { | 60 | { |
130 | struct mptcp_sock *msk; | 61 | struct mptcp_sock *msk; |
131 | @@ -XXX,XX +XXX,XX @@ static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *in | ||
132 | return msk; | ||
133 | } | ||
134 | |||
135 | +static int userspace_pm_address_announced(struct mptcp_sock *msk, | ||
136 | + struct mptcp_pm_addr_entry *local) | ||
137 | +{ | ||
138 | + int err; | ||
139 | + | ||
140 | + err = mptcp_userspace_pm_append_new_local_addr(msk, local, false); | ||
141 | + if (err < 0) | ||
142 | + return err; | ||
143 | + | ||
144 | + spin_lock_bh(&msk->pm.lock); | ||
145 | + | ||
146 | + if (mptcp_pm_alloc_anno_list(msk, &local->addr)) { | ||
147 | + msk->pm.add_addr_signaled++; | ||
148 | + mptcp_pm_announce_addr(msk, &local->addr, false); | ||
149 | + mptcp_pm_nl_addr_send_ack(msk); | ||
150 | + } | ||
151 | + | ||
152 | + spin_unlock_bh(&msk->pm.lock); | ||
153 | + | ||
154 | + return 0; | ||
155 | +} | ||
156 | + | ||
157 | int mptcp_pm_nl_announce_doit(struct sk_buff *skb, struct genl_info *info) | ||
158 | { | ||
159 | struct mptcp_pm_addr_entry addr_val; | ||
160 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_announce_doit(struct sk_buff *skb, struct genl_info *info) | ||
161 | goto announce_err; | ||
162 | } | ||
163 | |||
164 | - err = mptcp_userspace_pm_append_new_local_addr(msk, &addr_val, false); | ||
165 | - if (err < 0) { | ||
166 | - NL_SET_ERR_MSG_ATTR(info->extack, addr, | ||
167 | - "did not match address and id"); | ||
168 | - goto announce_err; | ||
169 | - } | ||
170 | - | ||
171 | lock_sock(sk); | ||
172 | - spin_lock_bh(&msk->pm.lock); | ||
173 | - | ||
174 | - if (mptcp_pm_alloc_anno_list(msk, &addr_val.addr)) { | ||
175 | - msk->pm.add_addr_signaled++; | ||
176 | - mptcp_pm_announce_addr(msk, &addr_val.addr, false); | ||
177 | - mptcp_pm_nl_addr_send_ack(msk); | ||
178 | - } | ||
179 | - | ||
180 | - spin_unlock_bh(&msk->pm.lock); | ||
181 | + err = userspace_pm_address_announced(msk, &addr_val); | ||
182 | release_sock(sk); | ||
183 | + if (err) | ||
184 | + NL_SET_ERR_MSG_ATTR(info->extack, addr, | ||
185 | + "did not match address and id"); | ||
186 | |||
187 | - err = 0; | ||
188 | announce_err: | ||
189 | sock_put(sk); | ||
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, | ||
225 | @@ -XXX,XX +XXX,XX @@ void mptcp_pm_remove_addr_entry(struct mptcp_sock *msk, | ||
226 | spin_unlock_bh(&msk->pm.lock); | ||
227 | } | ||
228 | |||
229 | +static int userspace_pm_address_removed(struct mptcp_sock *msk, u8 id) | ||
230 | +{ | ||
231 | + struct mptcp_pm_addr_entry *entry; | ||
232 | + | ||
233 | + if (id == 0) | ||
234 | + return mptcp_userspace_pm_remove_id_zero_address(msk); | ||
235 | + | ||
236 | + spin_lock_bh(&msk->pm.lock); | ||
237 | + entry = mptcp_userspace_pm_lookup_addr_by_id(msk, id); | ||
238 | + if (!entry) { | ||
239 | + spin_unlock_bh(&msk->pm.lock); | ||
240 | + return -EINVAL; | ||
241 | + } | ||
242 | + | ||
243 | + list_del_rcu(&entry->list); | ||
244 | + spin_unlock_bh(&msk->pm.lock); | ||
245 | + | ||
246 | + mptcp_pm_remove_addr_entry(msk, entry); | ||
247 | + | ||
248 | + sock_kfree_s((struct sock *)msk, entry, sizeof(*entry)); | ||
249 | + | ||
250 | + return 0; | ||
251 | +} | ||
252 | + | ||
253 | int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info) | ||
254 | { | ||
255 | - struct mptcp_pm_addr_entry *match; | ||
256 | struct mptcp_sock *msk; | ||
257 | struct nlattr *id; | ||
258 | int err = -EINVAL; | ||
259 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info) | ||
260 | |||
261 | sk = (struct sock *)msk; | ||
262 | |||
263 | - if (id_val == 0) { | ||
264 | - err = mptcp_userspace_pm_remove_id_zero_address(msk); | ||
265 | - goto out; | ||
266 | - } | ||
267 | - | ||
268 | lock_sock(sk); | ||
269 | - | ||
270 | - spin_lock_bh(&msk->pm.lock); | ||
271 | - match = mptcp_userspace_pm_lookup_addr_by_id(msk, id_val); | ||
272 | - if (!match) { | ||
273 | - spin_unlock_bh(&msk->pm.lock); | ||
274 | - release_sock(sk); | ||
275 | - goto out; | ||
276 | - } | ||
277 | - | ||
278 | - list_del_rcu(&match->list); | ||
279 | - spin_unlock_bh(&msk->pm.lock); | ||
280 | - | ||
281 | - mptcp_pm_remove_addr_entry(msk, match); | ||
282 | - | ||
283 | + err = userspace_pm_address_removed(msk, id_val); | ||
284 | release_sock(sk); | ||
285 | - | ||
286 | - sock_kfree_s(sk, match, sizeof(*match)); | ||
287 | - | ||
288 | - err = 0; | ||
289 | -out: | ||
290 | if (err) | ||
291 | NL_SET_ERR_MSG_ATTR_FMT(info->extack, id, | ||
292 | "address with id %u not found", | ||
293 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info) | ||
294 | return err; | ||
295 | } | ||
296 | |||
297 | +static int userspace_pm_subflow_established(struct mptcp_sock *msk, | ||
298 | + struct mptcp_pm_addr_entry *entry, | ||
299 | + struct mptcp_addr_info *remote) | ||
300 | +{ | ||
301 | + struct sock *sk = (struct sock *)msk; | ||
302 | + struct mptcp_pm_local local; | ||
303 | + int err; | ||
304 | + | ||
305 | + err = mptcp_userspace_pm_append_new_local_addr(msk, entry, false); | ||
306 | + if (err < 0) | ||
307 | + return err; | ||
308 | + | ||
309 | + local.addr = entry->addr; | ||
310 | + local.flags = entry->flags; | ||
311 | + local.ifindex = entry->ifindex; | ||
312 | + | ||
313 | + err = __mptcp_subflow_connect(sk, &local, remote); | ||
314 | + spin_lock_bh(&msk->pm.lock); | ||
315 | + if (err) | ||
316 | + mptcp_userspace_pm_delete_local_addr(msk, entry); | ||
317 | + else | ||
318 | + msk->pm.subflows++; | ||
319 | + spin_unlock_bh(&msk->pm.lock); | ||
320 | + | ||
321 | + return err; | ||
322 | +} | ||
323 | + | ||
324 | int mptcp_pm_nl_subflow_create_doit(struct sk_buff *skb, struct genl_info *info) | ||
325 | { | ||
326 | struct mptcp_pm_addr_entry entry = { 0 }; | ||
327 | struct mptcp_addr_info addr_r; | ||
328 | struct nlattr *raddr, *laddr; | ||
329 | - struct mptcp_pm_local local; | ||
330 | struct mptcp_sock *msk; | ||
331 | int err = -EINVAL; | ||
332 | struct sock *sk; | ||
333 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_subflow_create_doit(struct sk_buff *skb, struct genl_info *info) | ||
334 | goto create_err; | ||
335 | } | ||
336 | |||
337 | - err = mptcp_userspace_pm_append_new_local_addr(msk, &entry, false); | ||
338 | - if (err < 0) { | ||
339 | - NL_SET_ERR_MSG_ATTR(info->extack, laddr, | ||
340 | - "did not match address and id"); | ||
341 | - goto create_err; | ||
342 | - } | ||
343 | - | ||
344 | - local.addr = entry.addr; | ||
345 | - local.flags = entry.flags; | ||
346 | - local.ifindex = entry.ifindex; | ||
347 | - | ||
348 | lock_sock(sk); | ||
349 | - err = __mptcp_subflow_connect(sk, &local, &addr_r); | ||
350 | + err = userspace_pm_subflow_established(msk, &entry, &addr_r); | ||
351 | release_sock(sk); | ||
352 | |||
353 | if (err) | ||
354 | GENL_SET_ERR_MSG_FMT(info, "connect error: %d", err); | ||
355 | |||
356 | - spin_lock_bh(&msk->pm.lock); | ||
357 | - if (err) | ||
358 | - mptcp_userspace_pm_delete_local_addr(msk, &entry); | ||
359 | - else | ||
360 | - msk->pm.subflows++; | ||
361 | - spin_unlock_bh(&msk->pm.lock); | ||
362 | - | ||
363 | create_err: | ||
364 | sock_put(sk); | ||
365 | return err; | ||
366 | @@ -XXX,XX +XXX,XX @@ static struct sock *mptcp_nl_find_ssk(struct mptcp_sock *msk, | ||
367 | return NULL; | ||
368 | } | ||
369 | |||
370 | +static int userspace_pm_subflow_closed(struct mptcp_sock *msk, | ||
371 | + struct mptcp_pm_addr_entry *local, | ||
372 | + struct mptcp_addr_info *remote) | ||
373 | +{ | ||
374 | + struct sock *ssk, *sk = (struct sock *)msk; | ||
375 | + | ||
376 | + ssk = mptcp_nl_find_ssk(msk, &local->addr, remote); | ||
377 | + if (!ssk) | ||
378 | + return -ESRCH; | ||
379 | + | ||
380 | + spin_lock_bh(&msk->pm.lock); | ||
381 | + mptcp_userspace_pm_delete_local_addr(msk, local); | ||
382 | + spin_unlock_bh(&msk->pm.lock); | ||
383 | + mptcp_subflow_shutdown(sk, ssk, RCV_SHUTDOWN | SEND_SHUTDOWN); | ||
384 | + mptcp_close_ssk(sk, ssk, mptcp_subflow_ctx(ssk)); | ||
385 | + MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_RMSUBFLOW); | ||
386 | + | ||
387 | + return 0; | ||
388 | +} | ||
389 | + | ||
390 | int mptcp_pm_nl_subflow_destroy_doit(struct sk_buff *skb, struct genl_info *info) | ||
391 | { | ||
392 | struct mptcp_pm_addr_entry addr_l; | ||
393 | struct mptcp_addr_info addr_r; | ||
394 | struct nlattr *raddr, *laddr; | ||
395 | struct mptcp_sock *msk; | ||
396 | - struct sock *sk, *ssk; | ||
397 | int err = -EINVAL; | ||
398 | + struct sock *sk; | ||
399 | |||
400 | if (GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_ADDR) || | ||
401 | GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_ADDR_REMOTE)) | ||
402 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_subflow_destroy_doit(struct sk_buff *skb, struct genl_info *info | ||
403 | } | ||
404 | |||
405 | lock_sock(sk); | ||
406 | - ssk = mptcp_nl_find_ssk(msk, &addr_l.addr, &addr_r); | ||
407 | - if (!ssk) { | ||
408 | - GENL_SET_ERR_MSG(info, "subflow not found"); | ||
409 | - err = -ESRCH; | ||
410 | - goto release_sock; | ||
411 | - } | ||
412 | - | ||
413 | - spin_lock_bh(&msk->pm.lock); | ||
414 | - mptcp_userspace_pm_delete_local_addr(msk, &addr_l); | ||
415 | - spin_unlock_bh(&msk->pm.lock); | ||
416 | - mptcp_subflow_shutdown(sk, ssk, RCV_SHUTDOWN | SEND_SHUTDOWN); | ||
417 | - mptcp_close_ssk(sk, ssk, mptcp_subflow_ctx(ssk)); | ||
418 | - MPTCP_INC_STATS(sock_net(sk), MPTCP_MIB_RMSUBFLOW); | ||
419 | -release_sock: | ||
420 | + err = userspace_pm_subflow_closed(msk, &addr_l, &addr_r); | ||
421 | release_sock(sk); | ||
422 | + if (err) | ||
423 | + GENL_SET_ERR_MSG(info, "subflow not found"); | ||
424 | |||
425 | destroy_err: | ||
426 | sock_put(sk); | ||
427 | return err; | ||
428 | } | ||
429 | |||
430 | +static int userspace_pm_set_priority(struct mptcp_sock *msk, | ||
431 | + struct mptcp_pm_addr_entry *local, | ||
432 | + struct mptcp_addr_info *remote) | ||
433 | +{ | ||
434 | + struct mptcp_pm_addr_entry *entry; | ||
435 | + u8 bkup = 0; | ||
436 | + | ||
437 | + if (local->flags & MPTCP_PM_ADDR_FLAG_BACKUP) | ||
438 | + bkup = 1; | ||
439 | + | ||
440 | + spin_lock_bh(&msk->pm.lock); | ||
441 | + entry = mptcp_userspace_pm_lookup_addr(msk, &local->addr); | ||
442 | + if (entry) { | ||
443 | + if (bkup) | ||
444 | + entry->flags |= MPTCP_PM_ADDR_FLAG_BACKUP; | ||
445 | + else | ||
446 | + entry->flags &= ~MPTCP_PM_ADDR_FLAG_BACKUP; | ||
447 | + } | ||
448 | + spin_unlock_bh(&msk->pm.lock); | ||
449 | + | ||
450 | + return mptcp_pm_nl_mp_prio_send_ack(msk, &local->addr, remote, bkup); | ||
451 | +} | ||
452 | + | ||
453 | int mptcp_userspace_pm_set_flags(struct mptcp_pm_addr_entry *local, | ||
454 | struct genl_info *info) | ||
455 | { | ||
456 | struct mptcp_addr_info rem = { .family = AF_UNSPEC, }; | ||
457 | - struct mptcp_pm_addr_entry *entry; | ||
458 | struct nlattr *attr, *attr_rem; | ||
459 | struct mptcp_sock *msk; | ||
460 | int ret = -EINVAL; | ||
461 | struct sock *sk; | ||
462 | - u8 bkup = 0; | ||
463 | |||
464 | if (GENL_REQ_ATTR_CHECK(info, MPTCP_PM_ATTR_ADDR_REMOTE)) | ||
465 | return ret; | ||
466 | @@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_set_flags(struct mptcp_pm_addr_entry *local, | ||
467 | goto set_flags_err; | ||
468 | } | ||
469 | |||
470 | - if (local->flags & MPTCP_PM_ADDR_FLAG_BACKUP) | ||
471 | - bkup = 1; | ||
472 | - | ||
473 | - spin_lock_bh(&msk->pm.lock); | ||
474 | - entry = mptcp_userspace_pm_lookup_addr(msk, &local->addr); | ||
475 | - if (entry) { | ||
476 | - if (bkup) | ||
477 | - entry->flags |= MPTCP_PM_ADDR_FLAG_BACKUP; | ||
478 | - else | ||
479 | - entry->flags &= ~MPTCP_PM_ADDR_FLAG_BACKUP; | ||
480 | - } | ||
481 | - spin_unlock_bh(&msk->pm.lock); | ||
482 | - | ||
483 | lock_sock(sk); | ||
484 | - ret = mptcp_pm_nl_mp_prio_send_ack(msk, &local->addr, &rem, bkup); | ||
485 | + ret = userspace_pm_set_priority(msk, local, &rem); | ||
486 | release_sock(sk); | ||
487 | |||
488 | /* mptcp_pm_nl_mp_prio_send_ack() only fails in one case */ | ||
489 | @@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_addr(u8 id, struct mptcp_pm_addr_entry *addr, | 62 | @@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_addr(u8 id, struct mptcp_pm_addr_entry *addr, |
490 | sock_put(sk); | 63 | sock_put(sk); |
491 | return ret; | 64 | return ret; |
492 | } | 65 | } |
493 | + | 66 | + |
494 | +static struct mptcp_pm_ops mptcp_userspace_pm = { | 67 | +static struct mptcp_pm_ops mptcp_userspace_pm = { |
495 | + .address_announced = userspace_pm_address_announced, | 68 | + .get_local_id = mptcp_userspace_pm_get_local_id, |
496 | + .address_removed = userspace_pm_address_removed, | 69 | + .get_priority = mptcp_userspace_pm_get_priority, |
497 | + .subflow_established = userspace_pm_subflow_established, | ||
498 | + .subflow_closed = userspace_pm_subflow_closed, | ||
499 | + .get_local_id = userspace_pm_get_local_id, | ||
500 | + .get_priority = userspace_pm_get_priority, | ||
501 | + .set_priority = userspace_pm_set_priority, | ||
502 | + .type = MPTCP_PM_TYPE_USERSPACE, | 70 | + .type = MPTCP_PM_TYPE_USERSPACE, |
503 | + .owner = THIS_MODULE, | 71 | + .owner = THIS_MODULE, |
504 | +}; | 72 | +}; |
505 | + | 73 | + |
506 | +void __init mptcp_userspace_pm_init(void) | 74 | +void __init mptcp_userspace_pm_init(void) |
... | ... | diff view generated by jsdifflib |
... | ... | ||
---|---|---|---|
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_pm_initialize() and mptcp_pm_release(), to set and release | 4 | functions mptcp_pm_initialize() and mptcp_pm_release(), to set and release |
5 | this pointer. mptcp_pm_initialize() is invoked in mptcp_pm_data_reset(), | 5 | this pointer. mptcp_pm_initialize() is invoked in mptcp_pm_data_reset(), |
6 | while mptcp_pm_release() 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 | ||
9 | the pm_type sysctl, and then called into their respective interfaces | ||
10 | through "ops" of "msk->pm". | ||
11 | |||
12 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> | 8 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> |
13 | --- | 9 | --- |
14 | net/mptcp/pm.c | 49 ++++++++++++++++++++++++++++++++++++++-- | 10 | net/mptcp/pm.c | 33 +++++++++++++++++++++++++++++++++ |
15 | net/mptcp/pm_netlink.c | 11 ++++++++- | 11 | net/mptcp/protocol.c | 1 + |
16 | net/mptcp/pm_userspace.c | 28 +++++++++++++++++------ | 12 | net/mptcp/protocol.h | 3 +++ |
17 | net/mptcp/protocol.c | 10 +++++++- | 13 | 3 files changed, 37 insertions(+) |
18 | net/mptcp/protocol.h | 3 +++ | ||
19 | 5 files changed, 90 insertions(+), 11 deletions(-) | ||
20 | 14 | ||
21 | diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c | 15 | diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c |
22 | index XXXXXXX..XXXXXXX 100644 | 16 | index XXXXXXX..XXXXXXX 100644 |
23 | --- a/net/mptcp/pm.c | 17 | --- a/net/mptcp/pm.c |
24 | +++ b/net/mptcp/pm.c | 18 | +++ b/net/mptcp/pm.c |
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) | ||
49 | { | ||
50 | pr_debug("msk=%p\n", msk); | ||
51 | |||
52 | - if (msk->token) | ||
53 | + if (msk->token) { | ||
54 | mptcp_event(MPTCP_EVENT_CLOSED, msk, NULL, GFP_KERNEL); | ||
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) | ||
62 | @@ -XXX,XX +XXX,XX @@ void mptcp_pm_data_reset(struct mptcp_sock *msk) | 19 | @@ -XXX,XX +XXX,XX @@ void mptcp_pm_data_reset(struct mptcp_sock *msk) |
63 | WRITE_ONCE(pm->work_pending, 0); | 20 | pm->rm_list_tx.nr = 0; |
64 | WRITE_ONCE(pm->accept_addr, 0); | 21 | pm->rm_list_rx.nr = 0; |
65 | WRITE_ONCE(pm->accept_subflow, 0); | 22 | WRITE_ONCE(pm->pm_type, pm_type); |
66 | + | 23 | + rcu_read_lock(); |
67 | + rcu_read_lock(); | 24 | + mptcp_pm_initialize(msk, mptcp_pm_find(pm_type)); |
68 | + mptcp_pm_initialize(msk, mptcp_pm_find(pm_type)); | 25 | + rcu_read_unlock(); |
69 | + rcu_read_unlock(); | 26 | |
70 | } | 27 | if (pm_type == MPTCP_PM_TYPE_KERNEL) { |
71 | 28 | bool subflows_allowed = !!mptcp_pm_get_subflows_max(msk); | |
72 | WRITE_ONCE(pm->addr_signal, 0); | ||
73 | @@ -XXX,XX +XXX,XX @@ void mptcp_pm_unregister(struct mptcp_pm_ops *pm) | 29 | @@ -XXX,XX +XXX,XX @@ void mptcp_pm_unregister(struct mptcp_pm_ops *pm) |
74 | list_del_rcu(&pm->list); | 30 | list_del_rcu(&pm->list); |
75 | spin_unlock(&mptcp_pm_list_lock); | 31 | spin_unlock(&mptcp_pm_list_lock); |
76 | } | 32 | } |
77 | + | 33 | + |
... | ... | ||
102 | + if (pm->release) | 58 | + if (pm->release) |
103 | + pm->release(msk); | 59 | + pm->release(msk); |
104 | + | 60 | + |
105 | + bpf_module_put(pm, pm->owner); | 61 | + bpf_module_put(pm, pm->owner); |
106 | +} | 62 | +} |
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 | |||
144 | diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c | ||
145 | index XXXXXXX..XXXXXXX 100644 | ||
146 | --- a/net/mptcp/pm_userspace.c | ||
147 | +++ b/net/mptcp/pm_userspace.c | ||
148 | @@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
149 | if (new_entry.addr.port == msk_sport) | ||
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) | ||
162 | { | ||
163 | - return userspace_pm_get_priority(msk, skc); | ||
164 | + return msk->pm.ops && msk->pm.ops->get_priority ? | ||
165 | + msk->pm.ops->get_priority(msk, skc) : | ||
166 | + userspace_pm_get_priority(msk, skc); | ||
167 | } | ||
168 | |||
169 | static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *info) | ||
170 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_announce_doit(struct sk_buff *skb, struct genl_info *info) | ||
171 | } | ||
172 | |||
173 | lock_sock(sk); | ||
174 | - err = userspace_pm_address_announced(msk, &addr_val); | ||
175 | + err = msk->pm.ops && msk->pm.ops->address_announced ? | ||
176 | + msk->pm.ops->address_announced(msk, &addr_val) : | ||
177 | + userspace_pm_address_announced(msk, &addr_val); | ||
178 | release_sock(sk); | ||
179 | if (err) | ||
180 | NL_SET_ERR_MSG_ATTR(info->extack, addr, | ||
181 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_remove_doit(struct sk_buff *skb, struct genl_info *info) | ||
182 | sk = (struct sock *)msk; | ||
183 | |||
184 | lock_sock(sk); | ||
185 | - err = userspace_pm_address_removed(msk, id_val); | ||
186 | + err = msk->pm.ops && msk->pm.ops->address_removed ? | ||
187 | + msk->pm.ops->address_removed(msk, id_val) : | ||
188 | + userspace_pm_address_removed(msk, id_val); | ||
189 | release_sock(sk); | ||
190 | if (err) | ||
191 | NL_SET_ERR_MSG_ATTR_FMT(info->extack, id, | ||
192 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_subflow_create_doit(struct sk_buff *skb, struct genl_info *info) | ||
193 | } | ||
194 | |||
195 | lock_sock(sk); | ||
196 | - err = userspace_pm_subflow_established(msk, &entry, &addr_r); | ||
197 | + err = msk->pm.ops && msk->pm.ops->subflow_established ? | ||
198 | + msk->pm.ops->subflow_established(msk, &entry, &addr_r) : | ||
199 | + userspace_pm_subflow_established(msk, &entry, &addr_r); | ||
200 | release_sock(sk); | ||
201 | |||
202 | if (err) | ||
203 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_nl_subflow_destroy_doit(struct sk_buff *skb, struct genl_info *info | ||
204 | } | ||
205 | |||
206 | lock_sock(sk); | ||
207 | - err = userspace_pm_subflow_closed(msk, &addr_l, &addr_r); | ||
208 | + err = msk->pm.ops && msk->pm.ops->subflow_closed ? | ||
209 | + msk->pm.ops->subflow_closed(msk, &addr_l, &addr_r) : | ||
210 | + userspace_pm_subflow_closed(msk, &addr_l, &addr_r); | ||
211 | release_sock(sk); | ||
212 | if (err) | ||
213 | GENL_SET_ERR_MSG(info, "subflow not found"); | ||
214 | @@ -XXX,XX +XXX,XX @@ int mptcp_userspace_pm_set_flags(struct mptcp_pm_addr_entry *local, | ||
215 | } | ||
216 | |||
217 | lock_sock(sk); | ||
218 | - ret = userspace_pm_set_priority(msk, local, &rem); | ||
219 | + ret = msk->pm.ops && msk->pm.ops->set_priority ? | ||
220 | + msk->pm.ops->set_priority(msk, local, &rem) : | ||
221 | + userspace_pm_set_priority(msk, local, &rem); | ||
222 | release_sock(sk); | ||
223 | |||
224 | /* mptcp_pm_nl_mp_prio_send_ack() only fails in one case */ | ||
225 | diff --git a/net/mptcp/protocol.c b/net/mptcp/protocol.c | 63 | diff --git a/net/mptcp/protocol.c b/net/mptcp/protocol.c |
226 | index XXXXXXX..XXXXXXX 100644 | 64 | index XXXXXXX..XXXXXXX 100644 |
227 | --- a/net/mptcp/protocol.c | 65 | --- a/net/mptcp/protocol.c |
228 | +++ b/net/mptcp/protocol.c | 66 | +++ b/net/mptcp/protocol.c |
229 | @@ -XXX,XX +XXX,XX @@ static void __mptcp_destroy_sock(struct sock *sk) | 67 | @@ -XXX,XX +XXX,XX @@ static void __mptcp_destroy_sock(struct sock *sk) |
... | ... | ||
232 | mptcp_release_sched(msk); | 70 | mptcp_release_sched(msk); |
233 | + mptcp_pm_release(msk); | 71 | + mptcp_pm_release(msk); |
234 | 72 | ||
235 | sk->sk_prot->destroy(sk); | 73 | sk->sk_prot->destroy(sk); |
236 | 74 | ||
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: | ||
273 | diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h | 75 | diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h |
274 | index XXXXXXX..XXXXXXX 100644 | 76 | index XXXXXXX..XXXXXXX 100644 |
275 | --- a/net/mptcp/protocol.h | 77 | --- a/net/mptcp/protocol.h |
276 | +++ b/net/mptcp/protocol.h | 78 | +++ b/net/mptcp/protocol.h |
277 | @@ -XXX,XX +XXX,XX @@ struct mptcp_pm_data { | 79 | @@ -XXX,XX +XXX,XX @@ struct mptcp_pm_data { |
... | ... | diff view generated by jsdifflib |
New patch | |||
---|---|---|---|
1 | From: Geliang Tang <tanggeliang@kylinos.cn> | ||
1 | 2 | ||
3 | Now mptcp_pm_nl_get_local_id() and mptcp_userspace_pm_get_local_id() | ||
4 | helpers can be dropped, and mptcp_pm_get_local_id() can directly invoke | ||
5 | get_local_id() interface through "ops" of "msk->pm". | ||
6 | |||
7 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> | ||
8 | --- | ||
9 | net/mptcp/pm.c | 6 +++--- | ||
10 | net/mptcp/pm_netlink.c | 4 ++-- | ||
11 | net/mptcp/pm_userspace.c | 4 ++-- | ||
12 | net/mptcp/protocol.h | 4 ---- | ||
13 | 4 files changed, 7 insertions(+), 11 deletions(-) | ||
14 | |||
15 | diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c | ||
16 | index XXXXXXX..XXXXXXX 100644 | ||
17 | --- a/net/mptcp/pm.c | ||
18 | +++ b/net/mptcp/pm.c | ||
19 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc) | ||
20 | skc_local.addr.id = 0; | ||
21 | skc_local.flags = MPTCP_PM_ADDR_FLAG_IMPLICIT; | ||
22 | |||
23 | + if (!msk->pm.ops || !msk->pm.ops->get_local_id) | ||
24 | + return -ENOTSUPP; | ||
25 | mptcp_pm_param_set_contexts(¶m, &skc_local, NULL); | ||
26 | - if (mptcp_pm_is_userspace(msk)) | ||
27 | - return mptcp_userspace_pm_get_local_id(msk, ¶m); | ||
28 | - return mptcp_pm_nl_get_local_id(msk, ¶m); | ||
29 | + return msk->pm.ops->get_local_id(msk, ¶m); | ||
30 | } | ||
31 | |||
32 | bool mptcp_pm_is_backup(struct mptcp_sock *msk, struct sock_common *skc) | ||
33 | diff --git a/net/mptcp/pm_netlink.c b/net/mptcp/pm_netlink.c | ||
34 | index XXXXXXX..XXXXXXX 100644 | ||
35 | --- a/net/mptcp/pm_netlink.c | ||
36 | +++ b/net/mptcp/pm_netlink.c | ||
37 | @@ -XXX,XX +XXX,XX @@ static int mptcp_pm_nl_create_listen_socket(struct sock *sk, | ||
38 | return err; | ||
39 | } | ||
40 | |||
41 | -int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, | ||
42 | - struct mptcp_pm_param *param) | ||
43 | +static int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, | ||
44 | + struct mptcp_pm_param *param) | ||
45 | { | ||
46 | struct mptcp_pm_addr_entry *skc = ¶m->entry; | ||
47 | struct mptcp_pm_addr_entry *entry; | ||
48 | diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c | ||
49 | index XXXXXXX..XXXXXXX 100644 | ||
50 | --- a/net/mptcp/pm_userspace.c | ||
51 | +++ b/net/mptcp/pm_userspace.c | ||
52 | @@ -XXX,XX +XXX,XX @@ mptcp_userspace_pm_lookup_addr_by_id(struct mptcp_sock *msk, unsigned int id) | ||
53 | return NULL; | ||
54 | } | ||
55 | |||
56 | -int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
57 | - struct mptcp_pm_param *param) | ||
58 | +static int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
59 | + struct mptcp_pm_param *param) | ||
60 | { | ||
61 | __be16 msk_sport = ((struct inet_sock *) | ||
62 | inet_sk((struct sock *)msk))->inet_sport; | ||
63 | diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h | ||
64 | index XXXXXXX..XXXXXXX 100644 | ||
65 | --- a/net/mptcp/protocol.h | ||
66 | +++ b/net/mptcp/protocol.h | ||
67 | @@ -XXX,XX +XXX,XX @@ void mptcp_pm_param_set_contexts(struct mptcp_pm_param *param, | ||
68 | const struct mptcp_pm_addr_entry *entry, | ||
69 | const struct mptcp_addr_info *addr); | ||
70 | int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc); | ||
71 | -int mptcp_pm_nl_get_local_id(struct mptcp_sock *msk, | ||
72 | - struct mptcp_pm_param *param); | ||
73 | -int mptcp_userspace_pm_get_local_id(struct mptcp_sock *msk, | ||
74 | - struct mptcp_pm_param *param); | ||
75 | bool mptcp_pm_is_backup(struct mptcp_sock *msk, struct sock_common *skc); | ||
76 | bool mptcp_pm_nl_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc); | ||
77 | bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc); | ||
78 | -- | ||
79 | 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 a new path manager type: MPTCP_PM_TYPE_BPF_USERSPACE, | 3 | Now mptcp_pm_nl_is_backup() and mptcp_userspace_pm_is_backup() |
4 | which is used for the userspace path manager implemented in BPF. This | 4 | helpers can be dropped, and mptcp_pm_is_backup() can directly |
5 | path manager is also a userspace path manager, so add this new type in | 5 | invoke get_priority() interface through "ops" of "msk->pm". |
6 | mptcp_pm_is_userspace() too. | ||
7 | 6 | ||
8 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> | 7 | Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> |
9 | --- | 8 | --- |
10 | net/mptcp/protocol.h | 6 +++++- | 9 | net/mptcp/pm.c | 9 +++++---- |
11 | 1 file changed, 5 insertions(+), 1 deletion(-) | 10 | net/mptcp/pm_netlink.c | 8 -------- |
11 | net/mptcp/pm_userspace.c | 9 --------- | ||
12 | net/mptcp/protocol.h | 2 -- | ||
13 | 4 files changed, 5 insertions(+), 23 deletions(-) | ||
12 | 14 | ||
15 | diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c | ||
16 | index XXXXXXX..XXXXXXX 100644 | ||
17 | --- a/net/mptcp/pm.c | ||
18 | +++ b/net/mptcp/pm.c | ||
19 | @@ -XXX,XX +XXX,XX @@ int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc) | ||
20 | bool mptcp_pm_is_backup(struct mptcp_sock *msk, struct sock_common *skc) | ||
21 | { | ||
22 | struct mptcp_addr_info skc_local; | ||
23 | + struct mptcp_pm_param param; | ||
24 | |||
25 | mptcp_local_address((struct sock_common *)skc, &skc_local); | ||
26 | |||
27 | - if (mptcp_pm_is_userspace(msk)) | ||
28 | - return mptcp_userspace_pm_is_backup(msk, &skc_local); | ||
29 | - | ||
30 | - return mptcp_pm_nl_is_backup(msk, &skc_local); | ||
31 | + if (!msk->pm.ops || !msk->pm.ops->get_priority) | ||
32 | + return -ENOTSUPP; | ||
33 | + mptcp_pm_param_set_contexts(¶m, NULL, &skc_local); | ||
34 | + return msk->pm.ops->get_priority(msk, ¶m); | ||
35 | } | ||
36 | |||
37 | static int mptcp_pm_get_addr(u8 id, struct mptcp_pm_addr_entry *addr, | ||
38 | diff --git a/net/mptcp/pm_netlink.c b/net/mptcp/pm_netlink.c | ||
39 | index XXXXXXX..XXXXXXX 100644 | ||
40 | --- a/net/mptcp/pm_netlink.c | ||
41 | +++ b/net/mptcp/pm_netlink.c | ||
42 | @@ -XXX,XX +XXX,XX @@ static bool mptcp_pm_nl_get_priority(struct mptcp_sock *msk, | ||
43 | return backup; | ||
44 | } | ||
45 | |||
46 | -bool mptcp_pm_nl_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc) | ||
47 | -{ | ||
48 | - struct mptcp_pm_param param; | ||
49 | - | ||
50 | - mptcp_pm_param_set_contexts(¶m, NULL, skc); | ||
51 | - return mptcp_pm_nl_get_priority(msk, ¶m); | ||
52 | -} | ||
53 | - | ||
54 | #define MPTCP_PM_CMD_GRP_OFFSET 0 | ||
55 | #define MPTCP_PM_EV_GRP_OFFSET 1 | ||
56 | |||
57 | diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c | ||
58 | index XXXXXXX..XXXXXXX 100644 | ||
59 | --- a/net/mptcp/pm_userspace.c | ||
60 | +++ b/net/mptcp/pm_userspace.c | ||
61 | @@ -XXX,XX +XXX,XX @@ static bool mptcp_userspace_pm_get_priority(struct mptcp_sock *msk, | ||
62 | return backup; | ||
63 | } | ||
64 | |||
65 | -bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, | ||
66 | - struct mptcp_addr_info *skc) | ||
67 | -{ | ||
68 | - struct mptcp_pm_param param; | ||
69 | - | ||
70 | - mptcp_pm_param_set_contexts(¶m, NULL, skc); | ||
71 | - return mptcp_userspace_pm_get_priority(msk, ¶m); | ||
72 | -} | ||
73 | - | ||
74 | static struct mptcp_sock *mptcp_userspace_pm_get_sock(const struct genl_info *info) | ||
75 | { | ||
76 | struct mptcp_sock *msk; | ||
13 | diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h | 77 | diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h |
14 | index XXXXXXX..XXXXXXX 100644 | 78 | index XXXXXXX..XXXXXXX 100644 |
15 | --- a/net/mptcp/protocol.h | 79 | --- a/net/mptcp/protocol.h |
16 | +++ b/net/mptcp/protocol.h | 80 | +++ b/net/mptcp/protocol.h |
17 | @@ -XXX,XX +XXX,XX @@ enum mptcp_pm_status { | 81 | @@ -XXX,XX +XXX,XX @@ void mptcp_pm_param_set_contexts(struct mptcp_pm_param *param, |
18 | enum mptcp_pm_type { | 82 | const struct mptcp_addr_info *addr); |
19 | MPTCP_PM_TYPE_KERNEL = 0, | 83 | int mptcp_pm_get_local_id(struct mptcp_sock *msk, struct sock_common *skc); |
20 | MPTCP_PM_TYPE_USERSPACE, | 84 | bool mptcp_pm_is_backup(struct mptcp_sock *msk, struct sock_common *skc); |
21 | + MPTCP_PM_TYPE_BPF_USERSPACE, | 85 | -bool mptcp_pm_nl_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc); |
22 | 86 | -bool mptcp_userspace_pm_is_backup(struct mptcp_sock *msk, struct mptcp_addr_info *skc); | |
23 | __MPTCP_PM_TYPE_NR, | 87 | int mptcp_pm_nl_dump_addr(struct sk_buff *msg, |
24 | __MPTCP_PM_TYPE_MAX = __MPTCP_PM_TYPE_NR - 1, | 88 | struct netlink_callback *cb); |
25 | @@ -XXX,XX +XXX,XX @@ static inline bool mptcp_pm_should_rm_signal(struct mptcp_sock *msk) | 89 | int mptcp_userspace_pm_dump_addr(struct sk_buff *msg, |
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 | -- | 90 | -- |
38 | 2.43.0 | 91 | 2.43.0 | diff view generated by jsdifflib |