mirror of
https://github.com/fumiama/CMoe-Counter.git
synced 2024-11-24 06:10:11 +01:00
251 lines
9.2 KiB
C
251 lines
9.2 KiB
C
|
#include <simple_protobuf.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include "cmoe.h"
|
||
|
|
|
||
|
|
static uint32_t* items_len;
|
||
|
|
static counter_t counter;
|
||
|
|
|
||
|
|
static FILE* fp;
|
||
|
|
|
||
|
|
static char buf[sizeof(SIMPLE_PB)+sizeof(counter_t)];
|
||
|
|
|
||
|
|
static int del_user(FILE* fp, SIMPLE_PB* spb) {
|
||
|
|
counter_t *d = (counter_t *)spb->target;
|
||
|
|
uint32_t next = ftell(fp);
|
||
|
|
uint32_t this = next - spb->real_len;
|
||
|
|
fseek(fp, 0, SEEK_END);
|
||
|
|
uint32_t end = ftell(fp);
|
||
|
|
if(next == end) return ftruncate(fileno(fp), end - spb->real_len);
|
||
|
|
uint32_t cap = end - next;
|
||
|
|
char *data = malloc(cap);
|
||
|
|
if(data) {
|
||
|
|
fseek(fp, next, SEEK_SET);
|
||
|
|
if(fread(data, cap, 1, fp) == 1) {
|
||
|
|
if(!ftruncate(fileno(fp), end - spb->real_len)){
|
||
|
|
fseek(fp, this, SEEK_SET);
|
||
|
|
if(fwrite(data, cap, 1, fp) == 1) {
|
||
|
|
free(data);
|
||
|
|
fflush(fp);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
free(data);
|
||
|
|
}
|
||
|
|
return 2;
|
||
|
|
}
|
||
|
|
|
||
|
|
#define has_next(fp, ch) ((ch=getc(fp)),(feof(fp)?0:(ungetc(ch,fp),1)))
|
||
|
|
|
||
|
|
int main(int argc, char** argv) {
|
||
|
|
if (argc != 2) {
|
||
|
|
puts("Usage: editor dat.sp");
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
fp = fopen(argv[1], "rb+");
|
||
|
|
if (!fp) {
|
||
|
|
perror("fopen");
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
items_len = align_struct(sizeof(counter_t), 2, &counter.name, &counter.count);
|
||
|
|
if(!items_len) {
|
||
|
|
perror("align_struct");
|
||
|
|
return 2;
|
||
|
|
}
|
||
|
|
int go;
|
||
|
|
lop: do {
|
||
|
|
go = 1;
|
||
|
|
puts("\ncommands:");
|
||
|
|
puts("\ta string uint: add user with count.");
|
||
|
|
puts("\td string: delete user.");
|
||
|
|
puts("\tD string: delete users match the keyword (empty is not allowed).");
|
||
|
|
puts("\tg string: get user count.");
|
||
|
|
puts("\tm uint: print users < count.");
|
||
|
|
puts("\tM uint: delete users < count.");
|
||
|
|
puts("\tp string: print users match the keyword (empty for all).");
|
||
|
|
puts("\tq: quit.");
|
||
|
|
putchar('>');putchar(' ');
|
||
|
|
char c = getchar();
|
||
|
|
switch (c) {
|
||
|
|
case 'a':
|
||
|
|
while ((c = getchar()) == ' '); // skip space
|
||
|
|
memset(counter.name, 0, sizeof(counter.name));
|
||
|
|
counter.name[0] = c;
|
||
|
|
while ((c = getchar()) != ' ' && go < sizeof(counter.name)-1) counter.name[go++] = c;
|
||
|
|
while ((c = getchar()) == ' '); // skip space
|
||
|
|
ungetc(c, stdin);
|
||
|
|
scanf("%u", &counter.count);
|
||
|
|
while (getchar() != '\n'); // skip to endl
|
||
|
|
printf("set user '%s' to %u\n", counter.name, counter.count);
|
||
|
|
if (fseek(fp, 0, SEEK_END)) {
|
||
|
|
perror("fseek");
|
||
|
|
return 3;
|
||
|
|
}
|
||
|
|
if (set_pb(fp, items_len, sizeof(counter_t), &counter) != 2) {
|
||
|
|
puts("set pb error.");
|
||
|
|
return 4;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case 'd':
|
||
|
|
while ((c = getchar()) == ' '); // skip space
|
||
|
|
memset(counter.name, 0, sizeof(counter.name));
|
||
|
|
counter.name[0] = c;
|
||
|
|
while ((c = getchar()) != '\n' && go < sizeof(counter.name)-1) counter.name[go++] = c;
|
||
|
|
if (fseek(fp, 0, SEEK_SET)) {
|
||
|
|
perror("fseek");
|
||
|
|
return 5;
|
||
|
|
}
|
||
|
|
while(has_next(fp, c)) {
|
||
|
|
SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf);
|
||
|
|
counter_t *d = (counter_t *)spb->target;
|
||
|
|
if (strcmp(counter.name, d->name)) continue;
|
||
|
|
if(del_user(fp, spb)) {
|
||
|
|
perror("del_user");
|
||
|
|
return 6;
|
||
|
|
}
|
||
|
|
printf("del user '%s'.\n", d->name);
|
||
|
|
goto lop;
|
||
|
|
}
|
||
|
|
printf("no such user '%s'.\n", counter.name);
|
||
|
|
break;
|
||
|
|
case 'D':
|
||
|
|
if (getchar() == '\n') {
|
||
|
|
puts("empty D is not allowed.");
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
while ((c = getchar()) == ' '); // skip space
|
||
|
|
memset(counter.name, 0, sizeof(counter.name));
|
||
|
|
counter.name[0] = c;
|
||
|
|
while ((c = getchar()) != '\n' && go < sizeof(counter.name)-1) counter.name[go++] = c;
|
||
|
|
go = 1;
|
||
|
|
scanlopD:
|
||
|
|
if (fseek(fp, 0, SEEK_SET)) {
|
||
|
|
perror("fseek");
|
||
|
|
return 7;
|
||
|
|
}
|
||
|
|
while(has_next(fp, c)) {
|
||
|
|
SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf);
|
||
|
|
counter_t *d = (counter_t *)spb->target;
|
||
|
|
if (strstr(d->name, counter.name)) {
|
||
|
|
if(del_user(fp, spb)) {
|
||
|
|
perror("del_user");
|
||
|
|
return 8;
|
||
|
|
}
|
||
|
|
printf("del user '%s'.\n", d->name);
|
||
|
|
go++;
|
||
|
|
goto scanlopD;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (go == 1) puts("no result.");
|
||
|
|
break;
|
||
|
|
case 'g':
|
||
|
|
while ((c = getchar()) == ' '); // skip space
|
||
|
|
memset(counter.name, 0, sizeof(counter.name));
|
||
|
|
counter.name[0] = c;
|
||
|
|
while ((c = getchar()) != '\n' && go < sizeof(counter.name)-1) counter.name[go++] = c;
|
||
|
|
if (fseek(fp, 0, SEEK_SET)) {
|
||
|
|
perror("fseek");
|
||
|
|
return 5;
|
||
|
|
}
|
||
|
|
while(has_next(fp, c)) {
|
||
|
|
SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf);
|
||
|
|
counter_t *d = (counter_t *)spb->target;
|
||
|
|
if (strcmp(counter.name, d->name)) continue;
|
||
|
|
printf("user '%s' = %u.\n", d->name, d->count);
|
||
|
|
goto lop;
|
||
|
|
}
|
||
|
|
printf("no such user '%s'.\n", counter.name);
|
||
|
|
break;
|
||
|
|
case 'm':
|
||
|
|
while ((c = getchar()) == ' '); // skip space
|
||
|
|
ungetc(c, stdin);
|
||
|
|
scanf("%u", &counter.count);
|
||
|
|
while (getchar() != '\n'); // skip to endl
|
||
|
|
go = 1;
|
||
|
|
if (fseek(fp, 0, SEEK_SET)) {
|
||
|
|
perror("fseek");
|
||
|
|
return 7;
|
||
|
|
}
|
||
|
|
while(has_next(fp, c)) {
|
||
|
|
SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf);
|
||
|
|
counter_t *d = (counter_t *)spb->target;
|
||
|
|
if (d->count < counter.count) {
|
||
|
|
printf("'%s' = %u\n", d->name, d->count);
|
||
|
|
go++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (go == 1) puts("no result.");
|
||
|
|
break;
|
||
|
|
case 'M':
|
||
|
|
if (getchar() == '\n') {
|
||
|
|
puts("empty M is not allowed.");
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
while ((c = getchar()) == ' '); // skip space
|
||
|
|
ungetc(c, stdin);
|
||
|
|
scanf("%u", &counter.count);
|
||
|
|
while (getchar() != '\n'); // skip to endl
|
||
|
|
go = 1;
|
||
|
|
scanlopM:
|
||
|
|
if (fseek(fp, 0, SEEK_SET)) {
|
||
|
|
perror("fseek");
|
||
|
|
return 7;
|
||
|
|
}
|
||
|
|
while(has_next(fp, c)) {
|
||
|
|
SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf);
|
||
|
|
counter_t *d = (counter_t *)spb->target;
|
||
|
|
if (d->count < counter.count) {
|
||
|
|
if(del_user(fp, spb)) {
|
||
|
|
perror("del_user");
|
||
|
|
return 8;
|
||
|
|
}
|
||
|
|
printf("del user '%s'.\n", d->name);
|
||
|
|
go++;
|
||
|
|
goto scanlopM;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (go == 1) puts("no result.");
|
||
|
|
break;
|
||
|
|
case 'p':
|
||
|
|
if ((c = getchar()) != '\n') {
|
||
|
|
while ((c = getchar()) == ' '); // skip space
|
||
|
|
memset(counter.name, 0, sizeof(counter.name));
|
||
|
|
counter.name[0] = c;
|
||
|
|
while ((c = getchar()) != '\n' && go < sizeof(counter.name)-1) counter.name[go++] = c;
|
||
|
|
} else counter.name[0] = 0;
|
||
|
|
go = 1;
|
||
|
|
if (fseek(fp, 0, SEEK_SET)) {
|
||
|
|
perror("fseek");
|
||
|
|
return 7;
|
||
|
|
}
|
||
|
|
while(has_next(fp, c)) {
|
||
|
|
SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf);
|
||
|
|
counter_t *d = (counter_t *)spb->target;
|
||
|
|
if (!counter.name[0] || strstr(d->name, counter.name)) {
|
||
|
|
printf("'%s' = %u\n", d->name, d->count);
|
||
|
|
go++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (go == 1) puts("no result.");
|
||
|
|
break;
|
||
|
|
case 'q':
|
||
|
|
go = 0;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
printf("invalid operator '%c'.\n", c);
|
||
|
|
while (getchar() != '\n'); // skip to endl
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
} while (go);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void __attribute__((destructor)) defer_close_fp() {
|
||
|
|
if(fp) {
|
||
|
|
if (fclose(fp)) perror("fclose");
|
||
|
|
else puts("file closed.");
|
||
|
|
}
|
||
|
|
}
|