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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
|
/* -*- linux-c -*-
*
* /proc transport and control
* Copyright (C) 2005-2009 Red Hat Inc.
*
* This file is part of systemtap, and is free software. You can
* redistribute it and/or modify it under the terms of the GNU General
* Public License (GPL); either version 2, or (at your option) any
* later version.
*/
#include "../procfs.c" // for _stp_mkdir_proc_module()
#define STP_DEFAULT_BUFFERS 256
#ifdef STP_BULKMODE
extern int _stp_relay_flushing;
/* handle the per-cpu subbuf info read for relayfs */
static ssize_t _stp_proc_read(struct file *file, char __user *buf, size_t count, loff_t *ppos)
{
int num;
struct _stp_buf_info out;
int cpu = *(int *)(PDE(file->f_dentry->d_inode)->data);
if (!_stp_utt->rchan)
return -EINVAL;
out.cpu = cpu;
out.produced = atomic_read(&_stp_utt->rchan->buf[cpu]->subbufs_produced);
out.consumed = atomic_read(&_stp_utt->rchan->buf[cpu]->subbufs_consumed);
out.flushing = _stp_relay_flushing;
num = sizeof(out);
if (copy_to_user(buf, &out, num))
return -EFAULT;
return num;
}
/* handle the per-cpu subbuf info write for relayfs */
static ssize_t _stp_proc_write(struct file *file, const char __user *buf, size_t count, loff_t *ppos)
{
struct _stp_consumed_info info;
int cpu = *(int *)(PDE(file->f_dentry->d_inode)->data);
if (copy_from_user(&info, buf, count))
return -EFAULT;
relay_subbufs_consumed(_stp_utt->rchan, cpu, info.consumed);
return count;
}
static struct file_operations _stp_proc_fops = {
.owner = THIS_MODULE,
.read = _stp_proc_read,
.write = _stp_proc_write,
};
#endif /* STP_BULKMODE */
inline static int _stp_ctl_write_fs(int type, void *data, unsigned len)
{
struct _stp_buffer *bptr;
unsigned long flags;
#define WRITE_AGG
#ifdef WRITE_AGG
spin_lock_irqsave(&_stp_ctl_ready_lock, flags);
if (!list_empty(&_stp_ctl_ready_q)) {
bptr = (struct _stp_buffer *)_stp_ctl_ready_q.prev;
if ((bptr->len + len) <= STP_CTL_BUFFER_SIZE
&& type == STP_REALTIME_DATA
&& bptr->type == STP_REALTIME_DATA) {
memcpy(bptr->buf + bptr->len, data, len);
bptr->len += len;
spin_unlock_irqrestore(&_stp_ctl_ready_lock, flags);
return len;
}
}
spin_unlock_irqrestore(&_stp_ctl_ready_lock, flags);
#endif
return 0;
}
/* set the number of buffers to use to 'num' */
static int _stp_set_buffers(int num)
{
dbug_trans(1, "stp_set_buffers %d\n", num);
return _stp_mempool_resize(_stp_pool_q, num);
}
static int _stp_ctl_read_bufsize(char *page, char **start, off_t off, int count, int *eof, void *data)
{
int len = sprintf(page, "%d,%d\n", _stp_nsubbufs, _stp_subbuf_size);
if (len <= off + count)
*eof = 1;
*start = page + off;
len -= off;
if (len > count)
len = count;
if (len < 0)
len = 0;
return len;
}
static int _stp_register_ctl_channel_fs(void)
{
int i;
const char *dirname = "systemtap";
char buf[32];
#ifdef STP_BULKMODE
int j;
#endif
struct proc_dir_entry *de, *bs = NULL;
if (!_stp_mkdir_proc_module())
goto err0;
#ifdef STP_BULKMODE
/* now for each cpu "n", create /proc/systemtap/module_name/n */
stp_for_each_cpu(i) {
sprintf(buf, "%d", i);
de = create_proc_entry(buf, 0600, _stp_proc_root);
if (de == NULL)
goto err1;
de->uid = _stp_uid;
de->gid = _stp_gid;
de->proc_fops = &_stp_proc_fops;
de->data = _stp_kmalloc(sizeof(int));
if (de->data == NULL) {
remove_proc_entry(buf, _stp_proc_root);
goto err1;
}
*(int *)de->data = i;
}
bs = create_proc_read_entry("bufsize", 0, _stp_proc_root, _stp_ctl_read_bufsize, NULL);
#endif /* STP_BULKMODE */
/* create /proc/systemtap/module_name/.cmd */
de = create_proc_entry(".cmd", 0600, _stp_proc_root);
if (de == NULL)
goto err1;
de->uid = _stp_uid;
de->gid = _stp_gid;
de->proc_fops = &_stp_ctl_fops_cmd;
return 0;
err2:
remove_proc_entry(".cmd", _stp_proc_root);
err1:
#ifdef STP_BULKMODE
for (de = _stp_proc_root->subdir; de; de = de->next)
_stp_kfree(de->data);
stp_for_each_cpu(j) {
if (j == i)
break;
sprintf(buf, "%d", j);
remove_proc_entry(buf, _stp_proc_root);
}
if (bs)
remove_proc_entry("bufsize", _stp_proc_root);
#endif /* STP_BULKMODE */
_stp_rmdir_proc_module();
err0:
return -1;
}
static void _stp_unregister_ctl_channel_fs(void)
{
#ifdef STP_BULKMODE
char buf[32];
int i;
struct proc_dir_entry *de;
dbug_trans(1, "unregistering procfs\n");
for (de = _stp_proc_root->subdir; de; de = de->next)
_stp_kfree(de->data);
stp_for_each_cpu(i) {
sprintf(buf, "%d", i);
remove_proc_entry(buf, _stp_proc_root);
}
remove_proc_entry("bufsize", _stp_proc_root);
#endif /* STP_BULKMODE */
remove_proc_entry(".cmd", _stp_proc_root);
_stp_rmdir_proc_module();
}
|
