Cool_Tools/CMoe-Counter
1
0
Fork 0
mirror of https://github.com/fumiama/CMoe-Counter.git synced 2025-02-08 23:09:33 +01:00
Code Releases Activity

优化代码结构

Browse Source
This commit is contained in:
源文雨 2022-11-23 15:22:53 +08:00
parent 9719875b87
commit d909b0b3b5
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
cmoe.c
View File

@ -10,8 +10,8 @@
static uint32_t* items_len; static uint32_t* items_len;
static counter_t counter; static counter_t counter;
static char* DATFILE = "dat.sp"; static char* datfile = "dat.sp";
static char* TOKEN = "fumiama"; static char* token = "fumiama";
#define ADD_HEADER_PARAM(buf, offset, h, p) sprintf(buf + offset, (h), (p)) #define ADD_HEADER_PARAM(buf, offset, h, p) sprintf(buf + offset, (h), (p))
#define HEADER(content_type) HTTP200 SERVER_STRING CACHE_CTRL CONTENT_TYPE(content_type) #define HEADER(content_type) HTTP200 SERVER_STRING CACHE_CTRL CONTENT_TYPE(content_type)
@ -97,8 +97,8 @@ static uint32_t get_content_len(int isbig, uint16_t* len_type, char* cntstr) {
len_type = l;\ len_type = l;\
} }
static void return_count(char* name, char* theme) { static void return_count(char* name, char* theme) {
FILE* fp = fopen(DATFILE, "rb+"); FILE* fp = fopen(datfile, "rb+");
if(!fp) fp = fopen(DATFILE, "wb+"); if(!fp) fp = fopen(datfile, "wb+");
if(!fp) { if(!fp) {
http_error(HTTP500, "Open File Error."); http_error(HTTP500, "Open File Error.");
return; return;
@ -164,8 +164,8 @@ static void return_count(char* name, char* theme) {
} }
static int name_exist(char* name) { static int name_exist(char* name) {
FILE* fp = fopen(DATFILE, "rb+"); FILE* fp = fopen(datfile, "rb+");
if(!fp) fp = fopen(DATFILE, "wb+"); if(!fp) fp = fopen(datfile, "wb+");
if(fp) { if(fp) {
http_error(HTTP500, "Open File Error."); http_error(HTTP500, "Open File Error.");
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
@ -193,9 +193,9 @@ int main(int argc, char **argv) {
return 1; return 1;
} }
char* str = getenv("DATFILE"); char* str = getenv("DATFILE");
if(str != NULL) DATFILE = str; if(str != NULL) datfile = str;
str = getenv("TOKEN"); str = getenv("TOKEN");
if(str != NULL) TOKEN = str; if(str != NULL) token = str;
char* name = strstr(QS, "name="); char* name = strstr(QS, "name=");
items_len = align_struct(sizeof(counter_t), 2, &counter.name, &counter.count); items_len = align_struct(sizeof(counter_t), 2, &counter.name, &counter.count);
if(!items_len) { if(!items_len) {
@ -225,7 +225,7 @@ int main(int argc, char **argv) {
http_error(HTTP400, "Null Register Token."); http_error(HTTP400, "Null Register Token.");
return 5; return 5;
} }
if(strcmp(reg, TOKEN)) { if(strcmp(reg, token)) {
http_error(HTTP400, "Token Error."); http_error(HTTP400, "Token Error.");
return 6; return 6;
} }
@ -233,7 +233,7 @@ int main(int argc, char **argv) {
http_error(HTTP400, "Name Exist."); http_error(HTTP400, "Name Exist.");
return 7; return 7;
} }
FILE* fp = fopen(DATFILE, "ab+"); FILE* fp = fopen(datfile, "ab+");
if (!fp) { if (!fp) {
http_error(HTTP500, "Open File Error."); http_error(HTTP500, "Open File Error.");
return 8; return 8;