WiCom/WifiScanner.h at main · Thelmos/WiCom · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#pragma once

#include "Arduino.h"
#include <vector>
using std::vector;


class WifiScanner{
  public:
    WifiScanner();
    void start();
    uint8_t run();
    void setChannel(uint8_t channel);
    void stop();
    void startScan(bool async, bool show_hidden);
    void runScan();
    void startSniff();
    uint8_t runSniff();
    void stopSniff();
    void sniffer(uint8_t* buf, uint16_t len);
    uint8_t APCount = 0;
  private:
    void addStation(uint8_t* mac, int accesspointNum);
    int findAccesspoint(uint8_t* mac);
    bool macBroadcast(uint8_t* mac);
    bool macValid(uint8_t* mac);
    bool macMulticast(uint8_t* mac);

    uint8_t broadcast[6] = { 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF };

    struct Station {
        uint8_t   ap;
        uint8_t   ch;
        uint8_t * mac;
    };
    vector<Station*> stations;
    vector<uint16_t*> packets;
    uint16_t deauths = 0;
    uint16_t packetCount = 0;

    int STACount = 0;

};