devbox
小小林熬夜学编程
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

C++ Windows下连接WIFI_c++ 重连wifi

作者：小小林熬夜学编程 | 2024-03-16 22:11:17

赞

踩

c++ 重连wifi

  1. #pragma once
  2. #include <windows.h>
  3. #include <wlanapi.h>
  4. #pragma comment(lib, "wlanapi.lib")
  5.  
  6. //连接方式
  7. #define MODE_AUTO "auto"
  8. #define MODE_MANUAL "manual"
  9.  
  10. //认证方式
  11. #define AUTH_WPAPSK "WPAPSK"
  12. #define AUTH_WPA2PSK "WPA2PSK"
  13.  
  14. //加密方式
  15. #define ENCR_AES "AES"
  16. #define ENCR_TKIP "TKIP"
  17.  
  18. /*WIFI信息*/
  19. struct WifiInfo 
  20. {
  21. 	char account[64];
  22.  
  23. 	char password[64];
  24.  
  25. 	char mode[32];
  26.  
  27. 	char auth[32];
  28.  
  29. 	char encrypt[32];
  30. };
  31.  
  32. class WifiMgr
  33. {
  34. public:
  35. 	WifiMgr();
  36.  
  37. 	~WifiMgr();
  38.  
  39. 	bool connect(const char* account, const char* password, const char* mode = MODE_MANUAL,
  40. 		const char* auth = AUTH_WPAPSK, const char* encrypt = ENCR_AES);
  41.  
  42. 	bool connect(const WifiInfo& info);
  43.  
  44. 	bool reconnect();
  45.  
  46. 	bool disconnect(bool removeProFile = true);
  47.  
  48. 	bool connected() const;
  49.  
  50. 	const char* getLastError() const;
  51. protected:
  52. 	static void _stdcall wlanNotification(PWLAN_NOTIFICATION_DATA data, void* args);
  53. private:
  54. 	bool m_connected = false;
  55.  
  56. 	bool m_reconnect = false;
  57.  
  58. 	char m_lastError[256] = { "未知错误" };
  59.  
  60. 	HANDLE m_handle = INVALID_HANDLE_VALUE;
  61.  
  62. 	PWLAN_INTERFACE_INFO_LIST m_infoList = nullptr;
  63.  
  64. 	WifiInfo m_wifiInfo = { 0 };
  65. };
  1. #include "WifiMgr.h"
  2. #include <stdio.h>
  3. #include <string>
  4.  
  5. WifiMgr::WifiMgr()
  6. {
  7. }
  8.  
  9. WifiMgr::~WifiMgr()
  10. {
  11. 	if (m_connected)
  12. 	{
  13. 		disconnect();
  14. 	}
  15. }
  16.  
  17. bool WifiMgr::connect(const char* account, const char* password, const char* mode,
  18. 	const char* auth, const char* encrypt)
  19. {
  20. 	bool result = false;
  21. 	do
  22. 	{
  23. 		if (!m_reconnect)
  24. 		{
  25. 			strcpy(m_wifiInfo.account, account);
  26. 			strcpy(m_wifiInfo.password, password);
  27. 			strcpy(m_wifiInfo.mode, mode);
  28. 			strcpy(m_wifiInfo.auth, auth);
  29. 			strcpy(m_wifiInfo.encrypt, encrypt);
  30. 		}
  31.  
  32. 		unsigned long version = 0, value = 0;
  33. 		value = WlanOpenHandle(WLAN_API_VERSION, nullptr, &version, &m_handle);
  34. 		if (value != ERROR_SUCCESS)
  35. 		{
  36. 			sprintf(m_lastError, "打开WIFI句柄失败,错误代码:%lu", value);
  37. 			break;
  38. 		}
  39.  
  40. 		value = WlanEnumInterfaces(m_handle, nullptr, &m_infoList);
  41. 		if (value != ERROR_SUCCESS)
  42. 		{
  43. 			disconnect();
  44. 			sprintf(m_lastError, "枚举WIFI接口失败,错误代码:%lu", value);
  45. 			break;
  46. 		}
  47.  
  48. 		std::string character = std::string(password);
  49. 		for (std::string::size_type pos = 0; pos != std::string::npos; pos += strlen("&amp;"))
  50. 		{
  51. 			pos = character.find("&", pos);
  52. 			if (pos != std::string::npos)
  53. 				character.replace(pos, strlen("&"), "&amp;");
  54. 			else
  55. 				break;
  56. 		}
  57.  
  58. 		char xmlBuffer[4096] = { 0 };
  59. 		sprintf(xmlBuffer, "<?xml version =\"1.0\"?>\
  60.                              <WLANProfile xmlns =\"http://www.microsoft.com/networking/WLAN/profile/v1\">\
  61.                              <name>%s</name>\
  62.                              <SSIDConfig>\
  63.                              <SSID>\
  64.                              <name>%s</name>\
  65.                              </SSID>\
  66.                              </SSIDConfig>\
  67.                              <connectionType>ESS</connectionType>\
  68.                              <connectionMode>%s</connectionMode>\
  69.                              <autoSwitch>false</autoSwitch>\
  70.                              <MSM>\
  71.                              <security>\
  72.                              <authEncryption>\
  73.                              <authentication>%s</authentication>\
  74.                              <encryption>%s</encryption>\
  75.                              <useOneX>false</useOneX>\
  76.                              </authEncryption>\
  77.                              <sharedKey>\
  78.                              <keyType>passPhrase</keyType>\
  79.                              <protected>false</protected>\
  80.                              <keyMaterial>%s</keyMaterial>\
  81.                              </sharedKey>\
  82.                              </security>\
  83.                              </MSM>\
  84.                              </WLANProfile>",
  85. 			account, account, mode, auth, encrypt, character.c_str());
  86.  
  87. 		wchar_t fileData[4096] = { 0 };
  88. 		int length = strlen(xmlBuffer);
  89. 		for (int j = 0; j < length; j++)
  90. 		{
  91. 			mbtowc(&fileData[j], &xmlBuffer[j], 1);
  92. 		}
  93. 		fileData[length] = '\0';
  94.  
  95. 		WLAN_REASON_CODE reasonCode = 0;
  96. 		value = WlanSetProfile(m_handle, &m_infoList->InterfaceInfo[0].InterfaceGuid,
  97. 			0, fileData, nullptr, TRUE, nullptr, &reasonCode);
  98. 		if (value != ERROR_SUCCESS)
  99. 		{
  100. 			disconnect();
  101. 			sprintf(m_lastError, "设置WIFI配置文件失败,错误代码:%lu", value);
  102. 			break;
  103. 		}
  104.  
  105. 		wchar_t wideSsid[512] = { 0 };
  106. 		if (!MultiByteToWideChar(CP_ACP, 0, account, strlen(account) + 1,
  107. 			wideSsid, sizeof(wideSsid) / sizeof(wideSsid[0])))
  108. 		{
  109. 			disconnect();
  110. 			sprintf(m_lastError, "多字符转宽字符失败,错误代码:%lu", GetLastError());
  111. 			break;
  112. 		}
  113.  
  114. 		DOT11_SSID dot11Ssid = { 0 };
  115. 		memcpy(dot11Ssid.ucSSID, account, strlen(account) + 1);
  116. 		dot11Ssid.uSSIDLength = strlen(account);
  117.  
  118. 		WLAN_CONNECTION_PARAMETERS connectionParameters;
  119. 		memset(&connectionParameters, 0, sizeof(connectionParameters));
  120. 		connectionParameters.pDot11Ssid = &dot11Ssid;
  121. 		connectionParameters.wlanConnectionMode = wlan_connection_mode_profile;
  122. 		connectionParameters.strProfile = wideSsid;   /* clear 可以填nullptr */
  123. 		connectionParameters.dot11BssType = dot11_BSS_type_any;//dot11_BSS_type_independent;  /* msdn说可以dot11_BSS_type_any，实测不行 */
  124. 		connectionParameters.pDesiredBssidList = nullptr;
  125. 		connectionParameters.dwFlags = 0; /* WLAN_CONNECTION_ADHOC_JOIN_ONLY */
  126.  
  127. 		Sleep(300);
  128.  
  129. 		value = WlanConnect(m_handle, &m_infoList->InterfaceInfo[0].InterfaceGuid,
  130. 			&connectionParameters, nullptr);
  131. 		if (value != ERROR_SUCCESS)
  132. 		{
  133. 			disconnect();
  134. 			sprintf(m_lastError, "连接WIFI失败,错误代码:%lu", value);
  135. 			break;
  136. 		}
  137.  
  138. 		//value = WlanRegisterNotification(m_handle, WLAN_NOTIFICATION_SOURCE_ALL, false,
  139. 		//	wlanNotification, this, nullptr, nullptr);
  140. 		//if (value != ERROR_SUCCESS)
  141. 		//{
  142. 		//	sprintf(m_lastError, "注册通知失败,错误代码:%lu", value);
  143. 		//	break;
  144. 		//}
  145. 		m_connected = true;
  146. 		result = true;
  147. 	} while (false);
  148. 	return result;
  149. }
  150.  
  151. bool WifiMgr::connect(const WifiInfo& info)
  152. {
  153. 	return connect(info.account, info.password, info.mode, info.auth, info.encrypt);
  154. }
  155.  
  156. bool WifiMgr::reconnect()
  157. {
  158. 	bool result = false;
  159. 	m_reconnect = true;
  160. 	result = disconnect() && connect(m_wifiInfo);
  161. 	m_reconnect = false;
  162. 	return result;
  163. }
  164.  
  165. bool WifiMgr::disconnect(bool removeProFile)
  166. {
  167. 	bool result = true;
  168. 	do
  169. 	{
  170. 		unsigned long value = 0;
  171.  
  172. 		if (removeProFile)
  173. 		{
  174. 			wchar_t profile[256] = { 0 };
  175. 			if (MultiByteToWideChar(CP_ACP, 0, m_wifiInfo.account, strlen(m_wifiInfo.account) + 1,
  176. 				profile, sizeof(profile) / sizeof(profile[0])))
  177. 			{
  178. 				if (m_handle != INVALID_HANDLE_VALUE && m_infoList)
  179. 				{
  180. 					WlanDeleteProfile(m_handle, &(m_infoList->InterfaceInfo->InterfaceGuid), profile, nullptr);
  181. 				}
  182. 			}
  183. 		}
  184.  
  185. 		if (m_handle != INVALID_HANDLE_VALUE && m_infoList)
  186. 		{
  187. 			WlanDisconnect(m_handle, &(m_infoList->InterfaceInfo->InterfaceGuid), nullptr);
  188. 		}
  189.  
  190. 		if (m_handle != INVALID_HANDLE_VALUE)
  191. 		{
  192. 			WlanCloseHandle(m_handle, nullptr);
  193. 			m_handle = INVALID_HANDLE_VALUE;
  194. 		}
  195.  
  196. 		if (m_infoList)
  197. 		{
  198. 			WlanFreeMemory(m_infoList);
  199. 			m_infoList = nullptr;
  200. 		}
  201. 		m_connected = false;
  202. 	} while (false);
  203. 	return result;
  204. }
  205.  
  206. bool WifiMgr::connected() const
  207. {
  208. 	return m_connected;
  209. }
  210.  
  211. const char* WifiMgr::getLastError() const
  212. {
  213. 	return m_lastError;
  214. }
  215.  
  216. void WifiMgr::wlanNotification(PWLAN_NOTIFICATION_DATA data, void* args)
  217. {
  218. 	WifiMgr* wifi = static_cast<WifiMgr*>(args);
  219. 	if (!wifi)
  220. 		return;
  221.  
  222. 	unsigned long source = data->NotificationSource;
  223. 	unsigned long code = data->NotificationCode;
  224. 	unsigned long size = data->dwDataSize;
  225. 	const char* action = "unknown";
  226. 	if (source == WLAN_NOTIFICATION_SOURCE_ACM)
  227. 	{
  228. 		switch (source)
  229. 		{
  230. 		case wlan_notification_acm_start:
  231. 			action = "acm_start";
  232. 			break;
  233. 		case wlan_notification_acm_autoconf_enabled:
  234. 			action = "acm_autoconf_enabled";
  235. 			break;
  236. 		case wlan_notification_acm_autoconf_disabled:
  237. 			action = "acm_autoconf_disabled";
  238. 			break;
  239. 		case wlan_notification_acm_background_scan_enabled:
  240. 			action = "acm_background_scan_enabled";
  241. 			break;
  242. 		case wlan_notification_acm_background_scan_disabled:
  243. 			action = "acm_background_scan_disabled";
  244. 			break;
  245. 		case wlan_notification_acm_bss_type_change:
  246. 			action = "acm_bss_type_change";
  247. 			break;
  248. 		case wlan_notification_acm_power_setting_change:
  249. 			action = "acm_power_setting_change";
  250. 			break;
  251. 		case wlan_notification_acm_scan_complete:
  252. 			action = "acm_scan_complete";
  253. 			break;
  254. 		case wlan_notification_acm_scan_fail:
  255. 			action = "acm_scan_fail";
  256. 			break;
  257. 		case wlan_notification_acm_connection_start:
  258. 			action = "acm_connection_start";
  259. 			break;
  260. 		case wlan_notification_acm_connection_complete:
  261. 			action = "acm_connection_complete";
  262. 			break;
  263. 		case wlan_notification_acm_connection_attempt_fail:
  264. 			action = "acm_connection_attempt_fail";
  265. 			break;
  266. 		case wlan_notification_acm_filter_list_change:
  267. 			action = "acm_filter_list_change";
  268. 			break;
  269. 		case wlan_notification_acm_interface_arrival:
  270. 			action = "acm_interface_removal";
  271. 			break;
  272. 		case wlan_notification_acm_interface_removal:
  273. 			action = "acm_interface_removal";
  274. 			break;
  275. 		case wlan_notification_acm_profile_change:
  276. 			action = "acm_profile_change";
  277. 			break;
  278. 		case wlan_notification_acm_profile_name_change:
  279. 			action = "acm_profile_name_change";
  280. 			break;
  281. 		case wlan_notification_acm_profiles_exhausted:
  282. 			action = "acm_profiles_exhausted";
  283. 			break;
  284. 		case wlan_notification_acm_network_not_available:
  285. 			action = "acm_network_not_available";
  286. 			break;
  287. 		case wlan_notification_acm_network_available:
  288. 			action = "acm_network_available";
  289. 			break;
  290. 		case wlan_notification_acm_disconnecting:
  291. 			action = "acm_disconnecting";
  292. 			break;
  293. 		case wlan_notification_acm_disconnected:
  294. 			action = "acm_disconnected";
  295. 			break;
  296. 		case wlan_notification_acm_adhoc_network_state_change:
  297. 			action = "acm_adhoc_network_state_change";
  298. 			break;
  299. 		case wlan_notification_acm_end:
  300. 			action = "acm_end";
  301. 			break;
  302. 		default:break;
  303. 		}
  304. 	}
  305. 	else if (source == WLAN_NOTIFICATION_SOURCE_MSM)
  306. 	{
  307. 		switch (code)
  308. 		{
  309. 		case wlan_notification_msm_start:
  310. 			action = "msm_start";
  311. 			break;
  312. 		case wlan_notification_msm_associating:
  313. 		{
  314. 			auto ptr = ((PWLAN_MSM_NOTIFICATION_DATA)data->pData);
  315. 		}
  316. 		action = "msm_associating";
  317. 		break;
  318. 		case wlan_notification_msm_associated:
  319. 		{
  320. 			auto ptr = ((PWLAN_MSM_NOTIFICATION_DATA)data->pData);
  321. 		}
  322. 		action = "msm_associated";
  323. 		break;
  324. 		case wlan_notification_msm_authenticating:
  325. 		{
  326. 			auto ptr = ((PWLAN_MSM_NOTIFICATION_DATA)data->pData);
  327. 		}
  328. 		action = "msm_authenticating";
  329. 		break;
  330. 		case wlan_notification_msm_connected:
  331. 			action = "msm_connected";
  332. 			break;
  333. 		case wlan_notification_msm_roaming_start:
  334. 			action = "msm_roaming_start";
  335. 			break;
  336. 		case wlan_notification_msm_roaming_end:
  337. 			action = "msm_roaming_end";
  338. 			break;
  339. 		case wlan_notification_msm_radio_state_change:
  340. 			action = "msm_radio_state_change";
  341. 			break;
  342. 		case wlan_notification_msm_signal_quality_change:
  343. 			action = "msm_signal_quality_change";
  344. 			printf("信号质量:%lu\n", *(unsigned long*)data->pData);
  345. 			break;
  346. 		case wlan_notification_msm_disassociating:
  347. 			action = "msm_disassociating";
  348. 			break;
  349. 		case wlan_notification_msm_disconnected:
  350. 		{
  351. 			auto ptr = ((PWLAN_MSM_NOTIFICATION_DATA)data->pData);
  352. 		}
  353. 		action = "msm_disconnected";
  354. 		break;
  355. 		case wlan_notification_msm_peer_join:
  356. 			action = "msm_peer_join";
  357. 			break;
  358. 		case wlan_notification_msm_peer_leave:
  359. 			action = "msm_peer_leave";
  360. 			break;
  361. 		case wlan_notification_msm_adapter_removal:
  362. 			action = "msm_adapter_removal";
  363. 			break;
  364. 		case wlan_notification_msm_adapter_operation_mode_change:
  365. 			action = "msm_adapter_operation_mode_change";
  366. 			break;
  367. 		case wlan_notification_msm_end:
  368. 			action = "msm_end";
  369. 			break;
  370. 		default:break;
  371. 		}
  372. 	}
  373. 	printf("source %d,code %d,size %d,action %s\n", source, code, size, action);
  374. 	return;
  375. }

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/252322
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号