2 #include <linux/inet.h>
3 #include <linux/list.h>
4 #include <linux/module.h>
6 #include <linux/proc_fs.h>
7 #include <linux/rculist.h>
8 #include <linux/seq_file.h>
9 #include <linux/socket.h>
10 #include <net/inet_sock.h>
12 #include <net/net_namespace.h>
13 #include <net/netns/generic.h>
17 struct kcm_seq_muxinfo {
19 const struct file_operations *seq_fops;
20 const struct seq_operations seq_ops;
23 static struct kcm_mux *kcm_get_first(struct seq_file *seq)
25 struct net *net = seq_file_net(seq);
26 struct kcm_net *knet = net_generic(net, kcm_net_id);
28 return list_first_or_null_rcu(&knet->mux_list,
29 struct kcm_mux, kcm_mux_list);
32 static struct kcm_mux *kcm_get_next(struct kcm_mux *mux)
34 struct kcm_net *knet = mux->knet;
36 return list_next_or_null_rcu(&knet->mux_list, &mux->kcm_mux_list,
37 struct kcm_mux, kcm_mux_list);
40 static struct kcm_mux *kcm_get_idx(struct seq_file *seq, loff_t pos)
42 struct net *net = seq_file_net(seq);
43 struct kcm_net *knet = net_generic(net, kcm_net_id);
46 list_for_each_entry_rcu(m, &knet->mux_list, kcm_mux_list) {
54 static void *kcm_seq_next(struct seq_file *seq, void *v, loff_t *pos)
58 if (v == SEQ_START_TOKEN)
59 p = kcm_get_first(seq);
66 static void *kcm_seq_start(struct seq_file *seq, loff_t *pos)
72 return SEQ_START_TOKEN;
74 return kcm_get_idx(seq, *pos - 1);
77 static void kcm_seq_stop(struct seq_file *seq, void *v)
83 struct kcm_proc_mux_state {
84 struct seq_net_private p;
88 static int kcm_seq_open(struct inode *inode, struct file *file)
90 struct kcm_seq_muxinfo *muxinfo = PDE_DATA(inode);
92 return seq_open_net(inode, file, &muxinfo->seq_ops,
93 sizeof(struct kcm_proc_mux_state));
96 static void kcm_format_mux_header(struct seq_file *seq)
98 struct net *net = seq_file_net(seq);
99 struct kcm_net *knet = net_generic(net, kcm_net_id);
102 "*** KCM statistics (%d MUX) ****\n",
106 "%-14s %-10s %-16s %-10s %-16s %-8s %-8s %-8s %-8s %s",
118 /* XXX: pdsts header stuff here */
122 static void kcm_format_sock(struct kcm_sock *kcm, struct seq_file *seq,
126 " kcm-%-7u %-10llu %-16llu %-10llu %-16llu %-8d %-8d %-8d %-8s ",
132 kcm->sk.sk_receive_queue.qlen,
133 sk_rmem_alloc_get(&kcm->sk),
134 kcm->sk.sk_write_queue.qlen,
138 seq_printf(seq, "Psck-%u ", kcm->tx_psock->index);
141 seq_puts(seq, "TxWait ");
143 if (kcm->tx_wait_more)
144 seq_puts(seq, "WMore ");
147 seq_puts(seq, "RxWait ");
152 static void kcm_format_psock(struct kcm_psock *psock, struct seq_file *seq,
156 " psock-%-5u %-10llu %-16llu %-10llu %-16llu %-8d %-8d %-8d %-8d ",
158 psock->stats.rx_msgs,
159 psock->stats.rx_bytes,
160 psock->stats.tx_msgs,
161 psock->stats.tx_bytes,
162 psock->sk->sk_receive_queue.qlen,
163 atomic_read(&psock->sk->sk_rmem_alloc),
164 psock->sk->sk_write_queue.qlen,
165 atomic_read(&psock->sk->sk_wmem_alloc));
168 seq_puts(seq, "Done ");
170 if (psock->tx_stopped)
171 seq_puts(seq, "TxStop ");
173 if (psock->rx_stopped)
174 seq_puts(seq, "RxStop ");
177 seq_printf(seq, "Rsvd-%d ", psock->tx_kcm->index);
179 if (psock->ready_rx_msg)
180 seq_puts(seq, "RdyRx ");
186 kcm_format_mux(struct kcm_mux *mux, loff_t idx, struct seq_file *seq)
189 struct kcm_sock *kcm;
190 struct kcm_psock *psock;
192 /* mux information */
194 "%-6s%-8s %-10llu %-16llu %-10llu %-16llu %-8s %-8s %-8s %-8s ",
202 seq_printf(seq, "KCMs: %d, Psocks %d\n",
203 mux->kcm_socks_cnt, mux->psocks_cnt);
205 /* kcm sock information */
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);
213 list_for_each_entry(psock, &mux->psocks, psock_list) {
214 kcm_format_psock(psock, seq, i, &len);
217 spin_unlock_bh(&mux->lock);
220 static int kcm_seq_show(struct seq_file *seq, void *v)
222 struct kcm_proc_mux_state *mux_state;
224 mux_state = seq->private;
225 if (v == SEQ_START_TOKEN) {
227 kcm_format_mux_header(seq);
229 kcm_format_mux(v, mux_state->idx, seq);
235 static const struct file_operations kcm_seq_fops = {
236 .owner = THIS_MODULE,
237 .open = kcm_seq_open,
240 .release = seq_release_net,
243 static struct kcm_seq_muxinfo kcm_seq_muxinfo = {
245 .seq_fops = &kcm_seq_fops,
247 .show = kcm_seq_show,
248 .start = kcm_seq_start,
249 .next = kcm_seq_next,
250 .stop = kcm_seq_stop,
254 static int kcm_proc_register(struct net *net, struct kcm_seq_muxinfo *muxinfo)
256 struct proc_dir_entry *p;
259 p = proc_create_data(muxinfo->name, S_IRUGO, net->proc_net,
260 muxinfo->seq_fops, muxinfo);
265 EXPORT_SYMBOL(kcm_proc_register);
267 static void kcm_proc_unregister(struct net *net,
268 struct kcm_seq_muxinfo *muxinfo)
270 remove_proc_entry(muxinfo->name, net->proc_net);
272 EXPORT_SYMBOL(kcm_proc_unregister);
274 static int kcm_stats_seq_show(struct seq_file *seq, void *v)
276 struct kcm_psock_stats psock_stats;
277 struct kcm_mux_stats mux_stats;
279 struct kcm_psock *psock;
280 struct net *net = seq->private;
281 struct kcm_net *knet = net_generic(net, kcm_net_id);
283 memset(&mux_stats, 0, sizeof(mux_stats));
284 memset(&psock_stats, 0, sizeof(psock_stats));
286 mutex_lock(&knet->mutex);
288 aggregate_mux_stats(&knet->aggregate_mux_stats, &mux_stats);
289 aggregate_psock_stats(&knet->aggregate_psock_stats,
292 list_for_each_entry_rcu(mux, &knet->mux_list, kcm_mux_list) {
293 spin_lock_bh(&mux->lock);
294 aggregate_mux_stats(&mux->stats, &mux_stats);
295 aggregate_psock_stats(&mux->aggregate_psock_stats,
297 list_for_each_entry(psock, &mux->psocks, psock_list)
298 aggregate_psock_stats(&psock->stats, &psock_stats);
299 spin_unlock_bh(&mux->lock);
302 mutex_unlock(&knet->mutex);
305 "%-8s %-10s %-16s %-10s %-16s %-10s %-10s %-10s %-10s %-10s\n",
318 "%-8s %-10llu %-16llu %-10llu %-16llu %-10u %-10u %-10u %-10u %-10u\n",
324 mux_stats.tx_retries,
325 mux_stats.psock_attach,
326 mux_stats.psock_unattach_rsvd,
327 mux_stats.psock_unattach,
328 mux_stats.rx_ready_drops);
331 "%-8s %-10s %-16s %-10s %-16s %-10s %-10s %-10s %-10s %-10s %-10s %-10s %-10s %-10s\n",
348 "%-8s %-10llu %-16llu %-10llu %-16llu %-10llu %-10llu %-10u %-10u %-10u %-10u %-10u %-10u %-10u\n",
351 psock_stats.rx_bytes,
353 psock_stats.tx_bytes,
354 psock_stats.reserved,
355 psock_stats.unreserved,
356 psock_stats.rx_aborts,
357 psock_stats.rx_mem_fail,
358 psock_stats.rx_need_more_hdr,
359 psock_stats.rx_bad_hdr_len,
360 psock_stats.rx_msg_too_big,
361 psock_stats.rx_msg_timeouts,
362 psock_stats.tx_aborts);
367 static int kcm_stats_seq_open(struct inode *inode, struct file *file)
369 return single_open_net(inode, file, kcm_stats_seq_show);
372 static const struct file_operations kcm_stats_seq_fops = {
373 .owner = THIS_MODULE,
374 .open = kcm_stats_seq_open,
377 .release = single_release_net,
380 static int kcm_proc_init_net(struct net *net)
384 if (!proc_create("kcm_stats", S_IRUGO, net->proc_net,
385 &kcm_stats_seq_fops)) {
390 err = kcm_proc_register(net, &kcm_seq_muxinfo);
397 remove_proc_entry("kcm_stats", net->proc_net);
402 static void kcm_proc_exit_net(struct net *net)
404 kcm_proc_unregister(net, &kcm_seq_muxinfo);
405 remove_proc_entry("kcm_stats", net->proc_net);
408 static struct pernet_operations kcm_net_ops = {
409 .init = kcm_proc_init_net,
410 .exit = kcm_proc_exit_net,
413 int __init kcm_proc_init(void)
415 return register_pernet_subsys(&kcm_net_ops);
418 void __exit kcm_proc_exit(void)
420 unregister_pernet_subsys(&kcm_net_ops);
423 #endif /* CONFIG_PROC_FS */