wifi_link_tool.h

/*
wifi link tool 配网库
by:发明控 
版本v1.1.9
测试环境 sdk版本：3.0.0 arduino版本1.8.16
项目地址：https://github.com/bilibilifmk/wifi_link_tool 
*/
#include <Arduino.h>
#include <ESP8266WiFi.h>
#include <ESP8266HTTPClient.h>
#include <ESP8266WebServer.h>
#include <EEPROM.h>
#include <FS.h>
#include <DNSServer.h>
#include <ESP8266mDNS.h>
#define fs_server true
#ifndef FS_CONFIG
#include <wifi_link_tool_user_config.h>
#define fs_server false
#endif
void  ICACHE_RAM_ATTR blink();
#define colony_password  "keaino:1"
//组网密钥 可修改为自己的集群密钥 使用PSK 请保证密钥大于8位小于32位
const char *AP_name = "wifi_link_tool";
//修改后即不支持微信配网
/////////////////////////////////////////////////////不建议修改部分//////////////////////////////////////////////////////////////////
#define  wifilinktoolversion  "1.1.8"
String Version ="1.0.0";
String Hostname = "ESP8266";
int Signal_filtering = -200;
const byte DNS_PORT = 53;
int WiFi_State;
#define WiFi_State_Addr  0
bool wxscan=true;
int rstb=0;
int stateled=2;
IPAddress apIP(6, 6, 6, 6);
DNSServer dnsServer;
ESP8266WebServer webServer(80);

void info(void);
void pant(void);
void torest(void);
int link(void);
void sendBuffer(void);
void sendHeader(int code, String type, size_t _size);
void  wifi_link_tool_hex(int code, String type, const char* adr, size_t len);

void wwwroot(void);
void wifiConfig(void);
String wifi_type(int typecode);
void wifiScan(void);
void opera(void);
String gethttp_API(String url,int port);
void load(void);

//输出信息
void info(){
Serial.println("");	
Serial.print("wifi_link_tool Version:");	
Serial.print(wifilinktoolversion);	
Serial.println("");	
Serial.print("Program Version:");	
Serial.print(Version);	
Serial.print("");	
}

//心跳服务
void pant() {
	MDNS.update();
	dnsServer.processNextRequest();
	webServer.handleClient();
}
void torest() {
	WiFi.disconnect(true);
	delay(100);
	EEPROM.write(WiFi_State_Addr, 0);
	EEPROM.commit();
	delay(300);
	Serial.print("重置成功！正在重启。");
	ESP.restart();
}
int link() {
	int a=0;
	if(WiFi_State==1) {
		a=1;
	}
	return a;
}
#define buf_s 6000
char data_buffer[buf_s];
int bufwz = 0;
void sendBuffer() {
	if(bufwz > 0) {
		webServer.sendContent_P(data_buffer, bufwz);
		bufwz = 0;
	}
}
void sendToBuffer(String str) {
	size_t len = str.length();
	if(bufwz + len > buf_s) {
		webServer.sendContent_P(data_buffer, bufwz);
		bufwz = 0;
	}
	memcpy(data_buffer + bufwz, str.c_str(), len);
	bufwz += len;
}
void sendHeader(int code, String type, size_t _size) {
	webServer.setContentLength(_size);
	webServer.send(code, type, "");
}
void  wifi_link_tool_hex(int code, String type, const char* adr, size_t len) {
	sendHeader(code,type,len);
	webServer.sendContent_P(adr,len);
	sendBuffer();
}
void wwwroot() {
	if(fs_server) {
		if (WiFi_State == 1) {
			File file = SPIFFS.open("/index.html", "r");
			webServer.streamFile(file, "text/html");
			file.close();
		} else if (WiFi_State == 0) {
			File file = SPIFFS.open("/config.html", "r");
			webServer.streamFile(file, "text/html");
			file.close();
		}
	} else {
		if (WiFi_State == 1) {
			File file = SPIFFS.open("/index.html", "r");
			webServer.streamFile(file, "text/html");
			file.close();
		} else if (WiFi_State == 0) {
			#ifndef FS_CONFIG
			   wifi_link_tool_hex(200, "text/html", wifi_config, sizeof(wifi_config));
			#endif
		}
	}
}
void wifiConfig() {
	if (webServer.hasArg("ssid") && webServer.hasArg("password") && WiFi_State == 0) {
		int ssid_len = webServer.arg("ssid").length();
		int password_len = webServer.arg("password").length();
		//if ((ssid_len > 0) && (ssid_len < 33) && (password_len > 7) && (password_len < 65)) {
		if ((ssid_len > 0) && (ssid_len < 33) && (password_len < 65)) {
			String ssid_str = webServer.arg("ssid");
			String password_str = webServer.arg("password");
			const char *ssid = ssid_str.c_str();
			const char *password = password_str.c_str();
			WiFi.mode(WIFI_AP_STA);
			WiFi.disconnect(true);
			Serial.print("SSID: ");
			Serial.println(ssid);
			Serial.print("Password: ");
			Serial.println(password);
			WiFi.begin(ssid, password);
			Serial.print("尝试连接");
			unsigned long millis_time = millis();
			while ((WiFi.status() != WL_CONNECTED) && (millis() - millis_time < 80000)) {
				delay(500);
				Serial.print(".");
			}
			if (WiFi.status() == WL_CONNECTED) {
				digitalWrite(stateled, HIGH);
				Serial.println("");
				Serial.println("连接成功->"+WiFi.SSID());
				Serial.print("IP 地址: ");
				Serial.println(WiFi.localIP());
				//Serial.print("http://");
				Serial.println(Hostname);
				// webServer.send(200, "text/plain", "1");
				IPAddress ips;
				ips = WiFi.localIP();
				webServer.send(200, "text/plain", String(ips[0])+"."+String(ips[1])+"."+String(ips[2])+"."+String(ips[3]));
				delay(300);
				WiFi_State = 1;
				EEPROM.write(WiFi_State_Addr, 1);
				EEPROM.commit();
				delay(50);
		/* 取消注释后 配网成功后直接重启
        WiFi.softAPdisconnect();
        delay(1000);
        ESP.restart();
        */
			} else {
				Serial.println("链接失败");
				webServer.send(200, "text/plain", "0");
			}
		} else {
			Serial.println("密码错误");
			webServer.send(200, "text/plain", "0");
		}
	} else {
		Serial.println("参数错误");
		webServer.send(200, "text/plain", "0");
	}
}
String wifi_type(int typecode) {
	if (typecode == ENC_TYPE_NONE) return "Open";
	if (typecode == ENC_TYPE_WEP) return "WEP ";
	if (typecode == ENC_TYPE_TKIP) return "WPA ";
	if (typecode == ENC_TYPE_CCMP) return "WPA2";
	if (typecode == ENC_TYPE_AUTO) return "WPA*";
}
void wifiScan() {
	WiFi.disconnect();
	String req_json = "";
	Serial.println("Scan WiFi");
	int n = WiFi.scanNetworks();
		int m = 0;
  if (n > 0) {
		req_json = "{\"req\":[";
		for (int i = 0; i < n; i++) {
			if ((int)WiFi.RSSI(i) >= Signal_filtering)
			     //  if (1) {
				m++;
				String a="{\"ssid\":\"" + (String)WiFi.SSID(i) + "\"," + "\"encryptionType\":\"" + wifi_type(WiFi.encryptionType(i)) + "\"," + "\"rssi\":" + (int)WiFi.RSSI(i) + "},";
				if(a.length()>15)
				        req_json += a;
			}
		}
		req_json.remove(req_json.length() - 1);
		req_json += "]}";
		webServer.send(200, "text/json;charset=UTF-8", req_json);
		Serial.print("Found ");
		Serial.print(m);
		Serial.print(" WiFi!  >");
		Serial.print(Signal_filtering);
		Serial.println("dB");
	}

void opera() {
	if(webServer.arg("opera") == "sb") {
		webServer.send(200, "text/plain", Hostname);
	}
	if(webServer.arg("opera") == "reboot") {
		IPAddress ips;
		ips = WiFi.localIP();
		webServer.send(200, "text/plain", String(ips[0])+"."+String(ips[1])+"."+String(ips[2])+"."+String(ips[3]));
		delay(1000);
		WiFi.softAPdisconnect();
		ESP.restart();
	}
	if(webServer.arg("opera") == "version"){
		webServer.send(200, "text/plain", Version);
	}
		#ifndef OFF_colony

		if(webServer.arg("opera") == "SSID")
		if(webServer.arg("colony_password") ==colony_password)
		webServer.send(200, "text/plain",WiFi.SSID().c_str());
    	
	    if(webServer.arg("opera") == "PSK")
		if(webServer.arg("colony_password") ==colony_password)
		 webServer.send(200, "text/plain",WiFi.psk().c_str()); 	 
		
		#endif
	
	
}
// http get 请求函数 
String gethttp_API(String url){
  String payload="";
  WiFiClient client;
  HTTPClient http;
  if (http.begin(client,url)) { 
    int httpCode = http.GET();
    payload = http.getString();
    http.end();
 }
return payload;
}
//兼容以前httpget 
String gethttp_API(String url,int port){
return gethttp_API(url);
}
// ota升级服务函数 
void ota(){
	  webServer.on("/upota", HTTP_POST, []() {
      webServer.sendHeader("Connection", "close");
      webServer.send(200, "text/plain", (Update.hasError()) ? "FAIL" : "OK");
      ESP.restart();
    }, []() {
      HTTPUpload& upload =  webServer.upload();
      if (upload.status == UPLOAD_FILE_START) {
        Serial.setDebugOutput(true);
        WiFiUDP::stopAll();
        Serial.printf("升级文件: %s\n", upload.filename.c_str());
        uint32_t maxSketchSpace = (ESP.getFreeSketchSpace() - 0x1000) & 0xFFFFF000;
        if (!Update.begin(maxSketchSpace)) {
          Update.printError(Serial);
        }
      } else if (upload.status == UPLOAD_FILE_WRITE) {
        if (Update.write(upload.buf, upload.currentSize) != upload.currentSize) {
          Update.printError(Serial);
        }
      } else if (upload.status == UPLOAD_FILE_END) {
        if (Update.end(true)) { 
          Serial.printf("更新成功: %u\n等待重启...\n", upload.totalSize);
        } else {
          Update.printError(Serial);
        }
        Serial.setDebugOutput(false);
      }
      yield();
    });
}
void blink() {
	Serial.println("长按3秒后重置");
	bool res_state = true;
	unsigned int res_time = millis();
	while (millis() - res_time < 3000) {
		ESP.wdtFeed();
		//喂狗
		if (digitalRead(rstb) != LOW) {
			res_state = false;
			Serial.println("终止重置");
			break;
		}
	}
	if (res_state == true) {
		EEPROM.write(WiFi_State_Addr, 0);
		EEPROM.commit();
		if (WiFi.status() == WL_CONNECTED) {
			WiFi.disconnect(true);
		}
		delay(300);
		Serial.println("等待重启");
		for (int i=0;i<10;i++) {
			digitalWrite(stateled, LOW);
			res_time = millis();
			while (millis() - res_time < 200);
			digitalWrite(stateled, HIGH);
			res_time = millis();
			while (millis() - res_time < 200);
		}
		Serial.println("重置!");
		ESP.restart();
	}
}
//加载部分
void load() {
	//输出信息
	info();
	Serial.println("");
	if(fs_server) Serial.println("文件系统模式"); else Serial.println("二进制固化模式");
	WiFi.softAP("wif_link_tool_colony",colony_password, 3, 1);
    //启动加密网络 辅助集群系统
	WiFi.softAPConfig(apIP, apIP, IPAddress(255, 255, 255, 0));
	attachInterrupt(digitalPinToInterrupt(rstb), blink, FALLING);
	EEPROM.begin(4096);
	SPIFFS.begin();
	WiFi.hostname(Hostname);
	pinMode(stateled, OUTPUT);
	WiFi_State = EEPROM.read(WiFi_State_Addr);
	delay(300);
	if (WiFi_State == 1) {
		WiFi.mode(WIFI_STA);
		WiFi.begin();
		Serial.println("找到配置!");
		Serial.print("WiFi_link to " + WiFi.SSID());
		delay(500);
		unsigned millis_time = millis();
		while ((WiFi.status() != WL_CONNECTED) && (millis() - millis_time < 80000)) {
			delay(250);
			ESP.wdtFeed();
			//喂狗
			Serial.print("📡");
		}
		if (wxscan) {
			if (MDNS.begin(Hostname)) {
				Serial.println("");
				Serial.println("mDNS以启动");
			}
			MDNS.addService("http", "tcp", 80);
		}
		if (WiFi.status() == WL_CONNECTED) {
			Serial.print("IP 地址: ");
			Serial.println(WiFi.localIP());
			Serial.print("http://");
			Serial.println(Hostname);
			digitalWrite(stateled, HIGH);
			Serial.println("准备测试互联网通信！");
			String buf= gethttp_API("http://keai.icu/apiwyy/apitext",80);
			//String buf= gethttp_API("http://baidu.com",80);
			if(buf!=""){
				Serial.println("网络正常接口返回："+buf);
				Serial.println("向服务器发送一次请求");
				gethttp_API("http://keai.icu/wifilinktool/up",80);
				//向服务端发送一次get请求 服务端不会收集任何设备信息 只记录请求次数
			}
				
			else{
				Serial.println("请检查网络，或接口失效");
			}
				

		} else {
			Serial.println("连接失败!");
			Serial.println("SSID或密钥可能失效!");
			digitalWrite(stateled, LOW);
			delay(5000);
		}
	} else if (WiFi_State == 0) {
		digitalWrite(stateled, LOW);
		WiFi.disconnect(true);
		Serial.println("");
		Serial.print("启动WiFi配置 \n建立AP 名称 -->  ");
		Serial.println(AP_name);
		WiFi.mode(WIFI_AP_STA);
		WiFi.softAP(AP_name);

		#ifndef OFF_colony
        Serial.print("扫描网络环境尝试组网");
        WiFi.begin("wif_link_tool_colony",colony_password);
		unsigned millis_time = millis();
		while ((WiFi.status() != WL_CONNECTED) && (millis() - millis_time < 10000)) {
			delay(250);
			ESP.wdtFeed();
			Serial.print("-");
		}
		if(WiFi.status()==WL_CONNECTED)
		{
        Serial.println("发现集群 正在尝试加入集群");
        String getssid=gethttp_API("http://6.6.6.6/opera?opera=SSID&colony_password="+(String)colony_password,80);
		String getpsk=gethttp_API("http://6.6.6.6/opera?opera=PSK&colony_password="+(String)colony_password,80);
		String getsb=gethttp_API("http://6.6.6.6/opera?opera=sb",80);
		if(getssid!=""&& getpsk!=""){
        WiFi.disconnect(); 
		WiFi.begin(getssid, getpsk);
		millis_time = millis();  
        	while ((WiFi.status() != WL_CONNECTED) && (millis() - millis_time < 8000)) {
			delay(250);
			ESP.wdtFeed();
			Serial.print(".");
		}
		if(WiFi.status()==WL_CONNECTED)
		{
         Serial.println("网络信息贡献节点："+getsb);
		 Serial.println("SSID："+getssid+" password："+getpsk);
		 WiFi_State = 1;
		 EEPROM.write(WiFi_State_Addr, 1);
		 EEPROM.commit();
		 delay(50);
         Serial.println("等待重启"); 
		 delay(500);
		 ESP.restart();
		}else
		{
    	 Serial.println("组网失败 节点提供信息可能有误");
		 Serial.println("启动配网模式");
         WiFi.disconnect();
		}
		}else
		{
         Serial.println("组网失败 当前网络环境中可能不存在联网的节点");
		}

		}else
		{
        Serial.println("组网失败 当前网络环境中可能不存在当前密钥的集群");
		Serial.println("启动配网模式");
        WiFi.disconnect();
		}
	 #endif
	} else {
		Serial.println("初次启动");
		EEPROM.write(WiFi_State_Addr, 0);
		EEPROM.commit();
		delay(300);
		Serial.println("请重置系统！");
		while (1) {
			digitalWrite(stateled, LOW);
			delay(250);
			digitalWrite(stateled, HIGH);
			delay(250);
		}
	}
	Serial.println("正在启动http服务");
  if(WiFi_State == 0) {
    webServer.on("/", wwwroot);
    } else {
      if(fs_server)
   webServer.on("/", wwwroot);
  }
	webServer.on("/wificonfig", wifiConfig);
	webServer.on("/wifiscan", wifiScan);
	webServer.on("/opera", opera);
	//WiFi.softAPConfig(apIP, apIP, IPAddress(255, 255, 255, 0));
	dnsServer.start(DNS_PORT, "*", apIP);
	if (WiFi_State == 0) 
	webServer.onNotFound([]() {
		if(fs_server) {

			File file = SPIFFS.open("/config.html", "r");
			webServer.streamFile(file, "text/html");
			file.close();

		} else {
			#ifndef FS_CONFIG
			   wifi_link_tool_hex(200, "text/html", wifi_config, sizeof(wifi_config));
			#endif
		}
	}
	);
	webServer.begin();
	Serial.println("http服务启动完成");
	Serial.println("加载用户程序");
}