Support MTU reconfigure for Windows.

This commit is contained in:
Adam Ierymenko 2017-05-04 19:50:02 -07:00
parent b55422a528
commit d5ed025871
2 changed files with 16 additions and 3 deletions

View File

@ -875,6 +875,14 @@ void WindowsEthernetTap::scanMulticastGroups(std::vector<MulticastGroup> &added,
void WindowsEthernetTap::setMtu(unsigned int mtu) void WindowsEthernetTap::setMtu(unsigned int mtu)
{ {
if (mtu != _mtu) { if (mtu != _mtu) {
_mtu = mtu;
HKEY nwAdapters;
if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Control\\Class\\{4D36E972-E325-11CE-BFC1-08002BE10318}", 0, KEY_READ | KEY_WRITE, &nwAdapters) == ERROR_SUCCESS) {
char tmps[64];
unsigned int tmpsl = Utils::snprintf(tmps, sizeof(tmps), "%d", mtu);
RegSetKeyValueA(nwAdapters, _mySubkeyName.c_str(), "MTU", REG_SZ, tmps, tmpsl);
RegCloseKey(nwAdapters);
}
} }
} }
@ -1019,10 +1027,15 @@ void WindowsEthernetTap::threadMain()
bool writeInProgress = false; bool writeInProgress = false;
ULONGLONG timeOfLastBorkCheck = GetTickCount64(); ULONGLONG timeOfLastBorkCheck = GetTickCount64();
_initialized = true; _initialized = true;
unsigned int oldmtu = _mtu;
while (_run) { while (_run) {
DWORD waitResult = WaitForMultipleObjectsEx(writeInProgress ? 3 : 2,wait4,FALSE,2500,TRUE); DWORD waitResult = WaitForMultipleObjectsEx(writeInProgress ? 3 : 2,wait4,FALSE,2500,TRUE);
if (!_run) break; // will also break outer while(_run) if (!_run) break; // will also break outer while(_run) since _run is false
// Check for changes in MTU and break to restart tap device to reconfigure in this case
if (_mtu != oldmtu)
break;
// Check for issues with adapter and close/reopen if any are detected. This // Check for issues with adapter and close/reopen if any are detected. This
// check fixes a while boatload of Windows adapter 'coma' issues after // check fixes a while boatload of Windows adapter 'coma' issues after

View File

@ -131,7 +131,7 @@ private:
void *_arg; void *_arg;
MAC _mac; MAC _mac;
uint64_t _nwid; uint64_t _nwid;
unsigned int _mtu; volatile unsigned int _mtu;
Thread _thread; Thread _thread;
volatile HANDLE _tap; volatile HANDLE _tap;