tcp文件传输(udp发送消息确定是否接收)

原文链接  http://blog.bccn.net/liYX9553/61129


dialog.h:

#ifndef DIALOG_H
#define DIALOG_H
 
#include <QDialog>
#include <QTime>
 
class QTcpServer;
class QFile;
class QTcpSocket;
class QUdpSocket;
 
namespace Ui {
class Dialog;
}
 
class Dialog : public QDialog
{
    Q_OBJECT
 
public:
    explicit Dialog(QWidget *parent = 0);
    ~Dialog();
    void initser();// 初始化服务器。
 
 
private:
    Ui::Dialog *ui;
    qint16 tport;
    qint16 uport;
    qint16 u2port;
    qint64 totalBytes;
    qint64 writeBytes;
    qint64 tobeWriteBytes;
    qint64 payloadSize;//被初始化为第一个常量。
    QByteArray outBlock;
 
    QTcpSocket *clintcon;
    QUdpSocket *udpSocket;
    QTcpServer *tser;
    QString fileName;
    QString theFileName;
 
    QFile *locFile;
    QTime time;
protected:
    void SendFileName(QString fileName);
    QString getIP();
private slots:
    void sendMsg();
    void updClintProgress(qint64 numBytes);
    void getFileName(QString name);
    void refused();//关闭服务器。
    void on_sCloseBtn_clicked();
 
    void on_sOpenBtn_clicked();
 
    void on_sSendBtn_clicked();
 
signals:
    void sendFileName(QString fileName);
};
 
#endif // DIALOG_H


dialog.cpp:

#include "dialog.h"
#include "ui_dialog.h"
#include <QFile>
#include <QDebug>
#include <QMessageBox>
#include <QFileDialog>
#include <QTcpServer>
#include <QTcpSocket>
#include <QNetworkInterface>
#include "QObject"
#include <QUdpSocket>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include <qtcpserver.h>
 
Dialog::Dialog(QWidget *parent) :
    QDialog(parent),
    ui(new Ui::Dialog)
{
    ui->setupUi(this);
    udpSocket = new QUdpSocket(this);
    uport = 8888;
    u2port = 7777;
    udpSocket->bind(u2port, QUdpSocket::ShareAddress | QUdpSocket::ReuseAddressHint);
    connect(udpSocket, SIGNAL(readyRead()), this, SLOT(refused()));
    //sendMessage(NewParticipant);
    tport = 6666;
    tser = new QTcpServer(this);
    connect(tser,SIGNAL(newConnection()),this,SLOT(sendMsg()));
    connect(this,SIGNAL(sendFileName(QString)),this,SLOT(getFileName(QString)));
    initser();
}
 
Dialog::~Dialog()
{
    delete ui;
}
void Dialog::initser()//初始化服务器。
{
    totalBytes = 0;
    tobeWriteBytes = 0;
    writeBytes = 0;
    ui->sOpenBtn->setEnabled(true);//open按钮可用
    ui->sSendBtn->setEnabled(false);//发送按钮不可用
 
    ui->label_2->setText("请选择要发送的文件");
    tser->close();
}
void Dialog::refused()//关闭服务器。
{
    tser->close();
    ui->label_2->setText(tr("对方拒绝接收!"));
}
//发送文件
void Dialog::sendMsg()
{
    qDebug() << "--------sendMsg start-----------" ;
    ui->sSendBtn->setEnabled(false);//设置按钮不可用
    clintcon = tser->nextPendingConnection();//获取一个已经连接tcp套接字。
    connect(clintcon,SIGNAL(bytesWritten(qint64)),this,SLOT(updClintProgress(qint64)));
 
    ui->label_2->setText(tr("开始传送文件 %1!").arg(theFileName));
    locFile = new QFile(fileName);//只读方式打开文档
    if(!locFile->open((QFile::ReadOnly))){
         QMessageBox::warning(this,tr(""),tr("无法读取文件%1:\n %2").arg(fileName).arg(locFile->errorString()));
         return;
    }
    totalBytes = locFile->size();//文件的总字节数。
    qDebug() << "fileTotalBytes:" << totalBytes;
    QDataStream sendOut(&outBlock,QIODevice::WriteOnly);
    sendOut.setVersion(QDataStream::Qt_4_7);
    time.start();
    QString curFile = fileName.right(fileName.size() - fileName.lastIndexOf('/') -1 );//待发送文件名。
    qDebug() << "curFile:" << curFile;
    sendOut << qint64(0) << qint64(0) << curFile;
    totalBytes += outBlock.size();//文件名大小信息+实际文件大小。
    sendOut.device()->seek(0);//设置文件指针从0开始
    sendOut << totalBytes <<  qint64((outBlock.size() - sizeof(qint64)*2));
    tobeWriteBytes = totalBytes - clintcon->write(outBlock);//发送缓存区的数据,同时修改待发送的数据长度。
    qDebug() << "tobeWriteBytes:" << tobeWriteBytes;
    qDebug() << "outBlock:" << outBlock;
    outBlock.resize(0);
}
 
void Dialog::on_sCloseBtn_clicked()
{
    if(tser->isListening()){
        tser->close();//当tcp正在监听时,关闭tcp服务器端应用,即按下close键时就不监听tcp请求了
        if(locFile->isOpen()){
            locFile->close();
        }
        clintcon->abort();//Socket::终止当前的连接,禁止写入缓存区。
    }
    close();
}
 
void Dialog::on_sOpenBtn_clicked()
{
    fileName = QFileDialog::getOpenFileName(this, tr("Open File"),
                 "/home"
                 );//tr("Images (*.png *.xpm *.jpg *.txt *.gz)")
    //fileName = QFileDialog::getOpenFileName(this);
    if(theFileName.isEmpty()){
        theFileName = fileName.right(fileName.size() - fileName.lastIndexOf('/') - 1);
        ui->label_2->setText(tr("要发送的文件为:%1").arg(theFileName));
        qDebug() << "thefilename:" << theFileName;
    }
    qDebug() << "fileName:" << fileName;
    ui->sSendBtn->setEnabled(true);
}
 
void Dialog::on_sSendBtn_clicked()
{
        qint8 a;
        a = tser->listen(QHostAddress::Any,tport);
    if(!a){
        qDebug() << tser->errorString();
        close();
    }
    qDebug() << a;
    ui->label_2->setText(tr("等待对方接收...."));
    emit sendFileName(theFileName);
    qDebug() << "-----emit- success!------";
    ui->sSendBtn->setEnabled(false);
}
//进度条修改进度。
void Dialog::updClintProgress(qint64 numBytes)
{
    qDebug() << "udpClintProgress  start....";
    qApp->processEvents();
    //qDebug() << "writeBytes1:"<< writeBytes;
    //writeBytes += (int)numBytes;
    //qDebug() << "writeBytes2:" << writeBytes;
    if(tobeWriteBytes > 0){//没发送完。
        outBlock = locFile->read(qMin(tobeWriteBytes,payloadSize));//每次最多发送64k大小。
        tobeWriteBytes -= (int)clintcon->write(outBlock);
        writeBytes += (totalBytes - tobeWriteBytes);//已经发送了的。
        outBlock.resize(0);
    }else {
        locFile->close();
    }
    ui->progressBar->setMaximum(totalBytes);
    ui->progressBar->setValue(writeBytes);
    float useTime = time.elapsed();
    double speed = writeBytes/useTime*1000000;
 
    ui->label_2->setText(tr("(%1 m/s)").arg(speed));
    qDebug() << "writeBytes:"<< writeBytes;
    qDebug() << "totalBytes:" << totalBytes;
 
    if(writeBytes == totalBytes){
        locFile->close();
        tser->close();
        ui->label_2->setText(tr("文件:%1发送成功!").arg(theFileName));
    }
 
}
//获取文件名
void Dialog::getFileName(QString name)
{
        qDebug() << "------1getFileName--------" << name;
        theFileName = name;
        //qDebug() << "fileNamesize" << strlen(fileName);
        SendFileName(theFileName);
}
//发送文件名
void Dialog::SendFileName(QString fileName)
{
        QByteArray data;
        QDataStream out(&data, QIODevice::WriteOnly);
 
        QString address,clientaddr;
        address  = getIP();
        //clientaddr = getIP();
        //QString clientAddress = ui->userTableWidget->item(row, 2)->text();//(row,,2)为ip地址
        out << address << fileName;//发送本地ip,所发送的文件名
        qDebug() << "SendFileName :" << fileName;
        qDebug() << "server IP:" << address;
        int a= udpSocket->writeDatagram(data,data.length(),QHostAddress::Broadcast, uport);//向8888端口广播
        qDebug() << "a" << a;
}
//获取IP
QString Dialog::getIP()
{
    QList<QHostAddress> list = QNetworkInterface::allAddresses();
    foreach (QHostAddress address, list) {
        if(address.protocol() == QAbstractSocket::IPv4Protocol){
            if(address.toString().contains("192.168.1")||address.toString().contains("127.0"))
                continue;
            return address.toString();
        }
    }
    return 0;
}

client.h

#ifndef DIALOG_H
#define DIALOG_H
 
#include <QDialog>
#include <QHostAddress>
#include <QFile>
#include <QTime>
#include <QDebug>
#include <QMessageBox>
#include"thread.h"
enum MessageType{FileName, refuse};
class QTcpSocket;
class QUdpSocket;
namespace Ui {
class Dialog;
}
 
class Dialog : public QDialog
{
    Q_OBJECT
 
public:
    explicit Dialog(QWidget *parent = 0);
    ~Dialog();
 
    void setHostaddr(QHostAddress addr);                         //获取主机ip
    void setFileName(QString FileName);                         //获取文件保存路径
    void reciveFileNmae();
private slots:
 
    void on_cCancelBtn_clicked();
 
    void on_cClosebtn_clicked();
 
protected:
    void SendRefuse(MessageType type);
    void closeEvent(QCloseEvent *);
    void hansPendingFile(QString serAddress,QString fileName);
    QString getIP();
private:
    Thread *b;
    Ui::Dialog *ui;
    QUdpSocket *udpSocket;
    QTcpSocket *cClint;
 
    QHostAddress hostAddr;
    qint16 tport;
    qint16 uport;
    qint16 u2port;
    quint16 blockSize;
    qint64 totalBytes;
    qint64 recevBytes;
    qint64 fileNameSize;
 
    QString FileName;
    QString *SerAdress;
    QFile *locFile;                          //要接收的文件
 
    QByteArray inBlock;                     //缓存一次接收的数据
    QTime Time;
private slots:
    void processPendingDatarams();
    void newConn();
    void readMsg();
    void displayError(QAbstractSocket::SocketError);
 
signals:
 
};
 
#endif // DIALOG_H


client.cpp

#include "dialog.h"
#include "ui_dialog.h"
 
#include <QTcpSocket>
#include <QUdpSocket>
#include <QTime>
#include <QDebug>
#include <QMessageBox>
#include <QNetworkInterface>
#include <QFile>
#include <QFileDialog>
#include <QDateTime>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <qiodevice.h>
 
 
Dialog::Dialog(QWidget *parent) :
    QDialog(parent),
    ui(new Ui::Dialog)
{
    ui->setupUi(this);
    setWindowTitle("client");
    totalBytes = 0;
    recevBytes = 0;
    fileNameSize = 0;
    udpSocket = new QUdpSocket(this);
    uport = 8888;
    u2port = 7777;
    udpSocket->bind(QHostAddress::Broadcast, uport, QUdpSocket::ShareAddress| QUdpSocket::ReuseAddressHint);
    connect(udpSocket, SIGNAL(readyRead()), this, SLOT(processPendingDatarams()));
    tport = 6666;
    cClint = new QTcpSocket(this);
    connect(cClint,SIGNAL(readyRead()),this,SLOT(readMsg()));
    connect(cClint, SIGNAL(error(QAbstractSocket::SocketError)), this,
                SLOT(displayError(QAbstractSocket::SocketError)));
}
Dialog::~Dialog()
{
    delete ui;
}
// 设置地址
void Dialog::setHostaddr(QHostAddress addr)
{
    qDebug() << "--------setHostaddr---------";
    hostAddr = addr;
    qDebug() << "hostAddr:" << hostAddr;
    newConn();
}
//设置文件名
void Dialog::setFileName(QString FileName)
{
    locFile = new QFile(FileName);
}
void Dialog::newConn()                   // 设置与服务器建立连接的。
{
    qDebug() << "---------newConn start---------";
    blockSize = 0;
    cClint->abort();
    cClint->connectToHost(hostAddr,tport);
    Time.start();
}
//接收文件
void Dialog::readMsg(){
    qDebug() << "-----------readMsg start-----------";
    QDataStream in(cClint);
    in.setVersion(QDataStream::Qt_4_7);
 
    float useTime = Time.elapsed();
    if(recevBytes <= sizeof(qint64)*2){
        if((cClint->bytesAvailable() >= sizeof(qint64)*2)&& fileNameSize == 0){
            //接收数据总大小信息和数据文件名大小。
            in >> totalBytes >> fileNameSize;
            qDebug() << "totalBytes:" << totalBytes;
            qDebug() << "fileNameSize:" << fileNameSize;
            recevBytes += sizeof(qint64)*2;
            qDebug() << "1recevBytes:" << recevBytes;
        }
        if((cClint->bytesAvailable() >= fileNameSize) && (fileNameSize != 0)){
            //开始接收文件,并建立文件。
            in >> FileName;
            qDebug() << "FileName:" << FileName;
            recevBytes += fileNameSize;
            qDebug() << "2recevBytes:" << recevBytes;
            if(!locFile->open(QFile::WriteOnly)){
                QMessageBox::warning(this,tr("应用程序"),tr("无法读取文件%1: \n%2").arg(FileName).arg(locFile->errorString()));
                return;
            }
        }else{
            return;
        }
    } 
    qDebug() << "-----------BytesArray write newFile of start-----------";
    if(recevBytes < totalBytes){
        recevBytes += cClint->bytesAvailable();        //把缓存区的内容写入文件。
        inBlock = cClint->readAll();
        locFile->write(inBlock);
        inBlock.resize(0);                             //清空缓存区
    }
    ui->progressBar->setMaximum(totalBytes);
    ui->progressBar->setValue(recevBytes);//设置进度条的进度
    double speed = recevBytes/useTime;
    ui->label->setText(tr("(%1 m/s)").arg(speed));
    qDebug() << "recevBytes:" << recevBytes;
    qDebug() << "totalBytes:" << totalBytes;
    if(recevBytes == totalBytes){
        locFile->close();
        cClint->close();
        ui->label->setText(tr("接收文件 %1完毕 ").arg(FileName));
    }
}
 
void Dialog::closeEvent(QCloseEvent *)
{
    on_cClosebtn_clicked();
}
 
void Dialog::on_cCancelBtn_clicked()
{
    cClint->abort();
    if(locFile->isOpen()){
        locFile->close();
    }
}
 
void Dialog::on_cClosebtn_clicked()       //关闭服务
{
    cClint->abort();
    if(locFile->isOpen()){
        locFile->close();
    }
    close();
}
//处理接收到的文件名
void Dialog::hansPendingFile(QString serAddress,QString fileName)
{
    QString cIpAddress = getIP(); //"192.168.31.119";
    qDebug() << "cIpAddress:" << cIpAddress;
    int btn = QMessageBox::information(this,tr("接收文件"),
                                       tr("文件%1是否接收?").arg(fileName),
                                       QMessageBox::Yes,QMessageBox::No);
    if(btn == QMessageBox::Yes){
        QString name = QFileDialog::getSaveFileName(0,tr("保存文件"),fileName);
        if(!name.isEmpty()){
            Dialog *clint = new Dialog(this);
            setFileName(name);
            setHostaddr(QHostAddress(serAddress));//默认一台电脑用。
        }
    }
    else{
       SendRefuse(refuse);
        sleep(2);
        on_cClosebtn_clicked();
    }
}
QString Dialog::getIP()
{
    QList<QHostAddress> list = QNetworkInterface::allAddresses();
    foreach (QHostAddress address, list) {
        if(address.protocol() == QAbstractSocket::IPv4Protocol){
            if(address.toString().contains("192.168.1")||address.toString().contains("127.0"))
                continue;
            return address.toString();
           }
    }
    return 0;
}
//处理udp发送名字信号时函数
void Dialog::processPendingDatarams()
{
    qDebug() << "pending Datarams......." ;
    while(udpSocket->hasPendingDatagrams()){
        QByteArray datagram;
        datagram.resize(udpSocket->pendingDatagramSize());
        udpSocket->readDatagram(datagram.data(), datagram.size());
        QDataStream in(&datagram,QIODevice::ReadOnly);
        QString time = QDateTime::currentDateTime()
                .toString("yyyy-MM-dd hh:mm:ss");
 
        //QString clientAddress, fileName;
        QString fileName,serAddress;
        in >> serAddress >>  fileName;
        qDebug() << "pending fileName" << fileName;
        hansPendingFile(serAddress,fileName);
    }
}
void Dialog::SendRefuse(MessageType refuse)
{
        QByteArray data;
        QDataStream out(&data, QIODevice::WriteOnly);
        out << refuse;
        qDebug() << "data" << data;
        int a= udpSocket->writeDatagram(data,data.length(),QHostAddress::LocalHost, u2port);
        qDebug() << "------------writeDatagram a:" << a;        
}
void Dialog::displayError(QAbstractSocket::SocketError socketError)
{
    switch(socketError)
        {
        //RemoteHostClosedError为远处主机关闭了连接时发出的错误信号
        case QAbstractSocket::RemoteHostClosedError : break;
        default : qDebug() << cClint->errorString();
        }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值