亲宝软件园·资讯

展开

QT传输文件工具

一个眉头紧锁的码农 人气:0

最近因为一个事情很恼火,因为办公需要用到企业微信,但是企业微信只能在一个电脑上登陆,所以当别人发文件给你的时候,你只能一个电脑接收,创建共享文件夹也很麻烦,每次都需要去访问,很麻烦。所以准备自己写一个文件传输小工具。

功能就是能实现文件的双向传输,即客户端能传给服务端,服务端可以传给客户端。

使用的tcp通信,其实就是发消息,但是组合数据我是借鉴了IT1995大神写的代码。

先看下效果图

可以看到既可以接受文件也可进行发送文件,只要2台电脑在统一局域网内,就可发送和接受数据。

本地文件下出现了一份传输的文件。

直接看代码

.h

#ifndef WIDGET_H
#define WIDGET_H

#include <QWidget>
#include <QPushButton>
#include <QLineEdit>
#include <QTcpSocket>
#include <QTcpServer>
#include <QFile>
#include <QTextEdit>
#include <QProgressBar>

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 <QHBoxLayout>
#include <QLabel>
#include <QValidator>
#include <QMessageBox>
#include <QFileDialog>

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;
    }
}


这个小工具我会一直用的 

加载全部内容

相关教程
猜你喜欢
用户评论