1 // SPDX-License-Identifier: GPL-2.0 2 #include <linux/in.h> 3 #include <linux/inet.h> 4 #include <linux/list.h> 5 #include <linux/module.h> 6 #include <linux/net.h> 7 #include <linux/proc_fs.h> 8 #include <linux/rculist.h> 9 #include <linux/seq_file.h> 10 #include <linux/socket.h> 11 #include <net/inet_sock.h> 12 #include <net/kcm.h> 13 #include <net/net_namespace.h> 14 #include <net/netns/generic.h> 15 #include <net/tcp.h> 16 17 #ifdef CONFIG_PROC_FS 18 static struct kcm_mux *kcm_get_first(struct seq_file *seq) 19 { 20 struct net *net = seq_file_net(seq); 21 struct kcm_net *knet = net_generic(net, kcm_net_id); 22 23 return list_first_or_null_rcu(&knet->mux_list, 24 struct kcm_mux, kcm_mux_list); 25 } 26 27 static struct kcm_mux *kcm_get_next(struct kcm_mux *mux) 28 { 29 struct kcm_net *knet = mux->knet; 30 31 return list_next_or_null_rcu(&knet->mux_list, &mux->kcm_mux_list, 32 struct kcm_mux, kcm_mux_list); 33 } 34 35 static struct kcm_mux *kcm_get_idx(struct seq_file *seq, loff_t pos) 36 { 37 struct net *net = seq_file_net(seq); 38 struct kcm_net *knet = net_generic(net, kcm_net_id); 39 struct kcm_mux *m; 40 41 list_for_each_entry_rcu(m, &knet->mux_list, kcm_mux_list) { 42 if (!pos) 43 return m; 44 --pos; 45 } 46 return NULL; 47 } 48 49 static void *kcm_seq_next(struct seq_file *seq, void *v, loff_t *pos) 50 { 51 void *p; 52 53 if (v == SEQ_START_TOKEN) 54 p = kcm_get_first(seq); 55 else 56 p = kcm_get_next(v); 57 ++*pos; 58 return p; 59 } 60 61 static void *kcm_seq_start(struct seq_file *seq, loff_t *pos) 62 __acquires(rcu) 63 { 64 rcu_read_lock(); 65 66 if (!*pos) 67 return SEQ_START_TOKEN; 68 else 69 return kcm_get_idx(seq, *pos - 1); 70 } 71 72 static void kcm_seq_stop(struct seq_file *seq, void *v) 73 __releases(rcu) 74 { 75 rcu_read_unlock(); 76 } 77 78 struct kcm_proc_mux_state { 79 struct seq_net_private p; 80 int idx; 81 }; 82 83 static void kcm_format_mux_header(struct seq_file *seq) 84 { 85 struct net *net = seq_file_net(seq); 86 struct kcm_net *knet = net_generic(net, kcm_net_id); 87 88 seq_printf(seq, 89 "*** KCM statistics (%d MUX) ****\n", 90 knet->count); 91 92 seq_printf(seq, 93 "%-14s %-10s %-16s %-10s %-16s %-8s %-8s %-8s %-8s %s", 94 "Object", 95 "RX-Msgs", 96 "RX-Bytes", 97 "TX-Msgs", 98 "TX-Bytes", 99 "Recv-Q", 100 "Rmem", 101 "Send-Q", 102 "Smem", 103 "Status"); 104 105 /* XXX: pdsts header stuff here */ 106 seq_puts(seq, "\n"); 107 } 108 109 static void kcm_format_sock(struct kcm_sock *kcm, struct seq_file *seq, 110 int i, int *len) 111 { 112 seq_printf(seq, 113 " kcm-%-7u %-10llu %-16llu %-10llu %-16llu %-8d %-8d %-8d %-8s ", 114 kcm->index, 115 kcm->stats.rx_msgs, 116 kcm->stats.rx_bytes, 117 kcm->stats.tx_msgs, 118 kcm->stats.tx_bytes, 119 kcm->sk.sk_receive_queue.qlen, 120 sk_rmem_alloc_get(&kcm->sk), 121 kcm->sk.sk_write_queue.qlen, 122 "-"); 123 124 if (kcm->tx_psock) 125 seq_printf(seq, "Psck-%u ", kcm->tx_psock->index); 126 127 if (kcm->tx_wait) 128 seq_puts(seq, "TxWait "); 129 130 if (kcm->tx_wait_more) 131 seq_puts(seq, "WMore "); 132 133 if (kcm->rx_wait) 134 seq_puts(seq, "RxWait "); 135 136 seq_puts(seq, "\n"); 137 } 138 139 static void kcm_format_psock(struct kcm_psock *psock, struct seq_file *seq, 140 int i, int *len) 141 { 142 seq_printf(seq, 143 " psock-%-5u %-10llu %-16llu %-10llu %-16llu %-8d %-8d %-8d %-8d ", 144 psock->index, 145 psock->strp.stats.msgs, 146 psock->strp.stats.bytes, 147 psock->stats.tx_msgs, 148 psock->stats.tx_bytes, 149 psock->sk->sk_receive_queue.qlen, 150 atomic_read(&psock->sk->sk_rmem_alloc), 151 psock->sk->sk_write_queue.qlen, 152 refcount_read(&psock->sk->sk_wmem_alloc)); 153 154 if (psock->done) 155 seq_puts(seq, "Done "); 156 157 if (psock->tx_stopped) 158 seq_puts(seq, "TxStop "); 159 160 if (psock->strp.stopped) 161 seq_puts(seq, "RxStop "); 162 163 if (psock->tx_kcm) 164 seq_printf(seq, "Rsvd-%d ", psock->tx_kcm->index); 165 166 if (!psock->strp.paused && !psock->ready_rx_msg) { 167 if (psock->sk->sk_receive_queue.qlen) { 168 if (psock->strp.need_bytes) 169 seq_printf(seq, "RxWait=%u ", 170 psock->strp.need_bytes); 171 else 172 seq_printf(seq, "RxWait "); 173 } 174 } else { 175 if (psock->strp.paused) 176 seq_puts(seq, "RxPause "); 177 178 if (psock->ready_rx_msg) 179 seq_puts(seq, "RdyRx "); 180 } 181 182 seq_puts(seq, "\n"); 183 } 184 185 static void 186 kcm_format_mux(struct kcm_mux *mux, loff_t idx, struct seq_file *seq) 187 { 188 int i, len; 189 struct kcm_sock *kcm; 190 struct kcm_psock *psock; 191 192 /* mux information */ 193 seq_printf(seq, 194 "%-6s%-8s %-10llu %-16llu %-10llu %-16llu %-8s %-8s %-8s %-8s ", 195 "mux", "", 196 mux->stats.rx_msgs, 197 mux->stats.rx_bytes, 198 mux->stats.tx_msgs, 199 mux->stats.tx_bytes, 200 "-", "-", "-", "-"); 201 202 seq_printf(seq, "KCMs: %d, Psocks %d\n", 203 mux->kcm_socks_cnt, mux->psocks_cnt); 204 205 /* kcm sock information */ 206 i = 0; 207 spin_lock_bh(&mux->lock); 208 list_for_each_entry(kcm, &mux->kcm_socks, kcm_sock_list) { 209 kcm_format_sock(kcm, seq, i, &len); 210 i++; 211 } 212 i = 0; 213 list_for_each_entry(psock, &mux->psocks, psock_list) { 214 kcm_format_psock(psock, seq, i, &len); 215 i++; 216 } 217 spin_unlock_bh(&mux->lock); 218 } 219 220 static int kcm_seq_show(struct seq_file *seq, void *v) 221 { 222 struct kcm_proc_mux_state *mux_state; 223 224 mux_state = seq->private; 225 if (v == SEQ_START_TOKEN) { 226 mux_state->idx = 0; 227 kcm_format_mux_header(seq); 228 } else { 229 kcm_format_mux(v, mux_state->idx, seq); 230 mux_state->idx++; 231 } 232 return 0; 233 } 234 235 static const struct seq_operations kcm_seq_ops = { 236 .show = kcm_seq_show, 237 .start = kcm_seq_start, 238 .next = kcm_seq_next, 239 .stop = kcm_seq_stop, 240 }; 241 242 static int kcm_stats_seq_show(struct seq_file *seq, void *v) 243 { 244 struct kcm_psock_stats psock_stats; 245 struct kcm_mux_stats mux_stats; 246 struct strp_aggr_stats strp_stats; 247 struct kcm_mux *mux; 248 struct kcm_psock *psock; 249 struct net *net = seq->private; 250 struct kcm_net *knet = net_generic(net, kcm_net_id); 251 252 memset(&mux_stats, 0, sizeof(mux_stats)); 253 memset(&psock_stats, 0, sizeof(psock_stats)); 254 memset(&strp_stats, 0, sizeof(strp_stats)); 255 256 mutex_lock(&knet->mutex); 257 258 aggregate_mux_stats(&knet->aggregate_mux_stats, &mux_stats); 259 aggregate_psock_stats(&knet->aggregate_psock_stats, 260 &psock_stats); 261 aggregate_strp_stats(&knet->aggregate_strp_stats, 262 &strp_stats); 263 264 list_for_each_entry(mux, &knet->mux_list, kcm_mux_list) { 265 spin_lock_bh(&mux->lock); 266 aggregate_mux_stats(&mux->stats, &mux_stats); 267 aggregate_psock_stats(&mux->aggregate_psock_stats, 268 &psock_stats); 269 aggregate_strp_stats(&mux->aggregate_strp_stats, 270 &strp_stats); 271 list_for_each_entry(psock, &mux->psocks, psock_list) { 272 aggregate_psock_stats(&psock->stats, &psock_stats); 273 save_strp_stats(&psock->strp, &strp_stats); 274 } 275 276 spin_unlock_bh(&mux->lock); 277 } 278 279 mutex_unlock(&knet->mutex); 280 281 seq_printf(seq, 282 "%-8s %-10s %-16s %-10s %-16s %-10s %-10s %-10s %-10s %-10s\n", 283 "MUX", 284 "RX-Msgs", 285 "RX-Bytes", 286 "TX-Msgs", 287 "TX-Bytes", 288 "TX-Retries", 289 "Attach", 290 "Unattach", 291 "UnattchRsvd", 292 "RX-RdyDrops"); 293 294 seq_printf(seq, 295 "%-8s %-10llu %-16llu %-10llu %-16llu %-10u %-10u %-10u %-10u %-10u\n", 296 "", 297 mux_stats.rx_msgs, 298 mux_stats.rx_bytes, 299 mux_stats.tx_msgs, 300 mux_stats.tx_bytes, 301 mux_stats.tx_retries, 302 mux_stats.psock_attach, 303 mux_stats.psock_unattach_rsvd, 304 mux_stats.psock_unattach, 305 mux_stats.rx_ready_drops); 306 307 seq_printf(seq, 308 "%-8s %-10s %-16s %-10s %-16s %-10s %-10s %-10s %-10s %-10s %-10s %-10s %-10s %-10s %-10s %-10s\n", 309 "Psock", 310 "RX-Msgs", 311 "RX-Bytes", 312 "TX-Msgs", 313 "TX-Bytes", 314 "Reserved", 315 "Unreserved", 316 "RX-Aborts", 317 "RX-Intr", 318 "RX-Unrecov", 319 "RX-MemFail", 320 "RX-NeedMor", 321 "RX-BadLen", 322 "RX-TooBig", 323 "RX-Timeout", 324 "TX-Aborts"); 325 326 seq_printf(seq, 327 "%-8s %-10llu %-16llu %-10llu %-16llu %-10llu %-10llu %-10u %-10u %-10u %-10u %-10u %-10u %-10u %-10u %-10u\n", 328 "", 329 strp_stats.msgs, 330 strp_stats.bytes, 331 psock_stats.tx_msgs, 332 psock_stats.tx_bytes, 333 psock_stats.reserved, 334 psock_stats.unreserved, 335 strp_stats.aborts, 336 strp_stats.interrupted, 337 strp_stats.unrecov_intr, 338 strp_stats.mem_fail, 339 strp_stats.need_more_hdr, 340 strp_stats.bad_hdr_len, 341 strp_stats.msg_too_big, 342 strp_stats.msg_timeouts, 343 psock_stats.tx_aborts); 344 345 return 0; 346 } 347 348 static int kcm_proc_init_net(struct net *net) 349 { 350 if (!proc_create_net_single("kcm_stats", 0444, net->proc_net, 351 kcm_stats_seq_show, NULL)) 352 goto out_kcm_stats; 353 354 if (!proc_create_net("kcm", 0444, net->proc_net, &kcm_seq_ops, 355 sizeof(struct kcm_proc_mux_state))) 356 goto out_kcm; 357 358 return 0; 359 360 out_kcm: 361 remove_proc_entry("kcm_stats", net->proc_net); 362 out_kcm_stats: 363 return -ENOMEM; 364 } 365 366 static void kcm_proc_exit_net(struct net *net) 367 { 368 remove_proc_entry("kcm", net->proc_net); 369 remove_proc_entry("kcm_stats", net->proc_net); 370 } 371 372 static struct pernet_operations kcm_net_ops = { 373 .init = kcm_proc_init_net, 374 .exit = kcm_proc_exit_net, 375 }; 376 377 int __init kcm_proc_init(void) 378 { 379 return register_pernet_subsys(&kcm_net_ops); 380 } 381 382 void __exit kcm_proc_exit(void) 383 { 384 unregister_pernet_subsys(&kcm_net_ops); 385 } 386 387 #endif /* CONFIG_PROC_FS */ 388