#include "filetranseceiver.h"
#include <QDebug>
#include <QFile>
#include <QString>
#include <QTimer>
#include <iostream>
#include <string>
//#include "LogInfo.h"
#include <QCoreApplication>

FileTranseceiver::FileTranseceiver(uint16_t hostshort)
{
//	struct sockaddr_in server_addr;
//	bzero(&server_addr, sizeof(server_addr)); // 置字节字符串前n个字节为0，包括'\0'
//    server_addr.sin_family = AF_INET;
//    server_addr.sin_addr.s_addr = htons(INADDR_ANY); // 转小端,INADDR_ANY就是指定地址为0.0.0.0的地址
//    server_addr.sin_port = htons(5000);
    serverThead = NULL;
    sockfd = socket(PF_INET, SOCK_STREAM, 0);

    port=hostshort;
    struct sockaddr_in seraddr;
    seraddr.sin_family = AF_INET;
    seraddr.sin_port = htons(hostshort);
    seraddr.sin_addr.s_addr = htons(INADDR_ANY);
	int tmp = 1;
    setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &tmp, sizeof(int));
    if(bind(sockfd,(struct sockaddr *)&seraddr,sizeof(seraddr)) == -1)
    {
        qDebug() << "tftp server bind error";
        return;
    }

    if(listen(sockfd,5) == -1)
    {
        qDebug() << "listen error";
        return;
    }

    serverThead = new QThread();
    moveToThread(serverThead);
    QObject::connect(serverThead, SIGNAL(started()), this, SLOT(connectListen()));
    //connect(&m_thread, SIGNAL(finished()), this, SLOT(deleteLater()));
//    serverThead->start();
//    qDebug()<<"tftp server on...";
    if(hostshort==5002)
    {
        refresh_timer = new QTimer();
        connect(refresh_timer, SIGNAL(timeout()), this, SLOT(refreshtimerHandler()),Qt::DirectConnection);
    }
}
FileTranseceiver::~FileTranseceiver(void)
{
    if(serverThead) {
        disconnect(serverThead, SIGNAL(started()), this, SLOT(connectListen()));
        if (serverThead->isRunning()) {
            serverThead->quit();
            serverThead->wait();
        }
    }
    if(connfd)
        close(connfd);
    if(sockfd)
        close(sockfd);
}
void FileTranseceiver::start()
{
    if(serverThead)
    {
        serverThead->start();
    }
    qDebug()<<"tftp server on...";
}

void FileTranseceiver::connectListen()
{
    switch(port)
    {
    case 5001:
        connectListen5001();
        break;
    case 5002:
        connectListen5002();
        break;
    }
}

void FileTranseceiver::connectListen5001()
{

	struct msg info;
	struct sockaddr_in cliaddr;
    int len = sizeof(cliaddr);
    pid_t pid;
    int n;
    QString fileName;
    do
    {
        connfd = accept(sockfd,(struct sockaddr *)&cliaddr,(socklen_t *)&len);
        if(connfd == -1)
        {
            qDebug() << "accept";
            continue;
        }

        {
            //close(sockfd);
            while(1)
            {
                n = recv(connfd,&info,sizeof(info),0);
                if(n == -1)
                {
                    qDebug() << "recv";
                    //exit(EXIT_FAILURE);
                    break;
                }
                if(n == 0)
                {
                    qDebug() << "客户端退出";
                    break;
                }
                switch(info.type)
                {
                case DOWN:
                    ftp_download(connfd,info);
                    break;
                case UP:
                    upload_ftp(connfd,info);
                    fileName = QString(info.fname);
                    qDebug()<<"emit fileRecieved(fileName)";
                    emit fileRecieved(fileName);
                    break;
                case UPGRADE:
                    upload_and_upgrade(connfd,info);
                    QFile::remove("/home/plg/QtUI");
                     if(!QFile::copy("/tmp/QtUI", "/home/plg/QtUI"))
                    {
                        QMessageBox::about(0, tr("错误"), tr("更新失败！"));
                        break;
                    }
                    QMessageBox::information(0, tr("提示"), tr("软件更新成功,重启后生效!"));
                    break;
                default:
                    break;
                }
            }
            close(connfd);
        }
    }
    while(1);
}

//将某文件中的内容传给客户端
int FileTranseceiver::ftp_download(int connfd,struct msg info)
{
    FILE *fp;

    char name[100];
    memset(name,0,100);
    sprintf(name, DOC_DIR"/%s",info.fname);
    fp = fopen(name,"r");
    if(fp == NULL)
    {
        info.flag = FAIL;
        send(connfd,&info,sizeof(info),0);
        return -1;
    }
    else 
    {
        info.flag = SUCCESS;
        send(connfd,&info,sizeof(info),0);
    }

    unsigned int len;
    while(1)
    {
        len = fread(info.buf,1,sizeof(info.buf),fp);
        send(connfd,info.buf,len,0);

        if(len < sizeof(info.buf) && feof(fp) != 0)
            break;
    }

    fclose(fp); 
    return 0;
}

//接收客户端上传的文件内容
int FileTranseceiver::upload_ftp(int connfd,struct msg info)
{
    FILE *fp;
    char name[100];
    sprintf(name, "/tmp/%s",info.fname);
    fp = fopen(name,"w");
    if(fp == NULL)
    {   
        info.flag = FAIL;
        send(connfd,&info,sizeof(info),0);
        return -1;
    }
    else 
    {
        info.flag = SUCCESS;
        send(connfd,&info,sizeof(info),0);
    }
    unsigned int len;
    while(1)
    {
        len = recv(connfd,info.buf,sizeof(info.buf),0);
        fwrite(info.buf,1,len,fp);
        if(len < sizeof(info.buf))
            break;
    }
    fclose(fp);
    return 0;
}

int FileTranseceiver::upload_and_upgrade(int connfd,struct msg info)
{
    FILE *fp;
    char name[100];
    sprintf(name, "/tmp/%s",info.fname);
    fp = fopen(name,"wb");
    unsigned int len;
    while(1)
    {
        len = recv(connfd,info.buf,sizeof(info.buf),0);
        if(0<len)
            fwrite(info.buf,1,len,fp);
        else break;
    }
    fclose(fp);
    return 0;
}
QByteArray FileTranseceiver::read(int length)
{
    char buf[100];
    QByteArray b;
//recv file contant
unsigned int len;
while(1)
{
    len = recv(connfd,buf,length-b.size(),0);

    if(len ==0)
    {
        qDebug()<<"Connection Closed";
        break;
    }
    b.append(buf,len);

    if(b.size()==length)break;
}
return b;
}
/*
void MySleep100(unsigned int msec)
{

    QTime dieTime = QTime::currentTime().addMSecs(msec);

    while( QTime::currentTime() < dieTime )

        QCoreApplication::processEvents(QEventLoop::AllEvents, 100);

}
*/
static bool flagOfSockfdClosed = true;
void FileTranseceiver::connectListen5002()
{
    struct sockaddr_in cliaddr;
    int len = sizeof(cliaddr);
    do
    {
        connfd = accept(sockfd,(struct sockaddr *)&cliaddr,(socklen_t *)&len);
        if(connfd == -1)
        {
            qDebug() << "5002 Connection failure :"<<connfd;
            continue;
        }
        flagOfSockfdClosed = false;
       refresh_timer->start(5000);
        {
            while(1)
            {
                if(flagOfSockfdClosed)
                    break;
                QByteArray a=read(7);
                showData("read-a=",a);
                if(!a.startsWith(QByteArray::fromHex("5aa5")))
                {
                    qDebug() << "无效数据-Header";
                    break;
                }

                int InstructionCode=a.mid(2,1).toHex().toInt(0,16);
                int DataLenght=a.mid(3,4).toHex().toInt(0,16);

                if(DataLenght >60)
                {
                    qDebug() << "无效数据-DataLenght";
                    break;
                }

                QByteArray b=read(DataLenght+3);
                showData("read-b=",b);
                if(!b.endsWith(QByteArray::fromHex("a55a")))
                {
                    qDebug() << "无效数据-Ender";
                    break;
                }
                if(CheckSumIn(a+b))
                {
                    switch (InstructionCode) {
                    case 0x01:InstructionCode1(a+b); break;//启动指令
                    case 0x02:InstructionCode2(a+b); break;//停止指令
                    case 0x03:InstructionCode3(b.left(DataLenght));break;//喷码指令
                    case 0x04:InstructionCode4(a+b);break;//获取缓存指令
                    //case 0x05:break;//喷码完成
                    //case 0x06:break;//
                    case 0x07:InstructionCode7(a+b);break;//清空缓存
                    case 0x08:InstructionCode8(a+b);break;//丢弃废膜指令
                    default:
                        write(InstructionCode,QByteArray::fromHex("00"));
                        break;
                    }
                }
                else {
                    write(InstructionCode,QByteArray::fromHex("00"));
                    qDebug() << "无效数据-Checksum";
                    break;
                }
            }
            if(!flagOfSockfdClosed)
                close(connfd);
        }
        refresh_timer->stop();
    }
    while(1);
}
bool FileTranseceiver::CheckSumIn( QByteArray a)
{
    UCHAR add8=CheckSumIn((UCHAR*)a.data(),a.length()-3);
//    qDebug()<<"CheckSumIn: "<<add8;
    return add8==(UCHAR)a.at(a.length()-3);
}

UCHAR FileTranseceiver::CheckSumIn( UCHAR *buf, unsigned int len)
{
    unsigned int i;
    UCHAR checksum;
    checksum = 0;
    for (i = 0; i < (len - 0); i++)
    {
        checksum += buf[i];
    }
    return checksum;
}

void FileTranseceiver::showData(QString tag, QByteArray dat)const
{
    int i,len;
    QString s="";
    len = dat.length();
    for(i=0;i<len;i++)
    {
        s+=QString().sprintf("%02X ",(unsigned char)dat.at(i));
    }
    if(s != "5A A5 06 00 00 00 01 01 07 A5 5A " && s != "5A A5 05 00 00 00 01 01 06 A5 5A ")
        qDebug()<<tag<<s;
}
void FileTranseceiver::refreshtimerHandler()
{
    write(0x06,QByteArray::fromHex("01"));
}
void FileTranseceiver::write(int code,QByteArray d)
{
    mutex.lock();
    QByteArray s=QByteArray::fromHex("5aa5");
    s.append(code);
    s.append(QByteArray::fromHex(QString::number(d.size(),16).rightJustified(8,'0',true).toStdString().data()));
    s.append(d);
    UCHAR add8=CheckSumIn((UCHAR*)s.data(),s.length());
    s.append(add8);
    s.append(QByteArray::fromHex("a55a"));
    QString log=s.toHex();
    showData("write=",s);
    int ret = 0;
    ret == send(connfd,s.data(),s.length(),0);
    if(ret == -1) {
        qDebug()<<"Send Failure";
        close(connfd);
        refresh_timer->stop();
        flagOfSockfdClosed = true;
    }
    mutex.unlock();
}

void FileTranseceiver::InstructionCode1(QByteArray d){
    emit printer_operate_0x01();
}
void FileTranseceiver::InstructionCode2(QByteArray d){
    emit printer_operate_0x02();
    write(2,QByteArray::fromHex("01"));
}
void FileTranseceiver::InstructionCode3(QByteArray d)
{
    emit printer_operate_0x03(QString::fromUtf8(d));
    write(3,QByteArray::fromHex("01"));
}
void FileTranseceiver::InstructionCode4(QByteArray d){
    emit printer_operate_0x04();
}
void FileTranseceiver::InstructionCode5(QByteArray d){
    write(5,QByteArray::fromHex("01"));
}
void FileTranseceiver::InstructionCode6(QByteArray d){}

void FileTranseceiver::InstructionCode7(QByteArray d){
    emit clear_DataCache();
    write(7,QByteArray::fromHex("01"));
}

void FileTranseceiver::InstructionCode8(QByteArray d){
    emit drop_NextSignal();
    write(8,QByteArray::fromHex("01"));
}

//启动指令-回收槽就绪
void FileTranseceiver::printer_operate_0x01_result(int status)
{
    write(1,QByteArray::fromHex(status==0?"00":"01"));
}
void FileTranseceiver::WriteInstructionCodeAck(int InstructionCode,QByteArray data)
{
    write(InstructionCode,data);
}