Update subbrute

This commit is contained in:
Willy-JL
2023-07-14 01:17:57 +02:00
parent d8ccd03308
commit 27dca6920f
9 changed files with 159 additions and 42 deletions

View File

@@ -0,0 +1,64 @@
#include "radio_device_loader.h"
#include <applications/drivers/subghz/cc1101_ext/cc1101_ext_interconnect.h>
#include <lib/subghz/devices/cc1101_int/cc1101_int_interconnect.h>
static void radio_device_loader_power_on() {
uint8_t attempts = 0;
while(!furi_hal_power_is_otg_enabled() && attempts++ < 5) {
furi_hal_power_enable_otg();
//CC1101 power-up time
furi_delay_ms(10);
}
}
static void radio_device_loader_power_off() {
if(furi_hal_power_is_otg_enabled()) furi_hal_power_disable_otg();
}
bool radio_device_loader_is_connect_external(const char* name) {
bool is_connect = false;
bool is_otg_enabled = furi_hal_power_is_otg_enabled();
if(!is_otg_enabled) {
radio_device_loader_power_on();
}
const SubGhzDevice* device = subghz_devices_get_by_name(name);
if(device) {
is_connect = subghz_devices_is_connect(device);
}
if(!is_otg_enabled) {
radio_device_loader_power_off();
}
return is_connect;
}
const SubGhzDevice* radio_device_loader_set(
const SubGhzDevice* current_radio_device,
SubGhzRadioDeviceType radio_device_type) {
const SubGhzDevice* radio_device;
if(radio_device_type == SubGhzRadioDeviceTypeExternalCC1101 &&
radio_device_loader_is_connect_external(SUBGHZ_DEVICE_CC1101_EXT_NAME)) {
radio_device_loader_power_on();
radio_device = subghz_devices_get_by_name(SUBGHZ_DEVICE_CC1101_EXT_NAME);
subghz_devices_begin(radio_device);
} else if(current_radio_device == NULL) {
radio_device = subghz_devices_get_by_name(SUBGHZ_DEVICE_CC1101_INT_NAME);
} else {
radio_device_loader_end(current_radio_device);
radio_device = subghz_devices_get_by_name(SUBGHZ_DEVICE_CC1101_INT_NAME);
}
return radio_device;
}
void radio_device_loader_end(const SubGhzDevice* radio_device) {
furi_assert(radio_device);
radio_device_loader_power_off();
if(radio_device != subghz_devices_get_by_name(SUBGHZ_DEVICE_CC1101_INT_NAME)) {
subghz_devices_end(radio_device);
}
}

View File

@@ -0,0 +1,15 @@
#pragma once
#include <lib/subghz/devices/devices.h>
/** SubGhzRadioDeviceType */
typedef enum {
SubGhzRadioDeviceTypeInternal,
SubGhzRadioDeviceTypeExternalCC1101,
} SubGhzRadioDeviceType;
const SubGhzDevice* radio_device_loader_set(
const SubGhzDevice* current_radio_device,
SubGhzRadioDeviceType radio_device_type);
void radio_device_loader_end(const SubGhzDevice* radio_device);

View File

@@ -9,7 +9,7 @@
#define SUBBRUTE_TX_TIMEOUT 6
#define SUBBRUTE_MANUAL_TRANSMIT_INTERVAL 250
SubBruteWorker* subbrute_worker_alloc() {
SubBruteWorker* subbrute_worker_alloc(const SubGhzDevice* radio_device) {
SubBruteWorker* instance = malloc(sizeof(SubBruteWorker));
instance->state = SubBruteWorkerStateIDLE;
@@ -37,6 +37,8 @@ SubBruteWorker* subbrute_worker_alloc() {
instance->transmit_mode = false;
instance->radio_device = radio_device;
return instance;
}
@@ -56,6 +58,9 @@ void subbrute_worker_free(SubBruteWorker* instance) {
furi_thread_free(instance->thread);
subghz_devices_sleep(instance->radio_device);
radio_device_loader_end(instance->radio_device);
free(instance);
}
@@ -206,9 +211,7 @@ void subbrute_worker_stop(SubBruteWorker* instance) {
instance->worker_running = false;
furi_thread_join(instance->thread);
furi_hal_subghz_set_path(FuriHalSubGhzPathIsolate);
furi_hal_subghz_idle();
furi_hal_subghz_sleep();
subghz_devices_idle(instance->radio_device);
}
bool subbrute_worker_transmit_current_key(SubBruteWorker* instance, uint64_t step) {
@@ -320,20 +323,24 @@ void subbrute_worker_subghz_transmit(SubBruteWorker* instance, FlipperFormat* fl
instance->transmitter =
subghz_transmitter_alloc_init(instance->environment, instance->protocol_name);
subghz_transmitter_deserialize(instance->transmitter, flipper_format);
furi_hal_subghz_reset();
furi_hal_subghz_idle();
furi_hal_subghz_load_preset(instance->preset);
furi_hal_subghz_set_frequency_and_path(instance->frequency);
furi_hal_subghz_start_async_tx(subghz_transmitter_yield, instance->transmitter);
while(!furi_hal_subghz_is_async_tx_complete()) {
furi_delay_ms(timeout);
subghz_devices_reset(instance->radio_device);
subghz_devices_idle(instance->radio_device);
subghz_devices_load_preset(instance->radio_device, instance->preset, NULL);
subghz_devices_set_frequency(
instance->radio_device, instance->frequency); // TODO is freq valid check
if(subghz_devices_set_tx(instance->radio_device)) {
subghz_devices_start_async_tx(
instance->radio_device, subghz_transmitter_yield, instance->transmitter);
while(!subghz_devices_is_async_complete_tx(instance->radio_device)) {
furi_delay_ms(timeout);
}
subghz_devices_stop_async_tx(instance->radio_device);
}
furi_hal_subghz_stop_async_tx();
//furi_hal_subghz_set_path(FuriHalSubGhzPathIsolate);
furi_hal_subghz_idle();
//furi_hal_subghz_sleep();
subghz_devices_idle(instance->radio_device);
subghz_transmitter_stop(instance->transmitter);
subghz_transmitter_free(instance->transmitter);
instance->transmitter = NULL;
@@ -456,4 +463,18 @@ void subbrute_worker_timeout_dec(SubBruteWorker* instance) {
if(instance->tx_timeout_ms > 0) {
instance->tx_timeout_ms--;
}
}
bool subbrute_worker_is_tx_allowed(SubBruteWorker* instance, uint32_t value) {
furi_assert(instance);
bool res = false;
if(!subghz_devices_is_frequency_valid(instance->radio_device, value)) {
} else {
subghz_devices_set_frequency(instance->radio_device, value);
res = subghz_devices_set_tx(instance->radio_device);
subghz_devices_idle(instance->radio_device);
}
return res;
}

View File

@@ -1,6 +1,7 @@
#pragma once
#include "../subbrute_protocols.h"
#include "radio_device_loader.h"
typedef enum {
SubBruteWorkerStateIDLE,
@@ -13,7 +14,7 @@ typedef void (*SubBruteWorkerCallback)(void* context, SubBruteWorkerState state)
typedef struct SubBruteWorker SubBruteWorker;
SubBruteWorker* subbrute_worker_alloc();
SubBruteWorker* subbrute_worker_alloc(const SubGhzDevice* radio_device);
void subbrute_worker_free(SubBruteWorker* instance);
uint64_t subbrute_worker_get_step(SubBruteWorker* instance);
bool subbrute_worker_set_step(SubBruteWorker* instance, uint64_t step);
@@ -46,3 +47,5 @@ uint8_t subbrute_worker_get_timeout(SubBruteWorker* instance);
void subbrute_worker_timeout_inc(SubBruteWorker* instance);
void subbrute_worker_timeout_dec(SubBruteWorker* instance);
bool subbrute_worker_is_tx_allowed(SubBruteWorker* instance, uint32_t value);

View File

@@ -22,6 +22,7 @@ struct SubBruteWorker {
SubGhzTransmitter* transmitter;
const char* protocol_name;
uint8_t tx_timeout_ms;
const SubGhzDevice* radio_device;
// Initiated values
SubBruteAttacks attack; // Attack state