/*

 * Copyright (C) 2008 Mark Wong

 */

 

#include 

#include 

#include 

#include 

#include 

#include 

#include 

#include 

#include 

#include 

 

#define PROC "/proc"

 

#define GET_VALUE(v) 

		p = strchr(p, ':'); 

		++p; 

		++p; 

		q = strchr(p, 'n'); 

		length = q - p; 

		if (length >= BUFFERLEN) 

		{ 

			printf("ERROR - value is larger than the buffer: %dn", __LINE__); 

			exit(1); 

		} 

		strncpy(value, p, length); 

		value[length] = ''; 

		v = atoll(value);

 

#define BTOKB(b) b >> 10

#define BTOMB(b) b >> 20

 

#define BUFFERLEN 255

#define COMMANDLEN 1024

#define VALUELEN 63

 

#define NUM_STRINGS 8

 

struct io_node

{

	int pid;

	long long rchar;

	long long wchar;

	long long syscr;

	long long syscw;

	long long read_bytes;

	long long write_bytes;

	long long cancelled_write_bytes;

	char command[COMMANDLEN + 1];

	struct io_node *next;

};

 

struct io_node *head = NULL;

int command_flag = 0;

int idle_flag = 0;

int mb_flag = 0;

int kb_flag = 0;

int hr_flag = 0;

 

/* Prototypes */

char *format_b(long long);

struct io_node *get_ion(int);

struct io_node *new_ion(char *);

void upsert_data(struct io_node *);

 

char *

format_b(long long amt)

{

	static char retarray[NUM_STRINGS][16];

	static int	index = 0;

	register char *ret;

	register char tag = 'B';

 

	ret = retarray[index];

	index = (index + 1) % NUM_STRINGS;

 

	if (amt >= 10000) {

		amt = (amt + 512) / 1024;

		tag = 'K';

		if (amt >= 10000) {

			amt = (amt + 512) / 1024;

			tag = 'B';

			if (amt >= 10000) {

				amt = (amt + 512) / 1024;

				tag = 'G';

			}

		}

	}

 

	snprintf(ret, sizeof(retarray[index]) - 1, "%lld%c", amt, tag);

 

	return (ret);

}

 

int

get_cmdline(struct io_node *ion)

{

	int fd;

	int length;

	char filename[BUFFERLEN + 1];

	char buffer[COMMANDLEN + 1];

	char *p;

	char *q;

 

 

	length = snprintf(filename, BUFFERLEN, "%s/%d/cmdline", PROC, ion->pid);

	if (length == BUFFERLEN)

		printf("WARNING - filename length may be too big for buffer: %dn",

				__LINE__);

	fd = open(filename, O_RDONLY);

	if (fd == -1)

		return 1;

	length = read(fd, buffer, sizeof(buffer) - 1);

	close(fd);

	buffer[length] = '';

	if (length == 0)

		return 2;

	if (command_flag == 0)

	{

		/*

		 * The command is near the beginning; we don't need to be able to

		 * the entire stat file.

		 */

		p = strchr(buffer, '(');

		++p;

		q = strchr(p, ')');

		length = q - p;

	}

	else

		p = buffer;

	length = length command, p, length);

	ion->command[length] = '';

	return 0;

}

 

struct io_node *

get_ion(int pid)

{

	struct io_node *c = head;

 

	while (c != NULL)

	{

		if (c->pid == pid)

			break;

		c = c->next;

	}

	return c;

}

 

int

get_tcomm(struct io_node *ion)

{

	int fd;

	int length;

	char filename[BUFFERLEN + 1];

	char buffer[BUFFERLEN + 1];

	char *p;

	char *q;

 

	length = snprintf(filename, BUFFERLEN, "%s/%d/stat", PROC, ion->pid);

	if (length == BUFFERLEN)

		printf("WARNING - filename length may be too big for buffer: %dn",

				__LINE__);

	fd = open(filename, O_RDONLY);

	if (fd == -1)

		return 1;

	length = read(fd, buffer, sizeof(buffer) - 1);

	close(fd);

	/*

	 * The command is near the beginning; we don't need to be able to

	 * the entire stat file.

	 */

	p = strchr(buffer, '(');

	++p;

	q = strchr(p, ')');

	length = q - p;

	length = length command, p, length);

	ion->command[length] = '';

	return 0;

}

 

struct io_node *

insert_ion(struct io_node *ion)

{

	struct io_node *c;

	struct io_node *p;

 

	/* Check the head of the list as a special case. */

	if (ion->pid pid)

	{

		ion->next = head;

		head = ion;

		return head;

	}

 

	c = head->next;

	p = head;

	while (c != NULL)

	{

		if (ion->pid pid)

		{

			ion->next = c;

			p->next = ion;

			return head;

		}

		p = c;

		c = c->next;

	}

 

	/* Append to the end of the list. */

	if (c == NULL)

		p->next = ion;

 

	return head;

}

您可能感兴趣的文章: