使用clamav模块对数据流进行病毒检测

#define MAX_DATA_LEN 8000
#define MAX_FILE_LEN 1024*1024*10
#define START_CHECK_VIRUS  "zINSTREAM"

int virus_check(const char *szFileName){
    int sock,fd;
    int tmperror = 0;
    int len;
    int revl = 0;
    unsigned long int todo = MAX_FILE_LEN;
    char rbuf[256];
    uint32_t buf[MAX_DATA_LEN/sizeof(uint32_t)];
    sock = CreateConnectLocaleSocket("/tmp/clamd.socket", &tmperror);
    if (sock < 0){
        zlog_debug(cat,"CreateConnectLocaleSocket error");
        return -1;
    }
    fd = open (szFileName, O_RDONLY|0);
    if (fd < 0){
        zlog_debug(cat,"open [%s] err!",szFileName);
        return -1;
    }
    if(sendln(sock,START_CHECK_VIRUS, 10)){
        zlog_debug(cat,"sendln zINSTREAM erro \n");
        close(sock);
        return -1;
    }
    
    while((len = read(fd, &buf[1], sizeof(buf) - sizeof(uint32_t))) > 0){
        if((unsigned int)len > todo) 
            len = todo;
        buf[0] = htonl(len);
        if(sendln(sock, (const char *)buf, len+sizeof(uint32_t))){
            close(fd);
            close(sock);
            return -1;
        }
        todo -= len;
        if(!todo){
            len = 0;
            break;
        }
    }
    close(fd);

    *buf=0;
    sendln(sock, (const char *)buf, 4);

    memset(rbuf,0x00,sizeof(rbuf));
    revl = recv(sock, rbuf, sizeof(rbuf), 0);

    close(sock);
    return parse_Check_results(rbuf,revl);
}

int parse_Check_results(const char *bufdata,int bufLen){
    zlog_debug(cat,"parse_Check_results = [%s] ----------",bufdata);
    if(!strcmp(bufdata,"stream: OK")){
        return 0;
    }
    else{
        zlog_error(cat, "parse_Check_results  err, info = [%s]",bufdata);
    }
    return -2;
}

int sendln(int sockd, const char *line, unsigned int len){
    while(len)
        {
        int sent = send(sockd, line, len, 0);
        if(sent <= 0)
            {
            if(sent && errno == EINTR)
                {
                continue;
                }
            return 1;
            }
        line += sent;
        len -= sent;
        }
    return 0;
}


int CreateConnectLocaleSocket(char *pszHostString, int *pnResCode){
    int nConnectFd = 0;
    struct sockaddr_un struInAddr;//???? socket
    if((nConnectFd=socket(AF_UNIX,SOCK_STREAM,0))==-1){
        zlog_error(cat, "CreateConnectLocaleSocket err, errno = [%d] strerror(errno) = [%s]",
                   errno,strerror(errno));
        *pnResCode=errno;
        return -1;
        }
    struInAddr.sun_family = AF_UNIX;
    strcpy(struInAddr.sun_path, pszHostString);
    if(connect(nConnectFd,(struct sockaddr *)&struInAddr,sizeof(struct sockaddr_un))==-1)
        {
        zlog_error(cat, "CreateConnectLocaleSocket connect err, errno = [%d] strerror(errno) = [%s]",
                   errno,strerror(errno));
        close(nConnectFd);
        *pnResCode=errno;
        return -1;
        }
    return nConnectFd;
}