Initial commit: project docs and ignore rules
This commit is contained in:
Dr Jones
206
ESP32-C5-Toolkit/main/update_web_server.py
Normal file
206
ESP32-C5-Toolkit/main/update_web_server.py
Normal file
@@ -0,0 +1,206 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Update web_server.c with enhanced UI and add Bluetooth/advanced API endpoints
|
||||
"""
|
||||
|
||||
import re
|
||||
|
||||
# Read the enhanced UI C string
|
||||
with open('enhanced_ui_c_string.txt', 'r') as f:
|
||||
enhanced_html = f.read()
|
||||
|
||||
# Read current web_server.c
|
||||
with open('web_server.c', 'r') as f:
|
||||
web_server_content = f.read()
|
||||
|
||||
# Find the HTML section (from "static const char index_html[]" to "</html>";)
|
||||
pattern = r'(static const char index_html\[\] = ).*?("</html>";)'
|
||||
match = re.search(pattern, web_server_content, re.DOTALL)
|
||||
|
||||
if match:
|
||||
# Replace the HTML section
|
||||
# Extract the enhanced HTML (remove the "static const char index_html[] = " part from our file)
|
||||
enhanced_html_clean = enhanced_html.replace('static const char index_html[] = \\\n', '').rstrip(';')
|
||||
|
||||
# Replace in web_server.c
|
||||
new_content = web_server_content[:match.start()] + enhanced_html_clean + ';' + web_server_content[match.end():]
|
||||
|
||||
# Add Bluetooth API handlers before the register_uri_handlers function
|
||||
bt_handlers = '''
|
||||
// Bluetooth API handlers
|
||||
static esp_err_t api_bt_scan_start_handler(httpd_req_t *req) {
|
||||
httpd_resp_set_type(req, "application/json");
|
||||
bool success = bt_scan_start();
|
||||
if (success) {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"success\\",\\"message\\":\\"BT scan started\\"}");
|
||||
} else {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"error\\",\\"message\\":\\"Failed to start BT scan\\"}");
|
||||
}
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
static esp_err_t api_bt_scan_stop_handler(httpd_req_t *req) {
|
||||
httpd_resp_set_type(req, "application/json");
|
||||
bool success = bt_scan_stop();
|
||||
if (success) {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"success\\",\\"message\\":\\"BT scan stopped\\"}");
|
||||
} else {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"error\\",\\"message\\":\\"Failed to stop BT scan\\"}");
|
||||
}
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
static esp_err_t api_bt_devices_handler(httpd_req_t *req) {
|
||||
httpd_resp_set_type(req, "application/json");
|
||||
bt_device_t devices[50];
|
||||
int count = bt_get_devices(devices, 50);
|
||||
|
||||
cJSON *root = cJSON_CreateObject();
|
||||
cJSON *devices_array = cJSON_AddArrayToObject(root, "devices");
|
||||
|
||||
for (int i = 0; i < count; i++) {
|
||||
cJSON *device = cJSON_CreateObject();
|
||||
cJSON_AddStringToObject(device, "name", devices[i].name);
|
||||
char addr_str[18];
|
||||
snprintf(addr_str, sizeof(addr_str), "%02x:%02x:%02x:%02x:%02x:%02x",
|
||||
devices[i].addr[0], devices[i].addr[1], devices[i].addr[2],
|
||||
devices[i].addr[3], devices[i].addr[4], devices[i].addr[5]);
|
||||
cJSON_AddStringToObject(device, "addr", addr_str);
|
||||
cJSON_AddNumberToObject(device, "rssi", devices[i].rssi);
|
||||
cJSON_AddNumberToObject(device, "type", devices[i].adv_type);
|
||||
cJSON_AddItemToArray(devices_array, device);
|
||||
}
|
||||
|
||||
char *json_response = cJSON_PrintUnformatted(root);
|
||||
httpd_resp_sendstr(req, json_response);
|
||||
free(json_response);
|
||||
cJSON_Delete(root);
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
static esp_err_t api_bt_jam_start_handler(httpd_req_t *req) {
|
||||
httpd_resp_set_type(req, "application/json");
|
||||
char buf[256];
|
||||
int ret = httpd_req_recv(req, buf, sizeof(buf)-1);
|
||||
if (ret <= 0) {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"error\\",\\"message\\":\\"No data\\"}");
|
||||
return ESP_FAIL;
|
||||
}
|
||||
buf[ret] = 0;
|
||||
|
||||
cJSON *root = cJSON_Parse(buf);
|
||||
if (!root) {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"error\\",\\"message\\":\\"Bad JSON\\"}");
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
uint8_t *target_addr = NULL;
|
||||
uint32_t duration = 30;
|
||||
|
||||
cJSON *target = cJSON_GetObjectItem(root, "target");
|
||||
if (target && cJSON_IsString(target)) {
|
||||
// Parse target address
|
||||
target_addr = malloc(6);
|
||||
if (target_addr) {
|
||||
sscanf(target->valuestring, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx",
|
||||
&target_addr[0], &target_addr[1], &target_addr[2],
|
||||
&target_addr[3], &target_addr[4], &target_addr[5]);
|
||||
}
|
||||
}
|
||||
|
||||
cJSON *dur = cJSON_GetObjectItem(root, "duration");
|
||||
if (dur && cJSON_IsNumber(dur)) {
|
||||
duration = (uint32_t)dur->valueint;
|
||||
}
|
||||
|
||||
bool success = bt_jam_start(target_addr, duration);
|
||||
if (target_addr) free(target_addr);
|
||||
|
||||
cJSON_Delete(root);
|
||||
|
||||
if (success) {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"success\\",\\"message\\":\\"BT jamming started\\"}");
|
||||
} else {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"error\\",\\"message\\":\\"Failed to start jamming\\"}");
|
||||
}
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
static esp_err_t api_bt_jam_stop_handler(httpd_req_t *req) {
|
||||
httpd_resp_set_type(req, "application/json");
|
||||
bool success = bt_jam_stop();
|
||||
if (success) {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"success\\",\\"message\\":\\"BT jamming stopped\\"}");
|
||||
} else {
|
||||
httpd_resp_sendstr(req, "{\\"status\\":\\"error\\",\\"message\\":\\"Jamming not active\\"}");
|
||||
}
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
'''
|
||||
|
||||
# Insert BT handlers before register_uri_handlers
|
||||
register_pattern = r'(// Register URI handlers|esp_err_t register_uri_handlers)'
|
||||
register_match = re.search(register_pattern, new_content)
|
||||
if register_match:
|
||||
new_content = new_content[:register_match.start()] + bt_handlers + new_content[register_match.start():]
|
||||
|
||||
# Add BT URI registrations in register_uri_handlers function
|
||||
bt_registrations = '''
|
||||
// Register Bluetooth endpoints
|
||||
httpd_uri_t bt_scan_start_uri = {
|
||||
.uri = "/api/bt/scan/start",
|
||||
.method = HTTP_GET,
|
||||
.handler = api_bt_scan_start_handler,
|
||||
.user_ctx = NULL
|
||||
};
|
||||
httpd_register_uri_handler(server, &bt_scan_start_uri);
|
||||
|
||||
httpd_uri_t bt_scan_stop_uri = {
|
||||
.uri = "/api/bt/scan/stop",
|
||||
.method = HTTP_GET,
|
||||
.handler = api_bt_scan_stop_handler,
|
||||
.user_ctx = NULL
|
||||
};
|
||||
httpd_register_uri_handler(server, &bt_scan_stop_uri);
|
||||
|
||||
httpd_uri_t bt_devices_uri = {
|
||||
.uri = "/api/bt/devices",
|
||||
.method = HTTP_GET,
|
||||
.handler = api_bt_devices_handler,
|
||||
.user_ctx = NULL
|
||||
};
|
||||
httpd_register_uri_handler(server, &bt_devices_uri);
|
||||
|
||||
httpd_uri_t bt_jam_start_uri = {
|
||||
.uri = "/api/bt/jam/start",
|
||||
.method = HTTP_POST,
|
||||
.handler = api_bt_jam_start_handler,
|
||||
.user_ctx = NULL
|
||||
};
|
||||
httpd_register_uri_handler(server, &bt_jam_start_uri);
|
||||
|
||||
httpd_uri_t bt_jam_stop_uri = {
|
||||
.uri = "/api/bt/jam/stop",
|
||||
.method = HTTP_GET,
|
||||
.handler = api_bt_jam_stop_handler,
|
||||
.user_ctx = NULL
|
||||
};
|
||||
httpd_register_uri_handler(server, &bt_jam_stop_uri);
|
||||
|
||||
'''
|
||||
|
||||
# Find where to insert BT registrations (after deauth registrations)
|
||||
deauth_reg_pattern = r'(httpd_register_uri_handler\(server, &deauth_status_uri\);|// Register reboot endpoint)'
|
||||
deauth_reg_match = re.search(deauth_reg_pattern, new_content)
|
||||
if deauth_reg_match:
|
||||
new_content = new_content[:deauth_reg_match.start()] + bt_registrations + new_content[deauth_reg_match.start():]
|
||||
|
||||
# Write updated file
|
||||
with open('web_server.c', 'w') as f:
|
||||
f.write(new_content)
|
||||
|
||||
print("Updated web_server.c with enhanced UI and Bluetooth APIs")
|
||||
print("Added Bluetooth scan, devices, and jamming endpoints")
|
||||
else:
|
||||
print("ERROR: Could not find HTML section in web_server.c")
|
||||
Reference in New Issue
Block a user