在写入文件之前检查文件中是否存在字符串/单词

Question

我有点小问题。仅当尝试写入的相同字符串/单词不存在时，我才想写入文件。在我的例子中，它是一个 IP 地址和端口，用“:”分隔。如果我手动写入文件，例如 193...:80 和 193...:22，它指示它们但仍然写入到文件。我的错误在哪里:

#ifdef DEBUG
#define INITIAL_ALLOC 2
#else
#define INITIAL_ALLOC 512
#endif


void BlockIP(char * s);
char *read_line(FILE *fin);
void *right_line(int p,char * s,int ssh,int http);



int main(int argc, char const *argv[])
{
    //struct uci_context *uci;
    //uci = uci_init();

    char *ip = "193.2.2.1 193.2.6.6 193.168.1.1 193.5.5.5"
    //int uci_port = atoi(ucix_get_option(uci, "pack_mon", "manual_blocking", "port"));

    const char s[2] = " ";
    char *token;


    token = strtok(ip, s);

    while( token != NULL ) 
    {
        BlockIP(token);
        token = strtok(NULL, s);
    }

    return(0);
}


void BlockIP(char * s){
    int ssh = 22;
    int http = 80;
    char sshbuffer[2000]= "\0";
    char httpbuffer[2000]= "\0";

    char cmd2[2000] = "\0";
    char cmd[2000] = "\0";

    snprintf(cmd, sizeof(cmd),"iptables -I INPUT 1 -p tcp -s %s --dport 80 -j REJECT",s);
    snprintf(cmd2, sizeof(cmd2),"iptables -I INPUT 1 -p tcp -s %s --dport 22 -j REJECT",s);

    snprintf(sshbuffer, sizeof(sshbuffer),"%s:%d", s,ssh);
    snprintf(httpbuffer, sizeof(httpbuffer),"%s:%d", s,http);


    FILE *fp ,*ft;
    fp = popen(cmd, "r");

    pclose(fp);

    ft = popen(cmd2, "r");

    pclose(ft);

    FILE *fin;
    char *line;


    fin = fopen("/tmp/mymonlog", "r");

    if ( fin ) {
        while ( line = read_line(fin) ) {
            if (strstr(line, sshbuffer)){
                printf("Already exist %s\n",line);
            }else{
                right_line(1,s,ssh,http);

            }
            if(strstr(line, httpbuffer)){
                printf("Already exist %s\n",line);
            }else{
                right_line(2,s,ssh,http);
            }
            free(line);
        }
    }

    fclose(fin);
}

char *read_line(FILE *fin) {
    char *buffer;
    char *tmp;
    int read_chars = 0;
    int bufsize = INITIAL_ALLOC;
    char *line = malloc(bufsize);

    if ( !line ) {
        return NULL;
    }

    buffer = line;

    while ( fgets(buffer, bufsize - read_chars, fin) ) {
        read_chars = strlen(line);

        if ( line[read_chars - 1] == '\n' ) {
            line[read_chars - 1] = '\0';
            return line;
        }

        else {
            bufsize = 2 * bufsize;
            tmp = realloc(line, bufsize);
            if ( tmp ) {
                line = tmp;
                buffer = line + read_chars;
            }
            else {
                free(line);
                return NULL;
            }
        }
    }
    return NULL;
}


void *right_line(int p,char * s,int ssh,int http){

    FILE *pFile,*tFile;

    if(p == 1){
        pFile=fopen("/tmp/mymonlog", "a");


        if(pFile==NULL) {
            perror("Error opening file.");
        }else {
            fprintf(pFile, "%s:%d\n", s,ssh);
        }

        fclose(pFile);
    }else if(p == 2){
        tFile=fopen("/tmp/mymonlog", "a");

        if(tFile==NULL) {
            perror("Error opening file.");
        }
        else {
            fprintf(tFile, "%s:%d\n", s,http);
        }

        fclose(tFile);
    }
}

Answer 1

它仍然写入文件的原因是您以逐行的方式检查文件。

假设您的文件有:

193.2.2.1:22
193.2.2.1:80

并且您想检查 193.2.2.1:22 和 193.2.2.1:80

首先你阅读193.2.2.1:22行

strstr("193.2.2.1:22", "193.2.2.1:22")

这是匹配的，所以它不会写入文件。

然后你就这样做了

strstr("193.2.2.1:22", "193.2.2.1:80")

这不匹配，因此它将写入文件。

然后您从文件中读取下一行 193.2.2.1:80 并执行

strstr("193.2.2.1:80", "193.2.2.1:22")

这不匹配，因此它将写入文件。

然后你做

strstr("193.2.2.1:80", "193.2.2.1:80")

这是匹配的，所以它不会写入文件。

现在你的文件是:

193.2.2.1:22
193.2.2.1:80
193.2.2.1:80
193.2.2.1:22

结论:

不要逐行检查。在决定写入文件之前，您需要检查文件中的所有行。

一个简单的修复可能是这样的:

int save_ssh = 1;
int save_http = 1;
if ( fin ) {
    while ( line = read_line(fin) ) {
        if (strstr(line, sshbuffer)){
            printf("Already exist %s\n",line);
            save_ssh = 0;
        }
        if(strstr(line, httpbuffer)){
            printf("Already exist %s\n",line);
            save_http = 0;
        }
        free(line);
    }

    fclose(fin);

    if (save_ssh) right_line(1,s,ssh,http);
    if (save_http) right_line(2,s,ssh,http);
}