adulau/aha
1
0
Fork 0
mirror of https://github.com/adulau/aha.git synced 2024-12-27 11:16:11 +00:00
Code Issues Projects Releases Packages Wiki Activity

Nasty code to generate unique files with the output of sys_execve arguments

Browse source
This commit is contained in:
Gerard Wagener 2010-01-06 22:57:07 +01:00
parent 7025b6f9cb
commit e1e908b8cb
1 changed files with 56 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

62
arch/um/kernel/exec.c
View file

@ -17,6 +17,7 @@
#include "os.h"
#include "internal.h"
#include "aha.h"
#include "os.h"
void flush_thread(void)
{
void *data = NULL;
@ -70,19 +71,60 @@ long um_execve(char *file, char __user *__user *argv, char __user *__user *env)
return err;
}
/*
* My uuid hack wuuuurgs, performance bye bye it is already gone with the
* massive amount of IO
*
* The filename is returned through parameters and the length of the string
* is returned. On error negative value is returned. See snprintf
*/
int create_filename(char *fn, int size){
int a,b;
long ncycles;
/* Query the processor cycles and concatenate it with a prefix */
asm volatile("rdtsc" : "=a" (a), "=d" (b));
ncycles = ((long long )a|(long long)b<<32);
/* Return the length of the string, negative value on failure */
return snprintf(fn,size,"AHA_%lx.out",ncycles);
}
/*
* TODO need to extract PID and PPID?
*/
void dump_execve(char __user *file, char __user *__user *argv,
char __user *__user *env)
{
char *p;
char *a;
char *p, *a, *q;
struct openflags flg;
int mode = 0644;
int fd,cnt;
flg.w = 1;
flg.c = 1;
cnt = 0;
/*
* FIXME Disk access is an good awfull solution; UML can be put on a
* tmfs as mitigation
*/
p = kmalloc(MAX_DUMP_BUF,GFP_KERNEL);
if (p) {
/* Dump the file from execve */
q = kmalloc(MAX_DUMP_BUF, GFP_KERNEL);
if (p && q) {
if (create_filename(p,MAX_DUMP_BUF)<0)
return;
if ((fd = os_open_file(p,flg,mode))<0)
return;
/* Dump the file from execve */
if (strncpy_from_user(p,file,MAX_DUMP_BUF) > 0){
printk("AHA:execve>file=%s\n",p);
cnt = snprintf((char*)q,MAX_DUMP_BUF,"AHA:execve>file=%s\n",p);
/* Best effort: If we manage to write ok if not it is also ok */
if ((cnt>0) & (cnt < MAX_DUMP_BUF))
os_write_file(fd,q,cnt);
}
/* Dump the arguments */
for (;;) {
@ -91,11 +133,19 @@ void dump_execve(char __user *file, char __user *__user *argv,
if (!a)
break;
if (strncpy_from_user(p,a, MAX_DUMP_BUF) > 0) {
printk("AHA:argument=%s\n",p);
cnt=snprintf(q,cnt,"AHA:argument=%s\n",p);
if ((cnt>0) & (cnt<MAX_DUMP_BUF))
os_write_file(fd,q,cnt);
}
argv++;
}
cnt = snprintf(q,cnt,"DONE\n");
if ((cnt >0) & (cnt < MAX_DUMP_BUF))
os_write_file(fd,q,cnt);
os_close_file(fd);
kfree(p);
kfree(q);
}
}