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

优化unlock

Browse Source
This commit is contained in:
源文雨 2023-05-30 21:27:51 +08:00
parent c823a70d43
commit 09c44e87fb
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
cmoe.c
View File

@ -169,6 +169,12 @@ static int name_exist(FILE* fp, char* name) {
#define create_or_open(fp, filename) ((fp = fopen(filename, "rb+"))?(fp):(fp = fopen(filename, "wb+")))
#define flease(fp) { \
fflush(fp); \
flock(fileno(fp), LOCK_UN); \
fclose(fp); fp = NULL; \
}
#define QS (argv[2])
// Usage: cmoe method query_string
int main(int argc, char **argv) {
@ -210,8 +216,7 @@ int main(int argc, char **argv) {
return -3;
}
return_count(fp, name, theme);
flock(fileno(fp), LOCK_UN);
fclose(fp); fp = NULL;
flease(fp);
return 0;
}
reg = get_arg(reg + 4);
@ -232,13 +237,11 @@ int main(int argc, char **argv) {
return -5;
}
if(name_exist(fp, name)) {
flock(fileno(fp), LOCK_UN);
fclose(fp); fp = NULL;
flease(fp);
http_error(HTTP400, "Name Exist.");
return 7;
}
flock(fileno(fp), LOCK_UN);
fclose(fp); fp = NULL;
flease(fp);
fp = fopen(datfile, "ab+");
if (!fp) {
http_error(HTTP500, "Open File Error.");
@ -249,8 +252,7 @@ int main(int argc, char **argv) {
return -7;
}
add_user(name, 0, fp);
flock(fileno(fp), LOCK_UN);
fclose(fp); fp = NULL;
flease(fp);
char* msg = "<P>Success.\r\n";
if(headers(strlen(msg), "text/html")) {
write(1, "\0\0\0\0", 4);
@ -260,8 +262,5 @@ int main(int argc, char **argv) {
}
static void __attribute__((destructor)) defer_close_fp() {
if(fp) {
flock(fileno(fp), LOCK_UN);
fclose(fp);
}
if(fp) flease(fp);
}