HEADmaster

1 files changed, 45 insertions, 19 deletions

diff --git a/srv/c.c b/srv/c.c
index f402a28..e00b7d7 100644
--- a/srv/c.c
+++ b/srv/c.c
@@ -7,21 +7,24 @@
 #include <errno.h>
 #include <sys/mman.h>
 #include <unistd.h>
+#include <sys/param.h>
+#include <string.h>
 #define S0(x) (x ? x : "")
 struct entry {
 	uint64_t time __attribute__((packed));
 	uint64_t value __attribute__((packed));
 } __attribute__((packed));
-int najdi (struct entry * db, int first, int last, long long int čas) {
+int najdi (struct entry * db, int first, int last, uint64_t čas) {
 	if (first == last)
 		return first;
-	if (abs(first-last) == 1) {
-		if (labs(čas/1000-first*1000) < labs(čas/1000-last*1000))
+	if (last-first == 1) {
+		if (MAX(čas, be64toh(db[first].time))-MIN(čas, be64toh(db[first].time)) < MAX(čas, be64toh(db[last].time))-MIN(čas, be64toh(db[last].time)))
 			return first;
 		else
 			return last;
 	}
-	if ((long long int) be64toh(db[(first+last)/2].time/1000) > čas*1000)
+	uint64_t pol = be64toh(db[(first+last)/2].time);
+	if (pol < čas)
 		return najdi(db, (first+last)/2, last, čas);
 	else
 		return najdi(db, first, (first+last)/2-1, čas);
@@ -30,16 +33,17 @@ int main (int argc, char ** argv) {
 	int r = 0;
 	if (argc < 1+2)
 		error_at_line(1, 0, __FILE__, __LINE__,
-"uporaba: %s db infor/preveri/rast/seštevek [natančnost] [začetek UNIX] [konec UNIX]\n\t"
+"uporaba: %s db infor/preveri/rast/seštevek/uredi [natančnost] [začetek UNIX] [konec UNIX]\n\t"
 "informacije: pove UNIXus prvega in UNIXus zadnjega zapisa\n\t"
-"preveri: pove, če čas ni naraščajoč in kdaj se je števec resetiral\n\t"
+"preveri: najde znane napake v podatkovni zbirki\n\t"
 "rast: TSV s podatki <UNIXus>:<Wh skupno> najgosteje na [natančnost=0] ms\n\t"
 "seštevek: TSV s podatki <UNIXus>:<vatnih ur v [natančnost=10e3] ms>\n\t"
+"uredi: uredi pokvarjeno podatkovno zbirko, daemon ne sme teči med urejanjem\n\t"
 "če je nastavljena okoljska spremenljivka SEK, bodo izhodni časi v UNIX sekundah",
 			S0(argv[0]));
 	int fd = -1;
 	struct entry * db = NULL;
-	if ((fd = open(argv[1], O_RDONLY | O_CLOEXEC)) == -1)
+	if ((fd = open(argv[1], (argv[2][0] == 'u' ? O_RDWR : O_RDONLY) | O_CLOEXEC)) == -1)
 		error_at_line(2, 0, __FILE__, __LINE__, "open");
 	struct stat statbuf;
 	if (fstat(fd, &statbuf) == -1) {
@@ -48,11 +52,24 @@ int main (int argc, char ** argv) {
 		goto r;
 	}
 	unsigned entries = statbuf.st_size / sizeof(struct entry);
-	if (!(db = mmap(NULL, statbuf.st_size, PROT_READ, MAP_SHARED, fd, 0))) {
+	if (!(db = mmap(NULL, statbuf.st_size, (argv[2][0] == 'u' ? PROT_WRITE | PROT_READ : PROT_READ), MAP_SHARED, fd, 0))) {
 		error_at_line(0, errno, __FILE__, __LINE__, "mmap");
 		r = 4;
 		goto r;
 	}
+	if (argv[2][0] == 'u') {
+		unsigned o = 0;
+		for (unsigned r = 0; r < entries; r++) {
+			if (!be64toh(db[r].time)) {
+				o++;
+				continue;
+			}
+			memmove(db+r-o, db+r, sizeof(*db));
+		}
+		ftruncate(fd, sizeof(*db)*(entries-o));
+		printf("odstranjenih %u zapisov\n", o);
+		goto r;
+	}
 	if (argv[2][0] == 'i') {
 		uint64_t začetek = be64toh(db[0].time);
 		uint64_t konec = be64toh(db[entries-1].time);
@@ -68,13 +85,18 @@ int main (int argc, char ** argv) {
 	if (argv[2][0] == 'p') {
 		uint64_t prev_time = 0;
 		uint64_t prev_value = 0;
+		fprintf(stderr, "opis možnih napak:\n\tčas se je zavrtel nazaj ponazarja napako na strežniku, saj ni imel nastavljenega časa. podatkovne zbirke s to napako so trenutno neuporabne.\n\tštevec se je resetiral pomeni, da je merilnik dobil ukaz reset oziroma je bil vzpostavljen nov merilnik, ki meri od začetka. v prihodnosti bo ta bralnik sproti nadomestil vrednosti ob branju pri tej napaki, trenutno pa tega ne stori.\n\tničelni zapis v podatkovni zbirki se zgodi, ko med tekom daemona nenadno zmanjka elektrike - takrat obstaja možnost, da se bodo po ponovem zagonu v datoteko vpisali ničelni zapisi. podatkovne zbirke s to napako je treba urediti z ukazom uredi, drugače ni uporabna v bralniku.\n");
 		for (unsigned i = 0; i < entries; i++) {
-			if (be64toh(db[0].time) < prev_time)
-				printf("čas se je zavrtel nazaj: i=%u a=%" PRIu64 " b=%" PRIu64 "\n", i, prev_time, be64toh(db[0].time));
-			if (be64toh(db[0].value) < prev_value)
-				printf("števec se je resetiral: i=%u a=%" PRIu64 " b=%" PRIu64 "\n", i, prev_value, be64toh(db[0].value));
-			prev_time = be64toh(db[0].time);
-			prev_value = be64toh(db[0].value);
+			if (!be64toh(db[i].time)) {
+				printf("ničelni zapis v podatkovni zbriki (med pisanjem se je zgodil izpad elektrike): i=%u prev_time=%" PRIu64 "\n", i, prev_time);
+				continue;
+			}
+			if (be64toh(db[i].time) < prev_time)
+				printf("čas se je zavrtel nazaj: i=%u a=%" PRIu64 " b=%" PRIu64 "\n", i, prev_time, be64toh(db[i].time));
+			if (be64toh(db[i].value) < prev_value)
+				printf("števec se je resetiral: i=%u a=%" PRIu64 " b=%" PRIu64 "\n", i, prev_value, be64toh(db[i].value));
+			prev_time = be64toh(db[i].time);
+			prev_value = be64toh(db[i].value);
 		}
 		goto r;
 	}
@@ -85,16 +107,18 @@ int main (int argc, char ** argv) {
 		natančnost = atoi(argv[3]);
 	unsigned začetek = 0;
 	if (argc >= 1+4)
-		začetek = najdi(db, 0, entries-1, atoi(argv[4]));
-	unsigned konec = entries-1;
+		začetek = najdi(db, 0, entries-1, atoi(argv[4])*1e6);
+	uint64_t konec = -1;
 	if (argc >= 1+5)
-		konec = najdi(db, 0, entries-1, atoi(argv[5]));
+		konec = atoi(argv[5])*(uint64_t)1e6;
 	if (argv[2][0] == 's') {
 		uint64_t us = 0;
 		uint64_t Wh = 0;
 		unsigned long long prev_time = be64toh(db[začetek].time);
 		unsigned long long prev_value = be64toh(db[začetek].value);
-		for (unsigned i = začetek; i <= konec; i++) {
+		for (unsigned i = začetek; i < entries; i++) {
+			if (be64toh(db[i].time) > konec)
+				break;
 			if (us + (be64toh(db[i].time)-prev_time) >= natančnost*1000) {
 				uint64_t čas_začetek = be64toh(db[i].time)-us;
 				uint64_t čas_konec = be64toh(db[i].time);
@@ -113,7 +137,9 @@ int main (int argc, char ** argv) {
 		}
 	} else {
 		unsigned long long prev_print = 0;
-		for (unsigned i = začetek; i <= konec; i++) {
+		for (unsigned i = začetek; i < entries; i++) {
+			if (be64toh(db[i].time) > konec)
+				break;
 			if (be64toh(db[i].time)-prev_print > natančnost*1000) {
 				uint64_t čas = be64toh(db[i].time);
 				if (getenv("SEK"))