mayhem-firmware/firmware/application/ui_jammer.cpp

/*
 * Copyright (C) 2015 Jared Boone, ShareBrained Technology, Inc.
 *
 * This file is part of PortaPack.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#include "ui_jammer.hpp"
#include "ui_receiver.hpp"

#include "ch.h"
#include "evtimer.h"

#include "ff.h"
#include "hackrf_gpio.hpp"
#include "portapack.hpp"
#include "radio.hpp"

#include "hackrf_hal.hpp"
#include "portapack_shared_memory.hpp"
#include "portapack_persistent_memory.hpp"

#include <cstring>
#include <stdio.h>

using namespace hackrf::one;

namespace ui {

void JammerView::focus() {
	options_preset.focus();
}

JammerView::~JammerView() {
	transmitter_model.disable();
}

void JammerView::paint(Painter& painter) {
	(void)painter;
}

void JammerView::updfreq(uint8_t id, rf::Frequency f) {
	char finalstr[25] = {0};
	uint8_t c;
	
	auto mhz = to_string_dec_int(f / 1000000, 3);
	auto hz100 = to_string_dec_int((f / 100) % 10000, 4, '0');

	strcat(finalstr, mhz.c_str());
	strcat(finalstr, ".");
	strcat(finalstr, hz100.c_str());
	strcat(finalstr, "M");

	while (strlen(finalstr) < 10) {
		strcat(finalstr, " ");
	}

	if (id == 0) {
		range1_min = f;
		this->button_setfreq1_min.set_text(finalstr);
	}
	if (id == 1) {
		range1_max = f;
		this->button_setfreq1_max.set_text(finalstr);
	}
	if (id == 2) {
		range2_min = f;
		this->button_setfreq2_min.set_text(finalstr);
	}
	if (id == 3) {
		range2_max = f;
		this->button_setfreq2_max.set_text(finalstr);
	}
	if (id == 4) {
		range3_min = f;
		this->button_setfreq3_min.set_text(finalstr);
	}
	if (id == 5) {
		range3_max = f;
		this->button_setfreq3_max.set_text(finalstr);
	}
	
	rf::Frequency center;
	std::string bw;
	
	for (c = 0; c < 3; c++) {
		if (c == 0)	{
			center = (range1_min + range1_max) / 2;
			range1_center = center;
		}
		if (c == 1)	{
			center = (range2_min + range2_max) / 2;
			range2_center = center;
		}
		if (c == 2)	{
			center = (range3_min + range3_max) / 2;
			range3_center = center;
		}
		
		if (c == 0)	{
			range1_width = abs(range1_max - range1_min) / 1000;
			bw = to_string_dec_int(range1_width, 5);
		}
		if (c == 1)	{
			range2_width = abs(range2_max - range2_min) / 1000;
			bw = to_string_dec_int(range2_width, 5);
		}
		if (c == 2)	{
			range3_width = abs(range3_max - range3_min) / 1000;
			bw = to_string_dec_int(range3_width, 5);
		}
		
		auto center_mhz = to_string_dec_int(center / 1000000, 4);
		auto center_hz100 = to_string_dec_int((center / 100) % 10000, 4, '0');

		strcpy(finalstr, "C:");
		strcat(finalstr, center_mhz.c_str());
		strcat(finalstr, ".");
		strcat(finalstr, center_hz100.c_str());
		strcat(finalstr, "M W:");
		strcat(finalstr, bw.c_str());
		strcat(finalstr, "kHz");
		
		while (strlen(finalstr) < 23) {
			strcat(finalstr, " ");
		}
		
		if (c == 0) this->text_info1.set(finalstr);
		if (c == 1) this->text_info2.set(finalstr);
		if (c == 2) this->text_info3.set(finalstr);
	}
}

JammerView::JammerView(
	NavigationView& nav,
	TransmitterModel& transmitter_model
) : transmitter_model(transmitter_model)
{
	
	static constexpr Style style_val {
		.font = font::fixed_8x16,
		.background = Color::green(),
		.foreground = Color::black(),
	};
	
	static constexpr Style style_cancel {
		.font = font::fixed_8x16,
		.background = Color::red(),
		.foreground = Color::black(),
	};
	
	static constexpr Style style_info {
		.font = font::fixed_8x16,
		.background = Color::black(),
		.foreground = Color::grey(),
	};
	
	transmitter_model.set_modulation(TX_JAMMER);
	
	add_children({ {
		&text_type,
		&options_modulation,
		&text_sweep,
		&options_sweep,
		&text_preset,
		&options_preset,
		&text_hop,
		&options_hop,
		&checkbox_range1,
		&checkbox_range2,
		&checkbox_range3,
		&button_setfreq1_min,
		&button_setfreq1_max,
		&text_info1,
		&button_setfreq2_min,
		&button_setfreq2_max,
		&text_info2,
		&button_setfreq3_min,
		&button_setfreq3_max,
		&text_info3,
		&button_transmit,
		&button_exit
	} });
	
	button_transmit.set_style(&style_val);
	text_info1.set_style(&style_info);
	text_info2.set_style(&style_info);
	text_info3.set_style(&style_info);
	
	options_preset.on_change = [this](size_t n, OptionsField::value_t v) {
		(void)n;
		uint8_t c;
		
		for (c = 0; c < 3; c++) {
			updfreq(c*2, range_presets[v][c].min);
			updfreq((c*2)+1, range_presets[v][c].max);
			if (c == 0) checkbox_range1.set_value(range_presets[v][c].active);
			if (c == 1) checkbox_range2.set_value(range_presets[v][c].active);
			if (c == 2) checkbox_range3.set_value(range_presets[v][c].active);
		}
	};
	
	button_setfreq1_min.on_select = [this,&nav](Button&){
		auto new_view = new FrequencyKeypadView { nav, range1_min };
		new_view->on_changed = [this](rf::Frequency f) {
			updfreq(0, f);
		};
		nav.push(new_view);
	};
	button_setfreq1_max.on_select = [this,&nav](Button&){
		auto new_view = new FrequencyKeypadView { nav, range1_max };
		new_view->on_changed = [this](rf::Frequency f) {
			updfreq(1, f);
		};
		nav.push(new_view);
	};
	
	button_transmit.on_select = [this,&transmitter_model](Button&) {
		uint8_t i = 0;
		rf::Frequency t, range_lower;
		auto& message_map = context().message_map();
		
		message_map.unregister_handler(Message::ID::Retune);
		
		message_map.register_handler(Message::ID::Retune,
			[this,&transmitter_model](Message* const p) {
				const auto message = static_cast<const RetuneMessage*>(p);
				if (message->freq > 0) {
					transmitter_model.set_tuning_frequency(message->freq);
				}
			}
		);
		
		for (i = 0; i < 16; i++) {
			shared_memory.jammer_ranges[i].active = false;
		}
		
		if (range1_min > range1_max) {
			t = range1_min;
			range1_min = range1_max;
			range1_max = t;
		}
		range_lower = range1_min;
		for (i = 0;;) {
			if (range1_max - range_lower > 1000000) {
				shared_memory.jammer_ranges[i].center = range_lower + (1000000/2);
				shared_memory.jammer_ranges[i].width = 1000000 / 10;
				shared_memory.jammer_ranges[i].active = true;
				shared_memory.jammer_ranges[i].duration = 2280000/10;
				range_lower += 1000000;
			} else {
				shared_memory.jammer_ranges[i].center = (range1_max + range_lower) / 2;
				shared_memory.jammer_ranges[i].width = (range1_max - range_lower) / 10;
				shared_memory.jammer_ranges[i].active = true;
				shared_memory.jammer_ranges[i].duration = 2280000/10;
				break;
			}
			i++;
		}
		
		transmitter_model.set_tuning_frequency(shared_memory.jammer_ranges[0].center);
		
		if (jamming == true) {
			jamming = false;
			button_transmit.set_style(&style_val);
			button_transmit.set_text("START");
			transmitter_model.disable();
		} else {
			jamming = true;
			button_transmit.set_style(&style_cancel);
			button_transmit.set_text("STOP");
			transmitter_model.enable();
		}
	};

	button_exit.on_select = [&nav](Button&){
		nav.pop();
	};
	
}

} /* namespace ui */