Refactor

src/device.js

@@ -965,14 +965,10 @@ class Device extends DeviceBase {
 	async getProtectionState() {
 		if (this.isInDfuMode) {
 			let allSegmentsProtected = true;
-			const memMap = await this._dfu.getMemoryMap();
-			memMap.forEach(m => {
-				if (m.name === 'Internal Flash') {
-					m.segments.forEach(s => {
-						if (!(s.erasable === true && s.writable === false && s.readable === false)) {
-							allSegmentsProtected = false;
-						}
-					});
+			const memInfo = await this._dfu.getSegmentForInternalFlash();
+			memInfo.segments.forEach(s => {
+				if (!(s.erasable === true && s.writable === false && s.readable === false)) {
+					allSegmentsProtected = false;
 				}
 			});
 			// FIXME: Currently, device-os does not reliably distinguish the `overridden` value for different protection modes.

src/dfu.js

@@ -218,13 +218,9 @@ class Dfu {
 		await this._pollUntil((state) => state === DfuDeviceState.dfuMANIFEST);
 	}
 
-	async getMemoryMap() {
-		for (const i of this._allInterfaces) {
-			const ifaceName = await this._getStringDescriptor(i.iInterface);
-			const memInfo = this._parseMemoryDescriptor(ifaceName);
-			this._memoryMap.push(memInfo);
-		}
-		return this._memoryMap;
+	async getSegmentForInternalFlash() {
+		await this.setAltSetting(0);
+		return this._memoryInfo;
 	}
 
 	/**