Cool_Tools/CMoe-Counter
1
0
Fork 0
mirror of https://github.com/fumiama/CMoe-Counter.git synced 2024-11-27 23:20:49 +01:00
Code Releases Activity

优化流程

Browse Source
This commit is contained in:
源文雨 2021-09-12 00:19:08 +08:00
parent 61b453f06c
commit 7a9501065b
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cmoe.c
View File

@ -22,8 +22,7 @@ static void headers(uint32_t content_len, const char* content_type) {
char buf[1024]; char buf[1024];
uint32_t offset = 0; uint32_t offset = 0;
ADD_HERDER(H200 SERVER_STRING); ADD_HERDER(HTTP200 SERVER_STRING CACHE_CTRL);
ADD_HERDER(CACHE_CTRL);
ADD_HERDER_PARAM(CONTENT_TYPE, content_type); ADD_HERDER_PARAM(CONTENT_TYPE, content_type);
ADD_HERDER_PARAM(CONTENT_LEN "\r\n", content_len); ADD_HERDER_PARAM(CONTENT_LEN "\r\n", content_len);
content_len += offset; content_len += offset;
@ -31,7 +30,7 @@ static void headers(uint32_t content_len, const char* content_type) {
write(1, buf, offset); write(1, buf, offset);
} }
static void http_error(RESPCODE code, char* msg) { static void http_error(ERRCODE code, char* msg) {
uint32_t len = strlen(msg) + typel[code]; uint32_t len = strlen(msg) + typel[code];
char* str = malloc(len); char* str = malloc(len);
sprintf(str, types[code], msg); sprintf(str, types[code], msg);
@ -156,11 +155,12 @@ static void return_count(char* name, char* theme) {
} }
free(spb); free(spb);
user_exist = 1; user_exist = 1;
break; return;
} }
} else free(spb); } else free(spb);
} }
if(!user_exist) http_error(HTTP404, "No Such User."); if(!user_exist) http_error(HTTP404, "No Such User.");
fclose(fp);
} else http_error(HTTP500, "Open File Error."); } else http_error(HTTP500, "Open File Error.");
} }

10
cmoe.h
View File

@ -15,14 +15,14 @@ typedef struct COUNTER COUNTER;
#define SERVER_STRING "Server: CMoe Counter by Fumiama/1.0\r\n" #define SERVER_STRING "Server: CMoe Counter by Fumiama/1.0\r\n"
#define H200 "HTTP/1.0 200 OK\r\n" #define HTTP200 "HTTP/1.0 200 OK\r\n"
#define H400 "HTTP/1.0 400 BAD REQUEST\r\nContent-Type: text/html\r\n\r\n<P>%s\r\n" #define H400 "HTTP/1.0 400 BAD REQUEST\r\nContent-Type: text/html\r\n\r\n<P>%s\r\n"
#define H404 "HTTP/1.0 404 NOT FOUND\r\nContent-Type: text/html\r\n\r\n<HTML><TITLE>Not Found</TITLE>\r\n<BODY><P>%s\r\n</BODY></HTML>\r\n" #define H404 "HTTP/1.0 404 NOT FOUND\r\nContent-Type: text/html\r\n\r\n<HTML><TITLE>Not Found</TITLE>\r\n<BODY><P>%s\r\n</BODY></HTML>\r\n"
#define H500 "HTTP/1.0 500 Internal Server Error\r\nContent-Type: text/html\r\n\r\n<P>%s\r\n" #define H500 "HTTP/1.0 500 Internal Server Error\r\nContent-Type: text/html\r\n\r\n<P>%s\r\n"
const char* types[] = {H200, H400, H404, H500}; const char* types[] = {H400, H404, H500};
const uint8_t typel[] = {18, 59, 111, 69}; const uint8_t typel[] = {59, 111, 69};
enum RESPCODE {HTTP200, HTTP400, HTTP404, HTTP500}; enum ERRCODE {HTTP400, HTTP404, HTTP500};
typedef enum RESPCODE RESPCODE; typedef enum ERRCODE ERRCODE;
#define CONTENT_TYPE "Content-Type: %s\r\n" #define CONTENT_TYPE "Content-Type: %s\r\n"
#define CONTENT_LEN "Content-Length: %d\r\n" #define CONTENT_LEN "Content-Length: %d\r\n"