Blame view
net/dccp/probe.c
4.91 KB
e41542f51 [DCCP]: Introduce... |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
/* * dccp_probe - Observe the DCCP flow with kprobes. * * The idea for this came from Werner Almesberger's umlsim * Copyright (C) 2004, Stephen Hemminger <shemminger@osdl.org> * * Modified for DCCP from Stephen Hemminger's code * Copyright (C) 2006, Ian McDonald <ian.mcdonald@jandi.co.nz> * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ #include <linux/kernel.h> #include <linux/kprobes.h> #include <linux/socket.h> #include <linux/dccp.h> #include <linux/proc_fs.h> #include <linux/module.h> #include <linux/kfifo.h> #include <linux/vmalloc.h> |
5a0e3ad6a include cleanup: ... |
33 |
#include <linux/gfp.h> |
457c4cbc5 [NET]: Make /proc... |
34 |
#include <net/net_namespace.h> |
e41542f51 [DCCP]: Introduce... |
35 36 37 38 39 40 41 42 43 44 |
#include "dccp.h" #include "ccid.h" #include "ccids/ccid3.h" static int port; static int bufsize = 64 * 1024; static const char procname[] = "dccpprobe"; |
1e2f0e5e8 dccp: Fix sparse ... |
45 |
static struct { |
454654878 kfifo: move struc... |
46 |
struct kfifo fifo; |
e41542f51 [DCCP]: Introduce... |
47 48 |
spinlock_t lock; wait_queue_head_t wait; |
cdd04d98f [DCCP]: Convert d... |
49 |
struct timespec tstart; |
e41542f51 [DCCP]: Introduce... |
50 51 52 53 54 55 |
} dccpw; static void printl(const char *fmt, ...) { va_list args; int len; |
cdd04d98f [DCCP]: Convert d... |
56 |
struct timespec now; |
e41542f51 [DCCP]: Introduce... |
57 58 59 |
char tbuf[256]; va_start(args, fmt); |
cdd04d98f [DCCP]: Convert d... |
60 |
getnstimeofday(&now); |
e41542f51 [DCCP]: Introduce... |
61 |
|
cdd04d98f [DCCP]: Convert d... |
62 |
now = timespec_sub(now, dccpw.tstart); |
e41542f51 [DCCP]: Introduce... |
63 64 65 |
len = sprintf(tbuf, "%lu.%06lu ", (unsigned long) now.tv_sec, |
cdd04d98f [DCCP]: Convert d... |
66 |
(unsigned long) now.tv_nsec / NSEC_PER_USEC); |
e41542f51 [DCCP]: Introduce... |
67 68 |
len += vscnprintf(tbuf+len, sizeof(tbuf)-len, fmt, args); va_end(args); |
7acd72eb8 kfifo: rename kfi... |
69 |
kfifo_in_locked(&dccpw.fifo, tbuf, len, &dccpw.lock); |
e41542f51 [DCCP]: Introduce... |
70 71 72 73 74 75 |
wake_up(&dccpw.wait); } static int jdccp_sendmsg(struct kiocb *iocb, struct sock *sk, struct msghdr *msg, size_t size) { |
e41542f51 [DCCP]: Introduce... |
76 |
const struct inet_sock *inet = inet_sk(sk); |
996ccf490 dccp ccid-3: Remo... |
77 |
struct ccid3_hc_tx_sock *hc = NULL; |
e41542f51 [DCCP]: Introduce... |
78 |
|
71c262a3d dccp: API to quer... |
79 |
if (ccid_get_current_tx_ccid(dccp_sk(sk)) == DCCPC_CCID3) |
996ccf490 dccp ccid-3: Remo... |
80 |
hc = ccid3_hc_tx_sk(sk); |
e41542f51 [DCCP]: Introduce... |
81 |
|
c720c7e83 inet: rename some... |
82 83 |
if (port == 0 || ntohs(inet->inet_dport) == port || ntohs(inet->inet_sport) == port) { |
996ccf490 dccp ccid-3: Remo... |
84 |
if (hc) |
388d5e990 dccp ccid-3: Over... |
85 86 |
printl("%pI4:%u %pI4:%u %d %d %d %d %u %llu %llu %d ", |
c720c7e83 inet: rename some... |
87 88 |
&inet->inet_saddr, ntohs(inet->inet_sport), &inet->inet_daddr, ntohs(inet->inet_dport), size, |
996ccf490 dccp ccid-3: Remo... |
89 90 91 |
hc->tx_s, hc->tx_rtt, hc->tx_p, hc->tx_x_calc, hc->tx_x_recv >> 6, hc->tx_x >> 6, hc->tx_t_ipi); |
e41542f51 [DCCP]: Introduce... |
92 |
else |
21454aaad net: replace NIPQ... |
93 94 |
printl("%pI4:%u %pI4:%u %d ", |
c720c7e83 inet: rename some... |
95 96 97 |
&inet->inet_saddr, ntohs(inet->inet_sport), &inet->inet_daddr, ntohs(inet->inet_dport), size); |
e41542f51 [DCCP]: Introduce... |
98 99 100 101 102 103 104 |
} jprobe_return(); return 0; } static struct jprobe dccp_send_probe = { |
e1b7441e8 [DCCP]: Make dccp... |
105 106 107 |
.kp = { .symbol_name = "dccp_sendmsg", }, |
9e367d859 jprobes: remove J... |
108 |
.entry = jdccp_sendmsg, |
e41542f51 [DCCP]: Introduce... |
109 110 111 112 |
}; static int dccpprobe_open(struct inode *inode, struct file *file) { |
454654878 kfifo: move struc... |
113 |
kfifo_reset(&dccpw.fifo); |
cdd04d98f [DCCP]: Convert d... |
114 |
getnstimeofday(&dccpw.tstart); |
e41542f51 [DCCP]: Introduce... |
115 116 117 118 119 120 121 122 |
return 0; } static ssize_t dccpprobe_read(struct file *file, char __user *buf, size_t len, loff_t *ppos) { int error = 0, cnt = 0; unsigned char *tbuf; |
75202e768 [NET]: Fix compar... |
123 |
if (!buf) |
e41542f51 [DCCP]: Introduce... |
124 125 126 127 128 129 130 131 132 133 |
return -EINVAL; if (len == 0) return 0; tbuf = vmalloc(len); if (!tbuf) return -ENOMEM; error = wait_event_interruptible(dccpw.wait, |
e64c026dd kfifo: cleanup na... |
134 |
kfifo_len(&dccpw.fifo) != 0); |
e41542f51 [DCCP]: Introduce... |
135 136 |
if (error) goto out_free; |
7acd72eb8 kfifo: rename kfi... |
137 |
cnt = kfifo_out_locked(&dccpw.fifo, tbuf, len, &dccpw.lock); |
653252c23 net: Fix wrong in... |
138 |
error = copy_to_user(buf, tbuf, cnt) ? -EFAULT : 0; |
e41542f51 [DCCP]: Introduce... |
139 140 141 142 143 144 |
out_free: vfree(tbuf); return error ? error : cnt; } |
9a32144e9 [PATCH] mark stru... |
145 |
static const struct file_operations dccpprobe_fops = { |
e41542f51 [DCCP]: Introduce... |
146 147 148 |
.owner = THIS_MODULE, .open = dccpprobe_open, .read = dccpprobe_read, |
6038f373a llseek: automatic... |
149 |
.llseek = noop_llseek, |
e41542f51 [DCCP]: Introduce... |
150 |
}; |
d984e6197 dccp: Fix compile... |
151 152 153 154 155 156 157 158 159 160 |
static __init int setup_jprobe(void) { int ret = register_jprobe(&dccp_send_probe); if (ret) { request_module("dccp"); ret = register_jprobe(&dccp_send_probe); } return ret; } |
e41542f51 [DCCP]: Introduce... |
161 162 163 164 165 166 |
static __init int dccpprobe_init(void) { int ret = -ENOMEM; init_waitqueue_head(&dccpw.wait); spin_lock_init(&dccpw.lock); |
c1e13f256 kfifo: move out s... |
167 |
if (kfifo_alloc(&dccpw.fifo, bufsize, GFP_KERNEL)) |
454654878 kfifo: move struc... |
168 |
return ret; |
457c4cbc5 [NET]: Make /proc... |
169 |
if (!proc_net_fops_create(&init_net, procname, S_IRUSR, &dccpprobe_fops)) |
e41542f51 [DCCP]: Introduce... |
170 |
goto err0; |
d984e6197 dccp: Fix compile... |
171 |
ret = setup_jprobe(); |
e41542f51 [DCCP]: Introduce... |
172 173 174 175 176 177 178 |
if (ret) goto err1; pr_info("DCCP watch registered (port=%d) ", port); return 0; err1: |
457c4cbc5 [NET]: Make /proc... |
179 |
proc_net_remove(&init_net, procname); |
e41542f51 [DCCP]: Introduce... |
180 |
err0: |
454654878 kfifo: move struc... |
181 |
kfifo_free(&dccpw.fifo); |
e41542f51 [DCCP]: Introduce... |
182 183 184 185 186 187 |
return ret; } module_init(dccpprobe_init); static __exit void dccpprobe_exit(void) { |
454654878 kfifo: move struc... |
188 |
kfifo_free(&dccpw.fifo); |
457c4cbc5 [NET]: Make /proc... |
189 |
proc_net_remove(&init_net, procname); |
e41542f51 [DCCP]: Introduce... |
190 191 192 193 194 195 196 197 198 199 200 201 202 203 |
unregister_jprobe(&dccp_send_probe); } module_exit(dccpprobe_exit); MODULE_PARM_DESC(port, "Port to match (0=all)"); module_param(port, int, 0); MODULE_PARM_DESC(bufsize, "Log buffer size (default 64k)"); module_param(bufsize, int, 0); MODULE_AUTHOR("Ian McDonald <ian.mcdonald@jandi.co.nz>"); MODULE_DESCRIPTION("DCCP snooper"); MODULE_LICENSE("GPL"); |