增加中国IP缓存

2025-01-21 17:06:12 +08:00
parent 5028d24a32
commit 685b29a045
6 changed files with 241 additions and 100 deletions
--- a/cache.c
+++ b/cache.c
@@ -0,0 +1,150 @@
+#include "cache.h"
+
+char cn_ip[MAXIPSET_][MAXIPLEN] = { 0 };
+
+// 添加一个 IP 到集合（如果已存在则不添加）
+int add_cn_ip(char cn_ip[MAXIPSET_][MAXIPLEN], char *ip)
+{
+    if (ip == NULL || strlen(ip) >= MAXIPLEN) {
+        return -1;              // 错误：无效的 IP 地址或过长
+    }
+    // 检查是否已存在
+    for (int i = 0; i < MAXIPSET_; i++) {
+        if (cn_ip[i][0] != '\0' && strcmp(cn_ip[i], ip) == 0) {
+            return 1;           // IP 已存在，返回特殊代码
+        }
+    }
+    // 查找空位并添加
+    for (int i = 0; i < MAXIPSET_; i++) {
+        if (cn_ip[i][0] == '\0') { // 检查是否为空
+            strcpy(cn_ip[i], ip);
+            return 0;           // 成功添加
+        }
+    }
+    return -1;                  // 错误：集合已满
+}
+
+// 判断给定的IP地址是否在集合中
+int is_ip_in_set(char cn_ip[MAXIPSET_][MAXIPLEN], const char *ip)
+{
+    if (ip == NULL) {
+        return 0;               // 错误：无效的IP地址指针
+    }
+    for (int i = 0; i < MAXIPSET_; i++) {
+        if (cn_ip[i][0] != '\0' && strcmp(cn_ip[i], ip) == 0) {
+            return 1;           // 找到匹配的IP地址
+        }
+    }
+    return 0;                   // 没有找到匹配的IP地址
+}
+
+// 计算集合中非空IP的数量
+int cn_ip_len(char cn_ip[MAXIPSET_][MAXIPLEN])
+{
+    int count = 0;              // 用于计数非空IP
+    for (int i = 0; i < MAXIPSET_; i++) {
+        if (cn_ip[i][0] != '\0') {
+            count++;            // 非空IP计数
+        }
+    }
+    return count;
+}
+
+// 安全的清理IP地址集合，将所有位置置为空字符串
+void clear_ip_set(char cn_ip[MAXIPSET_][MAXIPLEN])
+{
+    if (cn_ip == NULL) {
+        return;                 // 如果指针无效，则直接返回
+    }
+    for (int i = 0; i < MAXIPSET_; i++) {
+        memset(cn_ip[i], '\0', MAXIPLEN);
+    }
+
+    return;
+}
+
+// 保存字符串到文件
+int append_string_to_file(const char *filepath, const char *str)
+{
+    // 检查输入是否为空
+    if (filepath == NULL || str == NULL) {
+        fprintf(stderr, "Invalid argument(s).\n");
+        return -1;
+    }
+    // 打开文件以追加模式
+    FILE *file = fopen(filepath, "a");
+    if (file == NULL) {
+        perror("Error opening file");
+        return -1;
+    }
+    // 追加字符串到文件
+    if (fprintf(file, "%s\n", str) < 0) {
+        perror("Error writing to file");
+        fclose(file);
+        return -1;
+    }
+    // 刷新缓冲区确保数据写入磁盘
+    if (fflush(file) != 0) {
+        perror("Error flushing file");
+        fclose(file);
+        return -1;
+    }
+    // 关闭文件
+    if (fclose(file) != 0) {
+        perror("Error closing file");
+        return -1;
+    }
+    // 成功返回
+    return 0;
+}
+
+int read_file_to_array(const char *filepath, char cn_ip[MAXIPSET_][MAXIPLEN], int *line_count)
+{
+    // 参数校验
+    if (filepath == NULL || cn_ip == NULL || line_count == NULL) {
+        fprintf(stderr, "Invalid argument(s).\n");
+        return -1;
+    }
+
+    FILE *file = fopen(filepath, "r");
+    if (file == NULL) {
+        fprintf(stderr, "Error opening file '%s': %s\n", filepath, strerror(errno));
+        return -2;
+    }
+
+    int count = 0;
+    while (fgets(cn_ip[count], MAXIPLEN, file)) {
+        // 去掉行尾的换行符
+        size_t len = strlen(cn_ip[count]);
+        if (len > 0 && cn_ip[count][len - 1] == '\n') {
+            cn_ip[count][len - 1] = '\0';
+        }
+        count++;
+        if (count >= MAXIPSET_) {
+            fprintf(stderr, "Exceeded maximum number of lines (%d).\n", MAXIPSET_);
+            fclose(file);
+            return -3;
+        }
+    }
+
+    if (ferror(file)) {
+        fprintf(stderr, "Error reading file '%s'.\n", filepath);
+        fclose(file);
+        return -4;
+    }
+
+    fclose(file);
+    *line_count = count;
+    return 0;
+}
+
+// 判断文件是否存在，返回 1 表示存在，0 表示不存在
+int file_exists_access(const char *filepath)
+{
+    if (filepath == NULL) {
+        fprintf(stderr, "Invalid argument.\n");
+        return 0;
+    }
+
+    return access(filepath, F_OK) == 0;
+}