功能描述
OLED显示SSID
全中文界面
支持中文SSID
Arduino OTA支持
硬件准备
ESP8266:NodeMCU WiFi
ESP32:WIFI+BLE+OLED开发板(OLED控制芯片:SSD1315兼容SSD1306)
OLED:1.3’ 控制芯片：SH1106
软件准备
Arduino IDE
U8g2lib
连线
按键:GPIO 0
板载LED:GPIO 16
GND - 地
VCC - 3.3v
OLED SCK - GPIO 14 （NodeMCU D5）
OLED MOSI - GPIO 13（NodeMCU D7）
OLED CS - GPIO 15 （NodeMCU D8）
OLED DC - GPIO 2 (NodeMCU D4)
程序
/*
WiFiScannerOLED.ino -- 一个简易ESP8266/ESP32网络扫描器.
扫描结果将会被显示在OLED12864上.
可以通过OTA进行升级
Written by CNflysky 20.3.10
last edited by CNflysky at 20.3.21*/
#define BOARD 32
/* 板卡选择,ESP32=32,ESP8266=8266 */
#if (BOARD == 32)
#include <WiFi.h>
#elif (BOARD == 8266)
#include <ESP8266WiFi.h>
#endif
#include <WiFiUdp.h>
#include <ArduinoOTA.h>
#include <U8g2lib.h>
#include <cstring>
#define LED 16 /* 板载LED选择 */
#define BUTTON 0 /* 板载按键选择，这里选择的是板上的Flash(Boot)按键作为扫描键 */
int ssids = 0;
bool networkfound = false;
const char *ssid = "TP-LINK_18A4"; /* 你的SSID */
const char *password = "jse105724482"; /* 你的密码 */
int state = 0; /* 0:正常运行 1:OTA升级模式 */
/* OLED构造函数，按需修改即可 */
U8G2_SSD1306_128X64_NONAME_F_SW_I2C u8g2( U8G2_R0, /* clock=*/ 4, /* data=*/ 5, /* reset=*/ U8X8_PIN_NONE );
typedef enum {
BUTTON_CLICK,
BUTTON_LONG_PRESS,
BUTTON_NOT_PRESSED
} buttonState;
struct wifis_t {
String ssid;
String encryption;
int rssi;
};
void setup() {
/* 初始化开发板 */
pinMode( LED, OUTPUT );
digitalWrite( LED, HIGH ); /* 本人开发板上的LED是低电平触发，所以将LED脚拉高，请按需选择 */
u8g2.setFont( u8g2_font_wqy12_t_gb2312 );
u8g2.begin();
u8g2.enableUTF8Print();
displayWelcome(); /* 显示欢迎语，可去除 */
WiFi.mode( WIFI_STA );
//配置OTA函数
ArduinoOTA.onStart( handleOTAStart );
ArduinoOTA.onEnd( handleOTAEnd );
ArduinoOTA.onProgress( handleOTAProgress );
ArduinoOTA.onError( handleOTAError );
delay( 100 );
}
void handleOTA() {
char * str = "升级模式";
char * str2 = "连接中...";
char * str3 = "未搜索到预保存SSID";
u8g2.clearBuffer();
u8g2.drawUTF8( u8g2.getDisplayWidth() / 2 - u8g2.getUTF8Width( str ) / 2, u8g2.getMaxCharHeight(), str );
if (networkfound == false) {
u8g2.sendBuffer();
int count = WiFi.scanNetworks();
for (int i = 0; i < count; ++i) {
if (!strcmp(ssid, WiFi.SSID(i).c_str())) {
networkfound = true;
}
}
}
if (networkfound == false) {
u8g2.drawUTF8( 0, u8g2.getMaxCharHeight() * 2, str3 );
u8g2.sendBuffer();
delay(3000);
} else {
if (WiFi.status() != WL_CONNECTED) {
u8g2.drawUTF8( u8g2.getDisplayWidth() / 2 - u8g2.getUTF8Width( str ) / 2, u8g2.getMaxCharHeight() * 2, str2 );
WiFi.begin(ssid, password);
delay(1000);
if (WiFi.status() != WL_CONNECTED)
WiFi.reconnect();
} else {
u8g2.setCursor( 0, u8g2.getMaxCharHeight() * 2 );
u8g2.print( WiFi.localIP() );
ArduinoOTA.begin();
ArduinoOTA.handle();
}
u8g2.sendBuffer();
}
}
void handleOTAStart() {
digitalWrite( LED, LOW );
}
void handleOTAEnd() {
digitalWrite( LED, HIGH );
char * str = "升级完成";
char * str2 = "三秒后自动重启";
u8g2.clearBuffer();
u8g2.drawUTF8( u8g2.getDisplayWidth() / 2 - u8g2.getUTF8Width( str ) / 2, u8g2.getMaxCharHeight(), str );
u8g2.drawUTF8( u8g2.getDisplayWidth() / 2 - u8g2.getUTF8Width( str2 ) / 2, u8g2.getMaxCharHeight() * 2, str2 );
u8g2.sendBuffer();
delay( 3000 );
ESP.restart();
}
void handleOTAError( ota_error_t error ) {
char * str = "升级失败";
u8g2.clearBuffer();
u8g2.drawUTF8( u8g2.getDisplayWidth() / 2 - u8g2.getUTF8Width( str ) / 2, u8g2.getDisplayHeight() / 2 + u8g2.getMaxCharHeight() / 2, str );
u8g2.sendBuffer();
}
void handleOTAProgress( unsigned int progress, unsigned int total ) {
u8g2.clearBuffer();
u8g2.setCursor( 0, u8g2.getMaxCharHeight() );
u8g2.print( "升级中:" );
u8g2.print( (progress / (total / 100) ) );
u8g2.print( "%" );
u8g2.sendBuffer();
}
struct wifis_t *getWiFisInfo( int count ) {
wifis_t *wifis = new wifis_t[count];
wifis_t switcher;
for ( int i = 0; i < count; ++i ) {
wifis[i].ssid = WiFi.SSID( i );
wifis[i].rssi = WiFi.RSSI( i );
/* 由于ESP8266和ESP32在加密方式的常量定义上有所不同，这里分为两部分 */
#if (BOARD == 32)
switch ( WiFi.encryptionType( i ) ) {
case WIFI_AUTH_OPEN:
wifis[i].encryption = "Open";
break;
case WIFI_AUTH_WEP:
wifis[i].encryption = "WEP";
break;
case WIFI_AUTH_WPA_PSK:
wifis[i].encryption = "WPA";
break;
case WIFI_AUTH_WPA2_PSK:
wifis[i].encryption = "WPA2";
break;
case WIFI_AUTH_WPA_WPA2_PSK:
wifis[i].encryption = "WPA/WPA2";
break;
case WIFI_AUTH_WPA2_ENTERPRISE:
wifis[i].encryption = "WPA2 EP";
break;
}
#elif (BOARD == 8266)
switch ( WiFi.encryptionType( i ) ) {
case ENC_TYPE_NONE:
wifis[i].encryption = "Open";
break;
case ENC_TYPE_WEP:
wifis[i].encryption = "WEP";
break;
case ENC_TYPE_TKIP:
wifis[i].encryption = "WPA";
break;
case ENC_TYPE_CCMP:
wifis[i].encryption = "WPA2";
break;
case ENC_TYPE_AUTO:
wifis[i].encryption = "WPA/WPA2";
break;
}
#endif
/* wifis[i].encryption=WiFi.encryptionType(i); */
}
/* 排序 */
for ( int i = 0; i < count; i++ ) {
for ( int j = 0; j < count; j++ ) {
if ( wifis[i].rssi * -1 < wifis[j].rssi * -1 ) {
switcher = wifis[i];
wifis[i] = wifis[j];
wifis[j] = switcher;
}
}
}
return (wifis);
}
void displayWelcome() {
char *str = "正在启动...";
u8g2.clearBuffer();
/* 将字符串显示在屏幕中央 */
u8g2.drawUTF8( u8g2.getDisplayWidth() / 2 - u8g2.getUTF8Width( str ) / 2, u8g2.getDisplayHeight() / 2 + u8g2.getMaxCharHeight() / 2, str );
u8g2.sendBuffer();
/* while(1); */
}
int getButtonState( int pin ) {
pinMode( pin, INPUT );
if ( digitalRead( pin ) == LOW ) {
/* 本人开发板上的按键按下为低电平，请根据你自己的情况修改 */
int pressTime = millis();
while ( digitalRead( pin ) == LOW );
int releaseTime = millis();
if ( releaseTime - pressTime <= 500 && releaseTime - pressTime >= 20 )
return (BUTTON_CLICK);
else if ( releaseTime - pressTime > 2000 ) {
return (BUTTON_LONG_PRESS);
}
} else return (BUTTON_NOT_PRESSED);
}
void displaySSIDs( int count, wifis_t *wifis ) {
int y = u8g2.getMaxCharHeight();
int offset = 11;
u8g2.drawUTF8( 0, y - 3, "网络名称" );
u8g2.drawUTF8( (u8g2.getDisplayWidth() - u8g2.getMaxCharWidth() * 3) - 10, y - 3, "强度" );
u8g2.drawUTF8( u8g2.getDisplayWidth() - u8g2.getMaxCharWidth() * 2 + 3, y - 3, "安全" );
for ( int i = 0; i < count; i++ ) {
if ( y < u8g2.getDisplayHeight() ) {
u8g2.setCursor( 0, y + offset );
u8g2.print( wifis[i].ssid );
/* Serial.println(wifis[i].ssid.isAlpha()); */
u8g2.print( " " );
u8g2.setCursor( (u8g2.getDisplayWidth() - u8g2.getMaxCharWidth() * 3) - 1, y + offset );
u8g2.print( wifis[i].rssi * -1 );
u8g2.print( " " );
u8g2.setCursor( u8g2.getDisplayWidth() - u8g2.getMaxCharWidth() * 2 + 3, y + offset );
/* 由于屏幕大小限制，没有显示全加密方式 */
if ( wifis[i].encryption != "Open" )
u8g2.print( "加密" );
else
u8g2.print( "开放" );
/* u8g2.println(wifis[i].encryption); */
y += u8g2.getMaxCharHeight();
}
}
}
void messageWhileNoSSID() {
char * str = "没有发现WiFi";
char * str2 = "按下按键开始扫描";
u8g2.clearBuffer();
u8g2.drawUTF8( u8g2.getDisplayWidth() / 2 - u8g2.getUTF8Width( str ) / 2, u8g2.getMaxCharHeight(), str );
u8g2.drawUTF8( u8g2.getDisplayWidth() / 2 - u8g2.getUTF8Width( str2 ) / 2, u8g2.getMaxCharHeight() * 2, str2 );
u8g2.sendBuffer();
}
void scanNetwork() {
digitalWrite( LED, LOW );
u8g2.clearBuffer();
char * str = "扫描中...";
u8g2.drawUTF8( u8g2.getDisplayWidth() / 2 - u8g2.getUTF8Width( str ) / 2, u8g2.getMaxCharHeight(), str );
u8g2.sendBuffer();
ssids = WiFi.scanNetworks();
digitalWrite( LED, HIGH );
}
void loop() {
int bstate = getButtonState( BUTTON );
if ( bstate == BUTTON_CLICK && state != 1 ) {
scanNetwork();
} else if ( bstate == BUTTON_LONG_PRESS && state != 1 ) {
state = 1;
} else {
if ( ssids > 0 && state == 0 ) {
wifis_t *wifis = getWiFisInfo( ssids );
u8g2.clearBuffer();
displaySSIDs( ssids, wifis );
u8g2.sendBuffer();
delete[] wifis;
} else if ( state == 1 ) {
handleOTA();
} else {
messageWhileNoSSID();
}
}
}
成果图
如何使用OTA升级功能
在代码中修改SSID和密码为你自己的WiFi名称和密码
在开机状态下，长按按钮三到五秒，松开后进入如下界面：
其中显示的IP地址 xxx.xxx.xxx.xxx就是ESP连接WiFi后得到的的IP地址。
打开Arduino IDE，检查工具/端口菜单，你应该会看到一个网络端口:
ESPxx-xxxxxxxxxxxx at xxx.xxx.xxx.xxx:
选择这个端口之后点击上传即可。