From fd6561ffec220c214c87ee2b4089eddf364e78aa Mon Sep 17 00:00:00 2001
From: lisaac <lisaac.cn@gmail.com>
Date: Tue, 18 Feb 2020 17:08:22 +0800
Subject: [PATCH] diskaman: fix extended partations issues

---
 luasrc/model/diskman.lua | 21 ++++++++++++++-------
 po/zh-cn/diskman.po      |  6 +++---
 2 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/luasrc/model/diskman.lua b/luasrc/model/diskman.lua
index 609cc35..136b38b 100755
--- a/luasrc/model/diskman.lua
+++ b/luasrc/model/diskman.lua
@@ -234,14 +234,21 @@ local get_parted_info = function(device)
           table.insert(partitions_temp[disk_temp["extended_partition_index"]]["logicals"], i)
         end
       elseif (p["number"] < 4) and (p["number"] > 0) then
-        local real_size_sec = tonumber(nixio.fs.readfile("/sys/block/"..device.."/"..p["name"].."/size")) * tonumber(disk_temp.phy_sec)
-        if real_size_sec ~= p["size"] then
-          disk_temp["extended_partition_index"] = i
-          p["type"] = "extended"
-          p["size"] = real_size_sec
-          p["fs"] = "-"
-          p["logicals"] = {}
+        local s = nixio.fs.readfile("/sys/block/"..device.."/"..p["name"].."/size")
+        if s then
+          local real_size_sec = tonumber(s) * tonumber(disk_temp.phy_sec)
+          -- if size not equal, it's an extended
+          if real_size_sec ~= p["size"] then
+            disk_temp["extended_partition_index"] = i
+            p["type"] = "extended"
+            p["size"] = real_size_sec
+            p["fs"] = "-"
+            p["logicals"] = {}
+          else
+            p["type"] = "primary"
+          end
         else
+          -- if not found in "/sys/block"
           p["type"] = "primary"
         end
       end
diff --git a/po/zh-cn/diskman.po b/po/zh-cn/diskman.po
index b385e68..c79b6d3 100644
--- a/po/zh-cn/diskman.po
+++ b/po/zh-cn/diskman.po
@@ -5,7 +5,7 @@ msgid "DiskMan"
 msgstr "DiskMan 磁盘管理"
 
 msgid "Manage Disks over LuCI."
-msgstr "通过LuCI管理磁盘"
+msgstr "通过 LuCI 管理磁盘"
 
 msgid "Rescan Disks"
 msgstr "重新扫描磁盘"
@@ -26,7 +26,7 @@ msgid "Partition Table"
 msgstr "分区表"
 
 msgid "SATA Version"
-msgstr "SATA版本"
+msgstr "SATA 版本"
 
 msgid "Health"
 msgstr "健康"
@@ -125,7 +125,7 @@ msgid "Device Info"
 msgstr "设备信息"
 
 msgid "Disk Man"
-msgstr "DiskMan磁盘管理"
+msgstr "DiskMan 磁盘管理"
 
 msgid "Partitions Info"
 msgstr "分区信息"