forked from Minki/linux
CRIS: check for memory allocation
Add checking for allocated memory. Indents and spaces are added to be familiar with the kernel coding style. Signed-off-by: Cyrill Gorcunov <gorcunov@gmail.com> Acked-by: Mikael Starvik <starvik@axis.com> Signed-off-by: Andrew Morton <akpm@linux-foundation.org> Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
This commit is contained in:
parent
c65808ef76
commit
542401d97a
@ -15,39 +15,47 @@ static int prof_running = 0;
|
|||||||
void
|
void
|
||||||
cris_profile_sample(struct pt_regs* regs)
|
cris_profile_sample(struct pt_regs* regs)
|
||||||
{
|
{
|
||||||
if (!prof_running)
|
if (!prof_running)
|
||||||
return;
|
return;
|
||||||
if (user_mode(regs))
|
|
||||||
*(unsigned int*)sample_buffer_pos = current->pid;
|
if (user_mode(regs))
|
||||||
else
|
*(unsigned int*)sample_buffer_pos = current->pid;
|
||||||
*(unsigned int*)sample_buffer_pos = 0;
|
else
|
||||||
*(unsigned int*)(sample_buffer_pos + 4) = instruction_pointer(regs);
|
*(unsigned int*)sample_buffer_pos = 0;
|
||||||
sample_buffer_pos += 8;
|
|
||||||
if (sample_buffer_pos == sample_buffer + SAMPLE_BUFFER_SIZE)
|
*(unsigned int*)(sample_buffer_pos + 4) = instruction_pointer(regs);
|
||||||
sample_buffer_pos = sample_buffer;
|
sample_buffer_pos += 8;
|
||||||
|
|
||||||
|
if (sample_buffer_pos == sample_buffer + SAMPLE_BUFFER_SIZE)
|
||||||
|
sample_buffer_pos = sample_buffer;
|
||||||
}
|
}
|
||||||
|
|
||||||
static ssize_t
|
static ssize_t
|
||||||
read_cris_profile(struct file *file, char __user *buf, size_t count, loff_t *ppos)
|
read_cris_profile(struct file *file, char __user *buf,
|
||||||
|
size_t count, loff_t *ppos)
|
||||||
{
|
{
|
||||||
unsigned long p = *ppos;
|
unsigned long p = *ppos;
|
||||||
if (p > SAMPLE_BUFFER_SIZE)
|
|
||||||
return 0;
|
if (p > SAMPLE_BUFFER_SIZE)
|
||||||
if (p + count > SAMPLE_BUFFER_SIZE)
|
return 0;
|
||||||
count = SAMPLE_BUFFER_SIZE - p;
|
|
||||||
if (copy_to_user(buf, sample_buffer + p,count))
|
if (p + count > SAMPLE_BUFFER_SIZE)
|
||||||
|
count = SAMPLE_BUFFER_SIZE - p;
|
||||||
|
if (copy_to_user(buf, sample_buffer + p,count))
|
||||||
return -EFAULT;
|
return -EFAULT;
|
||||||
memset(sample_buffer + p, 0, count);
|
|
||||||
*ppos += count;
|
memset(sample_buffer + p, 0, count);
|
||||||
return count;
|
*ppos += count;
|
||||||
|
|
||||||
|
return count;
|
||||||
}
|
}
|
||||||
|
|
||||||
static ssize_t
|
static ssize_t
|
||||||
write_cris_profile(struct file *file, const char __user *buf,
|
write_cris_profile(struct file *file, const char __user *buf,
|
||||||
size_t count, loff_t *ppos)
|
size_t count, loff_t *ppos)
|
||||||
{
|
{
|
||||||
sample_buffer_pos = sample_buffer;
|
sample_buffer_pos = sample_buffer;
|
||||||
memset(sample_buffer, 0, SAMPLE_BUFFER_SIZE);
|
memset(sample_buffer, 0, SAMPLE_BUFFER_SIZE);
|
||||||
}
|
}
|
||||||
|
|
||||||
static const struct file_operations cris_proc_profile_operations = {
|
static const struct file_operations cris_proc_profile_operations = {
|
||||||
@ -58,16 +66,23 @@ static const struct file_operations cris_proc_profile_operations = {
|
|||||||
static int
|
static int
|
||||||
__init init_cris_profile(void)
|
__init init_cris_profile(void)
|
||||||
{
|
{
|
||||||
struct proc_dir_entry *entry;
|
struct proc_dir_entry *entry;
|
||||||
sample_buffer = kmalloc(SAMPLE_BUFFER_SIZE, GFP_KERNEL);
|
|
||||||
sample_buffer_pos = sample_buffer;
|
sample_buffer = kmalloc(SAMPLE_BUFFER_SIZE, GFP_KERNEL);
|
||||||
entry = create_proc_entry("system_profile", S_IWUSR | S_IRUGO, NULL);
|
if (!sample_buffer) {
|
||||||
if (entry) {
|
return -ENOMEM;
|
||||||
entry->proc_fops = &cris_proc_profile_operations;
|
}
|
||||||
entry->size = SAMPLE_BUFFER_SIZE;
|
|
||||||
}
|
sample_buffer_pos = sample_buffer;
|
||||||
prof_running = 1;
|
|
||||||
return 0;
|
entry = create_proc_entry("system_profile", S_IWUSR | S_IRUGO, NULL);
|
||||||
|
if (entry) {
|
||||||
|
entry->proc_fops = &cris_proc_profile_operations;
|
||||||
|
entry->size = SAMPLE_BUFFER_SIZE;
|
||||||
|
}
|
||||||
|
prof_running = 1;
|
||||||
|
|
||||||
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
__initcall(init_cris_profile);
|
__initcall(init_cris_profile);
|
||||||
|
Loading…
Reference in New Issue
Block a user