X-Git-Url: http://git.harvie.cz/?p=svn%2FPrometheus-QoS%2F.git;a=blobdiff_plain;f=prometheus.c;fp=prometheus.c;h=7adff3cc6b493b5b9481a6284af7b6265e799b7d;hp=5a9492e3e7592cc4cf8fac51f5a98e674039016e;hb=4deb2d6f2b43232cf5e782ebb1deb9b0e041def9;hpb=dba7666bdacf8846560e5414b6c9535434e98f8b
diff --git a/prometheus.c b/prometheus.c
index 5a9492e..7adff3c 100644
--- a/prometheus.c
+++ b/prometheus.c
@@ -2,12 +2,12 @@
/* Prometheus QoS - you can "steal fire" from your ISP */
/* "fair-per-IP" quality of service (QoS) utility */
/* requires Linux 2.4.x or 2.6.x with HTB support */
-/* Copyright(C) 2005-2019 Michael Polak, Arachne Aerospace */
+/* Copyright(C) 2005-2017 Michael Polak, Arachne Aerospace */
/* iptables-restore support Copyright(C) 2007-2008 ludva */
/* Credit: CZFree.Net,Martin Devera,Netdave,Aquarius,Gandalf */
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
-/* Modified by: xChaos, 20190212
+/* Modified by: xChaos, 20171106
ludva, 20080415
Prometheus QoS is free software; you can redistribute it and/or
@@ -29,7 +29,7 @@
#include "cll1-0.6.2.h"
#include "ipstruct.h"
-const char *version = "0.9.1";
+const char *version = "0.9.0-b";
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
/* Versions: 0.9.0 is development release, 1.0 will be "stable" */
@@ -40,7 +40,7 @@ const char *version = "0.9.1";
/* Warning: unofficial Github mirror is not supported by author! */
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
-const char *stats_html_signature = "Statistics generated by Prometheus QoS version %s
GPL+Copyright(C)2005-2019 Michael Polak, Arachne Labs\n";
+const char *stats_html_signature = "Statistics generated by Prometheus QoS version %s
GPL+Copyright(C)2005-2017 Michael Polak, Arachne Labs\n";
#define STRLEN 512
#undef DEBUG
@@ -87,8 +87,6 @@ char *final_chain = "DROP"; /* REJECT would be better, but it is impossible
char *mark = "MARK";
char *mark_iptables = "MARK --set-mark ";
int dry_run = FALSE; /* preview - use puts() instead of system() */
-int mix_new_hosts = FALSE; /* execute only commands not already in log of iptables */
-char *log_file_mode = "w"; /* overwrite command and iptables log files, except append in -x mode */
char *iptablespreamble = "*mangle\n:PREROUTING ACCEPT [0:0]\n:POSTROUTING ACCEPT [0:0]\n:INPUT ACCEPT [0:0]\n:OUTPUT ACCEPT [0:0]\n:FORWARD ACCEPT [0:0]";
char *ip6preamble = "-A FORWARD -p ipv6-icmp -j ACCEPT\n-A POSTROUTING -p ipv6-icmp -j ACCEPT\n-A FORWARD -s fe80::/10 -j ACCEPT\n-A FORWARD -d ff00::/8 -j ACCEPT\n-A POSTROUTING -s fe80::/10 -j ACCEPT\n-A POSTROUTING -d ff00::/8 -j ACCEPT";
FILE *iptables_file = NULL;
@@ -128,12 +126,12 @@ int ip_count = 0;
FILE *log_file = NULL;
char *kwd = "via-prometheus"; /* /etc/hosts comment, eg. #qos-64-128 */
-const int highest_priority = 0; /* highest HTB priority (HTB built-in value is 0) */
-const int lowest_priority = 7; /* lowest HTB priority /include/uapi/linux/pkt_sched.h: #define TC_HTB_NUMPRIO 8 */
-const int idxtable_treshold1 = 24; /* this is no longer configurable */
-const int idxtable_treshold2 = 12; /* this is no longer configurable */
-const int idxtable_bitmask1 = 3; /* this is no longer configurable */
-const int idxtable_bitmask2 = 3; /* this is no longer configurable */
+const int highest_priority = 0; /* highest HTB priority (HTB built-in value is 0) */
+const int lowest_priority = 7; /* lowest HTB priority /include/uapi/linux/pkt_sched.h: #define TC_HTB_NUMPRIO 8 */
+const int idxtable_treshold1 = 24; /* this is no longer configurable */
+const int idxtable_treshold2 = 12; /* this is no longer configurable */
+const int idxtable_bitmask1 = 3; /* this is no longer configurable */
+const int idxtable_bitmask2 = 3; /* this is no longer configurable */
struct IP *ips = NULL, *networks = NULL, *ip, *sharedip;
struct Group *groups = NULL, *group;
@@ -142,7 +140,6 @@ struct Macro *macro, *macros = NULL;
struct Index *idxs = NULL, *idx, *metaindex;
struct Interface *interfaces = NULL, *interface;
struct QosFreeInterface *qosfreeinterfaces = NULL, *qosfreeinterface;
-struct Textfile *previous_commands = NULL, *previous_iptables = NULL, *previous_ip6tables = NULL, *previous_classmap = NULL, *textline;
#define FREE_CLASS 3
#define OVERLIMIT_CLASS 4
@@ -187,6 +184,7 @@ const char *tr_odd_even(void)
}
}
+
/* ====== iptables indexes are used to reduce complexity to log8(N) ===== */
char *index_id(char *ip, int bitmask);
@@ -380,31 +378,9 @@ void safe_run(char *cmd)
}
else
{
- int skip_cmd = FALSE;
- if(mix_new_hosts)
- {
- skip_cmd = FALSE;
- for_each(textline, previous_commands)
- {
- if(eq(textline->str, cmd))
- {
- skip_cmd = TRUE;
- break;
- }
- }
- }
-
- if(!skip_cmd)
- {
- if(mix_new_hosts)
- {
- printf("Executing command: %s\n", cmd);
- }
- else
- system(cmd);
- }
+ system(cmd);
}
- if(log_file && !mix_new_hosts)//!!debug
+ if(log_file)
{
fprintf(log_file,"%s\n",cmd);
}
@@ -412,41 +388,14 @@ void safe_run(char *cmd)
void iptables_save_line(char *line, int ipv6)
{
- const char *cmd = NULL;
-
if(ipv6)
{
- if(mix_new_hosts)
- for_each(textline, previous_ip6tables)
- if(eq(textline->str, line))
- {
- printf("Match: %s\n",line);
- return;
- }
-
- if(!mix_new_hosts) //!!debug
- fprintf(ip6tables_file,"%s\n",line);
- cmd = ip6tables;
+ fprintf(ip6tables_file,"%s\n",line);
}
else
{
- if(mix_new_hosts)
- for_each(textline, previous_iptables)
- if(eq(textline->str, line))
- {
- printf("Match: %s\n",line);
- return;
- }
-
- if(!mix_new_hosts) //!!debug
- fprintf(iptables_file,"%s\n",line);
- cmd = iptables;
+ fprintf(iptables_file,"%s\n",line);
}
- if(mix_new_hosts)
- {
-
- printf("Executing command: %s %s\n",cmd, line);
- }
}
#define IPv4 FALSE
@@ -520,7 +469,7 @@ program
int just_preview = FALSE; /* preview - generate just stats */
int start_shaping = FALSE; /* apply FUP - requires classmap file */
int stop_shaping = FALSE; /* lift FUP - requires classmap file */
- int reduce_ceil = 0; /* allow only rate+(ceil-rate)/2, /4, etc. */
+ int reduce_ceil = 0; /* allow only rate+(ceil-rate)/2, /4, etc. */
int just_logs = FALSE; /* just parse logs */
int run = FALSE;
int total = 0;
@@ -529,7 +478,7 @@ program
printf("\n\
Prometheus QoS - \"fair-per-IP\" Quality of Service setup utility.\n\
-Version %s - Copyright (C)2005-2019 Michael Polak, Arachne Labs\n\
+Version %s - Copyright (C)2005-2017 Michael Polak, Arachne Labs\n\
iptables-restore & burst tunning & classify modification by Ludva\n\
Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
@@ -538,21 +487,20 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
{
argument("-c") { nextargument(config); }
argument("-h") { nextargument(althosts);}
- argument("-d") { run = TRUE; dry_run = TRUE; }
- argument("-f") { run = TRUE; just_flush = TRUE; }
- argument("-9") { run = TRUE; just_flush = 9; }
- argument("-p") { run = TRUE; just_preview = TRUE; }
- argument("-q") { run = TRUE; just_preview = TRUE; stop_shaping = TRUE; }
- argument("-2") { run = TRUE; just_preview = TRUE; reduce_ceil = 2; }
- argument("-4") { run = TRUE; just_preview = TRUE; reduce_ceil = 4; }
- argument("-s") { run = TRUE; just_preview = TRUE; start_shaping = TRUE; }
- argument("-x") { run = TRUE; mix_new_hosts = TRUE; log_file_mode = "a"; }
- argument("-r") { run = TRUE; }
- argument("-n") { run = TRUE; nodelay = TRUE; }
- argument("-a") { run = TRUE; just_networks = TRUE; }
- argument("-l") { just_logs = TRUE; }
- argument("-m") { just_logs = TRUE; }
- argument("-y") { just_logs = TRUE; }
+ argument("-d") { run=TRUE; dry_run=TRUE; }
+ argument("-f") { run=TRUE; just_flush=TRUE; }
+ argument("-9") { run=TRUE; just_flush=9; }
+ argument("-p") { run=TRUE; just_preview=TRUE; }
+ argument("-q") { run=TRUE; just_preview=TRUE; stop_shaping=TRUE; }
+ argument("-2") { run=TRUE; just_preview=TRUE; reduce_ceil=2; }
+ argument("-4") { run=TRUE; just_preview=TRUE; reduce_ceil=4; }
+ argument("-s") { run=TRUE; just_preview=TRUE; start_shaping=TRUE; }
+ argument("-r") { run=TRUE; }
+ argument("-n") { run=TRUE; nodelay=TRUE; }
+ argument("-a") { run=TRUE; just_networks=TRUE; }
+ argument("-l") { just_logs=TRUE; }
+ argument("-m") { just_logs=TRUE; }
+ argument("-y") { just_logs=TRUE; }
argument("-?") { help(); exit(0); }
argument("--help") { help(); exit(0); }
argument("-v") { exit(0); }
@@ -638,7 +586,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
hosts = althosts;
}
- if(!mix_new_hosts && just_flush<9)
+ if(just_flush<9)
{
/*-----------------------------------------------------------------*/
puts("Parsing iptables verbose output ...");
@@ -682,6 +630,8 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
}
done; /* ugly macro end */
+
+
/*-----------------------------------------------------------------*/
printf("Parsing class defintion file %s ...\n", hosts);
/*-----------------------------------------------------------------*/
@@ -695,27 +645,6 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
}
*/
- if(mix_new_hosts)
- {
- char *eoln;
- /*-----------------------------------------------------------------*/
- printf("Loading log of previously executed commands %s ...\n", cmdlog);
- /*-----------------------------------------------------------------*/
- load(textline, previous_commands, cmdlog, Textfile, str);
- /*-----------------------------------------------------------------*/
- printf("Loading log of iptables-restore commands %s ...\n", iptablesfile);
- /*-----------------------------------------------------------------*/
- load(textline, previous_iptables, iptablesfile, Textfile, str);
- /*-----------------------------------------------------------------*/
- printf("Loading log of ip6tables-restore commands %s ...\n", ip6tablesfile);
- /*-----------------------------------------------------------------*/
- load(textline, previous_ip6tables, ip6tablesfile, Textfile, str);
- /*-----------------------------------------------------------------*/
- printf("Loading map of IP addresses to tc classes %s ...\n", classmap);
- /*-----------------------------------------------------------------*/
- load(textline, previous_classmap, classmap, Textfile, str);
- }
-
/*-----------------------------------------------------------------*/
puts("Resolving shared connections ...");
/*-----------------------------------------------------------------*/
@@ -756,7 +685,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
}
}
- if(!mix_new_hosts && enable_credit && just_flush<9)
+ if(enable_credit && just_flush<9)
{
/*-----------------------------------------------------------------*/
printf("Parsing credit file %s ...\n", credit);
@@ -778,36 +707,33 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
if(!just_preview)
{
- if(!mix_new_hosts)
- {
- /*-----------------------------------------------------------------*/
- puts("Initializing iptables and tc classes ...");
- /*-----------------------------------------------------------------*/
+ /*-----------------------------------------------------------------*/
+ puts("Initializing iptables and tc classes ...");
+ /*-----------------------------------------------------------------*/
- iptables_file = fopen(iptablesfile, "w");
- if(iptables_file == NULL)
- {
+ iptables_file = fopen(iptablesfile, "w");
+ if(iptables_file == NULL)
+ {
perror(iptablesfile);
exit(-1);
- }
- iptables_save_line(iptablespreamble, IPv4);
+ }
+ iptables_save_line(iptablespreamble, IPv4);
- if(ip6prefix)
+ if(ip6prefix)
+ {
+ ip6tables_file = fopen(ip6tablesfile, "w");
+ if(ip6tables_file == NULL)
{
- ip6tables_file = fopen(ip6tablesfile, "w");
- if(ip6tables_file == NULL)
- {
- perror(ip6tablesfile);
- exit(-1);
- }
- iptables_save_line(iptablespreamble, IPv6);
- iptables_save_line(ip6preamble, IPv6);
+ perror(ip6tablesfile);
+ exit(-1);
}
-
- run_iptables_restore();
+ iptables_save_line(iptablespreamble, IPv6);
+ iptables_save_line(ip6preamble, IPv6);
}
+
+ run_iptables_restore();
- log_file = fopen(cmdlog, log_file_mode); /* overwrite command and iptables log files, except append in -x mode */
+ log_file = fopen(cmdlog, "w");
if(log_file == NULL)
{
perror(cmdlog);
@@ -819,17 +745,14 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
sprintf(str,"%s qdisc del dev %s root 2>/dev/null", tc, interface->name);
safe_run(str);
}
-
- if(!mix_new_hosts)
- {
- iptables_file = fopen(iptablesfile, log_file_mode); /* overwrite command and iptables log files, except append in -x mode */
- iptables_save_line(iptablespreamble, IPv4);
- if(ip6prefix)
- {
- ip6tables_file = fopen(ip6tablesfile, log_file_mode); /* overwrite command and iptables log files, except append in -x mode */
- iptables_save_line(iptablespreamble, IPv6);
- iptables_save_line(ip6preamble, IPv6);
- }
+
+ iptables_file=fopen(iptablesfile,"w");
+ iptables_save_line(iptablespreamble, IPv4);
+ if(ip6prefix)
+ {
+ ip6tables_file=fopen(ip6tablesfile,"w");
+ iptables_save_line(iptablespreamble, IPv6);
+ iptables_save_line(ip6preamble, IPv6);
}
if(qos_free_zone && *qos_free_zone!='0') /* this is currently supported only for IPv4 */
@@ -1002,7 +925,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
exit(0);
}
- if(!mix_new_hosts && !just_preview)
+ if(!just_preview)
{
if(!dry_run && !nodelay && qos_free_delay)
{
@@ -1026,13 +949,10 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
}
}
- if(mix_new_hosts)
- {
- /*-----------------------------------------------------------------*/
- puts("Locating heavy downloaders and generating root classes ...");
- /*-----------------------------------------------------------------*/
- sort(ip,ips,desc_order_by,traffic);
- }
+ /*-----------------------------------------------------------------*/
+ puts("Locating heavy downloaders and generating root classes ...");
+ /*-----------------------------------------------------------------*/
+ sort(ip,ips,desc_order_by,traffic);
/*-----------------------------------------------------------------*/
for_each(interface, interfaces)
@@ -1267,7 +1187,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
json_traffic=json_preview;
}
- if(!mix_new_hosts && !dry_run && !just_flush)
+ if(!dry_run && !just_flush)
{
/*-----------------------------------------------------------------*/
printf("Writing json traffic overview %s ... ", json_traffic);
@@ -1298,8 +1218,8 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
printf("Statistics preview generated (-%c switch) - now exiting ...\n", swchar);
exit(0);
- }
-
+ }
+
i=0;
#ifdef DEBUG
printf("%-22s %-15s mark\n","name","ip");
@@ -1420,7 +1340,6 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
}
i++;
}
- /* classmap */
if(ip->min && f > 0)
{
fprintf(f, "%s %d\n", ip->addr, ip->mark);
@@ -1488,10 +1407,7 @@ Credit: CZFree.Net, Martin Devera, Netdave, Aquarius, Gandalf\n\n",version);
safe_run(str);
}
printf("Total IP count: %d\n", i);
- if(!mix_new_hosts)
- {
- run_iptables_restore();
- }
+ run_iptables_restore();
if(log_file)
{
fclose(log_file);