franta-hg@0
|
1 |
/**
|
franta-hg@0
|
2 |
* DJM-Fix
|
franta-hg@0
|
3 |
* Copyright © 2020 František Kučera (Frantovo.cz, GlobalCode.info)
|
franta-hg@0
|
4 |
*
|
franta-hg@0
|
5 |
* This program is free software: you can redistribute it and/or modify
|
franta-hg@0
|
6 |
* it under the terms of the GNU General Public License as published by
|
franta-hg@0
|
7 |
* the Free Software Foundation, version 3 of the License.
|
franta-hg@0
|
8 |
*
|
franta-hg@0
|
9 |
* This program is distributed in the hope that it will be useful,
|
franta-hg@0
|
10 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
franta-hg@0
|
11 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
franta-hg@0
|
12 |
* GNU General Public License for more details.
|
franta-hg@0
|
13 |
*
|
franta-hg@0
|
14 |
* You should have received a copy of the GNU General Public License
|
franta-hg@0
|
15 |
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
franta-hg@0
|
16 |
*/
|
franta-hg@1
|
17 |
#include <iostream>
|
franta-hg@5
|
18 |
#include <iomanip>
|
franta-hg@2
|
19 |
#include <thread>
|
franta-hg@6
|
20 |
#include <mutex>
|
franta-hg@5
|
21 |
#include <atomic>
|
franta-hg@2
|
22 |
#include <chrono>
|
franta-hg@2
|
23 |
#include <stdexcept>
|
franta-hg@6
|
24 |
#include <vector>
|
franta-hg@1
|
25 |
|
franta-hg@1
|
26 |
#include "DJMFix.h"
|
franta-hg@1
|
27 |
|
franta-hg@1
|
28 |
namespace djmfix {
|
franta-hg@1
|
29 |
|
franta-hg@6
|
30 |
using Bytes = std::vector<uint8_t>;
|
franta-hg@6
|
31 |
|
franta-hg@1
|
32 |
class DJMFixImpl : public DJMFix {
|
franta-hg@1
|
33 |
private:
|
franta-hg@2
|
34 |
MidiSender* midiSender;
|
franta-hg@2
|
35 |
std::thread keepAliveThread;
|
franta-hg@6
|
36 |
std::recursive_mutex midiMutex;
|
franta-hg@5
|
37 |
std::atomic<bool> running{false};
|
franta-hg@5
|
38 |
std::atomic<bool> stopped{false};
|
franta-hg@8
|
39 |
std::atomic<bool> sendKeepAlive{false};
|
franta-hg@8
|
40 |
Bytes seed2;
|
franta-hg@2
|
41 |
|
franta-hg@2
|
42 |
void run() {
|
franta-hg@2
|
43 |
while (!stopped) {
|
franta-hg@2
|
44 |
std::cerr << "DJMFixImpl::run()" << std::endl; // TODO: do not mess STDIO
|
franta-hg@8
|
45 |
if (sendKeepAlive) send({0xf0, 0x00, 0x40, 0x05, 0x00, 0x00, 0x00, 0x17, 0x00, 0x50, 0x01, 0xf7});
|
franta-hg@2
|
46 |
std::this_thread::sleep_for(std::chrono::milliseconds(200));
|
franta-hg@2
|
47 |
}
|
franta-hg@2
|
48 |
}
|
franta-hg@2
|
49 |
|
franta-hg@6
|
50 |
void send(const MidiMessage& midiMessage) {
|
franta-hg@6
|
51 |
std::lock_guard<std::recursive_mutex> lock(midiMutex);
|
franta-hg@6
|
52 |
midiSender->send(midiMessage);
|
franta-hg@6
|
53 |
}
|
franta-hg@6
|
54 |
|
franta-hg@6
|
55 |
std::string toString(const Bytes& midiMessage) {
|
franta-hg@5
|
56 |
std::stringstream result;
|
franta-hg@5
|
57 |
for (uint8_t b : midiMessage) result << std::hex << std::setw(2) << std::setfill('0') << (int) b;
|
franta-hg@5
|
58 |
return result.str();
|
franta-hg@5
|
59 |
}
|
franta-hg@5
|
60 |
|
franta-hg@6
|
61 |
Bytes normalize(const Bytes& data) {
|
franta-hg@6
|
62 |
if (data.size() % 2) throw std::invalid_argument("data before normalization must have even number of bytes");
|
franta-hg@6
|
63 |
Bytes result;
|
franta-hg@6
|
64 |
result.reserve(data.size() / 2);
|
franta-hg@6
|
65 |
for (size_t i = 0; i < data.size() / 2; i++) result.push_back((data[i * 2] & 0x0F) << 4 | (data[i * 2 + 1] & 0x0F));
|
franta-hg@6
|
66 |
return result;
|
franta-hg@6
|
67 |
}
|
franta-hg@6
|
68 |
|
franta-hg@8
|
69 |
Bytes denormalize(const Bytes& data) {
|
franta-hg@8
|
70 |
Bytes result;
|
franta-hg@8
|
71 |
result.reserve(data.size()*2);
|
franta-hg@8
|
72 |
for (size_t i = 0; i < data.size(); i++) {
|
franta-hg@8
|
73 |
result.push_back(data[i] >> 4);
|
franta-hg@8
|
74 |
result.push_back(data[i] & 0x0F);
|
franta-hg@8
|
75 |
}
|
franta-hg@8
|
76 |
return result;
|
franta-hg@8
|
77 |
}
|
franta-hg@8
|
78 |
|
franta-hg@6
|
79 |
uint32_t fnv32hash(const Bytes& buff) {
|
franta-hg@6
|
80 |
uint32_t hash = 0x811c9dc5;
|
franta-hg@6
|
81 |
for (uint8_t b : buff) hash = ((b^hash) * 0x1000193);
|
franta-hg@6
|
82 |
return hash;
|
franta-hg@6
|
83 |
}
|
franta-hg@6
|
84 |
|
franta-hg@6
|
85 |
Bytes toBytes(const uint32_t value) {
|
franta-hg@6
|
86 |
Bytes result;
|
franta-hg@6
|
87 |
result.reserve(4);
|
franta-hg@6
|
88 |
result.push_back(value >> 24);
|
franta-hg@6
|
89 |
result.push_back(value >> 16);
|
franta-hg@6
|
90 |
result.push_back(value >> 8);
|
franta-hg@6
|
91 |
result.push_back(value >> 0);
|
franta-hg@6
|
92 |
return result;
|
franta-hg@6
|
93 |
}
|
franta-hg@6
|
94 |
|
franta-hg@6
|
95 |
bool equals(Bytes a, Bytes b) {
|
franta-hg@6
|
96 |
if (a.size() != b.size()) return false;
|
franta-hg@6
|
97 |
for (size_t i = 0; i < a.size(); i++) if (a[i] != b[i]) return false;
|
franta-hg@6
|
98 |
return true;
|
franta-hg@6
|
99 |
}
|
franta-hg@6
|
100 |
|
franta-hg@6
|
101 |
template<typename T> std::vector<T> concat(const std::vector<T>& a, const std::vector<T>& b, const std::vector<T>& c = {}) {
|
franta-hg@6
|
102 |
std::vector<T> result;
|
franta-hg@6
|
103 |
result.reserve(a.size() + b.size() + c.size());
|
franta-hg@6
|
104 |
for (size_t i = 0; i < a.size(); i++) result.push_back(a[i]);
|
franta-hg@6
|
105 |
for (size_t i = 0; i < b.size(); i++) result.push_back(b[i]);
|
franta-hg@6
|
106 |
for (size_t i = 0; i < c.size(); i++) result.push_back(c[i]);
|
franta-hg@6
|
107 |
return result;
|
franta-hg@6
|
108 |
}
|
franta-hg@6
|
109 |
|
franta-hg@6
|
110 |
template<typename T> std::vector<T> xOR(const std::vector<T>& a, const std::vector<T>& b) {
|
franta-hg@6
|
111 |
if (a.size() != b.size()) throw std::invalid_argument("xor: both must be the same length");
|
franta-hg@6
|
112 |
std::vector<T> result;
|
franta-hg@6
|
113 |
result.reserve(a.size());
|
franta-hg@6
|
114 |
for (size_t i = 0; i < a.size(); i++) result.push_back(a[i] ^ b[i]);
|
franta-hg@6
|
115 |
return result;
|
franta-hg@6
|
116 |
}
|
franta-hg@6
|
117 |
|
franta-hg@1
|
118 |
public:
|
franta-hg@1
|
119 |
|
franta-hg@1
|
120 |
virtual ~DJMFixImpl() override {
|
franta-hg@1
|
121 |
std::cerr << "~DJMFixImpl()" << std::endl; // TODO: do not mess STDIO
|
franta-hg@2
|
122 |
if (running) stop();
|
franta-hg@2
|
123 |
}
|
franta-hg@2
|
124 |
|
franta-hg@2
|
125 |
void setMidiSender(MidiSender* midiSender) {
|
franta-hg@2
|
126 |
std::cerr << "DJMFixImpl::setMidiSender()" << std::endl; // TODO: do not mess STDIO
|
franta-hg@2
|
127 |
this->midiSender = midiSender;
|
franta-hg@1
|
128 |
}
|
franta-hg@1
|
129 |
|
franta-hg@6
|
130 |
virtual void receive(const MidiMessage& midiMessage) override {
|
franta-hg@5
|
131 |
std::cerr << "DJMFixImpl::receive(): size = " << midiMessage.size() << " data = " << toString(midiMessage) << std::endl; // TODO: do not mess STDIO
|
franta-hg@6
|
132 |
std::lock_guard<std::recursive_mutex> lock(midiMutex);
|
franta-hg@5
|
133 |
|
franta-hg@7
|
134 |
|
franta-hg@7
|
135 |
if (midiMessage.size() == 12 && midiMessage[9] == 0x11) {
|
franta-hg@7
|
136 |
send({0xf0, 0x00, 0x40, 0x05, 0x00, 0x00, 0x00, 0x17, 0x00, 0x12, 0x2a, 0x01, 0x0b, 0x50, 0x69, 0x6f, 0x6e, 0x65, 0x65, 0x72, 0x44, 0x4a, 0x02, 0x0b, 0x72, 0x65, 0x6b, 0x6f, 0x72, 0x64, 0x62, 0x6f, 0x78, 0x03, 0x12, 0x02, 0x09, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x03, 0x04, 0x08, 0x00, 0x00, 0x00, 0x00, 0xf7});
|
franta-hg@7
|
137 |
} else if (midiMessage.size() == 54 && midiMessage[9] == 0x13 && midiMessage[33] == 0x04 && midiMessage[43] == 0x03) {
|
franta-hg@6
|
138 |
Bytes hash1(midiMessage.begin() + 35, midiMessage.begin() + 35 + 8);
|
franta-hg@8
|
139 |
seed2 = Bytes(midiMessage.begin() + 45, midiMessage.begin() + 45 + 8);
|
franta-hg@6
|
140 |
hash1 = normalize(hash1);
|
franta-hg@6
|
141 |
seed2 = normalize(seed2);
|
franta-hg@6
|
142 |
std::cerr << "DJMFixImpl::receive(): got message with hash1 = " << toString(hash1) << " and seed2 = " << toString(seed2) << std::endl; // TODO: do not mess STDIO
|
franta-hg@6
|
143 |
|
franta-hg@6
|
144 |
Bytes seed0 = {0x68, 0x01, 0x31, 0xFB};
|
franta-hg@6
|
145 |
Bytes seed1 = {0x29, 0x00, 0x00, 0x00, 0x23, 0x48, 0x00, 0x00};
|
franta-hg@6
|
146 |
|
franta-hg@6
|
147 |
Bytes hash1check = toBytes(fnv32hash(concat(seed1, xOR(seed0, seed2))));
|
franta-hg@6
|
148 |
|
franta-hg@6
|
149 |
if (equals(hash1, hash1check)) {
|
franta-hg@8
|
150 |
std::cerr << "DJMFixImpl::receive(): hash1 verification: OK" << std::endl; // TODO: do not mess STDIO
|
franta-hg@8
|
151 |
Bytes hash2 = toBytes(fnv32hash(concat(seed2, xOR(seed0, seed2))));
|
franta-hg@8
|
152 |
send(concat({0xf0, 0x00, 0x40, 0x05, 0x00, 0x00, 0x00, 0x17, 0x00, 0x14, 0x38, 0x01, 0x0b, 0x50, 0x69, 0x6f, 0x6e, 0x65, 0x65, 0x72, 0x44, 0x4a, 0x02, 0x0b, 0x72, 0x65, 0x6b, 0x6f, 0x72, 0x64, 0x62, 0x6f, 0x78, 0x04, 0x0a}, concat(denormalize(hash2),{0x05, 0x16, 0x05, 0x09, 0x0b, 0x05, 0x04, 0x0b, 0x0f, 0x0e, 0x0e, 0x04, 0x04, 0x0a, 0x05, 0x0a, 0x0c, 0x08, 0x0e, 0x04, 0x0c, 0x05, 0xf7})));
|
franta-hg@6
|
153 |
} else {
|
franta-hg@8
|
154 |
std::cerr
|
franta-hg@8
|
155 |
<< "DJMFixImpl::receive(): hash1 verification failed: "
|
franta-hg@8
|
156 |
<< " midiMessage = " << toString(midiMessage)
|
franta-hg@8
|
157 |
<< " seed0 = " << toString(seed0)
|
franta-hg@8
|
158 |
<< " seed1 = " << toString(seed1)
|
franta-hg@8
|
159 |
<< " seed2 = " << toString(seed2)
|
franta-hg@8
|
160 |
<< " hash1 = " << toString(hash1)
|
franta-hg@8
|
161 |
<< " hash1check = " << toString(hash1check)
|
franta-hg@8
|
162 |
<< std::endl;
|
franta-hg@8
|
163 |
// TODO: graceful death
|
franta-hg@6
|
164 |
}
|
franta-hg@8
|
165 |
} else if (midiMessage.size() == 12 && midiMessage[9] == 0x15) {
|
franta-hg@8
|
166 |
sendKeepAlive = true;
|
franta-hg@5
|
167 |
}
|
franta-hg@5
|
168 |
|
franta-hg@1
|
169 |
}
|
franta-hg@1
|
170 |
|
franta-hg@1
|
171 |
void start() override {
|
franta-hg@1
|
172 |
std::cerr << "DJMFixImpl::start()" << std::endl; // TODO: do not mess STDIO
|
franta-hg@5
|
173 |
if (midiSender == nullptr) throw std::logic_error("need a midiSender when starting DJMFix");
|
franta-hg@5
|
174 |
|
franta-hg@5
|
175 |
// TODO: methods for parsing and constructing messages from parts (TLV)
|
franta-hg@6
|
176 |
send({0xf0, 0x00, 0x40, 0x05, 0x00, 0x00, 0x00, 0x17, 0x00, 0x50, 0x01, 0xf7});
|
franta-hg@2
|
177 |
|
franta-hg@2
|
178 |
keepAliveThread = std::thread(&DJMFixImpl::run, this);
|
franta-hg@2
|
179 |
running = true;
|
franta-hg@2
|
180 |
|
franta-hg@1
|
181 |
}
|
franta-hg@1
|
182 |
|
franta-hg@1
|
183 |
void stop() override {
|
franta-hg@2
|
184 |
stopped = true;
|
franta-hg@2
|
185 |
keepAliveThread.join();
|
franta-hg@2
|
186 |
running = false;
|
franta-hg@1
|
187 |
std::cerr << "DJMFixImpl::stop()" << std::endl; // TODO: do not mess STDIO
|
franta-hg@1
|
188 |
}
|
franta-hg@1
|
189 |
};
|
franta-hg@1
|
190 |
|
franta-hg@2
|
191 |
DJMFix* create() {
|
franta-hg@2
|
192 |
return new DJMFixImpl();
|
franta-hg@1
|
193 |
}
|
franta-hg@1
|
194 |
|
franta-hg@1
|
195 |
}
|