|
| 1 | +#include "QSSLClient.h" |
| 2 | +#include <QSslSocket> |
| 3 | +#include <QSslCertificate> |
| 4 | +#include <QFile> |
| 5 | +#include <QSslKey> |
| 6 | +#include <QJsonDocument> |
| 7 | +#include <QList> |
| 8 | +#include <QDebug> |
| 9 | + |
| 10 | +QSSLClient::QSSLClient(QObject *parent) : QObject(parent) |
| 11 | +{ |
| 12 | + m_key = new QSslKey; |
| 13 | + m_privateCertificate = new QSslCertificate; |
| 14 | + m_client = new QSslSocket; |
| 15 | + loadCertificate(); |
| 16 | + |
| 17 | + connect(m_client, &QSslSocket::readyRead, this, &QSSLClient::rx); |
| 18 | + connect(m_client, &QSslSocket::disconnected, this, &QSSLClient::serverDisconnect); |
| 19 | + connect(m_client, SIGNAL(sslErrors(QList<QSslError>)), this, SLOT(sslErrors(QList<QSslError>))); |
| 20 | + |
| 21 | + m_client->addCaCertificates(m_publicCertificateList); |
| 22 | + m_client->setPrivateKey(*m_key); |
| 23 | + m_client->setLocalCertificate(*m_privateCertificate); |
| 24 | + m_client->setPeerVerifyMode(QSslSocket::VerifyPeer); |
| 25 | + |
| 26 | + qDebug() << "QSSLClient load over"; |
| 27 | +} |
| 28 | + |
| 29 | +QSSLClient::~QSSLClient() |
| 30 | +{ |
| 31 | + delete m_privateCertificate; |
| 32 | + delete m_client; |
| 33 | + delete m_key; |
| 34 | +} |
| 35 | + |
| 36 | +void QSSLClient::connectServer() |
| 37 | +{ |
| 38 | + m_client->connectToHostEncrypted("localhost", 19999); |
| 39 | + if(m_client->waitForEncrypted(5000)){ |
| 40 | + |
| 41 | + qDebug() << "Authentication Suceeded"; |
| 42 | + } |
| 43 | + else{ |
| 44 | + |
| 45 | + qDebug("Unable to connect to server"); |
| 46 | + exit(0); |
| 47 | + } |
| 48 | +} |
| 49 | + |
| 50 | +void QSSLClient::sendMsg(const QString &msg) |
| 51 | +{ |
| 52 | + m_client->write(msg.toUtf8()); |
| 53 | +} |
| 54 | + |
| 55 | +void QSSLClient::closeSocket() |
| 56 | +{ |
| 57 | + if(m_client->disconnect()){ |
| 58 | + |
| 59 | + m_client->close(); |
| 60 | + qDebug() << "close success"; |
| 61 | + } |
| 62 | +} |
| 63 | + |
| 64 | +void QSSLClient::loadCertificate() |
| 65 | +{ |
| 66 | + QFile p12File(":/res/p_client.p12"); |
| 67 | + if(!p12File.open(QIODevice::ReadOnly)){ |
| 68 | + |
| 69 | + qDebug() << "The certificate file open failed!"; |
| 70 | + exit(0); |
| 71 | + } |
| 72 | + |
| 73 | + bool ok = QSslCertificate::importPkcs12(&p12File, m_key, m_privateCertificate, &m_publicCertificateList, "cccccc"); |
| 74 | + if(!ok){ |
| 75 | + |
| 76 | + qDebug() << "The certificate import error!"; |
| 77 | + exit(0); |
| 78 | + } |
| 79 | + p12File.close(); |
| 80 | +} |
| 81 | + |
| 82 | +void QSSLClient::sslErrors(const QList<QSslError> &errors) |
| 83 | +{ |
| 84 | + foreach (const QSslError &error, errors) |
| 85 | + qDebug() << error.errorString(); |
| 86 | +} |
| 87 | + |
| 88 | +void QSSLClient::rx() |
| 89 | +{ |
| 90 | + QString getMsg = m_client->readAll(); |
| 91 | + qDebug() << getMsg; |
| 92 | +} |
| 93 | + |
| 94 | +void QSSLClient::serverDisconnect() |
| 95 | +{ |
| 96 | +// m_client->close(); |
| 97 | +// exit(0); |
| 98 | +} |
0 commit comments