1aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger/* 2aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * rtacct.c Applet to display contents of /proc/net/rt_acct. 3aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * 4aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * This program is free software; you can redistribute it and/or 5aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * modify it under the terms of the GNU General Public License 6aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * as published by the Free Software Foundation; either version 7aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * 2 of the License, or (at your option) any later version. 8aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * 9aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru> 10aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * 11aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger */ 12aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 13aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <stdio.h> 14aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <stdlib.h> 15aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <unistd.h> 16aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <fcntl.h> 17aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <string.h> 18aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <errno.h> 19aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <time.h> 20aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <sys/time.h> 21aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <fnmatch.h> 22aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <sys/file.h> 23aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <sys/socket.h> 24aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <sys/un.h> 25aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <sys/poll.h> 26aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <sys/wait.h> 27aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <sys/stat.h> 28aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <sys/mman.h> 29aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <signal.h> 30aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <math.h> 31aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 32aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include "rt_names.h" 33aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 34aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#include <SNAPSHOT.h> 35aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 36aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint reset_history = 0; 37aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint ignore_history = 0; 38aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint no_output = 0; 39aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint no_update = 0; 40aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint scan_interval = 0; 41aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint time_constant = 0; 42aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint dump_zeros = 0; 43aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerunsigned long magic_number = 0; 44aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerdouble W; 45aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 464ffc44ca7c1ec7adb980b67ae1e1c8fa772777f1Yu Zhiguostatic int generic_proc_open(const char *env, const char *name) 47aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 48aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger char store[1024]; 49aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger char *p = getenv(env); 50aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (!p) { 51aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger p = getenv("PROC_ROOT") ? : "/proc"; 52aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger snprintf(store, sizeof(store)-1, "%s/%s", p, name); 53aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger p = store; 54aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 554ffc44ca7c1ec7adb980b67ae1e1c8fa772777f1Yu Zhiguo return open(p, O_RDONLY); 56aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 57aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 58aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint net_rtacct_open(void) 59aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 60aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger return generic_proc_open("PROC_NET_RTACCT", "net/rt_acct"); 61aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 62aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 63aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger__u32 rmap[256/4]; 64aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 65aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerstruct rtacct_data 66aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 67aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger __u32 ival[256*4]; 68aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 69aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger unsigned long long val[256*4]; 70aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger double rate[256*4]; 71737f15f6da0ed7512220f6fa5244a39777de4e0dshemminger char signature[128]; 72aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger}; 73aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 74aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerstruct rtacct_data kern_db_static; 75aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 76aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerstruct rtacct_data *kern_db = &kern_db_static; 77aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerstruct rtacct_data *hist_db; 78aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 79aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid nread(int fd, char *buf, int tot) 80aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 81aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int count = 0; 82aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 83aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger while (count < tot) { 84aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int n = read(fd, buf+count, tot-count); 85aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (n < 0) { 86aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (errno == EINTR) 87aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger continue; 88aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 89aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 90aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (n == 0) 91aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 92aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger count += n; 93aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 94aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 95aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 96aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 97aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger__u32 *read_kern_table(__u32 *tbl) 98aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 99aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger static __u32 *tbl_ptr; 100aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int fd; 101aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 102aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (magic_number) { 103aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (tbl_ptr != NULL) 104aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger return tbl_ptr; 105aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 106aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fd = open("/dev/mem", O_RDONLY); 107aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (fd < 0) { 108aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger perror("magic open"); 109aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 110aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 111aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger tbl_ptr = mmap(NULL, 4096, 112aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger PROT_READ, 113aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger MAP_SHARED, 114aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fd, magic_number); 115aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if ((unsigned long)tbl_ptr == ~0UL) { 116aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger perror("magic mmap"); 117aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 118aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 119aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger close(fd); 120aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger return tbl_ptr; 121aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 122aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 123aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fd = net_rtacct_open(); 124aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (fd >= 0) { 125aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger nread(fd, (char*)tbl, 256*16); 126aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger close(fd); 127aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } else { 128aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger memset(tbl, 0, 256*16); 129aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 130aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger return tbl; 131aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 132aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 133aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid format_rate(FILE *fp, double rate) 134aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 135aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger char temp[64]; 136aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 137aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (rate > 1024*1024) { 138aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger sprintf(temp, "%uM", (unsigned)rint(rate/(1024*1024))); 139aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, " %-10s", temp); 140aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } else if (rate > 1024) { 141aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger sprintf(temp, "%uK", (unsigned)rint(rate/1024)); 142aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, " %-10s", temp); 143aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } else 144aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, " %-10u", (unsigned)rate); 145aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 146aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 147aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid format_count(FILE *fp, unsigned long long val) 148aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 149aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (val > 1024*1024*1024) 150aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, " %10lluM", val/(1024*1024)); 151aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger else if (val > 1024*1024) 152aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, " %10lluK", val/1024); 153aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger else 154aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, " %10llu", val); 155aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 156aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 157aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid dump_abs_db(FILE *fp) 158aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 159aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int realm; 160aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger char b1[16]; 161aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 162aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (!no_output) { 163aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, "#%s\n", kern_db->signature); 164aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, 165aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 166aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 167aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 168aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 169aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 170aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"\n" 171ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger , "Realm", "BytesTo", "PktsTo", "BytesFrom", "PktsFrom"); 172aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, 173aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 174aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 175aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 176aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 177aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 178aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"\n" 179ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger , "", "BPSTo", "PPSTo", "BPSFrom", "PPSFrom"); 180aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 181aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 182aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 183aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (realm=0; realm<256; realm++) { 184aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int i; 185aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger unsigned long long *val; 186aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger double *rate; 187aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 188aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (!(rmap[realm>>5] & (1<<(realm&0x1f)))) 189aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger continue; 190aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 191aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger val = &kern_db->val[realm*4]; 192aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger rate = &kern_db->rate[realm*4]; 193aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 194aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (!dump_zeros && 195aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger !val[0] && !rate[0] && 196aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger !val[1] && !rate[1] && 197aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger !val[2] && !rate[2] && 198aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger !val[3] && !rate[3]) 199aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger continue; 200aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 201aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (hist_db) { 202aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger memcpy(&hist_db->val[realm*4], val, sizeof(*val)*4); 203aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 204aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 205aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (no_output) 206aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger continue; 207aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 208aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, "%-10s", rtnl_rtrealm_n2a(realm, b1, sizeof(b1))); 209aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (i = 0; i < 4; i++) 210ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger format_count(fp, val[i]); 211aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, "\n%-10s", ""); 212aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (i = 0; i < 4; i++) 213ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger format_rate(fp, rate[i]); 214aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, "\n"); 215aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 216aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 217aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 218aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 219aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid dump_incr_db(FILE *fp) 220aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 221aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int k, realm; 222aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger char b1[16]; 223aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 224aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (!no_output) { 225aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, "#%s\n", kern_db->signature); 226aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, 227aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 228aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 229aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 230aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 231aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 232aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"\n" 233ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger , "Realm", "BytesTo", "PktsTo", "BytesFrom", "PktsFrom"); 234aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, 235aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 236aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 237aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 238aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 239aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"%-10s " 240aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"\n" 241ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger , "", "BPSTo", "PPSTo", "BPSFrom", "PPSFrom"); 242aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 243aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 244aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (realm=0; realm<256; realm++) { 245aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int ovfl = 0; 246aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int i; 247aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger unsigned long long *val; 248aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger double *rate; 249aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger unsigned long long rval[4]; 250aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 251aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (!(rmap[realm>>5] & (1<<(realm&0x1f)))) 252aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger continue; 253aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 254aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger val = &kern_db->val[realm*4]; 255aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger rate = &kern_db->rate[realm*4]; 256aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 257aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (k=0; k<4; k++) { 258aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger rval[k] = val[k]; 259aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (rval[k] < hist_db->val[realm*4+k]) 260aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger ovfl = 1; 261aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger else 262aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger rval[k] -= hist_db->val[realm*4+k]; 263aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 264aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (ovfl) { 265aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (k=0; k<4; k++) 266aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger rval[k] = val[k]; 267aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 268aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (hist_db) { 269aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger memcpy(&hist_db->val[realm*4], val, sizeof(*val)*4); 270aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 271aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 272aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (no_output) 273aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger continue; 274aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 275aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (!dump_zeros && 276aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger !rval[0] && !rate[0] && 277aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger !rval[1] && !rate[1] && 278aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger !rval[2] && !rate[2] && 279aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger !rval[3] && !rate[3]) 280aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger continue; 281aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 282aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 283aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, "%-10s", rtnl_rtrealm_n2a(realm, b1, sizeof(b1))); 284aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (i = 0; i < 4; i++) 285ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger format_count(fp, rval[i]); 286aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, "\n%-10s", ""); 287aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (i = 0; i < 4; i++) 288ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger format_rate(fp, rate[i]); 289aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(fp, "\n"); 290aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 291aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 292aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 293aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 294aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerstatic int children; 295aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 296aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid sigchild(int signo) 297aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 298aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 299aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 300ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger/* Server side only: read kernel data, update tables, calculate rates. */ 301aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 302aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid update_db(int interval) 303aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 304aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int i; 305aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger __u32 *ival; 306aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger __u32 _ival[256*4]; 307aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 308aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger ival = read_kern_table(_ival); 309aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 310aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (i=0; i<256*4; i++) { 311aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger double sample; 312aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger __u32 incr = ival[i] - kern_db->ival[i]; 313aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 314aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (ival[i] == 0 && incr == 0 && 315aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger kern_db->val[i] == 0 && kern_db->rate[i] == 0) 316aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger continue; 317aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 318aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger kern_db->val[i] += incr; 319aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger kern_db->ival[i] = ival[i]; 320aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger sample = (double)(incr*1000)/interval; 321aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (interval >= scan_interval) { 322aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger kern_db->rate[i] += W*(sample-kern_db->rate[i]); 323aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } else if (interval >= 1000) { 324aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (interval >= time_constant) { 325aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger kern_db->rate[i] = sample; 326aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } else { 327aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger double w = W*(double)interval/scan_interval; 328aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger kern_db->rate[i] += w*(sample-kern_db->rate[i]); 329aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 330aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 331aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 332aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 333aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 334aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid send_db(int fd) 335aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 336aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int tot = 0; 337aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 338aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger while (tot < sizeof(*kern_db)) { 339aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int n = write(fd, ((char*)kern_db) + tot, sizeof(*kern_db)-tot); 340aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (n < 0) { 341aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (errno == EINTR) 342aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger continue; 343aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger return; 344aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 345aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger tot += n; 346aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 347aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 348aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 349aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 350aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 351aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger#define T_DIFF(a,b) (((a).tv_sec-(b).tv_sec)*1000 + ((a).tv_usec-(b).tv_usec)/1000) 352aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 353aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 354aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid pad_kern_table(struct rtacct_data *dat, __u32 *ival) 355aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 356aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int i; 357aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger memset(dat->rate, 0, sizeof(dat->rate)); 358aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (dat->ival != ival) 359aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger memcpy(dat->ival, ival, sizeof(dat->ival)); 360aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (i=0; i<256*4; i++) 361aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger dat->val[i] = ival[i]; 362aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 363aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 364aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingervoid server_loop(int fd) 365aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 366737f15f6da0ed7512220f6fa5244a39777de4e0dshemminger struct timeval snaptime = { 0 }; 367aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger struct pollfd p; 368aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger p.fd = fd; 369aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger p.events = p.revents = POLLIN; 370aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 371ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger sprintf(kern_db->signature, 372737f15f6da0ed7512220f6fa5244a39777de4e0dshemminger "%u.%lu sampling_interval=%d time_const=%d", 373ae665a522bd46bea44c5ea84c89c8b1731954170Stephen Hemminger (unsigned) getpid(), (unsigned long)random(), 374737f15f6da0ed7512220f6fa5244a39777de4e0dshemminger scan_interval/1000, time_constant/1000); 375aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 376aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger pad_kern_table(kern_db, read_kern_table(kern_db->ival)); 377aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 378aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger for (;;) { 379aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int status; 380aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int tdiff; 381aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger struct timeval now; 382aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger gettimeofday(&now, NULL); 383aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger tdiff = T_DIFF(now, snaptime); 384aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (tdiff >= scan_interval) { 385aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger update_db(tdiff); 386aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger snaptime = now; 387aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger tdiff = 0; 388aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 389aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (poll(&p, 1, tdiff + scan_interval) > 0 390aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger && (p.revents&POLLIN)) { 391aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int clnt = accept(fd, NULL, NULL); 392aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (clnt >= 0) { 393aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger pid_t pid; 394aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (children >= 5) { 395aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger close(clnt); 396aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } else if ((pid = fork()) != 0) { 397aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (pid>0) 398aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger children++; 399aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger close(clnt); 400aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } else { 401aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (tdiff > 0) 402aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger update_db(tdiff); 403aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger send_db(clnt); 404aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(0); 405aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 406aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 407aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 408aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger while (children && waitpid(-1, &status, WNOHANG) > 0) 409aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger children--; 410aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 411aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 412aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 413aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint verify_forging(int fd) 414aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 415aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger struct ucred cred; 416737f15f6da0ed7512220f6fa5244a39777de4e0dshemminger socklen_t olen = sizeof(cred); 417737f15f6da0ed7512220f6fa5244a39777de4e0dshemminger 418aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (getsockopt(fd, SOL_SOCKET, SO_PEERCRED, (void*)&cred, &olen) || 419aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger olen < sizeof(cred)) 420aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger return -1; 421aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (cred.uid == getuid() || cred.uid == 0) 422aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger return 0; 423aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger return -1; 424aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 425aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 426aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerstatic void usage(void) __attribute__((noreturn)); 427aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 428aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerstatic void usage(void) 429aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 430aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(stderr, 431aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger"Usage: rtacct [ -h?vVzrnasd:t: ] [ ListOfRealms ]\n" 432aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger ); 433aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 434aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 435aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 436aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemmingerint main(int argc, char *argv[]) 437aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger{ 438aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger char hist_name[128]; 439aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger struct sockaddr_un sun; 440aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int ch; 441aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger int fd; 442aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 443aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger while ((ch = getopt(argc, argv, "h?vVzrM:nasd:t:")) != EOF) { 444aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger switch(ch) { 445aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 'z': 446aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger dump_zeros = 1; 447aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger break; 448aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 'r': 449aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger reset_history = 1; 450aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger break; 451aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 'a': 452aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger ignore_history = 1; 453aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger break; 454aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 's': 455aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger no_update = 1; 456aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger break; 457aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 'n': 458aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger no_output = 1; 459aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger break; 460aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 'd': 461aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger scan_interval = 1000*atoi(optarg); 462aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger break; 463aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 't': 464aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (sscanf(optarg, "%d", &time_constant) != 1 || 465aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger time_constant <= 0) { 466aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(stderr, "rtacct: invalid time constant divisor\n"); 467aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 468aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 469aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger break; 470aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 'v': 471aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 'V': 472aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger printf("rtacct utility, iproute2-ss%s\n", SNAPSHOT); 473aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(0); 474aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 'M': 475aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger /* Some secret undocumented option, nobody 476aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger * is expected to ask about its sense. See? 477aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger */ 478aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger sscanf(optarg, "%lx", &magic_number); 479aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger break; 480aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case 'h': 481aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger case '?': 482aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger default: 483aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger usage(); 484aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 485aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 486aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 487aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger argc -= optind; 488aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger argv += optind; 489aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 490aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (argc) { 491aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger while (argc > 0) { 492aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger __u32 realm; 493aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (rtnl_rtrealm_a2n(&realm, argv[0])) { 494aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(stderr, "Warning: realm \"%s\" does not exist.\n", argv[0]); 495aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 496aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 497aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger rmap[realm>>5] |= (1<<(realm&0x1f)); 498aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger argc--; argv++; 499aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 500aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } else { 501aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger memset(rmap, ~0, sizeof(rmap)); 502aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger /* Always suppress zeros. */ 503aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger dump_zeros = 0; 504aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 505aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 506aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger sun.sun_family = AF_UNIX; 507aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger sun.sun_path[0] = 0; 508aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger sprintf(sun.sun_path+1, "rtacct%d", getuid()); 509aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 510aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (scan_interval > 0) { 511aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (time_constant == 0) 512aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger time_constant = 60; 513aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger time_constant *= 1000; 514aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger W = 1 - 1/exp(log(10)*(double)scan_interval/time_constant); 515aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) { 516aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger perror("rtacct: socket"); 517aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 518aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 519aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (bind(fd, (struct sockaddr*)&sun, 2+1+strlen(sun.sun_path+1)) < 0) { 520aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger perror("rtacct: bind"); 521aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 522aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 523aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (listen(fd, 5) < 0) { 524aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger perror("rtacct: listen"); 525aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 526aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 527a7a9ddbb675b7fbd184c3ab40817265b0b207a23Mike Frysinger if (daemon(0, 0)) { 528a7a9ddbb675b7fbd184c3ab40817265b0b207a23Mike Frysinger perror("rtacct: daemon"); 529a7a9ddbb675b7fbd184c3ab40817265b0b207a23Mike Frysinger exit(-1); 530a7a9ddbb675b7fbd184c3ab40817265b0b207a23Mike Frysinger } 531aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger signal(SIGPIPE, SIG_IGN); 532aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger signal(SIGCHLD, sigchild); 533aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger server_loop(fd); 534aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(0); 535aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 536aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 537aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (getenv("RTACCT_HISTORY")) 5388988b02ee1d04fdf07b913454b8d4e7b5e3092e5Stephen Hemminger snprintf(hist_name, sizeof(hist_name), "%s", getenv("RTACCT_HISTORY")); 539aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger else 540aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger sprintf(hist_name, "/tmp/.rtacct.u%d", getuid()); 541aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 542aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (reset_history) 543aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger unlink(hist_name); 544aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 545aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (!ignore_history || !no_update) { 546aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger struct stat stb; 547aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 548aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fd = open(hist_name, O_RDWR|O_CREAT|O_NOFOLLOW, 0600); 549aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (fd < 0) { 550aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger perror("rtacct: open history file"); 551aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 552aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 553aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (flock(fd, LOCK_EX)) { 554aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger perror("rtacct: flock history file"); 555aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 556aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 557aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (fstat(fd, &stb) != 0) { 558aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger perror("rtacct: fstat history file"); 559aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 560aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 561aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (stb.st_nlink != 1 || stb.st_uid != getuid()) { 562aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(stderr, "rtacct: something is so wrong with history file, that I prefer not to proceed.\n"); 563aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 564aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 565aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (stb.st_size != sizeof(*hist_db)) 5668988b02ee1d04fdf07b913454b8d4e7b5e3092e5Stephen Hemminger if (write(fd, kern_db, sizeof(*hist_db)) < 0) { 5678988b02ee1d04fdf07b913454b8d4e7b5e3092e5Stephen Hemminger perror("rtacct: write history file"); 5688988b02ee1d04fdf07b913454b8d4e7b5e3092e5Stephen Hemminger exit(-1); 5698988b02ee1d04fdf07b913454b8d4e7b5e3092e5Stephen Hemminger } 570aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 571aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger hist_db = mmap(NULL, sizeof(*hist_db), 572aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger PROT_READ|PROT_WRITE, 573aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger no_update ? MAP_PRIVATE : MAP_SHARED, 574aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fd, 0); 575aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 576aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if ((unsigned long)hist_db == ~0UL) { 577aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger perror("mmap"); 578aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(-1); 579aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 580aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 581aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (!ignore_history) { 582aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger FILE *tfp; 5839a230771c0a0b27e2f26bbbc29a93ad1ad591ce9Dan McGee long uptime = -1; 584aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if ((tfp = fopen("/proc/uptime", "r")) != NULL) { 585aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (fscanf(tfp, "%ld", &uptime) != 1) 586aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger uptime = -1; 587aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fclose(tfp); 588aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 589aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 590aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (uptime >= 0 && time(NULL) >= stb.st_mtime+uptime) { 591aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(stderr, "rtacct: history is aged out, resetting\n"); 592aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger memset(hist_db, 0, sizeof(*hist_db)); 593aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 594aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 595aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 596aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger close(fd); 597aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 598aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 599aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) >= 0 && 600aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger (connect(fd, (struct sockaddr*)&sun, 2+1+strlen(sun.sun_path+1)) == 0 601aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger || (strcpy(sun.sun_path+1, "rtacct0"), 602aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger connect(fd, (struct sockaddr*)&sun, 2+1+strlen(sun.sun_path+1)) == 0)) 603aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger && verify_forging(fd) == 0) { 604aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger nread(fd, (char*)kern_db, sizeof(*kern_db)); 605aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (hist_db && hist_db->signature[0] && 606aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger strcmp(kern_db->signature, hist_db->signature)) { 607aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(stderr, "rtacct: history is stale, ignoring it.\n"); 608aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger hist_db = NULL; 609aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 610aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger close(fd); 611aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } else { 612aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (fd >= 0) 613aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger close(fd); 614aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 615aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (hist_db && hist_db->signature[0] && 616aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger strcmp(hist_db->signature, "kernel")) { 617aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger fprintf(stderr, "rtacct: history is stale, ignoring it.\n"); 618aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger hist_db = NULL; 619aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 620aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 621aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger pad_kern_table(kern_db, read_kern_table(kern_db->ival)); 622aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger strcpy(kern_db->signature, "kernel"); 623aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger } 624aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 625aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger if (ignore_history || hist_db == NULL) 626aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger dump_abs_db(stdout); 627aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger else 628aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger dump_incr_db(stdout); 629aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger 630aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger exit(0); 631aba5acdfdb347d2c21fc67d613d83d4430ca3937osdl.org!shemminger} 632