Exemplo n.º 1
0
void init_netload(netdata* data, const char* device)
{
    memset( data, 0, sizeof(netdata) );
    strncpy( data->ifdata.if_name, device, 9 );
    data->ifdata.if_name[9] = '\0';
    
    init_osspecific( data );
    
    if (checkinterface(data) != TRUE)
	{
        data->correct_interface = FALSE;
		return;
	}
    
    /* init in a sane state */
    get_stat(data);
    data->backup_in  = data->stats.rx_bytes;
    data->backup_out = data->stats.tx_bytes;
    
    data->correct_interface = TRUE;
    
#ifdef DEBUG
    fprintf( stderr, "The netload plugin was initialized for '%s'.\n", device );
#endif /* DEBUG */
}
Exemplo n.º 2
0
void init_netload(netdata* data, const char* device)
{
    memset( data, 0, sizeof(netdata) );
    strncpy( data->ifdata.if_name, device, 9 );
    data->ifdata.if_name[9] = '\0';
    
    if (checkinterface(data) != TRUE)
	{
        data->correct_interface = FALSE;
		return;
	}
    
    /* init in a sane state */
    get_stat(data);
    data->backup_in  = data->stats.rx_bytes;
    data->backup_out = data->stats.tx_bytes;
    
    data->correct_interface = TRUE;
}
Exemplo n.º 3
0
int main (int argc, char **argv)
{
	/* data aq. */
	double backup_in, backup_out;
	double cur_in, cur_out;
	char dim_in[4], dim_out[4];

	/* getopt */
	int opt;
	extern char *optarg;
	int mode = OPT_NONE;
	int sampletime = SAMPLE_TIME;

	bzero(ifdata.if_name, sizeof(ifdata.if_name));

	/* getopt () */
	while ((opt = getopt(argc, argv, "d:hi:o:")) != EOF)
	{
		switch (opt)
		{
			case 'd':
				/* delay aka sample time */
				if (atoi(optarg) > 0) sampletime = atoi(optarg);
				else
				{
					fprintf(stderr, "dude, delay must be positive!\n");
					usage(argv[0], 1);
				}
				break;
			case 'i':
				/* interface */
				snprintf(ifdata.if_name, 9, "%s", optarg); // memcpy
				break;
			case 'h':
				/* usage */
				usage(argv[0], 0);
				break;
			case 'o':
				/* possible output options */
				mode |= getopmode((char *) optarg);
#ifdef CUSTOM_MODE
				/* check arg if running in custom mode */
				if (mode == OPT_CUSTOM)
				{
					/* when running in custom mode the last argument
					 * should be a string containing two %% spaceholders
					 * where incoming and outgoing traffic should
					 * be inserted
					 */
					p = argv[argc-1];
					if (((p = strstr(p, "%%%")) == NULL) ||
					    ((p = strstr(++p, "%%%")) == NULL))
						usage(argv[0], 1);
					/* is there a third argument? */
					if (strstr(++p, "%%%") != NULL)
						usage(argv[0], 1);
				}
#endif
				break;
			default:
				usage(argv[0], 1);
		}
	}
	/* sanity checks */
	if (ifdata.if_name[0] == '\0') /* no interface given */
		usage(argv[0], 1);
	if (checkinterface() != TRUE)
	{
		fprintf(stderr, "invalid interface %s\n", (char *) ifdata.if_name);
		return 1;
	}
	if (mode == OPT_NONE)
	{

		/* init in a sane state */
		get_stat();
		backup_in  = stats.rx_bytes;
		backup_out = stats.tx_bytes;
		bzero(dim_in, sizeof(dim_in));
		bzero(dim_out, sizeof(dim_out));

		/* main loop */
		for (;;)
		{
			/* update */
			get_stat();
			if (backup_in > stats.rx_bytes)
				cur_in = (int) stats.rx_bytes / sampletime;
			else
				cur_in = (int) (stats.rx_bytes - backup_in) / sampletime;
			if (backup_out > stats.tx_bytes)
				cur_out = (int) stats.tx_bytes / sampletime;
			else
				cur_out = (int) (stats.tx_bytes - backup_out) / sampletime;

			/* determine traffic dimension and recalc value
		 	 * currently MB/s, KB/s and B/s are supported
		 	 */
			if (cur_in > 1024*1024)
			{
				cur_in = cur_in / (1024*1024); snprintf(dim_in, 22, "MB");
			}
			else if (cur_in > 1024)
			{
				cur_in = cur_in / 1024; snprintf(dim_in, 2, "KB");
			}
			else
				snprintf(dim_in, 2, "B");

			if (cur_out > 1024*1024)
			{
				cur_out = cur_out / (1024*1024); snprintf(dim_out, 2, "MB");
			}
			else if (cur_out > 1024)
			{
				cur_out = cur_out / 1024; snprintf(dim_out, 2, "KB");
			}
			else
				snprintf(dim_out, 2, "B");

			/* print to screen */
			printf("in %lu %s/s out %lu %s/s\n",
					(unsigned long) cur_in, dim_in,
					(unsigned long) cur_out, dim_out);
			fflush(stdout);

			/* save 'new old' values */
			backup_in = stats.rx_bytes;
			backup_out = stats.tx_bytes;

			/* sleep */	
			sleep(sampletime);
		}
	}
	else
	{
		/* non-screen modes */
		get_stat();
		backup_in  = stats.rx_bytes;
		backup_out = stats.tx_bytes;
		bzero(dim_in, sizeof(dim_in));
		bzero(dim_out, sizeof(dim_out));
		sleep(sampletime);
		get_stat();

		if (backup_in > stats.rx_bytes)
			cur_in = (int) stats.rx_bytes / sampletime;
		else
			cur_in = (int) (stats.rx_bytes - backup_in) / sampletime;
		if (backup_out > stats.tx_bytes)
			cur_out = (int) stats.tx_bytes / sampletime;
		else
			cur_out = (int) (stats.tx_bytes - backup_out) / sampletime;
	
		/*	
		 * mode dependent output
		 */

		switch (mode)
		{
			case OPT_MRTG:
			case OPT_FULLMRTG:	/* for now */
				printf("%lu\n%lu\n",
					(unsigned long) cur_in, (unsigned long) cur_out);
				break;
#ifdef CUSTOM_MODE
			case OPT_CUSTOM:
				/* substitute %%% with %lu to insert values */
				p = argv[argc-1];
				p = strstr(p, "%%%");
				memcpy(p, "%lu", 3);
				p = strstr(++p, "%%%");
				memcpy(p, "%lu",3);
				printf(argv[argc-1],
					(unsigned long) cur_in, (unsigned long) cur_out);
				break;
#endif
			default:
				fprintf(stderr, "internal error in output. please report!\n");
		}
	}
	return 0;
}