提交 542401d9 编写于 作者: C Cyrill Gorcunov 提交者: Linus Torvalds

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: NCyrill Gorcunov <gorcunov@gmail.com>
Acked-by: NMikael Starvik <starvik@axis.com>
Signed-off-by: NAndrew Morton <akpm@linux-foundation.org>
Signed-off-by: NLinus Torvalds <torvalds@linux-foundation.org>
上级 c65808ef
...@@ -15,39 +15,47 @@ static int prof_running = 0; ...@@ -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 = { ...@@ -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);
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册