Restored led

Needed to fix case problems

Author: Spacehuhn

esp8266_deauther/led.cpp

@@ -0,0 +1,135 @@
+/*
+   Copyright (c) 2020 Stefan Kremser (@Spacehuhn)
+   This software is licensed under the MIT License. See the license file for details.
+   Source: github.com/spacehuhn/esp8266_deauther
+ */
+ 
+#include "led.h"
+
+#include "A_config.h" // Config for LEDs
+#include <Arduino.h> // digitalWrite, analogWrite, pinMode
+#include "language.h" // Strings used in printColor and tempDisable
+#include "settings.h" // used in update()
+#include "Attack.h" // used in update()
+#include "Scan.h" // used in update()
+
+// Inlcude libraries for Neopixel or LED_MY92xx if used
+#if defined(NEOPIXEL_LED) || defined(LED_NEOPIXEL_RGB) || defined(LED_NEOPIXEL_GRB)
+#include <Adafruit_NeoPixel.h>
+#elif defined(LED_MY92)
+#include <my92xx.h>
+#endif // if defined(NEOPIXEL_LED)
+
+extern Attack   attack;
+extern Scan     scan;
+
+namespace led {
+    // ===== PRIVATE ===== //
+    LED_MODE mode = OFF;
+
+    #if defined(LED_NEOPIXEL_RGB)
+        Adafruit_NeoPixel strip = Adafruit_NeoPixel(LED_NEOPIXEL_NUM, LED_NEOPIXEL_PIN, NEO_RGB + NEO_KHZ400);
+    #elif defined(LED_NEOPIXEL_GRB)
+        Adafruit_NeoPixel strip = Adafruit_NeoPixel(LED_NEOPIXEL_NUM, LED_NEOPIXEL_PIN, NEO_GRB + NEO_KHZ400);
+    #elif defined(LED_MY92)
+        my92xx myled = my92xx(LED_MY92_MODEL, LED_MY92_NUM, LED_MY92_DATA, LED_MY92_CLK, MY92XX_COMMAND_DEFAULT);
+    #endif // if defined(NEOPIXEL_LED)
+
+
+    void setColor(uint8_t r, uint8_t g, uint8_t b) {
+    #if defined(LED_DIGITAL)
+        if (LED_ANODE) {
+            if (LED_PIN_R < 255) digitalWrite(LED_PIN_R, r > 0);
+            if (LED_PIN_G < 255) digitalWrite(LED_PIN_G, g > 0);
+            if (LED_PIN_B < 255) digitalWrite(LED_PIN_B, b > 0);
+        } else {
+            if (LED_PIN_R < 255) digitalWrite(LED_PIN_R, r == 0);
+            if (LED_PIN_G < 255) digitalWrite(LED_PIN_G, g == 0);
+            if (LED_PIN_B < 255) digitalWrite(LED_PIN_B, b == 0);
+        }
+    #elif defined(LED_RGB)
+        if (r > 0) r = r * LED_MODE_BRIGHTNESS / 100;
+        if (g > 0) g = g * LED_MODE_BRIGHTNESS / 100;
+        if (b > 0) b = b * LED_MODE_BRIGHTNESS / 100;
+
+        if (LED_ANODE) {
+            r = 255 - r;
+            g = 255 - g;
+            b = 255 - b;
+        }
+
+        analogWrite(LED_PIN_R, r);
+        analogWrite(LED_PIN_G, g);
+        analogWrite(LED_PIN_B, b);
+    #elif defined(NEOPIXEL_LED)
+
+        for (size_t i = 0; i < LED_NEOPIXEL_NUM; i++) {
+            strip.setPixelColor(i, r, g, b);
+        }
+
+        strip.show();
+    #elif defined(LED_MY9291)
+        myled.setChannel(LED_MY92_CH_R, r);
+        myled.setChannel(LED_MY92_CH_G, g);
+        myled.setChannel(LED_MY92_CH_B, b);
+        myled.setChannel(LED_MY92_CH_BRIGHTNESS, LED_MODE_BRIGHTNESS);
+        myled.setState(true);
+        myled.update();
+    #endif // if defined(LED_DIGITAL)
+    }
+
+    // ===== PUBLIC ===== //
+    void setup() {
+        analogWriteRange(0xff);
+
+        #if defined(LED_DIGITAL) || defined(LED_RGB)
+            if (LED_PIN_R < 255) pinMode(LED_PIN_R, OUTPUT);
+            if (LED_PIN_G < 255) pinMode(LED_PIN_G, OUTPUT);
+            if (LED_PIN_B < 255) pinMode(LED_PIN_B, OUTPUT);
+        #elif defined(NEOPIXEL_LED)
+            strip.begin();
+            strip.setBrightness(LED_MODE_BRIGHTNESS);
+            strip.show();
+        #elif defined(LED_MY9291)
+            myled.setChannel(LED_MY92_CH_R, 0);
+            myled.setChannel(LED_MY92_CH_G, 0);
+            myled.setChannel(LED_MY92_CH_B, 0);
+            myled.setChannel(LED_MY92_CH_BRIGHTNESS, LED_MODE_BRIGHTNESS);
+            myled.setState(true);
+            myled.update();
+        #endif // if defined(LED_DIGITAL) || defined(LED_RGB)
+    }
+
+    void update() {
+        if (!settings::getLEDSettings().enabled) {
+            setMode(OFF);
+        } else if (scan.isScanning() && (scan.deauths < settings::getSnifferSettings().min_deauth_frames)) {
+            setMode(SCAN);
+        } else if (attack.isRunning()) {
+            setMode(ATTACK);
+        } else {
+            setMode(IDLE);
+        }
+    }
+
+    void setMode(LED_MODE new_mode, bool force) {
+        if ((new_mode != mode) || force) {
+            mode = new_mode;
+
+            switch (mode) {
+                case OFF:
+                    setColor(LED_MODE_OFF);
+                    break;
+                case SCAN:
+                    setColor(LED_MODE_SCAN);
+                    break;
+                case ATTACK:
+                    setColor(LED_MODE_ATTACK);
+                    break;
+                case IDLE:
+                    setColor(LED_MODE_IDLE);
+                    break;
+            }
+        }
+    }
+}

esp8266_deauther/led.h

@@ -0,0 +1,20 @@
+/*
+   Copyright (c) 2020 Stefan Kremser (@Spacehuhn)
+   This software is licensed under the MIT License. See the license file for details.
+   Source: github.com/spacehuhn/esp8266_deauther
+ */
+
+#pragma once
+
+enum LED_MODE {
+    OFF,
+    SCAN,
+    ATTACK,
+    IDLE
+};
+
+namespace led {
+        void setup();
+        void update();
+        void setMode(LED_MODE new_mode, bool force = false);
+}