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:
源文雨 2022-11-23 14:45:39 +08:00
parent 24ac4f3905
commit 9719875b87
2 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
cmoe.c
View File

@ -8,7 +8,7 @@
#include "cmoe.h" #include "cmoe.h"
static uint32_t* items_len; static uint32_t* items_len;
static COUNTER counter; static counter_t counter;
static char* DATFILE = "dat.sp"; static char* DATFILE = "dat.sp";
static char* TOKEN = "fumiama"; static char* TOKEN = "fumiama";
@ -28,7 +28,7 @@ static void _headers(uint32_t content_len, const char* h, size_t hlen) {
writev(1, &iov, 3); writev(1, &iov, 3);
} }
static void http_error(ERRCODE code, char* msg) { static void http_error(errcode_enum_t 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);
@ -48,7 +48,7 @@ static char* get_arg(const char* query) {
} }
static int del_user(FILE* fp, SIMPLE_PB* spb) { static int del_user(FILE* fp, SIMPLE_PB* spb) {
COUNTER *d = (COUNTER *)spb->target; counter_t *d = (counter_t *)spb->target;
uint32_t next = ftell(fp); uint32_t next = ftell(fp);
uint32_t this = next - spb->real_len; uint32_t this = next - spb->real_len;
fseek(fp, 0, SEEK_END); fseek(fp, 0, SEEK_END);
@ -77,7 +77,7 @@ static int add_user(char* name, uint32_t count, FILE* fp) {
counter.count = count; counter.count = count;
strncpy(counter.name, name, COUNTER_NAME_LEN-1); strncpy(counter.name, name, COUNTER_NAME_LEN-1);
fseek(fp, 0, SEEK_END); fseek(fp, 0, SEEK_END);
return !set_pb(fp, items_len, sizeof(COUNTER), &counter); return !set_pb(fp, items_len, sizeof(counter_t), &counter);
} }
static uint32_t get_content_len(int isbig, uint16_t* len_type, char* cntstr) { static uint32_t get_content_len(int isbig, uint16_t* len_type, char* cntstr) {
@ -105,10 +105,10 @@ static void return_count(char* name, char* theme) {
} }
flock(fileno(fp), LOCK_EX); flock(fileno(fp), LOCK_EX);
int ch, exist = 0, user_exist = 0; int ch, exist = 0, user_exist = 0;
char buf[sizeof(SIMPLE_PB)+sizeof(COUNTER)]; char buf[sizeof(SIMPLE_PB)+sizeof(counter_t)];
while(has_next(fp, ch)) { while(has_next(fp, ch)) {
SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf); SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf);
COUNTER *d = (COUNTER *)spb->target; counter_t *d = (counter_t *)spb->target;
if (strcmp(name, d->name)) continue; if (strcmp(name, d->name)) continue;
if(del_user(fp, spb)) { if(del_user(fp, spb)) {
http_error(HTTP500, "Unable to Delete Old Data."); http_error(HTTP500, "Unable to Delete Old Data.");
@ -171,11 +171,11 @@ static int name_exist(char* name) {
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
int ch, exist = 0; int ch, exist = 0;
char buf[sizeof(SIMPLE_PB)+sizeof(COUNTER)]; char buf[sizeof(SIMPLE_PB)+sizeof(counter_t)];
flock(fileno(fp), LOCK_EX); flock(fileno(fp), LOCK_EX);
while(has_next(fp, ch)) { while(has_next(fp, ch)) {
SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf); SIMPLE_PB *spb = read_pb_into(fp, (SIMPLE_PB*)buf);
COUNTER *d = (COUNTER *)spb->target; counter_t *d = (counter_t *)spb->target;
if (!strcmp(name, d->name)) { if (!strcmp(name, d->name)) {
fclose(fp); fclose(fp);
return 1; return 1;
@ -197,7 +197,7 @@ int main(int argc, char **argv) {
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), 2, &counter.name, &counter.count); items_len = align_struct(sizeof(counter_t), 2, &counter.name, &counter.count);
if(!items_len) { if(!items_len) {
http_error(HTTP500, "Align Struct Error."); http_error(HTTP500, "Align Struct Error.");
return 2; return 2;

8
cmoe.h
View File

@ -4,11 +4,11 @@
#include <stdint.h> #include <stdint.h>
#define COUNTER_NAME_LEN 64 #define COUNTER_NAME_LEN 64
struct COUNTER { struct counter_t {
char name[COUNTER_NAME_LEN]; char name[COUNTER_NAME_LEN];
uint32_t count; uint32_t count;
}; };
typedef struct COUNTER COUNTER; typedef struct counter_t counter_t;
#define SERVER_STRING "Server: CMoe Counter by Fumiama/1.0\r\n" #define SERVER_STRING "Server: CMoe Counter by Fumiama/1.0\r\n"
@ -18,8 +18,8 @@ typedef struct COUNTER COUNTER;
#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[] = {H400, H404, H500}; const char* types[] = {H400, H404, H500};
const uint8_t typel[] = {59, 111, 69}; const uint8_t typel[] = {59, 111, 69};
enum ERRCODE {HTTP400, HTTP404, HTTP500}; enum errcode_enum_t {HTTP400, HTTP404, HTTP500};
typedef enum ERRCODE ERRCODE; typedef enum errcode_enum_t errcode_enum_t;
#define CONTENT_TYPE(x) "Content-Type: " #x "\r\n" #define CONTENT_TYPE(x) "Content-Type: " #x "\r\n"
#define CONTENT_LEN "Content-Length: %d\r\n" #define CONTENT_LEN "Content-Length: %d\r\n"