#include "head.h"


int checkMD5(const unsigned char *MD5, MYSQL* mysql){
    int flag = 0;
    char sqlCommand[1024];
    bzero(sqlCommand, sizeof(sqlCommand));
    sprintf(sqlCommand, "select MD5 from vfsystem where MD5 = %s", MD5);
    int ret = mysql_query(mysql, sqlCommand);

    if(ret == 0){
        MYSQL_RES *result = mysql_store_result(mysql);
        // 行数
        MYSQL_ROW row = mysql_fetch_row(result);
        // 判断是否存在该md5
        if(row != NULL){
            LOGRECORD(INFO,"get MD5 from vfsystem");
            flag =1;        
        }
        mysql_free_result(result);
    }

    if(flag == 1){
        LOGRECORD(INFO,"MD5 exist in vfsystem");
        // 存在该md5则返回0
        return 0;
    }
    else{
        // 不存在该md5则返回-1
        return -1;
    }
}


int recvandcheck(int netfd, MYSQL* mysql, unsigned char *MD5){
    train_t train;
    bzero(&train, sizeof(train));

    //unsigned char md5FromSender[16];
    //bzero(md5FromSender, sizeof(md5FromSender));

    // 接收发送方的md5
    recvn(netfd, &train.length, sizeof(train.length));
    recvn(netfd, train.data, train.length);
    memcpy(MD5, train.data, train.length);
    for(int i=0; i<16; i++){
        printf("%02x", MD5[i]);
    }
    printf("\n\n");
    LOGRECORD(INFO,"get MD5 finish");

    int ret = checkMD5(MD5, mysql);
    return ret;
}