panki/esp-scandetect
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gauge mode, fixes, idk its been to long

Browse Source
This commit is contained in:
Felix Pankratz 2023-12-24 14:26:57 +01:00
parent 77773661d5
commit 6ac9b9a198
2 changed files with 69 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
platformio.ini
View File

@ -7,4 +7,3 @@ build_flags =
'-DESP_IDF_LWIP_HOOK_FILENAME="lwip_hooks.h"' '-DESP_IDF_LWIP_HOOK_FILENAME="lwip_hooks.h"'
lib_deps = lib_deps =
adafruit/Adafruit NeoPixel adafruit/Adafruit NeoPixel

57
src/main.cpp
View File

@ -12,12 +12,28 @@
#define PROTO_TCP 6 #define PROTO_TCP 6
#define PROTO_UDP 17 #define PROTO_UDP 17
char ssid[] = "MyWifi"; // your network SSID (name) enum mode{Disco, Gauge};
enum mode active_mode = Disco;
char ssid[] = "MySSID"; // your network SSID (name)
char pass[] = "MyPassphrase"; // your network password (use for WPA, or use as key for WEP) char pass[] = "MyPassphrase"; // your network password (use for WPA, or use as key for WEP)
Adafruit_NeoPixel pixels(NUMPIXELS, PIN_NEOPIXEL, NEO_GRB + NEO_KHZ800); Adafruit_NeoPixel pixels(NUMPIXELS, PIN_NEOPIXEL, NEO_GRB + NEO_KHZ800);
Adafruit_NeoPixel onboard_pixel(1, PIN_NEOPIXEL_ONBOARD, NEO_GRB + NEO_KHZ800); Adafruit_NeoPixel onboard_pixel(1, PIN_NEOPIXEL_ONBOARD, NEO_GRB + NEO_KHZ800);
long last_gauge_show = 0;
int packets_received = 0;
uint32_t gauge_colors[] = {
pixels.Color(0, 255, 0), // red
pixels.Color(128, 255, 0),
pixels.Color(191, 255, 0),
pixels.Color(255, 255, 0), // yellow
pixels.Color(255, 191, 0),
pixels.Color(255, 128, 0),
pixels.Color(255, 64, 0),
pixels.Color(255, 0, 0) // red
};
void ssh_attempt() { void ssh_attempt() {
printf("Someone connected to SSH!"); printf("Someone connected to SSH!");
pixels.fill(pixels.Color(255, 0, 0)); pixels.fill(pixels.Color(255, 0, 0));
@ -34,15 +50,11 @@ void ssh_attempt() {
} }
void icmp_ping() { void icmp_ping() {
pixels.setPixelColor(2, pixels.Color(255, 255, 255));
pixels.setPixelColor(3, pixels.Color(255, 255, 255)); pixels.setPixelColor(3, pixels.Color(255, 255, 255));
pixels.setPixelColor(6, pixels.Color(255, 255, 255));
pixels.setPixelColor(7, pixels.Color(255, 255, 255)); pixels.setPixelColor(7, pixels.Color(255, 255, 255));
pixels.show(); pixels.show();
delay(100); delay(100);
pixels.setPixelColor(2, 0);
pixels.setPixelColor(3, 0); pixels.setPixelColor(3, 0);
pixels.setPixelColor(6, 0);
pixels.setPixelColor(7, 0); pixels.setPixelColor(7, 0);
pixels.show(); pixels.show();
} }
@ -50,6 +62,7 @@ void icmp_ping() {
extern "C" { extern "C" {
void log_packet(uint16_t port, u16_t proto) { void log_packet(uint16_t port, u16_t proto) {
if (WiFi.status() == WL_CONNECTED) { if (WiFi.status() == WL_CONNECTED) {
if (active_mode == Disco) {
uint32_t color = pixels.ColorHSV(port); uint32_t color = pixels.ColorHSV(port);
switch (proto) { switch (proto) {
case PROTO_ICMP: case PROTO_ICMP:
@ -61,15 +74,32 @@ extern "C" {
} else { } else {
pixels.setPixelColor(0, color); pixels.setPixelColor(0, color);
pixels.setPixelColor(1, color); pixels.setPixelColor(1, color);
pixels.setPixelColor(2, color);
pixels.show(); pixels.show();
} }
break; break;
case PROTO_UDP: case PROTO_UDP:
pixels.setPixelColor(4, color); pixels.setPixelColor(4, color);
pixels.setPixelColor(5, color); pixels.setPixelColor(5, color);
pixels.setPixelColor(6, color);
pixels.show(); pixels.show();
break; break;
} }
} else if (active_mode == Gauge) {
packets_received++;
if (millis() - last_gauge_show > 200) {
int num_leds = round(log((double)packets_received));
printf("gauge: Packets: %d LEDs: %d", packets_received, num_leds);
if (num_leds > NUMPIXELS) {num_leds = NUMPIXELS;}
pixels.clear();
for(int i = 0; i < num_leds; i++) {
pixels.setPixelColor(i, gauge_colors[i]);
}
pixels.show();
packets_received = 0;
last_gauge_show = millis();
}
}
} }
} }
} }
@ -100,12 +130,27 @@ extern "C" void app_main()
} }
onboard_pixel.fill(onboard_pixel.Color(0, 255, 0)); onboard_pixel.fill(onboard_pixel.Color(0, 255, 0));
onboard_pixel.show(); onboard_pixel.show();
printf("\nConnected!\n"); printf("\nConnected!\n");
printf("SSID: %s\n", WiFi.SSID().c_str()); printf("SSID: %s\n", WiFi.SSID().c_str());
printf("IP: %s\n", WiFi.localIP().toString().c_str()); printf("IP: %s\n", WiFi.localIP().toString().c_str());
long last_touch = 0;
// Arduino-like loop() // Arduino-like loop()
while(true){ while(true){
; int capacity = touchRead(8);
if (capacity > 20000) {
long cur_time = millis();
if (cur_time - last_touch > 500) {
active_mode = mode((active_mode + 1) % (Gauge + 1));
pixels.clear();
printf("changed mode: %u!\n", active_mode);
if (active_mode == Gauge) {
last_gauge_show = millis();
}
last_touch = cur_time;
}
}
} }
} }