mirror of
https://github.com/varun-r-mallya/sysprof.git
synced 2025-12-31 20:36:25 +00:00
Initial revision
This commit is contained in:
Søren Sandmann Pedersen
343
process.c
Normal file
343
process.c
Normal file
@ -0,0 +1,343 @@
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <sys/stat.h>
|
||||
#include <errno.h>
|
||||
|
||||
#include "process.h"
|
||||
#include "binfile.h"
|
||||
|
||||
/* FIXME: All the interesting stuff in this file is from memprof
|
||||
* and copyright Red Hat.
|
||||
*/
|
||||
|
||||
/* FIXME: this should be done with getpagesize() */
|
||||
#define PAGE_SIZE 4096
|
||||
|
||||
static GHashTable *processes_by_cmdline;
|
||||
static GHashTable *processes_by_pid;
|
||||
|
||||
typedef struct Map Map;
|
||||
struct Map
|
||||
{
|
||||
char * filename;
|
||||
gulong start;
|
||||
gulong end;
|
||||
gboolean do_offset;
|
||||
|
||||
BinFile * bin_file;
|
||||
};
|
||||
|
||||
struct Process
|
||||
{
|
||||
char *cmdline;
|
||||
|
||||
GList *maps;
|
||||
GList *bad_pages;
|
||||
};
|
||||
|
||||
static void
|
||||
initialize (void)
|
||||
{
|
||||
if (!processes_by_cmdline)
|
||||
{
|
||||
processes_by_cmdline = g_hash_table_new (g_str_hash, g_str_equal);
|
||||
processes_by_pid = g_hash_table_new (g_direct_hash, g_direct_equal);
|
||||
}
|
||||
}
|
||||
|
||||
static gboolean
|
||||
should_offset (const char *filename, int pid)
|
||||
{
|
||||
gboolean result = TRUE;
|
||||
|
||||
struct stat stat1, stat2;
|
||||
char *progname = g_strdup_printf ("/proc/%d/exe", pid);
|
||||
|
||||
if (stat (filename, &stat1) == 0)
|
||||
{
|
||||
if (stat (progname, &stat2) == 0)
|
||||
result = !(stat1.st_ino == stat2.st_ino &&
|
||||
stat1.st_dev == stat2.st_dev);
|
||||
}
|
||||
|
||||
g_free (progname);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
static gboolean
|
||||
check_do_offset (const Map *map, int pid)
|
||||
{
|
||||
if (map->filename)
|
||||
return should_offset (map->filename, pid);
|
||||
else
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
static GList *
|
||||
read_maps (int pid)
|
||||
{
|
||||
char *name = g_strdup_printf ("/proc/%d/maps", pid);
|
||||
char buffer[1024];
|
||||
FILE *in;
|
||||
GList *result = NULL;
|
||||
|
||||
in = fopen (name, "r");
|
||||
if (!in)
|
||||
{
|
||||
g_free (name);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
while (fgets (buffer, sizeof (buffer) - 1, in))
|
||||
{
|
||||
char file[256];
|
||||
int count;
|
||||
guint start;
|
||||
guint end;
|
||||
|
||||
count = sscanf (
|
||||
buffer, "%x-%x %*15s %*x %*u:%*u %*u %255s", &start, &end, file);
|
||||
if (count == 3)
|
||||
{
|
||||
Map *map;
|
||||
|
||||
map = g_new (Map, 1);
|
||||
|
||||
map->filename = g_strdup (file);
|
||||
map->start = start;
|
||||
map->end = end;
|
||||
|
||||
map->do_offset = check_do_offset (map, pid);
|
||||
|
||||
map->bin_file = NULL;
|
||||
|
||||
result = g_list_prepend (result, map);
|
||||
}
|
||||
}
|
||||
|
||||
g_free (name);
|
||||
fclose (in);
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
static Process *
|
||||
create_process (const char *cmdline, int pid)
|
||||
{
|
||||
Process *p;
|
||||
|
||||
p = g_new (Process, 1);
|
||||
|
||||
p->cmdline = g_strdup_printf ("[%s]", cmdline);
|
||||
p->bad_pages = NULL;
|
||||
p->maps = NULL;
|
||||
|
||||
g_assert (!g_hash_table_lookup (processes_by_pid, GINT_TO_POINTER (pid)));
|
||||
g_assert (!g_hash_table_lookup (processes_by_cmdline, cmdline));
|
||||
|
||||
g_hash_table_insert (processes_by_pid, GINT_TO_POINTER (pid), p);
|
||||
g_hash_table_insert (processes_by_cmdline, g_strdup (cmdline), p);
|
||||
|
||||
return p;
|
||||
}
|
||||
|
||||
static Map *
|
||||
process_locate_map (Process *process, gulong addr)
|
||||
{
|
||||
GList *list;
|
||||
|
||||
for (list = process->maps; list != NULL; list = list->next)
|
||||
{
|
||||
Map *map = list->data;
|
||||
|
||||
if ((addr >= map->start) &&
|
||||
(addr < map->end))
|
||||
{
|
||||
return map;
|
||||
}
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static void
|
||||
process_free_maps (Process *process)
|
||||
{
|
||||
GList *list;
|
||||
|
||||
for (list = process->maps; list != NULL; list = list->next)
|
||||
{
|
||||
Map *map = list->data;
|
||||
|
||||
if (map->bin_file)
|
||||
bin_file_free (map->bin_file);
|
||||
|
||||
g_free (map);
|
||||
}
|
||||
|
||||
g_list_free (process->maps);
|
||||
}
|
||||
|
||||
static gboolean
|
||||
process_has_page (Process *process, gulong addr)
|
||||
{
|
||||
if (process_locate_map (process, addr))
|
||||
return TRUE;
|
||||
else
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
void
|
||||
process_ensure_map (Process *process, int pid, gulong addr)
|
||||
{
|
||||
/* Round don to closest page */
|
||||
addr = (addr - addr % PAGE_SIZE);
|
||||
|
||||
if (process_has_page (process, addr))
|
||||
return;
|
||||
|
||||
if (g_list_find (process->bad_pages, (gpointer)addr))
|
||||
return;
|
||||
|
||||
/* a map containing addr was not found */
|
||||
if (process->maps)
|
||||
process_free_maps (process);
|
||||
|
||||
process->maps = read_maps (pid);
|
||||
|
||||
if (!process_has_page (process, addr))
|
||||
g_list_prepend (process->bad_pages, (gpointer)addr);
|
||||
}
|
||||
|
||||
Process *
|
||||
process_get_from_pid (int pid)
|
||||
{
|
||||
Process *p;
|
||||
gchar *filename = NULL;
|
||||
gchar *cmdline = NULL;
|
||||
|
||||
initialize();
|
||||
|
||||
p = g_hash_table_lookup (processes_by_pid, GINT_TO_POINTER (pid));
|
||||
if (p)
|
||||
goto out;
|
||||
|
||||
filename = g_strdup_printf ("/proc/%d/cmdline", pid);
|
||||
|
||||
if (g_file_get_contents (filename, &cmdline, NULL, NULL))
|
||||
{
|
||||
p = g_hash_table_lookup (processes_by_cmdline, cmdline);
|
||||
if (p)
|
||||
goto out;
|
||||
}
|
||||
else
|
||||
{
|
||||
cmdline = g_strdup_printf ("[pid %d]", pid);
|
||||
}
|
||||
|
||||
p = create_process (cmdline, pid);
|
||||
|
||||
out:
|
||||
if (cmdline)
|
||||
g_free (cmdline);
|
||||
|
||||
if (filename)
|
||||
g_free (filename);
|
||||
|
||||
return p;
|
||||
}
|
||||
|
||||
|
||||
const Symbol *
|
||||
process_lookup_symbol (Process *process, gulong address)
|
||||
{
|
||||
static Symbol undefined;
|
||||
const Symbol *result;
|
||||
Map *map = process_locate_map (process, address);
|
||||
|
||||
if (!map)
|
||||
{
|
||||
if (undefined.name)
|
||||
g_free (undefined.name);
|
||||
undefined.name = g_strdup_printf ("??? (%s)", process->cmdline);
|
||||
undefined.address = 0xBABE0001;
|
||||
|
||||
return &undefined;
|
||||
}
|
||||
|
||||
if (!map->bin_file)
|
||||
map->bin_file = bin_file_new (map->filename);
|
||||
|
||||
if (map->do_offset)
|
||||
address -= map->start;
|
||||
|
||||
result = bin_file_lookup_symbol (map->bin_file, address);
|
||||
return result;
|
||||
}
|
||||
|
||||
const Symbol *
|
||||
process_lookup_symbol_with_filename (Process *process,
|
||||
int pid,
|
||||
gulong map_start,
|
||||
const char *filename,
|
||||
gulong address)
|
||||
{
|
||||
const Symbol *result;
|
||||
BinFile *bin_file;
|
||||
|
||||
if (!filename)
|
||||
return process_lookup_symbol (process, address);
|
||||
|
||||
bin_file = bin_file_new (filename);
|
||||
|
||||
if (should_offset (filename, pid))
|
||||
address -= map_start;
|
||||
|
||||
result = bin_file_lookup_symbol (bin_file, address);
|
||||
|
||||
bin_file_free (bin_file);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
const char *
|
||||
process_get_cmdline (Process *process)
|
||||
{
|
||||
return process->cmdline;
|
||||
}
|
||||
|
||||
static void
|
||||
free_process (gpointer key, gpointer value, gpointer data)
|
||||
{
|
||||
char *cmdline = key;
|
||||
Process *process = value;
|
||||
|
||||
#if 0
|
||||
g_print ("freeing: %p\n", process);
|
||||
memset (process, '\0', sizeof (Process));
|
||||
#endif
|
||||
g_free (process->cmdline);
|
||||
#if 0
|
||||
process->cmdline = "You are using free()'d memory";
|
||||
#endif
|
||||
process_free_maps (process);
|
||||
g_list_free (process->bad_pages);
|
||||
g_free (cmdline);
|
||||
|
||||
g_free (process);
|
||||
}
|
||||
|
||||
void
|
||||
process_flush_caches (void)
|
||||
{
|
||||
if (!processes_by_cmdline)
|
||||
return;
|
||||
g_hash_table_foreach (processes_by_cmdline, free_process, NULL);
|
||||
|
||||
g_hash_table_destroy (processes_by_cmdline);
|
||||
g_hash_table_destroy (processes_by_pid);
|
||||
|
||||
processes_by_cmdline = NULL;
|
||||
processes_by_pid = NULL;
|
||||
}
|
||||
Reference in New Issue
Block a user