-
Notifications
You must be signed in to change notification settings - Fork 3
/
dn-vnstat.sh
2268 lines (2071 loc) · 72.7 KB
/
dn-vnstat.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#!/bin/sh
#################################################
## ##
## vnStat on Merlin ##
## for AsusWRT-Merlin routers ##
## ##
## Concept by dev_null ##
## Implemented by Jack Yaz ##
## github.com/de-vnull/vnstat-on-merlin ##
## ##
#################################################
# Last Modified: 2024-Sep-22
#------------------------------------------------
######## Shellcheck directives ######
# shellcheck disable=SC1091
# shellcheck disable=SC2009
# shellcheck disable=SC2016
# shellcheck disable=SC2018
# shellcheck disable=SC2019
# shellcheck disable=SC2059
# shellcheck disable=SC2086
# shellcheck disable=SC2154
# shellcheck disable=SC2155
# shellcheck disable=SC2181
#################################################
### Start of script variables ###
readonly SCRIPT_NAME="dn-vnstat"
readonly SCRIPT_VERSION="v2.0.6"
SCRIPT_BRANCH="main"
SCRIPT_REPO="https://raw.githubusercontent.com/de-vnull/vnstat-on-merlin/$SCRIPT_BRANCH"
readonly SCRIPT_DIR="/jffs/addons/$SCRIPT_NAME.d"
readonly SCRIPT_WEBPAGE_DIR="$(readlink /www/user)"
readonly SCRIPT_WEB_DIR="$SCRIPT_WEBPAGE_DIR/$SCRIPT_NAME"
readonly SHARED_DIR="/jffs/addons/shared-jy"
readonly SHARED_REPO="https://raw.githubusercontent.com/jackyaz/shared-jy/master"
readonly SHARED_WEB_DIR="$SCRIPT_WEBPAGE_DIR/shared-jy"
[ -z "$(nvram get odmpid)" ] && ROUTER_MODEL="$(nvram get productid)" || ROUTER_MODEL="$(nvram get odmpid)"
[ -f /opt/bin/sqlite3 ] && SQLITE3_PATH=/opt/bin/sqlite3 || SQLITE3_PATH=/usr/sbin/sqlite3
### End of script variables ###
### Start of output format variables ###
readonly CRIT="\\e[41m"
readonly ERR="\\e[31m"
readonly WARN="\\e[33m"
readonly PASS="\\e[32m"
readonly BOLD="\\e[1m"
readonly SETTING="${BOLD}\\e[36m"
readonly CLEARFORMAT="\\e[0m"
### End of output format variables ###
# $1 = print to syslog, $2 = message to print, $3 = log level
Print_Output(){
if [ "$1" = "true" ]; then
logger -t "$SCRIPT_NAME" "$2"
fi
printf "${BOLD}${3}%s${CLEARFORMAT}\\n\\n" "$2"
}
### Check firmware version contains the "am_addons" feature flag ###
Firmware_Version_Check(){
if nvram get rc_support | grep -qF "am_addons"; then
return 0
else
return 1
fi
}
### Create "lock" file to ensure script only allows 1 concurrent process for certain actions ###
### Code for these functions inspired by https://github.com/Adamm00 - credit to @Adamm ###
Check_Lock(){
if [ -f "/tmp/$SCRIPT_NAME.lock" ]; then
ageoflock=$(($(date +%s) - $(date +%s -r /tmp/$SCRIPT_NAME.lock)))
if [ "$ageoflock" -gt 600 ]; then
Print_Output true "Stale lock file found (>600 seconds old) - purging lock" "$ERR"
kill "$(sed -n '1p' /tmp/$SCRIPT_NAME.lock)" >/dev/null 2>&1
Clear_Lock
echo "$$" > "/tmp/$SCRIPT_NAME.lock"
return 0
else
Print_Output true "Lock file found (age: $ageoflock seconds)" "$ERR"
if [ -z "$1" ]; then
exit 1
else
if [ "$1" = "webui" ]; then
echo 'var vnstatstatus = "LOCKED";' > /tmp/detect_vnstat.js
exit 1
fi
return 1
fi
fi
else
echo "$$" > "/tmp/$SCRIPT_NAME.lock"
return 0
fi
}
Clear_Lock(){
rm -f "/tmp/$SCRIPT_NAME.lock" 2>/dev/null
return 0
}
############################################################################
### Create "settings" in the custom_settings file, used by the WebUI for version information and script updates ###
### local is the version of the script installed, server is the version on Github ###
Set_Version_Custom_Settings(){
SETTINGSFILE="/jffs/addons/custom_settings.txt"
case "$1" in
local)
if [ -f "$SETTINGSFILE" ]; then
if [ "$(grep -c "dnvnstat_version_local" $SETTINGSFILE)" -gt 0 ]; then
if [ "$2" != "$(grep "dnvnstat_version_local" /jffs/addons/custom_settings.txt | cut -f2 -d' ')" ]; then
sed -i "s/dnvnstat_version_local.*/dnvnstat_version_local $2/" "$SETTINGSFILE"
fi
else
echo "dnvnstat_version_local $2" >> "$SETTINGSFILE"
fi
else
echo "dnvnstat_version_local $2" >> "$SETTINGSFILE"
fi
;;
server)
if [ -f "$SETTINGSFILE" ]; then
if [ "$(grep -c "dnvnstat_version_server" $SETTINGSFILE)" -gt 0 ]; then
if [ "$2" != "$(grep "dnvnstat_version_server" /jffs/addons/custom_settings.txt | cut -f2 -d' ')" ]; then
sed -i "s/dnvnstat_version_server.*/dnvnstat_version_server $2/" "$SETTINGSFILE"
fi
else
echo "dnvnstat_version_server $2" >> "$SETTINGSFILE"
fi
else
echo "dnvnstat_version_server $2" >> "$SETTINGSFILE"
fi
;;
esac
}
### Checks for changes to Github version of script and returns reason for change (version or md5/minor), local version and server version ###
Update_Check(){
echo 'var updatestatus = "InProgress";' > "$SCRIPT_WEB_DIR/detect_update.js"
doupdate="false"
localver=$(grep "SCRIPT_VERSION=" "/jffs/scripts/$SCRIPT_NAME" | grep -m1 -oE 'v[0-9]{1,2}([.][0-9]{1,2})([.][0-9]{1,2})')
/usr/sbin/curl -fsL --retry 3 "$SCRIPT_REPO/$SCRIPT_NAME.sh" | grep -qF "de-vnull" || { Print_Output true "404 error detected - stopping update" "$ERR"; return 1; }
serverver=$(/usr/sbin/curl -fsL --retry 3 "$SCRIPT_REPO/$SCRIPT_NAME.sh" | grep "SCRIPT_VERSION=" | grep -m1 -oE 'v[0-9]{1,2}([.][0-9]{1,2})([.][0-9]{1,2})')
if [ "$localver" != "$serverver" ]; then
doupdate="version"
Set_Version_Custom_Settings server "$serverver"
echo 'var updatestatus = "'"$serverver"'";' > "$SCRIPT_WEB_DIR/detect_update.js"
else
localmd5="$(md5sum "/jffs/scripts/$SCRIPT_NAME" | awk '{print $1}')"
remotemd5="$(curl -fsL --retry 3 "$SCRIPT_REPO/$SCRIPT_NAME.sh" | md5sum | awk '{print $1}')"
if [ "$localmd5" != "$remotemd5" ]; then
doupdate="md5"
Set_Version_Custom_Settings server "$serverver-hotfix"
echo 'var updatestatus = "'"$serverver-hotfix"'";' > "$SCRIPT_WEB_DIR/detect_update.js"
fi
fi
if [ "$doupdate" = "false" ]; then
echo 'var updatestatus = "None";' > "$SCRIPT_WEB_DIR/detect_update.js"
fi
echo "$doupdate,$localver,$serverver"
}
### Updates the script from Github including any secondary files ###
### Accepts arguments of:
### force - download from server even if no change detected
### unattended - don't return user to script CLI menu
Update_Version(){
if [ -z "$1" ]; then
updatecheckresult="$(Update_Check)"
isupdate="$(echo "$updatecheckresult" | cut -f1 -d',')"
localver="$(echo "$updatecheckresult" | cut -f2 -d',')"
serverver="$(echo "$updatecheckresult" | cut -f3 -d',')"
if [ "$isupdate" = "version" ]; then
Print_Output true "New version of $SCRIPT_NAME available - $serverver" "$PASS"
elif [ "$isupdate" = "md5" ]; then
Print_Output true "MD5 hash of $SCRIPT_NAME does not match - hotfix available - $serverver" "$PASS"
fi
if [ "$isupdate" != "false" ]; then
printf "\\n${BOLD}Do you want to continue with the update? (y/n)${CLEARFORMAT} "
read -r confirm
case "$confirm" in
y|Y)
printf "\\n"
Update_File shared-jy.tar.gz
Update_File vnstat-ui.asp
Update_File vnstat.conf
Update_File S33vnstat
/usr/sbin/curl -fsL --retry 3 "$SCRIPT_REPO/$SCRIPT_NAME.sh" -o "/jffs/scripts/$SCRIPT_NAME" && Print_Output true "$SCRIPT_NAME successfully updated"
chmod 0755 "/jffs/scripts/$SCRIPT_NAME"
Set_Version_Custom_Settings local "$serverver"
Set_Version_Custom_Settings server "$serverver"
Clear_Lock
PressEnter
exec "$0"
exit 0
;;
*)
printf "\\n"
Clear_Lock
return 1
;;
esac
else
Print_Output true "No updates available - latest is $localver" "$WARN"
Clear_Lock
fi
fi
if [ "$1" = "force" ]; then
serverver=$(/usr/sbin/curl -fsL --retry 3 "$SCRIPT_REPO/$SCRIPT_NAME.sh" | grep "SCRIPT_VERSION=" | grep -m1 -oE 'v[0-9]{1,2}([.][0-9]{1,2})([.][0-9]{1,2})')
Print_Output true "Downloading latest version ($serverver) of $SCRIPT_NAME" "$PASS"
Update_File shared-jy.tar.gz
Update_File vnstat-ui.asp
Update_File vnstat.conf
Update_File S33vnstat
/usr/sbin/curl -fsL --retry 3 "$SCRIPT_REPO/$SCRIPT_NAME.sh" -o "/jffs/scripts/$SCRIPT_NAME" && Print_Output true "$SCRIPT_NAME successfully updated"
chmod 0755 "/jffs/scripts/$SCRIPT_NAME"
Set_Version_Custom_Settings local "$serverver"
Set_Version_Custom_Settings server "$serverver"
Clear_Lock
if [ -z "$2" ]; then
PressEnter
exec "$0"
elif [ "$2" = "unattended" ]; then
exec "$0" postupdate
fi
exit 0
fi
}
Validate_Number(){
if [ "$1" -eq "$1" ] 2>/dev/null; then
return 0
else
return 1
fi
}
Validate_Bandwidth(){
if echo "$1" | /bin/grep -oq "^[0-9]*\.\?[0-9]\?[0-9]$"; then
return 0
else
return 1
fi
}
### Perform relevant actions for secondary files when being updated ###
Update_File(){
if [ "$1" = "vnstat-ui.asp" ]; then ### WebUI page
tmpfile="/tmp/$1"
Download_File "$SCRIPT_REPO/$1" "$tmpfile"
if ! diff -q "$tmpfile" "$SCRIPT_DIR/$1" >/dev/null 2>&1; then
if [ -f "$SCRIPT_DIR/$1" ]; then
Get_WebUI_Page "$SCRIPT_DIR/$1"
sed -i "\\~$MyPage~d" /tmp/menuTree.js
rm -f "$SCRIPT_WEBPAGE_DIR/$MyPage" 2>/dev/null
fi
Download_File "$SCRIPT_REPO/$1" "$SCRIPT_DIR/$1"
Print_Output true "New version of $1 downloaded" "$PASS"
Mount_WebUI
fi
rm -f "$tmpfile"
elif [ "$1" = "shared-jy.tar.gz" ]; then ### shared web resources
if [ ! -f "$SHARED_DIR/$1.md5" ]; then
Download_File "$SHARED_REPO/$1" "$SHARED_DIR/$1"
Download_File "$SHARED_REPO/$1.md5" "$SHARED_DIR/$1.md5"
tar -xzf "$SHARED_DIR/$1" -C "$SHARED_DIR"
rm -f "$SHARED_DIR/$1"
Print_Output true "New version of $1 downloaded" "$PASS"
else
localmd5="$(cat "$SHARED_DIR/$1.md5")"
remotemd5="$(curl -fsL --retry 3 "$SHARED_REPO/$1.md5")"
if [ "$localmd5" != "$remotemd5" ]; then
Download_File "$SHARED_REPO/$1" "$SHARED_DIR/$1"
Download_File "$SHARED_REPO/$1.md5" "$SHARED_DIR/$1.md5"
tar -xzf "$SHARED_DIR/$1" -C "$SHARED_DIR"
rm -f "$SHARED_DIR/$1"
Print_Output true "New version of $1 downloaded" "$PASS"
fi
fi
elif [ "$1" = "S33vnstat" ]; then ### Entware S script to launch vnstat
tmpfile="/tmp/$1"
Download_File "$SCRIPT_REPO/$1" "$tmpfile"
if ! diff -q "$tmpfile" "/opt/etc/init.d/$1" >/dev/null 2>&1; then
if [ -f /opt/etc/init.d/S33vnstat ]; then
/opt/etc/init.d/S33vnstat stop >/dev/null 2>&1
sleep 2
fi
Download_File "$SCRIPT_REPO/$1" "/opt/etc/init.d/$1"
chmod 0755 "/opt/etc/init.d/$1"
/opt/etc/init.d/S33vnstat start >/dev/null 2>&1
Print_Output true "New version of $1 downloaded" "$PASS"
fi
rm -f "$tmpfile"
elif [ "$1" = "vnstat.conf" ]; then ### vnstat config file
tmpfile="/tmp/$1"
Download_File "$SCRIPT_REPO/$1" "$tmpfile"
if [ ! -f "$SCRIPT_STORAGE_DIR/$1" ]; then
Download_File "$SCRIPT_REPO/$1" "$SCRIPT_STORAGE_DIR/$1.default"
Download_File "$SCRIPT_REPO/$1" "$SCRIPT_STORAGE_DIR/$1"
Print_Output true "$SCRIPT_STORAGE_DIR/$1 does not exist, downloading now." "$PASS"
elif [ -f "$SCRIPT_STORAGE_DIR/$1.default" ]; then
if ! diff -q "$tmpfile" "$SCRIPT_STORAGE_DIR/$1.default" >/dev/null 2>&1; then
Download_File "$SCRIPT_REPO/$1" "$SCRIPT_STORAGE_DIR/$1.default"
Print_Output true "New default version of $1 downloaded to $SCRIPT_STORAGE_DIR/$1.default, please compare against your $SCRIPT_STORAGE_DIR/$1" "$PASS"
fi
else
Download_File "$SCRIPT_REPO/$1" "$SCRIPT_STORAGE_DIR/$1.default"
Print_Output true "$SCRIPT_STORAGE_DIR/$1.default does not exist, downloading now. Please compare against your $SCRIPT_STORAGE_DIR/$1" "$PASS"
fi
rm -f "$tmpfile"
else
return 1
fi
}
Conf_FromSettings(){
SETTINGSFILE="/jffs/addons/custom_settings.txt"
TMPFILE="/tmp/dnvnstat_settings.txt"
if [ -f "$SETTINGSFILE" ]; then
if [ "$(grep "dnvnstat_" $SETTINGSFILE | grep -v "version" -c)" -gt 0 ]; then
Print_Output true "Updated settings from WebUI found, merging..." "$PASS"
cp -a "$SCRIPT_CONF" "$SCRIPT_CONF.bak"
cp -a "$SCRIPT_STORAGE_DIR/vnstat.conf" "$SCRIPT_STORAGE_DIR/vnstat.conf.bak"
grep "dnvnstat_" "$SETTINGSFILE" | grep -v "version" > "$TMPFILE"
sed -i "s/dnvnstat_//g;s/ /=/g" "$TMPFILE"
warningresetrequired="false"
while IFS='' read -r line || [ -n "$line" ]; do
SETTINGNAME="$(echo "$line" | cut -f1 -d'=' | awk '{ print toupper($1) }')"
SETTINGVALUE="$(echo "$line" | cut -f2 -d'=')"
if [ "$SETTINGNAME" != "MONTHROTATE" ]; then
if [ "$SETTINGNAME" = "DATAALLOWANCE" ]; then
if [ "$(echo "$SETTINGVALUE $(BandwidthAllowance check)" | awk '{print ($1 != $2)}')" -eq 1 ]; then
warningresetrequired="true"
fi
fi
sed -i "s/$SETTINGNAME=.*/$SETTINGNAME=$SETTINGVALUE/" "$SCRIPT_CONF"
elif [ "$SETTINGNAME" = "MONTHROTATE" ]; then
if [ "$SETTINGVALUE" != "$(AllowanceStartDay check)" ]; then
warningresetrequired="true"
fi
sed -i 's/^MonthRotate .*$/MonthRotate '"$SETTINGVALUE"'/' "$SCRIPT_STORAGE_DIR/vnstat.conf"
fi
done < "$TMPFILE"
grep 'dnvnstat_version' "$SETTINGSFILE" > "$TMPFILE"
sed -i "\\~dnvnstat_~d" "$SETTINGSFILE"
mv "$SETTINGSFILE" "$SETTINGSFILE.bak"
cat "$SETTINGSFILE.bak" "$TMPFILE" > "$SETTINGSFILE"
rm -f "$TMPFILE"
rm -f "$SETTINGSFILE.bak"
/opt/etc/init.d/S33vnstat restart >/dev/null 2>&1
TZ=$(cat /etc/TZ)
export TZ
if [ "$warningresetrequired" = "true" ]; then
Reset_Allowance_Warnings force
fi
Check_Bandwidth_Usage silent
Print_Output true "Merge of updated settings from WebUI completed successfully" "$PASS"
else
Print_Output false "No updated settings from WebUI found, no merge necessary" "$PASS"
fi
fi
}
### Create directories in filesystem if they do not exist ###
Create_Dirs(){
if [ ! -d "$SCRIPT_DIR" ]; then
mkdir -p "$SCRIPT_DIR"
fi
if [ ! -d "$SCRIPT_STORAGE_DIR" ]; then
mkdir -p "$SCRIPT_STORAGE_DIR"
fi
if [ ! -d "$IMAGE_OUTPUT_DIR" ]; then
mkdir -p "$IMAGE_OUTPUT_DIR"
fi
if [ ! -d "$CSV_OUTPUT_DIR" ]; then
mkdir -p "$CSV_OUTPUT_DIR"
fi
if [ ! -d "$SHARED_DIR" ]; then
mkdir -p "$SHARED_DIR"
fi
if [ ! -d "$SCRIPT_WEBPAGE_DIR" ]; then
mkdir -p "$SCRIPT_WEBPAGE_DIR"
fi
if [ ! -d "$SCRIPT_WEB_DIR" ]; then
mkdir -p "$SCRIPT_WEB_DIR"
fi
}
### Create symbolic links to /www/user for WebUI files to avoid file duplication ###
Create_Symlinks(){
rm -rf "${SCRIPT_WEB_DIR:?}/"* 2>/dev/null
ln -s /tmp/detect_vnstat.js "$SCRIPT_WEB_DIR/detect_vnstat.js" 2>/dev/null
ln -s "$SCRIPT_STORAGE_DIR/.vnstatusage" "$SCRIPT_WEB_DIR/vnstatusage.js" 2>/dev/null
ln -s "$VNSTAT_OUTPUT_FILE" "$SCRIPT_WEB_DIR/vnstatoutput.htm" 2>/dev/null
ln -s "$SCRIPT_CONF" "$SCRIPT_WEB_DIR/config.htm" 2>/dev/null
ln -s "$SCRIPT_STORAGE_DIR/vnstat.conf" "$SCRIPT_WEB_DIR/vnstatconf.htm" 2>/dev/null
ln -s "$IMAGE_OUTPUT_DIR" "$SCRIPT_WEB_DIR/images" 2>/dev/null
ln -s "$CSV_OUTPUT_DIR" "$SCRIPT_WEB_DIR/csv" 2>/dev/null
if [ ! -d "$SHARED_WEB_DIR" ]; then
ln -s "$SHARED_DIR" "$SHARED_WEB_DIR" 2>/dev/null
fi
}
##----------------------------------------##
## Modified by Martinski W. [2024-Sep-22] ##
##----------------------------------------##
Conf_Exists()
{
if [ -f "$SCRIPT_STORAGE_DIR/vnstat.conf" ]
then
restartvnstat="false"
if ! grep -q "^MaxBandwidth 1000" "$SCRIPT_STORAGE_DIR/vnstat.conf"; then
sed -i 's/^MaxBandwidth.*$/MaxBandwidth 1000/' "$SCRIPT_STORAGE_DIR/vnstat.conf"
restartvnstat="true"
fi
if ! grep -q "^TimeSyncWait 10" "$SCRIPT_STORAGE_DIR/vnstat.conf"; then
sed -i 's/^TimeSyncWait.*$/TimeSyncWait 10/' "$SCRIPT_STORAGE_DIR/vnstat.conf"
restartvnstat="true"
fi
if ! grep -q "^UpdateInterval 30" "$SCRIPT_STORAGE_DIR/vnstat.conf"; then
sed -i 's/^UpdateInterval.*$/UpdateInterval 30/' "$SCRIPT_STORAGE_DIR/vnstat.conf"
restartvnstat="true"
fi
if ! grep -q "^UnitMode 2" "$SCRIPT_STORAGE_DIR/vnstat.conf"; then
sed -i 's/^UnitMode.*$/UnitMode 2/' "$SCRIPT_STORAGE_DIR/vnstat.conf"
restartvnstat="true"
fi
if ! grep -q "^RateUnitMode 1" "$SCRIPT_STORAGE_DIR/vnstat.conf"; then
sed -i 's/^RateUnitMode.*$/RateUnitMode 1/' "$SCRIPT_STORAGE_DIR/vnstat.conf"
restartvnstat="true"
fi
if ! grep -q "^OutputStyle 0" "$SCRIPT_STORAGE_DIR/vnstat.conf"; then
sed -i 's/^OutputStyle.*$/OutputStyle 0/' "$SCRIPT_STORAGE_DIR/vnstat.conf"
restartvnstat="true"
fi
if ! grep -q '^MonthFormat "%Y-%m"' "$SCRIPT_STORAGE_DIR/vnstat.conf"; then
sed -i 's/^MonthFormat.*$/MonthFormat "%Y-%m"/' "$SCRIPT_STORAGE_DIR/vnstat.conf"
restartvnstat="true"
fi
if [ "$restartvnstat" = "true" ]; then
/opt/etc/init.d/S33vnstat restart >/dev/null 2>&1
Generate_Images silent
Generate_Stats silent
Check_Bandwidth_Usage silent
fi
else
Update_File vnstat.conf
fi
if [ -f "$SCRIPT_CONF" ]
then
dos2unix "$SCRIPT_CONF"
chmod 0644 "$SCRIPT_CONF"
sed -i -e 's/"//g' "$SCRIPT_CONF"
if ! grep -q "^DAILYEMAIL=" "$SCRIPT_CONF"; then
echo "DAILYEMAIL=none" >> "$SCRIPT_CONF"
fi
if ! grep -q "^USAGEEMAIL=" "$SCRIPT_CONF"; then
echo "USAGEEMAIL=false" >> "$SCRIPT_CONF"
fi
if ! grep -q "^DATAALLOWANCE=" "$SCRIPT_CONF"; then
echo "DATAALLOWANCE=1200.00" >> "$SCRIPT_CONF"
fi
if ! grep -q "^ALLOWANCEUNIT=" "$SCRIPT_CONF"; then
echo "ALLOWANCEUNIT=G" >> "$SCRIPT_CONF"
fi
if ! grep -q "^STORAGELOCATION=" "$SCRIPT_CONF"; then
echo "STORAGELOCATION=jffs" >> "$SCRIPT_CONF"
fi
if ! grep -q "^OUTPUTTIMEMODE=" "$SCRIPT_CONF"; then
echo "OUTPUTTIMEMODE=unix" >> "$SCRIPT_CONF"
fi
return 0
else
{ echo "DAILYEMAIL=none"; echo "USAGEEMAIL=false"; echo "DATAALLOWANCE=1200.00"
echo "ALLOWANCEUNIT=G"; echo "STORAGELOCATION=jffs"; echo "OUTPUTTIMEMODE=unix"
} > "$SCRIPT_CONF"
return 1
fi
}
### Add script hook to service-event and pass service_event argument and all other arguments passed to the service call ###
Auto_ServiceEvent(){
case $1 in
create)
if [ -f /jffs/scripts/service-event ]; then
STARTUPLINECOUNT=$(grep -c '# '"$SCRIPT_NAME" /jffs/scripts/service-event)
STARTUPLINECOUNTEX=$(grep -cx "/jffs/scripts/$SCRIPT_NAME service_event"' "$@" & # '"$SCRIPT_NAME" /jffs/scripts/service-event)
if [ "$STARTUPLINECOUNT" -gt 1 ] || { [ "$STARTUPLINECOUNTEX" -eq 0 ] && [ "$STARTUPLINECOUNT" -gt 0 ]; }; then
sed -i -e '/# '"$SCRIPT_NAME"'/d' /jffs/scripts/service-event
fi
if [ "$STARTUPLINECOUNTEX" -eq 0 ]; then
echo "/jffs/scripts/$SCRIPT_NAME service_event"' "$@" & # '"$SCRIPT_NAME" >> /jffs/scripts/service-event
fi
else
echo "#!/bin/sh" > /jffs/scripts/service-event
echo "" >> /jffs/scripts/service-event
echo "/jffs/scripts/$SCRIPT_NAME service_event"' "$@" & # '"$SCRIPT_NAME" >> /jffs/scripts/service-event
chmod 0755 /jffs/scripts/service-event
fi
;;
delete)
if [ -f /jffs/scripts/service-event ]; then
STARTUPLINECOUNT=$(grep -c '# '"$SCRIPT_NAME" /jffs/scripts/service-event)
if [ "$STARTUPLINECOUNT" -gt 0 ]; then
sed -i -e '/# '"$SCRIPT_NAME"'/d' /jffs/scripts/service-event
fi
fi
;;
esac
}
### Add script hook to post-mount and pass startup argument and all other arguments passed with the partition mount ###
Auto_Startup(){
case $1 in
create)
if [ -f /jffs/scripts/post-mount ]; then
STARTUPLINECOUNT=$(grep -c '# '"$SCRIPT_NAME" /jffs/scripts/post-mount)
STARTUPLINECOUNTEX=$(grep -cx "/jffs/scripts/$SCRIPT_NAME startup"' "$@" & # '"$SCRIPT_NAME" /jffs/scripts/post-mount)
if [ "$STARTUPLINECOUNT" -gt 1 ] || { [ "$STARTUPLINECOUNTEX" -eq 0 ] && [ "$STARTUPLINECOUNT" -gt 0 ]; }; then
sed -i -e '/# '"$SCRIPT_NAME"'/d' /jffs/scripts/post-mount
fi
if [ "$STARTUPLINECOUNTEX" -eq 0 ]; then
echo "/jffs/scripts/$SCRIPT_NAME startup"' "$@" & # '"$SCRIPT_NAME" >> /jffs/scripts/post-mount
fi
else
echo "#!/bin/sh" > /jffs/scripts/post-mount
echo "" >> /jffs/scripts/post-mount
echo "/jffs/scripts/$SCRIPT_NAME startup"' "$@" & # '"$SCRIPT_NAME" >> /jffs/scripts/post-mount
chmod 0755 /jffs/scripts/post-mount
fi
;;
delete)
if [ -f /jffs/scripts/post-mount ]; then
STARTUPLINECOUNT=$(grep -c '# '"$SCRIPT_NAME" /jffs/scripts/post-mount)
if [ "$STARTUPLINECOUNT" -gt 0 ]; then
sed -i -e '/# '"$SCRIPT_NAME"'/d' /jffs/scripts/post-mount
fi
fi
;;
esac
}
Auto_Cron(){
case $1 in
create)
STARTUPLINECOUNT=$(cru l | grep -c "${SCRIPT_NAME}_images")
if [ "$STARTUPLINECOUNT" -gt 0 ]; then
cru d "${SCRIPT_NAME}_images"
fi
STARTUPLINECOUNT=$(cru l | grep -c "${SCRIPT_NAME}_stats")
if [ "$STARTUPLINECOUNT" -gt 0 ]; then
cru d "${SCRIPT_NAME}_stats"
fi
STARTUPLINECOUNT=$(cru l | grep -c "${SCRIPT_NAME}_generate")
if [ "$STARTUPLINECOUNT" -eq 0 ]; then
cru a "${SCRIPT_NAME}_generate" "*/5 * * * * /jffs/scripts/$SCRIPT_NAME generate"
fi
STARTUPLINECOUNT=$(cru l | grep -c "${SCRIPT_NAME}_summary")
if [ "$STARTUPLINECOUNT" -eq 0 ]; then
cru a "${SCRIPT_NAME}_summary" "59 23 * * * /jffs/scripts/$SCRIPT_NAME summary"
fi
;;
delete)
STARTUPLINECOUNT=$(cru l | grep -c "${SCRIPT_NAME}_images")
if [ "$STARTUPLINECOUNT" -gt 0 ]; then
cru d "${SCRIPT_NAME}_images"
fi
STARTUPLINECOUNT=$(cru l | grep -c "${SCRIPT_NAME}_stats")
if [ "$STARTUPLINECOUNT" -gt 0 ]; then
cru d "${SCRIPT_NAME}_stats"
fi
STARTUPLINECOUNT=$(cru l | grep -c "${SCRIPT_NAME}_generate")
if [ "$STARTUPLINECOUNT" -gt 0 ]; then
cru d "${SCRIPT_NAME}_generate"
fi
STARTUPLINECOUNT=$(cru l | grep -c "${SCRIPT_NAME}_summary")
if [ "$STARTUPLINECOUNT" -gt 0 ]; then
cru d "${SCRIPT_NAME}_summary"
fi
;;
esac
}
Download_File(){
/usr/sbin/curl -fsL --retry 3 "$1" -o "$2"
}
Get_WebUI_Page(){
MyPage="none"
for i in 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20; do
page="/www/user/user$i.asp"
if [ -f "$page" ] && [ "$(md5sum < "$1")" = "$(md5sum < "$page")" ]; then
MyPage="user$i.asp"
return
elif [ "$MyPage" = "none" ] && [ ! -f "$page" ]; then
MyPage="user$i.asp"
fi
done
}
### function based on @dave14305's FlexQoS webconfigpage function ###
Get_WebUI_URL(){
urlpage=""
urlproto=""
urldomain=""
urlport=""
urlpage="$(sed -nE "/$SCRIPT_NAME/ s/.*url\: \"(user[0-9]+\.asp)\".*/\1/p" /tmp/menuTree.js)"
if [ "$(nvram get http_enable)" -eq 1 ]; then
urlproto="https"
else
urlproto="http"
fi
if [ -n "$(nvram get lan_domain)" ]; then
urldomain="$(nvram get lan_hostname).$(nvram get lan_domain)"
else
urldomain="$(nvram get lan_ipaddr)"
fi
if [ "$(nvram get ${urlproto}_lanport)" -eq 80 ] || [ "$(nvram get ${urlproto}_lanport)" -eq 443 ]; then
urlport=""
else
urlport=":$(nvram get ${urlproto}_lanport)"
fi
if echo "$urlpage" | grep -qE "user[0-9]+\.asp"; then
echo "${urlproto}://${urldomain}${urlport}/${urlpage}" | tr "A-Z" "a-z"
else
echo "WebUI page not found"
fi
}
### ###
### locking mechanism code credit to Martineau (@MartineauUK) ###
Mount_WebUI()
{
Print_Output true "Mounting WebUI tab for $SCRIPT_NAME" "$PASS"
LOCKFILE=/tmp/addonwebui.lock
FD=386
eval exec "$FD>$LOCKFILE"
flock -x "$FD"
Get_WebUI_Page "$SCRIPT_DIR/vnstat-ui.asp"
if [ "$MyPage" = "none" ]; then
Print_Output true "Unable to mount $SCRIPT_NAME WebUI page, exiting" "$CRIT"
flock -u "$FD"
return 1
fi
cp -f "$SCRIPT_DIR/vnstat-ui.asp" "$SCRIPT_WEBPAGE_DIR/$MyPage"
echo "$SCRIPT_NAME" > "$SCRIPT_WEBPAGE_DIR/$(echo $MyPage | cut -f1 -d'.').title"
if [ "$(uname -o)" = "ASUSWRT-Merlin" ]; then
if [ ! -f /tmp/index_style.css ]; then
cp -f /www/index_style.css /tmp/
fi
if ! grep -q '.menu_Addons' /tmp/index_style.css ; then
echo ".menu_Addons { background: url(ext/shared-jy/addons.png); }" >> /tmp/index_style.css
fi
umount /www/index_style.css 2>/dev/null
mount -o bind /tmp/index_style.css /www/index_style.css
if [ ! -f /tmp/menuTree.js ]; then
cp -f /www/require/modules/menuTree.js /tmp/
fi
sed -i "\\~$MyPage~d" /tmp/menuTree.js
if ! grep -q 'menuName: "Addons"' /tmp/menuTree.js ; then
lineinsbefore="$(( $(grep -n "exclude:" /tmp/menuTree.js | cut -f1 -d':') - 1))"
sed -i "$lineinsbefore"'i,\n{\nmenuName: "Addons",\nindex: "menu_Addons",\ntab: [\n{url: "javascript:var helpwindow=window.open('"'"'/ext/shared-jy/redirect.htm'"'"')", tabName: "Help & Support"},\n{url: "NULL", tabName: "__INHERIT__"}\n]\n}' /tmp/menuTree.js
fi
sed -i "/url: \"javascript:var helpwindow=window.open('\/ext\/shared-jy\/redirect.htm'/i {url: \"$MyPage\", tabName: \"$SCRIPT_NAME\"}," /tmp/menuTree.js
umount /www/require/modules/menuTree.js 2>/dev/null
mount -o bind /tmp/menuTree.js /www/require/modules/menuTree.js
fi
flock -u "$FD"
Print_Output true "Mounted $SCRIPT_NAME WebUI page as $MyPage" "$PASS"
}
Shortcut_Script(){
case $1 in
create)
if [ -d /opt/bin ] && [ ! -f "/opt/bin/$SCRIPT_NAME" ] && [ -f "/jffs/scripts/$SCRIPT_NAME" ]; then
ln -s "/jffs/scripts/$SCRIPT_NAME" /opt/bin
chmod 0755 "/opt/bin/$SCRIPT_NAME"
fi
;;
delete)
if [ -f "/opt/bin/$SCRIPT_NAME" ]; then
rm -f "/opt/bin/$SCRIPT_NAME"
fi
;;
esac
}
PressEnter(){
while true; do
printf "Press enter to continue..."
read -r key
case "$key" in
*)
break
;;
esac
done
return 0
}
Check_Requirements(){
CHECKSFAILED="false"
if [ "$(nvram get jffs2_scripts)" -ne 1 ]; then
nvram set jffs2_scripts=1
nvram commit
Print_Output true "Custom JFFS Scripts enabled" "$WARN"
fi
if [ ! -f /opt/bin/opkg ]; then
Print_Output false "Entware not detected!" "$ERR"
CHECKSFAILED="true"
fi
if ! Firmware_Version_Check; then
Print_Output false "Unsupported firmware version detected" "$ERR"
Print_Output false "$SCRIPT_NAME requires Merlin 384.15/384.13_4 or Fork 43E5 (or later)" "$ERR"
CHECKSFAILED="true"
fi
if [ "$CHECKSFAILED" = "false" ]; then
Print_Output false "Installing required packages from Entware" "$PASS"
opkg update
opkg install vnstat2
opkg install vnstati2
opkg install libjpeg-turbo >/dev/null 2>&1
opkg install jq
opkg install sqlite3-cli
opkg install p7zip
opkg install findutils
rm -f /opt/etc/vnstat.conf
return 0
else
return 1
fi
}
### Determine WAN interface using nvram ###
Get_WAN_IFace(){
if [ "$(nvram get wan0_proto)" = "pppoe" ] || [ "$(nvram get wan0_proto)" = "pptp" ] || [ "$(nvram get wan0_proto)" = "l2tp" ]; then
IFACE_WAN="ppp0"
else
IFACE_WAN="$(nvram get wan0_ifname)"
fi
echo "$IFACE_WAN"
}
ScriptStorageLocation(){
case "$1" in
usb)
sed -i 's/^STORAGELOCATION.*$/STORAGELOCATION=usb/' "$SCRIPT_CONF"
mkdir -p "/opt/share/$SCRIPT_NAME.d/"
mv "/jffs/addons/$SCRIPT_NAME.d/csv" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/images" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/config" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/config.bak" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/vnstat.conf" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/vnstat.conf.bak" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/vnstat.conf.default" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/.vnstatusage" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/vnstat.txt" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/.v2upgraded" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
mv "/jffs/addons/$SCRIPT_NAME.d/v1" "/opt/share/$SCRIPT_NAME.d/" 2>/dev/null
/opt/etc/init.d/S33vnstat restart >/dev/null 2>&1
SCRIPT_CONF="/opt/share/$SCRIPT_NAME.d/config"
ScriptStorageLocation load
;;
jffs)
sed -i 's/^STORAGELOCATION.*$/STORAGELOCATION=jffs/' "$SCRIPT_CONF"
mkdir -p "/jffs/addons/$SCRIPT_NAME.d/"
mv "/opt/share/$SCRIPT_NAME.d/csv" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/images" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/config" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/config.bak" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/vnstat.conf" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/vnstat.conf.bak" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/vnstat.conf.default" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/.vnstatusage" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/vnstat.txt" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/.v2upgraded" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
mv "/opt/share/$SCRIPT_NAME.d/v1" "/jffs/addons/$SCRIPT_NAME.d/" 2>/dev/null
/opt/etc/init.d/S33vnstat restart >/dev/null 2>&1
SCRIPT_CONF="/jffs/addons/$SCRIPT_NAME.d/config"
ScriptStorageLocation load
;;
check)
STORAGELOCATION="$(grep "^STORAGELOCATION=" "$SCRIPT_CONF" | cut -f2 -d"=")"
echo "${STORAGELOCATION:=jffs}"
;;
load)
STORAGELOCATION="$(grep "^STORAGELOCATION=" "$SCRIPT_CONF" | cut -f2 -d"=")"
if [ "$STORAGELOCATION" = "usb" ]; then
SCRIPT_STORAGE_DIR="/opt/share/$SCRIPT_NAME.d"
elif [ "$STORAGELOCATION" = "jffs" ]; then
SCRIPT_STORAGE_DIR="/jffs/addons/$SCRIPT_NAME.d"
fi
CSV_OUTPUT_DIR="$SCRIPT_STORAGE_DIR/csv"
IMAGE_OUTPUT_DIR="$SCRIPT_STORAGE_DIR/images"
VNSTAT_COMMAND="vnstat --config $SCRIPT_STORAGE_DIR/vnstat.conf"
VNSTATI_COMMAND="vnstati --config $SCRIPT_STORAGE_DIR/vnstat.conf"
VNSTAT_OUTPUT_FILE="$SCRIPT_STORAGE_DIR/vnstat.txt"
;;
esac
}
OutputTimeMode(){
case "$1" in
unix)
sed -i 's/^OUTPUTTIMEMODE.*$/OUTPUTTIMEMODE=unix/' "$SCRIPT_CONF"
Generate_CSVs
;;
non-unix)
sed -i 's/^OUTPUTTIMEMODE.*$/OUTPUTTIMEMODE=non-unix/' "$SCRIPT_CONF"
Generate_CSVs
;;
check)
OUTPUTTIMEMODE="$(grep "^OUTPUTTIMEMODE=" "$SCRIPT_CONF" | cut -f2 -d"=")"
echo "${OUTPUTTIMEMODE:=unix}"
;;
esac
}
Generate_CSVs(){
renice 15 $$
interface="$(grep "^Interface" "$SCRIPT_STORAGE_DIR/vnstat.conf" | awk '{print $2}' | sed 's/"//g')"
dbdir="$(grep "^DatabaseDir " "$SCRIPT_STORAGE_DIR/vnstat.conf" | awk '{print $2}' | sed 's/"//g')"
TZ=$(cat /etc/TZ)
export TZ
timenow=$(date +"%s")
{
echo ".headers off"
echo ".output /tmp/dn-vnstatiface"
echo "SELECT id FROM [interface] WHERE [name] = '$interface';"
} > /tmp/dn-vnstat.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat.sql
interfaceid="$(cat /tmp/dn-vnstatiface)"
rm -f /tmp/dn-vnstatiface
intervallist="fiveminute hour day"
for interval in $intervallist; do
metriclist="rx tx"
for metric in $metriclist; do
{
echo ".mode csv"
echo ".headers off"
echo ".output $CSV_OUTPUT_DIR/${metric}daily.tmp"
echo "SELECT '$metric' Metric,strftime('%s',[date],'utc') Time,[$metric] Value FROM $interval WHERE [interface] = '$interfaceid' AND strftime('%s',[date],'utc') >= strftime('%s',datetime($timenow,'unixepoch','-1 day'));"
} > /tmp/dn-vnstat.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat.sql
{
echo ".mode csv"
echo ".headers off"
echo ".output $CSV_OUTPUT_DIR/${metric}weekly.tmp"
echo "SELECT '$metric' Metric,strftime('%s',[date],'utc') Time,[$metric] Value FROM $interval WHERE [interface] = '$interfaceid' AND strftime('%s',[date],'utc') >= strftime('%s',datetime($timenow,'unixepoch','-7 day'));"
} > /tmp/dn-vnstat.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat.sql
{
echo ".mode csv"
echo ".headers off"
echo ".output $CSV_OUTPUT_DIR/${metric}monthly.tmp"
echo "SELECT '$metric' Metric,strftime('%s',[date],'utc') Time,[$metric] Value FROM $interval WHERE [interface] = '$interfaceid' AND strftime('%s',[date],'utc') >= strftime('%s',datetime($timenow,'unixepoch','-30 day'));"
} > /tmp/dn-vnstat.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat.sql
rm -f /tmp/dn-vnstat.sql
done
cat "$CSV_OUTPUT_DIR/rxdaily.tmp" "$CSV_OUTPUT_DIR/txdaily.tmp" > "$CSV_OUTPUT_DIR/DataUsage_${interval}_daily.htm" 2> /dev/null
cat "$CSV_OUTPUT_DIR/rxweekly.tmp" "$CSV_OUTPUT_DIR/txweekly.tmp" > "$CSV_OUTPUT_DIR/DataUsage_${interval}_weekly.htm" 2> /dev/null
cat "$CSV_OUTPUT_DIR/rxmonthly.tmp" "$CSV_OUTPUT_DIR/txmonthly.tmp" > "$CSV_OUTPUT_DIR/DataUsage_${interval}_monthly.htm" 2> /dev/null
sed -i 's/rx/Received/g;s/tx/Sent/g;1i Metric,Time,Value' "$CSV_OUTPUT_DIR/DataUsage_${interval}_daily.htm"
sed -i 's/rx/Received/g;s/tx/Sent/g;1i Metric,Time,Value' "$CSV_OUTPUT_DIR/DataUsage_${interval}_weekly.htm"
sed -i 's/rx/Received/g;s/tx/Sent/g;1i Metric,Time,Value' "$CSV_OUTPUT_DIR/DataUsage_${interval}_monthly.htm"
rm -f "$CSV_OUTPUT_DIR/rx"*
rm -f "$CSV_OUTPUT_DIR/tx"*
done
metriclist="rx tx"
for metric in $metriclist; do
{
echo ".mode csv"
echo ".headers off"
echo ".output $CSV_OUTPUT_DIR/week_this_${metric}.tmp"
echo "SELECT '$metric' Metric,strftime('%w', [date]) Time,[$metric] Value FROM day WHERE [interface] = '$interfaceid' AND strftime('%s',[date],'utc') >= strftime('%s',datetime($timenow,'unixepoch','start of day','+1 day','-7 day'));"
} > /tmp/dn-vnstat.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat.sql
{
echo ".mode csv"
echo ".headers off"
echo ".output $CSV_OUTPUT_DIR/week_prev_${metric}.tmp"
echo "SELECT '$metric' Metric,strftime('%w', [date]) Time,[$metric] Value FROM day WHERE [interface] = '$interfaceid' AND strftime('%s',[date],'utc') < strftime('%s',datetime($timenow,'unixepoch','start of day','+1 day','-7 day')) AND strftime('%s',[date],'utc') >= strftime('%s',datetime($timenow,'unixepoch','start of day','+1 day','-14 day'));"
} > /tmp/dn-vnstat.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat.sql
{
echo ".mode csv"
echo ".headers off"
echo ".output $CSV_OUTPUT_DIR/week_summary_this_${metric}.tmp"
echo "SELECT '$metric' Metric,'Current 7 days' Time,IFNULL(SUM([$metric]),'0') Value FROM day WHERE [interface] = '$interfaceid' AND strftime('%s',[date],'utc') >= strftime('%s',datetime($timenow,'unixepoch','start of day','+1 day','-7 day'));"
} > /tmp/dn-vnstat.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat.sql
{
echo ".mode csv"
echo ".headers off"
echo ".output $CSV_OUTPUT_DIR/week_summary_prev_${metric}.tmp"
echo "SELECT '$metric' Metric,'Previous 7 days' Time,IFNULL(SUM([$metric]),'0') Value FROM day WHERE [interface] = '$interfaceid' AND strftime('%s',[date],'utc') < strftime('%s',datetime($timenow,'unixepoch','start of day','+1 day','-7 day')) AND strftime('%s',[date],'utc') >= strftime('%s',datetime($timenow,'unixepoch','start of day','+1 day','-14 day'));"
} > /tmp/dn-vnstat.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat.sql
{
echo ".mode csv"
echo ".headers off"
echo ".output $CSV_OUTPUT_DIR/week_summary_prev2_${metric}.tmp"
echo "SELECT '$metric' Metric,'2 weeks ago' Time,IFNULL(SUM([$metric]),'0') Value FROM day WHERE [interface] = '$interfaceid' AND strftime('%s',[date],'utc') < strftime('%s',datetime($timenow,'unixepoch','start of day','+1 day','-14 day')) AND strftime('%s',[date],'utc') >= strftime('%s',datetime($timenow,'unixepoch','start of day','+1 day','-21 day'));"
} > /tmp/dn-vnstat.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat.sql
done
cat "$CSV_OUTPUT_DIR/week_this_rx.tmp" "$CSV_OUTPUT_DIR/week_this_tx.tmp" > "$CSV_OUTPUT_DIR/WeekThis.htm" 2> /dev/null
cat "$CSV_OUTPUT_DIR/week_prev_rx.tmp" "$CSV_OUTPUT_DIR/week_prev_tx.tmp" > "$CSV_OUTPUT_DIR/WeekPrev.htm" 2> /dev/null
sed -i 's/rx/Received/g;s/tx/Sent/g;1i Metric,Time,Value' "$CSV_OUTPUT_DIR/WeekThis.htm"
sed -i 's/rx/Received/g;s/tx/Sent/g;1i Metric,Time,Value' "$CSV_OUTPUT_DIR/WeekPrev.htm"
cat "$CSV_OUTPUT_DIR/week_summary_this_rx.tmp" "$CSV_OUTPUT_DIR/week_summary_this_tx.tmp" "$CSV_OUTPUT_DIR/week_summary_prev_rx.tmp" "$CSV_OUTPUT_DIR/week_summary_prev_tx.tmp" "$CSV_OUTPUT_DIR/week_summary_prev2_rx.tmp" "$CSV_OUTPUT_DIR/week_summary_prev2_tx.tmp" > "$CSV_OUTPUT_DIR/WeekSummary.htm" 2> /dev/null
sed -i 's/rx/Received/g;s/tx/Sent/g;1i Metric,Time,Value' "$CSV_OUTPUT_DIR/WeekSummary.htm"
rm -f "$CSV_OUTPUT_DIR/week"*
{
echo ".mode csv"
echo ".headers on"
echo ".output $CSV_OUTPUT_DIR/CompleteResults.htm"
echo "SELECT strftime('%s',[date],'utc') Time,[rx],[tx] FROM fiveminute WHERE strftime('%s',[date],'utc') >= strftime('%s',datetime($timenow,'unixepoch','-30 day')) ORDER BY strftime('%s', [date]) DESC;"
} > /tmp/dn-vnstat-complete.sql
"$SQLITE3_PATH" "$dbdir/vnstat.db" < /tmp/dn-vnstat-complete.sql
rm -f /tmp/dn-vnstat-complete.sql
dos2unix "$CSV_OUTPUT_DIR/"*.htm
tmpoutputdir="/tmp/${SCRIPT_NAME}results"
mkdir -p "$tmpoutputdir"
mv "$CSV_OUTPUT_DIR/CompleteResults"*.htm "$tmpoutputdir/."
OUTPUTTIMEMODE="$(OutputTimeMode check)"
if [ "$OUTPUTTIMEMODE" = "unix" ]; then
find "$tmpoutputdir/" -name '*.htm' -exec sh -c 'i="$1"; mv -- "$i" "${i%.htm}.csv"' _ {} \;
elif [ "$OUTPUTTIMEMODE" = "non-unix" ]; then
for i in "$tmpoutputdir/"*".htm"; do