151 lines
4.0 KiB
C
151 lines
4.0 KiB
C
#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;
|
||
}
|