en_message/qt_keyfile.cpp at master · wevsty/en_message · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
#include "qt_keyfile.h"

//QString OpenKeyFileDialog(QWidget *parent = Q_NULLPTR)
QString OpenKeyFileDialog(QWidget *parent)
{
    //QFileDialog
    QFileDialog qfd(parent);
    //设置模式
    qfd.setAcceptMode(QFileDialog::AcceptOpen);
    //返回文件名，无论文件是否存在
    qfd.setFileMode(QFileDialog::AnyFile);
    //设置标题
    qfd.setWindowTitle(QObject::tr("Open Key File"));
    //设置起始目录
    qfd.setDirectory(".");
    //设置显示过滤列表
    QStringList name_filter_list;
    name_filter_list << QObject::tr("Key files (*.key *.bin)")
                     << QObject::tr("Hex files (*.hex)")
                     << QObject::tr("Base64 files (*.base64)")
                     << QObject::tr("Text files (*.txt)")
                     << QObject::tr("Any files (*)");
    //设置显示过滤
    qfd.setNameFilters(name_filter_list);
    //显示对话框
    qfd.exec();
    //获取选择路径
    QStringList qlist_filepath = qfd.selectedFiles();
    if(qlist_filepath.size() == 0)
    {
        return QString();
    }
    else
    {
        return qlist_filepath.first();
    }
}

//QString SaveKeyFileDialog(QWidget *parent = Q_NULLPTR)
QString SaveKeyFileDialog(QWidget *parent)
{
    QFileDialog qfd(parent);
    qfd.setAcceptMode(QFileDialog::AcceptSave);
    //返回文件名，无论文件是否存在
    qfd.setFileMode(QFileDialog::AnyFile);
    //设置标题
    qfd.setWindowTitle(QObject::tr("Save Key File"));
    //设置起始目录
    qfd.setDirectory(".");
    //设置显示过滤列表
    QStringList name_filter_list;
    name_filter_list << QObject::tr("Key files (*.key *.bin)")
                     << QObject::tr("Hex files (*.hex)")
                     << QObject::tr("Base64 files (*.base64)")
                     << QObject::tr("Text files (*.txt)")
                     << QObject::tr("Any files (*)");
    //设置显示过滤
    qfd.setNameFilters(name_filter_list);
    //设置默认扩展名
    qfd.setDefaultSuffix(".key");
    //显示对话框
    qfd.exec();
    //获取选择路径
    QStringList qlist_filepath = qfd.selectedFiles();
    if(qlist_filepath.size() == 0)
    {
        return QString();
    }
    else
    {
        return qlist_filepath.first();
    }
}

bool IsFilePathExtension(const QString &FilePath,const QString &FileExtension)
{
    QFileInfo finfo(FilePath);
    if(finfo.suffix() == FileExtension)
    {
        return true;
    }
    return false;
}

QByteArray ReadKeyFromFile(const QString &KeyFilePath,qint64 MaxReadSize)
{

    QByteArray q_bin_bytes;
    QFile key_file(KeyFilePath);
    if (!key_file.open(QIODevice::ReadOnly))
    {
        qDebug() << "Open file failed.";
        return q_bin_bytes;
    }
    if(MaxReadSize <= 0)
    {
        q_bin_bytes = key_file.readAll();
    }
    else
    {
        q_bin_bytes = key_file.read(MaxReadSize);
    }
    key_file.close();
    return q_bin_bytes;
}

QByteArray ReadBase64KeyFromFile(const QString &KeyFilePath,qint64 MaxReadSize)
{
    QByteArray base64_array = ReadKeyFromFile(KeyFilePath,MaxReadSize);
    QByteArray bin_array = QByteArray::fromBase64(base64_array);
    return bin_array;
}

QByteArray ReadHexKeyFromFile(const QString &KeyFilePath,qint64 MaxReadSize)
{
    QByteArray hex_array = ReadKeyFromFile(KeyFilePath,MaxReadSize);
    QByteArray bin_array = QByteArray::fromHex(hex_array);
    return bin_array;
}

bool SaveKeyToFile(const QString &KeyFilePath,const QByteArray &BinData)
{
    QFile key_file(KeyFilePath);
    if (!key_file.open(QFile::WriteOnly))
    {
        qDebug() << "Open file failed.";
        return false;
    }
    key_file.write(BinData);
    key_file.close();
    return true;
}

bool SaveKeyToBase64File(const QString &KeyFilePath,const QByteArray &BinData)
{
    QByteArray bin_array = BinData.toBase64();
    return SaveKeyToFile(KeyFilePath,bin_array);
}

bool SaveKeyToHexFile(const QString &KeyFilePath,const QByteArray &BinData)
{
    QByteArray bin_array = BinData.toHex();
    return SaveKeyToFile(KeyFilePath,bin_array);
}

bool SaveKeyToFileWithExtension(const QString &KeyFilePath,const QByteArray &BinData)
{
    if(IsFilePathExtension(KeyFilePath,"base64")==true)
    {
        return SaveKeyToBase64File(KeyFilePath,BinData);
    }
    else if(IsFilePathExtension(KeyFilePath,"hex")==true)
    {
        return SaveKeyToHexFile(KeyFilePath,BinData);
    }
    else
    {
        return SaveKeyToFile(KeyFilePath,BinData);
    }
}