diff options
Diffstat (limited to 'src')
-rwxr-xr-x | src/parted-usb.sh | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/src/parted-usb.sh b/src/parted-usb.sh index 198323b..d6d0873 100755 --- a/src/parted-usb.sh +++ b/src/parted-usb.sh | |||
@@ -13,17 +13,22 @@ MiB() | |||
13 | KiB $(KiB "$1") | 13 | KiB $(KiB "$1") |
14 | } | 14 | } |
15 | 15 | ||
16 | get_rootfs_size() | 16 | get_size() |
17 | { | 17 | { |
18 | [ "$ROOTFS" ] || return | 18 | [ "$1" ] || return |
19 | if [ -b "$ROOTFS" ] | 19 | if [ -b "$1" ] |
20 | then | 20 | then |
21 | rootfs_size=$(blockdev --getsize64 "$ROOTFS") | 21 | blockdev --getsize64 "$1" |
22 | else | 22 | else |
23 | rootfs_size=$(stat -c%s "$ROOTFS") | 23 | stat -c'%s' "$1" |
24 | fi | 24 | fi |
25 | } | 25 | } |
26 | 26 | ||
27 | get_rootfs_size() | ||
28 | { | ||
29 | rootfs_size=$(get_size "$ROOTFS") | ||
30 | } | ||
31 | |||
27 | initialize_target() | 32 | initialize_target() |
28 | { | 33 | { |
29 | if [ -b "$target" ] | 34 | if [ -b "$target" ] |