communication.h

Idź do dokumentacji tego pliku.

#ifndef COMMUNICATION_H
#define COMMUNICATION_H

#include <QObject>
#include <QHostAddress>
#include <QUdpSocket>
#include <QList>
#include <QString>
#include <QHash>
#include <QTime>
#include <QTimer>

#include "fileTransfer.h"

uint qHash (const QHostAddress &ip);

class LanPacket
{
public:

        enum PacketType
        {
                Invalid=0,
                Welcome, Hi, Bye,
                Ping, Pong,
                SysPing, SysPong,
                Bmsg, BmsgOk,
                Pmsg, PmsgOk,
                FileReq, FileOk, FileRefuse
        };

        LanPacket ();

        LanPacket (const QByteArray &data);

        LanPacket (const enum PacketType packet_type);

        LanPacket (const enum PacketType packet_type, const QString& str_data, quint32 file_size=0, quint16 port=0);

        bool isValid () const;

        QByteArray constData ();

        PacketType getType () const;

        quint16 getSize () const;

        QString getString () const;

        quint32 getFileSize () const;

        quint16 getPort () const;

private:

        QByteArray packInt (quint32 num, bool bit32=false);

        void clearData ();

        bool validate ();

        bool    is_valid;       
        QByteArray packet_data; 
        LanPacket::PacketType packet_type;      
        quint16 data_size;      
        QString str_data;               
        quint32 file_size;      
        quint16 port;                           
};


class Communication : public QObject
{
Q_OBJECT
public:
         Communication ();

         ~Communication ();

        void welcome (const QString& nick, QHostAddress broadcast, quint16 port=2310);

        void bye ();

        void sendPMsg (const QHostAddress& user_ip, const QString& msg);

        void sendBMsg (const QString& msg);

        void sendFile (QHostAddress user_ip, QString file_name);

        void checkPing (QHostAddress user_ip);

        void acceptFileRequest (QHostAddress user_ip, const QString &path);

        void refuseFileRequest (QHostAddress user_ip);

        void cancelTranfer ();

signals:

        void connected ();

        void unableConnect ();

        void newUser(QHostAddress user_ip, QString user_nick);

        void leaveUser (QHostAddress user_ip);

        void recvPMsg (QHostAddress user_ip, QString msg);

        void recvBMsg (QHostAddress user_ip, QString msg);

        void fileRequest (QHostAddress user_ip, QString file_name, quint32 size);

        void acceptSendFile (QHostAddress user_ip, QString file_name);

        void refuseSendFile (QHostAddress user_ip, QString file_name);

        void transferEnd (QHostAddress user_ip, bool ok);

        void fileProgress (int perc);

        void pong (QHostAddress user_ip, qint32 time);

        void invalidNick ();


protected:

        struct SendFileTime
        {
                QString name;           
                QTime           time;           
        };

        struct RecvFileSize
        {
                QString name;           
                quint32 size;           
        };

        struct QueueItem
        {
                LanPacket packet;                               
                QTime                   last_send;              
                QString         md5;                                    
                QHash<QHostAddress, int> num_err;               
        };

        QString user_nick;                      
        QHostAddress broadcast; 
        QList<QHostAddress>     ip_list;        
        QUdpSocket      read_socket;                    
        bool                    is_connect;                                     
        bool                    connecting;                                     
        QHash<QHostAddress, QTime>      ping_list;              
        QHash<QHostAddress, SendFileTime> send_file_list;       
        QHash<QHostAddress, RecvFileSize> incoming_file;        
        QList<QueueItem> packet_queue;  
        QTimer          *packet_timer;                          
        QTimer          *ping_timer;                                    
        QTimer          *connect_timer;                         
        quint32         rand_num;                                                       
        QHostAddress my_ip;                                                     
        quint16         main_port;

        bool    is_transfer;                            
        FileTransfer*   file_transfer;  
        void connectOk (const QHostAddress &ip);

        void connectFail ();

        void checkConnection (const QString &place);

        void sendPacket (QHostAddress ip, LanPacket packet, bool control=false);

        void servicePacket (LanPacket& packet, QHostAddress ip);

        void removeUser (QHostAddress user_ip);

        void removeQueue (QHostAddress user_ip, LanPacket::PacketType pack_type, QString md5);




private slots:

        void readData ();

        void packetTimer ();

        void sysPingTimer ();

        void connectTimeOut ();

        void fileTransferEnd (bool ok);

        void fileTransferProgres (int perc);
};


#endif

---