DJMFix.cpp
author František Kučera <franta-hg@frantovo.cz>
Mon, 04 Jan 2021 00:15:56 +0100
branchv_0
changeset 10 4d95b089457d
parent 8 87dfa7c89294
child 12 15d87fdd6e6c
permissions -rw-r--r--
documentation
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
}