sshkeygenerator.cpp 4.68 KB
Newer Older
ck's avatar
ck committed
1
2
3
4
/**************************************************************************
**
** This file is part of Qt Creator
**
con's avatar
con committed
5
** Copyright (c) 2011 Nokia Corporation and/or its subsidiary(-ies).
ck's avatar
ck committed
6
7
8
**
** Contact: Nokia Corporation (qt-info@nokia.com)
**
con's avatar
con committed
9
** No Commercial Usage
ck's avatar
ck committed
10
**
con's avatar
con committed
11
12
13
14
** This file contains pre-release code and may not be distributed.
** You may use this file in accordance with the terms and conditions
** contained in the Technology Preview License Agreement accompanying
** this package.
ck's avatar
ck committed
15
16
17
18
19
20
21
22
23
24
**
** GNU Lesser General Public License Usage
**
** Alternatively, this file may be used under the terms of the GNU Lesser
** General Public License version 2.1 as published by the Free Software
** Foundation and appearing in the file LICENSE.LGPL included in the
** packaging of this file.  Please review the following information to
** ensure the GNU Lesser General Public License version 2.1 requirements
** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
**
con's avatar
con committed
25
26
27
28
29
30
** In addition, as a special exception, Nokia gives you certain additional
** rights.  These rights are described in the Nokia Qt LGPL Exception
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
**
** If you have questions regarding the use of this file, please contact
** Nokia at qt-info@nokia.com.
ck's avatar
ck committed
31
32
33
**
**************************************************************************/

34
35
#include "sshkeygenerator.h"

ck's avatar
ck committed
36
37
38
#include "sshbotanconversions_p.h"
#include "sshcapabilities_p.h"
#include "sshpacket_p.h"
39

ck's avatar
ck committed
40
41
42
43
44
45
46
47
#include <botan/auto_rng.h>
#include <botan/bigint.h>
#include <botan/der_enc.h>
#include <botan/dsa.h>
#include <botan/pem.h>
#include <botan/pkcs8.h>
#include <botan/rsa.h>
#include <botan/x509_key.h>
48

ck's avatar
ck committed
49
#include <QtCore/QDateTime>
50

51
namespace Utils {
52

ck's avatar
ck committed
53
54
using namespace Botan;
using namespace Internal;
55

kh1's avatar
kh1 committed
56
57
58
59
60
SshKeyGenerator::SshKeyGenerator()
    : m_type(Rsa)
    , m_format(OpenSsl)
{
}
ck's avatar
ck committed
61
62
63

bool SshKeyGenerator::generateKeys(KeyType type, PrivateKeyFormat format,
    int keySize)
64
{
kh1's avatar
kh1 committed
65
66
67
    m_type = type;
    m_format = format;

ck's avatar
ck committed
68
69
70
    try {
        AutoSeeded_RNG rng;
        KeyPtr key;
kh1's avatar
kh1 committed
71
        if (m_type == Rsa)
ck's avatar
ck committed
72
73
            key = KeyPtr(new RSA_PrivateKey(rng, keySize));
        else
74
            key = KeyPtr(new DSA_PrivateKey(rng, DL_Group(rng, DL_Group::DSA_Kosherizer,
ck's avatar
ck committed
75
                keySize)));
kh1's avatar
kh1 committed
76
77
        return m_format == Pkcs8
            ? generatePkcs8Keys(key) : generateOpenSslKeys(key);
ck's avatar
ck committed
78
79
    } catch (Botan::Exception &e) {
        m_error = tr("Error generating key: %1").arg(e.what());
80
81
        return false;
    }
ck's avatar
ck committed
82
}
83

ck's avatar
ck committed
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
bool SshKeyGenerator::generatePkcs8Keys(const KeyPtr &key)
{
    generatePkcs8Key(key, false);
    generatePkcs8Key(key, true);
    return true;
}

void SshKeyGenerator::generatePkcs8Key(const KeyPtr &key, bool privateKey)
{
    Pipe pipe;
    pipe.start_msg();
    QByteArray *keyData;
    if (privateKey) {
        PKCS8::encode(*key, pipe);
        keyData = &m_privateKey;
    } else {
        X509::encode(*key, pipe);
        keyData = &m_publicKey;
102
    }
ck's avatar
ck committed
103
104
105
106
107
    pipe.end_msg();
    keyData->resize(pipe.remaining(pipe.message_count() - 1));
    pipe.read(convertByteArray(*keyData), keyData->size(),
        pipe.message_count() - 1);
}
108

kh1's avatar
kh1 committed
109
bool SshKeyGenerator::generateOpenSslKeys(const KeyPtr &key)
ck's avatar
ck committed
110
111
112
113
{
    QList<BigInt> publicParams;
    QList<BigInt> allParams;
    QByteArray keyId;
kh1's avatar
kh1 committed
114
    if (m_type == Rsa) {
ck's avatar
ck committed
115
116
117
118
119
120
121
122
123
124
125
126
127
        const QSharedPointer<RSA_PrivateKey> rsaKey
            = key.dynamicCast<RSA_PrivateKey>();
        publicParams << rsaKey->get_e() << rsaKey->get_n();
        allParams << rsaKey->get_n() << rsaKey->get_e() << rsaKey->get_d()
            << rsaKey->get_p() << rsaKey->get_q();
        keyId = SshCapabilities::PubKeyRsa;
    } else {
        const QSharedPointer<DSA_PrivateKey> dsaKey
            = key.dynamicCast<DSA_PrivateKey>();
        publicParams << dsaKey->group_p() << dsaKey->group_q()
            << dsaKey->group_g() << dsaKey->get_y();
        allParams << publicParams << dsaKey->get_x();
        keyId = SshCapabilities::PubKeyDss;
128
129
    }

ck's avatar
ck committed
130
131
132
133
134
135
136
137
138
139
140
141
142
143
    QByteArray publicKeyBlob = AbstractSshPacket::encodeString(keyId);
    foreach (const BigInt &b, publicParams)
        publicKeyBlob += AbstractSshPacket::encodeMpInt(b);
    publicKeyBlob = publicKeyBlob.toBase64();
    const QByteArray id = "QtCreator/"
        + QDateTime::currentDateTime().toString(Qt::ISODate).toUtf8();
    m_publicKey = keyId + ' ' + publicKeyBlob + ' ' + id;

    DER_Encoder encoder;
    encoder.start_cons(SEQUENCE).encode (0U);
    foreach (const BigInt &b, allParams)
        encoder.encode(b);
    encoder.end_cons();
    const char * const label
kh1's avatar
kh1 committed
144
        = m_type == Rsa ? "RSA PRIVATE KEY" : "DSA PRIVATE KEY";
ck's avatar
ck committed
145
146
    m_privateKey
        = QByteArray(PEM_Code::encode (encoder.get_contents(), label).c_str());
147
148
149
    return true;
}

150
} // namespace Utils