307 lines
7.2 KiB
Plaintext
307 lines
7.2 KiB
Plaintext
#include "filetranseceiver.h"
|
|
#include <QDebug>
|
|
#include <QFile>
|
|
#include <QString>
|
|
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...";
|
|
}
|
|
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;
|
|
}
|
|
|
|
// pid = fork();
|
|
// if(pid == -1)
|
|
// {
|
|
// qDebug() << "fork";
|
|
// continue;
|
|
// }
|
|
// else if(pid == 0)
|
|
{
|
|
//close(sockfd);
|
|
while(1)
|
|
{
|
|
n = recv(connfd,&info,sizeof(info),0);
|
|
if(n == -1)
|
|
{
|
|
qDebug() << "recv";
|
|
//exit(EXIT_FAILURE);
|
|
break;
|
|
}
|
|
if(n == 0)
|
|
{
|
|
qDebug() << "客户端退出";
|
|
//exit(EXIT_SUCCESS);
|
|
break;
|
|
}
|
|
switch(info.type)
|
|
{
|
|
// case SEARCH:
|
|
// search_serverdir(connfd,info);
|
|
// break;
|
|
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);
|
|
}
|
|
// else
|
|
// {
|
|
// 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);
|
|
// QFile dl_file(QString::fromUtf8(name));
|
|
// if (!dl_file.exists()) {
|
|
// return -1;
|
|
// }
|
|
//qDebug()<<"receive file name:"<<QString(name);
|
|
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, DOC_DIR"/%s",info.fname);
|
|
sprintf(name, "/tmp/%s",info.fname);
|
|
// QFile ul_file(QString::fromUtf8(name));
|
|
// if (!ul_file.exists()) {
|
|
// return -1;
|
|
// }
|
|
//create and open file
|
|
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);
|
|
}
|
|
//recv file contant
|
|
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, DOC_DIR"/%s",info.fname);
|
|
sprintf(name, "/tmp/%s",info.fname);
|
|
// QFile ul_file(QString::fromUtf8(name));
|
|
// if (!ul_file.exists()) {
|
|
// return -1;
|
|
// }
|
|
//create and open file
|
|
fp = fopen(name,"wb");
|
|
// if(fp == NULL)
|
|
// {
|
|
// info.flag = FAIL;
|
|
// send(connfd,&info,sizeof(info),0);
|
|
// return -1;
|
|
// }
|
|
// else
|
|
// {
|
|
// info.flag = SUCCESS;
|
|
// send(connfd,&info,sizeof(info),0);
|
|
// }
|
|
//recv file contant
|
|
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;
|
|
}
|
|
|
|
|
|
void FileTranseceiver::connectListen5002()
|
|
{
|
|
}
|
|
bool FileTranseceiver::CheckSumIn( UCHAR *buf, unsigned int len)
|
|
{
|
|
unsigned int i;
|
|
UCHAR checksum;
|
|
checksum = 0;
|
|
for (i = 0; i < (len - 1); i++)
|
|
{
|
|
checksum += buf[i];
|
|
}
|
|
if (buf[len - 1] == checksum)
|
|
{
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
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));
|
|
}
|
|
qDebug()<<tag<<s;
|
|
}
|