最近因为一个事情很恼火,因为办公需要用到企业微信,但是企业微信只能在一个电脑上登陆,所以当别人发文件给你的时候,你只能一个电脑接收,创建共享文件夹也很麻烦,每次都需要去访问,很麻烦。所以准备自己写一个文件传输小工具。
功能就是能实现文件的双向传输,即客户端能传给服务端,服务端可以传给客户端。
使用的tcp通信,其实就是发消息,但是组合数据我是借鉴了IT1995大神写的代码。
先看下效果图
可以看到既可以接受文件也可进行发送文件,只要2台电脑在统一局域网内,就可发送和接受数据。
本地文件下出现了一份传输的文件。
直接看代码
.h
#ifndef WIDGET_H
#define WIDGET_H
#include
#include
#include
#include
#include
#include
#include
#include
class Widget : public QWidget
{
Q_OBJECT
public:
Widget(QWidget *parent = nullptr);
~Widget();
void Init();
private slots:
void onTcpConnected();
void onConnectClicked();
void ServerNewConnect();
void SocketReadData();
void onOpenFileClicked();
void onSendClicked();
void updateClientProgress(qint64 numBytes);
private:
QPushButton *m_pConnectBtn=nullptr;
QLineEdit *m_pIpAddressEdit=nullptr;
QLineEdit *m_pPortEdit=nullptr;
QWidget *m_pTitleWgt=nullptr;
QLineEdit *m_pFilePathEdit=nullptr;
QPushButton *m_pOpenFileBtn=nullptr;
QPushButton *m_pSendBtn=nullptr;
QTextEdit *m_pTextEdit=nullptr;
QProgressBar *m_pReceiverBar=nullptr;
QProgressBar *m_pSendBar=nullptr;
QTcpSocket *m_pTcpSocket=nullptr;
QTcpServer *m_pTcpServer=nullptr;
QTcpSocket *m_pTcpServerSocket=nullptr;
//------receiver
qint64 m_bytesReceived;
qint64 m_fileNameSize;
qint64 m_totalBytes;
QString m_fileName;
QFile *m_localFile;
QByteArray m_inBlock;
//send
QFile *m_ClientlocalFile;
QString m_ClientfileName;
qint64 m_ClienttotalBytes;
qint64 m_ClientbytesWritten=0;
qint64 m_ClientbytesToWrite;
qint64 m_ClinetpayloadSize;
QByteArray m_ClientoutBlock;
};
#endif // WIDGET_H
.cpp
#include "widget.h"
#include
#include
#include
#include
#include
Widget::Widget(QWidget *parent)
: QWidget(parent)
{
this->resize(800,600);
Init();
m_pTcpSocket=new QTcpSocket(this);
connect(m_pTcpSocket,&QTcpSocket::connected,this,&Widget::onTcpConnected); //若连接成功,则触发此信号
connect(m_pTcpSocket,SIGNAL(bytesWritten(qint64)),this,SLOT(updateClientProgress(qint64))); //发送数据
m_pTcpServer=new QTcpServer(this);
m_totalBytes=0;
m_bytesReceived=0;
m_fileNameSize=0;
connect(m_pTcpServer,&QTcpServer::newConnection,this,&Widget::ServerNewConnect);
if(!m_pTcpServer->listen(QHostAddress::Any, 2021)) //端口为2021
{
QMessageBox::warning(this,"Warning",m_pTcpServer->errorString(),QMessageBox::Ok);
return;
}
}
Widget::~Widget()
{
}
void Widget::Init()
{
m_pConnectBtn=new QPushButton(tr("Connect"),this);
m_pIpAddressEdit=new QLineEdit(this);
m_pPortEdit=new QLineEdit(this);
m_pPortEdit->setValidator(new QIntValidator());
m_pTitleWgt=new QWidget(this);
m_pIpAddressEdit->setFixedWidth(200);
m_pPortEdit->setFixedWidth(200);
m_pConnectBtn->setFixedSize(100,25);
QLabel *ipLabel=new QLabel(tr("IpAddress:"),this);
QLabel *portLabel=new QLabel(tr("Port:"),this);
ipLabel->setFixedWidth(60);
portLabel->setFixedWidth(40);
QHBoxLayout *titleLayout=new QHBoxLayout(this);
titleLayout->addWidget(ipLabel);
titleLayout->addWidget(m_pIpAddressEdit);
titleLayout->addWidget(portLabel);
titleLayout->addWidget(m_pPortEdit);
titleLayout->addWidget(m_pConnectBtn);
titleLayout->setMargin(5);
titleLayout->setSpacing(10);
titleLayout->addStretch();
m_pTitleWgt->setFixedHeight(40);
m_pTitleWgt->setLayout(titleLayout);
m_pIpAddressEdit->setText("192.168.2.110");
m_pPortEdit->setText("2021");
m_pPortEdit->setEnabled(false);
m_pFilePathEdit=new QLineEdit(this);
m_pOpenFileBtn=new QPushButton(tr("Open File"),this);
m_pSendBtn=new QPushButton(tr("Send"));
m_pFilePathEdit->setFixedWidth(500);
m_pOpenFileBtn->setFixedSize(100,25);
m_pSendBtn->setFixedSize(100,25);
m_pSendBtn->setEnabled(false);
QWidget *bottomWgt=new QWidget(this);
QHBoxLayout *bottomLayout=new QHBoxLayout(this);
bottomLayout->addWidget(m_pFilePathEdit);
bottomLayout->addWidget(m_pOpenFileBtn);
bottomLayout->addWidget(m_pSendBtn);
bottomLayout->setMargin(5);
bottomLayout->setSpacing(5);
bottomLayout->addStretch();
bottomWgt->setLayout(bottomLayout);
m_pTextEdit=new QTextEdit(this);
QLabel *receiverLabel=new QLabel(tr("Receiver Speed"),this);
QLabel *SendLabel=new QLabel(tr("Send Speed"),this);
receiverLabel->setFixedWidth(100);
SendLabel->setFixedWidth(100);
m_pReceiverBar=new QProgressBar(this);
m_pSendBar=new QProgressBar(this);
m_pReceiverBar->setFixedSize(300,30);
m_pSendBar->setFixedSize(300,30);
m_pReceiverBar->setOrientation(Qt::Horizontal);
m_pSendBar->setOrientation(Qt::Horizontal);
QWidget *receiverBarWgt=new QWidget(this);
QHBoxLayout *receiverBarLayout=new QHBoxLayout(this);
receiverBarLayout->addWidget(receiverLabel);
receiverBarLayout->addWidget(m_pReceiverBar);
receiverBarLayout->addStretch();
receiverBarLayout->setSpacing(5);
receiverBarWgt->setLayout(receiverBarLayout);
QWidget *sendBarWgt=new QWidget(this);
QHBoxLayout *sendBarLayout=new QHBoxLayout(this);
sendBarLayout->addWidget(SendLabel);
sendBarLayout->addWidget(m_pSendBar);
sendBarLayout->addStretch();
sendBarLayout->setSpacing(5);
sendBarWgt->setLayout(sendBarLayout);
connect(m_pConnectBtn,&QPushButton::clicked,this,&Widget::onConnectClicked);
connect(m_pOpenFileBtn,&QPushButton::clicked,this,&Widget::onOpenFileClicked);
connect(m_pSendBtn,&QPushButton::clicked,this,&Widget::onSendClicked);
QVBoxLayout *mainLayout=new QVBoxLayout(this);
mainLayout->addWidget(m_pTitleWgt);
mainLayout->addWidget(bottomWgt);
mainLayout->addWidget(receiverBarWgt);
mainLayout->addWidget(sendBarWgt);
mainLayout->addWidget(m_pTextEdit);
mainLayout->setMargin(0);
mainLayout->addStretch();
this->setLayout(mainLayout);
}
void Widget::onTcpConnected()
{
m_pTextEdit->append("Connect Server Success!");
}
void Widget::onConnectClicked()
{
QString strip=m_pIpAddressEdit->text();
QString strport=m_pPortEdit->text();
if(strip!=""&&strport!="")
{
m_pTcpSocket->connectToHost(strip,strport.toInt()); //请求连接
}
else
{
QMessageBox::warning(this,"Warning","IpAddress or Port is Null",QMessageBox::Ok);
}
}
void Widget::ServerNewConnect()
{
m_pTcpServerSocket = m_pTcpServer->nextPendingConnection(); //服务端接受消息
QObject::connect(m_pTcpServerSocket, &QTcpSocket::readyRead, this, &Widget::SocketReadData);
m_pTextEdit->append("Connect Client Success");
}
void Widget::SocketReadData()
{
QDataStream in(m_pTcpServerSocket);
in.setVersion(QDataStream::Qt_5_11);
if (m_bytesReceived<=sizeof(qint64)*2){
if((m_pTcpServerSocket->bytesAvailable()>=sizeof(qint64)*2)&&(m_fileNameSize==0)){
in>>m_totalBytes>>m_fileNameSize;
m_bytesReceived +=sizeof(qint64)*2;
}
if((m_pTcpServerSocket->bytesAvailable()>=m_fileNameSize)&&(m_fileNameSize!=0)){
in>>m_fileName;
m_bytesReceived+=m_fileNameSize;
m_localFile = new QFile(m_fileName);
if (!m_localFile->open(QFile::WriteOnly)){
qDebug() << "server: open file error!";
return;
}
}
else{
return;
}
}
if(m_bytesReceived<m_totalBytes) {
m_bytesReceived+=m_pTcpServerSocket->bytesAvailable();
m_inBlock = m_pTcpServerSocket->readAll();
m_localFile->write(m_inBlock);
m_inBlock.resize(0);
}
m_pReceiverBar->setMaximum(m_totalBytes);
m_pReceiverBar->setValue(m_bytesReceived);
if (m_bytesReceived==m_totalBytes){
m_localFile->close();
QString strSuccess=QString("File %1 ReceiverSucess").arg(m_fileName);
m_pTextEdit->append(strSuccess);
m_pTcpServerSocket->close();
m_totalBytes=0;
m_bytesReceived=0;
m_fileNameSize=0;
}
}
void Widget::onOpenFileClicked()
{
QString fileName = QFileDialog::getOpenFileName(this, tr("Open File"),
"/home",
tr("File (*.*)"));
if(fileName!="")
{
m_ClientfileName=fileName;
m_pSendBtn->setEnabled(true);
m_pFilePathEdit->setText(fileName);
}
}
void Widget::onSendClicked()
{
m_ClientoutBlock.clear();
m_ClientlocalFile=new QFile(m_ClientfileName);
if(!m_ClientlocalFile->open(QFile::ReadOnly)){
qDebug()<<"client:open file error!";
return;
}
m_ClienttotalBytes=m_ClientlocalFile->size();
QDataStream sendOut(&m_ClientoutBlock,QIODevice::WriteOnly);
sendOut.setVersion(QDataStream::Qt_5_11);
QString currentFileName=m_ClientfileName.right(m_ClientfileName.size()-m_ClientfileName.lastIndexOf('/')-1);
sendOut<<qint64(0)<<qint64(0)<<currentFileName;
m_ClienttotalBytes+=m_ClientoutBlock.size();
sendOut.device()->seek(0);
sendOut<<m_ClienttotalBytes<<qint64(m_ClientoutBlock.size()-sizeof(qint64)*2);
m_ClientbytesToWrite=m_ClienttotalBytes-m_pTcpSocket->write(m_ClientoutBlock);
m_ClientoutBlock.resize(0);
}
void Widget::updateClientProgress(qint64 numBytes)
{
m_ClientbytesWritten+=(int)numBytes;
if(m_ClientbytesToWrite>0){
m_ClientoutBlock=m_ClientlocalFile->read(qMin(m_ClientbytesToWrite,m_ClinetpayloadSize));
m_ClientbytesToWrite-=(int)m_pTcpSocket->write(m_ClientoutBlock);
m_ClientoutBlock.resize(0);
}
else{
m_ClientlocalFile->close();
}
m_pSendBar->setMaximum(m_ClienttotalBytes);
m_pSendBar->setValue(m_ClientbytesWritten);
if(m_ClientbytesWritten==m_ClienttotalBytes){
QString sendSuccess=QString("Send File %1 Success").arg(m_fileName);
m_pTextEdit->append(sendSuccess);
m_ClientlocalFile->close();
m_pTcpSocket->close();
m_ClientbytesWritten=0;
}
}
这个小工具我会一直用的